TpcSpaceChargeReconstruction.h

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

#ifndef TPCCALIB_TPCSPACECHARGERECONSTRUCTION_H
#define TPCCALIB_TPCSPACECHARGERECONSTRUCTION_H

#include <fun4all/SubsysReco.h>
#include <phparameter/PHParameterInterface.h>
#include <tpc/TpcDistortionCorrection.h>
#include <tpc/TpcClusterZCrossingCorrection.h>
#include <trackbase/ActsSurfaceMaps.h>
#include <trackbase/ActsTrackingGeometry.h>
#include <trackbase/ClusterErrorPara.h>

#include <Acts/Utilities/BinnedArray.hpp>
#include <Acts/Definitions/Algebra.hpp>
#include <Acts/Utilities/Logger.hpp>

#include <TString.h>

#include <memory>
#include <vector>

// forward declaration
class ActsGeometry;

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

class TFile;
class TH1;
class TH2;

class TpcSpaceChargeReconstruction: public SubsysReco, public PHParameterInterface
{
  public:

  TpcSpaceChargeReconstruction( const std::string& = "TPCSPACECHARGERECONSTRUCTION" );



  void set_use_micromegas( bool value )
  { m_use_micromegas = value; }

  void set_min_pt( double value ) 
  { m_min_pt = value; }
  

  void set_grid_dimensions( int phibins, int rbins, int zbins );


  void set_save_histograms( bool value ) { m_savehistograms = value; }
    
  void set_histogram_outputfile(const std::string &outputfile) {m_histogramfilename = outputfile;}
  

  void set_outputfile( const std::string& filename )
  { m_outputfile = filename; }


  int Init(PHCompositeNode*) override;

  int InitRun(PHCompositeNode*) override;

  int process_event(PHCompositeNode*) override;

  int End(PHCompositeNode*) override;

  void SetDefaultParameters() override;

  private:

  int load_nodes( PHCompositeNode* );
  
  void create_histograms();


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

  void process_tracks();

  bool accept_track( SvtxTrack* ) const;

  void process_track( SvtxTrack* );

  int get_cell_index( const Acts::Vector3& );

  int m_event = 0;
  
  std::string m_outputfile = "TpcSpaceChargeMatrices.root";

  bool m_use_micromegas = true;

  double m_min_pt = 0.5;

  ActsGeometry *m_tgeometry = nullptr;
  

  // residual cuts in r, phi plane
  float m_max_talpha = 0.6;
  float m_max_drphi = 0.5;

  // residual cuts in r, z plane
  float m_max_tbeta = 1.5;
  float m_max_dz = 0.5;
 
  ClusterErrorPara m_cluster_error_parametrization;
 
  std::unique_ptr<TpcSpaceChargeMatrixContainer> m_matrix_container;


  int m_total_tracks = 0;
  int m_accepted_tracks = 0;

  int m_total_clusters = 0;
  int m_accepted_clusters = 0;

  


  bool m_savehistograms = false;

  std::string m_histogramfilename = "TpcSpaceChargeReconstruction.root";
  std::unique_ptr<TFile> m_histogramfile;  
  
  using TH1_map_t = std::map<int,TH1*>;
  using TH2_map_t = std::map<int,TH2*>;
  
  TH1_map_t m_h_drphi;
  TH1_map_t m_h_dz;
  TH2_map_t m_h_drphi_alpha;
  TH2_map_t m_h_dz_beta;
  

  SvtxTrackMap* m_track_map = nullptr;
  TrkrClusterContainer* m_cluster_map = 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;

};

#endif