--- /dev/null	Thu Jan 01 00:00:00 1970 +0000
+++ b/OrthancStone/Sources/Toolbox/AlignedMatrix.h	Wed May 17 17:30:52 2023 +0200
@@ -0,0 +1,135 @@
+/**
+ * Stone of Orthanc
+ * Copyright (C) 2012-2016 Sebastien Jodogne, Medical Physics
+ * Department, University Hospital of Liege, Belgium
+ * Copyright (C) 2017-2022 Osimis S.A., Belgium
+ * Copyright (C) 2021-2022 Sebastien Jodogne, ICTEAM UCLouvain, Belgium
+ *
+ * This program is free software: you can redistribute it and/or
+ * modify it under the terms of the GNU Lesser General Public License
+ * as published by the Free Software Foundation, either version 3 of
+ * the License, or (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful, but
+ * WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
+ * Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public
+ * License along with this program. If not, see
+ * <http://www.gnu.org/licenses/>.
+ **/
+
+
+#pragma once
+
+#include "SimdIncludes.h"
+
+#if (ORTHANC_HAS_AVX2 == 1 || ORTHANC_HAS_SSE2 == 1 || ORTHANC_HAS_WASM_SIMD == 1)
+#  define ORTHANC_HAS_MATRIX_PRODUCT_TRANSPOSED_VECTORIZED  1
+#else
+#  define ORTHANC_HAS_MATRIX_PRODUCT_TRANSPOSED_VECTORIZED  0
+#endif
+
+
+#include <boost/noncopyable.hpp>
+#include <cassert>
+
+namespace OrthancStone
+{
+  /**
+   * 2D matrix whose rows are aligned for the largest SIMD
+   * instructions that are available.
+   **/
+  class AlignedMatrix : public boost::noncopyable
+  {
+  private:
+    class ProductTransposedVectorizedContext;
+      
+    unsigned int  rows_;
+    unsigned int  cols_;
+    size_t        pitch_;
+    size_t        pitchFloatPointer_;
+    float*        content_;
+
+    void Setup(unsigned int rows,
+               unsigned int cols);
+
+  public:
+    AlignedMatrix(unsigned int rows,
+                  unsigned int cols)
+    {
+      Setup(rows, cols);
+    }
+
+    ~AlignedMatrix();
+
+    unsigned int GetRows() const
+    {
+      return rows_;
+    }
+
+    unsigned int GetColumns() const
+    {
+      return cols_;
+    }
+
+    unsigned int GetPitch() const
+    {
+      return pitch_;
+    }
+
+    float* GetRowPointer(unsigned int row)
+    {
+      assert(row < rows_);
+      return content_ + row * pitchFloatPointer_;
+    }
+
+    const float* GetRowPointer(unsigned int row) const
+    {
+      assert(row < rows_);
+      return content_ + row * pitchFloatPointer_;
+    }
+
+    size_t GetIndex(unsigned int row,
+                    unsigned int col) const
+    {
+      assert(row < rows_ && col < cols_);
+      return row * pitchFloatPointer_ + col;
+    }
+
+    float GetValue(unsigned int row,
+                   unsigned int col) const
+    {
+      return content_[GetIndex(row, col)];
+    }
+
+    void SetValue(unsigned int row,
+                  unsigned int col,
+                  float value) const
+    {
+      content_[GetIndex(row, col)] = value;
+    }
+
+    void AddValue(unsigned int row,
+                  unsigned int col,
+                  float value)
+    {
+      content_[GetIndex(row, col)] += value;
+    }
+
+    void FillZeros();
+
+    // Computes "C = A * B" without SIMD operations
+    static void ProductPlain(AlignedMatrix& c,
+                             const AlignedMatrix& a,
+                             const AlignedMatrix& b);
+
+#if ORTHANC_HAS_MATRIX_PRODUCT_TRANSPOSED_VECTORIZED == 1
+    // Computes "C = A * B^T" using SIMD operations
+    static void ProductTransposedVectorized(AlignedMatrix& c,
+                                            const AlignedMatrix& a,
+                                            const AlignedMatrix& bt);
+#endif
+  };
+}