X-Git-Url: http://git.kpe.io/?p=ctsim.git;a=blobdiff_plain;f=src%2Fctrec.cpp;h=dc945e35bb9c1524003437177190a7b8f754ed8d;hp=0ec840217f0944b6a759c9235a57913c116dea91;hb=08f34bf3ba14d4f436f4d2ef0ee5af1d6eb266ac;hpb=f4a23943110823118f35756dd41fbd6707f04511 diff --git a/src/ctrec.cpp b/src/ctrec.cpp index 0ec8402..dc945e3 100644 --- a/src/ctrec.cpp +++ b/src/ctrec.cpp @@ -9,7 +9,7 @@ ** This is part of the CTSim program ** Copyright (C) 1983-2000 Kevin Rosenberg ** -** $Id: ctrec.cpp,v 1.10 2000/06/17 20:12:15 kevin Exp $ +** $Id: ctrec.cpp,v 1.14 2000/06/25 17:32:24 kevin Exp $ ** ** This program is free software; you can redistribute it and/or modify ** it under the terms of the GNU General Public License (version 2) as @@ -103,23 +103,24 @@ static void ReduceImageMPI (MPIWorld& mpiWorld, ImageFile* imLocal, ImageFile* i int ctrec_main (int argc, char * argv[]) { - ImageFile *imGlobal = NULL; Projections projGlobal; - char *pj_name, *im_filename = NULL; + ImageFile* imGlobal = NULL; + char* filenameProj = NULL; + char* filenameImage = NULL; string remark; - char filt_name[80]; char *endptr; - int opt_verbose = 0; - int opt_debug = 0; - int opt_trace = TRACE_NONE; - double opt_filter_param = 1; - FilterType opt_filter = FILTER_ABS_BANDLIMIT; - InterpolationType opt_interp = I_LINEAR; - int opt_interp_param = 1; - BackprojType opt_backproj = O_BPROJ_DIFF2; + int optVerbose = 0; + int optDebug = 0; + int optTrace = TRACE_NONE; + double optFilterParam = -1; + string optFilterName = SignalFilter::FILTER_ABS_BANDLIMIT_STR; + string optInterpName = Backprojector::INTERP_LINEAR_STR; + string optBackprojName = Backprojector::BPROJ_IDIFF2_STR; + // string optFilterMethodName = SignalFilter::FILTER_METHOD_CONVOLUTION_STR; + int optInterpParam = 1; int nx, ny; #ifdef HAVE_MPI - ImageFile *imLocal; + ImageFile* imLocal; int mpi_nview, mpi_ndet; double mpi_detinc, mpi_rotinc, mpi_phmlen; MPIWorld mpiWorld (argc, argv); @@ -140,48 +141,39 @@ ctrec_main (int argc, char * argv[]) switch (c) { case O_INTERP: - if ((opt_interp = opt_set_interpolation(optarg)) < 0) { - ctrec_usage(argv[0]); - return (1); - } + optInterpName = optarg; break; case O_FILTER: - if ((opt_filter = opt_set_filter(optarg)) < 0) { - ctrec_usage(argv[0]); - return (1); - } + optFilterName = optarg; break; case O_BACKPROJ: - if ((opt_backproj = opt_set_backproj(optarg)) < 0) { - ctrec_usage(argv[0]); - return (1); - } + optBackprojName = optarg; break; case O_FILTER_PARAM: - opt_filter_param = strtod(optarg, &endptr); + optFilterParam = strtod(optarg, &endptr); if (endptr != optarg + strlen(optarg)) { ctrec_usage(argv[0]); } break; case O_VERBOSE: - opt_verbose = 1; + optVerbose = 1; break; case O_DEBUG: - opt_debug = 1; + optDebug = 1; break; case O_TRACE: - if ((opt_trace = opt_set_trace(optarg)) < 0) { + if ((optTrace = opt_set_trace(optarg)) < 0) { ctrec_usage(argv[0]); return (1); } break; case O_VERSION: #ifdef VERSION - cout << "Version " << VERSION << endl; + cout << "Version " << VERSION << endl; #else - cout << "Unknown version number" << endl; + cout << "Unknown version number" << endl; #endif - exit(0); + return (0); case O_HELP: case '?': ctrec_usage(argv[0]); @@ -197,24 +189,24 @@ ctrec_main (int argc, char * argv[]) return (1); } - pj_name = argv[optind]; + filenameProj = argv[optind]; - im_filename = argv[optind + 1]; + filenameImage = argv[optind + 1]; nx = strtol(argv[optind + 2], &endptr, 10); ny = strtol(argv[optind + 3], &endptr, 10); - if (opt_filter == FILTER_G_HAMMING || opt_filter == FILTER_ABS_G_HAMMING) - snprintf (filt_name, sizeof(filt_name), "%s: alpha = %.2f", - name_of_filter (opt_filter), opt_filter_param); + ostringstream filterDesc; + if (optFilterParam >= 0) + filterDesc << optFilterName << ": alpha=" << optFilterParam; else - snprintf (filt_name, sizeof(filt_name), "%s", name_of_filter (opt_filter)); - + filterDesc << optFilterName; + ostringstream label; - label << "Reconstruct: " << nx << "x" << ny << ", " << filt_name << ", " << name_of_interpolation (opt_interp) << ", " << name_of_backproj(opt_backproj); + label << "Reconstruct: " << nx << "x" << ny << ", " << filterDesc.str() << ", " << optInterpName << ", " << optBackprojName; remark = label.str(); - if (opt_verbose) + if (optVerbose) cout << "Remark: " << remark << endl; #ifdef HAVE_MPI } @@ -222,8 +214,8 @@ ctrec_main (int argc, char * argv[]) #ifdef HAVE_MPI if (mpiWorld.getRank() == 0) { - projGlobal.read (pj_name); - if (opt_verbose) + projGlobal.read (filenameProj); + if (optVerbose) projGlobal.printScanInfo(); mpi_ndet = projGlobal.nDet(); @@ -234,14 +226,14 @@ ctrec_main (int argc, char * argv[]) } TimerCollectiveMPI timerBcast (mpiWorld.getComm()); - mpiWorld.getComm().Bcast (&opt_verbose, 1, MPI::INT, 0); - mpiWorld.getComm().Bcast (&opt_debug, 1, MPI::INT, 0); - mpiWorld.getComm().Bcast (&opt_trace, 1, MPI::INT, 0); - mpiWorld.getComm().Bcast (&opt_filter, 1, MPI::INT, 0); - mpiWorld.getComm().Bcast (&opt_interp, 1, MPI::INT, 0); - mpiWorld.getComm().Bcast (&opt_filter_param, 1, MPI::DOUBLE, 0); - mpiWorld.getComm().Bcast (&opt_interp_param, 1, MPI::INT, 0); - mpiWorld.getComm().Bcast (&opt_backproj, 1, MPI::INT, 0); + mpiWorld.BcastString (optBackprojName); + mpiWorld.BcastString (optFilterName); + mpiWorld.BcastString (optInterpName); + mpiWorld.getComm().Bcast (&optVerbose, 1, MPI::INT, 0); + mpiWorld.getComm().Bcast (&optDebug, 1, MPI::INT, 0); + mpiWorld.getComm().Bcast (&optTrace, 1, MPI::INT, 0); + mpiWorld.getComm().Bcast (&optFilterParam, 1, MPI::DOUBLE, 0); + mpiWorld.getComm().Bcast (&optInterpParam, 1, MPI::INT, 0); mpiWorld.getComm().Bcast (&mpi_ndet, 1, MPI::INT, 0); mpiWorld.getComm().Bcast (&mpi_nview, 1, MPI::INT, 0); mpiWorld.getComm().Bcast (&mpi_detinc, 1, MPI::DOUBLE, 0); @@ -249,7 +241,7 @@ ctrec_main (int argc, char * argv[]) mpiWorld.getComm().Bcast (&mpi_rotinc, 1, MPI::DOUBLE, 0); mpiWorld.getComm().Bcast (&nx, 1, MPI::INT, 0); mpiWorld.getComm().Bcast (&ny, 1, MPI::INT, 0); - if (opt_verbose) + if (optVerbose) timerBcast.timerEndAndReport ("Time to broadcast variables"); mpiWorld.setTotalWorkUnits (mpi_nview); @@ -260,37 +252,37 @@ ctrec_main (int argc, char * argv[]) projLocal.setRotInc (mpi_rotinc); TimerCollectiveMPI timerScatter (mpiWorld.getComm()); - ScatterProjectionsMPI (mpiWorld, projGlobal, projLocal, opt_debug); - if (opt_verbose) + ScatterProjectionsMPI (mpiWorld, projGlobal, projLocal, optDebug); + if (optVerbose) timerScatter.timerEndAndReport ("Time to scatter projections"); if (mpiWorld.getRank() == 0) { - imGlobal = new ImageFile (im_filename, nx, ny); + imGlobal = new ImageFile (filenameImage, nx, ny); imGlobal->fileCreate(); } imLocal = new ImageFile (nx, ny); #else - projGlobal.read (pj_name); - if (opt_verbose) + projGlobal.read (filenameProj); + if (optVerbose) projGlobal.printScanInfo(); - imGlobal = new ImageFile (im_filename, nx, ny); + imGlobal = new ImageFile (filenameImage, nx, ny); imGlobal->fileCreate(); #endif #ifdef HAVE_MPI TimerCollectiveMPI timerReconstruct (mpiWorld.getComm()); - proj_reconst (*imLocal, projLocal, opt_filter, opt_filter_param, opt_interp, opt_interp_param, opt_backproj, opt_trace); - if (opt_verbose) + projLocal.reconstruct (*imLocal, optFilterName.c_str(), optFilterParam, optInterpName.c_str(), optInterpParam, optBackprojName.c_str(), optTrace); + if (optVerbose) timerReconstruct.timerEndAndReport ("Time to reconstruct"); TimerCollectiveMPI timerReduce (mpiWorld.getComm()); ReduceImageMPI (mpiWorld, imLocal, imGlobal); - if (opt_verbose) + if (optVerbose) timerReduce.timerEndAndReport ("Time to reduce image"); #else - proj_reconst (*imGlobal, projGlobal, opt_filter, opt_filter_param, opt_interp, opt_interp_param, opt_backproj, opt_trace); + projGlobal.reconstruct (*imGlobal, optFilterName.c_str(), optFilterParam, optInterpName.c_str(), optInterpParam, optBackprojName.c_str(), optTrace); #endif #ifdef HAVE_MPI @@ -302,7 +294,7 @@ ctrec_main (int argc, char * argv[]) imGlobal->labelAdd (Array2dFileLabel::L_HISTORY, projGlobal.remark(), projGlobal.calcTime()); imGlobal->labelAdd (Array2dFileLabel::L_HISTORY, remark.c_str(), calcTime); imGlobal->fileClose (); - if (opt_verbose) + if (optVerbose) cout << "Run time: " << calcTime << " seconds" << endl; } #ifdef HAVE_MPI @@ -319,7 +311,7 @@ ctrec_main (int argc, char * argv[]) ////////////////////////////////////////////////////////////////////////////////////// #ifdef HAVE_MPI -static void ScatterProjectionsMPI (MPIWorld& mpiWorld, Projections& projGlobal, Projections& projLocal, const int opt_debug) +static void ScatterProjectionsMPI (MPIWorld& mpiWorld, Projections& projGlobal, Projections& projLocal, const int optDebug) { if (mpiWorld.getRank() == 0) { for (int iProc = 0; iProc < mpiWorld.getNumProcessors(); iProc++) { @@ -371,7 +363,17 @@ ReduceImageMPI (MPIWorld& mpiWorld, ImageFile* imLocal, ImageFile* imGlobal) int main (int argc, char* argv[]) { - return (ctrec_main(argc, argv)); + int retval = 1; + + try { + retval = ctrec_main(argc, argv); + } catch (exception e) { + cerr << "Exception: " << e.what() << endl; + } catch (...) { + cerr << "Unknown exception" << endl; + } + + return (retval); } #endif