d6/dd5/CaloEvaluator_8h_source.html

#ifndef G4EVAL_CALOEVALUATOR_H

#define G4EVAL_CALOEVALUATOR_H


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

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


#include <fun4all/SubsysReco.h>


#include <set>

#include <string>


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;


  //----------------------------------

  // 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

  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