PHG4TpcDistortion.h

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

// $Id: $

#ifndef G4TPC_PHG4TPCDISTORTION_H
#define G4TPC_PHG4TPCDISTORTION_H

#include <memory>
#include <string>

class TFile;
class TH3;
class TTree;

class PHG4TpcDistortion
{
 public:
  explicit PHG4TpcDistortion() = default;



  double get_x_distortion_cartesian(double x, double y, double z) const;

  double get_y_distortion_cartesian(double x, double y, double z) const;

  double get_z_distortion_cartesian(double x, double y, double z) const;
  
  double get_r_distortion(double r, double phi, double z) const;

  double get_rphi_distortion(double r, double phi, double z) const;

  double get_z_distortion(double r, double phi, double z) const;




  
  int Verbosity() const
  {
    return verbosity;
  }




  void set_do_static_distortions(bool value)
  {
    m_do_static_distortions = value;
  }

  void set_static_distortion_filename(const std::string &value)
  {
    m_static_distortion_filename = value;
  }

  void set_do_time_ordered_distortions(bool value)
  {
    m_do_time_ordered_distortions = value;
  }

  void set_time_ordered_distortion_filename(const std::string &value)
  {
    m_time_ordered_distortion_filename = value;
  }

  void Init();

  void load_event(int event_num);

  void Verbosity(const int ival)
  {
    verbosity = ival;
  }


 private:
  double get_distortion(char axis, double r, double phi, double z) const;

  int verbosity = 0;


  bool m_do_static_distortions = false;
  std::string m_static_distortion_filename;
  std::unique_ptr<TFile> m_static_tfile;
  TH3 *hDRint[2]={nullptr,nullptr};
  TH3 *hDPint[2]={nullptr,nullptr};
  TH3 *hDZint[2]={nullptr,nullptr};


  bool m_do_time_ordered_distortions = false;
  std::string m_time_ordered_distortion_filename;
  std::unique_ptr<TFile> m_time_ordered_tfile;
  TTree *TimeTree=nullptr;
  TH3 *TimehDR[2]={nullptr,nullptr};
  TH3 *TimehDP[2]={nullptr,nullptr};
  TH3 *TimehDZ[2]={nullptr,nullptr};
};

#endif /* G4TPC_PHG4TPCDISTORTION_H */