PHActsVertexFitter.h

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

#ifndef TRACKRECO_PHACTSVERTEXFITTER_H
#define TRACKRECO_PHACTSVERTEXFITTER_H

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

#include <Acts/Vertexing/Vertex.hpp>

#include <ActsExamples/EventData/TrkrClusterMultiTrajectory.hpp>

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

namespace Acts
{
  class TrackParameters;
}

using Trajectory = ActsExamples::TrkrClusterMultiTrajectory;

using BoundTrackParamVec = std::vector<const Acts::BoundTrackParameters *>;
using VertexTrackMap = std::map<const unsigned int,
                                BoundTrackParamVec>;

using ActsVertex = const Acts::Vertex<Acts::BoundTrackParameters>;

class PHActsVertexFitter : public SubsysReco
{
 public:
  PHActsVertexFitter(const std::string &name = "PHActsVertexFitter");
  ~PHActsVertexFitter() override {}
  int process_event(PHCompositeNode *topNode) override;
  int Init(PHCompositeNode *topNode) override;
  int InitRun(PHCompositeNode *topNode) override;
  int ResetEvent(PHCompositeNode *topNode) override;
  int End(PHCompositeNode *topNode) override;

  void updateSvtxVertexMap(bool updateSvtxVertexMap)
  {
    m_updateSvtxVertexMap = updateSvtxVertexMap;
  }

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

  VertexTrackMap getTracks();

  const Acts::BoundTrackParameters *makeTrackParam(const SvtxTrack *track) const;

  ActsVertex fitVertex(BoundTrackParamVec tracks,
                       Acts::Logging::Level logLevel) const;

  void fitVertices(std::vector<const Acts::BoundTrackParameters *> tracks);

  void createActsSvtxVertex(const unsigned int,
                            ActsVertex vertex);
  void updateSvtxVertex(const unsigned int,
                        ActsVertex vertex);

  int m_event = 0;

  std::map<const unsigned int, Trajectory> *m_actsFitResults;
  ActsTrackingGeometry *m_tGeometry;
  SvtxTrackMap *m_trackMap = nullptr;
  SvtxVertexMap *m_vertexMap = nullptr;
  SvtxVertexMap *m_actsVertexMap = nullptr;

  bool m_updateSvtxVertexMap = false;
};

#endif  //TRACKRECO_PHACTSVERTEXFITTER_H