[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();
   AliAnalysisTaskMaterial(const char *name);
   virtual ~AliAnalysisTaskMaterial();

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

   void SetIsMC(Bool_t isMC){fIsMC=isMC;}
   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
   TList *fOutputList; 
   TList *fEventList;
   TList *fRecGammaList;
   TList *fAllMCGammaList;
   TList *fAllMCConvGammaList;
   TTree* fTreeEvent;
   TTree* fTreeMaterialRec;
   TTree* fTreeMaterialAllGamma;
   TTree* fTreeMaterialConvGamma;
   Float_t fPrimVtxZ;
   Int_t fNContrVtx;
   Int_t fNESDtracksEta09;
   Int_t fNESDtracksEta0914;
   Int_t fNESDtracksEta14;
   Float_t fGammaMCPt;
   Float_t fGammaMCTheta;
   Float_t fGammaMCConvPt;
   Float_t fGammaMCConvTheta;
   TVectorF fMCConvCords;
   TVectorF fMCConvDaughterProp;
   Float_t fGammaPt;
   Float_t fGammaTheta;
   Float_t fGammaChi2NDF;
   TVectorF fRecCords;
   TVectorF fDaughterProp;
   UChar_t fKind;
  
   
   Bool_t fIsHeavyIon;
   Bool_t fIsMC;
   AliESDEvent *fESDEvent;
   AliMCEvent *fMCEvent;

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


   ClassDef(AliAnalysisTaskMaterial, 1);
};

#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:
a280ac15	24
72395bd9	25	AliAnalysisTaskMaterial();
a280ac15	26	AliAnalysisTaskMaterial(const char *name);
	27	virtual ~AliAnalysisTaskMaterial();
	28
	29	virtual void UserCreateOutputObjects();
	30	virtual void UserExec(Option_t *option);
	31	virtual void Terminate(Option_t *);
	32
72395bd9	33	void SetIsMC(Bool_t isMC){fIsMC=isMC;}
a280ac15	34	void SetV0Reader(AliV0ReaderV1 *v0Reader){fV0Reader=v0Reader;}
	35	void SetConversionCuts(AliConversionCuts* conversionCuts,Bool_t IsHeavyIon ){
	36	fConversionCuts=conversionCuts;
	37	fIsHeavyIon = IsHeavyIon;
	38	}
	39
	40	private:
	41
	42	void ProcessPhotons();
	43	void ProcessMCPhotons();
	44	void FillMCTree(Int_t stackPos);
ae947965	45	Int_t CountTracks0914();
ae947965	46	Int_t CountTracks09();
a280ac15	47
	48	AliV0ReaderV1 *fV0Reader;
	49	TClonesArray *fConversionGammas; //Reconstructed Photons;
	50	AliConversionCuts *fConversionCuts; // Cuts used by the V0Reader
72395bd9	51	TList *fOutputList;
	52	TList *fEventList;
	53	TList *fRecGammaList;
	54	TList *fAllMCGammaList;
	55	TList *fAllMCConvGammaList;
	56	TTree* fTreeEvent;
	57	TTree* fTreeMaterialRec;
	58	TTree* fTreeMaterialAllGamma;
	59	TTree* fTreeMaterialConvGamma;
	60	Float_t fPrimVtxZ;
	61	Int_t fNContrVtx;
	62	Int_t fNESDtracksEta09;
	63	Int_t fNESDtracksEta0914;
	64	Int_t fNESDtracksEta14;
	65	Float_t fGammaMCPt;
	66	Float_t fGammaMCTheta;
	67	Float_t fGammaMCConvPt;
	68	Float_t fGammaMCConvTheta;
31216736	69	TVectorF fMCConvCords;
31216736	70	TVectorF fMCConvDaughterProp;
72395bd9	71	Float_t fGammaPt;
	72	Float_t fGammaTheta;
	73	Float_t fGammaChi2NDF;
31216736	74	TVectorF fRecCords;
31216736	75	TVectorF fDaughterProp;
72395bd9	76	UChar_t fKind;
	77
	78
a280ac15	79	Bool_t fIsHeavyIon;
72395bd9	80	Bool_t fIsMC;
a280ac15	81	AliESDEvent *fESDEvent;
	82	AliMCEvent *fMCEvent;
	83
	84	AliAnalysisTaskMaterial(const AliAnalysisTaskMaterial&); // not implemented
	85	AliAnalysisTaskMaterial& operator=(const AliAnalysisTaskMaterial&); // not implemented
	86
	87
72395bd9	88	ClassDef(AliAnalysisTaskMaterial, 1);
ca91a3e1	89	};
	90
	91	#endif