Added treatment of TPC only

diff --git a/PWG2/RESONANCES/AliRsnAnalysisTaskSEBase.cxx b/PWG2/RESONANCES/AliRsnAnalysisTaskSEBase.cxx
index 8694a98da6882f7cfe0c543f6bacab5a1f8ced61..41d4fcec0f5db707cf872b47f8c5a454585b1e4a 100644 (file)
--- a/PWG2/RESONANCES/AliRsnAnalysisTaskSEBase.cxx
+++ b/PWG2/RESONANCES/AliRsnAnalysisTaskSEBase.cxx
@@ -263,11 +263,13 @@ AliRsnEvent * AliRsnAnalysisTaskSEBase::GetRsnEventFromInputType(const Short_t &
 
   switch (fInputType[index])
   {
-    case kAOD:   return GetRsnFromAOD(index);
-    case kESD:   return GetRsnFromESD(index);
-    case kESDMC: return GetRsnFromESDMC(index);
-    case kMC:    return GetRsnFromMC(index);
-    case kRSN:   return GetRsnFromRSN(index);
+    case kAOD:      return GetRsnFromAOD(index);
+    case kESD:      return GetRsnFromESD(index);
+    case kESDMC:    return GetRsnFromESDMC(index);
+    case kESDTPC:   return GetRsnFromESD(index);
+    case kESDMCTPC: return GetRsnFromESDMC(index);
+    case kMC:       return GetRsnFromMC(index);
+    case kRSN:      return GetRsnFromRSN(index);
     default:
       AliError("Type not supported ...");
       return (AliRsnEvent*) 0x0;
@@ -311,6 +313,9 @@ AliRsnEvent * AliRsnAnalysisTaskSEBase::GetRsnFromESD(const Short_t & index)
 // Gets RSN event from ESD
 //
 
+  // bool value to know if it is TPC only or not
+  Bool_t tpcOnly = (fInputType[index] == kESDTPC || fInputType[index] == kESDMCTPC);
+
   if (!fRsnESD[index])
   {
     AliError("fRsnESD not available.");
@@ -327,7 +332,7 @@ AliRsnEvent * AliRsnAnalysisTaskSEBase::GetRsnFromESD(const Short_t & index)
   // clear pevious event
   fRSN[index]->Clear();
 
-  if (!fReader.FillFromESD(fRSN[index], fRsnESD[index])) return (AliRsnEvent*) 0x0;
+  if (!fReader.FillFromESD(fRSN[index], fRsnESD[index], 0x0, tpcOnly)) return (AliRsnEvent*) 0x0;
 
   if (!fPID.Process(fRSN[index]))
   {
@@ -375,6 +380,9 @@ AliRsnEvent * AliRsnAnalysisTaskSEBase::GetRsnFromESDMC(const Short_t & index)
 // Gets RSN event from ESD and MC
 //
 
+  // bool value to know if it is TPC only or not
+  Bool_t tpcOnly = (fInputType[index] == kESDTPC || fInputType[index] == kESDMCTPC);
+
   if (!fRsnESD[index])
   {
     AliError("fRsnESD not available.");
@@ -393,7 +401,7 @@ AliRsnEvent * AliRsnAnalysisTaskSEBase::GetRsnFromESDMC(const Short_t & index)
   fRsnMC[index] = MCEvent();
 
   if (!fRsnMC[index]) return (AliRsnEvent *) 0x0;
-  if (!fReader.FillFromESD(fRSN[index], fRsnESD[index], fRsnMC[index])) return (AliRsnEvent*) 0x0;
+  if (!fReader.FillFromESD(fRSN[index], fRsnESD[index], fRsnMC[index], tpcOnly)) return (AliRsnEvent*) 0x0;
   if (!fPID.Process(fRSN[index]))
   {
     AliWarning("Failed PID");

diff --git a/PWG2/RESONANCES/AliRsnAnalysisTaskSEBase.h b/PWG2/RESONANCES/AliRsnAnalysisTaskSEBase.h
index 6de11401eefe4ba171b291848352fcfa92f06f21..bebf1a12b08fcf7b9b953145b8df57ed0509b1d0 100644 (file)
--- a/PWG2/RESONANCES/AliRsnAnalysisTaskSEBase.h
+++ b/PWG2/RESONANCES/AliRsnAnalysisTaskSEBase.h
@@ -38,6 +38,8 @@ class AliRsnAnalysisTaskSEBase : public AliAnalysisTaskSE
       kAOD = 0,\r
       kESD,\r
       kESDMC,\r
+      kESDTPC,\r
+      kESDMCTPC,\r
       kMC,\r
       kRSN,\r
       kLastIndex\r