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>

struct ActsSurfaceMaps;
struct ActsTrackingGeometry;
class SvtxTrack;
class SvtxTrackMap;
class TpcSpaceChargeMatrixContainer;
class TrkrCluster;
class TrkrClusterContainer;
class TrkrHitSetContainer;

class TFile;
class TH1;
class TH2;
class TH3;

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 );

  private:

  int load_nodes( PHCompositeNode* );

  void create_histograms();

  void process_tracks();

  void process_track( SvtxTrack* );

  int get_cell_index( const Acts::Vector3D& ) const;

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


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

  float m_max_drphi = 0.5;

  float m_max_dz = 0.5;

  std::unique_ptr<TpcSpaceChargeMatrixContainer> m_matrix_container;


  int m_total_clusters = 0;
  int m_accepted_clusters = 0;



  ActsSurfaceMaps* m_surfmaps = nullptr;

  ActsTrackingGeometry* m_tGeometry = nullptr;

  ActsTransformations m_transformer;

  TrkrHitSetContainer* m_hitsetcontainer = nullptr;

  SvtxTrackMap* m_track_map = nullptr;
  
  // clusters
  TrkrClusterContainer* m_cluster_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 = nullptr;

  TH2 *h_deltaz_layer = nullptr;

  TH3 *h_entries = nullptr;


};

#endif