Add blockingGeometryInfos.

This commit is contained in:
sunwen
2023-05-24 14:25:23 +08:00
parent db86d973a6
commit 5bd7a15238
5 changed files with 145 additions and 8 deletions

View File

@@ -1,11 +1,13 @@
#include <gtest/gtest.h>
#include "Function2D.h"
#include "Matrix.h"
#include "common/ceMatchedFilterHandling.h"
#include "common/common.h"
#include "common/convertfp16tofloat.h"
#include "common/getGeometryInfo.h"
#include "common/daraBlockCreation/getAscanBlock.h"
#include "common/dataBlockCreation/getAscanBlock.h"
#include "common/dataBlockCreation/blockingGeometryInfo.h"
#include "Parser.h"
#include "MatlabReader.h"
@@ -232,7 +234,7 @@ TEST_F(Common_Test, getAscanBlock) {
Aurora::Matrix::fromRawData(new double[18]{1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16,17,18}, 18),
Aurora::Matrix::fromRawData(new double[1]{1}, 1),
Aurora::Matrix::fromRawData(new double[18]{1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16,17,18}, 18));
for(int i=0; i<ascanBlock.getDataSize(); ++i)
for(int i=0; i<result.ascanBlock.getDataSize(); ++i)
{
EXPECT_DOUBLE_AE(ascanBlock[i], result.ascanBlock[i]);
}
@@ -240,7 +242,7 @@ TEST_F(Common_Test, getAscanBlock) {
EXPECT_DOUBLE_EQ(ascanBlock.getDimSize(1), result.ascanBlock.getDimSize(1));
auto gainBlock = m.read("gainBlock");
for(int i=0; i<gainBlock.getDataSize(); ++i)
for(int i=0; i<result.gainBlock.getDataSize(); ++i)
{
EXPECT_DOUBLE_AE(gainBlock[i], result.gainBlock[i]);
}
@@ -248,7 +250,7 @@ TEST_F(Common_Test, getAscanBlock) {
EXPECT_DOUBLE_EQ(gainBlock.getDimSize(1), result.gainBlock.getDimSize(1));
auto mpBlock = m.read("mpBlock");
for(int i=0; i<mpBlock.getDataSize(); ++i)
for(int i=0; i<result.mpBlock.getDataSize(); ++i)
{
EXPECT_DOUBLE_AE(mpBlock[i], result.mpBlock[i]);
}
@@ -256,7 +258,7 @@ TEST_F(Common_Test, getAscanBlock) {
EXPECT_DOUBLE_EQ(mpBlock.getDimSize(1), result.mpBlock.getDimSize(1));
auto rlBlock = m.read("rlBlock");
for(int i=0; i<rlBlock.getDataSize(); ++i)
for(int i=0; i<result.rlBlock.getDataSize(); ++i)
{
EXPECT_DOUBLE_AE(rlBlock[i], result.rlBlock[i]);
}
@@ -264,7 +266,7 @@ TEST_F(Common_Test, getAscanBlock) {
EXPECT_DOUBLE_EQ(rlBlock.getDimSize(1), result.rlBlock.getDimSize(1));
auto rnBlock = m.read("rnBlock");
for(int i=0; i<rnBlock.getDataSize(); ++i)
for(int i=0; i<result.rnBlock.getDataSize(); ++i)
{
EXPECT_DOUBLE_AE(rnBlock[i], result.rnBlock[i]);
}
@@ -272,7 +274,7 @@ TEST_F(Common_Test, getAscanBlock) {
EXPECT_DOUBLE_EQ(rnBlock.getDimSize(1), result.rnBlock.getDimSize(1));
auto slBlock = m.read("slBlock");
for(int i=0; i<slBlock.getDataSize(); ++i)
for(int i=0; i<result.slBlock.getDataSize(); ++i)
{
EXPECT_DOUBLE_AE(slBlock[i], result.slBlock[i]);
}
@@ -280,10 +282,49 @@ TEST_F(Common_Test, getAscanBlock) {
EXPECT_DOUBLE_EQ(slBlock.getDimSize(1), result.slBlock.getDimSize(1));
auto snBlock = m.read("snBlock");
for(int i=0; i<snBlock.getDataSize(); ++i)
for(int i=0; i<result.snBlock.getDataSize(); ++i)
{
EXPECT_DOUBLE_AE(snBlock[i], result.snBlock[i]);
}
EXPECT_DOUBLE_EQ(snBlock.getDimSize(0), result.snBlock.getDimSize(0));
EXPECT_DOUBLE_EQ(snBlock.getDimSize(1), result.snBlock.getDimSize(1));
}
TEST_F(Common_Test, blockingGeometryInfos) {
Recon::GeometryInfo geom;
MatlabReader geomReader("/home/sun/testData/getGeometryInfo.mat");
geom.receiverNormals = geomReader.read4d("receiverNormals");
geom.receiverPositions = geomReader.read4d("receiverPositions");
geom.senderNormals = geomReader.read4d("senderNormals");
geom.senderPositions = geomReader.read4d("senderPositions");
MatlabReader m("/home/sun/testData/blockingGeometryInfo.mat");
std::string dataPath = "/home/AScans_Data/volunteer_20230328/20230328T123237/";
Parser parser(dataPath);
auto rnBlock = m.read("rnBlock");
auto rlBlock = m.read("rlBlock");
auto snBlock = m.read("snBlock");
auto slBlock = m.read("slBlock");
auto mpBlock = m.read("mpBlock");
auto result = Recon::blockingGeometryInfos(geom, rnBlock, rlBlock, snBlock, slBlock, mpBlock);
auto receiverNormalBlock = m.read("receiverNormalBlock");
for(int i=0; i<result.receiverNormalBlock.getDataSize(); ++i)
{
EXPECT_DOUBLE_AE(receiverNormalBlock[i], result.receiverNormalBlock[i]);
}
auto receiverPositionBlock = m.read("receiverPositionBlock");
for(int i=0; i<result.receiverPositionBlock.getDataSize(); ++i)
{
EXPECT_DOUBLE_AE(receiverPositionBlock[i], result.receiverPositionBlock[i]);
}
auto senderNormalBlock = m.read("senderNormalBlock");
for(int i=0; i<result.senderNormalBlock.getDataSize(); ++i)
{
EXPECT_DOUBLE_AE(senderNormalBlock[i], result.senderNormalBlock[i]);
}
auto senderPositionBlock = m.read("senderPositionBlock");
for(int i=0; i<result.senderPositionBlock.getDataSize(); ++i)
{
EXPECT_DOUBLE_AE(senderPositionBlock[i], result.senderPositionBlock[i]);
}
}