X-Git-Url: http://git.kpe.io/?p=ctsim.git;a=blobdiff_plain;f=include%2Farray2dfile.h;h=8569d49025ba712c0dcaf34ca0189f0cab2d7b8f;hp=9ff8dca58196da35e9f3e7a13a3be0b38f583799;hb=f13a8c004b8f182b42d9e4df2bcd7c7f030bf1ad;hpb=6850134e4711a842cc7c0a306a9e4243df06b952 diff --git a/include/array2dfile.h b/include/array2dfile.h index 9ff8dca..8569d49 100644 --- a/include/array2dfile.h +++ b/include/array2dfile.h @@ -1,15 +1,13 @@ /***************************************************************************** ** FILE IDENTIFICATION ** -** Name: array2dfile.h +** Name: array2dfile.h ** Purpose: 2-dimension array file class -** Programmer: Kevin Rosenberg -** Date Started: June 2000 +** Programmer: Kevin Rosenberg +** Date Started: June 2000 ** ** This is part of the CTSim program -** Copyright (C) 1983-2000 Kevin Rosenberg -** -** $Id: array2dfile.h,v 1.4 2000/07/09 08:16:17 kevin Exp $ +** Copyright (c) 1983-2009 Kevin Rosenberg ** ** 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 @@ -28,10 +26,15 @@ #ifndef ARRAY2DFILE_H #define ARRAY2DFILE_H -#include -#include +#ifndef MSVC #include +#endif + +#include +#include #include +#include +#include #include #include #include @@ -39,96 +42,106 @@ #include "fnetorderstream.h" #include "array2d.h" -using namespace std; - class Array2dFileLabel { public: - static const int L_EMPTY = 0; - static const int L_HISTORY = 1; - static const int L_USER = 2; + enum { + L_EMPTY = 0, + L_HISTORY = 1, + L_USER = 2, + }; - Array2dFileLabel(); + Array2dFileLabel(); - Array2dFileLabel(const char* const str, double ctime = 0.); + Array2dFileLabel(const char* const str, double ctime = 0.); - Array2dFileLabel(const int type, const char* const str, double ctime = 0.); + Array2dFileLabel(const int type, const char* const str, double ctime = 0.); - ~Array2dFileLabel(); + ~Array2dFileLabel(); - const string& getLabelString (void) const - { return m_strLabel; } + const std::string& getLabelString (void) const + { return m_strLabel; } - kfloat64 getCalcTime (void) const - { return m_calcTime; } + kfloat64 getCalcTime (void) const + { return m_calcTime; } - void setCalcTime (kfloat64 calcTime) - { m_calcTime = calcTime; } + void setCalcTime (kfloat64 calcTime) + { m_calcTime = calcTime; } - void setLabelType (int labelType) - { m_labelType = labelType; } + void setLabelType (int labelType) + { m_labelType = labelType; } - int getLabelType (void) const - { return m_labelType; } + int getLabelType (void) const + { return m_labelType; } - string& setLabelString (const char* const str) - { m_strLabel = str; return (m_strLabel); } + std::string& setLabelString (const char* const str) + { m_strLabel = str; return (m_strLabel); } - string& setLabelString (const string& str) - { m_strLabel = str; return (m_strLabel); } + std::string& setLabelString (const std::string& str) + { m_strLabel = str; return (m_strLabel); } - void setDateTime (int year, int month, int day, int hour, int minute, int second); + void setDateTime (int year, int month, int day, int hour, int minute, int second); - void getDateTime (int& year, int& month, int& day, int& hour, int& minute, int& second) const; + void getDateTime (int& year, int& month, int& day, int& hour, int& minute, int& second) const; - const string& getDateString () const; + const std::string& getDateString () const; - void print (ostream& os) const; + void print (std::ostream& os) const; + void printBrief (std::ostream& os) const; + void printBrief (std::ostringstream& os) const; - Array2dFileLabel (const Array2dFileLabel& rhs); + Array2dFileLabel (const Array2dFileLabel& rhs); - Array2dFileLabel& operator= (const Array2dFileLabel& rhs); + Array2dFileLabel& operator= (const Array2dFileLabel& rhs); private: - void init (void); - - kuint16 m_labelType; - kuint16 m_year; - kuint16 m_month; - kuint16 m_day; - kuint16 m_hour; - kuint16 m_minute; - kuint16 m_second; - string m_strLabel; - kfloat64 m_calcTime; - - mutable string m_strDate; + void init (void); + + kuint16 m_labelType; + kuint16 m_year; + kuint16 m_month; + kuint16 m_day; + kuint16 m_hour; + kuint16 m_minute; + kuint16 m_second; + std::string m_strLabel; + kfloat64 m_calcTime; + + mutable std::string m_strDate; }; -class Array2dFile +class Array2dFile { public: - static const int PIXEL_INVALID = 0; - static const int PIXEL_INT8 = 1; - static const int PIXEL_UINT8 = 2; - static const int PIXEL_INT16 = 3; - static const int PIXEL_UINT16 = 4; - static const int PIXEL_INT32 = 5; - static const int PIXEL_UINT32 = 6; - static const int PIXEL_FLOAT32 = 7; - static const int PIXEL_FLOAT64 = 8; - - Array2dFile (int nx, int ny, int pixelSize, int pixelFormat = PIXEL_INVALID); + enum { + PIXEL_INVALID = 0, + PIXEL_INT8 = 1, + PIXEL_UINT8 = 2, + PIXEL_INT16 = 3, + PIXEL_UINT16 = 4, + PIXEL_INT32 = 5, + PIXEL_UINT32 = 6, + PIXEL_FLOAT32 = 7, + PIXEL_FLOAT64 = 8, + }; + + enum { + DATA_TYPE_INVALID = 0, + DATA_TYPE_REAL, + DATA_TYPE_COMPLEX, + }; + + Array2dFile (int nx, int ny, int pixelSize, int pixelFormat = PIXEL_INVALID, int dataType = DATA_TYPE_REAL); Array2dFile (void); ~Array2dFile (); - void setArraySize (int nx, int ny, int pixelSize, int pixelFormat = PIXEL_INVALID); + void setArraySize (int nx, int ny, int pixelSize, int pixelFormat = PIXEL_INVALID, int dataType = DATA_TYPE_REAL); void setArraySize (int nx, int ny); unsigned int getNumLabels (void) const - { return m_labels.size(); } + { return m_labels.size(); } const Array2dFileLabel& labelGet (int label_num) const; @@ -138,66 +151,94 @@ public: void labelAdd (int type, const char* const m_strLabel, double calc_time=0.); - void labelsCopy (Array2dFile& file, const char* const idStr = NULL); + void labelsCopy (const Array2dFile& file, const char* const idStr = NULL); void setPixelFormat (int type) - { m_pixelFormat = type; } + { m_pixelFormat = type; } void setPixelSize (int size) - { m_pixelSize = size; } + { m_pixelSize = size; } kuint32 nx (void) const - { return m_nx; } + { return m_nx; } kuint32 ny (void) const - { return m_ny; } + { return m_ny; } + + bool isComplex() const + { return m_dataType == DATA_TYPE_COMPLEX; } + + bool isReal() const + { return m_dataType == DATA_TYPE_REAL; } + + int dataType () const + { return static_cast(m_dataType); } + + void setDataType (int dataType) + { m_dataType = dataType; } void setAxisIncrement (double axisIncX, double axisIncY); - void setAxisExtent (double minX, double maxX, double minY, double maxY); + bool reallocRealToComplex (); + + bool reallocComplexToReal (); void getPixelValueRange (double& pvmin, double& pvmax) const; - + void setAxisExtent (double minX, double maxX, double minY, double maxY); + bool getAxisExtent (double& minX, double& maxX, double& minY, double& maxY) const + { if (! m_axisExtentKnown) return false; minX = m_minX; maxX = m_maxX; minY = m_minY; maxY=m_maxY; + return true; } + void doPixelOffsetScale (double offset, double scale); + kfloat64 axisIncrementX() const {return m_axisIncrementKnown ? m_axisIncrementX : 0.;} + kfloat64 axisIncrementY() const {return m_axisIncrementKnown ? m_axisIncrementY : 0.;} + void arrayDataClear (void); bool fileRead (const char* const filename); - bool fileRead (const string& filename); + bool fileRead (const std::string& filename); bool fileWrite (const char* const filename); - bool fileWrite (const string& filename); + bool fileWrite (const std::string& filename); - const string& getFilename (void) const - { return m_filename; } + const std::string& getFilename (void) const + { return m_filename; } - void printLabels (ostream& os) const; + void printLabels (std::ostream& os) const; + void printLabelsBrief (std::ostream& os) const; + void printLabelsBrief (std::ostringstream& os) const; - typedef vector::iterator labelIterator; - typedef vector::const_iterator constLabelIterator; + unsigned int nLabels() const + { return m_labels.size(); } - protected: - typedef vector labelContainer; + typedef std::vector::iterator labelIterator; + typedef std::vector::const_iterator constLabelIterator; - static const kuint16 m_signature = ('I'*256+'F'); - kuint16 m_headersize; - string m_filename; +protected: + typedef std::vector labelContainer; - kuint16 m_pixelSize; - kuint16 m_pixelFormat; - kuint16 m_axisIncrementKnown; - kfloat64 m_axisIncrementX, m_axisIncrementY; - kuint16 m_axisExtentKnown; - kfloat64 m_minX, m_maxX, m_minY, m_maxY; - kfloat64 m_offsetPV, m_scalePV; - kuint32 m_nx; - kuint32 m_ny; - kuint32 m_arraySize; - labelContainer m_labels; - kuint16 m_numFileLabels; - unsigned char** m_arrayData; + static const kuint16 m_signature; + kuint16 m_headersize; + std::string m_filename; + + kuint16 m_pixelSize; + kuint16 m_pixelFormat; + kuint16 m_axisIncrementKnown; + kfloat64 m_axisIncrementX, m_axisIncrementY; + kuint16 m_axisExtentKnown; + kfloat64 m_minX, m_maxX, m_minY, m_maxY; + kfloat64 m_offsetPV, m_scalePV; + kuint32 m_nx; + kuint32 m_ny; + kuint32 m_arraySize; + labelContainer m_labels; + kuint16 m_numFileLabels; + kuint16 m_dataType; + unsigned char** m_arrayData; + unsigned char** m_imaginaryArrayData; private: void init (void); @@ -216,9 +257,11 @@ private: bool labelSeek (int label_num); - void allocArray (void); - - void freeArray (void); + void allocArrays (); + void freeArrays (); + + void allocArray (unsigned char**& rppData); + void freeArray (unsigned char**& rppData); Array2dFile (const Array2dFile& rhs); // copy constructor Array2dFile& operator= (const Array2dFile&); // assignment operator