DoEmbedding.h

Go to the documentation of this file.

#ifndef _DO_EMBEDDING__H_
#define _DO_EMBEDDING__H_
#include <fun4all/SubsysReco.h>
class TFile;
class TTree;
class SQEvent;
class SQMCEvent;
class SQHitVector;
class SQTrackVector;
class SQDimuonVector;
 
class DoEmbedding: public SubsysReco {
  bool m_overwrite_rf_info;
  int m_hit_id_shift;
  int m_trk_id_shift;
  int m_dim_id_shift;
 
  SQEvent       * mi_evt;
  SQHitVector   * mi_vec_hit;
  SQMCEvent     * mi_sim_evt;
  SQTrackVector * mi_sim_vec_trk;
  SQDimuonVector* mi_sim_vec_dim;
  SQEvent       * mi_evt_emb; 
  SQMCEvent     * mi_sim_evt_emb; 
 
  std::vector<std::string> m_list_emb_file;
  int m_idx_emb_file;
  int m_idx_emb_evt;
  bool m_emb_data_has_sim_evt;
  bool m_emb_data_has_sim_trk;
  bool m_emb_data_has_sim_dim;
 
  TFile* m_file_emb;
  TTree* m_tree_emb;
  SQEvent*        m_emb_sqevt;
  SQMCEvent*      m_emb_sqmcevt;
  SQHitVector*    m_emb_sqvec_hit;
  SQTrackVector*  m_emb_sqvec_trk;
  SQDimuonVector* m_emb_sqvec_dim;
 
 public:
  DoEmbedding(const std::string name="DoEmbedding");
  virtual ~DoEmbedding();
  int Init(PHCompositeNode *topNode);
  int InitRun(PHCompositeNode *topNode);
  int process_event(PHCompositeNode *topNode);
  int End(PHCompositeNode *topNode);
 
  void DoOverwriteRfInfo(const bool overwrite) { m_overwrite_rf_info = overwrite; }
  bool DoOverwriteRfInfo() const        { return m_overwrite_rf_info; }
  void SetHitIdShift(const int id_shift) { m_hit_id_shift = id_shift; }
  int  GetHitIdShift() const      { return m_hit_id_shift; }
  void SetTrackIdShift(const int id_shift) { m_trk_id_shift = id_shift; }
  int  GetTrackIdShift() const      { return m_trk_id_shift; }
  void SetDimuonIdShift(const int id_shift) { m_dim_id_shift = id_shift; }
  int  GetDimuonIdShift() const      { return m_dim_id_shift; }
 
  void AddEmbDataFile(const char* fn_root);
  void AddEmbDataFiles(const char* fn_list);
  int  GetNumEmbEvents();
 
 protected:
  void OpenEmbDataFile(const char* fn_root);
  void CloseEmbDataFile();
  bool GetNextEmbEvent();
};
 
#endif // _DO_EMBEDDING__H_