Add option to write MC particles to the AOD, new lightweight class AliAODMCParticle...

[u/mrichter/AliRoot.git] / ANALYSIS / AliAnalysisTaskESDfilter.h

diff --git a/ANALYSIS/AliAnalysisTaskESDfilter.h b/ANALYSIS/AliAnalysisTaskESDfilter.h
index 7d419883dfbed3f8a21c8f5fb905ad4fdcc5ceec..1f66209289beed43bc248af0c2ae86d89305f7d3 100644 (file)
--- a/ANALYSIS/AliAnalysisTaskESDfilter.h
+++ b/ANALYSIS/AliAnalysisTaskESDfilter.h
@@ -13,6 +13,9 @@
 #include "AliAODTrack.h"
 #include "AliAODPid.h"
 class AliAnalysisFilter;
+class AliStack;
+
+
 
 class AliAnalysisTaskESDfilter : public AliAnalysisTaskSE
 {
@@ -21,34 +24,36 @@ class AliAnalysisTaskESDfilter : public AliAnalysisTaskSE
     AliAnalysisTaskESDfilter(const char* name);
     virtual ~AliAnalysisTaskESDfilter() {;}
     // Implementation of interface methods
-    virtual void UserCreateOutputObjects();
-    virtual void Init();
-    virtual void LocalInit() {Init();}
-    virtual void UserExec(Option_t *option);
-    virtual void Terminate(Option_t *option);
+    virtual void   UserCreateOutputObjects();
+    virtual void   Init();
+    virtual void   LocalInit() {Init();}
+    virtual void   UserExec(Option_t *option);
+    virtual void   Terminate(Option_t *option);
 
     virtual void ConvertESDtoAOD();
-
     // Setters
     virtual void SetTrackFilter(AliAnalysisFilter* trackF) {fTrackFilter = trackF;}
     virtual void SetKinkFilter (AliAnalysisFilter*  KinkF) {fKinkFilter  =  KinkF;}
     virtual void SetV0Filter   (AliAnalysisFilter*    V0F) {fV0Filter    =    V0F;}
-    virtual void SetPthreshold (Double_t p)             {fHighPthreshold =      p;}
-    virtual void SetPshape     (TF1 *func)             {fPtshape         =    func;}
+    virtual void SetPthreshold (Double_t p)                {fHighPthreshold =  p;}
+    virtual void SetPshape     (TF1 *func)                 {fPtshape        = func;}
 
     virtual void SetAODPID(AliESDtrack *esdtrack, AliAODTrack *aodtrack, AliAODPid *detpid, Double_t timezero);
-
+    void SetDetectorRawSignals(AliAODPid *aodpid, AliESDtrack *track, Double_t timezero);
 
  private:
     AliAnalysisTaskESDfilter(const AliAnalysisTaskESDfilter&);
     AliAnalysisTaskESDfilter& operator=(const AliAnalysisTaskESDfilter&);
+    void PrintMCInfo(AliStack *pStack,Int_t label); // for debugging
+
+    // Filtering
     AliAnalysisFilter* fTrackFilter; //  Track Filter
     AliAnalysisFilter* fKinkFilter;  //  Kink  Filter
-    AliAnalysisFilter* fV0Filter;    //  V0    Filter    
+    AliAnalysisFilter* fV0Filter;    //  V0    Filter
+    // PID
     Double_t     fHighPthreshold;    //  Pt threshold for detector signal setting
     TF1 *        fPtshape;           //  Pt spectrum distribution
-
-    ClassDef(AliAnalysisTaskESDfilter, 1); // Analysis task for standard ESD filtering
+    ClassDef(AliAnalysisTaskESDfilter, 2); // Analysis task for standard ESD filtering
 };
  
 #endif