[u/mrichter/AliRoot.git] / PWG2 / SPECTRA / AliAnalysisTaskCheckCascade.h

#ifndef ALIANALYSISTASKCHECKCASCADE_H
#define ALIANALYSISTASKCHECKCASCADE_H

/*  See cxx source for full Copyright notice */

//-----------------------------------------------------------------
//                 AliAnalysisTaskCheckCascade class
//            (AliAnalysisTaskCheckCascade)
//            This task is for QAing the Cascades from ESD and AOD
//              Origin:  AliAnalysisTaskESDCheckV0 by B.H. Nov2007, hippolyt@in2p3.fr
//            Adapted to Cascade : A.Maire Mar2008, antonin.maire@ires.in2p3.fr
//	      Modified :           A.Maire Jun2009, antonin.maire@ires.in2p3.fr
//-----------------------------------------------------------------

class TList;
class TH1F;
class TH2F;
 

#include "TString.h"

#include "AliAnalysisTaskSE.h"

class AliAnalysisTaskCheckCascade : public AliAnalysisTaskSE {
 public:
  AliAnalysisTaskCheckCascade();
  AliAnalysisTaskCheckCascade(const char *name);
  virtual ~AliAnalysisTaskCheckCascade() {}
  
  virtual void   UserCreateOutputObjects();
  virtual void   UserExec(Option_t *option);
  virtual void   Terminate(Option_t *);
  
  void SetCollidingSystems(Short_t collidingSystems = 0)     {fCollidingSystems = collidingSystems;}
  void SetAnalysisType    (const char* analysisType = "ESD") {fAnalysisType = analysisType;}
    
 private:
  	TString fAnalysisType;				// "ESD" or "AOD" analysis type	
	Short_t fCollidingSystems;			// 0 = pp collisions or 1 = AA collisions
	

		TList	*fListHistCascade;		//! List of Cascade histograms
	
	// - General histos (filled for any event)
	TH1F	*fHistTrackMultiplicity;		//! Track multiplicity distribution
	TH1F	*fHistCascadeMultiplicity;		//! Cascade multiplicity distribution


	// - Vertex Positions
	TH1F	*fHistVtxStatus;			//! Is there a tracking vertex in the event ?

		// Vtx coming from the full tracking
	TH1F	*fHistPosTrkgPrimaryVtxX;  		//! primary vertex position distribution in x 
	TH1F	*fHistPosTrkgPrimaryVtxY;		//! primary vertex position distribution in y
	TH1F	*fHistPosTrkgPrimaryVtxZ;  		//! primary vertex position distribution in z
	TH1F	*fHistTrkgPrimaryVtxRadius;		//! primary vertex (3D) radius distribution 

		// Best primary Vtx available for the event
	TH1F	*fHistPosBestPrimaryVtxX;  		//! (best) primary vertex position distribution in x 
	TH1F	*fHistPosBestPrimaryVtxY;		//! (best) primary vertex position distribution in y
	TH1F	*fHistPosBestPrimaryVtxZ;  		//! (best) primary vertex position distribution in z
	TH1F	*fHistBestPrimaryVtxRadius;		//! (best) primary vertex radius distribution 
	
		// Correlation Best Vtx / Full Tracking Vtx
	TH2F	*f2dHistTrkgPrimVtxVsBestPrimVtx;	//!  Radius of prim. Vtx from tracks Vs Radius of best Prim. Vtx
	
	
	// - Typical histos on the variables used for the selection of cascades
	TH1F	*fHistEffMassXi;      			//! reconstructed cascade effective mass
	TH1F	*fHistChi2Xi;         			//! chi2 value
	TH1F	*fHistDcaXiDaughters; 			//! dca between Xi's daughters
	TH1F 	*fHistDcaBachToPrimVertex;		//! dca of the bachelor track to primary vertex
	TH1F	*fHistXiCosineOfPointingAngle;		//! cosine of Xi pointing angle in a cascade
	TH1F	*fHistXiRadius;				//! (transverse) radius of the cascade vertex 
		
	// - Histos about ~ the "V0 selection part" of the cascade,  coming by inheritance from AliESDv0
	TH1F	*fHistMassLambdaAsCascDghter;		//! Test Invariant Mass of Lambda coming from Cascade
	TH1F	*fHistV0Chi2Xi;				//! V0 chi2 distribution, for the V0 associated to a cascade
	TH1F	*fHistDcaV0DaughtersXi;			//! Dca between V0 daughters, for the V0 associated to a cascade
	TH1F	*fHistDcaV0ToPrimVertexXi;		//! Dca of V0 to primary vertex, for the V0 associated to a cascade	
	TH1F	*fHistV0CosineOfPointingAngleXi;	//! Cosine of V0 pointing angle, for the V0 associated to a cascade
	TH1F	*fHistV0RadiusXi;			//! V0 (transverse) distance distribution, for the V0 associated to a cascade

	TH1F	*fHistDcaPosToPrimVertexXi;		//! Dca of V0 positive daughter to primary vertex, for the V0 associated to a cascade
	TH1F	*fHistDcaNegToPrimVertexXi;		//! Dca of V0 negative daughter to primary vertex, for the V0 associated to a cascade
	

	// - Effective mass histos for cascades.
	TH1F	*fHistMassXiMinus;			//! reconstructed cascade effective mass, under Xi- hyp.
	TH1F	*fHistMassXiPlus;			//! reconstructed cascade effective mass, under Xi+ hyp.
	TH1F	*fHistMassOmegaMinus;			//! reconstructed cascade effective mass, under Omega- hyp.
	TH1F	*fHistMassOmegaPlus;			//! reconstructed cascade effective mass, under Omega+ hyp.
	
	TH1F	*fHistMassWithCombPIDXiMinus;		//! reconstructed Xi- effective mass, with bach. comb PID
	TH1F	*fHistMassWithCombPIDXiPlus;		//! reconstructed Xi+ effective mass, with bach. comb PID
	TH1F	*fHistMassWithCombPIDOmegaMinus;	//! reconstructed Omega- effective mass, with bach. comb PID
	TH1F	*fHistMassWithCombPIDOmegaPlus;		//! reconstructed Omega+ effective mass, with bach. comb PID

	// - Complements for QA
	TH1F	*fHistXiTransvMom;     			//! Xi transverse momentum 
	TH1F	*fHistXiTotMom;     			//! Xi momentum norm
	
	TH1F	*fHistBachTransvMom;   			//! bachelor transverse momentum 
	TH1F	*fHistBachTotMom;  			//! bachelor momentum norm
				
	TH1F	*fHistChargeXi;				//! Charge sign of the cascade candidate
	TH1F	*fHistV0toXiCosineOfPointingAngle;	//! Cos. of Pointing angle between the V0 mom and the Xi-V0 vtx line
  
	TH1F	*fHistRapXi;				//! rapidity of Xi candidates
	TH1F	*fHistRapOmega;				//! rapidity of Omega candidates
	TH1F	*fHistEta;				//! eta distrib. of all the cascade candidates
	TH1F	*fHistTheta;				//! theta distrib. of all the cascade candidates
	TH1F	*fHistPhi;				//! phi distrib. of all the cascade candidates
	
	TH2F	*f2dHistArmenteros;			//! alpha(casc. cand.) Vs PtArm(casc. cand.)
	
	TH2F	*f2dHistEffMassLambdaVsEffMassXiMinus;	//! Xi- Eff mass Vs V0 Eff mass, under Xi- hyp.
	TH2F	*f2dHistEffMassXiVsEffMassOmegaMinus;	//! Xi- Eff mass Vs Omega- Eff mass, for negative cascades
	TH2F	*f2dHistEffMassLambdaVsEffMassXiPlus;	//! Xi+ Eff mass Vs V0 Eff mass, under Xi+ hyp.
	TH2F	*f2dHistEffMassXiVsEffMassOmegaPlus;	//! Xi+ Eff mass Vs Omega+ Eff mass, for positive cascades
	
	TH2F	*f2dHistXiRadiusVsEffMassXiMinus;	//! transv. casc. decay radius Vs Xi- Eff mass, under Xi- hyp.
	TH2F	*f2dHistXiRadiusVsEffMassXiPlus;	//! transv. casc. decay radius Vs Xi+ Eff mass, under Xi+ hyp.
	TH2F	*f2dHistXiRadiusVsEffMassOmegaMinus;	//! transv. casc. decay radius Vs Omega- Eff mass, under Omega- hyp.
	TH2F	*f2dHistXiRadiusVsEffMassOmegaPlus;	//! transv. casc. decay radius Vs Omega+ Eff mass, under Omega+ hyp.
	
	TH2F	*f2dHistXiPtVsEffMassXiMinus;		//! casc. transv. momemtum Vs Xi- Eff mass
	TH2F	*f2dHistXiPtVsEffMassXiPlus;		//! casc. transv. momemtum Vs Xi+ Eff mass
	TH2F	*f2dHistXiPtVsEffMassOmegaMinus;	//! casc. transv. momemtum Vs Omega- Eff mass
	TH2F	*f2dHistXiPtVsEffMassOmegaPlus;		//! casc. transv. momemtum Vs Omega+ Eff mass

  AliAnalysisTaskCheckCascade(const AliAnalysisTaskCheckCascade&);            // not implemented
  AliAnalysisTaskCheckCascade& operator=(const AliAnalysisTaskCheckCascade&); // not implemented
  
  ClassDef(AliAnalysisTaskCheckCascade, 3);
};

#endif
Commit	Line	Data
684aef2e	1	#ifndef ALIANALYSISTASKCHECKCASCADE_H
684aef2e	2	#define ALIANALYSISTASKCHECKCASCADE_H
eed7ee51	3
	4	/* See cxx source for full Copyright notice */
	5
	6	//-----------------------------------------------------------------
45ee0bcc	7	// AliAnalysisTaskCheckCascade class
45ee0bcc	8	// (AliAnalysisTaskCheckCascade)
684aef2e	9	// This task is for QAing the Cascades from ESD and AOD
eed7ee51	10	// Origin: AliAnalysisTaskESDCheckV0 by B.H. Nov2007, hippolyt@in2p3.fr
e9c3f0b6	11	// Adapted to Cascade : A.Maire Mar2008, antonin.maire@ires.in2p3.fr
e9c3f0b6	12	// Modified : A.Maire Jun2009, antonin.maire@ires.in2p3.fr
eed7ee51	13	//-----------------------------------------------------------------
	14
	15	class TList;
	16	class TH1F;
6dc8b253	17	class TH2F;
45ee0bcc	18
	19
	20	#include "TString.h"
eed7ee51	21
45ee0bcc	22	#include "AliAnalysisTaskSE.h"
eed7ee51	23
45ee0bcc	24	class AliAnalysisTaskCheckCascade : public AliAnalysisTaskSE {
eed7ee51	25	public:
45ee0bcc	26	AliAnalysisTaskCheckCascade();
	27	AliAnalysisTaskCheckCascade(const char *name);
	28	virtual ~AliAnalysisTaskCheckCascade() {}
eed7ee51	29
45ee0bcc	30	virtual void UserCreateOutputObjects();
45ee0bcc	31	virtual void UserExec(Option_t *option);
eed7ee51	32	virtual void Terminate(Option_t *);
eed7ee51	33
83db4a5d	34	void SetCollidingSystems(Short_t collidingSystems = 0) {fCollidingSystems = collidingSystems;}
83db4a5d	35	void SetAnalysisType (const char* analysisType = "ESD") {fAnalysisType = analysisType;}
e9c3f0b6	36
45ee0bcc	37	private:
	38	TString fAnalysisType; // "ESD" or "AOD" analysis type
	39	Short_t fCollidingSystems; // 0 = pp collisions or 1 = AA collisions
eed7ee51	40
eed7ee51	41
db166251	42	TList *fListHistCascade; //! List of Cascade histograms
eed7ee51	43
	44	// - General histos (filled for any event)
	45	TH1F *fHistTrackMultiplicity; //! Track multiplicity distribution
	46	TH1F *fHistCascadeMultiplicity; //! Cascade multiplicity distribution
	47
	48
	49	// - Vertex Positions
f42573c1	50	TH1F *fHistVtxStatus; //! Is there a tracking vertex in the event ?
eed7ee51	51
f42573c1	52	// Vtx coming from the full tracking
	53	TH1F *fHistPosTrkgPrimaryVtxX; //! primary vertex position distribution in x
	54	TH1F *fHistPosTrkgPrimaryVtxY; //! primary vertex position distribution in y
	55	TH1F *fHistPosTrkgPrimaryVtxZ; //! primary vertex position distribution in z
	56	TH1F *fHistTrkgPrimaryVtxRadius; //! primary vertex (3D) radius distribution
	57
	58	// Best primary Vtx available for the event
	59	TH1F *fHistPosBestPrimaryVtxX; //! (best) primary vertex position distribution in x
	60	TH1F *fHistPosBestPrimaryVtxY; //! (best) primary vertex position distribution in y
	61	TH1F *fHistPosBestPrimaryVtxZ; //! (best) primary vertex position distribution in z
	62	TH1F *fHistBestPrimaryVtxRadius; //! (best) primary vertex radius distribution
eed7ee51	63
f42573c1	64	// Correlation Best Vtx / Full Tracking Vtx
f42573c1	65	TH2F *f2dHistTrkgPrimVtxVsBestPrimVtx; //! Radius of prim. Vtx from tracks Vs Radius of best Prim. Vtx
eed7ee51	66
	67
	68
f42573c1	69	// - Typical histos on the variables used for the selection of cascades
eed7ee51	70	TH1F *fHistEffMassXi; //! reconstructed cascade effective mass
	71	TH1F *fHistChi2Xi; //! chi2 value
	72	TH1F *fHistDcaXiDaughters; //! dca between Xi's daughters
f42573c1	73	TH1F *fHistDcaBachToPrimVertex; //! dca of the bachelor track to primary vertex
f42573c1	74	TH1F *fHistXiCosineOfPointingAngle; //! cosine of Xi pointing angle in a cascade
eed7ee51	75	TH1F *fHistXiRadius; //! (transverse) radius of the cascade vertex
eed7ee51	76
f42573c1	77	// - Histos about ~ the "V0 selection part" of the cascade, coming by inheritance from AliESDv0
eed7ee51	78	TH1F *fHistMassLambdaAsCascDghter; //! Test Invariant Mass of Lambda coming from Cascade
f42573c1	79	TH1F *fHistV0Chi2Xi; //! V0 chi2 distribution, for the V0 associated to a cascade
f42573c1	80	TH1F *fHistDcaV0DaughtersXi; //! Dca between V0 daughters, for the V0 associated to a cascade
db166251	81	TH1F *fHistDcaV0ToPrimVertexXi; //! Dca of V0 to primary vertex, for the V0 associated to a cascade
	82	TH1F *fHistV0CosineOfPointingAngleXi; //! Cosine of V0 pointing angle, for the V0 associated to a cascade
	83	TH1F *fHistV0RadiusXi; //! V0 (transverse) distance distribution, for the V0 associated to a cascade
eed7ee51	84
db166251	85	TH1F *fHistDcaPosToPrimVertexXi; //! Dca of V0 positive daughter to primary vertex, for the V0 associated to a cascade
db166251	86	TH1F *fHistDcaNegToPrimVertexXi; //! Dca of V0 negative daughter to primary vertex, for the V0 associated to a cascade
eed7ee51	87
	88
	89	// - Effective mass histos for cascades.
	90	TH1F *fHistMassXiMinus; //! reconstructed cascade effective mass, under Xi- hyp.
	91	TH1F *fHistMassXiPlus; //! reconstructed cascade effective mass, under Xi+ hyp.
	92	TH1F *fHistMassOmegaMinus; //! reconstructed cascade effective mass, under Omega- hyp.
	93	TH1F *fHistMassOmegaPlus; //! reconstructed cascade effective mass, under Omega+ hyp.
f87cd3db	94
	95	TH1F *fHistMassWithCombPIDXiMinus; //! reconstructed Xi- effective mass, with bach. comb PID
	96	TH1F *fHistMassWithCombPIDXiPlus; //! reconstructed Xi+ effective mass, with bach. comb PID
	97	TH1F *fHistMassWithCombPIDOmegaMinus; //! reconstructed Omega- effective mass, with bach. comb PID
	98	TH1F *fHistMassWithCombPIDOmegaPlus; //! reconstructed Omega+ effective mass, with bach. comb PID
f42573c1	99
	100	// - Complements for QA
	101	TH1F *fHistXiTransvMom; //! Xi transverse momentum
	102	TH1F *fHistXiTotMom; //! Xi momentum norm
	103
	104	TH1F *fHistBachTransvMom; //! bachelor transverse momentum
	105	TH1F *fHistBachTotMom; //! bachelor momentum norm
	106
	107	TH1F *fHistChargeXi; //! Charge sign of the cascade candidate
	108	TH1F *fHistV0toXiCosineOfPointingAngle; //! Cos. of Pointing angle between the V0 mom and the Xi-V0 vtx line
eed7ee51	109
f42573c1	110	TH1F *fHistRapXi; //! rapidity of Xi candidates
	111	TH1F *fHistRapOmega; //! rapidity of Omega candidates
	112	TH1F *fHistEta; //! eta distrib. of all the cascade candidates
	113	TH1F *fHistTheta; //! theta distrib. of all the cascade candidates
	114	TH1F *fHistPhi; //! phi distrib. of all the cascade candidates
	115
	116	TH2F *f2dHistArmenteros; //! alpha(casc. cand.) Vs PtArm(casc. cand.)
45ee0bcc	117
	118	TH2F *f2dHistEffMassLambdaVsEffMassXiMinus; //! Xi- Eff mass Vs V0 Eff mass, under Xi- hyp.
	119	TH2F *f2dHistEffMassXiVsEffMassOmegaMinus; //! Xi- Eff mass Vs Omega- Eff mass, for negative cascades
	120	TH2F *f2dHistEffMassLambdaVsEffMassXiPlus; //! Xi+ Eff mass Vs V0 Eff mass, under Xi+ hyp.
	121	TH2F *f2dHistEffMassXiVsEffMassOmegaPlus; //! Xi+ Eff mass Vs Omega+ Eff mass, for positive cascades
	122
	123	TH2F *f2dHistXiRadiusVsEffMassXiMinus; //! transv. casc. decay radius Vs Xi- Eff mass, under Xi- hyp.
	124	TH2F *f2dHistXiRadiusVsEffMassXiPlus; //! transv. casc. decay radius Vs Xi+ Eff mass, under Xi+ hyp.
e9c3f0b6	125	TH2F *f2dHistXiRadiusVsEffMassOmegaMinus; //! transv. casc. decay radius Vs Omega- Eff mass, under Omega- hyp.
	126	TH2F *f2dHistXiRadiusVsEffMassOmegaPlus; //! transv. casc. decay radius Vs Omega+ Eff mass, under Omega+ hyp.
	127
	128	TH2F *f2dHistXiPtVsEffMassXiMinus; //! casc. transv. momemtum Vs Xi- Eff mass
	129	TH2F *f2dHistXiPtVsEffMassXiPlus; //! casc. transv. momemtum Vs Xi+ Eff mass
	130	TH2F *f2dHistXiPtVsEffMassOmegaMinus; //! casc. transv. momemtum Vs Omega- Eff mass
	131	TH2F *f2dHistXiPtVsEffMassOmegaPlus; //! casc. transv. momemtum Vs Omega+ Eff mass
eed7ee51	132
45ee0bcc	133	AliAnalysisTaskCheckCascade(const AliAnalysisTaskCheckCascade&); // not implemented
45ee0bcc	134	AliAnalysisTaskCheckCascade& operator=(const AliAnalysisTaskCheckCascade&); // not implemented
eed7ee51	135
45ee0bcc	136	ClassDef(AliAnalysisTaskCheckCascade, 3);
eed7ee51	137	};
	138
	139	#endif