PWGCF/FEMTOSCOPY/AliFemto/AliAnalysisTaskFemto.h

   1 //------------------------------------------------------
   2 // AliAnalysisTaskFemto - A task for the analysis framework
   3 // from the FEMTOSCOPY analysis of PWG2. Creates the necessary
   4 // connection between the ESD or AOD input and the femtoscopic
   5 // code.
   6 // Author: Adam Kisiel, OSU; Adam.Kisiel@cern.ch
   7 //------------------------------------------------------
   8 #ifndef ALIANALYSISTASKFEMTO_H
   9 #define ALIANALYSISTASKFEMTO_H
  10
  11 #include "TH1.h"
  12 #include "TString.h"
  13
  14 #include "AliESDInputHandler.h"
  15 #include "AliAODHandler.h"
  16 #include "AliAODInputHandler.h"
  17 #include "AliMCEventHandler.h"
  18 #include "AliESDEvent.h"
  19 #include "AliAODEvent.h"
  20 #include "AliMCEvent.h"
  21
  22 //#include "AliAnalysisTask.h"
  23 #include "AliAnalysisTaskSE.h"
  24 #include "AliAnalysisManager.h"
  25 #include "AliAnalysisDataContainer.h"
  26
  27 #include "AliFemtoEventReaderESDChain.h"
  28 #include "AliFemtoEventReaderESDChainKine.h"
  29 #include "AliFemtoEventReaderAODChain.h"
  30 #include "AliFemtoEventReaderStandard.h"
  31 #include "AliFemtoEventReaderKinematicsChain.h"
  32 #include "AliFemtoEventReaderKinematicsChainESD.h"
  33 #include "AliFemtoManager.h"
  34
  35 #include "AliESDpid.h"
  36 #include "AliAODpidUtil.h"
  37 #include "AliAODHeader.h"
  38
  39
  40 class AliAnalysisTaskFemto : public AliAnalysisTaskSE { //AliAnalysisTask
  41  public:
  42 AliAnalysisTaskFemto() : AliAnalysisTaskSE(), fESD(0), fESDpid(0), fAOD(0), fAODpidUtil(0), fAODheader(0), fStack(0), fOutputList(0), fReader(0x0), fManager(0x0), fAnalysisType(0), fConfigMacro(0), fConfigParams(0), fVerbose(kTRUE) {}
  43     AliAnalysisTaskFemto(TString name, TString aConfigMacro, TString aConfigParams, Bool_t aVerbose=kFALSE);
  44     AliAnalysisTaskFemto(TString name, TString aConfigMacro, Bool_t aVerbose=kFALSE);
  45   AliAnalysisTaskFemto(const AliAnalysisTaskFemto& aFemtoTask);
  46   virtual ~AliAnalysisTaskFemto();
  47
  48   AliAnalysisTaskFemto& operator=(const AliAnalysisTaskFemto& aFemtoTask);
  49
  50   virtual void   ConnectInputData(Option_t *);
  51   virtual void   CreateOutputObjects();
  52   virtual void   Exec(Option_t *option);
  53   virtual void   Terminate(Option_t *);
  54   virtual void   FinishTaskOutput();
  55
  56   void SetFemtoReaderESD(AliFemtoEventReaderESDChain *aReader);
  57   void SetFemtoReaderESDKine(AliFemtoEventReaderESDChainKine *aReader);
  58   void SetFemtoReaderAOD(AliFemtoEventReaderAODChain *aReader);
  59   void SetFemtoReaderStandard(AliFemtoEventReaderStandard *aReader);
  60   void SetFemtoManager(AliFemtoManager *aManager);
  61   void SetFemtoReaderKinematics(AliFemtoEventReaderKinematicsChain *aReader);
  62   void SetFemtoReaderKinematicsESD(AliFemtoEventReaderKinematicsChainESD *aReader);
  63
  64  private:
  65   AliESDEvent                 *fESD;          //! ESD object
  66   AliESDpid                   *fESDpid;       //! ESDpid object
  67   AliAODEvent                 *fAOD;          //! AOD object
  68   AliAODpidUtil               *fAODpidUtil;   // AliAODpidUtil object
  69   AliAODHeader                *fAODheader;    //AliAODHeader object (to get reference multiplicity in pp)
  70
  71   AliStack                    *fStack;        //! Stack from Kinematics
  72   TList                       *fOutputList;   //  AliFemto results list
  73   AliFemtoEventReader         *fReader;       //! Reference to the reader
  74   AliFemtoManager             *fManager;      //! AliFemto top-level manager
  75   int                          fAnalysisType; //  Mark ESD of AOD analysis
  76     TString                      fConfigMacro;  //  Config macro location
  77     TString                      fConfigParams; //  Config macro parameters
  78   Bool_t fVerbose;
  79
  80   ClassDef(AliAnalysisTaskFemto, 3); // example of analysis
  81 };
  82
  83 #endif
  84