PWG2/AOD/AliAnalysisTaskPWG2ESDfilter.h

   1 //-------------------------------------------------------------------------
   2 //     Task for the Analysis Framework
   3 // Creates a PWG2 specific AOD, from the ESD, adding information required
   4 // for PWG2 analysis.
   5 //  - Puts the per-track information into the AliPWG2AODTrack container,
   6 //    together with the link to the original AliAODTrack
   7 //
   8 //     Author: Adam Kisiel, OSU, Adam.Kisiel@cern.ch
   9 //-------------------------------------------------------------------------
  10 #ifndef ALIANALYSISTASKPWG2ESDFILTER_H
  11 #define ALIANALYSISTASKPWG2ESDFILTER_H
  12
  13 #include <AliAnalysisTask.h>
  14
  15 class AliESDEvent;
  16 class TChain;
  17 class AliAODEvent;
  18 class AliAnalysisFilter;
  19 class TTree;
  20 class TClonesArray;
  21 class TList;
  22
  23 class AliAnalysisTaskPWG2ESDfilter : public AliAnalysisTask
  24 {
  25  public:
  26     AliAnalysisTaskPWG2ESDfilter();
  27     AliAnalysisTaskPWG2ESDfilter(const char* name);
  28     virtual ~AliAnalysisTaskPWG2ESDfilter() {;}
  29     // Implementation of interface methods
  30     virtual void ConnectInputData(Option_t *option = "");
  31     virtual void CreateOutputObjects();
  32     virtual void Init();
  33     virtual void LocalInit() {Init();}
  34     virtual void Exec(Option_t *option);
  35     virtual void Terminate(Option_t *option);
  36     // Setters
  37     virtual void SetTrackFilter(AliAnalysisFilter* trackF) {fTrackFilter = trackF;}
  38     virtual void SetKinkFilter (AliAnalysisFilter*  KinkF) {fKinkFilter  =  KinkF;}
  39     virtual void SetV0Filter   (AliAnalysisFilter*    V0F) {fV0Filter    =    V0F;}
  40     virtual void SetDebugLevel(Int_t level) {fDebug = level;}
  41
  42  private:
  43     Int_t              fDebug;         //  Debug flag
  44     TTree*             fTree;          //! chained files
  45     AliESDEvent*       fESD;           //! ESD
  46     AliAODEvent*       fAOD;           //! AOD event
  47     TTree*             fTreeA;         //! AOD tree
  48     AliAnalysisFilter* fTrackFilter;   //  Track Filter
  49     AliAnalysisFilter* fKinkFilter;    //  Kink  Filter
  50     AliAnalysisFilter* fV0Filter;      //  V0    Filter
  51     TClonesArray*      fPWG2AODTracks; //! container for PWG2 specific information
  52
  53     ClassDef(AliAnalysisTaskPWG2ESDfilter, 1); // Analysis task for standard ESD filtering
  54 };
  55
  56 #endif