120 lines
5.9 KiB
C++
120 lines
5.9 KiB
C++
#include "getAScanBlockPreprocessed.h"
|
|
|
|
#include "CudaMatrix.h"
|
|
#include "Matrix.h"
|
|
#include "blockingGeometryInfo.h"
|
|
#include "removeDataFromArrays.h"
|
|
#include "getAscanBlock.h"
|
|
#include "config/config.h"
|
|
#include "common/getGeometryInfo.h"
|
|
#include "common/preprocessAscanBlock.h"
|
|
#include "transmissionReconstruction/dataFilter/dataFilter.h"
|
|
#include "reflectionReconstruction/dataFilter.h"
|
|
#include "Function2D.cuh"
|
|
|
|
using namespace Aurora;
|
|
using namespace Recon;
|
|
|
|
AscanBlockPreprocessed Recon::getAscanBlockPreprocessed(Parser* aParser, const Aurora::Matrix& aMp, const Aurora::Matrix& aSl, const Aurora::Matrix& aSn,
|
|
const Aurora::Matrix& aRl, const Aurora::Matrix& aRn, GeometryInfo& aGeom, const MeasurementInfo& aMeasInfo,
|
|
bool aApplyFilter, bool aTransReco)
|
|
{
|
|
AscanBlockPreprocessed result;
|
|
AscanBlock ascanBlock = getAscanBlock(aParser, aMp, aSl, aSn, aRl, aRn);
|
|
result.gainBlock = ascanBlock.gainBlock;
|
|
result.mpBlock = ascanBlock.mpBlock;
|
|
result.rlBlock = ascanBlock.rlBlock;
|
|
result.rnBlock = ascanBlock.rnBlock;
|
|
result.slBlock = ascanBlock.slBlock;
|
|
result.snBlock = ascanBlock.snBlock;
|
|
GeometryBlock geometryBlock = blockingGeometryInfos(aGeom, ascanBlock.rnBlock, ascanBlock.rlBlock, ascanBlock.snBlock, ascanBlock.slBlock, ascanBlock.mpBlock);
|
|
result.receiverPositionBlock = geometryBlock.receiverPositionBlock;
|
|
result.senderPositionBlock = geometryBlock.senderPositionBlock;
|
|
if(aApplyFilter)
|
|
{
|
|
Matrix usedData;
|
|
if(aTransReco)
|
|
{
|
|
usedData = filterTransmissionData(ascanBlock.slBlock, ascanBlock.snBlock, ascanBlock.rlBlock, ascanBlock.rnBlock,
|
|
aGeom.sensData, geometryBlock.senderNormalBlock, geometryBlock.receiverNormalBlock);
|
|
}
|
|
else
|
|
{
|
|
usedData = filterReflectionData(geometryBlock.receiverPositionBlock, geometryBlock.senderPositionBlock, geometryBlock.senderNormalBlock, reflectParams::constrictReflectionAngles);
|
|
}
|
|
|
|
ascanBlock.ascanBlock = removeDataFromArrays(ascanBlock.ascanBlock, usedData);
|
|
result.mpBlock = removeDataFromArrays(ascanBlock.mpBlock, usedData);
|
|
result.slBlock = removeDataFromArrays(ascanBlock.slBlock, usedData);
|
|
result.snBlock = removeDataFromArrays(ascanBlock.snBlock, usedData);
|
|
result.rlBlock = removeDataFromArrays(ascanBlock.rlBlock, usedData);
|
|
result.rnBlock = removeDataFromArrays(ascanBlock.rnBlock, usedData);
|
|
|
|
result.senderPositionBlock = removeDataFromArrays(geometryBlock.senderPositionBlock, usedData);
|
|
result.receiverPositionBlock = removeDataFromArrays(geometryBlock.receiverPositionBlock, usedData);
|
|
result.gainBlock = removeDataFromArrays(ascanBlock.gainBlock, usedData);
|
|
|
|
}
|
|
|
|
if (ascanBlock.ascanBlock.getDataSize() > 0)
|
|
{
|
|
result.ascanBlockPreprocessed = preprocessAscanBlock(ascanBlock.ascanBlock, aMeasInfo);
|
|
}
|
|
else
|
|
{
|
|
result.ascanBlockPreprocessed = ascanBlock.ascanBlock;
|
|
}
|
|
return result;
|
|
}
|
|
|
|
|
|
AscanBlockPreprocessedCuda Recon::getAscanBlockPreprocessedCuda(Parser* aParser, const Aurora::Matrix& aMp, const Aurora::Matrix& aSl, const Aurora::Matrix& aSn,
|
|
const Aurora::Matrix& aRl, const Aurora::Matrix& aRn, GeometryInfo& aGeom, const MeasurementInfo& aMeasInfo,
|
|
bool aApplyFilter, bool aTransReco)
|
|
{
|
|
AscanBlockPreprocessedCuda result;
|
|
AscanBlock ascanBlock = getAscanBlock(aParser, aMp, aSl, aSn, aRl, aRn);
|
|
|
|
result.ascanBlockPreprocessed = ascanBlock.ascanBlock.toDeviceMatrix();
|
|
result.gainBlock = ascanBlock.gainBlock.toDeviceMatrix();
|
|
result.mpBlock = ascanBlock.mpBlock;
|
|
result.rlBlock = ascanBlock.rlBlock;
|
|
result.rnBlock = ascanBlock.rnBlock;
|
|
result.slBlock = ascanBlock.slBlock;
|
|
result.snBlock = ascanBlock.snBlock;
|
|
GeometryBlock geometryBlock = blockingGeometryInfos(aGeom, ascanBlock.rnBlock, ascanBlock.rlBlock, ascanBlock.snBlock, ascanBlock.slBlock, ascanBlock.mpBlock);
|
|
|
|
result.receiverPositionBlock = geometryBlock.receiverPositionBlock;
|
|
result.senderPositionBlock = geometryBlock.senderPositionBlock;
|
|
if(aApplyFilter)
|
|
{
|
|
Matrix usedData;
|
|
if(aTransReco)
|
|
{
|
|
usedData = filterTransmissionData(ascanBlock.slBlock, ascanBlock.snBlock, ascanBlock.rlBlock, ascanBlock.rnBlock,
|
|
aGeom.sensData, geometryBlock.senderNormalBlock, geometryBlock.receiverNormalBlock);
|
|
}
|
|
else
|
|
{
|
|
usedData = filterReflectionData(geometryBlock.receiverPositionBlock, geometryBlock.senderPositionBlock, geometryBlock.senderNormalBlock, reflectParams::constrictReflectionAngles);
|
|
}
|
|
|
|
CudaMatrix usedDataDevice = usedData.toDeviceMatrix();
|
|
result.ascanBlockPreprocessed = valid(result.ascanBlockPreprocessed, usedDataDevice);
|
|
result.mpBlock = removeDataFromArrays(ascanBlock.mpBlock, usedData);
|
|
result.slBlock = removeDataFromArrays(ascanBlock.slBlock, usedData);
|
|
result.snBlock = removeDataFromArrays(ascanBlock.snBlock, usedData);
|
|
result.rlBlock = removeDataFromArrays(ascanBlock.rlBlock, usedData);
|
|
result.rnBlock = removeDataFromArrays(ascanBlock.rnBlock, usedData);
|
|
|
|
result.senderPositionBlock = removeDataFromArrays(geometryBlock.senderPositionBlock, usedData);
|
|
result.receiverPositionBlock = removeDataFromArrays(geometryBlock.receiverPositionBlock, usedData);
|
|
result.gainBlock = valid(result.gainBlock, usedDataDevice);
|
|
}
|
|
|
|
if (ascanBlock.ascanBlock.getDataSize() > 0)
|
|
{
|
|
result.ascanBlockPreprocessed = preprocessAscanBlockCuda(result.ascanBlockPreprocessed, aMeasInfo);
|
|
}
|
|
return result;
|
|
}
|