CaloEvaluator.h

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file CaloEvaluator.h

#ifndef G4EVAL_CALOEVALUATOR_H
#define G4EVAL_CALOEVALUATOR_H

//===============================================
//===============================================

#include <fun4all/SubsysReco.h>

#include <set>
#include <string>
#include <vector>

class CaloEvalStack;
class PHCompositeNode;
class TFile;
class TNtuple;
class TTree;  //Added by Barak

class CaloEvaluator : public SubsysReco
{
 public:
  CaloEvaluator(const std::string &name = "CALOEVALUATOR",
                const std::string &caloname = "CEMC",
                const std::string &filename = "g4eval_cemc.root");
  ~CaloEvaluator() override{};

  int Init(PHCompositeNode *topNode) override;
  int process_event(PHCompositeNode *topNode) override;
  int End(PHCompositeNode *topNode) override;

  void set_strict(bool b) { _strict = b; }
  // funtions to limit the tracing to only part of the event ---------
  // and speed up the evaluation

  // when tracing truth showers limit the trace to showers
  // that result from truth particles with a particular embed flag set
  // useful if you only want to know about that electron you
  // embedded into a central hijing event
  // (evaluation for truth objects with matching embed flag set unaffected)
  void add_truth_tracing_embed_flag(int flag)
  {
    _truth_trace_embed_flags.insert(flag);
  }

  // limit the tracing of truth particles to those above some
  // theshold energy. useful for tracing only high energy particles
  // and ignoring low energy truth particles from a hijing event
  // (evaluation for objects above threshold unaffected)
  void set_truth_tracing_energy_threshold(float thresh)
  {
    _truth_e_threshold = thresh;
  }

  // limit the tracing of towers and clusters back to the truth particles
  // to only those reconstructed objects above a particular energy
  // threshold (evaluation for objects above threshold unaffected)
  void set_reco_tracing_energy_threshold(float thresh)
  {
    _reco_e_threshold = thresh;
  }

  // functions to limit the output size ------------------
  // will no evaluate or write out these particular ntuples
  // mostly intended for size savings, but some time savings will result
  void set_do_gpoint_eval(bool b) { _do_gpoint_eval = b; }
  void set_do_gshower_eval(bool b) { _do_gshower_eval = b; }
  void set_do_tower_eval(bool b) { _do_tower_eval = b; }
  void set_do_cluster_eval(bool b) { _do_cluster_eval = b; }

 private:
  std::string _caloname;

  unsigned int _ievent;

  //Added by Barak
  unsigned int _towerID_debug;
  int _ieta_debug;
  int _iphi_debug;
  float _eta_debug;
  float _phi_debug;
  float _e_debug;
  float _x_debug;
  float _y_debug;
  float _z_debug;

  std::set<int> _truth_trace_embed_flags;
  float _truth_e_threshold;
  float _reco_e_threshold;

  CaloEvalStack *_caloevalstack;

  std::vector<unsigned char> _towerEtas;
  std::vector<unsigned char> _towerPhis;
  std::vector<float> _towerEnergies;

  //----------------------------------
  // evaluator output ntuples

  bool _strict;

  bool _do_gpoint_eval;
  bool _do_gshower_eval;
  bool _do_tower_eval;
  bool _do_cluster_eval;

  TNtuple *_ntp_gpoint;
  TNtuple *_ntp_gshower;
  TNtuple *_ntp_tower;
  TTree *_tower_debug;  //Added by Barak
  TTree *_cluster_tower_info; //Added by Apurva
  TNtuple *_ntp_cluster;

  // evaluator output file
  std::string _filename;
  TFile *_tfile;

  // subroutines
  void printInputInfo(PHCompositeNode *topNode);     
  void fillOutputNtuples(PHCompositeNode *topNode);  
  void printOutputInfo(PHCompositeNode *topNode);    
};

#endif  // G4EVAL_CALOEVALUATOR_H