sunwen/UR
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add logic to Detection

Browse Source
This commit is contained in:
kradchen
2023-05-19 15:56:31 +08:00
parent f1cd61bf59
commit 3dd9664ff6
1 changed files with 35 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
test/Detection_Test.cpp
View File

@@ -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();
}