PWGCF/Correlations/DPhi/AliAnalysisTaskContMC.h

   1 #ifndef ALIANALYSISTASKCONTMC_H
   2 #define ALIANALYSISTASKCONTMC_H
   3
   4 class TH1F;
   5 class TH2F;
   6 class TH3F;
   7 class AliAODEvent;
   8 class AliHelperPID;
   9
  10 #include "AliAnalysisTaskSE.h"
  11 #include "AliHelperPID.h"
  12
  13 class AliAnalysisTaskContMC : public AliAnalysisTaskSE
  14 {
  15  public:
  16
  17   // constructors
  18  AliAnalysisTaskContMC() : AliAnalysisTaskSE(), fAOD(0), fNSigmaPID(0), fIsMC(0), fOutput(0), fHistID(0)
  19     {}
  20   AliAnalysisTaskContMC(const char *name);
  21   virtual ~AliAnalysisTaskContMC() {}
  22
  23   AliHelperPID * GetPID()         {  return fNSigmaPID; }
  24   void SetPID      (AliHelperPID      * pid)   {   fNSigmaPID  = pid;}
  25
  26   void SetIsMC(Bool_t isMC = kFALSE)    {fIsMC = isMC; }
  27   Bool_t GetIsMC()           const           { return fIsMC;}
  28
  29   virtual void   UserCreateOutputObjects();
  30   virtual void   UserExec(Option_t *option);
  31   virtual void   Terminate(Option_t *);
  32
  33  private:
  34
  35   AliAODEvent           *fAOD;         //! AOD object
  36   AliHelperPID           *fNSigmaPID;  // NSigmaPID object
  37   Bool_t          fIsMC;// true if processing MC
  38   TList *fOutput; //! tlist with output
  39   TH3F *fHistID; //! histo
  40
  41   AliAnalysisTaskContMC(const AliAnalysisTaskContMC&);
  42   AliAnalysisTaskContMC& operator=(const AliAnalysisTaskContMC&);
  43
  44   ClassDef(AliAnalysisTaskContMC, 1);
  45 };
  46
  47 #endif