[u/mrichter/AliRoot.git] / PWG / FLOW / Tasks / AliAnalysisTaskPIDconfig.h

#ifndef ALIANALYSISTASKPIDconfig_H
#define ALIANALYSISTASKPIDconfig_H

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

/* $Id: AliAnalysisTaskPIDconfig.h */
// Author: Naghmeh Mohammadi, 10/07/2014

//==============================================================================
//
//
//
//
//==============================================================================

#include <TVectorDfwd.h>

#include "AliAnalysisTaskSE.h"
#include "AliPID.h"
#include "AliPIDResponse.h"
#include "AliCentrality.h"
#include "TCutG.h"


class AliESDEvent;
class AliAODEvent;
class AliESDTrack;
class AliAODTrack;
class AliPIDResponse;
class TList;
class AliVEvent;
class TH1F;
class TH2F;
class TH3F;
class TH3F;


class AliAnalysisTaskPIDconfig : public AliAnalysisTaskSE {
    
    
public:
    AliAnalysisTaskPIDconfig();
    AliAnalysisTaskPIDconfig(const char *name);
    virtual ~AliAnalysisTaskPIDconfig();
    
    virtual void UserCreateOutputObjects();
    virtual void UserExec(Option_t * /*option*/);
    
    void SetFilterBit(Double_t b){fFilterBit = b;}
    void SetCentralityPercentileMin(Int_t b){fCentralityPercentileMin = b;}
    void SetCentralityPercentileMax(Int_t b){fCentralityPercentileMax = b;}
    void SetCentralityEstimator(TString b){fCentralityEstimator = b;}
    void SetUseCentrality(Bool_t b=kTRUE){fUseCentrality = b;}
    void SetCentralityTrigger(Int_t b=AliVEvent::kMB){ fTriggerSelection = b;}
    void SetDCAxyCut(Int_t b){fDCAxyCut = b;}
    void SetDCAzCut(Int_t b){fDCAzCut = b;}
    void SetCutTPCmultiplicityOutliersAOD(Bool_t b){fCutTPCmultiplicityOutliersAOD = b;}
    void SetData2011(Bool_t b){fData2011 = b;}
    void CheckCentrality(AliVEvent *event,Bool_t &centralitypass); //to use only events with the correct centrality....
    void SetCuts(Bool_t b){fPIDcuts = b;}
    //void MultiplicityOutlierCut(AliVEvent *event,Bool_t &centralitypass,Int_t ntracks);
    void SetPIDcontoursList(TDirectory* b){fCutContourList = b;}
    //TGraph* GetPIDcontours(TString specie, Double_t Plow, Double_t Phigh,Int_t centMin, Int_t centMax){}
    void GetPIDContours();
    
protected:
    
    
private:
    AliVEvent             *fVevent;             //! event
    AliESDEvent           *fESD;                //! esd
    AliAODEvent           *fAOD;                //! aod
    AliPIDResponse        *fPIDResponse;        //! PID response Handler
    Int_t                  fTriggerSelection;   // trigger selection
    Int_t                  fCentralityPercentileMin;    // min centrality
    Int_t                  fCentralityPercentileMax;    // max cen
    Double_t               fFilterBit;                  // filterbit
    Double_t               fDCAxyCut;           // dca cut
    Double_t               fDCAzCut;            // dcz z
    Double_t               fLowPtPIDTPCnsigLow_Pion[6]; //nsig low cut Pion
    Double_t               fLowPtPIDTPCnsigHigh_Pion[6]; //nsig high cut Pion
    Double_t               fLowPtPIDTPCnsigLow_Kaon[6]; //nsig low cut Kaon
    Double_t               fLowPtPIDTPCnsigHigh_Kaon[6]; //nsig high cut Kaon
    Bool_t                 fData2011;           // 2011 data
    Bool_t                 fTriggerMB;          // minB trigger
    Bool_t                 fTriggerCentral;     // cen trigger
    Bool_t                 fUseCentrality;      // use centrality
    Bool_t                 fCutTPCmultiplicityOutliersAOD;      // do outlier cut
    Bool_t                 fPIDcuts;            // pid cuts
    TString                fCentralityEstimator;// cen estimator "V0M","TRK","TKL","ZDC","FMD"
    TFile                 *fContoursFile;       //! contours file
    TDirectory            *fCutContourList;     //! contour list
    TList                 *fListQA;             //! List of all lists
    TList                 *fListQAtpctof;       //! List with combined PID from TPC + TOF
    TList                 *fListQAInfo;         //! list q ainfo
    TH1F                  *fhistCentralityPass; //! cen histo   
    TH1F                  *fNoEvents;           //! event no    
    TH1F                  *fpVtxZ;              //! v vertex no
    TH2F                  *fhistDCABefore;      //! dca after hist
    TH2F                  *fhistDCAAfter;       //! another hist
    TH1F                  *fhistPhiDistBefore;  //! another hist
    TH1F                  *fhistPhiDistAfter;   //! another hist
    TH1F                  *fhistEtaDistBefore;  //! another hist
    TH1F                  *fhistEtaDistAfter;   //! another hist
    TH2F                  *fTPCvsGlobalMultBeforeOutliers;      //! another hist
    TH2F                  *fTPCvsGlobalMultAfterOutliers;       //! another hist
    TH2F                  *fTPCvsGlobalMultAfter;       //! another hist
    TH2F                  *fHistBetavsPTOFbeforePID;    //! another hist
    TH2F                  *fHistdEdxvsPTPCbeforePID;    //! another hist
    TH3F                  *fhistNsigmaP;        //! another hist
    TH2F                  *fhistTPCnSigmavsP;   //! another hist
    TH2F                  *fHistBetavsPTOFafterPID;     //! another hist
    TH2F                  *fHistdEdxvsPTPCafterPID;     //! another hist
    TH2F                  *fHistBetavsPTOFafterPID_2;     //! another hist
    TH2F                  *fHistdEdxvsPTPCafterPID_2;     //! another hist
    TH2F                  *fHistBetavsPTOFafterPIDTPCTOF; //! another hist
    TH2F                  *fHistdEdxvsPTPCafterPIDTPCTOF; //! another hist
    TH2F                  *fHistBetavsPTOFafterPIDTPConly; //! another hist
    TH2F                  *fHistdEdxvsPTPCafterPIDTPConly; //! another hist
    TH2F                  *fHistPion_BetavsPTOFafterPIDTPCTOF; //! another hist
    TH2F                  *fHistPion_dEdxvsPTPCafterPIDTPCTOF; //!another hist
    TH2F                  *fHistKaon_BetavsPTOFafterPIDTPCTOF; //! another hist
    TH2F                  *fHistKaon_dEdxvsPTPCafterPIDTPCTOF; //!another hist
    TH2F                  *fHistProton_BetavsPTOFafterPIDTPCTOF; //! another hist
    TH2F                  *fHistProton_dEdxvsPTPCafterPIDTPCTOF; //!another hist
    TH1F                  *fhistPionEtaDistAfter; //!another hist
    TH1F                  *fhistKaonEtaDistAfter; //!another hist
    TH1F                  *fhistProtonEtaDistAfter; //!another hist
    TCutG                 *fCutContour[150];    //! TCutG contours
    TGraph                *fCutGraph[150];      //! graphs
    
    
    //qa object initialisation
    void SetupTPCTOFqa();
    void SetupEventInfo();
    //
    
    AliAnalysisTaskPIDconfig(const AliAnalysisTaskPIDconfig &other);
    AliAnalysisTaskPIDconfig& operator=(const AliAnalysisTaskPIDconfig &other);
    
    ClassDef(AliAnalysisTaskPIDconfig,3)  // Task to properly set the PID response functions of all detectors
};

#endif
Commit	Line	Data
2c03c485	1	#ifndef ALIANALYSISTASKPIDconfig_H
	2	#define ALIANALYSISTASKPIDconfig_H
	3
	4	/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
	5	* See cxx source for full Copyright notice */
	6
	7	/* $Id: AliAnalysisTaskPIDconfig.h */
	8	// Author: Naghmeh Mohammadi, 10/07/2014
	9
	10	//==============================================================================
	11	//
	12	//
	13	//
	14	//
	15	//==============================================================================
	16
	17	#include <TVectorDfwd.h>
	18
	19	#include "AliAnalysisTaskSE.h"
7ff63eca	20	#include "AliPID.h"
2c03c485	21	#include "AliPIDResponse.h"
2c03c485	22	#include "AliCentrality.h"
6440143f	23	#include "TCutG.h"
2c03c485	24
	25
	26
	27	class AliESDEvent;
	28	class AliAODEvent;
	29	class AliESDTrack;
	30	class AliAODTrack;
	31	class AliPIDResponse;
	32	class TList;
	33	class AliVEvent;
	34	class TH1F;
	35	class TH2F;
7ff63eca	36	class TH3F;
2c03c485	37	class TH3F;
	38
	39
	40
	41	class AliAnalysisTaskPIDconfig : public AliAnalysisTaskSE {
7ff63eca	42
7ff63eca	43
2c03c485	44	public:
d802b6ea	45	AliAnalysisTaskPIDconfig();
	46	AliAnalysisTaskPIDconfig(const char *name);
	47	virtual ~AliAnalysisTaskPIDconfig();
7ff63eca	48
d802b6ea	49	virtual void UserCreateOutputObjects();
d802b6ea	50	virtual void UserExec(Option_t * /option/);
7ff63eca	51
d802b6ea	52	void SetFilterBit(Double_t b){fFilterBit = b;}
	53	void SetCentralityPercentileMin(Int_t b){fCentralityPercentileMin = b;}
	54	void SetCentralityPercentileMax(Int_t b){fCentralityPercentileMax = b;}
	55	void SetCentralityEstimator(TString b){fCentralityEstimator = b;}
	56	void SetUseCentrality(Bool_t b=kTRUE){fUseCentrality = b;}
	57	void SetCentralityTrigger(Int_t b=AliVEvent::kMB){ fTriggerSelection = b;}
	58	void SetDCAxyCut(Int_t b){fDCAxyCut = b;}
	59	void SetDCAzCut(Int_t b){fDCAzCut = b;}
	60	void SetCutTPCmultiplicityOutliersAOD(Bool_t b){fCutTPCmultiplicityOutliersAOD = b;}
	61	void SetData2011(Bool_t b){fData2011 = b;}
	62	void CheckCentrality(AliVEvent *event,Bool_t &centralitypass); //to use only events with the correct centrality....
	63	void SetCuts(Bool_t b){fPIDcuts = b;}
7ff63eca	64	//void MultiplicityOutlierCut(AliVEvent *event,Bool_t &centralitypass,Int_t ntracks);
cf83f565	65	void SetPIDcontoursList(TDirectory* b){fCutContourList = b;}
7ff63eca	66	//TGraph* GetPIDcontours(TString specie, Double_t Plow, Double_t Phigh,Int_t centMin, Int_t centMax){}
6440143f	67	void GetPIDContours();
7ff63eca	68
	69	protected:
	70
	71
	72
2c03c485	73	private:
cf83f565	74	AliVEvent *fVevent; //! event
	75	AliESDEvent *fESD; //! esd
	76	AliAODEvent *fAOD; //! aod
	77	AliPIDResponse *fPIDResponse; //! PID response Handler
	78	Int_t fTriggerSelection; // trigger selection
	79	Int_t fCentralityPercentileMin; // min centrality
	80	Int_t fCentralityPercentileMax; // max cen
	81	Double_t fFilterBit; // filterbit
	82	Double_t fDCAxyCut; // dca cut
	83	Double_t fDCAzCut; // dcz z
a4ce2f5e	84	Double_t fLowPtPIDTPCnsigLow_Pion[6]; //nsig low cut Pion
	85	Double_t fLowPtPIDTPCnsigHigh_Pion[6]; //nsig high cut Pion
	86	Double_t fLowPtPIDTPCnsigLow_Kaon[6]; //nsig low cut Kaon
	87	Double_t fLowPtPIDTPCnsigHigh_Kaon[6]; //nsig high cut Kaon
cf83f565	88	Bool_t fData2011; // 2011 data
	89	Bool_t fTriggerMB; // minB trigger
	90	Bool_t fTriggerCentral; // cen trigger
	91	Bool_t fUseCentrality; // use centrality
	92	Bool_t fCutTPCmultiplicityOutliersAOD; // do outlier cut
	93	Bool_t fPIDcuts; // pid cuts
	94	TString fCentralityEstimator;// cen estimator "V0M","TRK","TKL","ZDC","FMD"
	95	TFile *fContoursFile; //! contours file
	96	TDirectory *fCutContourList; //! contour list
	97	TList *fListQA; //! List of all lists
	98	TList *fListQAtpctof; //! List with combined PID from TPC + TOF
	99	TList *fListQAInfo; //! list q ainfo
	100	TH1F *fhistCentralityPass; //! cen histo
	101	TH1F *fNoEvents; //! event no
	102	TH1F *fpVtxZ; //! v vertex no
	103	TH2F *fhistDCABefore; //! dca after hist
	104	TH2F *fhistDCAAfter; //! another hist
	105	TH1F *fhistPhiDistBefore; //! another hist
	106	TH1F *fhistPhiDistAfter; //! another hist
	107	TH1F *fhistEtaDistBefore; //! another hist
	108	TH1F *fhistEtaDistAfter; //! another hist
	109	TH2F *fTPCvsGlobalMultBeforeOutliers; //! another hist
	110	TH2F *fTPCvsGlobalMultAfterOutliers; //! another hist
	111	TH2F *fTPCvsGlobalMultAfter; //! another hist
	112	TH2F *fHistBetavsPTOFbeforePID; //! another hist
	113	TH2F *fHistdEdxvsPTPCbeforePID; //! another hist
	114	TH3F *fhistNsigmaP; //! another hist
e6baed4c	115	TH2F *fhistTPCnSigmavsP; //! another hist
cf83f565	116	TH2F *fHistBetavsPTOFafterPID; //! another hist
cf83f565	117	TH2F *fHistdEdxvsPTPCafterPID; //! another hist
a4ce2f5e	118	TH2F *fHistBetavsPTOFafterPID_2; //! another hist
a4ce2f5e	119	TH2F *fHistdEdxvsPTPCafterPID_2; //! another hist
cea65e98	120	TH2F *fHistBetavsPTOFafterPIDTPCTOF; //! another hist
	121	TH2F *fHistdEdxvsPTPCafterPIDTPCTOF; //! another hist
	122	TH2F *fHistBetavsPTOFafterPIDTPConly; //! another hist
	123	TH2F *fHistdEdxvsPTPCafterPIDTPConly; //! another hist
74fe90ab	124	TH2F *fHistPion_BetavsPTOFafterPIDTPCTOF; //! another hist
	125	TH2F *fHistPion_dEdxvsPTPCafterPIDTPCTOF; //!another hist
	126	TH2F *fHistKaon_BetavsPTOFafterPIDTPCTOF; //! another hist
	127	TH2F *fHistKaon_dEdxvsPTPCafterPIDTPCTOF; //!another hist
	128	TH2F *fHistProton_BetavsPTOFafterPIDTPCTOF; //! another hist
	129	TH2F *fHistProton_dEdxvsPTPCafterPIDTPCTOF; //!another hist
	130	TH1F *fhistPionEtaDistAfter; //!another hist
	131	TH1F *fhistKaonEtaDistAfter; //!another hist
	132	TH1F *fhistProtonEtaDistAfter; //!another hist
	133	TCutG *fCutContour[150]; //! TCutG contours
	134	TGraph *fCutGraph[150]; //! graphs
7ff63eca	135
	136
	137	//qa object initialisation
	138	void SetupTPCTOFqa();
	139	void SetupEventInfo();
	140	//
	141
	142	AliAnalysisTaskPIDconfig(const AliAnalysisTaskPIDconfig &other);
	143	AliAnalysisTaskPIDconfig& operator=(const AliAnalysisTaskPIDconfig &other);
	144
cf83f565	145	ClassDef(AliAnalysisTaskPIDconfig,3) // Task to properly set the PID response functions of all detectors
2c03c485	146	};
	147
	148	#endif