[u/mrichter/AliRoot.git] / PWGHF / hfe / AliAnalysisTaskFlowTPCEMCalQCSP.h

#ifndef ALIANALYSISTASKFLOWTPCEMCALQCSP_H
#define ALIANALYSISTASKFLOWTPCEMCALQCSP_H

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

////////////////////////////////////////////////////////////////////////
//                                                                    //
//  Task for Heavy Flavour Electron Flow TPC plus EMCal               //
//                                                                    //
//  Author: Andrea Dubla (Utrecht University)                         //
//                                                                    //
////////////////////////////////////////////////////////////////////////

class THnSparse;
class TH2F;
class TLorentzVector;

class AliEMCALTrack;
class AliMagF;
class AliESDEvent;
class AliAODEvent;
class AliEMCALGeometry;
class AliEMCALRecoUtils;
class AliAnalysisFilter;
class AliESDtrackCuts;
class AliESDtrack;
class AliAODtrack;
class AliHFEcontainer;
class AliHFEcuts;
class AliHFEpid;
class AliHFEpidQAmanager;
class AliCFManager;
class AliFlowTrackCuts;
class AliFlowTrack;
class AliFlowEvent;
class AliFlowCandidateTrack;
class AliFlowEventSimple;
class AliCentrality;
class AliSelectNonHFE;

#include "AliAnalysisTaskSE.h"

class AliAnalysisTaskFlowTPCEMCalQCSP : public AliAnalysisTaskSE {
    
public:
    AliAnalysisTaskFlowTPCEMCalQCSP();
    AliAnalysisTaskFlowTPCEMCalQCSP(const char *name);
    virtual ~AliAnalysisTaskFlowTPCEMCalQCSP();
    
    void                                 SetEnableDebugMode() {fDebug = kTRUE; };
    void                                 SetCentralityParameters(Double_t CentralityMin, Double_t CentralityMax, const char* CentralityMethod); //select centrality
    void                                 CheckCentrality(AliAODEvent *event,Bool_t &centralitypass); //to use only events with the correct centrality....
    void                                 SelectPhotonicElectron(Int_t itrack,const AliAODTrack *track, Double_t eopinc, Double_t evPlAnglV0, Bool_t &fFlagPhotonicElec);
    void                                 SetInvariantMassCut(Double_t invmass) {fInvmassCut = invmass;};
    void                                 SetpTCuttrack(Double_t ptcut) {fpTCut = ptcut;};
    void                                 SetTrigger(Int_t trig) {fTrigger = trig;};
    void                                 SetAssoTPCCluster(Int_t tpc_clust) {fAssoTPCCluster = tpc_clust;};
    void                                 SetMultCorrelationCut(Bool_t multcut) {fMultCut = multcut;};
    void                                 SetAssoITSRefit(Bool_t itsref) {fAssoITSRefit = itsref;};
    void                                 SetFlowSideBands(Bool_t sidebandsflow){fSideBandsFlow = sidebandsflow;}
    void                                 SelectPhotonicElectronMethod(Bool_t dca){fDCA = dca;}
    void                                 Setphiminuspsi(Bool_t phipsi){fPhiminusPsi = phipsi;}
    void                                 SetPurity(Bool_t Purityel){fpurity = Purityel;}
    template <typename T> void           PlotVZeroMultiplcities(const T* event) const;
    template <typename T> void           SetNullCuts(T* aod);
    void                                 PrepareFlowEvent(Int_t iMulti, AliFlowEvent *FlowEv) const;
    virtual void                         UserCreateOutputObjects();
    virtual void                         UserExec(Option_t *option);
    virtual void                         Terminate(Option_t *);
    void                                 SetRPCuts(AliFlowTrackCuts *cutsRP) { fCutsRP = cutsRP; }
    void                                 SetHFECuts(AliHFEcuts * const cuts) { fCuts = cuts; };
    void                                 SetIDCuts(Double_t minTPC, Double_t maxTPC, Double_t minEovP, Double_t maxEovP, Double_t minM20, Double_t maxM20, Double_t minM02, Double_t maxM02, Double_t Dispersion);
    void                                 SetOpeningAngleflag(Bool_t opang){fOP_angle = opang;};
    void                                 SetOpeningAngleCut(Double_t opanglecut) {fOpeningAngleCut = opanglecut;};
    void                                 SetHistoForCentralityFlattening(TH1F *h,Double_t minCentr,Double_t maxCentr,Double_t centrRef=0.,Int_t switchTRand=0);
    Bool_t                               IsEventSelectedForCentrFlattening(Float_t centvalue);
    
    
    
    
    
    AliHFEpid *GetPID() const { return fPID; };
    
private:
    
    Bool_t ProcessCutStep(Int_t cutStep, AliVParticle *track);
    
    Bool_t               fDebug; //! enable debug mode
    AliAODEvent          *fAOD;                     //AOD object
    AliEMCALGeometry     *fGeom;                // emcal geometry
    TList                *fOutputList;          //output list
    AliHFEcuts           *fCuts;                 //Cut Collection
    Bool_t               fIdentifiedAsOutInz;    //Out Of Range in z
    Bool_t               fPassTheEventCut;       //Pass The Event Cut
    AliCFManager         *fCFM;                  //!Correction Framework Manager
    AliHFEpid            *fPID;                  //PID
    AliHFEpidQAmanager   *fPIDqa;               //! PID QA manager
    AliFlowTrackCuts     *fCutsRP; // track cuts for reference particles
    AliFlowTrackCuts     *fNullCuts; // dummy cuts for flow event tracks
    AliFlowEvent         *fFlowEvent; //! flow events Inclusive e
    const char           *fkCentralityMethod; // method used to determine centrality (V0 by default)
    Double_t             fCentrality; // event centrality for QA
    Double_t             fCentralityMin; // lower bound of cenrality bin
    Double_t             fCentralityMax; // upper bound of centrality bin
    Double_t             fInvmassCut;           //invariant mass cut value
    Double_t             fpTCut;    //pt cut value
    Int_t                fTrigger;          //invariant mass cut value
    TH1F                 *fPhi; //! QA plot of azimuthal distribution of tracks used for event plane estimation
    TH1F                 *fEta; //! QA plot of eta distribution of tracks used for event plane estimation
    TH1F                 *fVZEROA; //! QA plot vzeroa multiplicity (all tracks in event)
    TH1F                 *fVZEROC; //! QA plot vzeroc multiplicity (all tracks in event)
    TH1F                 *fTPCM; //! QA plot TPC multiplicity (tracks used for event plane estimation)
    TH1F                 *fNoEvents;            //!no of events
    TH2F                 *fTrkEovPBef;          //!track E/p before HFE pid
    //   TH2F                           *fdEdxBef;         //track dEdx vs p before HFE pid
    TH1F                 *fInclusiveElecPt; //! Inclusive elec pt
    TH2F                 *fTPCnsigma;               //!TPC n sigma vs p
    TH2F                 *fTPCnsigmaAft;                //!TPC n sigma vs p after HFE pid
    TH1F                 *fCentralityPass; // ! QA histogram of events that pass centrality cut
    TH1F                 *fCentralityNoPass; //! QA histogram of events that do not pass centrality cut
    TH1F                 *fInvmassLS1; //!LS Invmass for all rec par
    TH1F                 *fInvmassULS1;//!ULS Invmass for all rec par
    TH1F                 *fPhotoElecPt;             //!photonic elec pt
    TH1F                 *fSemiInclElecPt;  //!Semi inclusive ele pt
    TH1F                 *fULSElecPt; //!ULS elec Pt
    TH1F                 *fLSElecPt;//! LS elec pt
    Double_t             fminTPC;  //ID cuts tpc
    Double_t             fmaxTPC;  //ID cuts tpc
    Double_t             fminEovP;  //ID cuts eovp
    Double_t             fmaxEovP;//ID cuts eovp
    Double_t             fminM20;//ID cuts SS
    Double_t             fmaxM20;//ID cuts SS
    Double_t             fminM02;//ID cuts SS
    Double_t             fmaxM02;//ID cuts SS
    Double_t             fDispersion;//ID cuts SS
    TH2F                 *fMultCorAfterCuts; //! QA profile global and tpc multiplicity after outlier cut
    TH2F                 *fMultvsCentr; //! QA profile of centralty vs multiplicity
    TProfile             *fSubEventDPhiv2;//!lll
    TH1D                 *EPVzA;//!v0aep
    TH1D                 *EPVzC;//!v0cep
    TH1D                 *EPTPC;//!tpcep
    THnSparseF           *fV2Phi;//! v2 analysis of EP-V0
    THnSparseD           *fSparseElectronHadron;//! Trk matching sparse for v1 clusterizer
    TH1D                 *fvertex;//!huge ThNsparse
    TH2F                 *fMultCorBeforeCuts; //! QA profile global and tpc multiplicity after outlier cut
    Bool_t               fSideBandsFlow;//flow from side bands for contamination
    Bool_t               fPhiminusPsi;//flow from phi minus psi method
    AliFlowEvent         *fFlowEventCont; // flow events for elect Contamination
    Bool_t               fpurity; //for purity evaluation
    THnSparseD           *fSparseElectronpurity;//! Trk matching sparse for v1 clusterizer
    TH1F                 *fOpeningAngleLS;  //!opening angle for LS pairs
    TH1F                 *fOpeningAngleULS; //!opening angle for ULS pairs
    AliSelectNonHFE      *fNonHFE;//new elienos stuff
    Bool_t               fDCA;//selection PHelectron
    Double_t             fOpeningAngleCut; //openingAngle cut value
    Bool_t               fOP_angle; //to shitch on and off the op_angle cut
    Int_t                fAssoTPCCluster;//asso tpc cluster
    Bool_t               fAssoITSRefit;//asso its refit
    Bool_t               fMultCut;//for mult correlationcut
    TH2F                 *fMultCorAfterCentrBeforeCuts; //! QA profile global and tpc multiplicity after outlier cut
    TH2F                 *fMultCorAfterVZTRKComp;//!after cent comp
    TH1F                 *fCentralityBeforePileup;//!cent chneck
    TH1F                 *fCentralityAfterVZTRK;//!cent chneck2
    TH1F                 *fCentralityAfterCorrCut;//!cent chneck2
    TH2F                 *fMultCorAfterCorrCut;//!after cent comp
    TH1D                 *EPVz;//!v0cep
    TH1D                 *EPTPCp;//!tpcep
    TH1D                 *EPTPCn;//!tpcep
    TProfile             *fSubEventDPhiv2new;
    THnSparseF           *fV2Phivzerotot;//! v2 analysis of EP-V0
    TH1F                 *fHistCentrDistr;//-> isto for Centr Flat
    TH1F                 *fCentralityNoPassForFlattening; //! QA histogram of events that do not pass centrality cut for flattening
    TH1F                 *fInvmassLS1highpt; //!LS Invmass for all rec par high pt
    TH1F                 *fInvmassULS1highpt;//!ULS Invmass for all rec par high pt
    THnSparseF           *fSparsephipsiULS;//!ssss
    THnSparseF           *fSparsephipsiLS;//!ssss
    THnSparseF           *fSparseMassULS;//!ssss
    THnSparseF           *fSparseMassLS;//!ssssss
    
    
    AliAnalysisTaskFlowTPCEMCalQCSP(const AliAnalysisTaskFlowTPCEMCalQCSP&); // not implemented
    AliAnalysisTaskFlowTPCEMCalQCSP& operator=(const AliAnalysisTaskFlowTPCEMCalQCSP&); // not implemented
    
    ClassDef(AliAnalysisTaskFlowTPCEMCalQCSP, 2); //!example of analysis
};

#endif
Commit	Line	Data
040cc6fa	1	#ifndef ALIANALYSISTASKFLOWTPCEMCALQCSP_H
040cc6fa	2	#define ALIANALYSISTASKFLOWTPCEMCALQCSP_H
914042c2	3
	4	/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	5	* See cxx source for full Copyright notice */
	6
	7	////////////////////////////////////////////////////////////////////////
	8	// //
9d63a2f6	9	// Task for Heavy Flavour Electron Flow TPC plus EMCal //
914042c2	10	// //
	11	// Author: Andrea Dubla (Utrecht University) //
	12	// //
	13	////////////////////////////////////////////////////////////////////////
	14
	15	class THnSparse;
	16	class TH2F;
	17	class TLorentzVector;
	18
	19	class AliEMCALTrack;
	20	class AliMagF;
	21	class AliESDEvent;
	22	class AliAODEvent;
	23	class AliEMCALGeometry;
	24	class AliEMCALRecoUtils;
	25	class AliAnalysisFilter;
	26	class AliESDtrackCuts;
	27	class AliESDtrack;
	28	class AliAODtrack;
	29	class AliHFEcontainer;
	30	class AliHFEcuts;
	31	class AliHFEpid;
	32	class AliHFEpidQAmanager;
	33	class AliCFManager;
	34	class AliFlowTrackCuts;
	35	class AliFlowTrack;
	36	class AliFlowEvent;
	37	class AliFlowCandidateTrack;
	38	class AliFlowEventSimple;
	39	class AliCentrality;
20791315	40	class AliSelectNonHFE;
20791315	41
914042c2	42	#include "AliAnalysisTaskSE.h"
914042c2	43
a70c9e97	44	class AliAnalysisTaskFlowTPCEMCalQCSP : public AliAnalysisTaskSE {
96387d94	45
96387d94	46	public:
a70c9e97	47	AliAnalysisTaskFlowTPCEMCalQCSP();
	48	AliAnalysisTaskFlowTPCEMCalQCSP(const char *name);
	49	virtual ~AliAnalysisTaskFlowTPCEMCalQCSP();
96387d94	50
914042c2	51	void SetEnableDebugMode() {fDebug = kTRUE; };
	52	void SetCentralityParameters(Double_t CentralityMin, Double_t CentralityMax, const char* CentralityMethod); //select centrality
	53	void CheckCentrality(AliAODEvent *event,Bool_t &centralitypass); //to use only events with the correct centrality....
7fcd6b0f	54	void SelectPhotonicElectron(Int_t itrack,const AliAODTrack *track, Double_t eopinc, Double_t evPlAnglV0, Bool_t &fFlagPhotonicElec);
914042c2	55	void SetInvariantMassCut(Double_t invmass) {fInvmassCut = invmass;};
040cc6fa	56	void SetpTCuttrack(Double_t ptcut) {fpTCut = ptcut;};
914042c2	57	void SetTrigger(Int_t trig) {fTrigger = trig;};
8e8d9d2a	58	void SetAssoTPCCluster(Int_t tpc_clust) {fAssoTPCCluster = tpc_clust;};
ff36a82c	59	void SetMultCorrelationCut(Bool_t multcut) {fMultCut = multcut;};
8e8d9d2a	60	void SetAssoITSRefit(Bool_t itsref) {fAssoITSRefit = itsref;};
9d63a2f6	61	void SetFlowSideBands(Bool_t sidebandsflow){fSideBandsFlow = sidebandsflow;}
20791315	62	void SelectPhotonicElectronMethod(Bool_t dca){fDCA = dca;}
9d63a2f6	63	void Setphiminuspsi(Bool_t phipsi){fPhiminusPsi = phipsi;}
9d63a2f6	64	void SetPurity(Bool_t Purityel){fpurity = Purityel;}
914042c2	65	template <typename T> void PlotVZeroMultiplcities(const T* event) const;
	66	template <typename T> void SetNullCuts(T* aod);
	67	void PrepareFlowEvent(Int_t iMulti, AliFlowEvent *FlowEv) const;
	68	virtual void UserCreateOutputObjects();
	69	virtual void UserExec(Option_t *option);
	70	virtual void Terminate(Option_t *);
	71	void SetRPCuts(AliFlowTrackCuts *cutsRP) { fCutsRP = cutsRP; }
	72	void SetHFECuts(AliHFEcuts * const cuts) { fCuts = cuts; };
	73	void SetIDCuts(Double_t minTPC, Double_t maxTPC, Double_t minEovP, Double_t maxEovP, Double_t minM20, Double_t maxM20, Double_t minM02, Double_t maxM02, Double_t Dispersion);
20791315	74	void SetOpeningAngleflag(Bool_t opang){fOP_angle = opang;};
20791315	75	void SetOpeningAngleCut(Double_t opanglecut) {fOpeningAngleCut = opanglecut;};
9f508565	76	void SetHistoForCentralityFlattening(TH1F *h,Double_t minCentr,Double_t maxCentr,Double_t centrRef=0.,Int_t switchTRand=0);
9f508565	77	Bool_t IsEventSelectedForCentrFlattening(Float_t centvalue);
96387d94	78
acd6f996	79
96387d94	80
	81
	82
914042c2	83	AliHFEpid *GetPID() const { return fPID; };
96387d94	84
	85	private:
	86
914042c2	87	Bool_t ProcessCutStep(Int_t cutStep, AliVParticle *track);
96387d94	88
914042c2	89	Bool_t fDebug; //! enable debug mode
96387d94	90	AliAODEvent *fAOD; //AOD object
	91	AliEMCALGeometry *fGeom; // emcal geometry
	92	TList *fOutputList; //output list
	93	AliHFEcuts *fCuts; //Cut Collection
	94	Bool_t fIdentifiedAsOutInz; //Out Of Range in z
	95	Bool_t fPassTheEventCut; //Pass The Event Cut
	96	AliCFManager *fCFM; //!Correction Framework Manager
	97	AliHFEpid *fPID; //PID
	98	AliHFEpidQAmanager *fPIDqa; //! PID QA manager
914042c2	99	AliFlowTrackCuts *fCutsRP; // track cuts for reference particles
914042c2	100	AliFlowTrackCuts *fNullCuts; // dummy cuts for flow event tracks
041e6d56	101	AliFlowEvent *fFlowEvent; //! flow events Inclusive e
914042c2	102	const char *fkCentralityMethod; // method used to determine centrality (V0 by default)
	103	Double_t fCentrality; // event centrality for QA
	104	Double_t fCentralityMin; // lower bound of cenrality bin
	105	Double_t fCentralityMax; // upper bound of centrality bin
96387d94	106	Double_t fInvmassCut; //invariant mass cut value
040cc6fa	107	Double_t fpTCut; //pt cut value
96387d94	108	Int_t fTrigger; //invariant mass cut value
914042c2	109	TH1F *fPhi; //! QA plot of azimuthal distribution of tracks used for event plane estimation
	110	TH1F *fEta; //! QA plot of eta distribution of tracks used for event plane estimation
	111	TH1F *fVZEROA; //! QA plot vzeroa multiplicity (all tracks in event)
	112	TH1F *fVZEROC; //! QA plot vzeroc multiplicity (all tracks in event)
	113	TH1F *fTPCM; //! QA plot TPC multiplicity (tracks used for event plane estimation)
03b64410	114	TH1F *fNoEvents; //!no of events
03b64410	115	TH2F *fTrkEovPBef; //!track E/p before HFE pid
96387d94	116	// TH2F *fdEdxBef; //track dEdx vs p before HFE pid
03b64410	117	TH1F *fInclusiveElecPt; //! Inclusive elec pt
	118	TH2F *fTPCnsigma; //!TPC n sigma vs p
	119	TH2F *fTPCnsigmaAft; //!TPC n sigma vs p after HFE pid
914042c2	120	TH1F *fCentralityPass; // ! QA histogram of events that pass centrality cut
914042c2	121	TH1F *fCentralityNoPass; //! QA histogram of events that do not pass centrality cut
03b64410	122	TH1F *fInvmassLS1; //!LS Invmass for all rec par
	123	TH1F *fInvmassULS1;//!ULS Invmass for all rec par
	124	TH1F *fPhotoElecPt; //!photonic elec pt
	125	TH1F *fSemiInclElecPt; //!Semi inclusive ele pt
	126	TH1F *fULSElecPt; //!ULS elec Pt
	127	TH1F *fLSElecPt;//! LS elec pt
96387d94	128	Double_t fminTPC; //ID cuts tpc
	129	Double_t fmaxTPC; //ID cuts tpc
	130	Double_t fminEovP; //ID cuts eovp
	131	Double_t fmaxEovP;//ID cuts eovp
	132	Double_t fminM20;//ID cuts SS
	133	Double_t fmaxM20;//ID cuts SS
	134	Double_t fminM02;//ID cuts SS
	135	Double_t fmaxM02;//ID cuts SS
	136	Double_t fDispersion;//ID cuts SS
9d63a2f6	137	TH2F *fMultCorAfterCuts; //! QA profile global and tpc multiplicity after outlier cut
9d63a2f6	138	TH2F *fMultvsCentr; //! QA profile of centralty vs multiplicity
03b64410	139	TProfile *fSubEventDPhiv2;//!lll
	140	TH1D *EPVzA;//!v0aep
	141	TH1D *EPVzC;//!v0cep
	142	TH1D *EPTPC;//!tpcep
9d63a2f6	143	THnSparseF *fV2Phi;//! v2 analysis of EP-V0
9d63a2f6	144	THnSparseD *fSparseElectronHadron;//! Trk matching sparse for v1 clusterizer
03b64410	145	TH1D *fvertex;//!huge ThNsparse
9d63a2f6	146	TH2F *fMultCorBeforeCuts; //! QA profile global and tpc multiplicity after outlier cut
96387d94	147	Bool_t fSideBandsFlow;//flow from side bands for contamination
96387d94	148	Bool_t fPhiminusPsi;//flow from phi minus psi method
03b64410	149	AliFlowEvent *fFlowEventCont; // flow events for elect Contamination
96387d94	150	Bool_t fpurity; //for purity evaluation
9d63a2f6	151	THnSparseD *fSparseElectronpurity;//! Trk matching sparse for v1 clusterizer
03b64410	152	TH1F *fOpeningAngleLS; //!opening angle for LS pairs
03b64410	153	TH1F *fOpeningAngleULS; //!opening angle for ULS pairs
20791315	154	AliSelectNonHFE *fNonHFE;//new elienos stuff
96387d94	155	Bool_t fDCA;//selection PHelectron
	156	Double_t fOpeningAngleCut; //openingAngle cut value
	157	Bool_t fOP_angle; //to shitch on and off the op_angle cut
	158	Int_t fAssoTPCCluster;//asso tpc cluster
9f508565	159	Bool_t fAssoITSRefit;//asso its refit
9f508565	160	Bool_t fMultCut;//for mult correlationcut
ff36a82c	161	TH2F *fMultCorAfterCentrBeforeCuts; //! QA profile global and tpc multiplicity after outlier cut
	162	TH2F *fMultCorAfterVZTRKComp;//!after cent comp
	163	TH1F *fCentralityBeforePileup;//!cent chneck
	164	TH1F *fCentralityAfterVZTRK;//!cent chneck2
acd6f996	165	TH1F *fCentralityAfterCorrCut;//!cent chneck2
acd6f996	166	TH2F *fMultCorAfterCorrCut;//!after cent comp
03b64410	167	TH1D *EPVz;//!v0cep
03b64410	168	TH1D *EPTPCp;//!tpcep
9f508565	169	TH1D *EPTPCn;//!tpcep
	170	TProfile *fSubEventDPhiv2new;
	171	THnSparseF *fV2Phivzerotot;//! v2 analysis of EP-V0
03b64410	172	TH1F *fHistCentrDistr;//-> isto for Centr Flat
9f508565	173	TH1F *fCentralityNoPassForFlattening; //! QA histogram of events that do not pass centrality cut for flattening
03b64410	174	TH1F *fInvmassLS1highpt; //!LS Invmass for all rec par high pt
03b64410	175	TH1F *fInvmassULS1highpt;//!ULS Invmass for all rec par high pt
7fcd6b0f	176	THnSparseF *fSparsephipsiULS;//!ssss
	177	THnSparseF *fSparsephipsiLS;//!ssss
	178	THnSparseF *fSparseMassULS;//!ssss
	179	THnSparseF *fSparseMassLS;//!ssssss
96387d94	180
20791315	181
a70c9e97	182	AliAnalysisTaskFlowTPCEMCalQCSP(const AliAnalysisTaskFlowTPCEMCalQCSP&); // not implemented
a70c9e97	183	AliAnalysisTaskFlowTPCEMCalQCSP& operator=(const AliAnalysisTaskFlowTPCEMCalQCSP&); // not implemented
96387d94	184
a70c9e97	185	ClassDef(AliAnalysisTaskFlowTPCEMCalQCSP, 2); //!example of analysis
914042c2	186	};
	187
	188	#endif
	189