X-Git-Url: http://git.kpe.io/?p=ctsim.git;a=blobdiff_plain;f=libctsim%2Fprojections.cpp;h=49aaa9942b078ea417817adb45789a0de5df26bc;hp=8d3689567550a02ea1a4e670135193410552951a;hb=9703ace3c451ce079967284bf191783736dbc77f;hpb=99dd1d6ed10db1f669a5fe6af71225a50fc0ddfb diff --git a/libctsim/projections.cpp b/libctsim/projections.cpp index 8d36895..49aaa99 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.1 2000/06/19 02:59:34 kevin Exp $ +** $Id: projections.cpp,v 1.29 2000/12/16 02:31:00 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,7 +25,8 @@ ******************************************************************************/ #include "ct.h" - + +const kuint16 Projections::m_signature = ('P'*256 + 'J'); /* NAME * Projections Constructor for projections matrix storage @@ -39,24 +40,20 @@ */ Projections::Projections (const Scanner& scanner) + : m_projData(0) { - init (scanner.nView(), scanner.nDet()); - - m_phmLen = scanner.phmLen(); - m_rotInc = scanner.rotInc(); - m_detInc = scanner.detInc(); - m_rotStart = 0; - m_detStart = -scanner.radius() + (scanner.detInc() / 2); - m_phmLen = scanner.phmLen(); + initFromScanner (scanner); } Projections::Projections (const int nView, const int nDet) + : m_projData(0) { init (nView, nDet); } Projections::Projections (void) + : m_projData(0) { init (0, 0); } @@ -70,13 +67,41 @@ Projections::~Projections (void) void Projections::init (const int nView, const int nDet) { + m_label.setLabelType (Array2dFileLabel::L_HISTORY); m_nView = nView; m_nDet = nDet; - m_projData = NULL; newProjData (); - m_fd = -1; } +void +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(); + m_geometry = scanner.geometry(); + m_focalLength = scanner.focalLength(); + m_fieldOfView = scanner.fieldOfView(); + m_rotStart = 0; + m_detStart = -(scanner.detLen() / 2); +#if 0 + if (m_geometry == Scanner::GEOMETRY_EQUILINEAR) { + m_detInc /= 2; + cout << "Kludge: detInc /= 2 in Projections::initFromScanner" << endl; + } +#endif +} + +void +Projections::setNView (int nView) // used by MPI to reduce # of views +{ + deleteProjData(); + init (nView, m_nDet); +} // NAME // newProjData @@ -118,120 +143,129 @@ Projections::deleteProjData (void) /* NAME - * projections_write_header Write data header for projections file + * Projections::headerWwrite Write data header for projections file * */ bool -Projections::headerWrite (void) +Projections::headerWrite (fnetorderstream& fs) { - kuint32 _hsize = m_headerSize; + kuint16 _hsize = m_headerSize; + kuint16 _signature = m_signature; kuint32 _nView = m_nView; kuint32 _nDet = m_nDet; kuint32 _geom = m_geometry; - kuint32 _remarksize = m_remark.length(); + kuint16 _remarksize = m_remark.length(); + kuint16 _year = m_year; + kuint16 _month = m_month; + kuint16 _day = m_day; + kuint16 _hour = m_hour; + kuint16 _minute = m_minute; + kuint16 _second = m_second; + kfloat64 _calcTime = m_calcTime; kfloat64 _rotStart = m_rotStart; kfloat64 _rotInc = m_rotInc; kfloat64 _detStart = m_detStart; kfloat64 _detInc = m_detInc; kfloat64 _phmLen = m_phmLen; - - if (m_fd < 0) { - sys_error (ERR_SEVERE, "m_fd < 0 [projections_write_header]"); - return false; - } - - if (lseek (m_fd, (off_t) 0, SEEK_SET) != (off_t) 0) { - sys_error (ERR_WARNING, "Error seeking to beginning of fiel [projections_write_header]"); - return false; - } + kfloat64 _fieldOfView = m_fieldOfView; + kfloat64 _focalLength = m_focalLength; - if (! write_nint32 (&_hsize, m_fd) || - ! write_nint32 (&_nView, m_fd) || - ! write_nint32 (&_nDet, m_fd) || - ! write_nint32 (&_geom, m_fd) || - ! write_nfloat64 (&_calcTime, m_fd) || - ! write_nfloat64 (&_rotStart, m_fd) || - ! write_nfloat64 (&_rotInc, m_fd) || - ! write_nfloat64 (&_detStart, m_fd) || - ! write_nfloat64 (&_detInc, m_fd) || - ! write_nfloat64 (&_phmLen, m_fd) || - ! write_nint32 (&_remarksize, m_fd) || - (::write (m_fd, m_remark.c_str(), _remarksize) != _remarksize)) { - sys_error (ERR_SEVERE, "Error writing header information [projections_write_header] %ld", _remarksize); - return false; - } - m_headerSize = _hsize = lseek(m_fd, (off_t) 0, SEEK_CUR); - if ((lseek(m_fd, 0, SEEK_SET) != 0) || ! write_nint32 (&_hsize, m_fd)) { - sys_error (ERR_SEVERE, "Error writing header information [projections_write_header]"); + fs.seekp(0); + if (! fs) return false; - } + + fs.writeInt16 (_hsize); + fs.writeInt16 (_signature); + fs.writeInt32 (_nView); + fs.writeInt32 (_nDet); + fs.writeInt32 (_geom); + fs.writeFloat64 (_calcTime); + fs.writeFloat64 (_rotStart); + fs.writeFloat64 (_rotInc); + fs.writeFloat64 (_detStart); + fs.writeFloat64 (_detInc); + fs.writeFloat64 (_phmLen); + fs.writeFloat64 (_focalLength); + fs.writeFloat64 (_fieldOfView); + fs.writeInt16 (_year); + fs.writeInt16 (_month); + fs.writeInt16 (_day); + fs.writeInt16 (_hour); + fs.writeInt16 (_minute); + 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 true; } - /* NAME * projections_read_header Read data header for projections file * */ - bool -Projections::headerRead (void) +Projections::headerRead (fnetorderstream& fs) { - kuint32 _hsize; - kuint32 _nView; - kuint32 _nDet; - kuint32 _geom; - kuint32 _remarksize = 0; - kfloat64 _calcTime; - kfloat64 _rotStart; - kfloat64 _rotInc; - kfloat64 _detStart; - kfloat64 _detInc; - kfloat64 _phmLen; + kuint16 _hsize, _signature, _year, _month, _day, _hour, _minute, _second, _remarksize = 0; + kuint32 _nView, _nDet, _geom; + kfloat64 _calcTime, _rotStart, _rotInc, _detStart, _detInc, _phmLen, _focalLength, _fieldOfView; - if (m_fd < 0) { - sys_error (ERR_SEVERE, "m_fd < 0 [projections_read_header]"); - return false; - } - - if (lseek (m_fd, (off_t) 0, SEEK_SET) != (off_t) 0) { - sys_error (ERR_WARNING, "Error seeking to beginning of field [projections_read_header]"); - return false; + fs.seekg(0); + if (! fs) + return false; + + fs.readInt16 (_hsize); + fs.readInt16 (_signature); + fs.readInt32 (_nView); + fs.readInt32 (_nDet); + fs.readInt32 (_geom); + fs.readFloat64 (_calcTime); + fs.readFloat64 (_rotStart); + fs.readFloat64 (_rotInc); + fs.readFloat64 (_detStart); + fs.readFloat64 (_detInc); + fs.readFloat64 (_phmLen); + fs.readFloat64 (_focalLength); + fs.readFloat64 (_fieldOfView); + fs.readInt16 (_year); + fs.readInt16 (_month); + fs.readInt16 (_day); + fs.readInt16 (_hour); + 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; } - off_t testPos; - if (! read_nint32 (&_hsize, m_fd)) { - sys_error (ERR_SEVERE, "Error reading header size , _hsize=%d [projections_read_header]", _hsize); + if (_signature != m_signature) { + sys_error (ERR_SEVERE, "File %s does not have a valid projection file signature", m_filename.c_str()); return false; } - if ( ! read_nint32 (&_nView, m_fd) || - ! read_nint32 (&_nDet, m_fd) || - ! read_nint32 (&_geom, m_fd) || - ! read_nfloat64 (&_calcTime, m_fd) || - ! read_nfloat64 (&_rotStart, m_fd) || - ! read_nfloat64 (&_rotInc, m_fd) || - ! read_nfloat64 (&_detStart, m_fd) || - ! read_nfloat64 (&_detInc, m_fd) || - ! read_nfloat64 (&_phmLen, m_fd) || - ! read_nint32 (&_remarksize, m_fd)) - { - sys_error (ERR_SEVERE, "Error reading header information , _remarksize=%d [projections_read_header]", _remarksize); - return false; - } - - char remarkStorage[_remarksize+1]; - if (::read (m_fd, &remarkStorage, _remarksize) != _remarksize) { + + char* pszRemarkStorage = new char [_remarksize+1]; + fs.read (pszRemarkStorage, _remarksize); + if (! fs) { sys_error (ERR_SEVERE, "Error reading remark, _remarksize = %d", _remarksize); return false; } - remarkStorage[_remarksize] = 0; - m_remark = remarkStorage; + pszRemarkStorage[_remarksize] = 0; + m_remark = pszRemarkStorage; + delete pszRemarkStorage; - off_t _hsizeread = lseek(m_fd, 0, SEEK_CUR); - if (_hsizeread != _hsize) { + 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; } @@ -246,66 +280,188 @@ Projections::headerRead (void) m_detStart = _detStart; m_detInc = _detInc; m_phmLen = _phmLen; + m_focalLength = _focalLength; + m_fieldOfView = _fieldOfView; + m_year = _year; + m_month = _month; + m_day = _day; + 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; } +bool +Projections::read (const string& filename) +{ + return read (filename.c_str()); +} + bool Projections::read (const char* filename) { - if ((m_fd = open(filename, OPEN_RDONLY | O_BINARY)) < 0) { - sys_error (ERR_SEVERE, "Error opening projections file %s for input [open_projections]", filename); + frnetorderstream fileRead (filename, ios::in | ios::binary); + m_filename = filename; + + if (! fileRead) return false; - } - if (! headerRead()) { - sys_error (ERR_SEVERE, "Error reading projections file %s [open_projections]", filename); + if (! headerRead (fileRead)) return false; - } deleteProjData (); - newProjData (); + newProjData(); for (int i = 0; i < m_nView; i++) { - if (! detarrayRead (*m_projData[i], i)) + if (! detarrayRead (fileRead, *m_projData[i], i)) break; } - close (m_fd); - m_fd = -1; + fileRead.close(); + return true; +} + + +bool +Projections::copyViewData (const string& filename, ostream& os, int startView, int endView) +{ + return copyViewData (filename.c_str(), os, startView, endView); +} + +bool +Projections::copyViewData (const char* const filename, ostream& os, int startView, int endView) +{ + 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; + if (startView > nView - 1) + startView = nView; + if (endView < 0 || endView > nView - 1) + endView = nView - 1; + + if (startView > endView) { // swap if start > end + 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 (pViewData, sizeView); + os.write (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 string& filename, ostream& os) +{ + return copyHeader (filename.c_str(), os); +} + +bool +Projections::copyHeader (const char* const filename, ostream& os) +{ + frnetorderstream is (filename, ios::in | ios::binary); + kuint16 sizeHeader, signature; + is.readInt16 (sizeHeader); + is.readInt16 (signature); + is.seekg (0); + if (signature != m_signature) { + sys_error (ERR_FATAL, "Illegal signature in projection file %s", filename); + return false; + } + + unsigned char* pHdrData = new unsigned char [sizeHeader]; + is.read (pHdrData, sizeHeader); + if (is.fail()) { + sys_error (ERR_FATAL, "Error reading header"); + return false; + } + + os.write (pHdrData, sizeHeader); + if (os.fail()) { + sys_error (ERR_FATAL, "Error writing header"); + return false; + } return true; } +bool +Projections::write (const string& filename) +{ + return write (filename.c_str()); +} bool Projections::write (const char* filename) { - if ((m_fd = open (filename, O_WRONLY | O_CREAT | O_TRUNC | O_BINARY -#ifndef _WIN32 - , S_IWRITE | S_IREAD | S_IRGRP | S_IROTH + frnetorderstream fs (filename, ios::out | ios::binary | ios::trunc | ios::ate); + m_filename = filename; + if (! fs) { + 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 - )) < 0) { - sys_error (ERR_SEVERE, "Error opening file %s for output [projections_create]", filename); - return false; - } - if (! headerWrite ()) { - sys_error(ERR_SEVERE, "Error writing to file %s [create_projections]", filename); - return false; - } - headerWrite (); + if (! headerWrite (fs)) + return false; if (m_projData != NULL) { for (int i = 0; i < m_nView; i++) { - if (! detarrayWrite (*m_projData[i], i)) + if (! detarrayWrite (fs, *m_projData[i], i)) break; } } + if (! fs) + return false; - close (m_fd); - m_fd = -1; + fs.close(); return true; } @@ -320,7 +476,7 @@ Projections::write (const char* filename) */ bool -Projections::detarrayRead (DetectorArray& darray, const int iview) +Projections::detarrayRead (fnetorderstream& fs, DetectorArray& darray, const int iview) { const int detval_bytes = darray.nDet() * sizeof(kfloat32); const int detheader_bytes = sizeof(kfloat64) /* view_angle */ + sizeof(kint32) /* nDet */; @@ -330,32 +486,20 @@ Projections::detarrayRead (DetectorArray& darray, const int iview) kfloat64 view_angle; kuint32 nDet; - if (m_fd < 0) { - sys_error (ERR_SEVERE, "Tried to read detector array on closed file [detarrayRead]"); - return false; - } - - if (lseek(m_fd, start_data, SEEK_SET) != start_data) { - sys_error (ERR_SEVERE, "Error seeking detectory array [detarrayRead]"); - return false; - } + fs.seekg (start_data); - if (! read_nfloat64 (&view_angle, m_fd) || - ! read_nint32 (&nDet, m_fd)) { - sys_error (ERR_SEVERE, "Error reading detector array [detarrayRead]"); - return false; - } + fs.readFloat64 (view_angle); + fs.readInt32 (nDet); darray.setViewAngle (view_angle); // darray.setNDet ( nDet); - for (int i = 0; i < nDet; i++) { + for (unsigned int i = 0; i < nDet; i++) { kfloat32 detval; - if (! read_nfloat32 (&detval, m_fd)) { - sys_error (ERR_SEVERE, "Error reading detector array [detarrayRead]"); - return false; - } + fs.readFloat32 (detval); detval_ptr[i] = detval; } + if (! fs) + return false; return true; } @@ -376,7 +520,7 @@ Projections::detarrayRead (DetectorArray& darray, const int iview) */ bool -Projections::detarrayWrite (const DetectorArray& darray, const int iview) +Projections::detarrayWrite (fnetorderstream& fs, const DetectorArray& darray, const int iview) { const int detval_bytes = darray.nDet() * sizeof(float); const int detheader_bytes = sizeof(kfloat64) /* view_angle */ + sizeof(kint32) /* nDet */; @@ -386,74 +530,81 @@ Projections::detarrayWrite (const DetectorArray& darray, const int iview) kfloat64 view_angle = darray.viewAngle(); kuint32 nDet = darray.nDet(); - if (m_fd < 0) { - sys_error (ERR_SEVERE, "Tried to write detector array with no file open [detarrayWrite]"); + fs.seekp (start_data); + if (! fs) { + sys_error (ERR_SEVERE, "Error seeking detectory array [detarrayWrite]"); return false; } - if (lseek(m_fd, start_data, SEEK_SET) != start_data) { - sys_error (ERR_SEVERE, "Error seeking detectory array [detarrayWrite]"); - return false; - } + fs.writeFloat64 (view_angle); + fs.writeInt32 (nDet); - if ( ! write_nfloat64 (&view_angle, m_fd)) { - sys_error (ERR_SEVERE, "Error writing detector array [detarrayWrite]"); - return false; - } - if ( ! write_nint32 (&nDet, m_fd)) { - sys_error (ERR_SEVERE, "Error writing detector array [detarrayWrite]"); - return false; - } - for (int i = 0; i < nDet; i++) { - kfloat32 detval = detval_ptr[i]; - if ( ! write_nfloat32 (&detval, m_fd)) { - sys_error (ERR_SEVERE, "Error writing detector array [detarrayWrite]"); - return false; - } + for (unsigned int i = 0; i < nDet; i++) { + kfloat32 detval = detval_ptr[i]; + fs.writeFloat32 (detval); } + if (! fs) + return (false); + return true; } - /* NAME - * prt_projections Print projections data + * printProjectionData Print projections data * * SYNOPSIS - * prt_projections (proj) - * Projections& proj Projection data to be printed + * printProjectionData () */ void -Projections::printProjectionData (void) +Projections::printProjectionData () { - printf("Projections Print\n\n"); - printf("Description: %s\n", m_remark.c_str()); - printf("nView = %d nDet = %d\n", m_nView, m_nDet); - printf("rotStart = %8.4f rotInc = %8.4f\n", m_rotStart, m_rotInc); - printf("detStart = %8.4f detInc = %8.4f\n", m_detStart, m_detInc); + printProjectionData (0, nView() - 1); +} +void +Projections::printProjectionData (int startView, int endView) +{ + printf("Projections Data\n\n"); + printf("Description: %s\n", m_remark.c_str()); + printf("Geometry: %s\n", Scanner::convertGeometryIDToName (m_geometry)); + printf("nView = %8d nDet = %8d\n", m_nView, m_nDet); + printf("focalLength = %8.4f fieldOfView = %8.4f\n", m_focalLength, m_fieldOfView); + printf("rotStart = %8.4f rotInc = %8.4f\n", m_rotStart, m_rotInc); + printf("detStart = %8.4f detInc = %8.4f\n", m_detStart, m_detInc); if (m_projData != NULL) { - for (int ir = 0; ir < m_nView; ir++) { - DetectorValue* detval = m_projData[ir]->detValues(); - for (int id = 0; id < m_projData[ir]->nDet(); id++) - printf("%8.4f ", detval[id]); - printf("\n"); - } + if (startView < 0) + startView = 0; + if (startView > m_nView - 1) + startView = m_nView - 1; + if (endView > m_nView - 1) + endView = m_nView - 1; + for (int ir = startView; ir <= endView - 1; ir++) { + 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("\n"); + } } } void -Projections::printScanInfo (void) const +Projections::printScanInfo (ostringstream& os) const { - printf ("Number of detectors: %d\n", m_nDet); - printf (" Number of views: %d\n", m_nView); - printf (" Remark: %s\n", m_remark.c_str()); - printf (" phmLen: %f\n", m_phmLen); - printf (" detStart: %f\n", m_detStart); - printf (" detInc: %f\n", m_detInc); - printf (" rotStart: %f\n", m_rotStart); - printf (" rotInc: %f\n", m_rotInc); + os << "Number of detectors: " << m_nDet << "\n"; + os << " Number of views: " << m_nView<< "\n"; + os << " Remark: " << m_remark.c_str()<< "\n"; + os << " Geometry: " << Scanner::convertGeometryIDToName (m_geometry)<< "\n"; + os << " Focal Length: " << m_focalLength<< "\n"; + os << " Field Of View: " << m_fieldOfView<< "\n"; + os << " phmLen: " << m_phmLen<< "\n"; + os << " detStart: " << m_detStart<< "\n"; + os << " detInc: " << m_detInc<< "\n"; + os << " rotStart: " << m_rotStart<< "\n"; + os << " rotInc: " << m_rotInc<< "\n"; } +