Adds local PID parametrization file option to KFParticle_sPHENIX

Author: Michael Peters

offline/packages/KFParticle_sPHENIX/KFParticle_Tools.cc

@@ -1184,7 +1184,16 @@ float KFParticle_Tools::get_dEdx(PHCompositeNode *topNode, const KFParticle &dau
 
 void KFParticle_Tools::init_dEdx_fits()
 {
-  std::string dedx_fitparams = CDBInterface::instance()->getUrl("TPC_DEDX_FITPARAM");
+  std::string dedx_fitparams;
+  if (m_use_local_PID_file)
+  {
+    dedx_fitparams = m_local_PID_filename;
+  }
+  else
+  {
+    dedx_fitparams = CDBInterface::instance()->getUrl("TPC_DEDX_FITPARAM");
+  }
+
   TFile *filefit = TFile::Open(dedx_fitparams.c_str());
 
   if (!filefit->IsOpen())
@@ -1193,12 +1202,25 @@ void KFParticle_Tools::init_dEdx_fits()
     return;
   }
 
-  filefit->GetObject("f_piband", f_pion_plus);
-  filefit->GetObject("f_Kband", f_kaon_plus);
-  filefit->GetObject("f_pband", f_proton_plus);
-  filefit->GetObject("f_piminus_band", f_pion_minus);
-  filefit->GetObject("f_Kminus_band", f_kaon_minus);
-  filefit->GetObject("f_pbar_band", f_proton_minus);
+  if (m_use_local_PID_file)
+  {
+    // new method is independent of charge
+    filefit->GetObject("pi_band",f_pion_plus);
+    filefit->GetObject("K_band",f_kaon_plus);
+    filefit->GetObject("p_band",f_proton_plus);
+    filefit->GetObject("pi_band",f_pion_minus);
+    filefit->GetObject("K_band",f_kaon_minus);
+    filefit->GetObject("p_band",f_proton_minus);
+  }
+  else
+  {
+    filefit->GetObject("f_piband", f_pion_plus);
+    filefit->GetObject("f_Kband", f_kaon_plus);
+    filefit->GetObject("f_pband", f_proton_plus);
+    filefit->GetObject("f_piminus_band", f_pion_minus);
+    filefit->GetObject("f_Kminus_band", f_kaon_minus);
+    filefit->GetObject("f_pbar_band", f_proton_minus);
+  }
 
   pidMap.insert(std::pair<int, TF1 *>(-11, f_pion_plus));
   pidMap.insert(std::pair<int, TF1 *>(211, f_pion_plus));

offline/packages/KFParticle_sPHENIX/KFParticle_Tools.h

@@ -149,6 +149,8 @@ class KFParticle_Tools : protected KFParticle_MVA
   std::vector<float> m_intermediate_vertex_volume;
 
   bool m_use_PID{false};
+  bool m_use_local_PID_file{false};
+  std::string m_local_PID_filename = "";
   float m_dEdx_band_width{0.2};  // Fraction of expected dE/dx
 
   TF1 *f_pion_plus{nullptr};

offline/packages/KFParticle_sPHENIX/KFParticle_sPHENIX.h

@@ -394,6 +394,10 @@ class KFParticle_sPHENIX : public SubsysReco, public KFParticle_nTuple, public K
   void selectMotherByMassError(bool select = true) { m_select_by_mass_error = select; }
 
   void usePID(bool use = true){ m_use_PID = use; }
+
+  void useLocalPIDFile(bool use = false){ m_use_local_PID_file = use; }
+
+  void setLocalPIDFilename(std::string name){ m_local_PID_filename = name; }
 
   void setPIDacceptFraction(float frac = 0.2){ m_dEdx_band_width = frac; }