g4simulation/g4histos/G4HitTTree.h

0001 #ifndef G4HISTOS_G4HITTTREE_H
0002 #define G4HISTOS_G4HITTTREE_H
0003
0004 #include <fun4all/SubsysReco.h>
0005
0006 #include <string>
0007
0008 class Fun4AllHistoManager;
0009 class PHCompositeNode;
0010 class TH1;
0011 class TH2;
0012
0013 class G4HitTTree : public SubsysReco
0014 {
0015  public:
0016   G4HitTTree(const std::string &name = "HITTTREE");
0017   ~G4HitTTree() override = default;
0018
0019   //! full initialization
0020   int Init(PHCompositeNode *) override;
0021
0022   //! event processing method
0023   int process_event(PHCompositeNode *) override;
0024
0025   int End(PHCompositeNode *) override;
0026
0027   void Detector(const std::string &det);
0028   void BlackHoleName(const std::string &bh);
0029
0030   void SaveHits(const int i = 1) { savehits = i; }
0031
0032  private:
0033   std::string _detector;
0034   std::string _outnodename;
0035   std::string _hitnodename;
0036   std::string _absorbernodename;
0037   std::string _blackholenodename;
0038   int savehits{1};
0039   int evtno{0};
0040   Fun4AllHistoManager *hm{nullptr};
0041   TH1 *etot_hist{nullptr};
0042   TH2 *eion_etot_hist{nullptr};
0043 };
0044
0045 #endif