d5/d0f/PHActsTrackProjection_8h_source.html

#ifndef TRACKRECO_PHACTSTRACKPROJECTION_H

#define TRACKRECO_PHACTSTRACKPROJECTION_H


#include <fun4all/SubsysReco.h>

#include <trackbase/TrkrDefs.h>

#include <trackbase_historic/SvtxTrack.h>


#include <trackbase/ActsTrackingGeometry.h>


#include <Acts/Utilities/Definitions.hpp>

#include <Acts/Propagator/Propagator.hpp>

#include <Acts/Utilities/Result.hpp>

#include <Acts/Surfaces/CylinderSurface.hpp>

#include <Acts/EventData/TrackParameters.hpp>


#include <ActsExamples/EventData/TrkrClusterMultiTrajectory.hpp>


class PHCompositeNode;

class RawClusterContainer;

class RawTowerContainer;

class RawTowerGeomContainer;

class SvtxTrackMap;

class SvtxTrack;

class SvtxVertexMap;


#include <memory>

#include <map>

#include <string>


using BoundTrackParamPtr =

  std::unique_ptr<const Acts::BoundTrackParameters>;

using BoundTrackParamPtrResult = Acts::Result<BoundTrackParamPtr>;

using SurfacePtr = std::shared_ptr<const Acts::Surface>;

using Trajectory = ActsExamples::TrkrClusterMultiTrajectory;


class PHActsTrackProjection : public SubsysReco

{


 public:

  PHActsTrackProjection(const std::string& name

      = "PHActsTrackProjection");


  int Init(PHCompositeNode *topNode) override;

  int InitRun(PHCompositeNode *topNode) override;

  int process_event(PHCompositeNode *topNode) override;

  int End(PHCompositeNode *topNode) override;


 private:


  int getNodes(PHCompositeNode *topNode);

  int projectTracks(PHCompositeNode *topNode, int caloLayer);


  BoundTrackParamPtrResult propagateTrack(

        const Acts::BoundTrackParameters& params,

  const SurfacePtr &targetSurf);


  int setCaloContainerNodes(PHCompositeNode *topNode,

           const int caloLayer);


  int makeCaloSurfacePtrs(PHCompositeNode *topNode);


  void updateSvtxTrack(const Acts::BoundTrackParameters& params,

           SvtxTrack* svtxTrack,

           const int caloLayer);


  void getSquareTowerEnergies(int phiBin, int etaBin,

            double& energy3x3,

            double& energy5x5);


  void getClusterProperties(double phi, double eta,

          double& minIndex, double& minDphi,

          double& minDeta, double& minE);

  Acts::BoundTrackParameters makeTrackParams(SvtxTrack* track);

  double deltaPhi(const double& phi);

  Acts::Vector3D getVertex(SvtxTrack* track);


  ActsTrackingGeometry *m_tGeometry = nullptr;

  SvtxTrackMap *m_trackMap = nullptr;

  SvtxVertexMap *m_vertexMap = nullptr;

  std::map<const unsigned int, Trajectory> *m_trajectories{nullptr};


  const static int m_nCaloLayers = 3;

  std::vector<std::string> m_caloNames;

  std::vector<SvtxTrack::CAL_LAYER> m_caloTypes;

  std::map<std::string, SurfacePtr> m_caloSurfaces;


  RawTowerGeomContainer *m_towerGeomContainer = nullptr;

  RawTowerContainer *m_towerContainer = nullptr;

  RawClusterContainer *m_clusterContainer = nullptr;


  bool m_useCemcPosRecalib = false;


  int m_event = 0;

};


#endif