X-Git-Url: http://git.kpe.io/?p=ctsim.git;a=blobdiff_plain;f=include%2Farray2dfile.h;h=22e8b6173401fe6ed3d2dec6dccc4c2211e65af5;hp=6b6c5c25e94b366ed629e7911b46223571a5d6af;hb=7ec2cd66921180a624813dff9f8bac76c6b268cc;hpb=b25dbe4a67721d115636a008d508d6bd25f3bfaa diff --git a/include/array2dfile.h b/include/array2dfile.h index 6b6c5c2..22e8b61 100644 --- a/include/array2dfile.h +++ b/include/array2dfile.h @@ -9,7 +9,7 @@ ** This is part of the CTSim program ** Copyright (C) 1983-2000 Kevin Rosenberg ** -** $Id: array2dfile.h,v 1.5 2000/12/04 02:52:57 kevin Exp $ +** $Id: array2dfile.h,v 1.14 2001/01/01 10:14:34 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 @@ -28,28 +28,30 @@ #ifndef ARRAY2DFILE_H #define ARRAY2DFILE_H -#ifndef WIN32 +#ifndef MSVC #include #endif #include -#include -#include -#include -#include +#include +#include +#include +#include +#include +#include #include #include "ctsupport.h" #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(); @@ -59,7 +61,7 @@ public: ~Array2dFileLabel(); - const string& getLabelString (void) const + const std::string& getLabelString (void) const { return m_strLabel; } kfloat64 getCalcTime (void) const @@ -74,19 +76,19 @@ public: int getLabelType (void) const { return m_labelType; } - string& setLabelString (const char* const str) + std::string& setLabelString (const char* const str) { m_strLabel = str; return (m_strLabel); } - string& setLabelString (const string& str) + 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 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; Array2dFileLabel (const Array2dFileLabel& rhs); @@ -102,31 +104,39 @@ private: kuint16 m_hour; kuint16 m_minute; kuint16 m_second; - string m_strLabel; + std::string m_strLabel; kfloat64 m_calcTime; - mutable string m_strDate; + mutable std::string m_strDate; }; 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); @@ -154,12 +164,27 @@ public: kuint32 ny (void) const { 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 getPixelValueRange (double& pvmin, double& pvmax) const; + void setAxisExtent (double minX, double maxX, double minY, double maxY); void doPixelOffsetScale (double offset, double scale); @@ -167,26 +192,26 @@ public: 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 + const std::string& getFilename (void) const { return m_filename; } - void printLabels (ostream& os) const; + void printLabels (std::ostream& os) const; - typedef vector::iterator labelIterator; - typedef vector::const_iterator constLabelIterator; + typedef std::vector::iterator labelIterator; + typedef std::vector::const_iterator constLabelIterator; protected: - typedef vector labelContainer; + typedef std::vector labelContainer; - static const kuint16 m_signature = ('I'*256+'F'); + static const kuint16 m_signature; kuint16 m_headersize; - string m_filename; + std::string m_filename; kuint16 m_pixelSize; kuint16 m_pixelFormat; @@ -199,8 +224,10 @@ public: kuint32 m_ny; kuint32 m_arraySize; labelContainer m_labels; - kuint16 m_numFileLabels; + kuint16 m_numFileLabels; + kuint16 m_dataType; unsigned char** m_arrayData; + unsigned char** m_imaginaryArrayData; private: void init (void); @@ -219,9 +246,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