Fix detectAttVectorized

This commit is contained in:
kradchen
2023-05-22 14:07:27 +08:00
parent a4a972f99f
commit 5276d1cc42
2 changed files with 43 additions and 26 deletions

View File

@@ -73,24 +73,19 @@ TEST_F(Detection_Test, calculateAttenuation) {
TEST_F(Detection_Test, applyTimeWindowing) {
MatlabReader m("/home/krad/TestData/timeWindow.mat");
MatlabReader m("/home/krad/TestData/timeWindow2.mat");
auto AscanBlock = m.read("AscanBlock");
auto distBlock = m.read("distBlock");
auto sosBlock = m.read("sosBlock");
auto AscanBlockProcessed = m.read("AscanBlockProcessed");
auto distBlock = m.read("dists");
auto sosBlock = m.read("waterTempBlock");
auto AscanBlockProcessed = m.read("AscanBlock1");
auto startSearch = m.read("startSearch");
auto result = Recon::applyTimeWindowing(AscanBlock, 10000000, distBlock, sosBlock, 1482, 0, 0, 1400, 1650, false);
auto result = Recon::applyTimeWindowing(AscanBlock, 10000000, distBlock, sosBlock, 1.511948131508464e+03, 5.2, 1, 1450, 1550, false);
#pragma omp parallel for
for (size_t i = 0; i < AscanBlockProcessed.getDataSize(); i++)
{
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;
}
}
@@ -102,8 +97,8 @@ TEST_F(Detection_Test, detectTofVectorized) {
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");
auto sosWaterBlock = m.read("waterTempBlock");
auto sosWaterRefBlock = m.read("waterTempRefBlock");
double expectedSOSWater = 1.511948131508464e+03;
auto result = Recon::detectTofVectorized(
@@ -116,5 +111,14 @@ TEST_F(Detection_Test, detectTofVectorized) {
Recon::transParams::minSpeedOfSound,
Recon::transParams::maxSpeedOfSound, Recon::transParams::gaussWindow);
int size = result.getDataSize();
MatlabReader m2("/home/krad/TestData/tofResult.mat");
auto tof = m2.read("tof");
auto sosvalue = m2.read("sosValue");
EXPECT_EQ(tof.getDataSize(), result.getDataSize());
#pragma omp parallel for
for (size_t i = 0; i < tof.getDataSize(); i++)
{
EXPECT_DOUBLE_AE(tof[i],result[i])<<",index:"<<i;
}
}