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 <TString.h>

#include <memory>
#include <vector>

// forward declaration
class SvtxTrack;
class SvtxTrackMap;
class TpcSpaceChargeMatrixContainer;
class TrkrCluster;
class TrkrClusterContainer;

class TpcSpaceChargeReconstruction: public SubsysReco, public PHParameterInterface
{
  public:

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



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


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


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

  bool accept_track( SvtxTrack* ) const;

  void process_track( SvtxTrack* );

  int get_cell_index( TrkrCluster* ) const;

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

  bool m_use_micromegas = true;


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

  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;
  

  SvtxTrackMap* m_track_map = nullptr;
  TrkrClusterContainer* m_cluster_map = nullptr;

};

#endif