TpcDirectLaserReconstruction.h

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

#ifndef TPCCALIB_TPCDIRECTLASERRECONSTRUCTION_H
#define TPCCALIB_TPCDIRECTLASERRECONSTRUCTION_H

#include <fun4all/SubsysReco.h>
#include <phparameter/PHParameterInterface.h>
#include <trackbase_historic/ActsTransformations.h>

#include <memory>

class SvtxTrack;
class SvtxTrackMap;
class TpcSpaceChargeMatrixContainer;
class TrkrHitSetContainer;
class PHG4TpcCylinderGeomContainer;

class TFile;
class TH1;
class TH2;
class TH3;
class TH2;
class TVector3;
class TNtuple;

class TpcDirectLaserReconstruction: public SubsysReco, public PHParameterInterface
{

  public:

  TpcDirectLaserReconstruction( const std::string& = "TpcDirectLaserReconstruction" );

  int Init(PHCompositeNode*) override;

  int InitRun(PHCompositeNode*) override;

  int process_event(PHCompositeNode*) override;

  int End(PHCompositeNode*) override;

  void SetDefaultParameters() override;


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

  void set_savehistograms( bool value ) { m_savehistograms = value; }

  void set_histogram_outputfile(const std::string &outputfile)
  {m_histogramfilename = outputfile;}

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

  void set_max_zrange(float length)
  {m_max_zrange = length;}

  void set_max_dca(float length)
  {m_max_dca = length;}

  private:

  int load_nodes( PHCompositeNode* );

  void create_histograms();

  void process_tracks();

  void process_track( SvtxTrack* );

  int get_cell_index( const TVector3& ) const;

  int Locate(float r , float phi , float z);

  float GetRelPhi( float xorig, float yorig, float x, float y, float phiorig );

  float GetRelTheta( float xorig, float yorig, float zorig, float x, float y, float z, float thetaorig, float phiorig );

  bool sameSign(float num1, float num2);

  std::string m_outputfile = "TpcSpaceChargeMatrices.root";

  float m_max_zrange = 10.0; // cm


  // residual cuts in r, phi plane
  float m_max_dca = NAN;

  float m_max_drphi = NAN;

  float m_max_dz = NAN;

  float m_pedestal = 74.4;  // pedestal for hit ASDC values


  std::unique_ptr<TpcSpaceChargeMatrixContainer> m_matrix_container;


  int m_total_hits = 0;
  int m_matched_hits = 0;
  int m_accepted_clusters = 0;



  PHG4TpcCylinderGeomContainer *m_geom_container = nullptr;

  ActsGeometry *m_tGeometry = nullptr;

  ActsTransformations m_transformer;

  SvtxTrackMap* m_track_map = nullptr;
  
  TrkrHitSetContainer* m_hit_map = nullptr;



  bool m_savehistograms = false;
  std::string m_histogramfilename = "TpcDirectLaserReconstruction.root";
  std::unique_ptr<TFile> m_histogramfile = nullptr;

  TH2* h_dca_layer = nullptr;

  TH2 *h_deltarphi_layer_south = nullptr;
  TH2 *h_deltarphi_layer_north = nullptr;

  TH2 *h_deltaz_layer = nullptr;

  TH2 *h_deltar_r = nullptr;

  TH3 *h_entries = nullptr;
  TNtuple *h_hits = nullptr;
  TNtuple *h_hits_reco = nullptr;
  // adc spectra of ALL lasers
  TH1 *h_adc = nullptr;
  TH1 *h_adc_reco = nullptr;

  TH1 *h_adc_sum = nullptr;
  TH1 *h_adc_sum_reco = nullptr;

  TH1 *h_adc_sum_ratio_true = nullptr;
  TH1 *h_adc_sum_ratio = nullptr;

//_______________________________________

  TH1 *h_num_sum = nullptr;
  TH1 *h_num_sum_reco = nullptr;

  TH1 *h_num_sum_ratio_true = nullptr;
  TH1 *h_num_sum_ratio = nullptr;

//_______________________________________

  TH2 *h_adc_vs_DCA_true = nullptr;
  TH2 *h_adc_sum_ratio_lasrangle = nullptr;
  TH2 *h_num_sum_ratio_lasrangle = nullptr;
   
  //TNtuple *h_origins = nullptr;
  //TNtuple *h_assoc_hits = nullptr;
  TNtuple *h_bright_hits_laser1 = nullptr;
  TNtuple *h_bright_hits_laser2 = nullptr;
  TNtuple *h_bright_hits_laser3 = nullptr;
  TNtuple *h_bright_hits_laser4 = nullptr;

  TH2 *h_deltheta_delphi = nullptr;
  TH2 *h_deltheta_delphi_1 = nullptr;
  TH2 *h_deltheta_delphi_2 = nullptr;
  TH2 *h_deltheta_delphi_3 = nullptr;
  TH2 *h_deltheta_delphi_4 = nullptr;
  TH2 *h_deltheta_delphi_5 = nullptr;
  TH2 *h_deltheta_delphi_6 = nullptr;
  TH2 *h_deltheta_delphi_7 = nullptr;
  TH2 *h_deltheta_delphi_8 = nullptr;

  // for recording # of unique GEM modules hit for the number of associated hits (to be replaced with GEM Modules)
  TH1 *h_GEMs_hit = nullptr;
  TH1 *h_layers_hit = nullptr;

  TH2 *h_relangle_lasrangle = nullptr;
  TH2 *h_relangle_theta_lasrangle = nullptr;
  TH2 *h_relangle_phi_lasrangle = nullptr;

  TH2* h_xy = nullptr;
  TH2* h_xz = nullptr;
  TH2* h_xy_pca = nullptr;
  TH2* h_xz_pca = nullptr;
  TH2* h_dca_path = nullptr;
  TH2* h_zr = nullptr;
  TH2* h_zr_pca = nullptr;
  TH2* h_dz_z = nullptr;
  //TNtuple *h_clusters = nullptr;


};

#endif