r635: no message
[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.32 2001/03/13 04:44:25 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 (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   int m_iNumCoordinates;
88   int m_iNumView;
89   int m_iNumDet;
90   int m_iThetaRange;
91 };
92
93
94
95 // Projections
96 class Projections
97 {
98  public:
99
100   static const int POLAR_INTERP_INVALID;
101   static const int POLAR_INTERP_NEAREST;
102   static const int POLAR_INTERP_BILINEAR;
103   static const int POLAR_INTERP_BICUBIC;
104
105   Projections (const int nView, const int nDet);
106   Projections (const Scanner& scanner);
107   Projections ();
108   ~Projections ();
109
110   static const int getInterpCount() {return s_iInterpCount;}
111   static const char* const* getInterpNameArray() {return s_aszInterpName;}
112   static const char* const* getInterpTitleArray() {return s_aszInterpTitle;}
113   static int convertInterpNameToID (const char* const interpName);
114   static const char* convertInterpIDToName (const int interpID);
115   static const char* convertInterpIDToTitle (const int interpID);
116
117   void initFromScanner (const Scanner& scanner);
118   bool initFromSomatomAR_STAR (int iNViews, int iNDets, unsigned char* pData, unsigned long lDataLength);
119
120   void printProjectionData (int startView, int endView);
121   void printProjectionData ();
122   void printScanInfo (std::ostringstream& os) const;
123
124   bool read (const std::string& fname);
125   bool read (const char* fname);
126   bool write (const char* fname);
127   bool write (const std::string& fname);
128   bool detarrayRead (fnetorderstream& fs, DetectorArray& darray, const int view_num);
129   bool detarrayWrite (fnetorderstream& fs, const DetectorArray& darray, const int view_num);
130
131   Projections* interpolateToParallel();
132
133   bool convertPolar (ImageFile& rIF, int iInterpolation);
134   bool convertFFTPolar (ImageFile& rIF, int iInterpolation, int iZeropad);
135   bool calcArrayPolarCoordinates (unsigned int nx, unsigned int ny, double** ppdView, double** ppdDet);
136   void interpolatePolar (ImageFileArray& v, ImageFileArray& vImag, unsigned int nx, unsigned int ny, std::complex<double>** ppcDetValue,
137     double** ppdDet, double** ppdView, unsigned int nView, unsigned int nDet, int iInterpolate);
138
139   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;
140
141   void setNView (int nView);  // used in MPI to restrict # of views
142   void setRotInc (double rotInc) { m_rotInc = rotInc;}
143   void setDetInc (double detInc) { m_detInc = detInc;}
144   void setCalcTime (double calcTime) {m_calcTime = calcTime;}
145   void setRemark (const char* remark) {m_remark = remark; m_label.setLabelString(remark);}
146   void setRemark (const std::string& remark) {setRemark(remark.c_str());}
147
148   double detStart() const {return m_detStart;}
149   double rotStart() const {return m_rotStart;}
150   double calcTime() const {return m_calcTime;}
151   double viewLen() const {return m_dViewDiameter / SQRT2;}
152   const char*  remark() const {return m_remark.c_str();}
153   double detInc() const {return m_detInc;}
154   double rotInc() const {return m_rotInc;}
155   int nDet() const {return m_nDet;}
156   int nView() const {return m_nView;}
157   int geometry() const {return m_geometry;}
158   double focalLength() const {return m_dFocalLength;}
159   double sourceDetectorLength() const { return m_dSourceDetectorLength;}
160   double viewDiameter() const {return m_dViewDiameter; }
161   double phmLen() const { return m_dViewDiameter / SQRT2; }
162   void setPhmLen(double phmLen) { m_dViewDiameter = phmLen * SQRT2; }
163
164   const std::string& getFilename() const {return m_filename;}
165   Array2dFileLabel& getLabel() {return m_label;}
166   const Array2dFileLabel& getLabel() const {return m_label;}
167
168   DetectorArray& getDetectorArray (const int iview)
169       { return (*m_projData[iview]); }
170   
171   const DetectorArray& getDetectorArray (const int iview) const
172       { return (*m_projData[iview]); }
173
174   static bool copyHeader (const char* const filename, std::ostream& os);
175   static bool copyHeader (const std::string& filename, std::ostream& os);
176
177   static bool copyViewData (const char* const filename, std::ostream& os, int startView, int endView);
178   static bool copyViewData (const std::string& filename, std::ostream& os, int startView, int endView);
179   
180  private:
181   int m_headerSize;             // Size of disk file header 
182   int m_geometry;               // Geometry of scanner 
183   class DetectorArray **m_projData;     // Pointer to array of detarray_st pointers 
184   std::string m_remark;         // description of raysum data 
185   int m_nDet;                   // number of detectors in array 
186   int m_nView;                  // number of rotated views 
187   double m_calcTime;            // time required to calculate raysums 
188   double m_rotStart;            // starting view rotation
189   double m_rotInc;              // angle between rotations 
190   double m_detStart;            // distance of beginning detector to center phantom
191   double m_detInc;              // increment between detectors 
192   double m_dFocalLength;
193   double m_dSourceDetectorLength;
194   double m_dViewDiameter;
195   double m_dFanBeamAngle;
196   kuint32 m_year;                   // Creation date & time
197   kuint32 m_month;
198   kuint32 m_day;
199   kuint32 m_hour;
200   kuint32 m_minute;
201   kuint32 m_second;
202   std::string m_filename;
203   Array2dFileLabel m_label;
204
205   const static kuint16 m_signature;
206
207   static const char* const s_aszInterpName[];
208   static const char* const s_aszInterpTitle[];
209   static const int s_iInterpCount;
210
211   bool headerWrite (fnetorderstream& fs);
212   bool headerRead (fnetorderstream& fs);
213   void newProjData ();
214   void deleteProjData ();
215
216   void init (const int nView, const int nDet);
217
218   // prevent default methods
219   Projections& operator= (const Projections& rhs);   // assignment
220   Projections(const Projections& rhs);               // copy
221 };
222
223
224 #endif