From e3ce295dda02408d73695d530c52b2a93d4d3ef7 Mon Sep 17 00:00:00 2001
From: kradchen <chenjuijun@realcan.cn>
Date: Wed, 31 May 2023 14:57:04 +0800
Subject: [PATCH] Add reconstructArt(unfinished)

---
 .../reconstruction/reconstruction.cpp         | 27 +++++++++++++++++++
 .../reconstruction/reconstruction.h           | 21 ++++++++++++---
 2 files changed, 44 insertions(+), 4 deletions(-)

diff --git a/src/transmissionReconstruction/reconstruction/reconstruction.cpp b/src/transmissionReconstruction/reconstruction/reconstruction.cpp
index 2aa1667..3b29813 100644
--- a/src/transmissionReconstruction/reconstruction/reconstruction.cpp
+++ b/src/transmissionReconstruction/reconstruction/reconstruction.cpp
@@ -4,6 +4,11 @@
 #include "Function2D.h"
 #include "Function3D.h"
 #include "Matrix.h"
+#include "config/config.h"
+
+#include "CudaEnvInit.h"
+
+#include <algorithm>
 #include <cmath>
 #include <iostream>
 using namespace Aurora;
@@ -103,4 +108,26 @@ namespace Recon {
         result.senderCoordList = aVSenderCoordList;
         return result;
     }
+    
+    ArtResult reconstructArt(Aurora::Matrix &data, Aurora::Matrix &dataAtt,
+                             Aurora::Matrix &dims, 
+                             Aurora::Matrix &senderList,
+                             Aurora::Matrix &receiverList, Aurora::Matrix &res,
+                             double SOS_IN_WATER)
+    {
+        auto nTotalRays = size(senderList, 2);
+        int numIter = 1;
+        if (transParams::bentReconstruction)
+        {
+            numIter =transParams::bentIter+1; 
+        }
+        for (size_t i = 0; i < transParams::gpuSelectionList.getDataSize(); i++)
+        {
+            std::string msg;
+            if (!resetGPUDevice((int)transParams::gpuSelectionList[i],msg))
+            {
+                std::cerr<<msg<<std::endl;
+            }
+        }
+    }
 } // namespace Recon
\ No newline at end of file
diff --git a/src/transmissionReconstruction/reconstruction/reconstruction.h b/src/transmissionReconstruction/reconstruction/reconstruction.h
index b4d1ae0..2a04b3e 100644
--- a/src/transmissionReconstruction/reconstruction/reconstruction.h
+++ b/src/transmissionReconstruction/reconstruction/reconstruction.h
@@ -3,11 +3,18 @@
 #include "Matrix.h"
 namespace Recon {
 struct DiscretizePositionValues{
-   Aurora::Matrix senderCoordList;
+    Aurora::Matrix senderCoordList;
     Aurora::Matrix receiverCoordList;
-   Aurora::Matrix ddims;
-   Aurora::Matrix dims;
-   Aurora::Matrix res;
+    Aurora::Matrix ddims;
+    Aurora::Matrix dims;
+    Aurora::Matrix res;
+};
+struct ArtResult{
+    Aurora::Matrix outSOS;
+    Aurora::Matrix outATT;
+    Aurora::Matrix outAll;
+    bool bentReco;
+    Aurora::Matrix allHitMaps;
 };
 Aurora::Matrix calculateMinimalMaximalTransducerPositions(
    const Aurora::Matrix &aMSenderList, const Aurora::Matrix &aMReceiverList);
@@ -20,6 +27,12 @@ void slownessToSOS(Aurora::Matrix & aVF1, double aSOS_IN_WATER);
 
 DiscretizePositionValues discretizePositions(Aurora::Matrix &aVSenderCoordList, Aurora::Matrix &aVReceiverCoordList, double aNumPixelXY);
 
+ArtResult reconstructArt(Aurora::Matrix &data, Aurora::Matrix &dataAtt,
+                         Aurora::Matrix &dims, 
+                         Aurora::Matrix &senderList,
+                         Aurora::Matrix &receiverList, Aurora::Matrix &res,
+                         double SOS_IN_WATER);
+
 //TODO: exitBent, reconstructArt, saveTransmissionReconstruction.m
 
 }