Adapt task to newest Analysis Framework

diff --git a/PWG2/FEMTOSCOPY/AliFemto/AliAnalysisTaskFemto.cxx b/PWG2/FEMTOSCOPY/AliFemto/AliAnalysisTaskFemto.cxx
index cb6e522e347aaa69631c939a1ce5760d04d1cfc4..42d3fc66c567f76b6c575035ad5ecd1973ef8940 100644 (file)
--- a/PWG2/FEMTOSCOPY/AliFemto/AliAnalysisTaskFemto.cxx
+++ b/PWG2/FEMTOSCOPY/AliFemto/AliAnalysisTaskFemto.cxx
@@ -186,6 +186,17 @@ void AliAnalysisTaskFemto::Exec(Option_t *) {
       }    \r
     }\r
 \r
+    // Get ESD\r
+    AliESDInputHandler *esdH = dynamic_cast<AliESDInputHandler*> (AliAnalysisManager::GetAnalysisManager()->GetInputEventHandler());\r
+    \r
+    if (!esdH) {\r
+      Printf("ERROR: Could not get ESDInputHandler");\r
+      return;\r
+    } \r
+    else {\r
+      fESD = esdH->GetEvent();\r
+    }\r
+\r
     printf("Tracks in ESD: %d \n",fESD->GetNumberOfTracks());\r
 \r
     if (fESD->GetNumberOfTracks() >= 0) {\r
@@ -222,6 +233,19 @@ void AliAnalysisTaskFemto::Exec(Option_t *) {
       Printf("ERROR: fAOD not available");\r
       return;\r
     }\r
+\r
+    // Get AOD\r
+    AliAODInputHandler *aodH = dynamic_cast<AliAODInputHandler*> (AliAnalysisManager::GetAnalysisManager()->GetInputEventHandler());\r
+      \r
+    if (!aodH) {\r
+      Printf("ERROR: Could not get AODInputHandler");\r
+      return;\r
+    } \r
+    else {\r
+\r
+      fAOD = aodH->GetEvent();\r
+    }\r
+\r
     printf("Tracks in AOD: %d \n",fAOD->GetNumberOfTracks());\r
     \r
     if (fAOD->GetNumberOfTracks() > 0) {\r