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.1 2000/06/26 21:15:24 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 ///////////////////////////////////////////////////////////////////////////
36 // CLASS IMPLEMENTATION
38 // Name: Array2dFileLabel
39 // Purpose: Labels for Array2dFiles
40 ///////////////////////////////////////////////////////////////////////////
43 Array2dFileLabel::init (void)
46 m_labelType = L_EMPTY;
48 tm* lt = localtime(&t);
49 m_year = lt->tm_year + 1900;
50 m_month = lt->tm_mon + 1;
53 m_minute = lt->tm_min;
54 m_second = lt->tm_sec;
57 Array2dFileLabel::Array2dFileLabel()
62 Array2dFileLabel::Array2dFileLabel(const char* const str, double ctime = 0.)
71 Array2dFileLabel::Array2dFileLabel(const int type, const char* const str, double ctime = 0.)
80 Array2dFileLabel::~Array2dFileLabel()
85 Array2dFileLabel::setDateTime (int year, int month, int day, int hour, int minute, int second)
96 Array2dFileLabel::getDateTime (int& year, int& month, int& day, int& hour, int& minute, int& second) const
107 Array2dFileLabel::getDateString (void) const
110 oss << m_month <<"/"<< m_day <<"/"<< m_year << " " << m_hour <<":"<< m_minute <<":"<< m_second;
111 m_strDate = oss.str();
116 ///////////////////////////////////////////////////////////////////////////
117 // CLASS IMPLEMENTATION
120 // Purpose: Array2dFiles
121 ///////////////////////////////////////////////////////////////////////////
124 Array2dFile::Array2dFile (int x, int y, int pixelSize, int pixelFormat)
127 setArraySize (x, y, pixelSize, pixelFormat);
130 Array2dFile::~Array2dFile (void)
133 for (labelIterator l = m_labels.begin(); l != m_labels.end(); l++)
137 Array2dFile::Array2dFile (void)
143 Array2dFile::init (void)
146 m_pixelFormat = PIXEL_INVALID;
151 m_axisIncrementKnown = false;
152 m_axisIncrementX = m_axisIncrementY = 0;
153 m_axisExtentKnown = false;
154 m_minX = m_maxX = m_minY = m_maxY = 0;
161 Array2dFile::setArraySize (int x, int y, int pixelSize, int pixelFormat)
165 m_pixelSize = pixelSize;
166 m_arraySize = m_nx * m_ny * m_pixelSize;
167 m_pixelFormat = pixelFormat;
172 Array2dFile::setArraySize (int x, int y)
180 Array2dFile::allocArray (void)
185 m_arraySize = m_nx * m_ny * m_pixelSize;
186 m_arrayData = new unsigned char* [m_nx];
188 int columnBytes = m_ny * m_pixelSize;
189 for (int i = 0; i < m_nx; i++)
190 m_arrayData[i] = new unsigned char [columnBytes];
194 Array2dFile::freeArray (void)
197 for (int i = 0; i < m_nx; i++)
198 delete m_arrayData[i];
205 Array2dFile::fileWrite (const char* const filename)
207 m_filename = filename;
209 frnetorderstream fs (m_filename.c_str(), ios::out | ios::in | ios::trunc | ios::binary);
211 sys_error (ERR_WARNING, "Error opening file %s for writing [fileCreate]", m_filename.c_str());
214 if (! headerWrite(fs))
217 if (! arrayDataWrite (fs))
220 if (! labelsWrite (fs))
227 Array2dFile::fileRead (const char* const filename)
229 m_filename = filename;
231 frnetorderstream fs (m_filename.c_str(), ios::out | ios::in | ios::binary | ios::nocreate);
233 sys_error (ERR_WARNING, "Unable to open file %s [fileRead]", m_filename.c_str());
237 if (! headerRead(fs))
242 if (! arrayDataRead(fs))
245 if (! labelsRead (fs))
252 Array2dFile::setAxisIncrement (double incX, double incY)
254 m_axisIncrementKnown = true;
255 m_axisIncrementX = incX;
256 m_axisIncrementY = incY;
260 Array2dFile::setAxisExtent (double minX, double maxX, double minY, double maxY)
262 m_axisExtentKnown = true;
270 Array2dFile::headerRead (frnetorderstream& fs)
273 sys_error (ERR_WARNING, "Tried to read header with file closed [headerRead]");
278 kuint16 file_signature;
280 fs.readInt16 (m_headersize);
281 fs.readInt16 (file_signature);
282 fs.readInt16 (m_pixelFormat);
283 fs.readInt16 (m_pixelSize);
284 fs.readInt16 (m_numFileLabels);
287 fs.readInt16 (m_axisIncrementKnown);
288 fs.readFloat64 (m_axisIncrementX);
289 fs.readFloat64 (m_axisIncrementY);
290 fs.readInt16 (m_axisExtentKnown);
291 fs.readFloat64 (m_minX);
292 fs.readFloat64 (m_maxX);
293 fs.readFloat64 (m_minY);
294 fs.readFloat64 (m_maxY);
295 fs.readFloat64 (m_offsetPV);
296 fs.readFloat64 (m_scalePV);
298 int read_m_headersize = fs.tellg();
299 if (read_m_headersize != m_headersize) {
300 sys_error (ERR_WARNING, "Read m_headersize %d != file m_headersize %d", read_m_headersize, m_headersize);
303 if (file_signature != m_signature) {
304 sys_error (ERR_WARNING, "File signature %d != true signature %d", file_signature, m_signature);
313 Array2dFile::headerWrite (frnetorderstream& fs)
316 sys_error (ERR_WARNING, "Tried to write header with ! fs");
320 m_numFileLabels = m_labels.size();
323 fs.writeInt16 (m_headersize);
324 fs.writeInt16 (m_signature);
325 fs.writeInt16 (m_pixelFormat);
326 fs.writeInt16 (m_pixelSize);
327 fs.writeInt16 (m_numFileLabels);
328 fs.writeInt32 (m_nx);
329 fs.writeInt32 (m_ny);
330 fs.writeInt16 (m_axisIncrementKnown);
331 fs.writeFloat64 (m_axisIncrementX);
332 fs.writeFloat64 (m_axisIncrementY);
333 fs.writeInt16 (m_axisExtentKnown);
334 fs.writeFloat64 (m_minX);
335 fs.writeFloat64 (m_maxX);
336 fs.writeFloat64 (m_minY);
337 fs.writeFloat64 (m_maxY);
338 fs.writeFloat64 (m_offsetPV);
339 fs.writeFloat64 (m_scalePV);
341 m_headersize = fs.tellp();
343 fs.writeInt16 (m_headersize);
350 Array2dFile::arrayDataWrite (frnetorderstream& fs)
353 sys_error (ERR_WARNING, "Tried to arrayDataWrite with !fs");
360 fs.seekp (m_headersize);
361 int columnSize = m_ny * m_pixelSize;
362 for (unsigned int ix = 0; ix < m_nx; ix++) {
363 unsigned char* ptrColumn = m_arrayData[ix];
364 if (NativeBigEndian()) {
365 for (unsigned int iy = 0; iy < m_ny; iy++) {
366 ConvertReverseNetworkOrder (ptrColumn, m_pixelSize);
367 fs.write (ptrColumn, m_pixelSize);
368 ptrColumn += m_pixelSize;
371 fs.write (ptrColumn, columnSize);
379 Array2dFile::arrayDataRead (frnetorderstream& fs)
382 sys_error (ERR_WARNING, "Tried to arrayDataRead with ! fs");
389 fs.seekg (m_headersize);
390 int columnSize = m_ny * m_pixelSize;
391 for (int ix = 0; ix < m_nx; ix++) {
392 unsigned char* ptrColumn = m_arrayData[ix];
393 if (NativeBigEndian()) {
394 for (unsigned int iy = 0; iy < m_ny; iy++) {
395 fs.read (ptrColumn, m_pixelSize);
396 ConvertReverseNetworkOrder (ptrColumn, m_pixelSize);
397 ptrColumn += m_pixelSize;
400 fs.read (ptrColumn, columnSize);
407 Array2dFile::labelsRead (frnetorderstream& fs)
409 off_t pos = m_headersize + m_arraySize;
414 for (int i = 0; i < m_numFileLabels; i++) {
415 kuint16 labelType, year, month, day, hour, minute, second;
418 fs.readInt16 (labelType);
420 fs.readInt16 (month);
423 fs.readInt16 (minute);
424 fs.readInt16 (second);
425 fs.readFloat64 (calcTime);
428 fs.readInt16 (strLength);
429 char labelStr [strLength+1];
430 fs.read (labelStr, strLength);
431 labelStr[strLength] = 0;
433 Array2dFileLabel* pLabel = new Array2dFileLabel(labelType, labelStr, calcTime);
434 pLabel->setDateTime (year, month, day, hour, minute, second);
435 m_labels.push_back (pLabel);
442 Array2dFile::labelsWrite (frnetorderstream& fs)
444 off_t pos = m_headersize + m_arraySize;
447 for (constLabelIterator l = m_labels.begin(); l != m_labels.end(); l++) {
448 const Array2dFileLabel& label = **l;
449 kuint16 labelType = label.getLabelType();
450 kfloat64 calcTime = label.getCalcTime();
451 const char* const labelString = label.getLabelString().c_str();
452 int year, month, day, hour, minute, second;
453 kuint16 yearBuf, monthBuf, dayBuf, hourBuf, minuteBuf, secondBuf;
455 label.getDateTime (year, month, day, hour, minute, second);
456 yearBuf = year; monthBuf = month; dayBuf = day;
457 hourBuf = hour; minuteBuf = minute; secondBuf = second;
459 fs.writeInt16 (labelType);
460 fs.writeInt16 (yearBuf);
461 fs.writeInt16 (monthBuf);
462 fs.writeInt16 (dayBuf);
463 fs.writeInt16 (hourBuf);
464 fs.writeInt16 (minuteBuf);
465 fs.writeInt16 (secondBuf);
466 fs.writeFloat64 (calcTime);
467 kuint16 strlength = strlen (labelString);
468 fs.writeInt16 (strlength);
469 fs.write (labelString, strlength);
476 Array2dFile::labelAdd (const char* const lstr, double calc_time=0.)
478 labelAdd (Array2dFileLabel::L_HISTORY, lstr, calc_time);
483 Array2dFile::labelAdd (int type, const char* const lstr, double calc_time=0.)
485 Array2dFileLabel label (type, lstr, calc_time);
492 Array2dFile::labelAdd (const Array2dFileLabel& label)
494 Array2dFileLabel* pLabel = new Array2dFileLabel(label);
496 m_labels.push_back (pLabel);
500 Array2dFile::labelsCopy (Array2dFile& copyFile, const char* const idStr)
503 for (int i = 0; i < copyFile.getNumLabels(); i++) {
504 Array2dFileLabel l = copyFile.labelGet (i);
505 string lstr = l.getLabelString();
507 l.setLabelString (lstr);
513 Array2dFile::arrayDataClear (void)
516 int columnSize = m_ny * m_pixelSize;
517 for (int ix = 0; ix < m_nx; ix++)
518 memset (m_arrayData[ix], 0, columnSize);
523 Array2dFile::printLabels (ostream& os) const
525 int nlabels = getNumLabels();
527 for (constLabelIterator l = m_labels.begin(); l != m_labels.end(); l++) {
528 const Array2dFileLabel& label = **l;
530 if (label.getLabelType() == Array2dFileLabel::L_HISTORY) {
531 os << "History: " << endl;
532 os << " " << label.getLabelString() << endl;
533 os << " calc time = " << label.getCalcTime() << " secs" << endl;
534 os << " Timestamp = " << label.getDateString() << endl;
535 } else if (label.getLabelType() == Array2dFileLabel::L_USER) {
536 os << "Note: " << label.getLabelString() << endl;
537 os << " Timestamp = %s" << label.getDateString() << endl;
544 const Array2dFileLabel&
545 Array2dFile::labelGet (int i) const