Add SensitivityCalculation and its test

2023-05-12 14:13:56 +08:00
parent 2f98fc8819
commit 77e7d7be99
4 changed files with 397 additions and 3 deletions
--- a/test/Sensitivity_Test.cpp
+++ b/test/Sensitivity_Test.cpp
@@ -0,0 +1,126 @@
+#include <gtest/gtest.h>
+
+#include "Matrix.h"
+#include "transmissionReconstruction/dataFilter/sensitivityCalculations.h"
+#include "MatlabReader.h"
+
+inline double fourDecimalRound(double src){
+    return round(src*10000.0)/10000.0;
+}
+
+#define EXPECT_DOUBLE_AE(valueA,valueB)\
+    EXPECT_DOUBLE_EQ(fourDecimalRound(valueA),fourDecimalRound(valueB))
+
+#define ASSERT_DOUBLE_AE(valueA,valueB)\
+    ASSERT_DOUBLE_EQ(fourDecimalRound(valueA),fourDecimalRound(valueB))
+
+class Sensitivity_Test : public ::testing::Test {
+protected:
+    static void SetUpSensitivityTester() {
+
+    }
+
+    static void TearDownTestCase() {
+    }
+
+    void SetUp() {
+    }
+
+    void TearDown() {
+    }
+};
+
+TEST_F(Sensitivity_Test, getSensitivity) {
+    MatlabReader m("/home/krad/TestData/sensitivity.mat");
+
+    auto sensmap = m.read("sensmap");
+    auto senderNormal = m.read("senderNormal");
+    auto dirToReceiver = m.read("dirVector");
+
+    auto output = Recon::getSensitivity(sensmap, senderNormal, dirToReceiver);
+
+    auto sens = m.read("sens");
+    EXPECT_EQ(output.getDataSize(), sens.getDataSize());
+    for (size_t i = 0; i < output.getDataSize(); i++)
+    {
+        EXPECT_DOUBLE_AE(output.getData()[i],sens.getData()[i])<<", index:"<<i;
+    }
+    
+}
+
+TEST_F(Sensitivity_Test, precalcSensitivityForTAS){
+    MatlabReader m("/home/krad/TestData/sensTAS.mat");
+
+    auto sensmap = m.read("sensChar");
+    auto senderNormals = m.read("senderNormals");
+    auto senderPositions = m.read("senderPositions");
+    auto receiverPositions = m.read("receiverPositions");
+
+
+    auto output = Recon::precalcSensitivityForTAS(sensmap, senderPositions,senderNormals, receiverPositions);
+
+    auto sens = m.read("TASResult");
+    EXPECT_EQ(output.getDataSize(), sens.getDataSize());
+    for (size_t i = 0; i < output.getDataSize(); i++)
+    {
+        EXPECT_DOUBLE_AE(output.getData()[i],sens.getData()[i])<<", index:"<<i;
+    }
+}
+
+TEST_F(Sensitivity_Test, combineSensitivity){
+    MatlabReader m("/home/krad/TestData/combine.mat");
+
+    auto senderTASRange = m.read("senderTASRange");
+    auto senderElementRange = m.read("senderElementRange");
+    auto receiverTASRange = m.read("receiverTASRange");
+    auto receiverElementRange = m.read("receiverElementRange");
+    auto senderSens = m.read("TASResult");
+    auto receiverSens = m.read("TASResult2");
+
+
+    auto output = Recon::combineSensitivity(senderTASRange, senderElementRange, receiverTASRange, receiverElementRange, senderSens, receiverSens);
+
+    auto sens = m.read4d("combineResult");
+    EXPECT_EQ(output.size(), sens.size());
+    EXPECT_EQ(output[0].getDataSize(), sens[0].getDataSize());
+
+    for (size_t i = 0; i < output.size(); i++)
+    {
+        for (size_t j = 0; j < output[i].getDataSize(); j++)
+        {
+            EXPECT_DOUBLE_AE(output[i].getData()[j],sens[i].getData()[j])<<"Matrix index:"<<i<<", value index:"<<j;
+            
+        }
+    }
+}
+
+TEST_F(Sensitivity_Test,precalcSensitivity){
+    MatlabReader m("/home/krad/TestData/precalcSensitivity.mat");
+
+    auto senderTASRange = m.read("senderTASRange");
+    auto senderElementRange = m.read("senderElementRange");
+    auto receiverTASRange = m.read("receiverTASRange");
+    auto receiverElementRange = m.read("receiverElementRange");
+    MatlabReader m3("/home/krad/TestData/precalcSensitivity2.mat");
+    Recon::geo TASElements;
+    TASElements.senderPositions = m3.read4d("senderPositions");
+    TASElements.senderNormals = m3.read4d("senderNormals");
+    TASElements.receiverPositions = m3.read4d("receiverPositions");
+    TASElements.receiverNormals = m3.read4d("receiverNormals");
+    TASElements.sensChar = m.read("sensChar");
+    auto motorPos = Aurora::Matrix::fromRawData(new double[2]{1,2}, 2);
+    auto output = precalcSensitivity(senderTASRange, senderElementRange, receiverTASRange, receiverElementRange, motorPos, TASElements);
+    MatlabReader m2("/home/krad/TestData/precalcResult.mat");
+    auto sens = m2.read4d("result");
+    EXPECT_EQ(output.size(), sens.size());
+    EXPECT_EQ(output[0].getDataSize(), sens[0].getDataSize());
+
+    for (size_t i = 0; i < output.size(); i++)
+    {
+        for (size_t j = 0; j < output[i].getDataSize(); j++)
+        {
+            ASSERT_DOUBLE_AE(output[i].getData()[j],sens[i].getData()[j])<<"Matrix index:"<<i<<", value index:"<<j;
+            
+        }
+    }
+}