PHActsVertexFinder.h

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

#ifndef TRACKRECO_PHACTSVERTEXFINDER_H
#define TRACKRECO_PHACTSVERTEXFINDER_H

#include "PHInitVertexing.h"
#include <trackbase/ActsTrackingGeometry.h>

#include <trackbase/TrkrDefs.h>

#include <Acts/Utilities/Result.hpp>
#include <Acts/Vertexing/Vertex.hpp>

#include <ActsExamples/EventData/TrkrClusterMultiTrajectory.hpp>

class PHCompositeNode;
class SvtxTrack;
class SvtxTrackMap;
class SvtxVertexMap;
class SvtxVertex;

namespace Acts
{
  class TrackParameters;
}

using Trajectory = ActsExamples::TrkrClusterMultiTrajectory;
using VertexVector = std::vector<Acts::Vertex<Acts::BoundTrackParameters>>;
using TrackPtrVector = std::vector<const Acts::BoundTrackParameters*>;
using VertexMap = std::map<unsigned int, 
                           Acts::Vertex<Acts::BoundTrackParameters>>;

using KeyMap = std::map<const Acts::BoundTrackParameters*, const unsigned int>;

class PHActsVertexFinder: public PHInitVertexing 
{
  
 public:
  PHActsVertexFinder(const std::string &name = "PHActsVertexFinder");

  ~PHActsVertexFinder() override {}

  void setMaxVertices(int maxVertices)
    { m_maxVertices = maxVertices; }
  
  void setFieldMap(const std::string& fieldMap)
    { m_fieldMap = fieldMap; }
  
 protected:
  int Setup(PHCompositeNode *topNode) override;
  int Process(PHCompositeNode *topNode) override;
  int ResetEvent(PHCompositeNode *topNode) override;
  int End(PHCompositeNode *topNode) override;

 private:
  
  int createNodes(PHCompositeNode *topNode);
  int getNodes(PHCompositeNode *topNode);

  TrackPtrVector getTracks(KeyMap& keyMap);

  VertexVector findVertices(TrackPtrVector& tracks);

  void fillVertexMap(VertexVector& vertices,
         KeyMap& keyMap);
  
  void updateTrackDCA(const unsigned int trackKey,
          const Acts::Vector3D vertex);

  void checkTrackVertexAssociation();

  VertexMap *m_actsVertexMap;

  int m_event = 0;
  int m_maxVertices = 20;

  int m_goodFits = 0;
  int m_totalFits = 0;
  std::string m_fieldMap = "";

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

#endif // TRACKRECO_PHACTSVERTEXFINDER_H