PWGLF/SPECTRA/PiKaPr/TestAOD/AliAnalysisTaskSpectraAllChAOD.h

   1 #ifndef ALIANALYSISTASKSPECTRAALLCHAOD_H
   2 #define ALIANALYSISTASKSPECTRAALLCHAOD_H
   3
   4 /*  See cxx source for full Copyright notice */
   5
   6 //-------------------------------------------------------------------------
   7 //                      AliAnalysisTaskSpectraAllChAOD
   8 //
   9 //
  10 //
  11 //
  12 // Author: Leonardo Milano, CERN
  13 //-------------------------------------------------------------------------
  14
  15 class AliAODEvent;
  16 class AliSpectraAODTrackCuts;
  17 class AliSpectraAODEventCuts;
  18 class AliHelperPID;
  19
  20 #include "AliAnalysisTaskSE.h"
  21
  22 class AliAnalysisTaskSpectraAllChAOD : public AliAnalysisTaskSE
  23 {
  24  public:
  25   // constructors
  26  AliAnalysisTaskSpectraAllChAOD() : AliAnalysisTaskSE(),
  27     fAOD(0x0),
  28     fTrackCuts(0x0),
  29     fEventCuts(0x0),
  30     fHelperPID(0x0),
  31     fIsMC(0),
  32     fDoDoubleCounting(0),
  33     fFillOnlyEvents(0),
  34     fCharge(0),
  35     fVZEROside(0),
  36     fOutput(0x0),
  37     fnCentBins(20),
  38     fnQvecBins(40)
  39       {}
  40   AliAnalysisTaskSpectraAllChAOD(const char *name);
  41   virtual ~AliAnalysisTaskSpectraAllChAOD() {
  42     Printf("calling detructor of AliAnalysisTaskSpectraAllChAOD - To be implemented");
  43   }
  44
  45   void SetIsMC(Bool_t isMC = kFALSE)    {fIsMC = isMC; };
  46   Bool_t GetIsMC()           const           { return fIsMC;};
  47
  48   void SetDoDoubleCounting(Bool_t doDoubleCounting = kFALSE)    {fDoDoubleCounting = doDoubleCounting; };
  49   Bool_t GetDoDoubleCounting()           const           { return fDoDoubleCounting;};
  50
  51   void SetFillOnlyEvents(Bool_t fillOnlyEvents = kFALSE)    {fFillOnlyEvents = fillOnlyEvents; };
  52   Bool_t GetFillOnlyEvents()           const           { return fFillOnlyEvents;};
  53
  54   void SetCharge(Int_t charge = 0)    {fCharge = charge; };
  55   Int_t GetCharge()           const           { return fCharge;};
  56
  57   void SetVZEROside(Int_t side = 0)    {fVZEROside = side; };
  58   Int_t GetVZEROside()           const           { return fVZEROside;};
  59
  60   virtual void   UserCreateOutputObjects();
  61   virtual void   UserExec(Option_t *option);
  62   virtual void   Terminate(Option_t *);
  63
  64   AliSpectraAODTrackCuts      * GetTrackCuts()         {  return fTrackCuts; }
  65   AliSpectraAODEventCuts      * GetEventCuts()         {  return fEventCuts; }
  66   AliHelperPID                   * GetHelperPID()          { return fHelperPID; }
  67   TList                          * GetOutputList()         { return fOutput; }
  68
  69   void SetTrackCuts(AliSpectraAODTrackCuts * tc)       { fTrackCuts = tc; }
  70   void SetEventCuts(AliSpectraAODEventCuts * vc)       { fEventCuts = vc; }
  71   void SetHelperPID(AliHelperPID* pid)                     { fHelperPID = pid; }
  72   void SetnCentBins(Int_t val)                             { fnCentBins = val; }
  73   void SetnQvecBins(Int_t val)                             { fnQvecBins = val; }
  74
  75  private:
  76
  77   AliAODEvent                   * fAOD;                         //! AOD object
  78   AliSpectraAODTrackCuts      * fTrackCuts;                   // Track Cuts
  79   AliSpectraAODEventCuts      * fEventCuts;                   // Event Cuts
  80   AliHelperPID                   * fHelperPID;                    // points to class for PID
  81   Bool_t                          fIsMC;                         // true if processing MC
  82   Bool_t                          fDoDoubleCounting;           // true is double counting for Nsigma accepetd
  83   Bool_t                          fFillOnlyEvents;               // if true fill only NSparseHistEv
  84   Int_t                            fCharge;                      // charge to be selected
  85   Int_t                            fVZEROside;                  // 0: VZERO-A 1: VZERO-C
  86   TList                          * fOutput;                     // output list
  87   Int_t                            fnCentBins;                  // number of bins for the centrality axis
  88   Int_t                            fnQvecBins;                 // number of bins for the q vector axis
  89   AliAnalysisTaskSpectraAllChAOD(const AliAnalysisTaskSpectraAllChAOD&);
  90   AliAnalysisTaskSpectraAllChAOD& operator=(const AliAnalysisTaskSpectraAllChAOD&);
  91
  92   ClassDef(AliAnalysisTaskSpectraAllChAOD, 5);
  93 };
  94
  95 #endif