X-Git-Url: http://git.kpe.io/?p=ctsim.git;a=blobdiff_plain;f=libctsim%2Fprojections.cpp;h=22171504420064543e7d958bb168e461f9f9f79c;hp=d2ab2b69e189409ce08fb007b8116c0e6da7c3c9;hb=c6cda8844a491b71759e5dd5edba830d0b809cfd;hpb=9324f74c9b1dcbb6aae655dbb24ca05284a144df diff --git a/libctsim/projections.cpp b/libctsim/projections.cpp index d2ab2b6..2217150 100644 --- a/libctsim/projections.cpp +++ b/libctsim/projections.cpp @@ -8,7 +8,7 @@ ** This is part of the CTSim program ** Copyright (C) 1983-2000 Kevin Rosenberg ** -** $Id: projections.cpp,v 1.33 2000/12/17 22:30:09 kevin Exp $ +** $Id: projections.cpp,v 1.38 2001/01/06 15:33:15 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 @@ -25,35 +25,57 @@ ******************************************************************************/ #include "ct.h" - -const kuint16 Projections::m_signature = ('P'*256 + 'J'); + +const kuint16 Projections::m_signature = ('P'*256 + 'J'); + +const int Projections::POLAR_INTERP_INVALID = -1; +const int Projections::POLAR_INTERP_NEAREST = 0; +const int Projections::POLAR_INTERP_BILINEAR = 1; +const int Projections::POLAR_INTERP_BICUBIC = 2; + +const char* Projections::s_aszInterpName[] = +{ + {"nearest"}, + {"bilinear"}, +// {"bicubic"}, +}; + +const char* Projections::s_aszInterpTitle[] = +{ + {"Nearest"}, + {"Bilinear"}, +// {"Bicubic"}, +}; + +const int Projections::s_iInterpCount = sizeof(s_aszInterpName) / sizeof(char*); + /* NAME - * Projections Constructor for projections matrix storage - * - * SYNOPSIS - * proj = projections_create (filename, nView, nDet) - * Projections& proj Allocated projections structure & matrix - * int nView Number of rotated view - * int nDet Number of detectors - * - */ +* Projections Constructor for projections matrix storage +* +* SYNOPSIS +* proj = projections_create (filename, nView, nDet) +* Projections& proj Allocated projections structure & matrix +* int nView Number of rotated view +* int nDet Number of detectors +* +*/ Projections::Projections (const Scanner& scanner) - : m_projData(0) +: m_projData(0) { initFromScanner (scanner); } Projections::Projections (const int nView, const int nDet) - : m_projData(0) +: m_projData(0) { init (nView, nDet); } Projections::Projections (void) - : m_projData(0) +: m_projData(0) { init (0, 0); } @@ -63,6 +85,43 @@ Projections::~Projections (void) deleteProjData(); } +int +Projections::convertInterpNameToID (const char* const interpName) +{ + int interpID = POLAR_INTERP_INVALID; + + for (int i = 0; i < s_iInterpCount; i++) + if (strcasecmp (interpName, s_aszInterpName[i]) == 0) { + interpID = i; + break; + } + + return (interpID); +} + +const char* +Projections::convertInterpIDToName (const int interpID) +{ + static const char *interpName = ""; + + if (interpID >= 0 && interpID < s_iInterpCount) + return (s_aszInterpName[interpID]); + + return (interpName); +} + +const char* +Projections::convertInterpIDToTitle (const int interpID) +{ + static const char *interpTitle = ""; + + if (interpID >= 0 && interpID < s_iInterpCount) + return (s_aszInterpTitle[interpID]); + + return (interpTitle); +} + + void Projections::init (const int nView, const int nDet) @@ -71,6 +130,15 @@ Projections::init (const int nView, const int nDet) m_nView = nView; m_nDet = nDet; newProjData (); + + time_t t = time (NULL); + tm* lt = localtime (&t); + m_year = lt->tm_year; + m_month = lt->tm_mon; + m_day = lt->tm_mday; + m_hour = lt->tm_hour; + m_minute = lt->tm_min; + m_second = lt->tm_sec; } void @@ -79,7 +147,7 @@ Projections::initFromScanner (const Scanner& scanner) m_label.setLabelType (Array2dFileLabel::L_HISTORY); deleteProjData(); init (scanner.nView(), scanner.nDet()); - + m_phmLen = scanner.phmLen(); m_rotInc = scanner.rotInc(); m_detInc = scanner.detInc(); @@ -88,12 +156,6 @@ Projections::initFromScanner (const Scanner& scanner) m_fieldOfView = scanner.fieldOfView(); m_rotStart = 0; m_detStart = -(scanner.detLen() / 2); -#if 0 - if (m_geometry == Scanner::GEOMETRY_EQUILINEAR) { - m_detInc /= 2; - std::cout << "Kludge: detInc /= 2 in Projections::initFromScanner" << endl; - } -#endif } void @@ -111,10 +173,10 @@ Projections::newProjData (void) { if (m_projData) sys_error(ERR_WARNING, "m_projData != NULL [newProjData]"); - + if (m_nView > 0 && m_nDet) { m_projData = new DetectorArray* [m_nView]; - + for (int i = 0; i < m_nView; i++) m_projData[i] = new DetectorArray (m_nDet); } @@ -122,12 +184,12 @@ Projections::newProjData (void) /* NAME - * projections_free Free memory allocated to projections - * - * SYNOPSIS - * projections_free(proj) - * Projections& proj Projectionss to be deallocated - */ +* projections_free Free memory allocated to projections +* +* SYNOPSIS +* projections_free(proj) +* Projections& proj Projectionss to be deallocated +*/ void Projections::deleteProjData (void) @@ -135,7 +197,7 @@ Projections::deleteProjData (void) if (m_projData != NULL) { for (int i = 0; i < m_nView; i++) delete m_projData[i]; - + delete m_projData; m_projData = NULL; } @@ -143,9 +205,9 @@ Projections::deleteProjData (void) /* NAME - * Projections::headerWwrite Write data header for projections file - * - */ +* Projections::headerWwrite Write data header for projections file +* +*/ bool Projections::headerWrite (fnetorderstream& fs) @@ -162,7 +224,7 @@ Projections::headerWrite (fnetorderstream& fs) kuint16 _hour = m_hour; kuint16 _minute = m_minute; kuint16 _second = m_second; - + kfloat64 _calcTime = m_calcTime; kfloat64 _rotStart = m_rotStart; kfloat64 _rotInc = m_rotInc; @@ -171,11 +233,11 @@ Projections::headerWrite (fnetorderstream& fs) kfloat64 _phmLen = m_phmLen; kfloat64 _fieldOfView = m_fieldOfView; kfloat64 _focalLength = m_focalLength; - + fs.seekp(0); if (! fs) return false; - + fs.writeInt16 (_hsize); fs.writeInt16 (_signature); fs.writeInt32 (_nView); @@ -197,21 +259,21 @@ Projections::headerWrite (fnetorderstream& fs) fs.writeInt16 (_second); fs.writeInt16 (_remarksize); fs.write (m_remark.c_str(), _remarksize); - + m_headerSize = fs.tellp(); _hsize = m_headerSize; fs.seekp(0); fs.writeInt16 (_hsize); if (! fs) - return false; + return false; return true; } /* NAME - * projections_read_header Read data header for projections file - * - */ +* projections_read_header Read data header for projections file +* +*/ bool Projections::headerRead (fnetorderstream& fs) { @@ -221,8 +283,8 @@ Projections::headerRead (fnetorderstream& fs) fs.seekg(0); if (! fs) - return false; - + return false; + fs.readInt16 (_hsize); fs.readInt16 (_signature); fs.readInt32 (_nView); @@ -243,17 +305,17 @@ Projections::headerRead (fnetorderstream& fs) fs.readInt16 (_minute); fs.readInt16 (_second); fs.readInt16 (_remarksize); - + if (! fs) { - sys_error (ERR_SEVERE, "Error reading header information , _remarksize=%d [projections_read_header]", _remarksize); - return false; + sys_error (ERR_SEVERE, "Error reading header information , _remarksize=%d [projections_read_header]", _remarksize); + return false; } - + if (_signature != m_signature) { sys_error (ERR_SEVERE, "File %s does not have a valid projection file signature", m_filename.c_str()); return false; } - + char* pszRemarkStorage = new char [_remarksize+1]; fs.read (pszRemarkStorage, _remarksize); if (! fs) { @@ -262,12 +324,12 @@ Projections::headerRead (fnetorderstream& fs) } pszRemarkStorage[_remarksize] = 0; m_remark = pszRemarkStorage; - delete pszRemarkStorage; - + delete pszRemarkStorage; + off_t _hsizeread = fs.tellg(); if (!fs || _hsizeread != _hsize) { - sys_error (ERR_WARNING, "File header size read %ld != file header size stored %ld [read_projections_header]\n_remarksize=%ld", (long int) _hsizeread, _hsize, _remarksize); - return false; + sys_error (ERR_WARNING, "File header size read %ld != file header size stored %ld [read_projections_header]\n_remarksize=%ld", (long int) _hsizeread, _hsize, _remarksize); + return false; } m_headerSize = _hsize; @@ -288,12 +350,12 @@ Projections::headerRead (fnetorderstream& fs) m_hour = _hour; m_minute = _minute; m_second = _second; - + m_label.setLabelType (Array2dFileLabel::L_HISTORY); m_label.setLabelString (m_remark); m_label.setCalcTime (m_calcTime); m_label.setDateTime (m_year, m_month, m_day, m_hour, m_minute, m_second); - + return true; } @@ -307,27 +369,27 @@ Projections::read (const std::string& filename) bool Projections::read (const char* filename) { - m_filename = filename; -#ifdef MSVC - frnetorderstream fileRead (m_filename.c_str(), std::ios::in | std::ios::binary); -#else - frnetorderstream fileRead (m_filename.c_str(), std::ios::in | std::ios::binary | std::ios::nocreate); -#endif - + m_filename = filename; +#ifdef MSVC + frnetorderstream fileRead (m_filename.c_str(), std::ios::in | std::ios::binary); +#else + frnetorderstream fileRead (m_filename.c_str(), std::ios::in | std::ios::binary | std::ios::nocreate); +#endif + if (fileRead.fail()) return false; - + if (! headerRead (fileRead)) return false; - + deleteProjData (); newProjData(); - + for (int i = 0; i < m_nView; i++) { if (! detarrayRead (fileRead, *m_projData[i], i)) break; } - + fileRead.close(); return true; } @@ -336,7 +398,7 @@ Projections::read (const char* filename) bool Projections::copyViewData (const std::string& filename, std::ostream& os, int startView, int endView) { - return copyViewData (filename.c_str(), os, startView, endView); + return copyViewData (filename.c_str(), os, startView, endView); } bool @@ -345,56 +407,56 @@ Projections::copyViewData (const char* const filename, std::ostream& os, int sta frnetorderstream is (filename, ios::in | ios::binary); kuint16 sizeHeader, signature; kuint32 _nView, _nDet; - + is.readInt16 (sizeHeader); is.readInt16 (signature); is.readInt32 (_nView); is.readInt32 (_nDet); int nView = _nView; int nDet = _nDet; - + if (signature != m_signature) { sys_error (ERR_FATAL, "Illegal signature in projection file %s", filename); return false; } - + if (startView < 0) - startView = 0; + startView = 0; if (startView > nView - 1) - startView = nView; + startView = nView; if (endView < 0 || endView > nView - 1) - endView = nView - 1; - + endView = nView - 1; + if (startView > endView) { // swap if start > end - int tempView = endView; - endView = startView; - startView = tempView; + int tempView = endView; + endView = startView; + startView = tempView; } - + int sizeView = 8 /* view_angle */ + 4 /* nDet */ + (4 * nDet); unsigned char* pViewData = new unsigned char [sizeView]; - + for (int i = startView; i <= endView; i++) { - is.seekg (sizeHeader + i * sizeView); - is.read (reinterpret_cast(pViewData), sizeView); - os.write (reinterpret_cast(pViewData), sizeView); - if (is.fail() || os.fail()) - break; + is.seekg (sizeHeader + i * sizeView); + is.read (reinterpret_cast(pViewData), sizeView); + os.write (reinterpret_cast(pViewData), sizeView); + if (is.fail() || os.fail()) + break; } - + delete pViewData; if (is.fail()) sys_error (ERR_FATAL, "Error reading projection file"); if (os.fail()) sys_error (ERR_FATAL, "Error writing projection file"); - + return (! (is.fail() | os.fail())); } bool Projections::copyHeader (const std::string& filename, std::ostream& os) { - return copyHeader (filename.c_str(), os); + return copyHeader (filename.c_str(), os); } bool @@ -409,20 +471,20 @@ Projections::copyHeader (const char* const filename, std::ostream& os) sys_error (ERR_FATAL, "Illegal signature in projection file %s", filename); return false; } - + unsigned char* pHdrData = new unsigned char [sizeHeader]; is.read (reinterpret_cast(pHdrData), sizeHeader); if (is.fail()) { - sys_error (ERR_FATAL, "Error reading header"); - return false; + sys_error (ERR_FATAL, "Error reading header"); + return false; } - + os.write (reinterpret_cast(pHdrData), sizeHeader); if (os.fail()) { - sys_error (ERR_FATAL, "Error writing header"); - return false; + sys_error (ERR_FATAL, "Error writing header"); + return false; } - + return true; } @@ -441,43 +503,32 @@ Projections::write (const char* filename) sys_error (ERR_SEVERE, "Error opening file %s for output [projections_create]", filename); return false; } - -#ifdef HAVE_TIME - time_t t = time(NULL); - tm* lt = localtime(&t); - m_year = lt->tm_year; - m_month = lt->tm_mon; - m_day = lt->tm_mday; - m_hour = lt->tm_hour; - m_minute = lt->tm_min; - m_second = lt->tm_sec; -#endif - + if (! headerWrite (fs)) - return false; - + return false; + if (m_projData != NULL) { for (int i = 0; i < m_nView; i++) { if (! detarrayWrite (fs, *m_projData[i], i)) - break; + break; } } if (! fs) return false; - - fs.close(); - + + fs.close(); + return true; } /* NAME - * detarrayRead Read a Detector Array structure from the disk - * - * SYNOPSIS - * detarrayRead (proj, darray, view_num) - * DETARRAY *darray Detector array storage location to be filled - * int view_num View number to read - */ +* detarrayRead Read a Detector Array structure from the disk +* +* SYNOPSIS +* detarrayRead (proj, darray, view_num) +* DETARRAY *darray Detector array storage location to be filled +* int view_num View number to read +*/ bool Projections::detarrayRead (fnetorderstream& fs, DetectorArray& darray, const int iview) @@ -489,39 +540,39 @@ Projections::detarrayRead (fnetorderstream& fs, DetectorArray& darray, const int DetectorValue* detval_ptr = darray.detValues(); kfloat64 view_angle; kuint32 nDet; - + fs.seekg (start_data); - + fs.readFloat64 (view_angle); fs.readInt32 (nDet); darray.setViewAngle (view_angle); // darray.setNDet ( nDet); for (unsigned int i = 0; i < nDet; i++) { - kfloat32 detval; - fs.readFloat32 (detval); - detval_ptr[i] = detval; + kfloat32 detval; + fs.readFloat32 (detval); + detval_ptr[i] = detval; } if (! fs) return false; - + return true; } /* NAME - * detarrayWrite Write detector array data to the disk - * - * SYNOPSIS - * detarrayWrite (darray, view_num) - * DETARRAY *darray Detector array data to be written - * int view_num View number to write - * - * DESCRIPTION - * This routine writes the detarray data from the disk sequentially to - * the file that was opened with open_projections(). Data is written in - * binary format. - */ +* detarrayWrite Write detector array data to the disk +* +* SYNOPSIS +* detarrayWrite (darray, view_num) +* DETARRAY *darray Detector array data to be written +* int view_num View number to write +* +* DESCRIPTION +* This routine writes the detarray data from the disk sequentially to +* the file that was opened with open_projections(). Data is written in +* binary format. +*/ bool Projections::detarrayWrite (fnetorderstream& fs, const DetectorArray& darray, const int iview) @@ -539,27 +590,27 @@ Projections::detarrayWrite (fnetorderstream& fs, const DetectorArray& darray, co sys_error (ERR_SEVERE, "Error seeking detectory array [detarrayWrite]"); return false; } - + fs.writeFloat64 (view_angle); fs.writeInt32 (nDet); - + for (unsigned int i = 0; i < nDet; i++) { kfloat32 detval = detval_ptr[i]; fs.writeFloat32 (detval); } - + if (! fs) return (false); - + return true; } /* NAME - * printProjectionData Print projections data - * - * SYNOPSIS - * printProjectionData () - */ +* printProjectionData Print projections data +* +* SYNOPSIS +* printProjectionData () +*/ void Projections::printProjectionData () @@ -588,7 +639,7 @@ Projections::printProjectionData (int startView, int endView) printf("View %d: angle %f\n", ir, m_projData[ir]->viewAngle()); DetectorValue* detval = m_projData[ir]->detValues(); for (int id = 0; id < m_projData[ir]->nDet(); id++) - printf("%8.4f ", detval[id]); + printf("%8.4f ", detval[id]); printf("\n"); } } @@ -611,4 +662,195 @@ Projections::printScanInfo (std::ostringstream& os) const } +bool +Projections::convertPolar (ImageFile& rIF, int iInterpolationID) +{ + unsigned int nx = rIF.nx(); + unsigned int ny = rIF.ny(); + ImageFileArray v = rIF.getArray(); + ImageFileArray vImag = rIF.getImaginaryArray(); + + if (! v || nx == 0 || ny == 0) + return false; + + Array2d adView (nx, ny); + Array2d adDet (nx, ny); + double** ppdView = adView.getArray(); + double** ppdDet = adDet.getArray(); + + calcArrayPolarCoordinates (nx, ny, ppdView, ppdDet); + + std::complex** ppcDetValue = new std::complex* [m_nView]; + for (unsigned int iView = 0; iView < m_nView; iView++) { + ppcDetValue[iView] = new std::complex [m_nDet]; + for (unsigned int iDet = 0; iDet < m_nDet; iDet++) + ppcDetValue[iView][iDet] = std::complex(getDetectorArray (iView).detValues()[iDet], 0); + } + + interpolatePolar (v, vImag, nx, ny, ppcDetValue, ppdView, ppdDet, m_nView, m_nDet, iInterpolationID); + + for (iView = 0; iView < m_nView; iView++) + delete [] ppcDetValue[iView]; + delete [] ppcDetValue; + + return true; +} + + +bool +Projections::convertFFTPolar (ImageFile& rIF, int iInterpolationID, int iZeropad) +{ + unsigned int nx = rIF.nx(); + unsigned int ny = rIF.ny(); + ImageFileArray v = rIF.getArray(); + if (! rIF.isComplex()) + rIF.convertRealToComplex(); + ImageFileArray vImag = rIF.getImaginaryArray(); + + if (! v || nx == 0 || ny == 0) + return false; + + Array2d adView (nx, ny); + Array2d adDet (nx, ny); + double** ppdView = adView.getArray(); + double** ppdDet = adDet.getArray(); + + std::complex** ppcDetValue = new std::complex* [m_nView]; + unsigned int iView; + double* pdDet = new double [m_nDet]; + fftw_complex* pcIn = new fftw_complex [m_nDet]; + fftw_plan plan = fftw_create_plan (m_nDet, FFTW_FORWARD, FFTW_IN_PLACE); + + for (iView = 0; iView < m_nView; iView++) { + ppcDetValue[iView] = new std::complex [m_nDet]; + unsigned int iDet; + for (iDet = 0; iDet < m_nDet; iDet++) { + pcIn[iDet].re = getDetectorArray(iView).detValues()[iDet]; + pcIn[iDet].im = 0; + } + fftw_one (plan, pcIn, NULL); + for (iDet = 0; iDet < m_nDet; iDet++) + ppcDetValue[iView][iDet] = std::complex (pcIn[iDet].re, pcIn[iDet].im); + Fourier::shuffleFourierToNaturalOrder (ppcDetValue[iView], m_nDet); + } + + fftw_destroy_plan (plan); + delete [] pcIn; + + calcArrayPolarCoordinates (nx, ny, ppdView, ppdDet); + + interpolatePolar (v, vImag, nx, ny, ppcDetValue, ppdView, ppdDet, m_nView, m_nDet, iInterpolationID); + + for (iView = 0; iView < m_nView; iView++) + delete [] ppcDetValue[iView]; + delete [] ppcDetValue; + + return true; +} + +void +Projections::calcArrayPolarCoordinates (unsigned int nx, unsigned int ny, double** ppdView, double** ppdDet) +{ + double xMin = -m_phmLen / 2; + double xMax = xMin + m_phmLen; + double yMin = -m_phmLen / 2; + double yMax = yMin + m_phmLen; + + double xInc = (xMax - xMin) / nx; // size of cells + double yInc = (yMax - yMin) / ny; + + int iDetCenter = (m_nDet - 1) / 2; // index refering to L=0 projection + + if (m_geometry != Scanner::GEOMETRY_PARALLEL) { + sys_error (ERR_WARNING, "convertPolar supports Parallel only"); + } + + double x = xMin + xInc / 2; // Rectang coords of center of pixel + for (unsigned int ix = 0; ix < nx; x += xInc, ix++) { + double y = yMin + yInc / 2; + for (unsigned int iy = 0; iy < ny; y += yInc, iy++) { + double r = ::sqrt (x * x + y * y); + double phi = atan2 (y, x); + + if (phi >= PI) { + phi -= PI; + } else if (phi < 0) { + phi += PI; + } else + r = -r; + + ppdView[ix][iy] = (phi - m_rotStart) / m_rotInc; + ppdDet[ix][iy] = (r / m_detInc) + iDetCenter; + } + } +} + +void +Projections::interpolatePolar (ImageFileArray& v, ImageFileArray& vImag, + unsigned int nx, unsigned int ny, std::complex** ppcDetValue, + double** ppdView, double** ppdDet, unsigned int nView, unsigned int nDet, int iInterpolationID) +{ + for (unsigned int ix = 0; ix < ny; ix++) { + for (unsigned int iy = 0; iy < ny; iy++) { + if (iInterpolationID == POLAR_INTERP_NEAREST) { + int iView = nearest (ppdView[ix][iy]); + int iDet = nearest (ppdDet[ix][iy]); + if (iView == nView) { + iView = 0; + // iDet = m_nDet - iDet; + } + if (iDet >= 0 && iDet < nDet && iView >= 0 && iView < nView) { + v[ix][iy] = ppcDetValue[iView][iDet].real(); + if (vImag) + vImag[ix][iy] = ppcDetValue[iView][iDet].imag(); + } else { + sys_error (ERR_SEVERE, "Can't find projection data for ix=%d,iy=%d with radView=%f and radDet=%f", + ix, iy, ppdView[ix][iy], ppdDet[ix][iy]); + v[ix][iy] = 0; + } + } else if (iInterpolationID == POLAR_INTERP_BILINEAR) { + int iFloorView = ppdView[ix][iy]; + double dFracView = ppdView[ix][iy] - iFloorView; + int iFloorDet = ppdDet[ix][iy]; + double dFracDet = ppdDet[ix][iy] - iFloorDet; + + if (iFloorDet >= 0 && iFloorView >= 0) { + std::complex v1 = ppcDetValue[iFloorView][iFloorDet]; + std::complex v2, v3, v4; + if (iFloorView < nView - 1) + v2 = ppcDetValue[iFloorView + 1][iFloorDet]; + else + v2 = ppcDetValue[0][iFloorDet]; + if (iFloorDet < nDet - 1) + v4 = ppcDetValue[iFloorView][iFloorDet+1]; + else + v4 = v1; + if (iFloorView < nView - 1 && iFloorDet < nDet - 1) + v3 = ppcDetValue [iFloorView+1][iFloorDet+1]; + else if (iFloorView < nView - 1) + v3 = v2; + else + v3 = ppcDetValue[0][iFloorDet+1]; + std::complex vInterp = (1 - dFracView) * (1 - dFracDet) * v1 + + dFracView * (1 - dFracDet) * v2 + dFracView * dFracDet * v3 + + dFracDet * (1 - dFracView) * v4; + v[ix][iy] = vInterp.real(); + if (vImag) + vImag[ix][iy] = vInterp.imag(); + } else { + sys_error (ERR_SEVERE, "Can't find projection data for ix=%d,iy=%d with radView=%f and radDet=%f", + ix, iy, ppdView[ix][iy], ppdDet[ix][iy]); + v[ix][iy] = 0; + if (vImag) + vImag[ix][iy] = 0; + } + } else if (iInterpolationID == POLAR_INTERP_BICUBIC) { + v[ix][iy] =0; + if (vImag) + vImag[ix][iy] = 0; + } + } + } +} +