[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"

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 void Terminate(const Option_t *);


         void SetMoveParticleAccordingToVertex(Bool_t flag){fMoveParticleAccordingToVertex = flag;}
         void SetIsHeavyIon(Bool_t flag){fIsHeavyIon = flag;}
         void SetConversionCutList(Int_t nCuts, TList *CutArray){
                        fnCuts= nCuts;
                        fCutGammaArray = CutArray;
                }
         void SetElectronCutList(TList *CutArray){
                        fCutElectronArray = CutArray;
                }
         void SetMesonCutList(TList *CutArray){
                        fCutMesonArray = CutArray;
                }
        	

	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;

		
    AliV0ReaderV1 *fV0Reader;
    AliDalitzElectronSelector* fElecSelector;
    AliGammaConversionAODBGHandler **fBGHandler;
    AliESDEvent *fESDEvent;
    AliMCEvent *fMCEvent;
    AliStack *fMCStack;
    TList **fCutFolder;
    TList **fESDList;
    TList **fBackList;
    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;
    TH2F **hESDEposEnegPsiPairDPhi;
    TH2F **hESDEposEnegInvMassPt;
    TH2F **hESDEposEnegLikeSignBackInvMassPt;
    TH2F **hESDMotherInvMassPt;
    TH2F **hESDPi0MotherInvMassPt;
    TH2F **hESDPi0MotherDiffInvMassPt;
    THnSparseF **sESDMotherInvMassPtZM;
    TH2F **hESDMotherBackInvMassPt;
    THnSparseF **sESDMotherBackInvMassPtZM;
    TH1F **hMCPi0Pt;
    TH1F **hMCPi0GGPt;
    TH1F **hMCEtaPt;
    TH1F **hMCEtaGGPt;
    TH1F **hMCPi0InAccPt;
    TH1F **hMCEtaInAccPt;
    TH2F **hESDTrueMotherInvMassPt;
    TH2F **hESDTrueMotherPi0GGInvMassPt;
    TH2F **hESDTruePrimaryMotherInvMassMCPt;
    TH2F **hESDTruePrimaryPi0DalitzESDPtMCPt;
    TH2F **hESDTrueSecondaryMotherInvMassPt;
    TH2F **hESDTrueSecondaryMotherFromK0sInvMassPt;
    TH2F **hESDTrueBckGGInvMassPt;
    TH2F **hESDTrueBckContInvMassPt;
    TH2F **hESDTrueMotherGGInvMassPt;
    TH1F **hESDTrueConvGammaPt;
    TH1I **hNEvents;
    TH1I **hNGoodESDTracks;
        
    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;
   
	private:
		AliAnalysisTaskGammaConvDalitzV1( const AliAnalysisTaskGammaConvDalitzV1& ); // Not implemented
		AliAnalysisTaskGammaConvDalitzV1& operator=( const AliAnalysisTaskGammaConvDalitzV1& ); // Not implemented

		ClassDef( AliAnalysisTaskGammaConvDalitzV1, 2 );
};

#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"
	15
	16	class AliESDInputHandler;
	17	class AliMCEventHandler;
	18	class AliESDEvent;
	19	class AliESDtrack;
	20	class AliESDtrackCuts;
	21	class AliESDpidCuts;
	22	class AliV0Reader;
	23	class AliGammaConversionHistograms;
	24	class AliTriggerAnalysis;
	25
	26	class AliAnalysisTaskGammaConvDalitzV1: public AliAnalysisTaskSE
	27	{
	28	public:
	29
	30	AliAnalysisTaskGammaConvDalitzV1();
	31	AliAnalysisTaskGammaConvDalitzV1( const char* name );
	32	virtual ~AliAnalysisTaskGammaConvDalitzV1();
	33
	34	virtual void UserExec(Option_t *);
	35	virtual void UserCreateOutputObjects();
	36	virtual void Terminate(const Option_t *);
	37
	38
	39
	40
	41	void SetMoveParticleAccordingToVertex(Bool_t flag){fMoveParticleAccordingToVertex = flag;}
	42	void SetIsHeavyIon(Bool_t flag){fIsHeavyIon = flag;}
	43	void SetConversionCutList(Int_t nCuts, TList *CutArray){
	44	fnCuts= nCuts;
	45	fCutGammaArray = CutArray;
	46	}
	47	void SetElectronCutList(TList *CutArray){
	48	fCutElectronArray = CutArray;
	49	}
	50	void SetMesonCutList(TList *CutArray){
	51	fCutMesonArray = CutArray;
	52	}
	53
	54
	55	private:
	56
2bb2434e	57	void InitBack();
	58	void ProcessPhotonCandidates();
	59	void ProcessTruePhotonCandidates(AliAODConversionPhoton*);
	60	void ProcessTrueMesonCandidates(AliAODConversionMother Pi0Candidate, AliAODConversionPhoton TrueGammaCandidate, AliAODConversionPhoton *TrueVirtualGammaCandidate);
	61	void MoveParticleAccordingToVertex(AliAODConversionPhoton* particle,const AliGammaConversionAODBGHandler::GammaConversionVertex *vertex);
	62	void ProcessElectronCandidates();
	63	void ProcessMCParticles();
	64	void CountESDTracks();
	65	void CalculatePi0DalitzCandidates();
	66	void CalculateBackground();
	67	void UpdateEventByEventData();
a280ac15	68	Double_t GetPsiPair( const AliESDtrack trackPos, const AliESDtrack trackNeg ) const;
a280ac15	69
2bb2434e	70
	71
	72
	73	AliV0ReaderV1 *fV0Reader;
	74	AliDalitzElectronSelector* fElecSelector;
	75	AliGammaConversionAODBGHandler **fBGHandler;
	76	AliESDEvent *fESDEvent;
	77	AliMCEvent *fMCEvent;
	78	AliStack *fMCStack;
	79	TList **fCutFolder;
	80	TList **fESDList;
	81	TList **fBackList;
	82	TList **fTrueList;
	83	TList **fMCList;
	84	TList *fOutputContainer;
	85	TClonesArray *fReaderGammas;
	86	vector<Int_t> fSelectorElectronIndex;
	87	vector<Int_t> fSelectorPositronIndex;
	88	TList *fGoodGammas;
	89	TList *fGoodVirtualGammas;
	90	TList *fGoodElectrons;
	91	TList *fGoodPositrons;
	92	TList *fCutGammaArray;
	93	TList *fCutElectronArray;
	94	TList *fCutMesonArray;
	95	TList **fGammasPool;
	96	AliConversionCuts *fConversionCuts;
	97	TH1F **hESDConvGammaPt;
	98	TH1F **hESDDalitzElectronPt;
	99	TH1F **hESDDalitzPositronPt;
	100	TH2F **hESDEposEnegPsiPairDPhi;
a280ac15	101	TH2F **hESDEposEnegInvMassPt;
0a2b2b4b	102	TH2F **hESDEposEnegLikeSignBackInvMassPt;
2bb2434e	103	TH2F **hESDMotherInvMassPt;
a280ac15	104	TH2F **hESDPi0MotherInvMassPt;
a280ac15	105	TH2F **hESDPi0MotherDiffInvMassPt;
2bb2434e	106	THnSparseF **sESDMotherInvMassPtZM;
	107	TH2F **hESDMotherBackInvMassPt;
	108	THnSparseF **sESDMotherBackInvMassPtZM;
	109	TH1F **hMCPi0Pt;
a280ac15	110	TH1F **hMCPi0GGPt;
2bb2434e	111	TH1F **hMCEtaPt;
a280ac15	112	TH1F **hMCEtaGGPt;
2bb2434e	113	TH1F **hMCPi0InAccPt;
	114	TH1F **hMCEtaInAccPt;
	115	TH2F **hESDTrueMotherInvMassPt;
	116	TH2F **hESDTrueMotherPi0GGInvMassPt;
	117	TH2F **hESDTruePrimaryMotherInvMassMCPt;
	118	TH2F **hESDTruePrimaryPi0DalitzESDPtMCPt;
	119	TH2F **hESDTrueSecondaryMotherInvMassPt;
	120	TH2F **hESDTrueSecondaryMotherFromK0sInvMassPt;
	121	TH2F **hESDTrueBckGGInvMassPt;
	122	TH2F **hESDTrueBckContInvMassPt;
	123	TH2F **hESDTrueMotherGGInvMassPt;
	124	TH1F **hESDTrueConvGammaPt;
	125	TH1I **hNEvents;
	126	TH1I **hNGoodESDTracks;
	127
	128	TRandom3 fRandom;
	129	Double_t *fUnsmearedPx;
	130	Double_t *fUnsmearedPy;
	131	Double_t *fUnsmearedPz;
	132	Double_t *fUnsmearedE;
	133	Int_t fnCuts;
	134	Int_t fiCut;
	135	Int_t fNumberOfESDTracks;
	136	Bool_t fMoveParticleAccordingToVertex;
	137	Bool_t fIsHeavyIon;
	138	Bool_t fDoMesonAnalysis;
	139
	140	private:
	141	AliAnalysisTaskGammaConvDalitzV1( const AliAnalysisTaskGammaConvDalitzV1& ); // Not implemented
	142	AliAnalysisTaskGammaConvDalitzV1& operator=( const AliAnalysisTaskGammaConvDalitzV1& ); // Not implemented
	143
	144	ClassDef( AliAnalysisTaskGammaConvDalitzV1, 2 );
	145	};
	146
	147	#endif // ALIANALYSISTASKGAMMACONVDALITZV1_H
	148