Including the AliESDtrackCuts and the plugin

[u/mrichter/AliRoot.git] / PWG2 / EBYE / AliAnalysisTaskBF.cxx

diff --git a/PWG2/EBYE/AliAnalysisTaskBF.cxx b/PWG2/EBYE/AliAnalysisTaskBF.cxx
index b2b344f75490601383cd7a4934414bbf3afad5c6..594ce38476b744ec9defa4a45e1892a24fa9313f 100755 (executable)
--- a/PWG2/EBYE/AliAnalysisTaskBF.cxx
+++ b/PWG2/EBYE/AliAnalysisTaskBF.cxx
@@ -17,6 +17,7 @@
 #include "AliMCEventHandler.h"
 #include "AliMCEvent.h"
 #include "AliStack.h"
+#include "AliESDtrackCuts.h"
 
 #include "AliBalance.h"
 
@@ -32,7 +33,8 @@ AliAnalysisTaskBF::AliAnalysisTaskBF(const char *name)
   : AliAnalysisTaskSE(name), 
     fBalance(0),
     fList(0),
-    fHistEventStats(0) {
+    fHistEventStats(0),
+    fESDtrackCuts(0) {
   // Constructor
 
   // Define input and output slots here
@@ -67,6 +69,8 @@ void AliAnalysisTaskBF::UserCreateOutputObjects() {
     fHistEventStats->GetXaxis()->SetBinLabel(i,gCutName[i-1].Data());
   fList->Add(fHistEventStats);
 
+  if(fESDtrackCuts) fList->Add(fESDtrackCuts);
+
   // Post output data.
   PostData(1, fBalance);
   PostData(2, fList);
@@ -105,6 +109,10 @@ void AliAnalysisTaskBF::UserExec(Option_t *) {
            Printf("ERROR: Could not receive track %d", iTracks);
            continue;
          }
+
+         //ESD track cuts
+         if(fESDtrackCuts) 
+           if(!fESDtrackCuts->AcceptTrack(track)) continue;
          array->Add(track);
        } //track loop
       }//vertex object valid