PHTpcResiduals.h

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

#ifndef TRACKRECO_PHTPCRESIDUALS_H
#define TRACKRECO_PHTPCRESIDUALS_H

#include <fun4all/SubsysReco.h>
#include <tpc/TpcDistortionCorrection.h>
#include <tpc/TpcClusterZCrossingCorrection.h>
#include <trackbase/ActsGeometry.h>
#include <trackbase/ClusterErrorPara.h>
#include <trackbase/TrkrDefs.h>
#include <trackbase_historic/ActsTransformations.h>

#include <Acts/Utilities/Result.hpp>
#include <Acts/EventData/TrackParameters.hpp>

#include <memory>
#include <optional>

class PHCompositeNode;
class SvtxTrack;
class SvtxTrackMap;
class TpcSpaceChargeMatrixContainer;
class TrkrCluster;
class TrkrClusterContainer;

class TFile;
class TH1;
class TH2;
class TTree;

class PHTpcResiduals : public SubsysReco
{

 public:

  PHTpcResiduals(const std::string &name = "PHTpcResiduals");
  ~PHTpcResiduals() override = default;

  int Init(PHCompositeNode *topNode) override;
  int InitRun(PHCompositeNode *topNode) override;
  int process_event(PHCompositeNode *topNode) override;
  int End(PHCompositeNode *topNode) override;


  void setMaxTrackAlpha(float maxTAlpha) 
  { m_maxTAlpha = maxTAlpha;}
  
  void setMaxTrackBeta(float maxTBeta)
  { m_maxTBeta = maxTBeta; }
  
  void setMaxTrackResidualDrphi(float maxResidualDrphi) 
  { m_maxResidualDrphi = maxResidualDrphi;}
  
  void setMaxTrackResidualDz(float maxResidualDz)
  { m_maxResidualDz = maxResidualDz; }
  

  void setMinPt( double value ) 
  { m_minPt = value; }
  
  void setGridDimensions(const int phiBins, const int rBins, const int zBins);

  void setSavehistograms( bool ) {}
    
  void setHistogramOutputfile(const std::string&) {}

  void setOutputfile(const std::string &outputfile)
  {m_outputfile = outputfile;}

  void setUseMicromegas( bool value )
  { m_useMicromegas = value; }

  private:

  using BoundTrackParam = 
    const Acts::BoundTrackParameters;
  
  using BoundTrackParamPair = std::pair<float,BoundTrackParam>;

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


  Acts::Vector3 getGlobalPosition(TrkrDefs::cluskey, TrkrCluster*, short int crossing) const;

  int processTracks(PHCompositeNode *topNode);

  bool checkTrack(SvtxTrack* track) const;
  void processTrack(SvtxTrack* track);

  void addTrackState( SvtxTrack* track, TrkrDefs::cluskey key, float pathlength, const Acts::BoundTrackParameters& params );
  
  int getCell(const Acts::Vector3& loc);

  Acts::BoundTrackParameters makeTrackParams(SvtxTrack* ) const;

  ActsTransformations m_transformer;
  
  SvtxTrackMap *m_trackMap = nullptr;
  ActsGeometry *m_tGeometry = nullptr;
  TrkrClusterContainer *m_clusterContainer = nullptr;

  // crossing z correction
  TpcClusterZCrossingCorrection m_clusterCrossingCorrection;
  
  // distortion corrections
  TpcDistortionCorrectionContainer* m_dcc_static = nullptr;
  TpcDistortionCorrectionContainer* m_dcc_average = nullptr;
  TpcDistortionCorrectionContainer* m_dcc_fluctuation = nullptr;

  TpcDistortionCorrection m_distortionCorrection;
  
  float m_maxTAlpha = 0.6;
  float m_maxResidualDrphi = 0.5; // cm
  float m_maxTBeta = 1.5;
  float m_maxResidualDz = 0.5; // cm

  static constexpr float m_phiMin = 0;
  static constexpr float m_phiMax = 2. * M_PI;

  static constexpr float m_rMin = 20; // cm
  static constexpr float m_rMax = 78; // cm

  static constexpr int m_minClusCount = 10;

  static constexpr unsigned int m_nLayersTpc = 48;
  static constexpr float m_zMin = -105.5; // cm
  static constexpr float m_zMax = 105.5;  // cm
 
  ClusterErrorPara m_cluster_error_parametrization;
  
  std::unique_ptr<TpcSpaceChargeMatrixContainer> m_matrix_container;
  
  // TODO: check if needed
  int m_event = 0;
  
  bool m_useMicromegas = true;

  double m_minPt = 0.5;
  
  std::string m_outputfile = "TpcSpaceChargeMatrices.root";

  short int m_crossing = 0;
  

  int m_total_tracks = 0;
  int m_accepted_tracks = 0;

  int m_total_clusters = 0;
  int m_accepted_clusters = 0;

};

#endif