PWGCF/Correlations/DPhi/AliAnalysisTaskContMC.h

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