JetEvaluator.h

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

#ifndef G4EVAL_JETEVALUATOR_H
#define G4EVAL_JETEVALUATOR_H

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

#include <fun4all/SubsysReco.h>

#include <string>

class JetEvalStack;
class PHCompositeNode;
class TFile;
class TNtuple;

class JetEvaluator : public SubsysReco
{
 public:
  JetEvaluator(const std::string &name = "JETEVALUATOR",
               const std::string &recojetname = "AntiKt_Tower_r0.3",
               const std::string &truthjetname = "AntiKt_Truth_r0.3",
               const std::string &filename = "g4eval_jets.root");
  ~JetEvaluator() override{};

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

  void set_strict(bool b) { _strict = b; }

 private:
  std::string _recojetname;
  std::string _truthjetname;

  unsigned long _ievent = 0;

  JetEvalStack *_jetevalstack = nullptr;

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

  bool _strict = false;

  bool _do_recojet_eval = true;
  bool _do_truthjet_eval = true;

  TNtuple *_ntp_recojet = nullptr;
  TNtuple *_ntp_truthjet = nullptr;

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

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

#endif  // G4EVAL_JETEVALUATOR_H