r327: Added textfile output
[ctsim.git] / libctsim / phantom.cpp
index d07b6a32eae0035242d8c467b5838a55e066655f..d8fa862f625724968e328fc86be4eabdb03deaea 100644 (file)
@@ -9,7 +9,7 @@
 **  This is part of the CTSim program
 **  Copyright (C) 1983-2000 Kevin Rosenberg
 **
-**  $Id: phantom.cpp,v 1.8 2000/07/18 03:14:35 kevin Exp $
+**  $Id: phantom.cpp,v 1.24 2001/01/02 05:33:37 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
 
 #include "ct.h"
 
+const int PhantomElement::POINTS_PER_CIRCLE = 360;\r
+const double PhantomElement::SCALE_PELEM_EXTENT=0.005;  // increase pelem limits by 0.5% \r
+
+const int Phantom::PHM_INVALID = -1;
+const int Phantom::PHM_HERMAN = 0;
+const int Phantom::PHM_B_HERMAN = 1;
+const int Phantom::PHM_SHEPP_LOGAN = 2;
+const int Phantom::PHM_B_SHEPP_LOGAN = 3;
+const int Phantom::PHM_UNITPULSE = 4;
+
+const char* Phantom::s_aszPhantomName[] = 
+{
+  {"herman"},
+  {"herman-b"},
+  {"shepp-logan"},
+  {"shepp-logan-b"},
+  {"unitpulse"},
+};
+
+const char* Phantom::s_aszPhantomTitle[] = 
+{
+  {"Herman Head"},
+  {"Herman Head (Bordered)"},
+  {"Shepp-Logan"},
+  {"Shepp-Logan (Bordered)"},
+  {"Unit Pulse"},
+};
+
+const int Phantom::s_iPhantomCount = sizeof(s_aszPhantomName) / sizeof(const char*);
+
 
 // CLASS IDENTIFICATION
-//   Phanton
+//   Phantom
 //
 
-Phantom::Phantom (void)
+Phantom::Phantom ()
 {
   init ();
 }
@@ -45,7 +75,7 @@ Phantom::Phantom (const char* const phmName)
 }
 
 void 
-Phantom::init (void)
+Phantom::init ()
 {
   m_nPElem = 0;
   m_xmin = 1E30;
@@ -58,7 +88,7 @@ Phantom::init (void)
   m_id = PHM_INVALID;
 }
 
-Phantom::~Phantom (void)
+Phantom::~Phantom ()
 {
   for (PElemIterator i = m_listPElem.begin(); i != m_listPElem.end(); i++) {
     delete *i;
@@ -67,39 +97,37 @@ Phantom::~Phantom (void)
 
 
 const char*
-Phantom::convertPhantomIDToName (const PhantomID phmID)
+Phantom::convertPhantomIDToName (int phmID)
 {
-  const char *name = "";
+  static const char *name = "";
 
-  if (phmID == PHM_HERMAN)
-    name = PHM_HERMAN_STR;
-  else if (phmID == PHM_BHERMAN)
-    name = PHM_BHERMAN_STR;
-  else if (phmID == PHM_ROWLAND)
-    name = PHM_ROWLAND_STR;
-  else if (phmID == PHM_BROWLAND)
-    name = PHM_BROWLAND_STR;
-  else if (phmID == PHM_UNITPULSE)
-    name = PHM_UNITPULSE_STR;
+  if (phmID >= 0 && phmID < s_iPhantomCount)
+      return (s_aszPhantomName[phmID]);
 
   return (name);
 }
+
+const char*
+Phantom::convertPhantomIDToTitle (int phmID)
+{
+  static const char *title = "";
+
+  if (phmID >= 0 && phmID < s_iPhantomCount)
+      return (s_aszPhantomName[phmID]);
+
+  return (title);
+}
       
-Phantom::PhantomID
+int
 Phantom::convertNameToPhantomID (const char* const phmName) 
 {
-  PhantomID id = PHM_INVALID;
+  int id = PHM_INVALID;
 
-  if (strcasecmp (phmName, PHM_HERMAN_STR) == 0)
-    id = PHM_HERMAN;
-  else if (strcasecmp (phmName, PHM_BHERMAN_STR) == 0)
-    id = PHM_BHERMAN;
-  else if (strcasecmp (phmName, PHM_ROWLAND_STR) == 0)
-    id = PHM_ROWLAND;
-  else if (strcasecmp (phmName, PHM_BROWLAND_STR) == 0)
-    id = PHM_BROWLAND;
-  else if (strcasecmp (phmName, PHM_UNITPULSE_STR) == 0)
-    id = PHM_UNITPULSE;
+  for (int i = 0; i < s_iPhantomCount; i++)
+      if (strcasecmp (phmName, s_aszPhantomName[i]) == 0) {
+         id = i;
+         break;
+      }
 
   return (id);
 }
@@ -108,7 +136,7 @@ Phantom::convertNameToPhantomID (const char* const phmName)
 bool
 Phantom::createFromPhantom (const char* const phmName)
 {
-  PhantomID phmid = convertNameToPhantomID (phmName);
+  int phmid = convertNameToPhantomID (phmName);
   if (phmid == PHM_INVALID) {
     m_fail = true;
     m_failMessage = "Invalid phantom name ";
@@ -122,21 +150,21 @@ Phantom::createFromPhantom (const char* const phmName)
 }
 
 bool
-Phantom::createFromPhantom (const PhantomID phmid)
+Phantom::createFromPhantom (const int phmid)
 {
   switch (phmid) 
     {
     case PHM_HERMAN:
       addStdHerman();
       break;
-    case PHM_BHERMAN:
+    case PHM_B_HERMAN:
       addStdHermanBordered();
       break;
-    case PHM_ROWLAND:
-      addStdRowland();
+    case PHM_SHEPP_LOGAN:
+      addStdSheppLogan();
       break;
-    case PHM_BROWLAND:
-      addStdRowlandBordered ();
+    case PHM_B_SHEPP_LOGAN:
+      addStdSheppLoganBordered();
       break;
     case PHM_UNITPULSE:
       m_composition = P_UNIT_PULSE;
@@ -170,36 +198,43 @@ Phantom::createFromPhantom (const PhantomID phmid)
 bool
 Phantom::createFromFile (const char* const fname)
 {
-  bool stoploop = false;
-  bool retval = false;
+  bool bGoodFile = true;
   FILE *fp;
 
   if ((fp = fopen (fname, "r")) == NULL)
     return (false);
 
-  do {
+  m_name = fname;
+
+  while (1) {
     double cx, cy, u, v, rot, dens;
     char pelemtype[80];
-    int n = fscanf (fp, "%79s %lf %lf %lf %lf %lf %lf",
-               pelemtype, &cx, &cy, &u, &v, &rot, &dens);
+
+    int status = fscanf (fp, "%79s %lf %lf %lf %lf %lf %lf", pelemtype, &cx, &cy, &u, &v, &rot, &dens);
     
-    if (n == EOF || n == 0) {  /* end of file */
-      stoploop = true;
-      retval = false;
-    } else if (n != 7) {
-      stoploop = true;
-      retval = false;
-    } else {
-      addPElem (pelemtype, cx, cy, u, v, rot, dens);
-      retval = true;
+    if (status == static_cast<int>(EOF)) 
+      break;
+    else if (status != 7) {
+      sys_error (ERR_WARNING, "Insufficient fields reading phantom file %s [Phantom::createFromFile]", fname);
+      bGoodFile = false;
     }
-  } while (stoploop == false);
+    addPElem (pelemtype, cx, cy, u, v, rot, dens);
+  }
   
   fclose (fp);
 
-  return (retval);
+  return (bGoodFile);
 }
 
+bool\r
+Phantom::fileWrite (const char* const fname)\r
+{\r
+  fstream file (fname, ios::out);\r
+\r
+  if (! file.fail())\r
+    printDefinitions (file);\r
+  return ! file.fail();\r
+}\r
 
 /* NAME
  *   addPElem          Add pelem
@@ -248,26 +283,32 @@ Phantom::addPElem (const char *type, const double cx, const double cy, const dou
  */
 
 void 
-Phantom::print (void) const
+Phantom::print (std::ostream& os) const
 {
-  printf("PRINTING Phantom\n\n");
-  printf("number of pelems in Phantom = %d\n", m_nPElem);
-  printf("limits: xmin=%8.2g  ymin=%8.2g  xmax=%8.2g  ymax=%8.2g\n",
-        m_xmin, m_ymin, m_xmax, m_ymax);
+  os << "Number of PElements: " << m_nPElem << "\n";
+  os << "Limits: xmin=" << m_xmin << ", ymin=" << m_ymin << ", xmax=" << m_xmax << ", ymax=" << m_ymax << "\n";
   
-  for (PElemIterator i = m_listPElem.begin(); i != m_listPElem.end(); i++) {
-      printf("PELEM:\n");
-      printf("# pts=%3d        atten = %7.4f   rot = %7.2f (deg)\n",
-            (*i)->nOutlinePoints(), (*i)->atten(), convertRadiansToDegrees ((*i)->rot()));
-    
-    printf("xmin=%7.3g  ymin=%7.3g  xmax=%7.3g  ymax=%7.3g\n",
-          (*i)->xmin(), (*i)->ymin(), (*i)->xmax(), (*i)->ymax());
-    
-    //    for (int i = 0; i < m_nPoints; i++)
-    //      printf("\t%8.3g    %8.3g\n", i->xOutline()[i], i->yOutline()[i]);
+  for (PElemConstIterator i = m_listPElem.begin(); i != m_listPElem.end(); i++) {\r
+    const PhantomElement& rPE = **i;
+      os << "PhantomElement: nPoints=" << rPE.nOutlinePoints();\r
+      os << ", atten=" << rPE.atten() << " rot=" << convertRadiansToDegrees (rPE.rot()) << "\n";
+      os << "xmin=" << rPE.xmin() << ", ymin=" << rPE.ymin() << ", xmax=" << rPE.xmax() << ", ymax=" << rPE.ymax() << "\n";
+    \r
+      if (false)
+        for (int i = 0; i < rPE.nOutlinePoints(); i++)
+          os << rPE.xOutline()[i] << "," << rPE.yOutline()[i] << "\n";
   }
 }
-
+\r
+void\r
+Phantom::printDefinitions (std::ostream& os) const\r
+{\r
+  for (PElemConstIterator i = m_listPElem.begin(); i != m_listPElem.end(); i++) {\r
+    const PhantomElement& rPE = **i;\r
+    rPE.printDefinition (os);\r
+  }\r
+}
+\r
 
 /* NAME
  *   show              Show vector outline of Phantom to user
@@ -278,31 +319,32 @@ Phantom::print (void) const
 
 #ifdef HAVE_SGP
 void 
-Phantom::show (void) const
+Phantom::show () const
 {
-  double wsize = m_xmax - m_xmin;
-  double xmin = m_xmin;
-  double ymin = m_ymin;
-  double xmax, ymax;
-  SGP_ID gid;
+  SGPDriver driverSGP ("Phantom Show");
+  SGP sgp (driverSGP);
 
-  if ((m_ymax - m_ymin) > wsize)
-      wsize = m_ymax - m_ymin;
-  wsize *= 1.1;
+  show (sgp);
 
-  xmax = xmin + wsize;
-  ymax = ymin + wsize; 
-  
-  printf("Drawing Phantom:\n\n");
-  printf("    data limits: %9.3g, %9.3g, %9.3g, %9.3g\n", m_xmin, m_ymin, m_xmax, m_ymax);
-  printf("    window size: %9.3g, %9.3g, %9.3g, %9.3g\n", xmin, ymin, xmax, ymax);
+  std::cout << "Press return to continue";
+  cio_kb_getc();
+}
+
+void 
+Phantom::show (SGP& sgp) const
+{
+  double wsize = m_xmax - m_xmin;
+  if ((m_ymax - m_ymin) > wsize) 
+      wsize = m_ymax - m_ymin;
+  wsize *= 1.01;
+  double halfWindow = wsize / 2;
 
-  gid = sgp2_init (0, 0, "Phantom Show");
-  sgp2_window (xmin, ymin, xmax, ymax);
+  double xcent = m_xmin + (m_xmax - m_xmin) / 2;
+  double ycent = m_ymin + (m_ymax - m_ymin) / 2;
 
-  draw();
+  sgp.setWindow (xcent - halfWindow, ycent - halfWindow, xcent + halfWindow, ycent + halfWindow);
 
-  sgp2_close (gid);
+  draw (sgp);
 }
 #endif
 
@@ -316,16 +358,16 @@ Phantom::show (void) const
 
 #ifdef HAVE_SGP
 void 
-Phantom::draw (void) const
+Phantom::draw (SGP& sgp) const
 {
   for (PElemIterator i = m_listPElem.begin(); i != m_listPElem.end(); i++)
-    sgp2_polyline_abs ((*i)->xOutline(), (*i)->yOutline(), (*i)->nOutlinePoints());
+    sgp.polylineAbs ((*i)->xOutline(), (*i)->yOutline(), (*i)->nOutlinePoints());
 }
 #endif
 
 
 /* NAME
- *   addStdRowland             Make head phantom of S.W. Rowland
+ *   addStdSheppLogan  Make head phantom of Shepp-Logan
  *
  * REFERENCES
  *   S. W. Rowland, "Computer Implementation of Image Reconstruction
@@ -334,7 +376,7 @@ Phantom::draw (void) const
  */
 
 void 
-Phantom::addStdRowland (void)
+Phantom::addStdSheppLogan ()
 {
   addPElem ("ellipse",  0.0000,  0.0000, 0.6900,  0.9200,   0.0,  1.00);
   addPElem ("ellipse",  0.0000, -0.0184, 0.6624,  0.8740,   0.0, -0.98);
@@ -350,10 +392,10 @@ Phantom::addStdRowland (void)
 }
 
 void 
-Phantom::addStdRowlandBordered (void)
+Phantom::addStdSheppLoganBordered ()
 {
-  addStdRowland ();
-  addPElem ("ellipse", 0.000, 0.0000, 0.7500, 1.000, 0.0, 0.00);
+  addStdSheppLogan ();
+  addPElem ("rectangle", 0.000, 0.0000, 0.8600, 1.150, 0.0, 0.00);
 }
 
 /* NAME
@@ -365,7 +407,7 @@ Phantom::addStdRowlandBordered (void)
  */
 
 void 
-Phantom::addStdHerman (void)
+Phantom::addStdHerman ()
 {
   addPElem ("ellipse",  0.000,  1.50,  0.375, 0.3000,  90.00, -0.003);
   addPElem ("ellipse",  0.675, -0.75,  0.225, 0.1500, 140.00,  0.010);
@@ -385,10 +427,10 @@ Phantom::addStdHerman (void)
 }
 
 void
-Phantom::addStdHermanBordered (void)
+Phantom::addStdHermanBordered ()
 {
   addStdHerman();
-  addPElem ("ellipse",  0.000, 0.000, 8.650, 8.650,  0.00, 0.000);
+  addPElem ("rectangle",  0.000, 0.00, 10.780, 8.110,  90.00, 0.000);
 }
 
 
@@ -406,11 +448,11 @@ Phantom::addStdHermanBordered (void)
 void
 Phantom::convertToImagefile (ImageFile& im, const int in_nsample, const int trace) const
 {
-  convertToImagefile (im, in_nsample, trace, 0, im.nx());
+  convertToImagefile (im, in_nsample, trace, 0, im.nx(), true);
 }
 
 void 
-Phantom::convertToImagefile (ImageFile& im, const int in_nsample, const int trace, const int colStart, const int colCount) const
+Phantom::convertToImagefile (ImageFile& im, const int in_nsample, const int trace, const int colStart, const int colCount, bool bStoreAtColumnPos) const
 {
   int nx = im.nx();
   int ny = im.ny();
@@ -452,8 +494,12 @@ Phantom::convertToImagefile (ImageFile& im, const int in_nsample, const int trac
   ImageFileArray v = im.getArray();
 
   for (int ix = 0; ix < colCount; ix++)
-      for (int iy = 0; iy < ny; iy++)
-         v[ix][iy] = 0;
+    for (int iy = 0; iy < ny; iy++) {
+      int iColStore = ix;
+      if (bStoreAtColumnPos)
+               iColStore += colStart;
+      v[iColStore][iy] = 0;
+    }
 
   double x_start = xmin + (colStart * xinc);
   for (PElemConstIterator pelem = m_listPElem.begin(); pelem != m_listPElem.end(); pelem++) {
@@ -461,11 +507,14 @@ Phantom::convertToImagefile (ImageFile& im, const int in_nsample, const int trac
       double x, y, xi, yi;
       int ix, iy, kx, ky;
       for (ix = 0, x = x_start; ix < colCount; ix++, x += xinc) {
-         for (iy = 0, y = ymin; iy < ny; iy++, y += yinc) {
-             for (kx = 0, xi = x + kxofs; kx < nsample; kx++, xi += kxinc) {
-                 for (ky = 0, yi = y + kyofs; ky < nsample; ky++, yi += kyinc)
-                     if (rPElem.isPointInside (xi, yi, PHM_COORD) == TRUE)
-                         v[ix][iy] += rPElem.atten();
+       int iColStore = ix;
+       if (bStoreAtColumnPos)
+         iColStore += colStart;
+       for (iy = 0, y = ymin; iy < ny; iy++, y += yinc) {
+         for (kx = 0, xi = x + kxofs; kx < nsample; kx++, xi += kxinc) {
+           for (ky = 0, yi = y + kyofs; ky < nsample; ky++, yi += kyinc)
+             if (rPElem.isPointInside (xi, yi, PHM_COORD) == TRUE)
+                       v[iColStore][iy] += rPElem.atten();
              } // for kx
          } /* for iy */
       }  /* for ix */
@@ -473,11 +522,16 @@ Phantom::convertToImagefile (ImageFile& im, const int in_nsample, const int trac
   
 
   if (nsample > 1) {
-    double factor = 1.0 / (nsample * nsample);
-
-    for (int ix = 0; ix < colCount; ix++)
-      for (int iy = 0; iy < ny; iy++)
-       v[ix][iy] *= factor;
+    double factor = 1.0 / static_cast<double>(nsample * nsample);
+
+\r
+    for (int ix = 0; ix < colCount; ix++) {
+               int iColStore = ix;\r
+               if (bStoreAtColumnPos)\r
+                       iColStore += colStart;\r
+               for (int iy = 0; iy < ny; iy++)
+                       v[iColStore][iy] *= factor;\r
+       }
   }
 }
 
@@ -514,12 +568,20 @@ PhantomElement::PhantomElement (const char *type, const double cx, const double
   m_rectLimits[2] = m_xmax;   m_rectLimits[3] = m_ymax;
 }
 
+\r
 
-PhantomElement::~PhantomElement (void)
+PhantomElement::~PhantomElement ()
 {
     delete m_xOutline;
     delete m_yOutline;
 }
+\r
+void\r
+PhantomElement::printDefinition (std::ostream& os) const\r
+{\r
+  os << convertTypeToName (m_type) << " " << m_cx << " " << m_cy << " " << m_u << " "\r
+    << m_v << " " << convertRadiansToDegrees (m_rot) << " " << m_atten << "\n";\r
+}\r
 
 PhmElemType
 PhantomElement::convertNameToType (const char* const typeName)
@@ -541,9 +603,29 @@ PhantomElement::convertNameToType (const char* const typeName)
 
     return (type);
 }
+\r
+const char* const\r
+PhantomElement::convertTypeToName (PhmElemType iType)\r
+{\r
+  static char* pszType = "Unknown";\r
+\r
+  if (iType == PELEM_RECTANGLE)\r
+    pszType = "rectangle";\r
+  else if (iType == PELEM_TRIANGLE)\r
+    pszType = "triangle";\r
+  else if (iType == PELEM_ELLIPSE)\r
+    pszType = "ellipse";\r
+  else if (iType == PELEM_SECTOR)\r
+    pszType = "sector";\r
+  else if (iType == PELEM_SEGMENT)\r
+    pszType = "segment";\r
+\r
+  return pszType;\r
+}\r
+\r
 
 void 
-PhantomElement::makeTransformMatrices (void)
+PhantomElement::makeTransformMatrices ()
 {
   GRFMTX_2D temp;
 
@@ -583,7 +665,7 @@ PhantomElement::makeTransformMatrices (void)
  */
 
 void
-PhantomElement::makeVectorOutline (void)
+PhantomElement::makeVectorOutline ()
 {
   double radius, theta, start, stop;
   double xfact, yfact;