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

Add determineOptimalPulse function

Browse Source
This commit is contained in:
kradchen
2023-06-09 17:55:40 +08:00
parent 2bb12bf078
commit 5e003c7348
3 changed files with 121 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

43
test/Reconstruction_Test.cpp
View File

@@ -4,6 +4,7 @@
#include "Function1D.h"
#include "MatlabReader.h"
#include "MatlabWriter.h"
#include "Matrix.h"
#include "Sparse.h"
#include "config/config.h"
@@ -13,6 +14,8 @@
#include "transmissionReconstruction/reconstruction/reconstruction.h"
#include "transmissionReconstruction/reconstruction/solvingEquationSystem/TVAL/TVAL.h"
#include "reflectionReconstruction/preprocessData/determineOptimalPulse.h"
@@ -42,6 +45,20 @@ protected:
}
};
TEST_F(Reconstruction_Test, determineOptimalPulse) {
Recon::reflectParams::imageResolution = 8.6381e-04;
Recon::reflectParams::optPulseFactor = 24;
auto result = Recon::determineOptimalPulse(1.0000e-07,3000);
EXPECT_EQ(3000, result.getDataSize());
ASSERT_DOUBLE_AE(result[2], 0.0025);
ASSERT_DOUBLE_AE(result[4], 0.0078);
// for(size_t i=0; i<result.getDataSize(); ++i)
// {
// ASSERT_DOUBLE_AE(f1[i], result.outSOS[i])<<"index:"<<i;
// }
}
TEST_F(Reconstruction_Test, reconstructArt) {
MatlabReader m("/home/sun/testData/reconstructArt.mat");
auto data = m.read("data");
@@ -261,3 +278,29 @@ TEST_F(Reconstruction_Test,traceStraightRayBresenham){
EXPECT_DOUBLE_AE(result.path[7],2);
EXPECT_EQ(6, result.pathLen);
}
TEST_F(Reconstruction_Test,callTval3){
TVALOptions opt;
opt.nonneg = false;
opt.bent = false;
opt.tol = 1E-10;
opt.maxit = 50;
opt.TVnorm = 2;
opt.disp = false;
opt.mu0 = 100;
opt.mu = 100;
opt.beta = 1;
opt.beta0 = 1;
MatlabReader m("/home/sun/testData/buildMatrixTest.mat");
auto j1 = m.read("j1");
auto i1 = m.read("i1");
auto s1 = m.read("s1");
Aurora::Sparse M(i1-1,j1-1,s1,734989,1196032);
MatlabReader m2("/home/sun/testData/tval3gpu3d.mat");
auto b = m2.read("b");
auto dims = Aurora::Matrix::fromRawData(new double[3]{128,128,73}, 1, 3);
auto result = Recon::callTval3(M, b, dims, 0,opt);
auto outSOS = Recon::slownessToSOS(result, 1.498206569328594e+03) ;
MatlabWriter w2("/home/krad/transmissionSOS111.mat");
w2.write(outSOS, "SOS");
}