Changeset 10964


Ignore:
Timestamp:
06/09/11 21:16:15 (13 years ago)
Author:
tbretz
Message:
First version of writing data into binary files.
File:
1 edited

Legend:

Unmodified
Added
Removed
  • trunk/FACT++/src/EventBuilderWrapper.h

    r10949 r10964  
    3131    DataFileImp(uint32_t id) : fRunId(id) { }
    3232
    33     virtual bool OpenFile(uint32_t runid, RUN_HEAD* h) = 0;
     33    virtual bool OpenFile(RUN_HEAD* h) = 0;
    3434    virtual bool Write(EVENT *) = 0;
    3535    virtual bool Close(RUN_TAIL * = 0) = 0;
    3636
    3737    uint32_t GetRunId() const { return fRunId; }
    38 };
    39 
    40 
    41 #include "FAD.h"
    42 
    43 class DataFileRaw : public DataFileImp
    44 {
    45 public:
    46     DataFileRaw(uint32_t id) : DataFileImp(id)  { }
    47     ~DataFileRaw() { Close(); }
    48 
    49     virtual bool OpenFile(uint32_t, RUN_HEAD* ) { return true; }
    50     virtual bool Write(EVENT *) { return true; }
    51     virtual bool Close(RUN_TAIL * = 0) { return true; }
    52 };
    53 
    54 
    55 class DataFileFits : public DataFileImp
    56 {
    57     CCfits::FITS*  fFile;        /// The pointer to the CCfits FITS file
    58     CCfits::Table* fTable;       /// The pointer to the CCfits binary table
    59 
    60     uint64_t fNumRows;                ///the number of rows that have been written already to the FITS file.
    61     uint32_t fRoi;                ///the number of rows that have been written already to the FITS file.
    62 
    63 public:
    64     DataFileFits(uint32_t runid) : DataFileImp(runid), fFile(0)
    65     {
    66     }
    67 
    68     // --------------------------------------------------------------------------
    69     //
    70     //! Default destructor
    71     //! The Fits file SHOULD have been closed already, otherwise the informations
    72     //! related to the RUN_TAIL will NOT be written to the file.
    73     //
    74     ~DataFileFits() { Close(); }
    75 
    76     // --------------------------------------------------------------------------
    77     //
    78     //! Add a new column to the vectors storing the column data.
    79     //! @param names the vector of string storing the columns names
    80     //! @param types the vector of string storing the FITS data format
    81     //! @param numElems the number of elements in this column
    82     //! @param type the char describing the FITS data format
    83     //! @param name the name of the particular column to be added.
    84     //
    85     inline void AddColumnEntry(vector<string>& names, vector<string>& types, int numElems, char type, string name)
    86     {
    87         names.push_back(name);
    88 
    89         ostringstream str;
    90         if (numElems != 1)
    91             str << numElems;
    92         str << type;
    93         types.push_back(str.str());
    94     }
    95 
    96     // --------------------------------------------------------------------------
    97     //
    98     //! Writes a single header key entry
    99     //! @param name the name of the key
    100     //! @param value its value
    101     //! @param comment the comment associated to that key
    102     //
    103     //FIXME this function is a duplicate from the class Fits. should we try to merge it ?
    104     template <typename T>
    105     void WriteKey(const string &name, const T &value, const string &comment)
    106     {
    107         try
    108         {
    109             fTable->addKey(name, value, comment);
    110         }
    111         catch (CCfits::FitsException e)
    112         {
    113             ostringstream str;
    114             str << "Could not add header key ";
    115             //TODO pipe the error message somewhere
    116         }
    117     }
    118 
    119     template <typename T>
    120     void WriteKey(const string &name, const int idx, const T &value, const string &comment)
    121     {
    122         ostringstream str;
    123         str << name << idx;
    124 
    125         WriteKey(str.str(), value, comment);
    126     }
    12738
    12839    // --------------------------------------------------------------------------
     
    13344    //! @param extension a string containing the extension to be appened to the file name
    13445    //
    135     string FormFileName(uint32_t runNumber, uint32_t runType, string extension)
     46    string FormFileName(uint32_t runType, string extension)
    13647    {
    13748        //TODO where am I supposed to get the base directory from ?
    13849        //TODO also, for creating subsequent directories, should I use the functions from the dataLogger ?
    13950        string baseDirectory = "./Run";
     51
    14052        ostringstream result;
    141         result << baseDirectory;
    142         result << Time::fmt("/%Y/%m/%d/") << (Time() - boost::posix_time::time_duration(12,0,0));
    143         result << setfill('0') << setw(8) << runNumber;
    144         result << "_001_";
     53//        result << baseDirectory;
     54//        result << Time::fmt("/%Y/%m/%d/") << (Time() - boost::posix_time::time_duration(12,0,0));
     55        result << setfill('0') << setw(8) << fRunId;
     56        result << ".001_";
    14557        switch (runType)
    14658        {
     59        case -1:
     60            result << 'T';
     61            break;
    14762        case 0:
    14863            result << 'D';
     
    15570            break;
    15671        case 3:
    157             result << 'Y';
     72            result << 'N';
    15873            break;
    15974        default:
    160             //TODO pipe this error message to the appropriate error stream
    161             cout << "Error unexpected event" << endl;
     75            result << runType;
    16276        };
    163         result << "_data." << extension;
     77        result << "." << extension;
    16478
    16579        return result.str();
     80    }
     81
     82};
     83
     84
     85#include "FAD.h"
     86
     87class DataFileRaw : public DataFileImp
     88{
     89    ofstream fOut;
     90
     91    off_t fPosTail;
     92
     93    uint32_t fCounter;
     94
     95
     96    // WRITE uint32_t 0xFAC77e1e  (FACT Tele)
     97    // ===
     98    // WRITE uint32_t TYPE(>0)          == 1
     99    // WRITE uint32_t ID(>0)            == 0
     100    // WRITE uint32_t VERSION(>0)       == 1
     101    // WRITE uint32_t LENGTH
     102    // -
     103    // WRITE uint32_t TELESCOPE ID
     104    // WRITE uint32_t RUNID
     105    // ===
     106    // WRITE uint32_t TYPE(>0)          == 2
     107    // WRITE uint32_t ID(>0)            == 0
     108    // WRITE uint32_t VERSION(>0)       == 1
     109    // WRITE uint32_t LENGTH
     110    // -
     111    // WRITE          HEADER
     112    // ===
     113    // [ 40 TIMES
     114    //    WRITE uint32_t TYPE(>0)       == 3
     115    //    WRITE uint32_t ID(>0)         == 0..39
     116    //    WRITE uint32_t VERSION(>0)    == 1
     117    //    WRITE uint32_t LENGTH
     118    //    -
     119    //    WRITE          BOARD-HEADER
     120    // ]
     121    // ===
     122    // WRITE uint32_t TYPE(>0)          == 4
     123    // WRITE uint32_t ID(>0)            == 0
     124    // WRITE uint32_t VERSION(>0)       == 1
     125    // WRITE uint32_t LENGTH
     126    // -
     127    // WRITE          FOOTER (empty)
     128    // ===
     129    // [ N times
     130    //    WRITE uint32_t TYPE(>0)       == 10
     131    //    WRITE uint32_t ID(>0)         == counter
     132    //    WRITE uint32_t VERSION(>0)    == 1
     133    //    WRITE uint32_t LENGTH HEADER
     134    //    -
     135    //    WRITE          HEADER+DATA
     136    // ]
     137    // ===
     138    // WRITE uint32_t TYPE   ==0
     139    // WRITE uint32_t VERSION==0
     140    // WRITE uint32_t LENGTH ==0
     141    // ===
     142    // Go back and write footer
     143
     144public:
     145    DataFileRaw(uint32_t id) : DataFileImp(id)  { }
     146    ~DataFileRaw() { Close(); }
     147
     148    void WriteBlockHeader(uint32_t type, uint32_t ver, uint32_t cnt, uint32_t len)
     149    {
     150        const uint32_t val[4] = { ver, type, cnt, len };
     151
     152        fOut.write(reinterpret_cast<const char*>(val), sizeof(val));
     153    }
     154
     155    template<typename T>
     156        void WriteValue(const T &t)
     157    {
     158        fOut.write(reinterpret_cast<const char*>(&t), sizeof(T));
     159    }
     160
     161    enum
     162    {
     163        kIdentifier = 1,
     164        kRunHeader,
     165        kBoardHeader,
     166        kRunSummary,
     167        kEvent,
     168    };
     169
     170    virtual bool OpenFile(RUN_HEAD *h)
     171    {
     172        const string name = FormFileName(h->RunType, "bin");
     173
     174        errno = 0;
     175        fOut.open(name.c_str(), ios_base::out);
     176        if (!fOut)
     177        {
     178            //ostringstream str;
     179            //str << "Open file " << name << ": " << strerror(errno) << " (errno=" << errno << ")";
     180            //Error(str);
     181
     182            return false;
     183        }
     184
     185        fCounter = 0;
     186
     187        static uint32_t FACT = 0xFAC77e1e;
     188
     189        fOut.write(reinterpret_cast<char*>(&FACT), 4);
     190
     191        WriteBlockHeader(kIdentifier, 1, 0, 8);
     192        WriteValue(uint32_t(0));
     193        WriteValue(GetRunId());
     194
     195        WriteBlockHeader(kRunHeader, 1, 0, sizeof(RUN_HEAD)-sizeof(PEVNT_HEADER*));
     196        fOut.write(reinterpret_cast<char*>(h), sizeof(RUN_HEAD)-sizeof(PEVNT_HEADER*));
     197
     198        for (int i=0; i<40; i++)
     199        {
     200            WriteBlockHeader(kBoardHeader, 1, i, sizeof(PEVNT_HEADER));
     201            fOut.write(reinterpret_cast<char*>(h->FADhead+i), sizeof(PEVNT_HEADER));
     202        }
     203
     204        const vector<char> block(sizeof(uint32_t)+sizeof(RUN_TAIL));
     205        WriteBlockHeader(kRunSummary, 1, 0, block.size());
     206
     207        fPosTail = fOut.tellp();
     208        fOut.write(block.data(), block.size());
     209
     210        if (!fOut)
     211        {
     212            //ostringstream str;
     213            //str << "Open file " << name << ": " << strerror(errno) << " (errno=" << errno << ")";
     214            //Error(str);
     215
     216            return false;
     217        }
     218
     219        return true;
     220    }
     221    virtual bool Write(EVENT *evt)
     222    {
     223        const int sh = sizeof(PEVNT_HEADER)+(NPIX-1)*evt->Roi*2;
     224
     225        WriteBlockHeader(kEvent, 1, fCounter++, sh);
     226        fOut.write(reinterpret_cast<char*>(evt)+2, sh-2);
     227        return true;
     228    }
     229    virtual bool Close(RUN_TAIL *tail= 0)
     230    {
     231        fOut.seekp(fPosTail);
     232
     233        WriteValue(uint32_t(1));
     234        fOut.write(reinterpret_cast<char*>(tail), sizeof(RUN_TAIL));
     235
     236        if (!fOut)
     237        {
     238            //ostringstream str;
     239            //str << "Open file " << name << ": " << strerror(errno) << " (errno=" << errno << ")";
     240            //Error(str);
     241
     242            return false;
     243        }
     244
     245        fOut.close();
     246
     247        if (!fOut)
     248        {
     249            //ostringstream str;
     250            //str << "Open file " << name << ": " << strerror(errno) << " (errno=" << errno << ")";
     251            //Error(str);
     252
     253            return false;
     254        }
     255
     256        return true;
     257    }
     258};
     259
     260
     261class DataFileFits : public DataFileImp
     262{
     263    CCfits::FITS*  fFile;        /// The pointer to the CCfits FITS file
     264    CCfits::Table* fTable;       /// The pointer to the CCfits binary table
     265
     266    uint64_t fNumRows;                ///the number of rows that have been written already to the FITS file.
     267
     268public:
     269    DataFileFits(uint32_t runid) : DataFileImp(runid), fFile(0)
     270    {
     271    }
     272
     273    // --------------------------------------------------------------------------
     274    //
     275    //! Default destructor
     276    //! The Fits file SHOULD have been closed already, otherwise the informations
     277    //! related to the RUN_TAIL will NOT be written to the file.
     278    //
     279    ~DataFileFits() { Close(); }
     280
     281    // --------------------------------------------------------------------------
     282    //
     283    //! Add a new column to the vectors storing the column data.
     284    //! @param names the vector of string storing the columns names
     285    //! @param types the vector of string storing the FITS data format
     286    //! @param numElems the number of elements in this column
     287    //! @param type the char describing the FITS data format
     288    //! @param name the name of the particular column to be added.
     289    //
     290    inline void AddColumnEntry(vector<string>& names, vector<string>& types, int numElems, char type, string name)
     291    {
     292        names.push_back(name);
     293
     294        ostringstream str;
     295        if (numElems != 1)
     296            str << numElems;
     297        str << type;
     298        types.push_back(str.str());
     299    }
     300
     301    // --------------------------------------------------------------------------
     302    //
     303    //! Writes a single header key entry
     304    //! @param name the name of the key
     305    //! @param value its value
     306    //! @param comment the comment associated to that key
     307    //
     308    //FIXME this function is a duplicate from the class Fits. should we try to merge it ?
     309    template <typename T>
     310    void WriteKey(const string &name, const T &value, const string &comment)
     311    {
     312        try
     313        {
     314            fTable->addKey(name, value, comment);
     315        }
     316        catch (CCfits::FitsException e)
     317        {
     318            ostringstream str;
     319            str << "Could not add header key ";
     320            //TODO pipe the error message somewhere
     321        }
     322    }
     323
     324    template <typename T>
     325    void WriteKey(const string &name, const int idx, const T &value, const string &comment)
     326    {
     327        ostringstream str;
     328        str << name << idx;
     329
     330        WriteKey(str.str(), value, comment);
    166331    }
    167332
     
    172337    //! @param h a pointer to the RUN_HEAD structure that contains the informations relative to this run
    173338    //
    174     bool OpenFile(uint32_t /*runid*/, RUN_HEAD* h)
     339    bool OpenFile(RUN_HEAD* h)
    175340    {
    176341        //Form filename, based on runid and run-type
    177         string fileName = FormFileName(h->FADhead[0].runnumber, h->RunType, "fits");
     342        const string fileName = FormFileName(h->RunType, "fits");
    178343
    179344        //create the FITS object
     
    214379        AddColumnEntry(colNames, dataTypes, NPIX*h->Nroi,   'U', "Data");
    215380
    216         fRoi = h->Nroi;
    217 
    218381        //actually create the table
    219382        try
     
    228391            }
    229392        }
    230         catch(CCfits::FitsException e)
     393        catch (const CCfits::FitsException &e)
    231394        {
    232395            ostringstream str;
     
    356519        fNumRows++;
    357520
    358         const int sh = sizeof(PEVNT_HEADER)-1;
     521        const int sh = sizeof(PEVNT_HEADER)+(NPIX-1)*e->Roi*2;
    359522
    360523        // column size pointer
    361524        size_t col = 1;
    362         if (!WriteColumns(col, sh + NPIX*fRoi*2, e))
     525        if (!WriteColumns(col, sh, e))
    363526            return true;
    364527
     
    654817        try
    655818        {
    656             if (!file->OpenFile(runid, h))
     819            if (!file->OpenFile(h))
    657820                return 0;
    658821        }
Note: See TracChangeset for help on using the changeset viewer.