98 lines
3.6 KiB
C++
98 lines
3.6 KiB
C++
|
#include "solve.h"
|
|||
|
|
|
|||
|
|
#include <cstddef>
|
|||
|
|
|
|||
|
|
#include "Function3D.h"
|
|||
|
|
#include "Matrix.h"
|
|||
|
|
|
|||
|
|
#include "config/config.h"
|
|||
|
|
|
|||
|
|
#include "transmissionReconstruction/reconstruction/solvingEquationSystem/TVAL/TVAL.h"
|
|||
|
|
#include "tvalstruct.h"
|
|||
|
|
|
|||
|
|
namespace Recon
|
|||
|
|
{
|
|||
|
|
struct TVAL3SolverOptions{
|
|||
|
|
Aurora::Matrix gpuSelectionList;
|
|||
|
|
double TVAL3MU;
|
|||
|
|
double TVAL3MU0;
|
|||
|
|
double TVAL3Beta;
|
|||
|
|
double TVAL3Beta0;
|
|||
|
|
bool nonNeg = false;
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
Aurora::Matrix solve( Aurora::Sparse& M, const Aurora::Matrix& b, const Aurora::Matrix& dims, int niter, TVAL3SolverOptions solverOptions){
|
|||
|
|
if (Recon::transParams::name.empty()){
|
|||
|
|
Recon::transParams::name = "TVAL3";
|
|||
|
|
}
|
|||
|
|
if (Recon::transParams::name == "TVAL3")
|
|||
|
|
{
|
|||
|
|
//callTval3
|
|||
|
|
TVALOptions opt;
|
|||
|
|
opt.bent = false;
|
|||
|
|
opt.tol = 1E-10;
|
|||
|
|
opt.maxit = niter;
|
|||
|
|
opt.TVnorm = 2;
|
|||
|
|
opt.disp = false;
|
|||
|
|
opt.mu0 = solverOptions.TVAL3MU0;
|
|||
|
|
opt.mu = solverOptions.TVAL3MU;
|
|||
|
|
opt.beta = solverOptions.TVAL3Beta;
|
|||
|
|
opt.beta0 = solverOptions.TVAL3Beta0;
|
|||
|
|
int device = (int)solverOptions.gpuSelectionList[0];
|
|||
|
|
return callTval3(M, b, dims, device, opt);
|
|||
|
|
}
|
|||
|
|
//SART
|
|||
|
|
else{
|
|||
|
|
//TODO:待实现，先实现默认的TVAL3
|
|||
|
|
return Aurora::Matrix();
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
std::vector<std::vector<Aurora::Matrix>> solveParameterIterator(Aurora::Sparse M, const Aurora::Matrix &b,
|
|||
|
|
const Aurora::Matrix &dims, bool oneIter, bool nonNeg)
|
|||
|
|
{
|
|||
|
|
if (Recon::transParams::name == "TVAL3"){
|
|||
|
|
std::vector<std::vector<Aurora::Matrix>> result(Recon::transParams::muValues.getDataSize());
|
|||
|
|
if (Recon::transParams::muValues.isNull()){
|
|||
|
|
Recon::transParams::muValues = Aurora::ones(1,1);
|
|||
|
|
Recon::transParams::muValues[0] = 24;
|
|||
|
|
}
|
|||
|
|
if (Recon::transParams::betaValues.isNull()){
|
|||
|
|
Recon::transParams::betaValues = Aurora::ones(1,1);
|
|||
|
|
Recon::transParams::betaValues[0] = 1;
|
|||
|
|
}
|
|||
|
|
if (oneIter){
|
|||
|
|
auto temp = Aurora::ones(1,1);
|
|||
|
|
temp[0] = Recon::transParams::muValues[0];
|
|||
|
|
Recon::transParams::muValues = temp;
|
|||
|
|
auto temp2 = Aurora::ones(1,1);
|
|||
|
|
temp2[0] = Recon::transParams::betaValues[0];
|
|||
|
|
Recon::transParams::betaValues = temp2;
|
|||
|
|
}
|
|||
|
|
TVAL3SolverOptions options;
|
|||
|
|
options.gpuSelectionList = Recon::transParams::gpuSelectionList;
|
|||
|
|
for (size_t i = 0; i < Recon::transParams::muValues.getDataSize(); i++)
|
|||
|
|
{
|
|||
|
|
options.TVAL3MU = Recon::transParams::muValues[i];
|
|||
|
|
options.TVAL3MU0 = Recon::transParams::muValues[i];
|
|||
|
|
std::vector<Aurora::Matrix> solveResult(Recon::transParams::betaValues.getDataSize());
|
|||
|
|
for (size_t j = 0; j < Recon::transParams::betaValues.getDataSize(); j++)
|
|||
|
|
{
|
|||
|
|
options.TVAL3Beta = Recon::transParams::betaValues[i];
|
|||
|
|
options.TVAL3Beta0 = Recon::transParams::betaValues[i];
|
|||
|
|
options.nonNeg = nonNeg;
|
|||
|
|
solveResult[j] = solve(M, b, dims, transParams::maxIter, options);
|
|||
|
|
}
|
|||
|
|
result[i] = solveResult;
|
|||
|
|
}
|
|||
|
|
return result;
|
|||
|
|
}
|
|||
|
|
//SART
|
|||
|
|
else{
|
|||
|
|
std::vector<std::vector<Aurora::Matrix>> result;
|
|||
|
|
//TODO:暂时未实现
|
|||
|
|
return result;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
}
|