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 iteration(const short int iter) { m_iteration = iter; }
0028
0029  private:
0030   int getNodes(PHCompositeNode *topNode);
0031   int createNodes(PHCompositeNode *topNode);
0032   void addClustersToIterationMap(TrackSeed *seed);
0033   short int m_iteration = 1;
0034
0035   SvtxTrackMap *m_trackMap = nullptr;
0036   TrkrClusterIterationMap *m_iterMap = nullptr;
0037 };
0038
0039 #endif  // TRACKINGITERATIONCOUNTER_H