PHActsToSvtxTracks.h

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file PHActsToSvtxTracks.h

#ifndef TRACKRECO_PHACTSTOSVTXTRACKS_H
#define TRACKRECO_PHACTSTOSVTXTRACKS_H

#include <fun4all/SubsysReco.h>
#include <trackbase/TrkrDefs.h>

#include <Acts/Utilities/BinnedArray.hpp>
#include <Acts/Utilities/Definitions.hpp>
#include <Acts/Utilities/Logger.hpp>

#include <Acts/EventData/TrackParameters.hpp>

#include <ActsExamples/EventData/Track.hpp>
#include <ActsExamples/EventData/TrkrClusterSourceLink.hpp>
#include <ActsExamples/EventData/TrkrClusterMultiTrajectory.hpp>

#include <trackbase/ActsTrackingGeometry.h>

#include <boost/bimap.hpp>

#include <map>
#include <string>
#include <vector>

class PHCompositeNode;
class SvtxTrackMap;
class SvtxTrack;
class SvtxVertexMap;
class MakeActsGeometry;

using SourceLink = ActsExamples::TrkrClusterSourceLink;
using Trajectory = ActsExamples::TrkrClusterMultiTrajectory;

class PHActsToSvtxTracks : public SubsysReco
{
 public:
  PHActsToSvtxTracks(const std::string &name = "PHActsToSvtxTracks");
  ~PHActsToSvtxTracks() override {}

  int End(PHCompositeNode *topNode) override;
  int Init(PHCompositeNode *topNode) override;
  int InitRun(PHCompositeNode *topNode) override;
  int process_event(PHCompositeNode *topNode) override;
  int ResetEvent(PHCompositeNode *topNode) override;
  
  void setSvtxTrackMapName(std::string &name)
  { m_svtxMapName = name;}

 private:

  int createNodes(PHCompositeNode *topNode);

  int getNodes(PHCompositeNode *topNode);

  void createSvtxTrack(const unsigned int trackKey, Trajectory traj);

  SvtxTrackMap *m_svtxTrackMap = nullptr;
  SvtxVertexMap *m_svtxVertexMap = nullptr;
  ActsTrackingGeometry *m_tGeometry = nullptr;
  std::map<const unsigned int, Trajectory> *m_actsFitResults = nullptr;

  std::string m_svtxMapName = "SvtxTrackMap";

};

#endif