packages/trackreco/TrackingIterationCounter.h

0001 // Tell emacs that this is a C++ source
0002 //  -*- C++ -*-.
0003 #ifndef TRACKINGITERATIONCOUNTER_H
0004 #define TRACKINGITERATIONCOUNTER_H
0005
0006 #include <fun4all/SubsysReco.h>
0007
0008 #include <string>
0009
0010 class PHCompositeNode;
0011 class SvtxTrackMap;
0012 class TrackSeed;
0013 class TrkrClusterIterationMap;
0014
0015 class TrackingIterationCounter : public SubsysReco
0016 {
0017  public:
0018   TrackingIterationCounter(const std::string &name = "TrackingIterationCounter");
0019
0020   ~TrackingIterationCounter() override;
0021
0022   int Init(PHCompositeNode *topNode) override;
0023   int InitRun(PHCompositeNode *topNode) override;
0024   int process_event(PHCompositeNode *topNode) override;
0025   int End(PHCompositeNode *topNode) override;
0026
0027   void setTrackMapName(const std::string &name) { m_trackMapName = name; }
0028   void iteration(const short int iter) { m_iteration = iter; }
0029
0030  private:
0031   int getNodes(PHCompositeNode *topNode);
0032   int createNodes(PHCompositeNode *topNode);
0033   void addClustersToIterationMap(TrackSeed *seed);
0034   short int m_iteration = 1;
0035
0036   std::string m_trackMapName = "SvtxTrackMap";
0037   SvtxTrackMap *m_trackMap = nullptr;
0038   TrkrClusterIterationMap *m_iterMap = nullptr;
0039 };
0040
0041 #endif  // TRACKINGITERATIONCOUNTER_H