Event/RawFile/include/RawFile/CgemIndexFile.h

Go to the documentation of this file.

#ifndef _CGEMINDEXFILE_H_
#define _CGEMINDEXFILE_H_
 
#include "RawFile/RawDataCache.h"
#include <cstdint>
#include <string>
#include <vector>
#include <fstream>
 
struct CgemSubFileMeta {
  uint32_t lastL1id;
  uint32_t pos;
};
 
struct CgemEventIndex {
  uint32_t l1id;
  uint32_t pos;
  uint32_t size;
};
 
class CgemIndexFile final {
public:
  const uint32_t kIndexFileMarker{ 0x6c6c7566 };      // "full" index
  const uint32_t kSubIndexFileMarker{ 0x66627573 };   // "subf" (sub-file) index
  const uint32_t kMaxNumFiles{ 1020 };  // total 8k (32 bytes file header + 1020*8 bytes meta)
 
  CgemIndexFile();
  ~CgemIndexFile();
 
  // open an exist index file for reading
  bool open4read( const std::string& file );
 
  // create a new index file for writing
  bool open4write( const std::string& file );
 
  // extract the indices from a raw data file and append to this object
  bool append_from( const std::string& file );
 
  // clear the status and release resources
  void close();
 
  // dump the brief meta-info of the index file
  void show();
  // dump the meta-info of the currently loaded sub-file after loadSubFile()
  void showSubFile();
  void showSubFileRange(const uint32_t start, const uint32_t end);
 
  // load the context of the nth sub-file
  bool loadSubFile(const uint32_t n);
 
  // get the CGEM data with the given L1ID
  uint32_t* getCgemData(const uint32_t l1id);
 
  // getters
  uint32_t numFiles() { return m_numFiles; }
  uint32_t numEvents() { return m_subFile->i_numEvents; }
  CgemSubFileMeta* getMeta() { return m_meta->data<CgemSubFileMeta>(); }
  CgemEventIndex* getIndices() { return m_subFile->i_fidx->data<CgemEventIndex>(); }
 
 
private:
  struct CgemSubFileContext {
    uint32_t      i_numEvents{ 0 };  // number of events in this sub-file
    RawDataCache* i_fidx{ nullptr }; // the event indices of this sub-file
    std::string   i_datafn;          // the related data file name
    std::ifstream i_datafs;          // the related data input stream
 
    CgemSubFileContext() { i_fidx = new RawDataCache(); }
    ~CgemSubFileContext() { delete i_fidx; }
  };
 
  bool                m_isWrite{ false };
  uint32_t            m_numFiles{ 0 };      // in full-file context
  RawDataCache*       m_meta{ nullptr };    // in full-file context
  RawDataCache*       m_data{ nullptr };    // for the CGEM data
  CgemSubFileContext* m_subFile{ nullptr }; // currently avtive sub-file
  std::fstream        m_fs;
  std::vector<CgemSubFileContext*> m_subFiles; // sub-file caches for reading
};
 
#endif