[u/mrichter/AliRoot.git] / PWGGA / EMCALTasks / AliAnalysisTaskPi0V2.h

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

/* $Id: AliAnalysisTaskPi0V2.h 45956 2010-12-10 12:55:37Z agheata $ */
/* AliAnalysisTaskPi0V2.h
 *
 * Template task producing a P_t spectrum and pseudorapidity distribution.
 * Includes explanations of physics and primary track selections
 *
 * Based on tutorial example from offline pages
 * Edited by Arvinder Palaha
 */
#ifndef ALIANALYSISTASKPI0V2_H
#define ALIANALYSISTASKPI0V2_H

class TH1F;
class TH2F;
class TH3F;
class TList;
class AliESDCaloCluster;
class AliESDtrackCuts;
class AliESDEvent;
class THnSparse;
class TClonesArray;
class TString;

#ifndef ALIANALYSISTASKSE_H
#include "AliAnalysisTaskSE.h"
#endif

class AliAnalysisTaskPi0V2 : public AliAnalysisTaskSE {
 public:
    AliAnalysisTaskPi0V2();
    AliAnalysisTaskPi0V2(const char *name);
    virtual ~AliAnalysisTaskPi0V2();
    
    virtual void     UserCreateOutputObjects();
    virtual void     UserExec(Option_t *option);
    virtual void     Terminate(Option_t *);
    Double_t		GetCrossEnergy(const AliVCluster *cluster, Short_t &idmax) const;
    Double_t		GetMaxCellEnergy(const AliVCluster *cluster, Short_t &id) const;
    Bool_t		IsWithinFiducialVolume(Short_t id) const;
    Bool_t		IsGoodCluster(const AliESDCaloCluster *c) const;
    Bool_t		IsGoodPion(const TLorentzVector& p1, const TLorentzVector& p2) const;
    void		FillPion(const TLorentzVector& p1, const TLorentzVector& p2, Double_t EPV0r, Double_t EPV0A, Double_t EPV0C, Double_t EPTPC);
    void 		GetMom(TLorentzVector& p, const AliESDCaloCluster *c, Double_t *vertex);		
    void		SetEventMethod(Double_t e )	{ fEvtSelect  =e ;}
    void		SetVtxCut(Double_t v )	        { fVtxCut     =v ;}
    void		SetClusNcell(Double_t c )	{ fNcellCut   =c ;}
    void		SetClusE(Double_t e )	        { fECut       =e ;}
    void		SetClusEta(Double_t e )	        { fEtaCut     =e ;}
    void		SetClusM02(Double_t m )	        { fM02Cut     =m ;}
    void		SetPi0Asy(Double_t a )	        { fPi0AsyCut  =a ;}
    void                SetTracksName(const char *n)    { fTracksName =n ;}

    
 private:
    TList           		*fOutput;        //! Output list
    AliESDEvent			*fESD;          //!ESD object
    TString                     fTracksName;    // name of track collection
    TClonesArray                *fTracks;       //! pico tracks specific for Skim ESD
    Double_t 			fEvtSelect;	// 1 = MB+Semi+Central, 2 = MB+Semi, 3 = MB;
    Double_t			fVtxCut;	// vertex cut
    Double_t			fNcellCut;      // N cells Cut
    Double_t			fECut;		// Cluster E cut
    Double_t			fEtaCut;	// Cluster Eta Cut
    Double_t			fM02Cut;	// Cluster long axis cut
    Bool_t			fPi0AsyCut;	// pion Asymetry cut 0=off 1=on
    Double_t			fCentrality;	//! Centrality
    Double_t			fEPTPC;		//! Evt plane TPC
    Double_t			fEPTPCreso;	//! resolution of TPC method
    Double_t			fEPV0;		//! EP V0
    Double_t			fEPV0A;		//! EP V0A
    Double_t			fEPV0C;		//! EP V0C
    Double_t			fEPV0Ar;	//! EP V0A reduced
    Double_t			fEPV0Cr;	//! EP V0C reduced
    Double_t			fEPV0r;		//! EP V0 reduced
    Double_t			fEPV0AR4;	//! EP V0A ring4 only
    Double_t			fEPV0AR5;	//! EP V0A ring5 only
    Double_t			fEPV0AR6;	//! EP V0A ring6 only
    Double_t			fEPV0AR7;	//! EP V0A ring7 only
    Double_t			fEPV0CR0;	//! EP V0C ring0 only	
    Double_t			fEPV0CR1;	//! EP V0C ring1 only	
    Double_t			fEPV0CR2;	//! EP V0C ring2 only	
    Double_t			fEPV0CR3;	//! EP V0C ring3 only	

    TH1F			*hEvtCount;	//!
    TH1F			*hAllcentV0;	//!
    TH1F			*hAllcentV0r;	//!
    TH1F			*hAllcentV0A;	//!
    TH1F			*hAllcentV0C;	//!
    TH1F			*hAllcentTPC;   //!
    TH2F			*hEPTPC;	//! 2-D histo EPTPC  vs cent
    TH2F			*hresoTPC;	//! 2-D histo TPC resolution vs cent
    TH2F			*hEPV0;		//! 2-D histo EPV0   vs cent
    TH2F			*hEPV0A;	//! 2-D histo EPV0A  vs cent
    TH2F			*hEPV0C;	//! 2-D histo EPV0C  vs cent
    TH2F			*hEPV0Ar;	//! 2-D histo EPV0Ar vs cent
    TH2F			*hEPV0Cr;	//! 2-D histo EPV0Cr vs cent
    TH2F			*hEPV0r;	//! 2-D histo EPV0r  vs cent
    TH2F			*hEPV0AR4;	//! 2-D histo EPV0AR4 vs cent
    TH2F			*hEPV0AR7;	//! 2-D histo EPV0AR7 vs cent
    TH2F			*hEPV0CR0;	//! 2-D histo EPV0AR0 vs cent
    TH2F			*hEPV0CR3;	//! 2-D histo EPV0AR3 vs cent

    TH2F			*hdifV0A_V0CR0;		//! 2-D histo diff V0A, V0CR0 vs cent
    TH2F			*hdifV0A_V0CR3;		//! 2-D histo diff V0A, V0CR3 vs cent
    TH2F			*hdifV0ACR0_V0CR3;	//! 2-D histo diff V0CR0, V0CR3 vs cent
    TH2F			*hdifV0C_V0AR4;		//! 2-D histo diff V0C, V0AR4 vs cent
    TH2F			*hdifV0C_V0AR7;		//! 2-D histo diff V0C, V0AR7 vs cent
    TH2F			*hdifV0AR4_V0AR7;	//! 2-D histo diff V0AR7, V0AR4 vs cent

    TH2F			*hdifV0A_V0C;		//! 2-D histo diff V0A - V0C
    TH2F			*hdifV0A_TPC;		//! 2-D histo diff V0A - TPC
    TH2F			*hdifV0C_TPC;		//! 2-D histo diff V0C - TPC
    TH2F			*hdifV0C_V0A;		//! 2-D histo diff V0C - V0A

    TH3F			*hdifEMC_EPV0;		//! 3-D histo dif phi in EMC with EPV0
    TH3F			*hdifEMC_EPV0A;		//! 3-D histo dif phi in EMC with EPV0A
    TH3F			*hdifEMC_EPV0C;		//! 3-D histo dif phi in EMC with EPV0C

    TH3F			*hdifful_EPV0;		//! 3-D histo dif phi in full with EPV0
    TH3F			*hdifful_EPV0A;		//! 3-D histo dif phi in full with EPV0A
    TH3F			*hdifful_EPV0C;		//! 3-D histo dif phi in full with EPV0C

    TH3F			*hdifout_EPV0;		//! 3-D histo dif phi out EMC with EPV0
    TH3F			*hdifout_EPV0A;		//! 3-D histo dif phi out EMC with EPV0A
    TH3F			*hdifout_EPV0C;		//! 3-D histo dif phi out EMC with EPV0C

    TH3F			*hdifEMC_EPTPC;		//! 3-D histo dif phi in EMC with EPTPC
    TH3F			*hdifful_EPTPC;		//! 3-D histo dif phi in full with EPTPC
    TH3F			*hdifout_EPTPC;		//! 3-D histo dif phi out EMC with EPTPC

    THnSparse                   *fHEPV0r;	        //! Flow 4-D Histo
    THnSparse                   *fHEPV0A;	        //! Flow 4-D Histo
    THnSparse                   *fHEPV0C;	        //! Flow 4-D Histo
    THnSparse                   *fHEPTPC;	        //! Flow 4-D Histo

    
    
    AliAnalysisTaskPi0V2(const AliAnalysisTaskPi0V2&); // not implemented
    AliAnalysisTaskPi0V2& operator=(const AliAnalysisTaskPi0V2&); // not implemented
    
    ClassDef(AliAnalysisTaskPi0V2, 4); // example of analysis
};

#endif
Commit	Line	Data
04b116e8	1	/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
04b116e8	2	* See cxx source for full Copyright notice */
3c40321c	3
04b116e8	4	/* $Id: AliAnalysisTaskPi0V2.h 45956 2010-12-10 12:55:37Z agheata $ */
	5	/* AliAnalysisTaskPi0V2.h
	6	*
	7	* Template task producing a P_t spectrum and pseudorapidity distribution.
	8	* Includes explanations of physics and primary track selections
	9	*
	10	* Based on tutorial example from offline pages
	11	* Edited by Arvinder Palaha
	12	*/
8071d5b2	13	#ifndef ALIANALYSISTASKPI0V2_H
8071d5b2	14	#define ALIANALYSISTASKPI0V2_H
3c40321c	15
	16	class TH1F;
	17	class TH2F;
04b116e8	18	class TH3F;
3c40321c	19	class TList;
	20	class AliESDCaloCluster;
	21	class AliESDtrackCuts;
	22	class AliESDEvent;
04b116e8	23	class THnSparse;
e5ba16b4	24	class TClonesArray;
e5ba16b4	25	class TString;
3c40321c	26
04b116e8	27	#ifndef ALIANALYSISTASKSE_H
3c40321c	28	#include "AliAnalysisTaskSE.h"
04b116e8	29	#endif
3c40321c	30
	31	class AliAnalysisTaskPi0V2 : public AliAnalysisTaskSE {
	32	public:
965c985f	33	AliAnalysisTaskPi0V2();
8f40bd27	34	AliAnalysisTaskPi0V2(const char *name);
04b116e8	35	virtual ~AliAnalysisTaskPi0V2();
3c40321c	36
04b116e8	37	virtual void UserCreateOutputObjects();
	38	virtual void UserExec(Option_t *option);
	39	virtual void Terminate(Option_t *);
	40	Double_t GetCrossEnergy(const AliVCluster *cluster, Short_t &idmax) const;
	41	Double_t GetMaxCellEnergy(const AliVCluster *cluster, Short_t &id) const;
	42	Bool_t IsWithinFiducialVolume(Short_t id) const;
	43	Bool_t IsGoodCluster(const AliESDCaloCluster *c) const;
	44	Bool_t IsGoodPion(const TLorentzVector& p1, const TLorentzVector& p2) const;
	45	void FillPion(const TLorentzVector& p1, const TLorentzVector& p2, Double_t EPV0r, Double_t EPV0A, Double_t EPV0C, Double_t EPTPC);
	46	void GetMom(TLorentzVector& p, const AliESDCaloCluster c, Double_t vertex);
670ffa5c	47	void SetEventMethod(Double_t e ) { fEvtSelect =e ;}
	48	void SetVtxCut(Double_t v ) { fVtxCut =v ;}
	49	void SetClusNcell(Double_t c ) { fNcellCut =c ;}
	50	void SetClusE(Double_t e ) { fECut =e ;}
	51	void SetClusEta(Double_t e ) { fEtaCut =e ;}
	52	void SetClusM02(Double_t m ) { fM02Cut =m ;}
	53	void SetPi0Asy(Double_t a ) { fPi0AsyCut =a ;}
e5ba16b4	54	void SetTracksName(const char *n) { fTracksName =n ;}
04b116e8	55
3c40321c	56
3c40321c	57	private:
90340bdc	58	TList *fOutput; //! Output list
04b116e8	59	AliESDEvent *fESD; //!ESD object
e5ba16b4	60	TString fTracksName; // name of track collection
e5ba16b4	61	TClonesArray *fTracks; //! pico tracks specific for Skim ESD
670ffa5c	62	Double_t fEvtSelect; // 1 = MB+Semi+Central, 2 = MB+Semi, 3 = MB;
	63	Double_t fVtxCut; // vertex cut
	64	Double_t fNcellCut; // N cells Cut
	65	Double_t fECut; // Cluster E cut
	66	Double_t fEtaCut; // Cluster Eta Cut
	67	Double_t fM02Cut; // Cluster long axis cut
	68	Bool_t fPi0AsyCut; // pion Asymetry cut 0=off 1=on
04b116e8	69	Double_t fCentrality; //! Centrality
8f40bd27	70	Double_t fEPTPC; //! Evt plane TPC
04b116e8	71	Double_t fEPTPCreso; //! resolution of TPC method
04b116e8	72	Double_t fEPV0; //! EP V0
8f40bd27	73	Double_t fEPV0A; //! EP V0A
8f40bd27	74	Double_t fEPV0C; //! EP V0C
04b116e8	75	Double_t fEPV0Ar; //! EP V0A reduced
04b116e8	76	Double_t fEPV0Cr; //! EP V0C reduced
8f40bd27	77	Double_t fEPV0r; //! EP V0 reduced
04b116e8	78	Double_t fEPV0AR4; //! EP V0A ring4 only
	79	Double_t fEPV0AR5; //! EP V0A ring5 only
	80	Double_t fEPV0AR6; //! EP V0A ring6 only
	81	Double_t fEPV0AR7; //! EP V0A ring7 only
	82	Double_t fEPV0CR0; //! EP V0C ring0 only
	83	Double_t fEPV0CR1; //! EP V0C ring1 only
	84	Double_t fEPV0CR2; //! EP V0C ring2 only
	85	Double_t fEPV0CR3; //! EP V0C ring3 only
	86
ef7e23cf	87	TH1F *hEvtCount; //!
ebaf288d	88	TH1F *hAllcentV0; //!
	89	TH1F *hAllcentV0r; //!
	90	TH1F *hAllcentV0A; //!
	91	TH1F *hAllcentV0C; //!
	92	TH1F *hAllcentTPC; //!
04b116e8	93	TH2F *hEPTPC; //! 2-D histo EPTPC vs cent
	94	TH2F *hresoTPC; //! 2-D histo TPC resolution vs cent
	95	TH2F *hEPV0; //! 2-D histo EPV0 vs cent
	96	TH2F *hEPV0A; //! 2-D histo EPV0A vs cent
	97	TH2F *hEPV0C; //! 2-D histo EPV0C vs cent
	98	TH2F *hEPV0Ar; //! 2-D histo EPV0Ar vs cent
	99	TH2F *hEPV0Cr; //! 2-D histo EPV0Cr vs cent
	100	TH2F *hEPV0r; //! 2-D histo EPV0r vs cent
	101	TH2F *hEPV0AR4; //! 2-D histo EPV0AR4 vs cent
	102	TH2F *hEPV0AR7; //! 2-D histo EPV0AR7 vs cent
	103	TH2F *hEPV0CR0; //! 2-D histo EPV0AR0 vs cent
	104	TH2F *hEPV0CR3; //! 2-D histo EPV0AR3 vs cent
	105
	106	TH2F *hdifV0A_V0CR0; //! 2-D histo diff V0A, V0CR0 vs cent
	107	TH2F *hdifV0A_V0CR3; //! 2-D histo diff V0A, V0CR3 vs cent
	108	TH2F *hdifV0ACR0_V0CR3; //! 2-D histo diff V0CR0, V0CR3 vs cent
	109	TH2F *hdifV0C_V0AR4; //! 2-D histo diff V0C, V0AR4 vs cent
	110	TH2F *hdifV0C_V0AR7; //! 2-D histo diff V0C, V0AR7 vs cent
	111	TH2F *hdifV0AR4_V0AR7; //! 2-D histo diff V0AR7, V0AR4 vs cent
	112
	113	TH2F *hdifV0A_V0C; //! 2-D histo diff V0A - V0C
	114	TH2F *hdifV0A_TPC; //! 2-D histo diff V0A - TPC
	115	TH2F *hdifV0C_TPC; //! 2-D histo diff V0C - TPC
	116	TH2F *hdifV0C_V0A; //! 2-D histo diff V0C - V0A
	117
93df010a	118	TH3F *hdifEMC_EPV0; //! 3-D histo dif phi in EMC with EPV0
	119	TH3F *hdifEMC_EPV0A; //! 3-D histo dif phi in EMC with EPV0A
	120	TH3F *hdifEMC_EPV0C; //! 3-D histo dif phi in EMC with EPV0C
	121
	122	TH3F *hdifful_EPV0; //! 3-D histo dif phi in full with EPV0
	123	TH3F *hdifful_EPV0A; //! 3-D histo dif phi in full with EPV0A
	124	TH3F *hdifful_EPV0C; //! 3-D histo dif phi in full with EPV0C
	125
	126	TH3F *hdifout_EPV0; //! 3-D histo dif phi out EMC with EPV0
	127	TH3F *hdifout_EPV0A; //! 3-D histo dif phi out EMC with EPV0A
	128	TH3F *hdifout_EPV0C; //! 3-D histo dif phi out EMC with EPV0C
	129
	130	TH3F *hdifEMC_EPTPC; //! 3-D histo dif phi in EMC with EPTPC
	131	TH3F *hdifful_EPTPC; //! 3-D histo dif phi in full with EPTPC
	132	TH3F *hdifout_EPTPC; //! 3-D histo dif phi out EMC with EPTPC
04b116e8	133
8f40bd27	134	THnSparse *fHEPV0r; //! Flow 4-D Histo
	135	THnSparse *fHEPV0A; //! Flow 4-D Histo
	136	THnSparse *fHEPV0C; //! Flow 4-D Histo
	137	THnSparse *fHEPTPC; //! Flow 4-D Histo
04b116e8	138
04b116e8	139
3c40321c	140
04b116e8	141	AliAnalysisTaskPi0V2(const AliAnalysisTaskPi0V2&); // not implemented
	142	AliAnalysisTaskPi0V2& operator=(const AliAnalysisTaskPi0V2&); // not implemented
	143
e5ba16b4	144	ClassDef(AliAnalysisTaskPi0V2, 4); // example of analysis
3c40321c	145	};
04b116e8	146
3c40321c	147	#endif
04b116e8	148