From aff5106a376c274e0c5834468c29be56e5df17d4 Mon Sep 17 00:00:00 2001
From: kradchen <chenjuijun@realcan.cn>
Date: Wed, 10 May 2023 10:39:40 +0800
Subject: [PATCH] Add convertToLinearIndices to common

---
 src/common/common.cpp | 30 +++++++++++++++++++++++++++++-
 src/common/common.h   |  2 +-
 test/Common_Test.cpp  | 27 +++++++++++++++++++++++++++
 3 files changed, 57 insertions(+), 2 deletions(-)

diff --git a/src/common/common.cpp b/src/common/common.cpp
index 37a5cfb..67d45c5 100644
--- a/src/common/common.cpp
+++ b/src/common/common.cpp
@@ -52,4 +52,32 @@ namespace Recon
         // Data2 = ifft(Data2, [], 1);
         return Aurora::Matrix();
     }
-}
\ No newline at end of file
+
+    Aurora::Matrix convertToLinearIndices(Aurora::Matrix aVMatrixSize, Aurora::Matrix aMCoordinates)
+    {
+        if (aVMatrixSize.getDataSize() != aMCoordinates.getDimSize(1)){
+            std::cerr<<"convertToLinearIndices fail, Dimension of coordinates and matrix do not fit."<<std::endl;
+            return Aurora::Matrix();
+        }
+        size_t columns = aMCoordinates.getDimSize(1);
+        auto coord_col1= aMCoordinates(Aurora::$,0).toMatrix();
+        auto coord_col2Sub1= aMCoordinates(Aurora::$,1).toMatrix()-1;
+        auto matrixSize_1 = aVMatrixSize.getData()[0];
+        switch (columns) {
+            case 3:
+            {
+                auto matrixSize_2 = aVMatrixSize.getData()[1];
+                auto coord_col3Sub1= aMCoordinates(Aurora::$,2).toMatrix()-1;
+                return coord_col1 + coord_col2Sub1 * matrixSize_1 + coord_col3Sub1 * matrixSize_1 * matrixSize_2;
+            }
+            case 2:
+            {
+                return coord_col1 + coord_col2Sub1 * matrixSize_1;
+            }
+            default:
+                std::cerr<<"convertToLinearIndices fail,NotImplemented, Converting into linear indices only implemented for 2D and 3D matrices."<<std::endl;
+                return Aurora::Matrix();
+        }
+    }
+}
+
diff --git a/src/common/common.h b/src/common/common.h
index ddcbb4a..1a51ffa 100644
--- a/src/common/common.h
+++ b/src/common/common.h
@@ -12,6 +12,6 @@ namespace Recon {
      */
     Aurora::Matrix reconstructBandpasssubsampling(Aurora::Matrix aMatrix, double aAScanReconstructionFreq, double SampleRate);
 
-    
+    Aurora::Matrix convertToLinearIndices(Aurora::Matrix aVMatrixSize, Aurora::Matrix aMCoordinates);
 }
 #endif //RECON_COMMON_H
diff --git a/test/Common_Test.cpp b/test/Common_Test.cpp
index 75d2839..7ee6265 100644
--- a/test/Common_Test.cpp
+++ b/test/Common_Test.cpp
@@ -1,6 +1,7 @@
 #include <gtest/gtest.h>
 
 #include "common/ceMatchedFilterHandling.h"
+#include "common/common.h"
 #include "common/convertfp16tofloat.h"
 #include "MatlabReader.h"
 
@@ -55,4 +56,30 @@ TEST_F(Common_Test, convertfp16tofloat) {
         EXPECT_EQ(result[i], output.getData()[i])<<"index:"<<i<<",input:"<< ((short*)ma.getData())[i]<<",input2:"<<input.get()[i];
     }
 
+}
+
+TEST_F(Common_Test, convertToLinearIndices) {
+    //2D
+    {
+        double *dataA = new double[6]{1, 9, 4, 0, .01, 1.9};
+        auto A = Aurora::Matrix::fromRawData(dataA, 3, 2);
+        double *dataB = new double[2]{3, 2};
+        auto B = Aurora::Matrix::fromRawData(dataB, 2, 1);
+        auto output =Recon::convertToLinearIndices(B,A);
+        EXPECT_DOUBLE_AE(-2,output.getData()[0]);
+        EXPECT_DOUBLE_AE(6.03,output.getData()[1]);
+        EXPECT_DOUBLE_AE(6.7,output.getData()[2]);
+    }
+    //3D
+    {
+        double *dataA = new double[12]{1, 9, 4, 0, .01, 1.9, 1, 9, 4, 0, .01, 1.9};
+        auto A = Aurora::Matrix::fromRawData(dataA, 2, 3, 2);
+        double *dataB = new double[3]{3, 2, 2};
+        auto B = Aurora::Matrix::fromRawData(dataB, 3, 1);
+        auto output =Recon::convertToLinearIndices(B,A);
+        EXPECT_DOUBLE_AE(4.06,output.getData()[0]);
+        EXPECT_DOUBLE_AE(11.4,output.getData()[1]);
+    }
+
+
 }
\ No newline at end of file