[u/mrichter/AliRoot.git] / PWGCF / FEMTOSCOPY / AliFemto / AliAnalysisTaskFemto.h

//------------------------------------------------------
// AliAnalysisTaskFemto - A task for the analysis framework
// from the FEMTOSCOPY analysis of PWG2. Creates the necessary
// connection between the ESD or AOD input and the femtoscopic
// code.
// Author: Adam Kisiel, OSU; Adam.Kisiel@cern.ch
//------------------------------------------------------
#ifndef ALIANALYSISTASKFEMTO_H
#define ALIANALYSISTASKFEMTO_H

#include "TH1.h"

#include "AliESDInputHandler.h"
#include "AliAODHandler.h"
#include "AliAODInputHandler.h"
#include "AliMCEventHandler.h"
#include "AliESDEvent.h"
#include "AliAODEvent.h"
#include "AliMCEvent.h"

#include "AliAnalysisTask.h"
#include "AliAnalysisManager.h"
#include "AliAnalysisDataContainer.h"

#include "AliFemtoEventReaderESDChain.h"
#include "AliFemtoEventReaderESDChainKine.h"
#include "AliFemtoEventReaderAODChain.h"
#include "AliFemtoEventReaderStandard.h"
#include "AliFemtoEventReaderKinematicsChain.h"
#include "AliFemtoManager.h"

#include "AliESDpid.h"
#include "AliAODpidUtil.h"


class AliAnalysisTaskFemto : public AliAnalysisTask {
 public:
  AliAnalysisTaskFemto() : AliAnalysisTask(), fESD(0), fESDpid(0), fAOD(0), fAODpidUtil(0), fStack(0), fOutputList(0), fReader(0x0), fManager(0x0), fAnalysisType(0), fConfigMacro(0), fConfigParams(0) {}
  AliAnalysisTaskFemto(const char *name, const char *aConfigMacro, const char *aConfigParams);
  AliAnalysisTaskFemto(const char *name, const char *aConfigMacro);
  AliAnalysisTaskFemto(const AliAnalysisTaskFemto& aFemtoTask);
  virtual ~AliAnalysisTaskFemto();
  
  AliAnalysisTaskFemto& operator=(const AliAnalysisTaskFemto& aFemtoTask);

  virtual void   ConnectInputData(Option_t *);
  virtual void   CreateOutputObjects();
  virtual void   Exec(Option_t *option);
  virtual void   Terminate(Option_t *);
  virtual void   FinishTaskOutput();

  void SetFemtoReaderESD(AliFemtoEventReaderESDChain *aReader);
  void SetFemtoReaderESDKine(AliFemtoEventReaderESDChainKine *aReader);
  void SetFemtoReaderAOD(AliFemtoEventReaderAODChain *aReader);
  void SetFemtoReaderStandard(AliFemtoEventReaderStandard *aReader);
  void SetFemtoManager(AliFemtoManager *aManager);
  void SetFemtoReaderKinematics(AliFemtoEventReaderKinematicsChain *aReader);

 private:
  AliESDEvent                 *fESD;          //! ESD object
  AliESDpid                   *fESDpid;       //! ESDpid object
  AliAODEvent                 *fAOD;          //! AOD object
  AliAODpidUtil               *fAODpidUtil;   // AliAODpidUtil object

  AliStack                    *fStack;        //! Stack from Kinematics
  TList                       *fOutputList;   //  AliFemto results list
  AliFemtoEventReader         *fReader;       //! Reference to the reader
  AliFemtoManager             *fManager;      //! AliFemto top-level manager 
  int                          fAnalysisType; //  Mark ESD of AOD analysis
  char                        *fConfigMacro;  //  Config macro location
  char                        *fConfigParams; //  Config macro parameters


  ClassDef(AliAnalysisTaskFemto, 3); // example of analysis
};

#endif
Commit	Line	Data
76ce4b5b	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