EnergyHelper.h

#ifndef ENERGYHELPER_H
#define ENERGYHELPER_H

#include "HelperBase.h"

#include "GeometryHelper.h"

#include "lardataobj/AnalysisBase/Calorimetry.h"
#include "lardataobj/RecoBase/Cluster.h"
#include "lardataobj/RecoBase/Hit.h"
#include "lardataobj/RecoBase/PFParticle.h"
#include "lardataobj/RecoBase/Shower.h"
#include "lardataobj/RecoBase/Track.h"

#include "canvas/Persistency/Common/FindManyP.h"
#include "canvas/Persistency/Common/FindOneP.h"
#include "lardata/DetectorInfoServices/DetectorPropertiesService.h"
#include "TPrincipal.h"
#include "uboone/UBXSec/Algorithms/TrackQuality.h"
#include "uboone/Database/TPCEnergyCalib/TPCEnergyCalibProvider.h"
#include "uboone/Database/TPCEnergyCalib/TPCEnergyCalibService.h"
#include "lardataobj/AnalysisBase/ParticleID.h"
#include "lardataobj/AnalysisBase/BackTrackerMatchingData.h"

#include "uboone/ParticleID/Algorithms/uB_PlaneIDBitsetHelperFunctions.h"

namespace lee {

class EnergyHelper : public HelperBase {
public:
  explicit EnergyHelper();
  ~EnergyHelper() = default;

  void reconfigure(fhicl::ParameterSet const &pset);

  void dQdx(const recob::Shower *shower_obj,
            std::vector<art::Ptr<recob::Cluster>> *clusters,
            art::FindManyP<recob::Hit> *hits_per_cluster,
            art::FindManyP<simb::MCParticle, anab::BackTrackerHitMatchingData> *mcps_per_hit,
            std::vector<double> &dqdx,
            std::vector<std::vector<double>> &dqdx_hits,
            std::vector<double> &pitches);

  double PID(art::Ptr<anab::ParticleID> selected_pid,
                           std::string AlgName,
                           anab::kVariableType VariableType,
                           anab::kTrackDir TrackDirection,
                           int pdgCode);

  void dEdx_from_dQdx(std::vector<double> &dedx,
                      std::vector<double> dqdx);

  void PCA(std::vector<art::Ptr<recob::Cluster>> *clusters,
           art::FindManyP<recob::Hit> *hits_per_cluster,
           std::vector<std::vector<double>> &pca_planes);

  void get_cali(std::vector<art::Ptr<recob::SpacePoint>> *spcpnts,
                art::FindManyP<recob::Hit> *hits_per_spcpnts,
                std::vector<double> &cali_corr);

  void energy_from_hits(std::vector<art::Ptr<recob::Cluster>> *clusters,
                        art::FindManyP<recob::Hit> *hits_per_cluster,
                        art::FindManyP<simb::MCParticle, anab::BackTrackerHitMatchingData> *mcps_per_hit,
                        std::vector<int> &nHits,
                        std::vector<double> &pfenergy);

  void cluster_residuals(std::vector<art::Ptr<recob::Cluster>> *clusters,
                         art::FindManyP<recob::Hit> *hits_per_cluster,
                         double &mean_v,
                         double &std_v);

  void track_dQdx(std::vector<art::Ptr<anab::Calorimetry>> *calos,
                  std::vector<double> &dqdx,
                  std::vector<double> &dedx);

  void dQdx_cali(const recob::Shower *shower_obj,
                 std::vector<double> &dqdx_cali);

  bool is_hit_data(art::FindManyP<simb::MCParticle, anab::BackTrackerHitMatchingData> *mcps_per_hit,
                                 size_t hit_key);

private:
  std::vector<double> _data_gain = {236.41, 228.83, 242.72}; // DocDB 14754
  std::vector<double> _mc_gain = {193.05, 196.85, 196.85};  // Plane 0, plane 1, plane 2
  std::vector<double> _gain;
  const lariov::TPCEnergyCalibProvider &_energy_calib_provider = art::ServiceHandle<lariov::TPCEnergyCalibService>()->GetProvider();
  const detinfo::DetectorProperties *_detprop = lar::providerFrom<detinfo::DetectorPropertiesService>();
  double _drift = _detprop->DriftVelocity() * 1e-3;
  double _readout_window = 4.8;
  double _from_tick_to_ns = _readout_window / _detprop->ReadOutWindowSize() * 1e6;
  double _wire_spacing = 0.3;
  double _work_function = 23 / 1e6;
  double _betap;
  double _alpha;
  double _recombination_factor;
  double _dQdx_rectangle_length;
  double _dQdx_rectangle_width;
  bool m_isOverlaidSample;
  GeometryHelper geo_helper;
};
} // namespace lee

#endif