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

#ifndef ALIANALYSISTASKVNV0_H
#define ALIANALYSISTASKVNV0_H

// ROOT includes
#include <TObject.h>
#include <TClonesArray.h>
#include "TTree.h"
#include <TList.h>
#include <TProfile.h>

// AliRoot includes
#include <AliAnalysisTaskSE.h>
#include <AliAODEvent.h>
#include "AliFlowBayesianPID.h"
#include "AliFlowVZEROResults.h"
#include "AliFlowVZEROQA.h"

class TH2F;
class AliESDtrackCuts;
class TProfile2D;

class AliAnalysisTaskVnV0 : public AliAnalysisTaskSE {
 public:
  AliAnalysisTaskVnV0();
  AliAnalysisTaskVnV0(const char *name);

  virtual ~AliAnalysisTaskVnV0();

  virtual void   UserCreateOutputObjects();
  virtual void   UserExec(Option_t *option);
  virtual void   Terminate(Option_t *); 
  
  Double_t GetVtxCut() { return fVtxCut; }   
  Double_t GetEtaCut() { return fEtaCut; }     
  Double_t GetMinPt() { return fMinPt; }   

  virtual void  SetVtxCut(Double_t vtxCut){fVtxCut = vtxCut;}
  virtual void  SetEtaCut(Double_t etaCut){fEtaCut = etaCut;}
  virtual void  SetMinPt(Double_t value) {fMinPt = value;}   
  virtual void  SetMinDistV0(Double_t value){fMinDistV0=value;}
  virtual void  SetMaxDistV0(Double_t value){fMaxDistV0=value;}
  virtual void SetV2(Bool_t val){fV2 = val;};
  virtual void SetV3(Bool_t val){fV3 = val;};

  virtual void SetMC(Bool_t flag = kTRUE){fIsMC = flag;};
  virtual void SetQA(Bool_t flag = kTRUE){fQAsw = flag;};

  void OpenInfoCalbration(Int_t run);

  void SetFillDCAinfo(Bool_t flag=kTRUE){fFillDCA = flag;};

  void SetModulationDEDx(Bool_t flag=kTRUE){fModulationDEDx=flag;};

  void SetAfter2011(Bool_t stat=kTRUE){fIsAfter2011=stat;};
  Bool_t GetAfter2011() const {return fIsAfter2011;};

  void SetTPCclusterN(Int_t ncl){fNcluster=ncl;};
  static Bool_t IsPsiComputed(){return fgIsPsiComputed;};
  static Float_t GetPsi2V0A(){return fgPsi2v0a;};
  static Float_t GetPsi2V0C(){return fgPsi2v0c;};
  static Float_t GetPsi2TPC(){return fgPsi2tpc;};
  static Float_t GetPsi3V0A(){return fgPsi3v0a;};
  static Float_t GetPsi3V0C(){return fgPsi3v0c;};
  static Float_t GetPsi3TPC(){return fgPsi3tpc;};
  static Float_t GetPsi2V0AMC(){return fgPsi2v0aMC;};
  static Float_t GetPsi2V0CMC(){return fgPsi2v0cMC;};
  static Float_t GetPsi2TPCMC(){return fgPsi2tpcMC;};
  static Float_t GetPsi3V0AMC(){return fgPsi3v0aMC;};
  static Float_t GetPsi3V0CMC(){return fgPsi3v0cMC;};
  static Float_t GetPsi3TPCMC(){return fgPsi3tpcMC;};

 private:
  AliAnalysisTaskVnV0(const AliAnalysisTaskVnV0 &old); 
  AliAnalysisTaskVnV0& operator=(const AliAnalysisTaskVnV0 &source); 

  Int_t PassesAODCuts(AliAODv0 *myV0, AliAODEvent *tAOD,Int_t specie);

  static Bool_t fgIsPsiComputed; // flag which return if event was processed
  static Float_t fgPsi2v0a,fgPsi2v0c,fgPsi2tpc; // current Psi2
  static Float_t fgPsi3v0a,fgPsi3v0c,fgPsi3tpc; // current Psi3
  static Float_t fgPsi2v0aMC,fgPsi2v0cMC,fgPsi2tpcMC; // current Psi2
  static Float_t fgPsi3v0aMC,fgPsi3v0cMC,fgPsi3tpcMC; // current Psi3

  virtual Float_t GetVertex(AliAODEvent* aod) const;
  virtual void Analyze(AliAODEvent* aodEvent, Float_t v0Centr); 
  virtual void SelectK0s();

  Double_t     fVtxCut;             // Vtx cut on z position in cm
  Double_t     fEtaCut;             // Eta cut used to select particles
  Double_t     fMinPt;              // Min pt - for histogram limits
  Double_t     fMinDistV0;          // Minimal distance for V0s
  Double_t     fMaxDistV0;          // Maximal distance for V0s

  Bool_t fV2; // switch to set the armonics
  Bool_t fV3; // switch to set the armonics
  Bool_t fIsMC; // if MC
  Bool_t fQAsw;   // if QA
  Bool_t fIsAfter2011; // switch for 2011 and later runs

  static const Int_t nCentrBin = 9;          //! # cenrality bins

  //
  // Cuts and options
  //

  Int_t fRun;                       //! current run checked to load VZERO calibrations

  Int_t fNcluster;           // Numer of TPC cluster required
  TList *fList;              //! List for output objects
  TList *fList2;             //! List for output objects
  TList *fList3;             //! List for output objects
  TList *fList4;             //! List for output objects
  //
  // Output objects
  TProfile *fMultV0;                //! object containing VZERO calibration information
  Float_t fV0Cpol;          //! loaded by OADB
  Float_t fV0Apol;          //! loaded by OADB
  Float_t fMeanQ[nCentrBin][2][2];           //! and recentering
  Float_t fWidthQ[nCentrBin][2][2];          //! ...
  Float_t fMeanQv3[nCentrBin][2][2];         //! also for v3
  Float_t fWidthQv3[nCentrBin][2][2];        //! ...

  TProfile *fHResTPCv0A2;   //! TProfile for subevent resolution (output)
  TProfile *fHResTPCv0C2;   //! TProfile for subevent resolution (output)
  TProfile *fHResv0Cv0A2;   //! TProfile for subevent resolution (output)
  TProfile *fHResTPCv0A3;    //! also for v3
  TProfile *fHResTPCv0C3;   //! also for v3
  TProfile *fHResv0Cv0A3;   //! also for v3

  TH2F *fPhiRPv0A;          //! EP distribution vs. centrality (v2)
  TH2F *fPhiRPv0C;          //! EP distribution vs. centrality (v2)
  TH2F *fPhiRPv0Av3;      //! EP distribution vs. centrality (v3)
  TH2F *fPhiRPv0Cv3;      //! EP distribution vs. centrality (v3)

  AliFlowVZEROQA *fQA;            //! QA histos (v2)
  AliFlowVZEROQA *fQA2;           //! QA histos (v2)
  AliFlowVZEROQA *fQAv3;        //! QA histos (v3)
  AliFlowVZEROQA *fQA2v3;       //! QA histos (v3)

  AliFlowBayesianPID *fPID;            //! PID class for the Bayesian probabilities
 
  TTree *fTree;                        //! tree to debug EP (if needed)

  Float_t fCentrality;  //! current centrality for the tree
  Float_t evPlAngV0ACor2;   //! subevent EPs (v2)
  Float_t evPlAngV0CCor2;   //! subevent EPs (v2)
  Float_t evPlAng2;   //! subevent EPs (v2)
  Float_t evPlAngV0ACor3;   //! subevent EPs (v3)
  Float_t evPlAngV0CCor3;   //! subevent EPs (v3)
  Float_t evPlAng3;   //! subevent EPs (v3)

  AliFlowVZEROResults *fContAllChargesV0A; //! results
  AliFlowVZEROResults *fContAllChargesV0C; //! results
  AliFlowVZEROResults *fContAllChargesV0Av3; //! results
  AliFlowVZEROResults *fContAllChargesV0Cv3; //! results
  AliFlowVZEROResults *fContAllChargesMC; //! results

  // MC
  TProfile *fHResMA2;   //! TProfile for subevent resolution (output)
  TProfile *fHResMC2;   //! TProfile for subevent resolution (output)
  TProfile *fHResAC2;   //! TProfile for subevent resolution (output)
  TProfile *fHResMA3;    //! also for v3
  TProfile *fHResMC3;   //! also for v3
  TProfile *fHResAC3;   //! also for v3

  AliFlowVZEROResults *fContAllChargesMCA; //! results
  AliFlowVZEROResults *fContAllChargesMCC; //! results
  AliFlowVZEROResults *fContAllChargesMCAv3; //! results
  AliFlowVZEROResults *fContAllChargesMCCv3; //! results

  Bool_t fFillDCA; // require to fill also DCA info
  TH2D *fHdcaPt[nCentrBin][7]; //! DCA distribution (for MC primary)
  TH2D *fHdcaPtSec[nCentrBin][7]; //! DCA distribution (for MC secondary, not used for data)
  AliFlowVZEROResults *fContQApid; //! QA pid object

  Bool_t fModulationDEDx; //add a modulation on the dE/dx response w.r.t. EP (kFALSE default)

  Float_t fZvtx; //! primary vertex z coordinate
  Int_t fNK0s; //! number of K0s in my private selection
  Float_t fPhiK0s[1000]; //! phi of K0s in my private selection
  Float_t fPtK0s[1000];//! pt of K0s in my private selection
  Int_t fNpiPos; //! number of positive pions for K0s selection
  Int_t fNpiNeg; //! number of negative pions for K0s selection
  Int_t fIPiPos[1000]; //! position in the AOD stack of positive pions for K0s
  Int_t fIPiNeg[1000]; //! position in the AOD stack of negative pions for K0s

  TH2D *fHKsPhi; //! Ks phi distribution
  TH2D *fHKsPhiEP; //! EP distribution
  TH2D *fHK0sMass; //! K0s mass vs. pt (private selection)
  TH2D *fHK0sMass2; //! K0s mass vs. pt (standard selection)
  TH2D *fHK0vsLambda; //! K0s vs lambda mass (in private K0s selection)
  TProfile2D *fHctauPtEP; //! ctau vs DeltaPhi,pt
  TH2F *fHctauAt1EP;    //! ctau vs. DeltaPhi at 1 GeV/c

  AliESDtrackCuts *fCutsDaughter; // daughter cut for K0s standard selection
  ClassDef(AliAnalysisTaskVnV0, 8);    //Analysis task v2 and v3 analysis on AOD
};

#endif
Commit	Line	Data
afa8df58	1	#ifndef ALIANALYSISTASKVNV0_H
	2	#define ALIANALYSISTASKVNV0_H
	3
	4	// ROOT includes
	5	#include <TObject.h>
	6	#include <TClonesArray.h>
	7	#include "TTree.h"
	8	#include <TList.h>
	9	#include <TProfile.h>
	10
	11	// AliRoot includes
	12	#include <AliAnalysisTaskSE.h>
	13	#include <AliAODEvent.h>
afa8df58	14	#include "AliFlowBayesianPID.h"
587d006a	15	#include "AliFlowVZEROResults.h"
243fbce7	16	#include "AliFlowVZEROQA.h"
afa8df58	17
afa8df58	18	class TH2F;
a640d925	19	class AliESDtrackCuts;
a31d07c8	20	class TProfile2D;
afa8df58	21
	22	class AliAnalysisTaskVnV0 : public AliAnalysisTaskSE {
	23	public:
	24	AliAnalysisTaskVnV0();
	25	AliAnalysisTaskVnV0(const char *name);
	26
	27	virtual ~AliAnalysisTaskVnV0();
	28
	29	virtual void UserCreateOutputObjects();
	30	virtual void UserExec(Option_t *option);
	31	virtual void Terminate(Option_t *);
	32
	33	Double_t GetVtxCut() { return fVtxCut; }
	34	Double_t GetEtaCut() { return fEtaCut; }
	35	Double_t GetMinPt() { return fMinPt; }
	36
	37	virtual void SetVtxCut(Double_t vtxCut){fVtxCut = vtxCut;}
	38	virtual void SetEtaCut(Double_t etaCut){fEtaCut = etaCut;}
	39	virtual void SetMinPt(Double_t value) {fMinPt = value;}
a640d925	40	virtual void SetMinDistV0(Double_t value){fMinDistV0=value;}
a640d925	41	virtual void SetMaxDistV0(Double_t value){fMaxDistV0=value;}
afa8df58	42	virtual void SetV2(Bool_t val){fV2 = val;};
	43	virtual void SetV3(Bool_t val){fV3 = val;};
	44
587d006a	45	virtual void SetMC(Bool_t flag = kTRUE){fIsMC = flag;};
	46	virtual void SetQA(Bool_t flag = kTRUE){fQAsw = flag;};
	47
afa8df58	48	void OpenInfoCalbration(Int_t run);
afa8df58	49
0f25ad32	50	void SetFillDCAinfo(Bool_t flag=kTRUE){fFillDCA = flag;};
	51
	52	void SetModulationDEDx(Bool_t flag=kTRUE){fModulationDEDx=flag;};
	53
a31d07c8	54	void SetAfter2011(Bool_t stat=kTRUE){fIsAfter2011=stat;};
	55	Bool_t GetAfter2011() const {return fIsAfter2011;};
	56
a640d925	57	void SetTPCclusterN(Int_t ncl){fNcluster=ncl;};
6fa5bb6e	58	static Bool_t IsPsiComputed(){return fgIsPsiComputed;};
	59	static Float_t GetPsi2V0A(){return fgPsi2v0a;};
	60	static Float_t GetPsi2V0C(){return fgPsi2v0c;};
	61	static Float_t GetPsi2TPC(){return fgPsi2tpc;};
	62	static Float_t GetPsi3V0A(){return fgPsi3v0a;};
	63	static Float_t GetPsi3V0C(){return fgPsi3v0c;};
	64	static Float_t GetPsi3TPC(){return fgPsi3tpc;};
a640d925	65	static Float_t GetPsi2V0AMC(){return fgPsi2v0aMC;};
	66	static Float_t GetPsi2V0CMC(){return fgPsi2v0cMC;};
	67	static Float_t GetPsi2TPCMC(){return fgPsi2tpcMC;};
	68	static Float_t GetPsi3V0AMC(){return fgPsi3v0aMC;};
	69	static Float_t GetPsi3V0CMC(){return fgPsi3v0cMC;};
	70	static Float_t GetPsi3TPCMC(){return fgPsi3tpcMC;};
6fa5bb6e	71
afa8df58	72	private:
243fbce7	73	AliAnalysisTaskVnV0(const AliAnalysisTaskVnV0 &old);
243fbce7	74	AliAnalysisTaskVnV0& operator=(const AliAnalysisTaskVnV0 &source);
a31d07c8	75
a640d925	76	Int_t PassesAODCuts(AliAODv0 myV0, AliAODEvent tAOD,Int_t specie);
afa8df58	77
6fa5bb6e	78	static Bool_t fgIsPsiComputed; // flag which return if event was processed
6fa5bb6e	79	static Float_t fgPsi2v0a,fgPsi2v0c,fgPsi2tpc; // current Psi2
a31d07c8	80	static Float_t fgPsi3v0a,fgPsi3v0c,fgPsi3tpc; // current Psi3
a640d925	81	static Float_t fgPsi2v0aMC,fgPsi2v0cMC,fgPsi2tpcMC; // current Psi2
a640d925	82	static Float_t fgPsi3v0aMC,fgPsi3v0cMC,fgPsi3tpcMC; // current Psi3
6fa5bb6e	83
afa8df58	84	virtual Float_t GetVertex(AliAODEvent* aod) const;
afa8df58	85	virtual void Analyze(AliAODEvent* aodEvent, Float_t v0Centr);
a31d07c8	86	virtual void SelectK0s();
a31d07c8	87
243fbce7	88	Double_t fVtxCut; // Vtx cut on z position in cm
	89	Double_t fEtaCut; // Eta cut used to select particles
	90	Double_t fMinPt; // Min pt - for histogram limits
a640d925	91	Double_t fMinDistV0; // Minimal distance for V0s
a640d925	92	Double_t fMaxDistV0; // Maximal distance for V0s
afa8df58	93
243fbce7	94	Bool_t fV2; // switch to set the armonics
	95	Bool_t fV3; // switch to set the armonics
	96	Bool_t fIsMC; // if MC
	97	Bool_t fQAsw; // if QA
a31d07c8	98	Bool_t fIsAfter2011; // switch for 2011 and later runs
afa8df58	99
243fbce7	100	static const Int_t nCentrBin = 9; //! # cenrality bins
afa8df58	101
	102	//
	103	// Cuts and options
	104	//
afa8df58	105
243fbce7	106	Int_t fRun; //! current run checked to load VZERO calibrations
afa8df58	107
a640d925	108	Int_t fNcluster; // Numer of TPC cluster required
243fbce7	109	TList *fList; //! List for output objects
	110	TList *fList2; //! List for output objects
	111	TList *fList3; //! List for output objects
	112	TList *fList4; //! List for output objects
afa8df58	113	//
afa8df58	114	// Output objects
243fbce7	115	TProfile *fMultV0; //! object containing VZERO calibration information
	116	Float_t fV0Cpol; //! loaded by OADB
	117	Float_t fV0Apol; //! loaded by OADB
	118	Float_t fMeanQ[nCentrBin][2][2]; //! and recentering
	119	Float_t fWidthQ[nCentrBin][2][2]; //! ...
	120	Float_t fMeanQv3[nCentrBin][2][2]; //! also for v3
	121	Float_t fWidthQv3[nCentrBin][2][2]; //! ...
	122
	123	TProfile *fHResTPCv0A2; //! TProfile for subevent resolution (output)
	124	TProfile *fHResTPCv0C2; //! TProfile for subevent resolution (output)
	125	TProfile *fHResv0Cv0A2; //! TProfile for subevent resolution (output)
	126	TProfile *fHResTPCv0A3; //! also for v3
	127	TProfile *fHResTPCv0C3; //! also for v3
	128	TProfile *fHResv0Cv0A3; //! also for v3
	129
	130	TH2F *fPhiRPv0A; //! EP distribution vs. centrality (v2)
	131	TH2F *fPhiRPv0C; //! EP distribution vs. centrality (v2)
	132	TH2F *fPhiRPv0Av3; //! EP distribution vs. centrality (v3)
	133	TH2F *fPhiRPv0Cv3; //! EP distribution vs. centrality (v3)
	134
	135	AliFlowVZEROQA *fQA; //! QA histos (v2)
	136	AliFlowVZEROQA *fQA2; //! QA histos (v2)
	137	AliFlowVZEROQA *fQAv3; //! QA histos (v3)
	138	AliFlowVZEROQA *fQA2v3; //! QA histos (v3)
	139
	140	AliFlowBayesianPID *fPID; //! PID class for the Bayesian probabilities
afa8df58	141
243fbce7	142	TTree *fTree; //! tree to debug EP (if needed)
afa8df58	143
243fbce7	144	Float_t fCentrality; //! current centrality for the tree
	145	Float_t evPlAngV0ACor2; //! subevent EPs (v2)
	146	Float_t evPlAngV0CCor2; //! subevent EPs (v2)
	147	Float_t evPlAng2; //! subevent EPs (v2)
	148	Float_t evPlAngV0ACor3; //! subevent EPs (v3)
	149	Float_t evPlAngV0CCor3; //! subevent EPs (v3)
	150	Float_t evPlAng3; //! subevent EPs (v3)
afa8df58	151
243fbce7	152	AliFlowVZEROResults *fContAllChargesV0A; //! results
	153	AliFlowVZEROResults *fContAllChargesV0C; //! results
	154	AliFlowVZEROResults *fContAllChargesV0Av3; //! results
	155	AliFlowVZEROResults *fContAllChargesV0Cv3; //! results
	156	AliFlowVZEROResults *fContAllChargesMC; //! results
587d006a	157
08371bdc	158	// MC
	159	TProfile *fHResMA2; //! TProfile for subevent resolution (output)
	160	TProfile *fHResMC2; //! TProfile for subevent resolution (output)
	161	TProfile *fHResAC2; //! TProfile for subevent resolution (output)
	162	TProfile *fHResMA3; //! also for v3
	163	TProfile *fHResMC3; //! also for v3
	164	TProfile *fHResAC3; //! also for v3
587d006a	165
08371bdc	166	AliFlowVZEROResults *fContAllChargesMCA; //! results
	167	AliFlowVZEROResults *fContAllChargesMCC; //! results
	168	AliFlowVZEROResults *fContAllChargesMCAv3; //! results
	169	AliFlowVZEROResults *fContAllChargesMCCv3; //! results
	170
0f25ad32	171	Bool_t fFillDCA; // require to fill also DCA info
	172	TH2D *fHdcaPt[nCentrBin][7]; //! DCA distribution (for MC primary)
	173	TH2D *fHdcaPtSec[nCentrBin][7]; //! DCA distribution (for MC secondary, not used for data)
	174	AliFlowVZEROResults *fContQApid; //! QA pid object
	175
	176	Bool_t fModulationDEDx; //add a modulation on the dE/dx response w.r.t. EP (kFALSE default)
	177
a31d07c8	178	Float_t fZvtx; //! primary vertex z coordinate
	179	Int_t fNK0s; //! number of K0s in my private selection
	180	Float_t fPhiK0s[1000]; //! phi of K0s in my private selection
	181	Float_t fPtK0s[1000];//! pt of K0s in my private selection
	182	Int_t fNpiPos; //! number of positive pions for K0s selection
	183	Int_t fNpiNeg; //! number of negative pions for K0s selection
	184	Int_t fIPiPos[1000]; //! position in the AOD stack of positive pions for K0s
	185	Int_t fIPiNeg[1000]; //! position in the AOD stack of negative pions for K0s
	186
	187	TH2D *fHKsPhi; //! Ks phi distribution
	188	TH2D *fHKsPhiEP; //! EP distribution
	189	TH2D *fHK0sMass; //! K0s mass vs. pt (private selection)
	190	TH2D *fHK0sMass2; //! K0s mass vs. pt (standard selection)
	191	TH2D *fHK0vsLambda; //! K0s vs lambda mass (in private K0s selection)
	192	TProfile2D *fHctauPtEP; //! ctau vs DeltaPhi,pt
	193	TH2F *fHctauAt1EP; //! ctau vs. DeltaPhi at 1 GeV/c
	194
	195	AliESDtrackCuts *fCutsDaughter; // daughter cut for K0s standard selection
	196	ClassDef(AliAnalysisTaskVnV0, 8); //Analysis task v2 and v3 analysis on AOD
afa8df58	197	};
	198
	199	#endif