9c0cce0a810b29f56e6d36a4c262dcc718b94d1f
[ctsim.git] / include / projections.h
1 /*****************************************************************************
2 ** FILE IDENTIFICATION
3 **
4 **   Name:          projections.h
5 **   Purpose:       Header file for Projections class
6 **   Programmer:    Kevin Rosenberg
7 **   Date Started:  July 1, 1984
8 **
9 **  This is part of the CTSim program
10 **  Copyright (c) 1983-2001 Kevin Rosenberg
11 **
12 **  $Id: projections.h,v 1.34 2001/03/13 14:53:44 kevin Exp $
13 **
14 **
15 **  This program is free software; you can redistribute it and/or modify
16 **  it under the terms of the GNU General Public License (version 2) as
17 **  published by the Free Software Foundation.
18 **
19 **  This program is distributed in the hope that it will be useful,
20 **  but WITHOUT ANY WARRANTY; without even the implied warranty of
21 **  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
22 **  GNU General Public License for more details.
23 **
24 **  You should have received a copy of the GNU General Public License
25 **  along with this program; if not, write to the Free Software
26 **  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
27 ******************************************************************************/
28
29 #ifndef PROJECTIONS_H
30 #define PROJECTIONS_H
31
32 class Scanner;
33 class DetectorArray;
34 class Array2dFileLabel;
35 class fnetorderstream;
36
37 #include "array2dfile.h"
38 #include "imagefile.h"
39 #include <complex>
40 #include <vector>
41
42
43 //used for rebinning divergent beam projections to parallel
44 class ParallelRaysumCoordinate {  
45 public:
46   double m_dT;      // Distance from center of origin
47   double m_dTheta;  // perpendicular angle to origin
48   double m_dRaysum;
49
50   static bool compareByTheta (ParallelRaysumCoordinate* a, ParallelRaysumCoordinate* b)
51   { return a->m_dTheta == b->m_dTheta ? b->m_dT > a->m_dT : b->m_dTheta > a->m_dTheta; }
52
53   // sort first by T, then Theta
54   static bool compareByT (ParallelRaysumCoordinate* a, ParallelRaysumCoordinate* b)
55   { return a->m_dT == b->m_dT ? b->m_dTheta > a->m_dTheta : b->m_dT > a->m_dT; }
56
57 };
58
59 class ParallelRaysums {
60 public:
61
62   enum {
63     THETA_RANGE_UNCONSTRAINED = 0,
64     THETA_RANGE_NORMALIZE_TO_TWOPI,
65     THETA_RANGE_FOLD_TO_PI,
66   };
67
68   ParallelRaysums (const Projections* pProjections, int iThetaRange);
69   ~ParallelRaysums ();
70
71   typedef std::vector<ParallelRaysumCoordinate*> CoordinateContainer;
72   typedef CoordinateContainer::iterator CoordinateIterator;
73
74   CoordinateContainer& getCoordinates() {return m_vecpCoordinates;}
75   int getNumCoordinates() const {return m_iNumCoordinates;}
76   void getLimits (double* dMinT, double* dMaxT, double* dMinTheta, double* dMaxTheta) const;
77   CoordinateContainer& getSortedByT();
78   CoordinateContainer& getSortedByTheta();
79   double interpolate (double* pdX, double* pdY, int n, double dXValue, int* iLastFloor = NULL);
80   void getThetaAndRaysumsForT (int iT, double* pdTheta, double* pdRaysum);
81   void getDetPositions (double* pdDetPos);
82
83 private:
84   CoordinateContainer m_vecpCoordinates;
85   CoordinateContainer m_vecpSortedByT;
86   CoordinateContainer m_vecpSortedByTheta;
87   ParallelRaysumCoordinate* m_pCoordinates;
88   int m_iNumCoordinates;
89   int m_iNumView;
90   int m_iNumDet;
91   int m_iThetaRange;
92 };
93
94
95
96 // Projections
97 class Projections
98 {
99  public:
100
101   static const int POLAR_INTERP_INVALID;
102   static const int POLAR_INTERP_NEAREST;
103   static const int POLAR_INTERP_BILINEAR;
104   static const int POLAR_INTERP_BICUBIC;
105
106   Projections (const int nView, const int nDet);
107   Projections (const Scanner& scanner);
108   Projections ();
109   ~Projections ();
110
111   static const int getInterpCount() {return s_iInterpCount;}
112   static const char* const* getInterpNameArray() {return s_aszInterpName;}
113   static const char* const* getInterpTitleArray() {return s_aszInterpTitle;}
114   static int convertInterpNameToID (const char* const interpName);
115   static const char* convertInterpIDToName (const int interpID);
116   static const char* convertInterpIDToTitle (const int interpID);
117
118   void initFromScanner (const Scanner& scanner);
119   bool initFromSomatomAR_STAR (int iNViews, int iNDets, unsigned char* pData, unsigned long lDataLength);
120
121   void printProjectionData (int startView, int endView);
122   void printProjectionData ();
123   void printScanInfo (std::ostringstream& os) const;
124
125   bool read (const std::string& fname);
126   bool read (const char* fname);
127   bool write (const char* fname);
128   bool write (const std::string& fname);
129   bool detarrayRead (fnetorderstream& fs, DetectorArray& darray, const int view_num);
130   bool detarrayWrite (fnetorderstream& fs, const DetectorArray& darray, const int view_num);
131
132   Projections* interpolateToParallel() const;
133
134   bool convertPolar (ImageFile& rIF, int iInterpolation);
135   bool convertFFTPolar (ImageFile& rIF, int iInterpolation, int iZeropad);
136   void calcArrayPolarCoordinates (unsigned int nx, unsigned int ny, double** ppdView, double** ppdDet, 
137     int iNumDetWithZeros, double dZeropadRatio);
138   void interpolatePolar (ImageFileArray& v, ImageFileArray& vImag, unsigned int nx, unsigned int ny, std::complex<double>** ppcDetValue,
139     double** ppdDet, double** ppdView, unsigned int nView, unsigned int nDet, unsigned int nDetWithZeros, 
140     int iInterpolate);
141
142   bool reconstruct (ImageFile& im, const char* const filterName, double filt_param, const char* const filterMethodName, const int zeropad, const char* frequencyFilterName, const char* const interpName, int interp_param, const char* const backprojName, const int trace) const;
143
144   void setNView (int nView);  // used in MPI to restrict # of views
145   void setRotInc (double rotInc) { m_rotInc = rotInc;}
146   void setDetInc (double detInc) { m_detInc = detInc;}
147   void setCalcTime (double calcTime) {m_calcTime = calcTime;}
148   void setRemark (const char* remark) {m_remark = remark; m_label.setLabelString(remark);}
149   void setRemark (const std::string& remark) {setRemark(remark.c_str());}
150
151   double detStart() const {return m_detStart;}
152   double rotStart() const {return m_rotStart;}
153   double calcTime() const {return m_calcTime;}
154   double viewLen() const {return m_dViewDiameter / SQRT2;}
155   const char*  remark() const {return m_remark.c_str();}
156   double detInc() const {return m_detInc;}
157   double rotInc() const {return m_rotInc;}
158   int nDet() const {return m_nDet;}
159   int nView() const {return m_nView;}
160   int geometry() const {return m_geometry;}
161   double focalLength() const {return m_dFocalLength;}
162   double sourceDetectorLength() const { return m_dSourceDetectorLength;}
163   double viewDiameter() const {return m_dViewDiameter; }
164   double phmLen() const { return m_dViewDiameter / SQRT2; }
165   void setPhmLen(double phmLen) { m_dViewDiameter = phmLen * SQRT2; }
166
167   const std::string& getFilename() const {return m_filename;}
168   Array2dFileLabel& getLabel() {return m_label;}
169   const Array2dFileLabel& getLabel() const {return m_label;}
170
171   DetectorArray& getDetectorArray (const int iview)
172       { return (*m_projData[iview]); }
173   
174   const DetectorArray& getDetectorArray (const int iview) const
175       { return (*m_projData[iview]); }
176
177   static bool copyHeader (const char* const filename, std::ostream& os);
178   static bool copyHeader (const std::string& filename, std::ostream& os);
179
180   static bool copyViewData (const char* const filename, std::ostream& os, int startView, int endView);
181   static bool copyViewData (const std::string& filename, std::ostream& os, int startView, int endView);
182   
183  private:
184   int m_headerSize;             // Size of disk file header 
185   int m_geometry;               // Geometry of scanner 
186   class DetectorArray **m_projData;     // Pointer to array of detarray_st pointers 
187   std::string m_remark;         // description of raysum data 
188   int m_nDet;                   // number of detectors in array 
189   int m_nView;                  // number of rotated views 
190   double m_calcTime;            // time required to calculate raysums 
191   double m_rotStart;            // starting view rotation
192   double m_rotInc;              // angle between rotations 
193   double m_detStart;            // distance of beginning detector to center phantom
194   double m_detInc;              // increment between detectors 
195   double m_dFocalLength;
196   double m_dSourceDetectorLength;
197   double m_dViewDiameter;
198   double m_dFanBeamAngle;
199   kuint32 m_year;                   // Creation date & time
200   kuint32 m_month;
201   kuint32 m_day;
202   kuint32 m_hour;
203   kuint32 m_minute;
204   kuint32 m_second;
205   std::string m_filename;
206   Array2dFileLabel m_label;
207
208   const static kuint16 m_signature;
209
210   static const char* const s_aszInterpName[];
211   static const char* const s_aszInterpTitle[];
212   static const int s_iInterpCount;
213
214   bool headerWrite (fnetorderstream& fs);
215   bool headerRead (fnetorderstream& fs);
216   void newProjData ();
217   void deleteProjData ();
218
219   void init (const int nView, const int nDet);
220
221   // prevent default methods
222   Projections& operator= (const Projections& rhs);   // assignment
223   Projections(const Projections& rhs);               // copy
224 };
225
226
227 #endif