PWG2/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
  13 #include "AliESDInputHandler.h"
  14 #include "AliAODHandler.h"
  15 #include "AliAODInputHandler.h"
  16 #include "AliMCEventHandler.h"
  17 #include "AliESDEvent.h"
  18 #include "AliAODEvent.h"
  19 #include "AliMCEvent.h"
  20
  21 #include "AliAnalysisTask.h"
  22 #include "AliAnalysisManager.h"
  23 #include "AliAnalysisDataContainer.h"
  24
  25 #include "AliFemtoEventReaderESDChain.h"
  26 #include "AliFemtoEventReaderESDChainKine.h"
  27 #include "AliFemtoEventReaderAODChain.h"
  28 #include "AliFemtoEventReaderStandard.h"
  29 #include "AliFemtoEventReaderKinematicsChain.h"
  30 #include "AliFemtoManager.h"
  31
  32 #include "AliESDpid.h"
  33 #include "AliAODpidUtil.h"
  34
  35
  36 class AliAnalysisTaskFemto : public AliAnalysisTask {
  37  public:
  38   AliAnalysisTaskFemto() : AliAnalysisTask(), fESD(0), fESDpid(0), fAOD(0), fAODpidUtil(0), fStack(0), fOutputList(0), fReader(0x0), fManager(0x0), fAnalysisType(0), fConfigMacro(0), fConfigParams(0) {}
  39   AliAnalysisTaskFemto(const char *name, const char *aConfigMacro, const char *aConfigParams);
  40   AliAnalysisTaskFemto(const char *name, const char *aConfigMacro);
  41   AliAnalysisTaskFemto(const AliAnalysisTaskFemto& aFemtoTask);
  42   virtual ~AliAnalysisTaskFemto();
  43
  44   AliAnalysisTaskFemto& operator=(const AliAnalysisTaskFemto& aFemtoTask);
  45
  46   virtual void   ConnectInputData(Option_t *);
  47   virtual void   CreateOutputObjects();
  48   virtual void   Exec(Option_t *option);
  49   virtual void   Terminate(Option_t *);
  50   virtual void   FinishTaskOutput();
  51
  52   void SetFemtoReaderESD(AliFemtoEventReaderESDChain *aReader);
  53   void SetFemtoReaderESDKine(AliFemtoEventReaderESDChainKine *aReader);
  54   void SetFemtoReaderAOD(AliFemtoEventReaderAODChain *aReader);
  55   void SetFemtoReaderStandard(AliFemtoEventReaderStandard *aReader);
  56   void SetFemtoManager(AliFemtoManager *aManager);
  57   void SetFemtoReaderKinematics(AliFemtoEventReaderKinematicsChain *aReader);
  58
  59  private:
  60   AliESDEvent                 *fESD;          //! ESD object
  61   AliESDpid                   *fESDpid;       //! ESDpid object
  62   AliAODEvent                 *fAOD;          //! AOD object
  63   AliAODpidUtil               *fAODpidUtil;   // AliAODpidUtil object
  64
  65   AliStack                    *fStack;        //! Stack from Kinematics
  66   TList                       *fOutputList;   //  AliFemto results list
  67   AliFemtoEventReader         *fReader;       //! Reference to the reader
  68   AliFemtoManager             *fManager;      //! AliFemto top-level manager
  69   int                          fAnalysisType; //  Mark ESD of AOD analysis
  70   char                        *fConfigMacro;  //  Config macro location
  71   char                        *fConfigParams; //  Config macro parameters
  72
  73
  74   ClassDef(AliAnalysisTaskFemto, 3); // example of analysis
  75 };
  76
  77 #endif
  78