source: trunk/FACT++/src/Fits.cc@ 10954

// **************************************************************************
/** @class FactFits

@brief FITS writter for the FACT project. 

The FactFits class is able to open, manage and update FITS files. 

The file columns should be given to the class before the file is openned. Once
a file has been created, the structure of its columns cannot be changed. Only
row can be added.

This class relies on the CCfits and CFitsIO packages.

*/
// **************************************************************************
#include "Fits.h"

#include "Time.h"
#include "Converter.h"
#include "MessageImp.h"

//for file stats
#include <sys/stat.h>

using namespace std;
using namespace CCfits;

// --------------------------------------------------------------------------
//
//! This gives a standard variable to the file writter. 
//! This variable should not be related to the DIM service being logged. 
//! @param desc the description of the variable to add
//! @param dataFormat the FITS data format corresponding to the variable to add.
//! @param dataPointer the memory location where the variable is stored
//! @param numDataBytes the number of bytes taken by the variable
//
void Fits::AddStandardColumn(Description& desc, string dataFormat, void* dataPointer, long unsigned int numDataBytes)
{
        //check if entry already exist
        for (vector<Description>::iterator it=fStandardColDesc.begin(); it != fStandardColDesc.end(); it++)
                if (it->name == desc.name)
                        return;
        fStandardColDesc.push_back(desc);
        fStandardFormats.push_back(dataFormat);
        fStandardPointers.push_back(dataPointer);
        fStandardNumBytes.push_back(numDataBytes);
}
// --------------------------------------------------------------------------
//
//! This gives the file writter access to the DIM data
//! @param desc a vector containing the description of all the columns to log
//! @param dataFormat a vector containing the FITS data format of all the columsn to log
//! @param dataPointer the memory location where the DIM data starts
//! @param numDataBytes the number of bytes taken by the DIM data. 
//      
void Fits::InitDataColumns(vector<Description> desc, vector<string>& dataFormat, void* dataPointer, int numDataBytes)
{//we will copy this information here. It duplicates the data, which is not great, but it is the easiest way of doing it right now
        if (desc.size() == dataFormat.size())
        {
                fDataColDesc = desc;
        }
        else
        {
                fDataColDesc.clear();
                for (unsigned int i=0;i<dataFormat.size();i++)
                {
                        ostringstream stt;
                        stt << "Data" << i;
                        fDataColDesc.push_back(Description(stt.str(), "comment", "unit"));
                }
        }               
        fDataFormats = dataFormat;
        fDataPointer = dataPointer;
        fDataNumBytes = numDataBytes;   
}
// --------------------------------------------------------------------------
//
//! This opens the FITS file (after the columns have been passed)
//! @param fileName the filename with complete or relative path of the file to open
//! @param tableName the name of the table that will receive the logged data.
//! @param file a pointer to an existing FITS file. If NULL, file will be opened and managed internally
//! @param fitsCounter a pointer to the integer keeping track of the opened FITS files
//! @param out a pointer to the MessageImp that should be used to log errors
//! @param runNumber the runNumber for which this file is opened. 0 means nightly file.
//
void Fits::Open(const string& fileName, const string& tableName, FITS* file, int* fitsCounter, MessageImp* out, int runNumber)
{               
//      if (fMess)
//              delete fMess;
//      fMess = new MessageImp(out);
    fRunNumber = runNumber;
        fMess = out;
        fFileName = fileName;
        if (file == NULL)
        {
                try
                {
                        fFile = new FITS(fileName, RWmode::Write);
                }
                catch (CCfits::FitsException e)
                {                       
                        ostringstream str;
                        str << "Could not open FITS file " << fileName << " reason: " << e.message();
                        fMess->Error(str);
                        fFile = NULL;
                        return;
                }       
                fOwner = true;
                fNumOpenFitsFiles = fitsCounter;
                (*fNumOpenFitsFiles)++;
        }
        else
        {
                fFile = file;
                fOwner = false;
        }
        //concatenate the standard and data columns
        //do it the inneficient way first: its easier and faster to code.
        vector<string> allNames;
        vector<string> allDataTypes;
        vector<string> allUnits;
        fTotalNumBytes = 0;
        for (unsigned int i=0;i<fStandardColDesc.size();i++)
        {
                allNames.push_back(fStandardColDesc[i].name);
                allDataTypes.push_back(fStandardFormats[i]);
                allUnits.push_back(fStandardColDesc[i].unit);
                fTotalNumBytes += fStandardNumBytes[i];
        }
        //for (int i=static_cast<int>(fDataColDesc.size())-1;i>=0;i--)
        for (int i=0; i< static_cast<int>(fDataColDesc.size()); i++)
        {
                if (fDataColDesc[i].name != "")
                        allNames.push_back(fDataColDesc[i].name);
                else 
                {
                        ostringstream stt;
                        stt << "Data" << i;
                        allNames.push_back(stt.str());
                }
                allDataTypes.push_back(fDataFormats[i]);
                allUnits.push_back(fDataColDesc[i].unit);       
        }
        fTotalNumBytes += fDataNumBytes;
        
        bool updating = false;
        try
        {
                string factTableName = tableName;
                //first, let's check if the table already exist in the file
                vector<string> tryToLoadName;
                tryToLoadName.push_back(factTableName);
                fFile->read(tryToLoadName);
            const multimap< string, CCfits::ExtHDU * >& extMap = fFile->extension();
            if (extMap.find(factTableName) == extMap.end())
            {
                for (multimap<string, CCfits::ExtHDU*>::const_iterator it=extMap.begin(); it!= extMap.end(); it++)
                    fMess->Debug(it->first);
                fTable = fFile->addTable(factTableName, 0, allNames, allDataTypes, allUnits);
            }
            else
            {
                fTable = dynamic_cast<CCfits::Table*>(extMap.find(factTableName)->second);
            }
            if (!fTable)
            {
                fMess->Error("The table " + factTableName + " could not be created nor loaded. skipping it");
                return;
            }
            fTable->makeThisCurrent();
                fCopyBuffer = new unsigned char[fTotalNumBytes]; 
                fNumRows = fTable->rows();
                if (fNumRows !=0)
                {//If the file already existed, then we must load its data to memory before writing to it.
                        BinTable* bTable = dynamic_cast<BinTable*>(fTable);
                        if (!bTable)
                        {
                                fMess->Error("The table " + factTableName + " could not be converted to a binary table. skipping");
                                return;
                        }       
                        //read the table binary data.
                        vector<string> colName;
                        bTable->readData(true, colName);

                        //double check that the data was indeed read from the disk. Go through the fTable instead as colName is empty (yes, it is !)
                        map<string, Column*> cMap = fTable->column();
                        map<string, Column*>::iterator cMapIt;

                        for (cMapIt = cMap.begin(); cMapIt != cMap.end(); cMapIt++)
                        {
                                if (!cMapIt->second->isRead())
                                {
                                        fMess->Error("Column " + cMapIt->first + "Could not be read back from the disk");
                                        return; 
                                }       
                        }
                        updating = true;
                }

        }
        catch(CCfits::FitsException e)
        {
                ostringstream str;
                str << "Could not open or create FITS table " << tableName << " in  file " << fileName << " reason: " << e.message();
                fMess->Error(str);
                fTable = NULL;
        }
                        
        //As requested by Roland, the reference MjD is 0.0
        fRefMjD = 0;//* static_cast<double*>(fStandardPointers[0]);
        if (!updating)
                WriteHeaderKeys();
}
// --------------------------------------------------------------------------
//
//!This writes a single header key in the currently open file.
//!@param name the key
//!@param value the key value
//!@param a comment explaining the meaning of the key
template <typename T>
void Fits::WriteSingleHeaderKey(string name, T value, string comment)
{
        try
        {
                fTable->addKey(name, value, comment);
        }
        catch (CCfits::FitsException e)
        {
                ostringstream str;
                str << "Could not add header keys in file " << fFileName << " reason: " << e.message();
                fMess->Error(str);
        }
}
// --------------------------------------------------------------------------
//
//! This writes the standard header 
//
void Fits::WriteHeaderKeys()
{
        if (!fTable)
                return;
        string name;
        string comment;

        string stringValue;

        WriteSingleHeaderKey("EXTREL", 1.0f, "Release Number");
        WriteSingleHeaderKey("TELESCOP", "FACT", "Telescope that acquired this data");
        WriteSingleHeaderKey("ORIGIN", "ISDC", "Institution that wrote the file");
        WriteSingleHeaderKey("CREATOR", "FACT++ DataLogger", "Program that wrote this file");
        stringValue = Time().GetAsStr();
        stringValue[10]= 'T';
        WriteSingleHeaderKey("DATE", stringValue, "File creation data");
        WriteSingleHeaderKey("TIMESYS", "TT", "Time frame system");
        WriteSingleHeaderKey("TIMEUNIT", "d", "Time unit");
        WriteSingleHeaderKey("TIMEREF", "UTC", "Time reference frame");
        WriteSingleHeaderKey("MJDREF", fRefMjD, "Modified Julian Date of origin");
    WriteSingleHeaderKey("TSTOP", fEndMjD, "Time of the last receied data");
}
// --------------------------------------------------------------------------
//
//! This writes one line of data to the file.
//! @param conv the converter corresponding to the service being logged
//
void Fits::Write(Converter* conv)
{
    //first copy the standard variables to the copy buffer
    int shift = 0;
    for (unsigned int i=0;i<fStandardNumBytes.size();i++)
    {
        const char *charSrc = reinterpret_cast<char*>(fStandardPointers[i]);
        reverse_copy(charSrc, charSrc+fStandardNumBytes[i], &fCopyBuffer[shift]);
        shift += fStandardNumBytes[i];
    }

    try
    {
        //now take care of the DIM data. The Converter is here for that purpose
        conv->ToFits(&fCopyBuffer[shift], fDataPointer, fDataNumBytes);
    }
    catch (const runtime_error &e)
    {
        ostringstream str;
        str << fFileName << ": " << e.what();
        fMess->Error(str);
        return;
    }

    fTable->makeThisCurrent();

    int status(0);
    if (fits_insert_rows(fTable->fitsPointer(), fNumRows, 1, &status))
    {
        ostringstream str;
        str << "Inserting row into " << fFileName << " failed (fits_insert_rows, rc=" << status << ")";
        fMess->Error(str);
        // FIXME: What is a proper action here?
    }

    fNumRows++;

    //the first standard variable is the current MjD
    if (fEndMjD==0)
    {
        const double doubleValue = *reinterpret_cast<double*>(fStandardPointers[0]);
        WriteSingleHeaderKey("TSTART", doubleValue,
                             "Time of the first received data");
    }
    fEndMjD = *reinterpret_cast<double*>(fStandardPointers[0]);

    //data copied to buffer, can write to fits
    fits_write_tblbytes(fFile->fitsPointer(), fNumRows, 1, fTotalNumBytes, fCopyBuffer, &status);
    if (status)
    {
        char text[30];//max length of cfitsio error strings (from doc)
        fits_get_errstatus(status, text);
        ostringstream str;
        str << "Writing FITS row " << fNumRows << " in " << fFileName << ": " << text << " (file_write_tblbytes, rc=" << status << ")";
        fMess->Error(str);
        // FIXME: What is a proper action here?
    }
}

// --------------------------------------------------------------------------
//
//! This closes the currently openned FITS file. 
//! it also updates the header to reflect the time of the last logged row
//      
void Fits::Close() 
{
//WARNING: do NOT delete the table as it gets deleted by the fFile object
//                      if (fTable != NULL)
//                              delete fTable;

        WriteSingleHeaderKey("TSTOP", fEndMjD, "Time of the last receied data");
        if (fFile != NULL && fOwner)
        {
//          fFile->flush();
            delete fFile;
        }
        fFile = NULL;
        if (fCopyBuffer != NULL)
                delete [] fCopyBuffer;
        fCopyBuffer = NULL;
        if (fOwner && fNumOpenFitsFiles != NULL)
                (*fNumOpenFitsFiles)--;
//fMess is the MessageImp part of the dataLogger itself. Thus it should NOT be deleted by the Fits files destructor.
//      if (fMess)
//              delete fMess;
        fMess = NULL;
}

// --------------------------------------------------------------------------
//! Returns the size on the disk of the Fits file being written.
int Fits::GetWrittenSize() const
{
        if (!IsOpen())
                return 0;
                
        struct stat st;
        if (stat(fFileName.c_str(), &st))
                return 0;
        else
                return st.st_size;
}