storage_manager.hh

Go to the documentation of this file.

#ifndef STORAGE_MANAGER_HH
#define STORAGE_MANAGER_HH
#include <set>
#include <TFile.h>
#include <TChain.h>

#include "decoder_base.hh"
#include "trig_info.hh"
#include "pulse_info.hh"
#include "user_info.hh"
#include "pmt_waveform.hh"
#include "tpc_waveform.hh"

class storage_manager : public decoder_base {

public:

  enum MODE{
    READ,       
    WRITE,      
    BOTH,       
    UNDEFINED   
  };

  enum STATUS{
    INIT,       
    OPENED,     
    READY_IO,   
    OPERATING,  
    CLOSED      
  };

  storage_manager(MODE mode=UNDEFINED);

  virtual ~storage_manager(){};

  void set_data_to_read(DATA_STRUCT::DATA_TYPE type,bool read=true)
  {_read_data_array[type]=read;};

  void set_data_to_write(DATA_STRUCT::DATA_TYPE type,bool write=true)
  {_write_data_array[type]=write;};

  void set_io_mode(MODE mode)         {_mode=mode;};

  //void set_in_filename(std::string name) {_in_fname=name;};
  void add_in_filename(std::string name) {_in_fnames.push_back(name);};

  void set_out_filename(std::string name) {_out_fname=name;};

  MODE io_mode() const         {return _mode;};

  std::vector<std::string> input_filename() const {return _in_fnames;};

  std::string output_filename() const {return _out_fname;};

  STATUS status() const        {return _status;};

  void reset();

  bool open();  

  bool is_open();

  bool is_ready_io();

  bool next_event(); 

  bool go_to(UInt_t index);

  bool close(); 

  data_base* get_data(DATA_STRUCT::DATA_TYPE type);

  static storage_manager* get() 
  { if(!me) me= new storage_manager(); return me; };
  
  static void kill() 
  { if(me) delete me;};

  inline UInt_t get_entries() const {return _nevents;};

  inline UInt_t get_index()   const {return _index;};

  inline UInt_t get_entries_read() const {return _nevents_read;};

  inline UInt_t get_entires_written() const {return _nevents_written;};
  
private:

  static storage_manager* me; 

  void create_data_ptr(DATA_STRUCT::DATA_TYPE type);

  void delete_data_ptr(DATA_STRUCT::DATA_TYPE type);

  bool read_event();
  
  bool write_event();

  bool prepare_tree();

  UInt_t _index;

  UInt_t _nevents, _nevents_read, _nevents_written;

  STATUS _status;

  MODE _mode;

  bool _read_data_array[DATA_STRUCT::DATA_TYPE_MAX];

  bool _write_data_array[DATA_STRUCT::DATA_TYPE_MAX];

  data_base *_ptr_data_array[DATA_STRUCT::DATA_TYPE_MAX];

  std::string _out_fname;
  std::vector<std::string> _in_fnames;

  const std::string _treename;

  //TFile *_fin, *_fout;
  TFile *_fout;

  //TTree *_in_tree, *_out_tree;  
  TChain *_in_ch[DATA_STRUCT::DATA_TYPE_MAX];
  TTree  *_out_ch[DATA_STRUCT::DATA_TYPE_MAX];

};

#endif
 // end of doxygen group storage_manager

---