TPC-GemGain/src/TPCGemGainCalb.h

0001 // Tell emacs that this is a C++ source
0002 //  -*- C++ -*-.
0003 #ifndef TPCGEMGAINCALB_H
0004 #define TPCGEMGAINCALB_H
0005
0006 #include <string>
0007
0008 #include <fun4all/SubsysReco.h>
0009
0010 #include <trackbase/TrkrDefs.h>
0011 #include <trackbase/TrkrClusterContainer.h>
0012
0013 class PHCompositeNode;
0014 class SvtxTrackMap;
0015 class SvtxTrack;
0016 class SvtxVertexMap;
0017
0018 class TF1;
0019 class TNtuple;
0020 class TFile;
0021
0022 class TPCGemGainCalb : public SubsysReco
0023 {
0024  public:
0025
0026  TPCGemGainCalb(const std::string &name = "TPCGemGainCalb");
0027
0028   virtual ~TPCGemGainCalb();
0029
0030   void set_track_map_name(const std::string &map_name) { _track_map_name = map_name; }
0031
0032  //! run initialization
0033   int InitRun(PHCompositeNode *topNode);
0034
0035   //! event processing
0036   int process_event(PHCompositeNode *topNode);
0037
0038   int EndRun(PHCompositeNode *topNode);
0039
0040   //! end of process
0041 int End(PHCompositeNode *topNode);
0042
0043  protected:
0044
0045  private:
0046
0047   int GetNodes(PHCompositeNode* topNode);
0048
0049   std::string _track_map_name;
0050
0051   SvtxTrackMap *_track_map{nullptr};
0052   SvtxTrack *_track{nullptr};
0053   SvtxVertexMap *_vertex_map{nullptr};
0054   TrkrClusterContainer *_cluster_map;
0055
0056   TNtuple *ntp{nullptr};
0057   TFile *fout;
0058
0059 };
0060
0061 #endif // TPCGEMGAINCALB_H