Implemented handy getters + renamed some of the enums to avoid conflicts with root...

[u/mrichter/AliRoot.git] / PWGLF / SPECTRA / PiKaPr / TestAOD / AliSpectraAODHistoManager.cxx

diff --git a/PWGLF/SPECTRA/PiKaPr/TestAOD/AliSpectraAODHistoManager.cxx b/PWGLF/SPECTRA/PiKaPr/TestAOD/AliSpectraAODHistoManager.cxx
index 40b328c2f43c1260e6f322fd9067ce4dbfc74c3c..86fb6172f7dbb4facb3565cebd97b5a79411044c 100644 (file)
--- a/PWGLF/SPECTRA/PiKaPr/TestAOD/AliSpectraAODHistoManager.cxx
+++ b/PWGLF/SPECTRA/PiKaPr/TestAOD/AliSpectraAODHistoManager.cxx
@@ -245,3 +245,46 @@ Long64_t AliSpectraAODHistoManager::Merge(TCollection* list)
   return count+1;
 }
 
+
+TH1* AliSpectraAODHistoManager::GetHistogram1D(UInt_t histoType, UInt_t particleType, UInt_t charge) {
+  // GetHistogram using particle ID and histogram type
+  Int_t baseId = -1;
+
+  switch(histoType) {
+  case kHistPtGenTruePrimary:
+    baseId = kHistPtGenTruePrimaryPionPlus;
+    break;
+  case kHistPtRecSigma:
+    baseId = kHistPtRecSigmaPionPlus;
+    break;
+  case kHistPtRecTruePrimary:
+    baseId = kHistPtRecTruePrimaryPionPlus;
+    break;
+  case kHistPtRecSigmaPrimary:
+    baseId = kHistPtRecSigmaPrimaryPionPlus;
+    break;
+  case kHistPtRecSigmaSecondaryMaterial:
+    baseId = kHistPtRecSigmaSecondaryMaterialPionPlus;
+    break;
+  case kHistPtRecSigmaSecondaryWeakDecay:
+    baseId = kHistPtRecSigmaSecondaryWeakDecayPionPlus;
+    break;
+  default:
+    baseId = -1;
+  }
+  
+  if (baseId < 0)
+    AliFatal("Wrong histogram type");
+
+  baseId = baseId + particleType + 3*(charge);
+  //  cout << "ID " << baseId << endl;
+
+  return GetHistogram(baseId);
+}
+
+TH2* AliSpectraAODHistoManager::GetHistogram2D(UInt_t histoType, UInt_t particleType, UInt_t charge){
+  // returns histo based on ids, casting it to TH2*
+  return (TH2*) GetHistogram1D(histoType,particleType,charge);
+
+
+}