[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 "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;
    }
    
    // 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 **fTrueList;
    TList **fMCList;
    TList *fOutputContainer;
    TClonesArray *fReaderGammas;
    TList *fGoodGammas;
    TList *fCutArray;
    AliConversionCuts *fConversionCuts;
    TH1F **hESDConvGammaPt;
    TH2F **hESDMotherInvMassPt;
    THnSparseF **sESDMotherInvMassPtZM;
    TH2F **hESDMotherBackInvMassPt;
    THnSparseF **sESDMotherBackInvMassPtZM;
    TH2F **hESDMotherInvMassEalpha;
    TH1F **hMCAllGammaPt;
	 TH1F **hMCAllGammaPtAddedSig;
    TH1F **hMCDecayGammaPi0Pt;
	 TH1F **hMCDecayGammaPi0PtAddedSig;
    TH1F **hMCDecayGammaRhoPt;
	 TH1F **hMCDecayGammaRhoPtAddedSig;
    TH1F **hMCDecayGammaEtaPt;
	 TH1F **hMCDecayGammaEtaPtAddedSig;
    TH1F **hMCDecayGammaOmegaPt;
	 TH1F **hMCDecayGammaOmegaPtAddedSig;
    TH1F **hMCDecayGammaEtapPt;
	 TH1F **hMCDecayGammaEtapPtAddedSig;
    TH1F **hMCDecayGammaPhiPt;
	 TH1F **hMCDecayGammaPhiPtAddedSig;
    TH1F **hMCConvGammaPt;
	 TH1F **hMCConvGammaPtAddedSig;
    TH1F **hMCPi0Pt;
    TH1F **hMCEtaPt;
    TH1F **hMCPi0InAccPt;
    TH1F **hMCEtaInAccPt;
	 TH1F **hMCPi0PtAddedSig;
    TH1F **hMCEtaPtAddedSig;
    TH1F **hMCPi0InAccPtAddedSig;
    TH1F **hMCEtaInAccPtAddedSig;
    TH2F **hESDTrueMotherInvMassPt;
	 TH2F **hESDTrueMotherInvMassPtAddedSig;
    TH2F **hESDTruePrimaryMotherInvMassMCPt;
	 TH2F **hESDTruePrimaryMotherInvMassMCPtAddedSig;
    TH2F **hESDTruePrimaryPi0ESDPtMCPt;
	 TH2F **hESDTruePrimaryPi0ESDPtMCPtAddedSig;
    TH2F **hESDTrueSecondaryMotherInvMassPt;
	 TH2F **hESDTrueSecondaryMotherInvMassPtAddedSig;
    TH2F **hESDTrueSecondaryMotherFromK0sInvMassPt;
	 TH2F **hESDTrueSecondaryMotherFromK0sInvMassPtAddedSig;
    TH2F **hESDTrueBckGGInvMassPt;
    TH2F **hESDTrueBckContInvMassPt;
    TH2F **hESDTrueMotherDalitzInvMassPt;
	 TH2F **hESDTrueMotherDalitzInvMassPtAddedSig;
    TH1F **hESDTrueConvGammaPt;
	 TH1F **hESDTrueConvGammaPtAddedSig;
    TH1F **hESDTrueElecCombPt;
    TH1F **hESDTruePionCombPt;
    TH1F **hESDTruePrimaryConvGammaPt;
	 TH1F **hESDTruePrimaryConvGammaPtAddedSig;
    TH2F **hESDTruePrimaryConvGammaESDPtMCPt;
	 TH2F **hESDTruePrimaryConvGammaESDPtMCPtAddedSig;
    TH1F **hESDTrueSecondaryConvGammaPt;
	 TH1F **hESDTrueSecondaryConvGammaPtAddedSig;
    TH1F **hESDTrueSecondaryConvGammaFromK0sPt;
	 TH1F **hESDTrueSecondaryConvGammaFromK0sPtAddedSig;
    TH1F **hESDTrueSecondaryConvGammaFromXFromK0sPt;
	 TH1F **hESDTrueSecondaryConvGammaFromXFromK0sPtAddedSig;
    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;
    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"
	9	#include "TH3.h"
	10
	11
	12	class AliAnalysisTaskGammaConvV1 : public AliAnalysisTaskSE {
	13	public:
	14
	15	AliAnalysisTaskGammaConvV1();
	16	AliAnalysisTaskGammaConvV1(const char *name);
	17	virtual ~AliAnalysisTaskGammaConvV1();
	18
	19	virtual void UserCreateOutputObjects();
	20	virtual void UserExec(Option_t *);
	21	virtual void Terminate(const Option_t*);
	22	void InitBack();
	23
	24	void SetIsHeavyIon(Bool_t flag){fIsHeavyIon = flag;}
	25	void SetDoMesonAnalysis(Bool_t flag){fDoMesonAnalysis = flag;}
	26	void ProcessPhotonCandidates();
	27	void CalculatePi0Candidates();
	28	void CalculateBackground();
	29	void ProcessMCParticles();
	30	void ProcessTruePhotonCandidates( AliAODConversionPhoton* TruePhotonCandidate);
	31	void ProcessTrueMesonCandidates( AliAODConversionMother Pi0Candidate, AliAODConversionPhoton TrueGammaCandidate0, AliAODConversionPhoton *TrueGammaCandidate1);
	32	void RotateParticle(AliAODConversionPhoton *gamma);
	33	void SetConversionCutList(Int_t nCuts, TList *CutArray){
	34	fnCuts = nCuts;
	35	fCutArray = CutArray;
	36	}
	37
	38	// BG HandlerSettings
	39	void SetMoveParticleAccordingToVertex(Bool_t flag){fMoveParticleAccordingToVertex = flag;}
	40	void CountESDTracks();
	41	void MoveParticleAccordingToVertex(AliAODConversionPhoton* particle,const AliGammaConversionAODBGHandler::GammaConversionVertex *vertex);
	42	void UpdateEventByEventData();
	43
	44	protected:
	45	AliV0ReaderV1 *fV0Reader;
	46	AliGammaConversionAODBGHandler **fBGHandler;
	47	AliESDEvent *fESDEvent;
	48	AliMCEvent *fMCEvent;
	49	AliStack *fMCStack;
	50	TList **fCutFolder;
	51	TList **fESDList;
	52	TList **fBackList;
	53	TList **fTrueList;
	54	TList **fMCList;
	55	TList *fOutputContainer;
	56	TClonesArray *fReaderGammas;
	57	TList *fGoodGammas;
	58	TList *fCutArray;
	59	AliConversionCuts *fConversionCuts;
	60	TH1F **hESDConvGammaPt;
	61	TH2F **hESDMotherInvMassPt;
	62	THnSparseF **sESDMotherInvMassPtZM;
	63	TH2F **hESDMotherBackInvMassPt;
	64	THnSparseF **sESDMotherBackInvMassPtZM;
65	TH2F **hESDMotherInvMassEalpha;
66	TH1F **hMCAllGammaPt;
67	TH1F **hMCAllGammaPtAddedSig;
68	TH1F **hMCDecayGammaPi0Pt;
69	TH1F **hMCDecayGammaPi0PtAddedSig;
70	TH1F **hMCDecayGammaRhoPt;
71	TH1F **hMCDecayGammaRhoPtAddedSig;
72	TH1F **hMCDecayGammaEtaPt;
73	TH1F **hMCDecayGammaEtaPtAddedSig;
74	TH1F **hMCDecayGammaOmegaPt;
75	TH1F **hMCDecayGammaOmegaPtAddedSig;
76	TH1F **hMCDecayGammaEtapPt;
77	TH1F **hMCDecayGammaEtapPtAddedSig;
78	TH1F **hMCDecayGammaPhiPt;
79	TH1F **hMCDecayGammaPhiPtAddedSig;
80	TH1F **hMCConvGammaPt;
81	TH1F **hMCConvGammaPtAddedSig;
82	TH1F **hMCPi0Pt;
83	TH1F **hMCEtaPt;
84	TH1F **hMCPi0InAccPt;
85	TH1F **hMCEtaInAccPt;
86	TH1F **hMCPi0PtAddedSig;
87	TH1F **hMCEtaPtAddedSig;
88	TH1F **hMCPi0InAccPtAddedSig;
89	TH1F **hMCEtaInAccPtAddedSig;
90	TH2F **hESDTrueMotherInvMassPt;
91	TH2F **hESDTrueMotherInvMassPtAddedSig;
92	TH2F **hESDTruePrimaryMotherInvMassMCPt;
93	TH2F **hESDTruePrimaryMotherInvMassMCPtAddedSig;
94	TH2F **hESDTruePrimaryPi0ESDPtMCPt;
95	TH2F **hESDTruePrimaryPi0ESDPtMCPtAddedSig;
96	TH2F **hESDTrueSecondaryMotherInvMassPt;
97	TH2F **hESDTrueSecondaryMotherInvMassPtAddedSig;
98	TH2F **hESDTrueSecondaryMotherFromK0sInvMassPt;
99	TH2F **hESDTrueSecondaryMotherFromK0sInvMassPtAddedSig;
100	TH2F **hESDTrueBckGGInvMassPt;
101	TH2F **hESDTrueBckContInvMassPt;
102	TH2F **hESDTrueMotherDalitzInvMassPt;
103	TH2F **hESDTrueMotherDalitzInvMassPtAddedSig;
104	TH1F **hESDTrueConvGammaPt;
105	TH1F **hESDTrueConvGammaPtAddedSig;
106	TH1F **hESDTrueElecCombPt;
107	TH1F **hESDTruePionCombPt;
108	TH1F **hESDTruePrimaryConvGammaPt;
109	TH1F **hESDTruePrimaryConvGammaPtAddedSig;
110	TH2F **hESDTruePrimaryConvGammaESDPtMCPt;
111	TH2F **hESDTruePrimaryConvGammaESDPtMCPtAddedSig;
112	TH1F **hESDTrueSecondaryConvGammaPt;
113	TH1F **hESDTrueSecondaryConvGammaPtAddedSig;
114	TH1F **hESDTrueSecondaryConvGammaFromK0sPt;
115	TH1F **hESDTrueSecondaryConvGammaFromK0sPtAddedSig;
116	TH1F **hESDTrueSecondaryConvGammaFromXFromK0sPt;
117	TH1F **hESDTrueSecondaryConvGammaFromXFromK0sPtAddedSig;
118	TH1I **hNEvents;
119	TH1I **hNGoodESDTracks;
120	TH1I **hNV0Tracks;
121
122	TRandom3 fRandom;
123	Double_t *fUnsmearedPx;
124	Double_t *fUnsmearedPy;
125	Double_t *fUnsmearedPz;
126	Double_t *fUnsmearedE;
127	Int_t fnCuts;
128	Int_t fiCut;
129	Int_t fNumberOfESDTracks;
130	Bool_t fMoveParticleAccordingToVertex;
131	Bool_t fIsHeavyIon;
132	Bool_t fDoMesonAnalysis;
133
134	ClassDef(AliAnalysisTaskGammaConvV1, 2);
135	};
136
137	#endif