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

#ifndef ALIANALYSISTASKGAMMACONVDALITZV1_H
#define ALIANALYSISTASKGAMMACONVDALITZV1_H

/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */

// Analysis task for pi0->e+e-gamma (Dalitz decay)

#include "AliAnalysisTaskSE.h"
#include "AliV0ReaderV1.h"
#include "AliKFConversionPhoton.h"
#include "AliDalitzElectronSelector.h"
#include "AliConversionMesonCuts.h"
#include "AliGammaConversionAODBGHandler.h"
#include "TProfile2D.h"

class AliESDInputHandler;
class AliMCEventHandler;
class AliESDEvent;
class AliESDtrack;
class AliESDtrackCuts;
class AliESDpidCuts;
class AliV0Reader;
class AliGammaConversionHistograms;
class AliTriggerAnalysis;

class AliAnalysisTaskGammaConvDalitzV1: public AliAnalysisTaskSE
{
  public:

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

	virtual void UserExec(Option_t *);
	virtual void UserCreateOutputObjects();
	virtual Bool_t Notify();
	virtual void Terminate(const Option_t *);


	void SetMoveParticleAccordingToVertex(Bool_t flag){fMoveParticleAccordingToVertex = flag;}
         
	void SetIsHeavyIon(Int_t flag){
		if (flag == 1 || flag ==2 ){
			fIsHeavyIon = 1;    
		} else {
			fIsHeavyIon = 0;    
		}
	}
	
	void SetIsMC(Bool_t isMC){fIsMC=isMC;}
	void SetConversionCutList(Int_t nCuts, TList *CutArray){
		fnCuts= nCuts;
		fCutGammaArray = CutArray;
	}
	void SetElectronCutList(TList *CutArray){
		fCutElectronArray = CutArray;
	}
	void SetMesonCutList(TList *CutArray){
		fCutMesonArray = CutArray;
	}
	void SetDoChicAnalysis(Bool_t flag){ fDoChicAnalysis = flag; }
	void SetDoMesonQA(Bool_t flag){ fDoMesonQA = flag; }
  

	private:

		void InitBack();
		void ProcessPhotonCandidates();
		void ProcessTruePhotonCandidates(AliAODConversionPhoton*);
                void ProcessTrueMesonCandidates(AliAODConversionMother *Pi0Candidate, AliAODConversionPhoton *TrueGammaCandidate, AliAODConversionPhoton *TrueVirtualGammaCandidate);
                void MoveParticleAccordingToVertex(AliAODConversionPhoton* particle,const AliGammaConversionAODBGHandler::GammaConversionVertex *vertex);
    		void ProcessElectronCandidates();
		void ProcessMCParticles();
		void CountESDTracks();
                void CalculatePi0DalitzCandidates();
                void CalculateBackground();
                void UpdateEventByEventData();
                Double_t GetPsiPair( const AliESDtrack *trackPos, const AliESDtrack *trackNeg ) const;
		Bool_t IsDalitz(TParticle *fMCMother) const;
                Bool_t IsPi0DalitzDaughter( Int_t label ) const;
		
		
    AliV0ReaderV1 *fV0Reader;
    AliDalitzElectronSelector* fElecSelector;
    AliGammaConversionAODBGHandler **fBGHandler;
    AliESDEvent *fESDEvent;
    AliMCEvent *fMCEvent;
    AliStack *fMCStack;
    TList **fCutFolder;
    TList **fESDList;
    TList **fBackList;
    TList **fMotherList;
    TList **fTrueList;
    TList **fMCList;
    TList *fOutputContainer;
    TClonesArray *fReaderGammas;
    vector<Int_t> fSelectorElectronIndex;
    vector<Int_t> fSelectorPositronIndex;
    TList *fGoodGammas;
    TList *fGoodVirtualGammas;
    TList *fGoodElectrons;
    TList *fGoodPositrons;
    TList *fCutGammaArray;
    TList *fCutElectronArray;
    TList *fCutMesonArray;
    TList **fGammasPool;
    AliConversionCuts *fConversionCuts;
    TH1F **hESDConvGammaPt;
    TH1F **hESDDalitzElectronPt;
    TH1F **hESDDalitzPositronPt;
    TH1F **hESDDalitzElectronPhi;
    TH1F **hESDDalitzPositronPhi;
    TH1F **hESDDalitzElectronAfterPt;
    TH1F **hESDDalitzPositronAfterPt;
    TH1F **hESDDalitzElectronAfterPhi;
    TH1F **hESDDalitzPositronAfterPhi;
    TH2F **hESDDalitzElectronAfterNFindClsTPC;
    TH2F **hESDDalitzPositronAfterNFindClsTPC;
    TH2F **hESDDalitzPosEleAfterDCAxy;
    TH2F **hESDDalitzPosEleAfterDCAz;
    TH2F **hESDDalitzPosEleAfterTPCdEdx;
    TH2F **hESDDalitzPosEleAfterTPCdEdxSignal;
    TH1F **hESDMotherPhi;
    TH2F **hESDEposEnegPsiPairDPhi;
    TH2F **hESDEposEnegInvMassPt;
    TH2F **hESDEposEnegLikeSignBackInvMassPt;
    TH2F **hESDMotherInvMassPt;
    TH2F **hESDPi0MotherInvMassPt;
    TH2F **hESDPi0MotherDiffInvMassPt;
    TH2F **hESDPi0MotherDiffLimInvMassPt;
    THnSparseF **sESDMotherInvMassPtZM;
    TH2F **hESDMotherBackInvMassPt;
    THnSparseF **sESDMotherBackInvMassPtZM;
    TH1F **hMCAllGammaPt;
    TH1F **hMCConvGammaPt;
    TH1F **hMCConvGammaRSPt;
    TH1F **hMCAllPositronsPt;
    TH1F **hMCAllElectronsPt;
    TH1F **hMCPi0DalitzGammaPt;
    TH1F **hMCPi0DalitzElectronPt;
    TH1F **hMCPi0DalitzPositronPt;
    TH1F **hMCPi0Pt;
    TH1F **hMCPi0GGPt;
    TH1F **hMCEtaPt;
    TH1F **hMCEtaGGPt;
    TH1F **hMCPi0InAccPt;
    TH1F **hMCEtaInAccPt;
    TH1F **hMCChiCPt;
    TH1F **hMCChiCInAccPt;
    TH2F **hESDEposEnegTruePi0DalitzInvMassPt;
    TH2F **hESDEposEnegTrueEtaDalitzInvMassPt;
    TH2F **hESDEposEnegTruePhotonInvMassPt;
    TH2F **hESDEposEnegTrueJPsiInvMassPt;
    TH2F **hESDTrueMotherChiCInvMassPt;
    TH2F **hESDTrueMotherChiCDiffInvMassPt;
    TH2F **hESDTrueMotherInvMassPt;
    TH2F **hESDTrueMotherDalitzInvMassPt;
    TH2F **hESDTrueMotherPi0GGInvMassPt;
    TH2F **hESDTruePrimaryMotherInvMassMCPt;
    TH2F **hESDTruePrimaryPi0DalitzESDPtMCPt;
    TH2F **hESDTrueSecondaryMotherInvMassPt;
    TH2F **hESDTrueSecondaryMotherFromK0sInvMassPt;
    TH2F **hESDTrueBckGGInvMassPt;
    TH2F **hESDTrueBckContInvMassPt;
    TH2F **hESDTrueMotherGGInvMassPt;
    TH1F **hESDTrueConvGammaPt;
    TH1F **hESDTruePositronPt;
    TH1F **hESDTrueElectronPt;
    TH1F **hESDTrueSecConvGammaPt;
    TH1F **hESDTrueSecPositronPt;
    TH1F **hESDTrueSecElectronPt;
    TH1F **hESDTruePi0DalitzConvGammaPt;
    TH1F **hESDTruePi0DalitzPositronPt;
    TH1F **hESDTruePi0DalitzElectronPt;
    TH1F **hESDTruePi0DalitzSecConvGammaPt;
    TH1F **hESDTruePi0DalitzSecPositronPt;
    TH1F **hESDTruePi0DalitzSecElectronPt;
      //if(fDoMesonAnalysis){


    TH1I **hNEvents;
    TH1I **hNGoodESDTracks;
    TProfile **hEtaShift;
        
    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;
    Bool_t fDoChicAnalysis;
    Bool_t fDoMesonQA;
    Bool_t fIsFromMBHeader;
		Bool_t fIsMC;

	private:
		AliAnalysisTaskGammaConvDalitzV1( const AliAnalysisTaskGammaConvDalitzV1& ); // Not implemented
		AliAnalysisTaskGammaConvDalitzV1& operator=( const AliAnalysisTaskGammaConvDalitzV1& ); // Not implemented

		ClassDef( AliAnalysisTaskGammaConvDalitzV1, 3 );
};

#endif // ALIANALYSISTASKGAMMACONVDALITZV1_H
Commit	Line	Data
2bb2434e	1	#ifndef ALIANALYSISTASKGAMMACONVDALITZV1_H
	2	#define ALIANALYSISTASKGAMMACONVDALITZV1_H
	3
	4	/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	5	* See cxx source for full Copyright notice */
	6
	7	// Analysis task for pi0->e+e-gamma (Dalitz decay)
	8
	9	#include "AliAnalysisTaskSE.h"
	10	#include "AliV0ReaderV1.h"
	11	#include "AliKFConversionPhoton.h"
	12	#include "AliDalitzElectronSelector.h"
	13	#include "AliConversionMesonCuts.h"
	14	#include "AliGammaConversionAODBGHandler.h"
8a52eed5	15	#include "TProfile2D.h"
2bb2434e	16
	17	class AliESDInputHandler;
	18	class AliMCEventHandler;
	19	class AliESDEvent;
	20	class AliESDtrack;
	21	class AliESDtrackCuts;
	22	class AliESDpidCuts;
	23	class AliV0Reader;
	24	class AliGammaConversionHistograms;
	25	class AliTriggerAnalysis;
	26
	27	class AliAnalysisTaskGammaConvDalitzV1: public AliAnalysisTaskSE
	28	{
	29	public:
	30
	31	AliAnalysisTaskGammaConvDalitzV1();
	32	AliAnalysisTaskGammaConvDalitzV1( const char* name );
	33	virtual ~AliAnalysisTaskGammaConvDalitzV1();
	34
	35	virtual void UserExec(Option_t *);
	36	virtual void UserCreateOutputObjects();
ae947965	37	virtual Bool_t Notify();
2bb2434e	38	virtual void Terminate(const Option_t *);
	39
	40
	41
	42
ae947965	43	void SetMoveParticleAccordingToVertex(Bool_t flag){fMoveParticleAccordingToVertex = flag;}
	44
	45	void SetIsHeavyIon(Int_t flag){
	46	if (flag == 1 \|\| flag ==2 ){
	47	fIsHeavyIon = 1;
	48	} else {
	49	fIsHeavyIon = 0;
	50	}
	51	}
	52
	53	void SetIsMC(Bool_t isMC){fIsMC=isMC;}
	54	void SetConversionCutList(Int_t nCuts, TList *CutArray){
	55	fnCuts= nCuts;
	56	fCutGammaArray = CutArray;
	57	}
	58	void SetElectronCutList(TList *CutArray){
	59	fCutElectronArray = CutArray;
	60	}
	61	void SetMesonCutList(TList *CutArray){
	62	fCutMesonArray = CutArray;
	63	}
fcc79cf5	64	void SetDoChicAnalysis(Bool_t flag){ fDoChicAnalysis = flag; }
fcc79cf5	65	void SetDoMesonQA(Bool_t flag){ fDoMesonQA = flag; }
ae947965	66
2bb2434e	67
	68	private:
	69
2bb2434e	70	void InitBack();
	71	void ProcessPhotonCandidates();
	72	void ProcessTruePhotonCandidates(AliAODConversionPhoton*);
	73	void ProcessTrueMesonCandidates(AliAODConversionMother Pi0Candidate, AliAODConversionPhoton TrueGammaCandidate, AliAODConversionPhoton *TrueVirtualGammaCandidate);
	74	void MoveParticleAccordingToVertex(AliAODConversionPhoton* particle,const AliGammaConversionAODBGHandler::GammaConversionVertex *vertex);
	75	void ProcessElectronCandidates();
	76	void ProcessMCParticles();
	77	void CountESDTracks();
	78	void CalculatePi0DalitzCandidates();
	79	void CalculateBackground();
	80	void UpdateEventByEventData();
a280ac15	81	Double_t GetPsiPair( const AliESDtrack trackPos, const AliESDtrack trackNeg ) const;
ae947965	82	Bool_t IsDalitz(TParticle *fMCMother) const;
ccfa8c0d	83	Bool_t IsPi0DalitzDaughter( Int_t label ) const;
2bb2434e	84
	85
	86
	87	AliV0ReaderV1 *fV0Reader;
	88	AliDalitzElectronSelector* fElecSelector;
	89	AliGammaConversionAODBGHandler **fBGHandler;
	90	AliESDEvent *fESDEvent;
	91	AliMCEvent *fMCEvent;
	92	AliStack *fMCStack;
	93	TList **fCutFolder;
	94	TList **fESDList;
	95	TList **fBackList;
fcc79cf5	96	TList **fMotherList;
2bb2434e	97	TList **fTrueList;
	98	TList **fMCList;
	99	TList *fOutputContainer;
	100	TClonesArray *fReaderGammas;
	101	vector<Int_t> fSelectorElectronIndex;
	102	vector<Int_t> fSelectorPositronIndex;
	103	TList *fGoodGammas;
	104	TList *fGoodVirtualGammas;
	105	TList *fGoodElectrons;
	106	TList *fGoodPositrons;
	107	TList *fCutGammaArray;
	108	TList *fCutElectronArray;
	109	TList *fCutMesonArray;
	110	TList **fGammasPool;
	111	AliConversionCuts *fConversionCuts;
	112	TH1F **hESDConvGammaPt;
	113	TH1F **hESDDalitzElectronPt;
	114	TH1F **hESDDalitzPositronPt;
4803eb1f	115	TH1F **hESDDalitzElectronPhi;
	116	TH1F **hESDDalitzPositronPhi;
	117	TH1F **hESDDalitzElectronAfterPt;
	118	TH1F **hESDDalitzPositronAfterPt;
	119	TH1F **hESDDalitzElectronAfterPhi;
	120	TH1F **hESDDalitzPositronAfterPhi;
	121	TH2F **hESDDalitzElectronAfterNFindClsTPC;
	122	TH2F **hESDDalitzPositronAfterNFindClsTPC;
	123	TH2F **hESDDalitzPosEleAfterDCAxy;
	124	TH2F **hESDDalitzPosEleAfterDCAz;
	125	TH2F **hESDDalitzPosEleAfterTPCdEdx;
	126	TH2F **hESDDalitzPosEleAfterTPCdEdxSignal;
	127	TH1F **hESDMotherPhi;
2bb2434e	128	TH2F **hESDEposEnegPsiPairDPhi;
a280ac15	129	TH2F **hESDEposEnegInvMassPt;
0a2b2b4b	130	TH2F **hESDEposEnegLikeSignBackInvMassPt;
2bb2434e	131	TH2F **hESDMotherInvMassPt;
a280ac15	132	TH2F **hESDPi0MotherInvMassPt;
a280ac15	133	TH2F **hESDPi0MotherDiffInvMassPt;
ae947965	134	TH2F **hESDPi0MotherDiffLimInvMassPt;
2bb2434e	135	THnSparseF **sESDMotherInvMassPtZM;
	136	TH2F **hESDMotherBackInvMassPt;
	137	THnSparseF **sESDMotherBackInvMassPtZM;
ccfa8c0d	138	TH1F **hMCAllGammaPt;
ae947965	139	TH1F **hMCConvGammaPt;
ae947965	140	TH1F **hMCConvGammaRSPt;
ccfa8c0d	141	TH1F **hMCAllPositronsPt;
	142	TH1F **hMCAllElectronsPt;
	143	TH1F **hMCPi0DalitzGammaPt;
	144	TH1F **hMCPi0DalitzElectronPt;
	145	TH1F **hMCPi0DalitzPositronPt;
2bb2434e	146	TH1F **hMCPi0Pt;
a280ac15	147	TH1F **hMCPi0GGPt;
2bb2434e	148	TH1F **hMCEtaPt;
a280ac15	149	TH1F **hMCEtaGGPt;
2bb2434e	150	TH1F **hMCPi0InAccPt;
2bb2434e	151	TH1F **hMCEtaInAccPt;
0f8c33c1	152	TH1F **hMCChiCPt;
0f8c33c1	153	TH1F **hMCChiCInAccPt;
ccfa8c0d	154	TH2F **hESDEposEnegTruePi0DalitzInvMassPt;
	155	TH2F **hESDEposEnegTrueEtaDalitzInvMassPt;
	156	TH2F **hESDEposEnegTruePhotonInvMassPt;
0f8c33c1	157	TH2F **hESDEposEnegTrueJPsiInvMassPt;
0f8c33c1	158	TH2F **hESDTrueMotherChiCInvMassPt;
ae947965	159	TH2F **hESDTrueMotherChiCDiffInvMassPt;
2bb2434e	160	TH2F **hESDTrueMotherInvMassPt;
4803eb1f	161	TH2F **hESDTrueMotherDalitzInvMassPt;
2bb2434e	162	TH2F **hESDTrueMotherPi0GGInvMassPt;
	163	TH2F **hESDTruePrimaryMotherInvMassMCPt;
	164	TH2F **hESDTruePrimaryPi0DalitzESDPtMCPt;
	165	TH2F **hESDTrueSecondaryMotherInvMassPt;
	166	TH2F **hESDTrueSecondaryMotherFromK0sInvMassPt;
	167	TH2F **hESDTrueBckGGInvMassPt;
	168	TH2F **hESDTrueBckContInvMassPt;
	169	TH2F **hESDTrueMotherGGInvMassPt;
	170	TH1F **hESDTrueConvGammaPt;
ccfa8c0d	171	TH1F **hESDTruePositronPt;
ccfa8c0d	172	TH1F **hESDTrueElectronPt;
ae947965	173	TH1F **hESDTrueSecConvGammaPt;
	174	TH1F **hESDTrueSecPositronPt;
	175	TH1F **hESDTrueSecElectronPt;
ccfa8c0d	176	TH1F **hESDTruePi0DalitzConvGammaPt;
	177	TH1F **hESDTruePi0DalitzPositronPt;
	178	TH1F **hESDTruePi0DalitzElectronPt;
ae947965	179	TH1F **hESDTruePi0DalitzSecConvGammaPt;
	180	TH1F **hESDTruePi0DalitzSecPositronPt;
	181	TH1F **hESDTruePi0DalitzSecElectronPt;
ccfa8c0d	182	//if(fDoMesonAnalysis){
	183
	184
2bb2434e	185	TH1I **hNEvents;
2bb2434e	186	TH1I **hNGoodESDTracks;
8a52eed5	187	TProfile **hEtaShift;
2bb2434e	188
	189	TRandom3 fRandom;
	190	Double_t *fUnsmearedPx;
	191	Double_t *fUnsmearedPy;
	192	Double_t *fUnsmearedPz;
	193	Double_t *fUnsmearedE;
	194	Int_t fnCuts;
	195	Int_t fiCut;
	196	Int_t fNumberOfESDTracks;
	197	Bool_t fMoveParticleAccordingToVertex;
	198	Bool_t fIsHeavyIon;
	199	Bool_t fDoMesonAnalysis;
fcc79cf5	200	Bool_t fDoChicAnalysis;
fcc79cf5	201	Bool_t fDoMesonQA;
ae947965	202	Bool_t fIsFromMBHeader;
	203	Bool_t fIsMC;
	204
2bb2434e	205	private:
	206	AliAnalysisTaskGammaConvDalitzV1( const AliAnalysisTaskGammaConvDalitzV1& ); // Not implemented
	207	AliAnalysisTaskGammaConvDalitzV1& operator=( const AliAnalysisTaskGammaConvDalitzV1& ); // Not implemented
	208
ae947965	209	ClassDef( AliAnalysisTaskGammaConvDalitzV1, 3 );
2bb2434e	210	};
	211
	212	#endif // ALIANALYSISTASKGAMMACONVDALITZV1_H
	213