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 "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

View File

@@ -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
}