[u/mrichter/AliRoot.git] / PWGGA / GammaConv / AliAnalysisTaskMaterial.h

#ifndef AliAnalysisTaskMaterial_cxx
#define AliAnalysisTaskMaterial_cxx

#include "AliAnalysisTaskSE.h"
#include "AliConversionPhotonBase.h"
#include "TH1.h"
#include "TH2.h"
#include "TTreeStream.h"
#include "AliLog.h"
#include <vector>
#include "AliV0ReaderV1.h"
#include "AliConversionCuts.h"
#include "TList.h"
#include "AliStack.h"
#include "TClonesArray.h"


using namespace std;


class AliAnalysisTaskMaterial : public AliAnalysisTaskSE{

 public:

   AliAnalysisTaskMaterial(const char *name);
   virtual ~AliAnalysisTaskMaterial();

   virtual void   UserCreateOutputObjects();
   virtual void   UserExec(Option_t *option);
   virtual void   Terminate(Option_t *);

   void SetV0Reader(AliV0ReaderV1 *v0Reader){fV0Reader=v0Reader;}
   void SetConversionCuts(AliConversionCuts* conversionCuts,Bool_t IsHeavyIon ){
      fConversionCuts=conversionCuts;
      fIsHeavyIon = IsHeavyIon;
   }

 private:

   void ProcessPhotons();
   void ProcessMCPhotons();
   void FillMCTree(Int_t stackPos);
   Int_t CountTracks0914();
   Int_t CountTracks09();

   AliV0ReaderV1 *fV0Reader;
   TClonesArray *fConversionGammas; //Reconstructed Photons;
   AliConversionCuts *fConversionCuts; // Cuts used by the V0Reader
   TTreeSRedirector *fStreamMaterial;
   TTreeSRedirector *fStreamResolution;
   Bool_t fIsHeavyIon;
   TList *fOutputList;
   AliESDEvent *fESDEvent;
   AliMCEvent *fMCEvent;

   AliAnalysisTaskMaterial(const AliAnalysisTaskMaterial&); // not implemented
   AliAnalysisTaskMaterial& operator=(const AliAnalysisTaskMaterial&); // not implemented


   ClassDef(AliAnalysisTaskMaterial, 0);
};

#endif
Commit	Line	Data
ca91a3e1	1	#ifndef AliAnalysisTaskMaterial_cxx
	2	#define AliAnalysisTaskMaterial_cxx
	3
	4	#include "AliAnalysisTaskSE.h"
	5	#include "AliConversionPhotonBase.h"
	6	#include "TH1.h"
	7	#include "TH2.h"
	8	#include "TTreeStream.h"
	9	#include "AliLog.h"
	10	#include <vector>
	11	#include "AliV0ReaderV1.h"
	12	#include "AliConversionCuts.h"
	13	#include "TList.h"
	14	#include "AliStack.h"
	15	#include "TClonesArray.h"
	16
	17
	18	using namespace std;
	19
	20
	21	class AliAnalysisTaskMaterial : public AliAnalysisTaskSE{
	22
a280ac15	23	public:
	24
	25	AliAnalysisTaskMaterial(const char *name);
	26	virtual ~AliAnalysisTaskMaterial();
	27
	28	virtual void UserCreateOutputObjects();
	29	virtual void UserExec(Option_t *option);
	30	virtual void Terminate(Option_t *);
	31
	32	void SetV0Reader(AliV0ReaderV1 *v0Reader){fV0Reader=v0Reader;}
	33	void SetConversionCuts(AliConversionCuts* conversionCuts,Bool_t IsHeavyIon ){
	34	fConversionCuts=conversionCuts;
	35	fIsHeavyIon = IsHeavyIon;
	36	}
	37
	38	private:
	39
	40	void ProcessPhotons();
	41	void ProcessMCPhotons();
	42	void FillMCTree(Int_t stackPos);
ae947965	43	Int_t CountTracks0914();
ae947965	44	Int_t CountTracks09();
a280ac15	45
	46	AliV0ReaderV1 *fV0Reader;
	47	TClonesArray *fConversionGammas; //Reconstructed Photons;
	48	AliConversionCuts *fConversionCuts; // Cuts used by the V0Reader
	49	TTreeSRedirector *fStreamMaterial;
	50	TTreeSRedirector *fStreamResolution;
	51	Bool_t fIsHeavyIon;
	52	TList *fOutputList;
	53	AliESDEvent *fESDEvent;
	54	AliMCEvent *fMCEvent;
	55
	56	AliAnalysisTaskMaterial(const AliAnalysisTaskMaterial&); // not implemented
	57	AliAnalysisTaskMaterial& operator=(const AliAnalysisTaskMaterial&); // not implemented
	58
	59
	60	ClassDef(AliAnalysisTaskMaterial, 0);
ca91a3e1	61	};
	62
	63	#endif