TracksInJets/src/TracksInJets.h

0001
0002 // Tell emacs that this is a C++ source
0003 //  -*- C++ -*-.
0004 #ifndef TRACKSINJETS_H
0005 #define TRACKSINJETS_H
0006
0007 #include <fun4all/SubsysReco.h>
0008 #include "TTree.h"
0009 #include <string>
0010 #include "TH2F.h"
0011
0012 class PHCompositeNode;
0013 class TH3;
0014
0015 class TracksInJets : public SubsysReco
0016 {
0017  public:
0018
0019   TracksInJets(const std::string &recojetname = "AntiKt_Tower_r04",
0020            const std::string &outputfilename = "tracksinjets.root");
0021
0022   ~TracksInJets() override;
0023
0024   /** Called during initialization.
0025       Typically this is where you can book histograms, and e.g.
0026       register them to Fun4AllServer (so they can be output to file
0027       using Fun4AllServer::dumpHistos() method).
0028   */
0029   int Init(PHCompositeNode *topNode) override;
0030
0031   /** Called for first event when run number is known.
0032       Typically this is where you may want to fetch data from
0033       database, because you know the run number. A place
0034       to book histograms which have to know the run number.
0035   */
0036   int InitRun(PHCompositeNode *topNode) override;
0037
0038   /** Called for each event.
0039       This is where you do the real work.
0040   */
0041   int process_event(PHCompositeNode *topNode) override;
0042
0043   /// Clean up internals after each event.
0044   int ResetEvent(PHCompositeNode *topNode) override;
0045
0046   /// Called at the end of each run.
0047   int EndRun(const int runnumber) override;
0048
0049   /// Called at the end of all processing.
0050   int End(PHCompositeNode *topNode) override;
0051
0052   /// Reset
0053   int Reset(PHCompositeNode * /*topNode*/) override;
0054
0055   void Print(const std::string &what = "ALL") const override;
0056
0057   bool isAA; // Declare isAA as a public member variable
0058
0059  private:
0060
0061   std::string m_recoJetName;
0062   float m_trk_pt_cut;
0063   float m_jetRadius;
0064   std::string m_outputFileName;
0065   TH3 *m_h_track_vs_calo_pt;
0066   TH2F *m_h_track_pt;
0067 };
0068
0069 #endif // TRACKSINJETS_H