PWG0/dNdPt/AlidNdPtTask.h

   1 #ifndef ALIDNDPTTASK_H
   2 #define ALIDNDPTTASK_H
   3
   4 //------------------------------------------------------------------------------
   5 // Task for dNdPt analysis.
   6 //
   7 // Author: J.Otwinowski 04/11/2008
   8 //------------------------------------------------------------------------------
   9
  10 class AliESDEvent;
  11 class AliMCEvent;
  12 class AlidNdPtEventCuts;
  13 class AlidNdPtAcceptanceCuts;
  14 class AliESDtrackCuts;
  15 class AlidNdPt;
  16 class AlidNdPtAnalysis;
  17 class AlidNdPtCorrection;
  18 class AliMagFMaps;
  19 class TList;
  20
  21 #include "dNdPt/AlidNdPtHelper.h"
  22 #include "AliAnalysisTaskSE.h"
  23
  24 class AlidNdPtTask : public AliAnalysisTaskSE {
  25  public:
  26   AlidNdPtTask(const char *name = "AlidNdPtTask");
  27   virtual ~AlidNdPtTask();
  28
  29   virtual void   UserCreateOutputObjects();
  30   virtual void   UserExec(Option_t *option);
  31   virtual void   Terminate(Option_t *);
  32   virtual Bool_t Notify();
  33
  34   Bool_t AddAnalysisObject(AlidNdPt *pObj);
  35   void SetUseMCInfo(Bool_t info)           { fUseMCInfo = info; }
  36
  37   // Use centrality - if yes, which one
  38   void  SetUseCentrality(Int_t cent)   { fUseCentrality = cent; }
  39   Int_t GetUseCentrality()             { return fUseCentrality; }
  40
  41   // Centrality bin to be used
  42   void  SetUseCentralityBin(Int_t bin) { fUseCentralityBin = bin; }
  43   Int_t GetUseCentralityBin()          { return fUseCentralityBin; }
  44
  45  private:
  46
  47   // Calculate centrality
  48   Int_t CalculateCentralityBin();
  49
  50   AliESDEvent *fESD;    //! ESD event
  51   AliMCEvent *fMC;      //! MC event
  52   TList* fOutput;       //! list send on output slot 0
  53   TIterator *fPitList;  //! iterator over the output objetcs
  54   TList *fCompList;     // list of comparison objects
  55
  56   Bool_t fUseMCInfo;        // use MC information
  57
  58   Int_t fUseCentrality;     // use centrality (0=off(default),1=VZERO,2=SPD)
  59   Int_t fUseCentralityBin;  // centrality bin to be used
  60
  61   AlidNdPtTask(const AlidNdPtTask&); // not implemented
  62   AlidNdPtTask& operator=(const AlidNdPtTask&); // not implemented
  63
  64   ClassDef(AlidNdPtTask, 2); // example of analysis
  65 };
  66
  67 #endif