X-Git-Url: http://git.kpe.io/?p=ctsim.git;a=blobdiff_plain;f=tools%2Fphm2pj.cpp;h=dc045f411df43bd29a05a8d10b23148e12cd1dac;hp=d7397e1a0e935e29d1ab7756af7923e116f36c47;hb=1a050c98763fbbc0662731b0b76953acede6f5d7;hpb=c8b19dfaffba9f06d8b6c40cb1bb83a8964867f7 diff --git a/tools/phm2pj.cpp b/tools/phm2pj.cpp index d7397e1..dc045f4 100644 --- a/tools/phm2pj.cpp +++ b/tools/phm2pj.cpp @@ -32,7 +32,7 @@ enum { O_PHANTOM, O_DESC, O_NRAY, O_ROTANGLE, O_PHMFILE, O_GEOMETRY, O_FOCAL_LENGTH, O_CENTER_DETECTOR_LENGTH, O_VIEW_RATIO, O_SCAN_RATIO, O_OFFSETVIEW, O_TRACE, O_VERBOSE, O_HELP, O_DEBUG, O_VERSION }; -static struct option phm2pj_options[] = +static struct option phm2pj_options[] = { {"phantom", 1, 0, O_PHANTOM}, {"phmfile", 1, 0, O_PHMFILE}, @@ -56,7 +56,7 @@ static struct option phm2pj_options[] = static const char* g_szIdStr = "$Id$"; -void +void phm2pj_usage (const char *program) { std::cout << "usage: " << fileBasename(program) << " outfile ndet nview [--phantom phantom-name] [--phmfile filename] [OPTIONS]\n"; @@ -99,7 +99,7 @@ phm2pj_usage (const char *program) void GatherProjectionsMPI (MPIWorld& mpiWorld, Projections& pjGlobal, Projections& pjLocal, const int opt_debug); #endif -int +int phm2pj_main (int argc, char* const argv[]) { Phantom phm; @@ -122,22 +122,22 @@ phm2pj_main (int argc, char* const argv[]) double opt_rotangle = -1; char* endptr = NULL; char* endstr; - + #ifdef HAVE_MPI MPIWorld mpiWorld (argc, argv); #endif - + Timer timerProgram; - + #ifdef HAVE_MPI if (mpiWorld.getRank() == 0) { #endif while (1) { int c = getopt_long(argc, argv, "", phm2pj_options, NULL); - + if (c == -1) break; - + switch (c) { case O_PHANTOM: optPhmName = optarg; @@ -218,15 +218,15 @@ phm2pj_main (int argc, char* const argv[]) return (1); } break; - case O_OFFSETVIEW: - opt_offsetview = strtol(optarg, &endptr, 10); - endstr = optarg + strlen(optarg); - if (endptr != endstr) { - std::cerr << "Error setting --offsetview to %s" << optarg << std::endl; - phm2pj_usage(argv[0]); - return (1); - } - break; + case O_OFFSETVIEW: + opt_offsetview = strtol(optarg, &endptr, 10); + endstr = optarg + strlen(optarg); + if (endptr != endstr) { + std::cerr << "Error setting --offsetview to %s" << optarg << std::endl; + phm2pj_usage(argv[0]); + return (1); + } + break; case O_VERSION: #ifdef VERSION @@ -244,7 +244,7 @@ phm2pj_main (int argc, char* const argv[]) return (1); } } - + if (optPhmName == "" && optPhmFileName == "") { std::cerr << "No phantom defined\n" << std::endl; phm2pj_usage(argv[0]); @@ -254,7 +254,7 @@ phm2pj_main (int argc, char* const argv[]) phm2pj_usage(argv[0]); return (1); } - + opt_outfile = argv[optind]; opt_ndet = strtol(argv[optind+1], &endptr, 10); endstr = argv[optind+1] + strlen(argv[optind+1]); @@ -270,14 +270,14 @@ phm2pj_main (int argc, char* const argv[]) phm2pj_usage(argv[0]); return (1); } - + if (opt_rotangle < 0) { if (optGeometryName.compare ("parallel") == 0) opt_rotangle = 0.5; else opt_rotangle = 1.0; } - + std::ostringstream desc; desc << "phm2pj: NDet=" << opt_ndet << ", Nview=" << opt_nview << ", NRay=" << opt_nray << ", RotAngle=" << opt_rotangle << "OffsetView =" << opt_offsetview << ", Geometry=" << optGeometryName << ", "; if (optPhmFileName.length()) { @@ -289,7 +289,7 @@ phm2pj_main (int argc, char* const argv[]) desc << ": " << opt_desc; } opt_desc = desc.str(); - + if (optPhmName != "") { phm.createFromPhantom (optPhmName.c_str()); if (phm.fail()) { @@ -298,7 +298,7 @@ phm2pj_main (int argc, char* const argv[]) return (1); } } - + if (optPhmFileName != "") { #ifdef HAVE_MPI std::cerr << "Can not read phantom from file in MPI mode\n"; @@ -306,11 +306,11 @@ phm2pj_main (int argc, char* const argv[]) #endif phm.createFromFile (optPhmFileName.c_str()); } - + #ifdef HAVE_MPI } #endif - + #ifdef HAVE_MPI TimerCollectiveMPI timerBcast(mpiWorld.getComm()); mpiWorld.BcastString (optPhmName); @@ -327,54 +327,54 @@ phm2pj_main (int argc, char* const argv[]) mpiWorld.getComm().Bcast (&opt_trace, 1, MPI::INT, 0); if (opt_verbose) timerBcast.timerEndAndReport ("Time to broadcast variables"); - + if (mpiWorld.getRank() > 0 && optPhmName != "") phm.createFromPhantom (optPhmName.c_str()); #endif - + opt_rotangle *= TWOPI; - Scanner scanner (phm, optGeometryName.c_str(), opt_ndet, opt_nview, opt_offsetview, opt_nray, - opt_rotangle, dOptFocalLength, dOptCenterDetectorLength, dOptViewRatio, dOptScanRatio); + Scanner scanner (phm, optGeometryName.c_str(), opt_ndet, opt_nview, opt_offsetview, opt_nray, + opt_rotangle, dOptFocalLength, dOptCenterDetectorLength, dOptViewRatio, dOptScanRatio); if (scanner.fail()) { std::cout << "Scanner Creation Error: " << scanner.failMessage() << std::endl; return (1); } #ifdef HAVE_MPI mpiWorld.setTotalWorkUnits (opt_nview); - + Projections pjGlobal; if (mpiWorld.getRank() == 0) pjGlobal.initFromScanner (scanner); - + if (opt_verbose) { std::ostringstream os; pjGlobal.printScanInfo(os); std::cout << os.str(); } - + Projections pjLocal (scanner); pjLocal.setNView (mpiWorld.getMyLocalWorkUnits()); - + if (opt_debug) std::cout << "pjLocal->nview = " << pjLocal.nView() << " (process " << mpiWorld.getRank() << ")\n";; - + TimerCollectiveMPI timerProject (mpiWorld.getComm()); scanner.collectProjections (pjLocal, phm, mpiWorld.getMyStartWorkUnit(), mpiWorld.getMyLocalWorkUnits(), false, opt_trace); if (opt_verbose) timerProject.timerEndAndReport ("Time to collect projections"); - + TimerCollectiveMPI timerGather (mpiWorld.getComm()); GatherProjectionsMPI (mpiWorld, pjGlobal, pjLocal, opt_debug); - if (opt_verbose) + if (opt_verbose) timerGather.timerEndAndReport ("Time to gather projections"); - + #else Projections pjGlobal (scanner); scanner.collectProjections (pjGlobal, phm, 0, opt_nview, opt_offsetview, true, opt_trace); #endif - + #ifdef HAVE_MPI - if (mpiWorld.getRank() == 0) + if (mpiWorld.getRank() == 0) #endif { pjGlobal.setCalcTime (timerProgram.timerEnd()); @@ -390,7 +390,7 @@ phm2pj_main (int argc, char* const argv[]) std::cout << "Run time: " << pjGlobal.calcTime() << " seconds\n"; } } - + return (0); } @@ -410,12 +410,12 @@ void GatherProjectionsMPI (MPIWorld& mpiWorld, Projections& pjGlobal, Projection double viewAngle = detArray.viewAngle(); int nDet = detArray.nDet(); DetectorValue* detval = detArray.detValues(); - + mpiWorld.getComm().Send(&viewAngle, 1, MPI::DOUBLE, 0, 0); mpiWorld.getComm().Send(&nDet, 1, MPI::INT, 0, 0); mpiWorld.getComm().Send(detval, nDet, MPI::FLOAT, 0, 0); } - + if (mpiWorld.getRank() == 0) { for (int iProc = 0; iProc < mpiWorld.getNumProcessors(); iProc++) { for (int iw = mpiWorld.getStartWorkUnit(iProc); iw <= mpiWorld.getEndWorkUnit(iProc); iw++) { @@ -424,7 +424,7 @@ void GatherProjectionsMPI (MPIWorld& mpiWorld, Projections& pjGlobal, Projection int nDet; DetectorArray& detArray = pjGlobal.getDetectorArray(iw); DetectorValue* detval = detArray.detValues(); - + mpiWorld.getComm().Recv(&viewAngle, 1, MPI::DOUBLE, iProc, 0, status); mpiWorld.getComm().Recv(&nDet, 1, MPI::INT, iProc, 0, status); mpiWorld.getComm().Recv(detval, nDet, MPI::FLOAT, iProc, 0, status); @@ -437,11 +437,11 @@ void GatherProjectionsMPI (MPIWorld& mpiWorld, Projections& pjGlobal, Projection #ifndef NO_MAIN -int +int main (int argc, char* argv[]) { int retval = 1; - + try { retval = phm2pj_main(argc, argv); #if HAVE_DMALLOC @@ -452,7 +452,7 @@ main (int argc, char* argv[]) } catch (...) { std::cerr << "Unknown exception\n"; } - + return (retval); } #endif