packages/TrackingDiagnostics/TrackContainerCombiner.h

0001 // Tell emacs that this is a C++ source
0002 //  -*- C++ -*-.
0003 #ifndef TRACKCONTAINERCOMBINER_H
0004 #define TRACKCONTAINERCOMBINER_H
0005
0006 #include <fun4all/SubsysReco.h>
0007
0008 #include <string>
0009
0010 class TrackSeedContainer;
0011 class SvtxTrackMap;
0012 class PHCompositeNode;
0013
0014 class TrackContainerCombiner : public SubsysReco
0015 {
0016  public:
0017   TrackContainerCombiner(const std::string &name = "TrackContainerCombiner");
0018
0019   ~TrackContainerCombiner() override;
0020
0021   int InitRun(PHCompositeNode *topNode) override;
0022   int process_event(PHCompositeNode *) override;
0023   int End(PHCompositeNode *) override;
0024
0025   void newContainerName(const std::string &name) { m_newContainerName = name; }
0026   void oldContainerName(const std::string &name) { m_oldContainerName = name; }
0027
0028  private:
0029   void mergeSeeds();
0030   int getNodes(PHCompositeNode *topNode);
0031   std::string m_newContainerName = "SiliconTrackSeedContainer";
0032   std::string m_oldContainerName = "SiliconTrackSeedContainerIt1";
0033   bool m_seedContainer = true;
0034   TrackSeedContainer *m_newSeedContainer = nullptr;
0035   TrackSeedContainer *m_oldSeedContainer = nullptr;
0036 };
0037
0038 #endif  // TRACKCONTAINERCOMBINER_H