[u/mrichter/AliRoot.git] / PWGDQ / dielectron / AliAnalysisTaskDielectronReadAODBranch.h

#ifndef ALIANALYSISTASKDIELECTRONREADAODBRANCH_H
#define ALIANALYSISTASKDIELECTRONREADAODBRANCH_H

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

////////////////////////////////////////////////////////////////////////////
//  Class to retrieve the branch of the dielectron candidates stored in   //
//  filtered AOD files (AliAOD.Dielectron.root). It is possible to        //
//  apply tighter cuts to candidates stored in the branch and do the      //
//  matching with MC truth                                                //
//                                                                        //
//  Contacts : G. Bruno / Giuseppe.Bruno@ba.infn.it,                      //
//             F. Fionda / Fiorella.Fionda@ba.infn.it,                    //
//             C. Di Giglio / Carmelo.Digiglio@ba.infn.it                 // 
////////////////////////////////////////////////////////////////////////////

#include "AliAnalysisTaskSE.h"

class TNtuple;
class TH1F;
class TH2F;
class AliDielectronPair;

class AliAnalysisTaskDielectronReadAODBranch : public AliAnalysisTaskSE
{
	public:
		AliAnalysisTaskDielectronReadAODBranch();
		AliAnalysisTaskDielectronReadAODBranch(const char *name);
		virtual ~AliAnalysisTaskDielectronReadAODBranch();

		// Implementation of interface methods
		virtual void UserCreateOutputObjects();
		virtual void Init();
		virtual void LocalInit() {Init();}
		virtual void UserExec(Option_t *option);
		virtual void Terminate(Option_t *option);
              
                //setters
                void SetPtLeg(Double_t cutPt){ fPtCut = cutPt;} 
                void SetSpdFirstRequired(Bool_t spdfirst){fSpdFirstRequired = spdfirst;}
                void SetNclsTPC(Int_t nCls){fClsTPC = nCls;}
                void SetPairType(Int_t type){fPairType = type;}
                void SetPtJpsi(Double_t ptjpsi) {fPtJpsi = ptjpsi;}
                void SetInvMassSignalRegion(Double_t lowLimit, Double_t upLimit){fInvMassSignalLimits[0]=lowLimit; fInvMassSignalLimits[1]=upLimit;}
                void SetInvMassSidebandRegion(Double_t lowLimit, Double_t upLimit){fInvMassSideBandsLimits[0]=lowLimit; fInvMassSideBandsLimits[1]=upLimit;}
                void SetHasMC(Bool_t mcFlag) {fHasMC = mcFlag;}


     enum {
    kITSin=0x0001,kITSout=0x0002,kITSrefit=0x0004,kITSpid=0x0008,
      kTPCin=0x0010,kTPCout=0x0020,kTPCrefit=0x0040,kTPCpid=0x0080,
      kTRDin=0x0100,kTRDout=0x0200,kTRDrefit=0x0400,kTRDpid=0x0800,
      kTOFin=0x1000,kTOFout=0x2000,kTOFrefit=0x4000,kTOFpid=0x8000,
      kHMPIDout=0x10000,kHMPIDpid=0x20000
  };

	private:
		AliAnalysisTaskDielectronReadAODBranch(const AliAnalysisTaskDielectronReadAODBranch &c);
		AliAnalysisTaskDielectronReadAODBranch& operator= (const AliAnalysisTaskDielectronReadAODBranch &c);

		TList      *fOutput;                 // output list with histo
	        TNtuple    *fNtupleJPSI;             // ntupla to store invariant mass and psproper of candidates
                TH1F       *fNentries;               // number of filtered events
		TH1F       *fInvMass;                // invariant mass of candidates in branch after tighter cuts application
		TH1F       *fInvMassNoCuts;          // invariant mass of candidates in branch without cuts
                TH1F       *fpsproperSignal;         // psproper distribution in the mass signal region           
		TH1F       *fpsproperSidebands;      // psproper distribution in the mass sidebands region           
		TH1F       *fpsproperAll;            // whole psproper distribution          
                TH1F       *fpsproperUnder;          // psproper distribution in the lower mass sidebands region 
                TH1F       *fpsproperUpper;          // psproper distribution in the upper mass sidebands region
		TH2F       *fLxyVsPtleg1;            // psproper vs pt for leg1
		TH2F       *fLxyVsPtleg2;            // psproper vs pt for leg2          
		TH2F       *fLxyVsPt;                // psproper vs pt jpsi 
		TH2F       *fMeeVsPt;                // invariant mass vs pt jpsi
		TH2F       *fMeeVsLxy;               // invariant mass vs psproper

		// QA plots
		TH1F       *fprimvtxZ;               // z coord of primary vertex        
		TH1F       *fsecvtxZ;                // z coord of secondary vertex        
		TH1F       *fprimvtxX;               // x coord of primary vertex          
		TH1F       *fsecvtxX;                // x coord of secondary vertex           
		TH1F       *fprimvtxY;               // y coord of primary vertex          
		TH1F       *fsecvtxY;                // y coord of secondary vertex          
		TH1F       *fPt;                     // pT(J/psi) distribution               
		TH1F       *fPtLeg1;                 // pT(leg1)  distribution. Warning: mixture of pos and neg tracks               
		TH1F       *fPtLeg2;                 // pT(leg2)  distribution. Warning: mixture of pos and neg tracks               
		TH2F       *fdEdxP;                  // dEdx vs P for legs
		Bool_t     fHasMC;                   // flag for read MC branch
                TObjArray  *fobj;                    // TObjArray with stored reconstructed candidates
		TObjArray  *fobjMC;                  // TObjArray with MC tracks
 
                // cuts on candidates
                Double_t  fPtCut;                    // ptLeg 
                Bool_t    fSpdFirstRequired;         // spd any/first 
                Int_t     fClsTPC;                   // #clsTPC
                Int_t     fPairType;                 // pair Type
                Double_t  fPtJpsi;                   // pt jpsi
                Double_t  *fInvMassSignalLimits;     // invariant mass signal region to extract psproper distribution
                Double_t  *fInvMassSideBandsLimits;  // invariant mass sideband region to extract psproper distribution 

                Int_t     fSecondary;                // 1(0) to select secondary (prompt) jpsi
		ClassDef(AliAnalysisTaskDielectronReadAODBranch,2); // AliAnalysisTaskSE for the MC association of heavy-flavour decay candidates
};


#endif
Commit	Line	Data
ba15fdfb	1	#ifndef ALIANALYSISTASKDIELECTRONREADAODBRANCH_H
	2	#define ALIANALYSISTASKDIELECTRONREADAODBRANCH_H
	3
	4	/* Copyright(c) 1998-2009, ALICE Experiment at CERN, All rights reserved. *
	5	* See cxx source for full Copyright notice */
	6
	7	////////////////////////////////////////////////////////////////////////////
	8	// Class to retrieve the branch of the dielectron candidates stored in //
	9	// filtered AOD files (AliAOD.Dielectron.root). It is possible to //
	10	// apply tighter cuts to candidates stored in the branch and do the //
	11	// matching with MC truth //
	12	// //
	13	// Contacts : G. Bruno / Giuseppe.Bruno@ba.infn.it, //
	14	// F. Fionda / Fiorella.Fionda@ba.infn.it, //
	15	// C. Di Giglio / Carmelo.Digiglio@ba.infn.it //
	16	////////////////////////////////////////////////////////////////////////////
	17
	18	#include "AliAnalysisTaskSE.h"
	19
	20	class TNtuple;
	21	class TH1F;
	22	class TH2F;
bc7a3220	23	class AliDielectronPair;
ba15fdfb	24
	25	class AliAnalysisTaskDielectronReadAODBranch : public AliAnalysisTaskSE
	26	{
	27	public:
	28	AliAnalysisTaskDielectronReadAODBranch();
	29	AliAnalysisTaskDielectronReadAODBranch(const char *name);
	30	virtual ~AliAnalysisTaskDielectronReadAODBranch();
	31
	32	// Implementation of interface methods
	33	virtual void UserCreateOutputObjects();
	34	virtual void Init();
	35	virtual void LocalInit() {Init();}
	36	virtual void UserExec(Option_t *option);
	37	virtual void Terminate(Option_t *option);
ba15fdfb	38
	39	//setters
	40	void SetPtLeg(Double_t cutPt){ fPtCut = cutPt;}
	41	void SetSpdFirstRequired(Bool_t spdfirst){fSpdFirstRequired = spdfirst;}
	42	void SetNclsTPC(Int_t nCls){fClsTPC = nCls;}
	43	void SetPairType(Int_t type){fPairType = type;}
	44	void SetPtJpsi(Double_t ptjpsi) {fPtJpsi = ptjpsi;}
	45	void SetInvMassSignalRegion(Double_t lowLimit, Double_t upLimit){fInvMassSignalLimits[0]=lowLimit; fInvMassSignalLimits[1]=upLimit;}
	46	void SetInvMassSidebandRegion(Double_t lowLimit, Double_t upLimit){fInvMassSideBandsLimits[0]=lowLimit; fInvMassSideBandsLimits[1]=upLimit;}
	47	void SetHasMC(Bool_t mcFlag) {fHasMC = mcFlag;}
	48
	49
	50	enum {
	51	kITSin=0x0001,kITSout=0x0002,kITSrefit=0x0004,kITSpid=0x0008,
	52	kTPCin=0x0010,kTPCout=0x0020,kTPCrefit=0x0040,kTPCpid=0x0080,
	53	kTRDin=0x0100,kTRDout=0x0200,kTRDrefit=0x0400,kTRDpid=0x0800,
	54	kTOFin=0x1000,kTOFout=0x2000,kTOFrefit=0x4000,kTOFpid=0x8000,
	55	kHMPIDout=0x10000,kHMPIDpid=0x20000
	56	};
	57
	58	private:
	59	AliAnalysisTaskDielectronReadAODBranch(const AliAnalysisTaskDielectronReadAODBranch &c);
	60	AliAnalysisTaskDielectronReadAODBranch& operator= (const AliAnalysisTaskDielectronReadAODBranch &c);
	61
	62	TList *fOutput; // output list with histo
	63	TNtuple *fNtupleJPSI; // ntupla to store invariant mass and psproper of candidates
	64	TH1F *fNentries; // number of filtered events
	65	TH1F *fInvMass; // invariant mass of candidates in branch after tighter cuts application
	66	TH1F *fInvMassNoCuts; // invariant mass of candidates in branch without cuts
	67	TH1F *fpsproperSignal; // psproper distribution in the mass signal region
	68	TH1F *fpsproperSidebands; // psproper distribution in the mass sidebands region
	69	TH1F *fpsproperAll; // whole psproper distribution
	70	TH1F *fpsproperUnder; // psproper distribution in the lower mass sidebands region
	71	TH1F *fpsproperUpper; // psproper distribution in the upper mass sidebands region
	72	TH2F *fLxyVsPtleg1; // psproper vs pt for leg1
	73	TH2F *fLxyVsPtleg2; // psproper vs pt for leg2
	74	TH2F *fLxyVsPt; // psproper vs pt jpsi
	75	TH2F *fMeeVsPt; // invariant mass vs pt jpsi
	76	TH2F *fMeeVsLxy; // invariant mass vs psproper
	77
	78	// QA plots
	79	TH1F *fprimvtxZ; // z coord of primary vertex
	80	TH1F *fsecvtxZ; // z coord of secondary vertex
	81	TH1F *fprimvtxX; // x coord of primary vertex
	82	TH1F *fsecvtxX; // x coord of secondary vertex
	83	TH1F *fprimvtxY; // y coord of primary vertex
	84	TH1F *fsecvtxY; // y coord of secondary vertex
	85	TH1F *fPt; // pT(J/psi) distribution
	86	TH1F *fPtLeg1; // pT(leg1) distribution. Warning: mixture of pos and neg tracks
	87	TH1F *fPtLeg2; // pT(leg2) distribution. Warning: mixture of pos and neg tracks
	88	TH2F *fdEdxP; // dEdx vs P for legs
	89	Bool_t fHasMC; // flag for read MC branch
	90	TObjArray *fobj; // TObjArray with stored reconstructed candidates
	91	TObjArray *fobjMC; // TObjArray with MC tracks
	92
	93	// cuts on candidates
	94	Double_t fPtCut; // ptLeg
	95	Bool_t fSpdFirstRequired; // spd any/first
	96	Int_t fClsTPC; // #clsTPC
	97	Int_t fPairType; // pair Type
	98	Double_t fPtJpsi; // pt jpsi
	99	Double_t *fInvMassSignalLimits; // invariant mass signal region to extract psproper distribution
	100	Double_t *fInvMassSideBandsLimits; // invariant mass sideband region to extract psproper distribution
	101
bc7a3220	102	Int_t fSecondary; // 1(0) to select secondary (prompt) jpsi
ba15fdfb	103	ClassDef(AliAnalysisTaskDielectronReadAODBranch,2); // AliAnalysisTaskSE for the MC association of heavy-flavour decay candidates
	104	};
	105
	106
	107	#endif