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

#ifndef ALIANLYSISTASKGAMMACONVV1_cxx
#define ALIANLYSISTASKGAMMACONVV1_cxx

#include "AliAnalysisTaskSE.h"
#include "AliESDtrack.h"
#include "AliV0ReaderV1.h"
#include "AliKFConversionPhoton.h"
#include "AliGammaConversionAODBGHandler.h"
#include "AliConversionMesonCuts.h"
#include "TH3.h"


class AliAnalysisTaskGammaConvV1 : public AliAnalysisTaskSE {
 public:

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

   virtual void   UserCreateOutputObjects();
   virtual void   UserExec(Option_t *);
   virtual void   Terminate(const Option_t*);
   void InitBack();

   void SetIsHeavyIon(Bool_t flag){fIsHeavyIon = flag;}
   void SetDoMesonAnalysis(Bool_t flag){fDoMesonAnalysis = flag;}
   void ProcessPhotonCandidates();
   void CalculatePi0Candidates();
   void CalculateBackground();
   void ProcessMCParticles();
   void ProcessTruePhotonCandidates( AliAODConversionPhoton* TruePhotonCandidate);
   void ProcessTrueMesonCandidates( AliAODConversionMother *Pi0Candidate, AliAODConversionPhoton *TrueGammaCandidate0, AliAODConversionPhoton *TrueGammaCandidate1);
   void RotateParticle(AliAODConversionPhoton *gamma);
   void SetConversionCutList(Int_t nCuts, TList *CutArray){
      fnCuts = nCuts;
      fCutArray = CutArray;
   }
   void SetMesonCutList(Int_t nCuts, TList *CutArray){
      fnCuts = nCuts;
      fMesonCutArray = CutArray;
   }

   // BG HandlerSettings
   void SetMoveParticleAccordingToVertex(Bool_t flag){fMoveParticleAccordingToVertex = flag;}
   void CountESDTracks();
   void MoveParticleAccordingToVertex(AliAODConversionPhoton* particle,const AliGammaConversionAODBGHandler::GammaConversionVertex *vertex);
   void UpdateEventByEventData();

 protected:
   AliV0ReaderV1 *fV0Reader;
   AliGammaConversionAODBGHandler **fBGHandler;
   AliESDEvent *fESDEvent;
   AliMCEvent *fMCEvent;
   AliStack *fMCStack;
   TList **fCutFolder;
   TList **fESDList;
   TList **fBackList;
   TList **fMotherList;
   TList **fTrueList;
   TList **fMCList;
   TList **fHeaderNameList;
   TList *fOutputContainer;
   TClonesArray *fReaderGammas;
   TList *fGoodGammas;
   TList *fCutArray;
   AliConversionCuts *fConversionCuts;
	TList *fMesonCutArray;
	AliConversionMesonCuts *fMesonCuts;
   TH1F **hESDConvGammaPt;
   TH2F **hESDMotherInvMassPt;
   THnSparseF **sESDMotherInvMassPtZM;
   TH2F **hESDMotherBackInvMassPt;
   THnSparseF **sESDMotherBackInvMassPtZM;
   TH2F **hESDMotherInvMassEalpha;
   TH1F **hMCAllGammaPt;
   TH1F **hMCDecayGammaPi0Pt;
   TH1F **hMCDecayGammaRhoPt;
   TH1F **hMCDecayGammaEtaPt;
   TH1F **hMCDecayGammaOmegaPt;
   TH1F **hMCDecayGammaEtapPt;
   TH1F **hMCDecayGammaPhiPt;
   TH1F **hMCConvGammaPt;
   TH1F **hMCConvGammaR;
   TH1F **hMCConvGammaEta;
   TH1F **hMCPi0Pt;
   TH1F **hMCEtaPt;
   TH1F **hMCPi0InAccPt;
   TH1F **hMCEtaInAccPt;
   TH2F **hESDTrueMotherInvMassPt;
   TH2F **hESDTruePrimaryMotherInvMassMCPt;
   TH2F **hESDTruePrimaryPi0ESDPtMCPt;
   TH2F **hESDTrueSecondaryMotherInvMassPt;
   TH2F **hESDTrueSecondaryMotherFromK0sInvMassPt;
   TH2F **hESDTrueBckGGInvMassPt;
   TH2F **hESDTrueBckContInvMassPt;
   TH2F **hESDTrueMotherDalitzInvMassPt;
   TH1F **hESDTrueConvGammaPt;
   TH1F **hESDTrueTwoElecCombPt;
   TH1F **hESDTrueTwoPionCombPt;
   TH1F **hESDTrueElecPionCombPt;
   TH1F **hESDTrueCombPt;
   TH1F **hESDTruePrimaryConvGammaPt;
   TH1F **hESDTruePrimaryConvGammaR;
   TH1F **hESDTruePrimaryConvGammaEta;
   TH2F **hESDTruePrimaryConvGammaESDPtMCPt;
   TH1F **hESDTrueSecondaryConvGammaPt;
   TH1F **hESDTrueSecondaryConvGammaFromK0sPt;
   TH1F **hESDTrueSecondaryConvGammaFromXFromK0sPt;
   TH1I **hNEvents;
   TH1I **hNGoodESDTracks;
   TH1I **hNV0Tracks;

   TRandom3 fRandom;
   Double_t *fUnsmearedPx;
   Double_t *fUnsmearedPy;
   Double_t *fUnsmearedPz;
   Double_t *fUnsmearedE;
   Int_t fnCuts;
   Int_t fiCut;
   Int_t fNumberOfESDTracks;
   Int_t fNumberOfHijingPythiaParticles;
   Bool_t fMoveParticleAccordingToVertex;
   Bool_t fIsHeavyIon;
   Bool_t fDoMesonAnalysis;

   ClassDef(AliAnalysisTaskGammaConvV1, 2);
};

#endif
Commit	Line	Data
2bb2434e	1	#ifndef ALIANLYSISTASKGAMMACONVV1_cxx
	2	#define ALIANLYSISTASKGAMMACONVV1_cxx
	3
	4	#include "AliAnalysisTaskSE.h"
	5	#include "AliESDtrack.h"
	6	#include "AliV0ReaderV1.h"
	7	#include "AliKFConversionPhoton.h"
	8	#include "AliGammaConversionAODBGHandler.h"
ca91a3e1	9	#include "AliConversionMesonCuts.h"
2bb2434e	10	#include "TH3.h"
	11
	12
	13	class AliAnalysisTaskGammaConvV1 : public AliAnalysisTaskSE {
ca91a3e1	14	public:
2bb2434e	15
ca91a3e1	16	AliAnalysisTaskGammaConvV1();
	17	AliAnalysisTaskGammaConvV1(const char *name);
	18	virtual ~AliAnalysisTaskGammaConvV1();
2bb2434e	19
ca91a3e1	20	virtual void UserCreateOutputObjects();
	21	virtual void UserExec(Option_t *);
	22	virtual void Terminate(const Option_t*);
	23	void InitBack();
2bb2434e	24
ca91a3e1	25	void SetIsHeavyIon(Bool_t flag){fIsHeavyIon = flag;}
	26	void SetDoMesonAnalysis(Bool_t flag){fDoMesonAnalysis = flag;}
	27	void ProcessPhotonCandidates();
	28	void CalculatePi0Candidates();
	29	void CalculateBackground();
	30	void ProcessMCParticles();
	31	void ProcessTruePhotonCandidates( AliAODConversionPhoton* TruePhotonCandidate);
	32	void ProcessTrueMesonCandidates( AliAODConversionMother Pi0Candidate, AliAODConversionPhoton TrueGammaCandidate0, AliAODConversionPhoton *TrueGammaCandidate1);
	33	void RotateParticle(AliAODConversionPhoton *gamma);
	34	void SetConversionCutList(Int_t nCuts, TList *CutArray){
	35	fnCuts = nCuts;
	36	fCutArray = CutArray;
	37	}
	38	void SetMesonCutList(Int_t nCuts, TList *CutArray){
	39	fnCuts = nCuts;
	40	fMesonCutArray = CutArray;
	41	}
2bb2434e	42
ca91a3e1	43	// BG HandlerSettings
	44	void SetMoveParticleAccordingToVertex(Bool_t flag){fMoveParticleAccordingToVertex = flag;}
	45	void CountESDTracks();
	46	void MoveParticleAccordingToVertex(AliAODConversionPhoton* particle,const AliGammaConversionAODBGHandler::GammaConversionVertex *vertex);
	47	void UpdateEventByEventData();
	48
	49	protected:
	50	AliV0ReaderV1 *fV0Reader;
	51	AliGammaConversionAODBGHandler **fBGHandler;
	52	AliESDEvent *fESDEvent;
	53	AliMCEvent *fMCEvent;
	54	AliStack *fMCStack;
	55	TList **fCutFolder;
	56	TList **fESDList;
	57	TList **fBackList;
	58	TList **fMotherList;
	59	TList **fTrueList;
	60	TList **fMCList;
	61	TList **fHeaderNameList;
	62	TList *fOutputContainer;
	63	TClonesArray *fReaderGammas;
	64	TList *fGoodGammas;
	65	TList *fCutArray;
	66	AliConversionCuts *fConversionCuts;
	67	TList *fMesonCutArray;
	68	AliConversionMesonCuts *fMesonCuts;
	69	TH1F **hESDConvGammaPt;
	70	TH2F **hESDMotherInvMassPt;
	71	THnSparseF **sESDMotherInvMassPtZM;
	72	TH2F **hESDMotherBackInvMassPt;
	73	THnSparseF **sESDMotherBackInvMassPtZM;
	74	TH2F **hESDMotherInvMassEalpha;
	75	TH1F **hMCAllGammaPt;
	76	TH1F **hMCDecayGammaPi0Pt;
	77	TH1F **hMCDecayGammaRhoPt;
	78	TH1F **hMCDecayGammaEtaPt;
	79	TH1F **hMCDecayGammaOmegaPt;
	80	TH1F **hMCDecayGammaEtapPt;
	81	TH1F **hMCDecayGammaPhiPt;
	82	TH1F **hMCConvGammaPt;
	83	TH1F **hMCConvGammaR;
	84	TH1F **hMCConvGammaEta;
	85	TH1F **hMCPi0Pt;
	86	TH1F **hMCEtaPt;
	87	TH1F **hMCPi0InAccPt;
	88	TH1F **hMCEtaInAccPt;
	89	TH2F **hESDTrueMotherInvMassPt;
	90	TH2F **hESDTruePrimaryMotherInvMassMCPt;
	91	TH2F **hESDTruePrimaryPi0ESDPtMCPt;
	92	TH2F **hESDTrueSecondaryMotherInvMassPt;
	93	TH2F **hESDTrueSecondaryMotherFromK0sInvMassPt;
	94	TH2F **hESDTrueBckGGInvMassPt;
	95	TH2F **hESDTrueBckContInvMassPt;
	96	TH2F **hESDTrueMotherDalitzInvMassPt;
	97	TH1F **hESDTrueConvGammaPt;
	98	TH1F **hESDTrueTwoElecCombPt;
	99	TH1F **hESDTrueTwoPionCombPt;
	100	TH1F **hESDTrueElecPionCombPt;
	101	TH1F **hESDTrueCombPt;
	102	TH1F **hESDTruePrimaryConvGammaPt;
	103	TH1F **hESDTruePrimaryConvGammaR;
	104	TH1F **hESDTruePrimaryConvGammaEta;
	105	TH2F **hESDTruePrimaryConvGammaESDPtMCPt;
	106	TH1F **hESDTrueSecondaryConvGammaPt;
107	TH1F **hESDTrueSecondaryConvGammaFromK0sPt;
108	TH1F **hESDTrueSecondaryConvGammaFromXFromK0sPt;
109	TH1I **hNEvents;
110	TH1I **hNGoodESDTracks;
111	TH1I **hNV0Tracks;
112
113	TRandom3 fRandom;
114	Double_t *fUnsmearedPx;
115	Double_t *fUnsmearedPy;
116	Double_t *fUnsmearedPz;
117	Double_t *fUnsmearedE;
118	Int_t fnCuts;
119	Int_t fiCut;
120	Int_t fNumberOfESDTracks;
121	Int_t fNumberOfHijingPythiaParticles;
122	Bool_t fMoveParticleAccordingToVertex;
123	Bool_t fIsHeavyIon;
124	Bool_t fDoMesonAnalysis;
125
126	ClassDef(AliAnalysisTaskGammaConvV1, 2);
2bb2434e	127	};
	128
	129	#endif