Warning, /coresoftware/offline/packages/trackreco/PHActsToSvtxTracks.h.outdated is written in an unsupported language. File is not indexed.
0001 #ifndef TRACKRECO_PHACTSTOSVTXTRACKS_H
0002 #define TRACKRECO_PHACTSTOSVTXTRACKS_H
0003
0004 #include <fun4all/SubsysReco.h>
0005 #include <trackbase/TrkrDefs.h>
0006
0007 /// Acts includes to create all necessary definitions
0008 #include <Acts/Utilities/BinnedArray.hpp>
0009 #include <Acts/Utilities/Definitions.hpp>
0010 #include <Acts/Utilities/Logger.hpp>
0011
0012 #include <Acts/EventData/TrackParameters.hpp>
0013
0014 #include <ActsExamples/EventData/Track.hpp>
0015 #include <ActsExamples/EventData/TrkrClusterSourceLink.hpp>
0016 #include <ActsExamples/EventData/TrkrClusterMultiTrajectory.hpp>
0017
0018 #include <trackbase/ActsTrackingGeometry.h>
0019
0020 #include <boost/bimap.hpp>
0021
0022 #include <map>
0023 #include <string>
0024 #include <vector>
0025
0026 class PHCompositeNode;
0027 class SvtxTrackMap;
0028 class SvtxTrack;
0029 class SvtxVertexMap;
0030 class MakeActsGeometry;
0031
0032 using SourceLink = ActsExamples::TrkrClusterSourceLink;
0033 using Trajectory = ActsExamples::TrkrClusterMultiTrajectory;
0034
0035 class PHActsToSvtxTracks : public SubsysReco
0036 {
0037 public:
0038 /// Default constructor and destructor
0039 PHActsToSvtxTracks(const std::string &name = "PHActsToSvtxTracks");
0040 ~PHActsToSvtxTracks() override {}
0041
0042 /// Inherited SubsysReco functions
0043 int End(PHCompositeNode *topNode) override;
0044 int Init(PHCompositeNode *topNode) override;
0045 int InitRun(PHCompositeNode *topNode) override;
0046 int process_event(PHCompositeNode *topNode) override;
0047 int ResetEvent(PHCompositeNode *topNode) override;
0048
0049 void setSvtxTrackMapName(std::string &name)
0050 { m_svtxMapName = name;}
0051
0052 private:
0053
0054 /// Create track seed node if it doesn't exist yet
0055 int createNodes(PHCompositeNode *topNode);
0056
0057 /// Get nodes off node tree needed to execute module
0058 int getNodes(PHCompositeNode *topNode);
0059
0060 void createSvtxTrack(const unsigned int trackKey, Trajectory traj);
0061
0062 SvtxTrackMap *m_svtxTrackMap = nullptr;
0063 SvtxVertexMap *m_svtxVertexMap = nullptr;
0064 ActsTrackingGeometry *m_tGeometry = nullptr;
0065 std::map<const unsigned int, Trajectory> *m_actsFitResults = nullptr;
0066
0067 std::string m_svtxMapName = "SvtxTrackMap";
0068
0069 };
0070
0071 #endif
