From 3dd9664ff67c358ff2a67c7261089084c48af853 Mon Sep 17 00:00:00 2001
From: kradchen <chenjuijun@realcan.cn>
Date: Fri, 19 May 2023 15:56:31 +0800
Subject: [PATCH] Add logic to Detection

---
 test/Detection_Test.cpp | 35 +++++++++++++++++++++++++++++++++++
 1 file changed, 35 insertions(+)

diff --git a/test/Detection_Test.cpp b/test/Detection_Test.cpp
index 75722dc..3a362b8 100644
--- a/test/Detection_Test.cpp
+++ b/test/Detection_Test.cpp
@@ -4,6 +4,7 @@
 #include "Function1D.h"
 #include "MatlabReader.h"
 #include "Matrix.h"
+#include "config/config.h"
 #include "transmissionReconstruction/detection/detection.h"
 
 
@@ -16,6 +17,9 @@ inline double fourDecimalRound(double src){
 #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 Detection_Test : public ::testing::Test {
 protected:
     static void SetUpDetectionTester() {
@@ -81,5 +85,36 @@ TEST_F(Detection_Test, applyTimeWindowing) {
     {
         EXPECT_DOUBLE_AE(AscanBlockProcessed[i],result.AscanBlockProcessed[i])<<",index:"<<i;
     }
+    auto AscanBlockProcessed2 = m.read("AscanBlockProcessed2");
+    result = Recon::applyTimeWindowing(AscanBlock, 10000000, distBlock, sosBlock, 1482, 0, 0,  1400, 1650, true);
+    for (size_t i = 0; i < AscanBlockProcessed2.getDataSize(); i++)
+    {
+        EXPECT_DOUBLE_AE(AscanBlockProcessed2[i],result.AscanBlockProcessed[i])<<",index:"<<i;
+    }
  
+}
+
+TEST_F(Detection_Test, detectTofVectorized) {
+
+    MatlabReader m("/home/krad/TestData/getBlockOfTransmissionData.mat");
+
+    auto AscanBlock = m.read("AscanBlock");
+    auto AscanRefBlock = m.read("AscanRefBlock");
+    auto distBlock = m.read("dists");
+    auto distBlockRef = m.read("distRefBlock");
+    auto sosWaterBlock = m.read("sosWaterBlock");
+    auto sosWaterRefBlock = m.read("sosWaterRefBlock");
+    double expectedSOSWater = 1.511948131508464e+03;
+
+    auto result = Recon::detectTofVectorized(
+        AscanBlock, AscanRefBlock, distBlock, distBlockRef, sosWaterBlock,
+        sosWaterRefBlock, expectedSOSWater,
+        Recon::transParams::useTimeWindowing,
+        Recon::transParams::aScanReconstructionFrequency,
+        Recon::transParams::offsetElectronic,
+        Recon::transParams::detectionWindowSOS,
+        Recon::transParams::minSpeedOfSound,
+        Recon::transParams::maxSpeedOfSound, Recon::transParams::gaussWindow);
+
+    int size = result.getDataSize();
 }
\ No newline at end of file