EMJetVal.h

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

// Tell emacs that this is a C++ source
//  -*- C++ -*-.
#ifndef EMJETVAL_H
#define EMJETVAL_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>


using namespace fastjet;

class PHCompositeNode;
class TTree;

class EMJetVal : public SubsysReco
{
 public:

  EMJetVal(const std::string &recojetname = "AntiKt_Tower_r04",
           const std::string &truthjetname = "AntiKt_Truth_r04",
           const std::string &outputfilename = "myjetanalysis.root");
 
 std::vector<fastjet::PseudoJet> eventVector;
    
  int retrieveEvent(const fastjet::PseudoJet& jet);

  //  int retrieveEvent(vector<PseudoJet>& event);

  ~EMJetVal() 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;
  TFile* outFile;
  TH1F *_h_R04_z_sj_10_20;
  TH1F *_h_R04_theta_sj_10_20;
  TH1F *_h_R04_z_g_10_20;
  TH1F *_h_R04_theta_g_10_20;
  TH1F *_hmult_R04;
  //softDrop multiplicity hists
  TH1F *_hmult_R04_pT_10_20GeV;
  TH1F *_hjetpT_R04;
  TH1F *_hjeteta_R04;
  //correlation figures
  TH2D *correlation_theta_10_20;
  TH2D *correlation_z_10_20;


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

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