PWGLF/SPECTRA/PiKaPr/TestAOD/AliAnalysisTaskSpectraAOD.h

   1 #ifndef ALIANALYSISTASKSPECTRAAOD_H
   2 #define ALIANALYSISTASKSPECTRAAOD_H
   3
   4 /*  See cxx source for full Copyright notice */
   5
   6 //-------------------------------------------------------------------------
   7 //                      AliAnalysisTaskSpectraAOD
   8 //
   9 //
  10 //
  11 //
  12 // Author: Michele Floris, CERN
  13 //-------------------------------------------------------------------------
  14
  15 class TH1F;
  16 class TH2F;
  17 class AliAODEvent;
  18 class AliSpectraAODHistoManager;
  19 class AliSpectraAODTrackCuts;
  20 class AliSpectraAODEventCuts;
  21 class AliSpectraAODPID;
  22 #include "AliSpectraAODHistoManager.h"
  23 #include "AliAnalysisTaskSE.h"
  24
  25
  26 class AliAnalysisTaskSpectraAOD : public AliAnalysisTaskSE
  27 {
  28 public:
  29
  30    // constructors
  31   AliAnalysisTaskSpectraAOD() : AliAnalysisTaskSE(), fAOD(0), fHistMan(0), fTrackCuts(0), fEventCuts(0), fPID(0), fIsMC(0), fNRebin(0)
  32  {}
  33   AliAnalysisTaskSpectraAOD(const char *name);
  34    virtual ~AliAnalysisTaskSpectraAOD() {}
  35
  36    void SetIsMC(Bool_t isMC = kFALSE)    {fIsMC = isMC; };
  37    Bool_t GetIsMC()           const           { return fIsMC;};
  38
  39    virtual void   UserCreateOutputObjects();
  40    virtual void   UserExec(Option_t *option);
  41    virtual void   Terminate(Option_t *);
  42
  43    AliSpectraAODHistoManager * GetHistoManager()         {  return fHistMan; }
  44    AliSpectraAODTrackCuts * GetTrackCuts()         {  return fTrackCuts; }
  45    AliSpectraAODEventCuts * GetEventCuts()         {  return fEventCuts; }
  46    AliSpectraAODPID * GetPID()         {  return fPID; }
  47
  48    void SetTrackCuts(AliSpectraAODTrackCuts * tc)   {   fTrackCuts = tc;   }
  49    void SetEventCuts(AliSpectraAODEventCuts * vc)   {   fEventCuts = vc;   }
  50    void SetPID      (AliSpectraAODPID      * pid)   {   fPID       = pid;  }
  51    void SetNRebin(Int_t nreb){fNRebin=nreb;}
  52    Int_t   GetNRebin() {return fNRebin;}
  53
  54 private:
  55
  56    AliAODEvent           * fAOD;         //! AOD object
  57    AliSpectraAODHistoManager      * fHistMan;       // Histogram Manager
  58    AliSpectraAODTrackCuts      * fTrackCuts;     // Track Cuts
  59    AliSpectraAODEventCuts      * fEventCuts;     // Event Cuts
  60    AliSpectraAODPID             * fPID;// PID class
  61    Bool_t          fIsMC;// true if processing MC
  62    Int_t      fNRebin; //rebin of histos
  63    AliAnalysisTaskSpectraAOD(const AliAnalysisTaskSpectraAOD&);
  64    AliAnalysisTaskSpectraAOD& operator=(const AliAnalysisTaskSpectraAOD&);
  65
  66    ClassDef(AliAnalysisTaskSpectraAOD, 1);
  67 };
  68
  69 #endif