PHActsTrackProjection.h

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

#ifndef TRACKRECO_PHACTSTRACKPROJECTION_H
#define TRACKRECO_PHACTSTRACKPROJECTION_H

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

#include <trackbase/ActsGeometry.h>

#include "ActsPropagator.h"

#include <Acts/Definitions/Algebra.hpp>
#include <Acts/EventData/TrackParameters.hpp>
#include <Acts/Surfaces/CylinderSurface.hpp>
#include <Acts/Utilities/Result.hpp>

#include <ActsExamples/EventData/Trajectories.hpp>

class PHCompositeNode;
class RawClusterContainer;
class TowerInfoContainer;
class RawTowerGeomContainer;
class SvtxTrackMap;
class SvtxTrack;
class SvtxVertexMap;

#include <map>
#include <memory>
#include <string>

class PHActsTrackProjection : public SubsysReco
{
 public:
  using BoundTrackParam =
      const Acts::BoundTrackParameters;
  using SurfacePtr = std::shared_ptr<const Acts::Surface>;
  using Trajectory = ActsExamples::Trajectories;
  using BoundTrackParamResult = ActsPropagator::BTPPairResult;

  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;

  void useConstField(bool field) { m_constField = field; }
  void setConstFieldVal(float b) { m_constFieldVal = b; }

  void setLayerRadius(SvtxTrack::CAL_LAYER layer,
                      const float rad)
  {
    if (m_caloRadii.find(layer) != m_caloRadii.end())
    {
      m_caloRadii[layer] = rad;
    }
    else
    {
      m_caloRadii.insert(std::make_pair(layer, rad));
    }
  }

 private:
  int getNodes(PHCompositeNode *topNode);
  int projectTracks(int caloLayer);

  BoundTrackParamResult propagateTrack(
      const Acts::BoundTrackParameters &params,
      const int caloLayer,
      const SurfacePtr &targetSurf);

  int setCaloContainerNodes(PHCompositeNode *topNode,
                            const int caloLayer);

  int makeCaloSurfacePtrs(PHCompositeNode *topNode);

  void updateSvtxTrack(const ActsPropagator::BoundTrackParamPair &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);
  double deltaPhi(const double &phi);
 
  ActsGeometry *m_tGeometry = nullptr;
  SvtxTrackMap *m_trackMap = nullptr;
  SvtxVertexMap *m_vertexMap = 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;
  std::map<SvtxTrack::CAL_LAYER, float> m_caloRadii;

  RawTowerGeomContainer *m_towerGeomContainer = nullptr;
  TowerInfoContainer *m_towerContainer = nullptr;
  RawClusterContainer *m_clusterContainer = nullptr;

  bool m_constField = true;
  float m_constFieldVal = 1.4;
  bool m_useCemcPosRecalib = false;
  bool m_calosAvailable = true;

  int m_event = 0;
};

#endif