Various updates

[u/mrichter/AliRoot.git] / PWG2 / FORWARD / analysis2 / AliFMDCorrELossFit.cxx

diff --git a/PWG2/FORWARD/analysis2/AliFMDCorrELossFit.cxx b/PWG2/FORWARD/analysis2/AliFMDCorrELossFit.cxx
index 15885536fb98d5f7ca7cdd8d479cd408fb666ae6..774db2fe88d55e91cf82a8b66e3bc7ca6f927a9a 100644 (file)
--- a/PWG2/FORWARD/analysis2/AliFMDCorrELossFit.cxx
+++ b/PWG2/FORWARD/analysis2/AliFMDCorrELossFit.cxx
@@ -738,6 +738,48 @@ AliFMDCorrELossFit::SetFit(UShort_t  d, Char_t r, Double_t eta,
   }
   return kTRUE;
 }
+//____________________________________________________________________
+AliFMDCorrELossFit::ELossFit*
+AliFMDCorrELossFit::GetFit(UShort_t  d, Char_t r, Int_t etabin) const
+{
+  // 
+  // Get the fit corresponding to the specified parameters 
+  // 
+  // Parameters:
+  //    d      Detector 
+  //    r      Ring 
+  //    etabin Eta bin (1 based)
+  // 
+  // Return:
+  //    Fit parameters or null in case of problems 
+  //
+  TObjArray* ringArray = GetRingArray(d, r);
+  if (!ringArray)                                   return 0;
+  if (etabin <= 0 || etabin >= fEtaAxis.GetNbins()) return 0;
+  if      (etabin > ringArray->GetEntriesFast())    return 0;
+  else if (etabin >= ringArray->GetEntriesFast())   etabin--;
+  else if (!ringArray->At(etabin))                  etabin++;
+  return static_cast<ELossFit*>(ringArray->At(etabin));
+}
+//____________________________________________________________________
+AliFMDCorrELossFit::ELossFit*
+AliFMDCorrELossFit::GetFit(UShort_t  d, Char_t r, Double_t eta) const
+{
+  // 
+  // Find the fit corresponding to the specified parameters 
+  // 
+  // Parameters:
+  //    d   Detector 
+  //    r   Ring 
+  //    eta Eta value 
+  // 
+  // Return:
+  //    Fit parameters or null in case of problems 
+  //
+  Int_t etabin = FindEtaBin(eta);
+  return GetFit(d, r, etabin);
+}
+
 //____________________________________________________________________
 AliFMDCorrELossFit::ELossFit*
 AliFMDCorrELossFit::FindFit(UShort_t  d, Char_t r, Int_t etabin) const