track_EEC/src/trackEEC.h

0001 // Tell emacs that this is a C++ source
0002 //  -*- C++ -*-.
0003 #ifndef trackEEC_H
0004 #define trackEEC_H
0005
0006 #include <fun4all/SubsysReco.h>
0007 #include <fun4all/Fun4AllReturnCodes.h>
0008
0009 #include <phool/PHCompositeNode.h>
0010 #include <phool/getClass.h>
0011
0012 #include "TFile.h"
0013 #include "TH1.h"
0014 #include "TH2.h"
0015
0016 #include <fastjet/PseudoJet.hh>
0017 #include <fastjet/JetDefinition.hh>
0018 #include <fastjet/Selector.hh>
0019
0020 #include <vector>
0021 #include <format>
0022 #include <string>
0023
0024 class PHCompositeNode;
0025 class SvtxTrack;
0026
0027
0028 class trackEEC : public SubsysReco
0029 {
0030     public:
0031
0032         trackEEC(const std::string &name = "trackEEC");
0033         ~trackEEC () override = default;
0034
0035         int Init(PHCompositeNode *topNode) override;
0036
0037         int process_event(PHCompositeNode *topNode) override;
0038
0039         int End(PHCompositeNode *topNode) override;
0040
0041         void SetOutfileName(std::string name){m_outfileName = name;};
0042
0043     private:
0044
0045         float m_pi{0.13957};
0046
0047         TFile *outfile{nullptr};
0048         std::string m_outfileName{""};
0049         TH1D *h_nEvents{nullptr};
0050         TH1D *h_crossing{nullptr};
0051         TH1D *h_nintt{nullptr};
0052         TH1D *h_chi2_ndf{nullptr};
0053         TH1D *h_nClus{nullptr};
0054         TH1D *h_trackSpec{nullptr};
0055         TH2D *h_track_etaPhi{nullptr};
0056         TH1D *h_jetCons{nullptr};
0057         TH1D *h_jetSpec{nullptr};
0058         TH2D *h_jetConsSpec{nullptr};
0059         TH2D *h_jet_etaPhi{nullptr};
0060         TH1D *h_jetMass{nullptr};
0061         TH1D *h_inJetAll{nullptr};
0062         TH1D *h_inJetAll_Q2{nullptr};
0063         TH1D *h_inJet[5]{nullptr};
0064         TH1D *h_inJet_Q2[5]{nullptr};
0065         TH1D *h_wEEC{nullptr};
0066         TH1D *h_wEEC_jet{nullptr};
0067         TH1D *h_wEEC_jet_Q2{nullptr};
0068         TH1D *h_wEEC_dijet{nullptr};
0069         TH1D *h_wEEC_dijet_Q2{nullptr};
0070
0071         float jet_pTs[6] = {10,20,30,40,60,1000};
0072
0073         std::string m_trackMapName = "SvtxTrackMap";
0074
0075         fastjet::JetDefinition jetDef{ fastjet::antikt_algorithm, 0.4};
0076         fastjet::Selector jetSel = fastjet::SelectorAbsEtaMax(1.1 - 0.4) && fastjet::SelectorPtMin(10.0);
0077 };
0078
0079 #endif