Analysis Software
Documentation for sPHENIX simulation software
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
G4SnglTree.h
Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file G4SnglTree.h
1 #ifndef G4HISTOS_G4SNGLTREE_H
2 #define G4HISTOS_G4SNGLTREE_H
3 
4 #include "G4EvtTree.h"
5 
6 #include <fun4all/SubsysReco.h>
7 
8 #include <map>
9 #include <set>
10 #include <string>
11 
12 // Forward declarations
13 class PHCompositeNode;
14 class PHG4HitContainer;
15 class TFile;
16 class TTree;
17 
18 class G4SnglTree : public SubsysReco
19 {
20  public:
22  G4SnglTree(const std::string &name = "G4SnglTree", const std::string &filename = "G4SnglTree.root");
23 
25  ~G4SnglTree() override {}
26 
28  int Init(PHCompositeNode *) override;
29 
31  int process_event(PHCompositeNode *) override;
32 
34  int process_hit(PHG4HitContainer *hits, const std::string &dName, int detid, int &nhits);
35 
37  int End(PHCompositeNode *) override;
38 
39  void AddNode(const std::string &name, const int detid = 0);
40 
41  protected:
42  int nblocks;
43  // std::vector<TH2 *> nhit_edep;
45  std::set<std::string> _node_postfix;
46  std::map<std::string, int> _detid;
47 
48  TTree *g4tree;
50  TFile *outfile;
51 };
52 
53 #endif