EIC Software
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
PHTrackSeeding.h
Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file PHTrackSeeding.h
1 
7 #ifndef TRACKRECO_PHTRACKSEEDING_H
8 #define TRACKRECO_PHTRACKSEEDING_H
9 
10 // PHENIX includes
11 #include <fun4all/SubsysReco.h>
12 
13 // STL includes
14 #include <string>
15 
16 // forward declarations
17 class PHCompositeNode;
18 
19 //class SvtxClusterMap;
24 class SvtxVertexMap;
25 class SvtxTrackMap;
26 class AssocInfoContainer;
27 
32 class PHTrackSeeding : public SubsysReco
33 {
34  public:
35  PHTrackSeeding(const std::string &name = "PHTrackSeeding");
36  ~PHTrackSeeding() override {}
37 
38  int InitRun(PHCompositeNode *topNode) override;
39  int process_event(PHCompositeNode *topNode) override;
40  int End(PHCompositeNode *topNode) override;
41  void set_track_map_name(const std::string &map_name) { _track_map_name = map_name; }
42  void SetUseTruthClusters(bool setit){_use_truth_clusters = setit;}
43  void SetIteration(int iter){_n_iteration = iter;}
44  protected:
47  virtual int Setup(PHCompositeNode *topNode);
48 
51  virtual int Process(PHCompositeNode *topNode) = 0;
52 
54  virtual int End() = 0;
55 
64  std::string _track_map_name = "SvtxTrackMap";
65 
66  bool _use_truth_clusters = false;
67 
68  private:
69 
71  int CreateNodes(PHCompositeNode *topNode);
72 
74  int GetNodes(PHCompositeNode *topNode);
75 };
76 
77 #endif