1 /*****************************************************************************
4 ** Name: array2dfile.cpp
5 ** Purpose: 2-dimension array file class
6 ** Programmer: Kevin Rosenberg
7 ** Date Started: June 2000
9 ** This is part of the CTSim program
10 ** Copyright (C) 1983-2000 Kevin Rosenberg
12 ** $Id: array2dfile.cpp,v 1.22 2000/12/29 19:30:08 kevin Exp $
14 ** This program is free software; you can redistribute it and/or modify
15 ** it under the terms of the GNU General Public License (version 2) as
16 ** published by the Free Software Foundation.
18 ** This program is distributed in the hope that it will be useful,
19 ** but WITHOUT ANY WARRANTY; without even the implied warranty of
20 ** MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
21 ** GNU General Public License for more details.
23 ** You should have received a copy of the GNU General Public License
24 ** along with this program; if not, write to the Free Software
25 ** Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
26 ******************************************************************************/
28 #include "array2dfile.h"
35 const kuint16 Array2dFile::m_signature = ('I'*256+'F');
37 ///////////////////////////////////////////////////////////////////////////
38 // CLASS IMPLEMENTATION
40 // Name: Array2dFileLabel
41 // Purpose: Labels for Array2dFiles
42 ///////////////////////////////////////////////////////////////////////////
45 Array2dFileLabel::init (void)
48 m_labelType = L_EMPTY;
50 tm* lt = localtime(&t);
55 m_minute = lt->tm_min;
56 m_second = lt->tm_sec;
59 Array2dFileLabel::Array2dFileLabel()
64 Array2dFileLabel::Array2dFileLabel(const char* const str, double ctime)
73 Array2dFileLabel::Array2dFileLabel(const int type, const char* const str, double ctime)
82 Array2dFileLabel::~Array2dFileLabel()
87 Array2dFileLabel::setDateTime (int year, int month, int day, int hour, int minute, int second)
98 Array2dFileLabel::getDateTime (int& year, int& month, int& day, int& hour, int& minute, int& second) const
109 Array2dFileLabel::getDateString (void) const
112 snprintf (szDate, sizeof(szDate), "%2d/%02d/%4d %02d:%02d:%02d",
\r
113 m_month + 1, m_day, m_year + 1900, m_hour, m_minute, m_second);
119 Array2dFileLabel::Array2dFileLabel (const Array2dFileLabel& rhs)
121 m_calcTime = rhs.m_calcTime;
122 m_labelType = rhs.m_labelType;
123 m_strLabel = rhs.m_strLabel;
124 m_year = rhs.m_year; m_month = rhs.m_month; m_day = rhs.m_day;
125 m_hour = rhs.m_hour; m_minute = rhs.m_minute; m_second = rhs.m_second;
129 Array2dFileLabel::operator= (const Array2dFileLabel& rhs)
131 m_calcTime = rhs.m_calcTime;
132 m_labelType = rhs.m_labelType;
133 m_strLabel = rhs.m_strLabel;
134 m_year = rhs.m_year; m_month = rhs.m_month; m_day = rhs.m_day;
135 m_hour = rhs.m_hour; m_minute = rhs.m_minute; m_second = rhs.m_second;
141 Array2dFileLabel::print (std::ostream& os) const
143 if (m_labelType == L_HISTORY) {
144 os << "History: " << std::endl;
145 os << " " << m_strLabel << std::endl;
146 os << " calc time = " << m_calcTime << " secs" << std::endl;
147 os << " Timestamp = " << getDateString() << std::endl;
148 } else if (m_labelType == L_USER) {
149 os << "Note: " << m_strLabel << std::endl;
150 os << " Timestamp = %s" << getDateString() << std::endl;
152 os << "Unknown (" << m_labelType << "): " << m_strLabel << std::endl;
153 os << " Timestamp = %s" << getDateString() << std::endl;
158 ///////////////////////////////////////////////////////////////////////////
159 // CLASS IMPLEMENTATION
162 // Purpose: Array2dFiles
163 ///////////////////////////////////////////////////////////////////////////
166 Array2dFile::Array2dFile (int x, int y, int pixelSize, int pixelFormat, int dataType)
169 setArraySize (x, y, pixelSize, pixelFormat, dataType);
172 Array2dFile::~Array2dFile (void)
175 for (labelIterator l = m_labels.begin(); l != m_labels.end(); l++)
179 Array2dFile::Array2dFile (void)
185 Array2dFile::init (void)
188 m_pixelFormat = PIXEL_INVALID;
189 m_arrayData = NULL;
\r
190 m_imaginaryArrayData = NULL;
\r
191 m_dataType = DATA_TYPE_INVALID;
195 m_axisIncrementKnown = false;
196 m_axisIncrementX = m_axisIncrementY = 0;
197 m_axisExtentKnown = false;
198 m_minX = m_maxX = m_minY = m_maxY = 0;
205 Array2dFile::setArraySize (int x, int y, int pixelSize, int pixelFormat, int dataType)
207 m_pixelSize = pixelSize;
208 m_pixelFormat = pixelFormat;
\r
209 m_dataType = dataType;
214 Array2dFile::setArraySize (int x, int y)
222 Array2dFile::reallocComplexToReal ()
\r
224 if (m_dataType != DATA_TYPE_COMPLEX)
\r
227 freeArray (m_imaginaryArrayData);
\r
228 m_dataType = DATA_TYPE_REAL;
\r
235 Array2dFile::reallocRealToComplex ()
\r
237 if (m_dataType != DATA_TYPE_REAL)
\r
240 allocArray (m_imaginaryArrayData);
\r
241 m_dataType = DATA_TYPE_COMPLEX;
\r
249 Array2dFile::allocArrays ()
252 freeArray (m_arrayData);
\r
253 if (m_imaginaryArrayData)
\r
254 freeArray (m_imaginaryArrayData);
\r
256 allocArray (m_arrayData);
\r
257 if (m_dataType == DATA_TYPE_COMPLEX)
\r
258 allocArray (m_imaginaryArrayData);
\r
262 Array2dFile::allocArray (unsigned char**& rppData)
\r
264 m_arraySize = m_nx * m_ny * m_pixelSize;
265 rppData = new unsigned char* [m_nx];
\r
266 int columnBytes = m_ny * m_pixelSize;
267 for (unsigned int i = 0; i < m_nx; i++)
268 rppData[i] = new unsigned char [columnBytes];
\r
272 Array2dFile::freeArrays ()
275 freeArray (m_arrayData);
\r
277 if (m_imaginaryArrayData)
\r
278 freeArray (m_imaginaryArrayData);
\r
283 Array2dFile::freeArray (unsigned char**& rppData)
\r
285 for (unsigned int i = 0; i < m_nx; i++)
\r
293 Array2dFile::fileWrite (const std::string& filename)
295 return fileWrite (filename.c_str());
299 Array2dFile::fileWrite (const char* const filename)
301 m_filename = filename;
303 frnetorderstream fs (m_filename.c_str(), std::ios::out | std::ios::in | std::ios::trunc | std::ios::binary);
305 sys_error (ERR_WARNING, "Error opening file %s for writing [fileCreate]", m_filename.c_str());
308 if (! headerWrite(fs))
311 if (! arrayDataWrite (fs))
314 if (! labelsWrite (fs))
321 Array2dFile::fileRead (const std::string& filename)
323 return fileRead (filename.c_str());
327 Array2dFile::fileRead (const char* const filename)
329 m_filename = filename;
332 frnetorderstream fs (m_filename.c_str(), std::ios::out | std::ios::in | std::ios::binary);
\r
334 frnetorderstream fs (m_filename.c_str(), std::ios::out | std::ios::in | std::ios::binary | std::ios::nocreate);
\r
337 sys_error (ERR_WARNING, "Unable to open file %s [fileRead]", m_filename.c_str());
341 if (! headerRead(fs))
346 if (! arrayDataRead(fs))
349 if (! labelsRead (fs))
356 Array2dFile::setAxisIncrement (double incX, double incY)
358 m_axisIncrementKnown = true;
359 m_axisIncrementX = incX;
360 m_axisIncrementY = incY;
364 Array2dFile::setAxisExtent (double minX, double maxX, double minY, double maxY)
366 m_axisExtentKnown = true;
374 Array2dFile::headerRead (frnetorderstream& fs)
377 sys_error (ERR_WARNING, "Tried to read header with file closed [headerRead]");
382 kuint16 file_signature;
384 fs.readInt16 (m_headersize);
385 fs.readInt16 (file_signature);
386 fs.readInt16 (m_pixelFormat);
387 fs.readInt16 (m_pixelSize);
388 fs.readInt16 (m_numFileLabels);
390 fs.readInt32 (m_ny);
\r
391 fs.readInt16 (m_dataType);
392 fs.readInt16 (m_axisIncrementKnown);
393 fs.readFloat64 (m_axisIncrementX);
394 fs.readFloat64 (m_axisIncrementY);
395 fs.readInt16 (m_axisExtentKnown);
396 fs.readFloat64 (m_minX);
397 fs.readFloat64 (m_maxX);
398 fs.readFloat64 (m_minY);
399 fs.readFloat64 (m_maxY);
400 fs.readFloat64 (m_offsetPV);
401 fs.readFloat64 (m_scalePV);
403 int read_m_headersize = fs.tellg();
404 if (read_m_headersize != m_headersize) {
405 sys_error (ERR_WARNING, "Read m_headersize %d != file m_headersize %d", read_m_headersize, m_headersize);
408 if (file_signature != m_signature) {
409 sys_error (ERR_WARNING, "File signature %d != true signature %d", file_signature, m_signature);
418 Array2dFile::headerWrite (frnetorderstream& fs)
421 sys_error (ERR_WARNING, "Tried to write header with ! fs");
425 m_numFileLabels = m_labels.size();
428 fs.writeInt16 (m_headersize);
429 fs.writeInt16 (m_signature);
430 fs.writeInt16 (m_pixelFormat);
431 fs.writeInt16 (m_pixelSize);
432 fs.writeInt16 (m_numFileLabels);
433 fs.writeInt32 (m_nx);
434 fs.writeInt32 (m_ny);
\r
435 fs.writeInt16 (m_dataType);
436 fs.writeInt16 (m_axisIncrementKnown);
437 fs.writeFloat64 (m_axisIncrementX);
438 fs.writeFloat64 (m_axisIncrementY);
439 fs.writeInt16 (m_axisExtentKnown);
440 fs.writeFloat64 (m_minX);
441 fs.writeFloat64 (m_maxX);
442 fs.writeFloat64 (m_minY);
443 fs.writeFloat64 (m_maxY);
444 fs.writeFloat64 (m_offsetPV);
445 fs.writeFloat64 (m_scalePV);
447 m_headersize = static_cast<kuint16>(fs.tellp());
449 fs.writeInt16 (m_headersize);
456 Array2dFile::arrayDataWrite (frnetorderstream& fs)
459 sys_error (ERR_WARNING, "Tried to arrayDataWrite with !fs");
466 fs.seekp (m_headersize);
467 int columnSize = m_ny * m_pixelSize;
468 for (unsigned int ix = 0; ix < m_nx; ix++) {
\r
469 unsigned char* ptrColumn = m_arrayData[ix];
\r
470 if (NativeBigEndian()) {
\r
471 for (unsigned int iy = 0; iy < m_ny; iy++) {
\r
472 ConvertReverseNetworkOrder (ptrColumn, m_pixelSize);
\r
473 fs.write (reinterpret_cast<const char*>(ptrColumn), m_pixelSize);
\r
474 ptrColumn += m_pixelSize;
\r
477 fs.write (reinterpret_cast<const char*>(ptrColumn), columnSize);
\r
479 if (m_dataType == DATA_TYPE_COMPLEX) {
\r
480 for (unsigned int ix = 0; ix < m_nx; ix++) {
\r
481 unsigned char* ptrColumn = m_imaginaryArrayData[ix];
\r
482 if (NativeBigEndian()) {
\r
483 for (unsigned int iy = 0; iy < m_ny; iy++) {
\r
484 ConvertReverseNetworkOrder (ptrColumn, m_pixelSize);
\r
485 fs.write (reinterpret_cast<const char*>(ptrColumn), m_pixelSize);
\r
486 ptrColumn += m_pixelSize;
\r
489 fs.write (reinterpret_cast<const char*>(ptrColumn), columnSize);
\r
498 Array2dFile::arrayDataRead (frnetorderstream& fs)
501 sys_error (ERR_WARNING, "Tried to arrayDataRead with ! fs");
508 fs.seekg (m_headersize);
509 int columnSize = m_ny * m_pixelSize;
510 for (unsigned int ix = 0; ix < m_nx; ix++) {
\r
511 unsigned char* ptrColumn = m_arrayData[ix];
\r
512 if (NativeBigEndian()) {
\r
513 for (unsigned int iy = 0; iy < m_ny; iy++) {
\r
514 fs.read (reinterpret_cast<char*>(ptrColumn), m_pixelSize);
\r
515 ConvertReverseNetworkOrder (ptrColumn, m_pixelSize);
\r
516 ptrColumn += m_pixelSize;
\r
519 fs.read (reinterpret_cast<char*>(ptrColumn), columnSize);
\r
521 if (m_dataType == DATA_TYPE_COMPLEX) {
\r
522 for (unsigned int ix = 0; ix < m_nx; ix++) {
\r
523 unsigned char* ptrColumn = m_imaginaryArrayData[ix];
\r
524 if (NativeBigEndian()) {
\r
525 for (unsigned int iy = 0; iy < m_ny; iy++) {
\r
526 fs.read (reinterpret_cast<char*>(ptrColumn), m_pixelSize);
\r
527 ConvertReverseNetworkOrder (ptrColumn, m_pixelSize);
\r
528 ptrColumn += m_pixelSize;
\r
531 fs.read (reinterpret_cast<char*>(ptrColumn), columnSize);
\r
539 Array2dFile::labelsRead (frnetorderstream& fs)
541 off_t pos = m_headersize + m_arraySize;
546 for (int i = 0; i < m_numFileLabels; i++) {
547 kuint16 labelType, year, month, day, hour, minute, second;
550 fs.readInt16 (labelType);
552 fs.readInt16 (month);
555 fs.readInt16 (minute);
556 fs.readInt16 (second);
557 fs.readFloat64 (calcTime);
560 fs.readInt16 (strLength);
561 char* pszLabelStr = new char [strLength+1];
562 fs.read (pszLabelStr, strLength);
563 pszLabelStr[strLength] = 0;
565 Array2dFileLabel* pLabel = new Array2dFileLabel (labelType, pszLabelStr, calcTime);
566 delete pszLabelStr;
\r
568 pLabel->setDateTime (year, month, day, hour, minute, second);
569 m_labels.push_back (pLabel);
\r
576 Array2dFile::labelsWrite (frnetorderstream& fs)
578 off_t pos = m_headersize + m_arraySize;
581 for (constLabelIterator l = m_labels.begin(); l != m_labels.end(); l++) {
582 const Array2dFileLabel& label = **l;
583 kuint16 labelType = label.getLabelType();
584 kfloat64 calcTime = label.getCalcTime();
585 const char* const labelString = label.getLabelString().c_str();
586 int year, month, day, hour, minute, second;
587 kuint16 yearBuf, monthBuf, dayBuf, hourBuf, minuteBuf, secondBuf;
589 label.getDateTime (year, month, day, hour, minute, second);
590 yearBuf = year; monthBuf = month; dayBuf = day;
591 hourBuf = hour; minuteBuf = minute; secondBuf = second;
593 fs.writeInt16 (labelType);
594 fs.writeInt16 (yearBuf);
595 fs.writeInt16 (monthBuf);
596 fs.writeInt16 (dayBuf);
597 fs.writeInt16 (hourBuf);
598 fs.writeInt16 (minuteBuf);
599 fs.writeInt16 (secondBuf);
600 fs.writeFloat64 (calcTime);
601 kuint16 strlength = strlen (labelString);
602 fs.writeInt16 (strlength);
603 fs.write (labelString, strlength);
610 Array2dFile::labelAdd (const char* const lstr, double calc_time)
612 labelAdd (Array2dFileLabel::L_HISTORY, lstr, calc_time);
617 Array2dFile::labelAdd (int type, const char* const lstr, double calc_time)
619 Array2dFileLabel label (type, lstr, calc_time);
626 Array2dFile::labelAdd (const Array2dFileLabel& label)
628 Array2dFileLabel* pLabel = new Array2dFileLabel(label);
630 m_labels.push_back (pLabel);
634 Array2dFile::labelsCopy (Array2dFile& copyFile, const char* const pszId)
639 for (unsigned int i = 0; i < copyFile.getNumLabels(); i++) {
640 Array2dFileLabel l (copyFile.labelGet (i));
641 std::string lstr = l.getLabelString();
643 l.setLabelString (lstr);
649 Array2dFile::arrayDataClear (void)
652 int columnSize = m_ny * m_pixelSize;
\r
653 for (unsigned int ix = 0; ix < m_nx; ix++)
\r
654 memset (m_arrayData[ix], 0, columnSize);
\r
656 if (m_imaginaryArrayData) {
\r
657 int columnSize = m_ny * m_pixelSize;
\r
658 for (unsigned int ix = 0; ix < m_nx; ix++)
\r
659 memset (m_arrayData[ix], 0, columnSize);
\r
664 Array2dFile::printLabels (std::ostream& os) const
666 for (constLabelIterator l = m_labels.begin(); l != m_labels.end(); l++) {
667 const Array2dFileLabel& label = **l;
675 const Array2dFileLabel&
676 Array2dFile::labelGet (int i) const