Add reconstructArt(unfinished)

This commit is contained in:
kradchen
2023-05-31 14:57:04 +08:00
parent 30ea494d51
commit e3ce295dda
2 changed files with 44 additions and 4 deletions

View File

@@ -4,6 +4,11 @@
#include "Function2D.h" #include "Function2D.h"
#include "Function3D.h" #include "Function3D.h"
#include "Matrix.h" #include "Matrix.h"
#include "config/config.h"
#include "CudaEnvInit.h"
#include <algorithm>
#include <cmath> #include <cmath>
#include <iostream> #include <iostream>
using namespace Aurora; using namespace Aurora;
@@ -103,4 +108,26 @@ namespace Recon {
result.senderCoordList = aVSenderCoordList; result.senderCoordList = aVSenderCoordList;
return result; 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 } // namespace Recon

View File

@@ -9,6 +9,13 @@ struct DiscretizePositionValues{
Aurora::Matrix dims; Aurora::Matrix dims;
Aurora::Matrix res; Aurora::Matrix res;
}; };
struct ArtResult{
Aurora::Matrix outSOS;
Aurora::Matrix outATT;
Aurora::Matrix outAll;
bool bentReco;
Aurora::Matrix allHitMaps;
};
Aurora::Matrix calculateMinimalMaximalTransducerPositions( Aurora::Matrix calculateMinimalMaximalTransducerPositions(
const Aurora::Matrix &aMSenderList, const Aurora::Matrix &aMReceiverList); 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); 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 //TODO: exitBent, reconstructArt, saveTransmissionReconstruction.m
} }