| 1 | #ifndef FACT_Fits
|
|---|
| 2 | #define FACT_Fits
|
|---|
| 3 |
|
|---|
| 4 | #include <CCfits/CCfits>
|
|---|
| 5 | #include <vector>
|
|---|
| 6 |
|
|---|
| 7 | #include "Description.h"
|
|---|
| 8 |
|
|---|
| 9 | class Converter;
|
|---|
| 10 | class MessageImp;
|
|---|
| 11 |
|
|---|
| 12 | using namespace std;
|
|---|
| 13 |
|
|---|
| 14 | class Fits
|
|---|
| 15 | {
|
|---|
| 16 | private:
|
|---|
| 17 | ///The CCfits object to the FITS file
|
|---|
| 18 | CCfits::FITS* fFile;
|
|---|
| 19 | ///Flag indicating whether the FITS object should be managed internally or not.
|
|---|
| 20 | bool fOwner;
|
|---|
| 21 | ///The CCfits Table
|
|---|
| 22 | CCfits::Table* fTable;
|
|---|
| 23 | ///The current number of Rows in the table
|
|---|
| 24 | int fNumRows;
|
|---|
| 25 | ///Name of the "standard", i.e. data found in every fits file
|
|---|
| 26 | ///TODO make these variable static so that they are shared by every object.
|
|---|
| 27 | ///TODO add also a static boolean to initialize these only once
|
|---|
| 28 | vector<Description> fStandardColDesc;
|
|---|
| 29 | ///Format of the standard columns.
|
|---|
| 30 | vector<string> fStandardFormats;
|
|---|
| 31 | ///the pointers to the standard variables
|
|---|
| 32 | vector<void*> fStandardPointers;
|
|---|
| 33 | ///the number of bytes taken by each standard variable
|
|---|
| 34 | vector<int> fStandardNumBytes;
|
|---|
| 35 | ///the vector of data column names
|
|---|
| 36 | vector<Description> fDataColDesc;
|
|---|
| 37 | ///the data format of the data columns
|
|---|
| 38 | vector<string> fDataFormats;
|
|---|
| 39 | ///the pointer to the contiguous memory location where the data is stored (i.e. the dim data pointer)
|
|---|
| 40 | void* fDataPointer;
|
|---|
| 41 | ///the size of the data, in bytes.
|
|---|
| 42 | int fDataNumBytes;
|
|---|
| 43 | ///the copy buffer. Required to put the standard and data variable in contguous memory
|
|---|
| 44 | unsigned char* fCopyBuffer;
|
|---|
| 45 | ///the total number of bytes per FITS row
|
|---|
| 46 | int fTotalNumBytes;
|
|---|
| 47 | ///to keep track of the time of the latest written entry (to update the header when closing the file)
|
|---|
| 48 | double fEndMjD;
|
|---|
| 49 | ///Write the FITS header keys
|
|---|
| 50 | bool WriteHeaderKeys();
|
|---|
| 51 | public:
|
|---|
| 52 | ///Name of the openned file. For querying stats
|
|---|
| 53 | string fFileName;
|
|---|
| 54 | private:
|
|---|
| 55 | ///Keep track of number of opened fits
|
|---|
| 56 | int* fNumOpenFitsFiles;
|
|---|
| 57 | ///were to log the errors
|
|---|
| 58 | MessageImp* fMess;
|
|---|
| 59 | public:
|
|---|
| 60 | ///current run number being logged
|
|---|
| 61 | uint32_t fRunNumber;
|
|---|
| 62 |
|
|---|
| 63 | Fits() : fFile(NULL),
|
|---|
| 64 | fOwner(false),
|
|---|
| 65 | fTable(NULL),
|
|---|
| 66 | fNumRows(0),
|
|---|
| 67 | fDataPointer(NULL),
|
|---|
| 68 | fDataNumBytes(0),
|
|---|
| 69 | fCopyBuffer(NULL),
|
|---|
| 70 | fTotalNumBytes(0),
|
|---|
| 71 | fEndMjD(0.0),
|
|---|
| 72 | fFileName(""),
|
|---|
| 73 | fNumOpenFitsFiles(NULL),
|
|---|
| 74 | fMess(NULL),
|
|---|
| 75 | fRunNumber(0)
|
|---|
| 76 | {}
|
|---|
| 77 |
|
|---|
| 78 | virtual ~Fits()
|
|---|
| 79 | {
|
|---|
| 80 | if (IsOpen())
|
|---|
| 81 | Close();
|
|---|
| 82 | }
|
|---|
| 83 | ///returns wether or not the file is currently open or not
|
|---|
| 84 | bool IsOpen() const {return fFile != NULL;}
|
|---|
| 85 |
|
|---|
| 86 | ///Adds a column that exists in all FITS files
|
|---|
| 87 | void AddStandardColumn(const Description& desc, const string &dataFormat, void* dataPointer, long unsigned int numDataBytes);
|
|---|
| 88 |
|
|---|
| 89 | ///Adds columns specific to the service being logged.
|
|---|
| 90 | void InitDataColumns(const vector<Description> &desc, vector<string>& dataFormat, void* dataPointer, int numDataBytes);
|
|---|
| 91 |
|
|---|
| 92 | ///Opens a FITS file
|
|---|
| 93 | bool Open(const string& fileName, const string& tableName, CCfits::FITS* file, int* fitsCounter, MessageImp* out, int runNumber);//ostream& out);
|
|---|
| 94 |
|
|---|
| 95 | ///Write one line of data. Use the given converter.
|
|---|
| 96 | bool Write(Converter* conv);
|
|---|
| 97 |
|
|---|
| 98 | ///Close the currently opened file.
|
|---|
| 99 | void Close();
|
|---|
| 100 |
|
|---|
| 101 | ///Get the size currently written on the disk
|
|---|
| 102 | int GetWrittenSize() const;
|
|---|
| 103 | private:
|
|---|
| 104 | template <typename T>
|
|---|
| 105 | bool WriteSingleHeaderKey(const string &name, const T &value, const string &comment);
|
|---|
| 106 | CCfits::Table* findSuitableTableInFitsFile(const string& tableName,const vector<string>& allNames, const vector<string>& allDataTypes, const vector<string>& allUnits);
|
|---|
| 107 |
|
|---|
| 108 | };//Fits
|
|---|
| 109 |
|
|---|
| 110 |
|
|---|
| 111 | #endif /*FITS_H_*/
|
|---|
| 112 |
|
|---|