Cosmetics

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

diff --git a/ANALYSIS/AliAnalysisTaskESDfilter.h b/ANALYSIS/AliAnalysisTaskESDfilter.h
index 5e8adda3f1aaf01a00ab8675ea08b687b6470ce9..97bcac882a4bf43cd7fac38fd4be4140514ceffa 100644 (file)
--- a/ANALYSIS/AliAnalysisTaskESDfilter.h
+++ b/ANALYSIS/AliAnalysisTaskESDfilter.h
@@ -7,9 +7,14 @@
 /* $Id: AliAnalysisTaskESDfilter.h 24429 2008-03-12 10:27:50Z jgrosseo $ */
 
 #include <TList.h> 
+#include <TF1.h> 
 #include "AliAnalysisTaskSE.h"
-
+#include "AliESDtrack.h"
+#include "AliAODTrack.h"
+#include "AliAODPid.h"
 class AliAnalysisFilter;
+class AliStack;
+class AliESDtrack;
 
 class AliAnalysisTaskESDfilter : public AliAnalysisTaskSE
 {
@@ -18,26 +23,40 @@ 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 SetTrackFilter   (AliAnalysisFilter*   trackF) {fTrackFilter    =   trackF;}
+    virtual void SetKinkFilter    (AliAnalysisFilter*    KinkF) {fKinkFilter     =    KinkF;}
+    virtual void SetV0Filter      (AliAnalysisFilter*      V0F) {fV0Filter       =      V0F;}
+    virtual void SetCascadeFilter (AliAnalysisFilter* CascadeF) {fCascadeFilter  = CascadeF;}
+    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, Double_t bfield);
+    void SetDetectorRawSignals(AliAODPid *aodpid, AliESDtrack *track, Double_t timezero, Double_t bfield);
 
  private:
     AliAnalysisTaskESDfilter(const AliAnalysisTaskESDfilter&);
     AliAnalysisTaskESDfilter& operator=(const AliAnalysisTaskESDfilter&);
-    AliAnalysisFilter* fTrackFilter; //  Track Filter
-    AliAnalysisFilter* fKinkFilter;  //  Kink  Filter
-    AliAnalysisFilter* fV0Filter;    //  V0    Filter    
-    ClassDef(AliAnalysisTaskESDfilter, 1); // Analysis task for standard ESD filtering
+    void PrintMCInfo(AliStack *pStack,Int_t label); // for debugging
+    Double_t Chi2perNDF(AliESDtrack* track);
+    
+    // Filtering
+    AliAnalysisFilter* fTrackFilter;      //  Track   Filter
+    AliAnalysisFilter* fKinkFilter;       //  Kink    Filter
+    AliAnalysisFilter* fV0Filter;         //  V0      Filter
+    AliAnalysisFilter* fCascadeFilter;    //  Cascade Filter
+    // PID
+    Double_t     fHighPthreshold;    //  Pt threshold for detector signal setting
+    TF1 *        fPtshape;           //  Pt spectrum distribution
+
+    ClassDef(AliAnalysisTaskESDfilter, 3); // Analysis task for standard ESD filtering
 };
  
 #endif