EIC Software
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
PHTpcTrackSeedCircleFit.h
Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file PHTpcTrackSeedCircleFit.h
1 // Tell emacs that this is a C++ source
2 // -*- C++ -*-.
3 #ifndef PHTPCTRACKSEEDCIRCLEFIT_H
4 #define PHTPCTRACKSEEDCIRCLEFIT_H
5 
6 #include <fun4all/SubsysReco.h>
10 
11 #include <string>
12 #include <vector>
13 
14 class PHCompositeNode;
15 class SvtxTrackMap;
16 class SvtxTrack;
17 class TrkrCluster;
18 class TF1;
20 
22 {
23  public:
24 
25  PHTpcTrackSeedCircleFit(const std::string &name = "PHTpcTrackSeedCircleFit");
26 
27  ~PHTpcTrackSeedCircleFit() override = default;
28 
29  int InitRun(PHCompositeNode* topNode) override;
30 
31  int process_event(PHCompositeNode*) override;
32 
33  int End(PHCompositeNode*) override;
34 
35  void use_truth_clusters(bool truth)
36  { _use_truth_clusters = truth; }
37 
38  void set_track_map_name(const std::string &map_name) { _track_map_name = map_name; }
39  void SetIteration(int iter){_n_iteration = iter;}
40 
41  private:
42 
43  int GetNodes(PHCompositeNode* topNode);
44  std::vector<TrkrCluster*> getTrackClusters(SvtxTrack *);
46 
50 
51  bool _use_truth_clusters = false;
58 
59  int _n_iteration = 0;
60  std::string _track_map_name = "SvtxTrackMap";
61 
62 };
63 
64 #endif // PHTRACKSEEDVERTEXASSOCIATION_H