JetValidation.h

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

// Tell emacs that this is a C++ source
//  -*- C++ -*-.
#ifndef JETVALIDATION_H
#define JETVALIDATION_H

#include <fun4all/SubsysReco.h>
#include <jetbase/Jetv1.h>
#include <jetbase/Jetv2.h>
#include <fastjet/PseudoJet.hh>

#include <string>
#include <vector>

#include <TFile.h>
#include <TH1F.h>
#include <TH2D.h>


class PHCompositeNode;
class TTree;

class JetValidation : public SubsysReco
{
 public:

  JetValidation(const std::string &recojetname = "AntiKt_Tower_r04",
                const std::string &truthjetname = "AntiKt_Truth_r04",
                const std::string &outputfilename = "myjetanalysis.root");

  ~JetValidation() override;

  void
    setEtaRange(double low, double high)
  {
    m_etaRange.first = low;
    m_etaRange.second = high;
  }
 void
   setPtRange(double low, double high)
 {
   m_ptRange.first = low;
   m_ptRange.second = high;
 }
 void
   doTruth(int flag)
 {
   m_doTruthJets = flag;
 }
 void
   doSeeds(int flag)
 {
   m_doSeeds = flag;
 }
 void
   doUnsub(int flag)
 {
   m_doUnsubJet = flag;
 }
  int Init(PHCompositeNode *topNode) override;

  int InitRun(PHCompositeNode *topNode) override;

  int process_event(PHCompositeNode *topNode) override;

  int ResetEvent(PHCompositeNode *topNode) override;

  int EndRun(const int runnumber) override;

  int End(PHCompositeNode *topNode) override;

  int Reset(PHCompositeNode * /*topNode*/) override;

  void Print(const std::string &what = "ALL") const override;

 private:
  std::string m_recoJetName;
  std::string m_truthJetName;
  std::string m_outputFileName;
  std::pair<double, double> m_etaRange;
  std::pair<double, double> m_ptRange;
  int m_doTruthJets;
  int m_doSeeds;
  int m_doUnsubJet;

  TTree *m_T;

  int m_event;
  int m_nTruthJet;
  int m_nJet;
  float m_totalCalo;
  int m_centrality;
  float m_impactparam;
  float m_zvtx;

  std::vector<int> m_id;
  std::vector<int> m_nComponent;
  std::vector<float> m_eta;
  std::vector<float> m_phi;
  std::vector<float> m_e;
  std::vector<float> m_pt;

  std::vector<float> m_unsub_pt;
  std::vector<float> m_sub_et;

  std::vector<int> m_truthID;
  std::vector<int> m_truthNComponent;
  std::vector<float> m_truthEta;
  std::vector<float> m_truthPhi;
  std::vector<float> m_truthE;
  std::vector<float> m_truthPt;
  std::vector<float> m_truthdR;

  std::vector<float> m_eta_rawseed;
  std::vector<float> m_phi_rawseed;
  std::vector<float> m_pt_rawseed;
  std::vector<float> m_e_rawseed;
  std::vector<int> m_rawseed_cut;
  std::vector<float> m_eta_subseed;
  std::vector<float> m_phi_subseed;
  std::vector<float> m_pt_subseed;
  std::vector<float> m_e_subseed;
  std::vector<int> m_subseed_cut;
};

#endif // JETVALIDATION_H