PWGLF/SPECTRA/AntiprotonToProton/AliProtonCorrectionAnalysisTask.h

   1 #ifndef ALIPROTONCORRECTIONTASK_H
   2 #define ALIPROTONCORRECTIONTASK_H
   3 #include "AliAnalysisTask.h"
   4
   5 class TList;
   6 class AliESDEvent;
   7 class AliAODEvent;
   8 class AliMCEvent;
   9 class AliProtonFeedDownAnalysis;
  10 class AliProtonAbsorptionCorrection;
  11 class AliProtonSpectraCorrection;
  12 class AliProtonAnalysisBase;
  13
  14
  15
  16 class AliProtonCorrectionAnalysisTask : public AliAnalysisTask {
  17  public:
  18         AliProtonCorrectionAnalysisTask();
  19         AliProtonCorrectionAnalysisTask(const char *name);
  20         virtual ~AliProtonCorrectionAnalysisTask() {}
  21
  22         virtual void   ConnectInputData(Option_t *);
  23         virtual void   CreateOutputObjects();
  24         virtual void   Exec(Option_t *option);
  25         virtual void   Terminate(Option_t *);
  26
  27         void SetAnalysisObjectAbsorptionCorrection(AliProtonAbsorptionCorrection* const analysis) ;
  28         void SetAnalysisObjectFeedDown(AliProtonFeedDownAnalysis* const analysis);
  29         void SetAnalysisObjectSpectraCorrection(AliProtonSpectraCorrection* const analysis);
  30         void SetBaseAnalysis(AliProtonAnalysisBase* const baseAnalysis) { fProtonAnalysisBase = baseAnalysis;}
  31  private:
  32         AliESDEvent *fESD;    //ESD object
  33         AliAODEvent *fAOD;    //AOD object
  34         AliMCEvent  *fMC;     //MC object
  35
  36         TList  *fList; //TList output object
  37
  38         AliProtonAnalysisBase *fProtonAnalysisBase; //base analysis object
  39         AliProtonAbsorptionCorrection *fProtonAbsorptionCorrection;//analysis object
  40         AliProtonFeedDownAnalysis *fProtonFeedDownAnalysis; //analysis object
  41         AliProtonSpectraCorrection *fProtonSpectraCorrection;//analysis object
  42
  43         TH1F *fStatHist;
  44
  45
  46         Bool_t fIsOn_AliProtonAbsorptionCorrection;
  47         Bool_t fIsOn_AliProtonFeedDownAnalysis;
  48         Bool_t fIsOn_AliProtonSpectraCorrection;
  49
  50         AliProtonCorrectionAnalysisTask(const AliProtonCorrectionAnalysisTask&); // not implemented
  51         AliProtonCorrectionAnalysisTask& operator=(const AliProtonCorrectionAnalysisTask&); // not implemented
  52
  53 ClassDef(AliProtonCorrectionAnalysisTask, 1); // example of analysis
  54 };
  55
  56 #endif
  57
  58
  59