1 #ifndef ALIANALYSISTASKSEDVSMULTIPLICITY_H
2 #define ALIANALYSISTASKSEDVSMULTIPLICITY_H
4 /* Copyright(c) 1998-2008, ALICE Experiment at CERN, All rights reserved. *
5 * See cxx source for full Copyright notice */
9 //*************************************************************************
10 // Class AliAnalysisTaskSEDvsMultiplicity
11 // AliAnalysisTaskSE for the D meson vs. multiplcity analysis
12 // Authors: Renu Bala, Zaida Conesa del Valle, Francesco Prino
13 //*************************************************************************
24 #include "AliRDHFCutsDplustoKpipi.h"
25 #include "AliAnalysisTaskSE.h"
26 #include "AliAnalysisVertexingHF.h"
27 #include "AliNormalizationCounter.h"
28 #include "AliAODMCHeader.h"
29 #include "AliAODMCParticle.h"
30 #include "AliVertexingHFUtils.h"
31 #include "AliVEvent.h"
34 class AliAnalysisTaskSEDvsMultiplicity : public AliAnalysisTaskSE
38 AliAnalysisTaskSEDvsMultiplicity();
39 AliAnalysisTaskSEDvsMultiplicity(const char *name, Int_t pdgMeson, AliRDHFCuts* cuts, Bool_t switchPPb);
40 virtual ~AliAnalysisTaskSEDvsMultiplicity();
43 void SetMassLimits(Double_t lowlimit, Double_t uplimit);
44 void SetMassLimits(Int_t pdg, Double_t range);
45 Double_t GetUpperMassLimit() const {return fUpmasslimit;}
46 Double_t GetLowerMassLimit() const {return fLowmasslimit;}
47 void SetNMassBins(Int_t nbins){fNMassBins=nbins;}
48 Int_t GetNMassBins() const{return fNMassBins;}
49 Bool_t GetSubtractTrackletsFromDaughters() const {return fSubtractTrackletsFromDau;}
51 void SetImpactParameterBinning(Int_t nbins, Double_t dmin, Double_t dmax){
57 void SetReadMC(Bool_t readMC=kTRUE){fReadMC=readMC;}
58 void SetMCOption(Int_t option=0){ fMCOption = option; }
59 void SetIsPPbData(Bool_t flag=kTRUE){
62 void SetUseBit(Bool_t use=kTRUE){fUseBit=use;}
63 void SetDoImpactParameterHistos(Bool_t doImp=kTRUE){fDoImpPar=doImp;}
64 void SetKeepEstimatorCorrelationPlots(Bool_t use=kTRUE){fKeepCorrPlots=use;}
66 void SetMultiplVsZProfileLHC10b(TProfile* hprof){
67 if(fMultEstimatorAvg[0]) delete fMultEstimatorAvg[0];
68 fMultEstimatorAvg[0]=new TProfile(*hprof);
70 void SetMultiplVsZProfileLHC10c(TProfile* hprof){
71 if(fMultEstimatorAvg[1]) delete fMultEstimatorAvg[1];
72 fMultEstimatorAvg[1]=new TProfile(*hprof);
74 void SetMultiplVsZProfileLHC10d(TProfile* hprof){
75 if(fMultEstimatorAvg[2]) delete fMultEstimatorAvg[2];
76 fMultEstimatorAvg[2]=new TProfile(*hprof);
78 void SetMultiplVsZProfileLHC10e(TProfile* hprof){
79 if(fMultEstimatorAvg[3]) delete fMultEstimatorAvg[3];
80 fMultEstimatorAvg[3]=new TProfile(*hprof);
83 void SetMultiplVsZProfileLHC13b(TProfile* hprof){
84 if(fMultEstimatorAvg[0]) delete fMultEstimatorAvg[0];
85 fMultEstimatorAvg[0]=new TProfile(*hprof);
87 void SetMultiplVsZProfileLHC13c(TProfile* hprof){
88 if(fMultEstimatorAvg[1]) delete fMultEstimatorAvg[1];
89 fMultEstimatorAvg[1]=new TProfile(*hprof);
92 void SetReferenceMultiplcity(Double_t rmu){fRefMult=rmu;}
95 void UseMCNchWeight(Bool_t flag) { fUseNchWeight = flag; }
96 void SetHistoNchWeight(TH1F *h){
97 if(fHistoMCNch) delete fHistoMCNch;
98 fHistoMCNch = new TH1F(*h);
101 void SetSubtractTrackletsFromDaughters(Bool_t opt){fSubtractTrackletsFromDau=opt;}
102 Int_t CheckOrigin(TClonesArray* arrayMC, AliAODMCParticle *mcPartCandidate) const;
104 enum { kNtrk10=0, kNtrk10to16=1, kVZERO=2, kNtrk03=3, kNtrk05=4, kVZEROA=5 };
105 void SetMultiplicityEstimator(Int_t value){ fMultiplicityEstimator=value; }
106 Int_t GetMultiplicityEstimator(){ return fMultiplicityEstimator; }
107 enum { kEta10=0, kEta10to16=1, kEtaVZERO=2, kEta03=3, kEta05=5, kEtaVZEROA=5 };
108 void SetMCPrimariesEstimator(Int_t value){ fMCPrimariesEstimator=value; }
109 Int_t GetMCPrimariesEstimator(){ return fMCPrimariesEstimator; }
111 // Implementation of interface methods
112 virtual void UserCreateOutputObjects();
114 virtual void LocalInit() {Init();}
115 virtual void UserExec(Option_t *option);
116 virtual void Terminate(Option_t *option);
120 AliAnalysisTaskSEDvsMultiplicity(const AliAnalysisTaskSEDvsMultiplicity &source);
121 AliAnalysisTaskSEDvsMultiplicity& operator=(const AliAnalysisTaskSEDvsMultiplicity& source);
123 TProfile* GetEstimatorHistogram(const AliVEvent *event);
124 void CreateImpactParameterHistos();
125 void CreateMeasuredNchHisto();
126 void FillMCMassHistos(TClonesArray *arrayMC, Int_t labD, Int_t countMult,Double_t nchWeight);
128 TList *fOutput; //! list send on output slot 1
129 TList *fListCuts; //list of cuts
130 TList *fOutputCounters; //! list send on output slot 3
131 TList *fListProfiles; //list of profile histos for z-vtx correction
133 TH1F *fHistNEvents; //!hist. for No. of events
135 TH2F* fHistNtrEta16vsNtrEta1EvSel; //!hist. for Ntracklets in eta<1.6 vs. eta<1.
136 TH2F* fHistNtrEta05vsNtrEta1EvSel; //!hist. for Ntracklets in eta<0.5 vs. eta<1.
137 TH2F* fHistNtrEta03vsNtrEta1EvSel; //!hist. for Ntracklets in eta<0.3 vs. eta<1.
138 TH2F* fHistNtrEtaV0AvsNtrEta1EvSel; //!hist. for Ntracklets in eta-V0A vs. eta<1.
139 TH2F* fHistNtrEtaV0MvsNtrEta1EvSel; //!hist. for Ntracklets in eta-V0M vs. eta<1.
140 TH2F* fHistNtrCorrEta1vsNtrRawEta1EvSel; //!hist. for Ntracklets in eta<1 with and w/o corrections
141 TH2F* fHistNtrEta16vsNtrEta1EvWithCand; //!hist. for Ntracklets in eta<1.6 vs. eta<1. for events with a candidate
142 TH2F* fHistNtrEta05vsNtrEta1EvWithCand; //!hist. for Ntracklets in eta<0.5 vs. eta<1. for events with a candidate
143 TH2F* fHistNtrEta03vsNtrEta1EvWithCand; //!hist. for Ntracklets in eta<0.3 vs. eta<1. for events with a candidate
144 TH2F* fHistNtrEtaV0AvsNtrEta1EvWithCand; //!hist. for Ntracklets in eta-V0A vs. eta<1. for events with a candidate
145 TH2F* fHistNtrEtaV0MvsNtrEta1EvWithCand; //!hist. for Ntracklets in eta-V0M vs. eta<1. for events with a candidate
146 TH2F* fHistNtrCorrEta1vsNtrRawEta1EvWithCand; //!hist. for Ntracklets in eta<1 with and w/o corrections for events with a candidate
147 TH2F* fHistNtrEta16vsNtrEta1EvWithD; //!hist. for Ntracklets in eta<1.6 vs. eta<1. for events with a candidate in D mass peak
148 TH2F* fHistNtrEta05vsNtrEta1EvWithD; //!hist. for Ntracklets in eta<0.5 vs. eta<1. for events with a candidate in D mass peak
149 TH2F* fHistNtrEta03vsNtrEta1EvWithD; //!hist. for Ntracklets in eta<0.3 vs. eta<1. for events with a candidate in D mass peak
150 TH2F* fHistNtrEtaV0AvsNtrEta1EvWithD; //!hist. for Ntracklets in eta-V0A vs. eta<1. for events with a candidate in D mass peak
151 TH2F* fHistNtrEtaV0MvsNtrEta1EvWithD; //!hist. for Ntracklets in eta-V0M vs. eta<1. for events with a candidate in D mass peak
152 TH2F* fHistNtrCorrEta1vsNtrRawEta1EvWithD; //!hist. for Ntracklets in eta<1 with and w/o corrections for events with a candidate in D mass peak
154 TH2F* fHistNtrVsZvtx; //! hist of ntracklets vs Zvertex
155 TH2F* fHistNtrCorrVsZvtx; //! hist of ntracklets vs Zvertex
157 TH2F* fHistNtrVsNchMC; //! hist of ntracklets vs Nch (Generated)
158 TH2F* fHistNtrCorrVsNchMC; //! hist of ntracklets vs Nch (Generated)
159 TH2F* fHistNtrVsNchMCPrimary; //! hist of ntracklets vs Nch (Primary)
160 TH2F* fHistNtrCorrVsNchMCPrimary; //! hist of ntracklets vs Nch (Primary)
161 TH2F* fHistNtrVsNchMCPhysicalPrimary; //! hist of ntracklets vs Nch (Physical Primary)
162 TH2F* fHistNtrCorrVsNchMCPhysicalPrimary; //! hist of ntracklets vs Nch (Physical Primary)
163 TH1F* fHistGenPrimaryParticlesInelGt0; //!hist. of geenrated multiplcity
164 TH3F* fHistNchMCVsNchMCPrimaryVsNchMCPhysicalPrimary; //! hist of Nch (generated) vs Nch (Primary) vs Nch (Physical Primary)
166 TH1F* fHistNtrUnCorrPSSel; //! hist. of ntracklets for physics selection only selected events
167 TH1F* fHistNtrUnCorrPSTrigSel; //! hist. of ntracklets for physics selection + trigger name selected events
168 TH1F* fHistNtrUnCorrPSTrigPileUpSel; //! hist. of ntracklets for physics selection + trigger name + pileup selected events
169 TH1F* fHistNtrUnCorrPSTrigPileUpVtxSel; //! hist. of ntracklets for physics selection + trigger name + pileup + with-vertex selected events
170 TH1F* fHistNtrUnCorrPSTrigPileUpVtxContSel; //! hist. of ntracklets for physics selection + trigger name + pileup + with-vertex-contrib selected events
171 TH1F* fHistNtrUnCorrPSTrigPileUpVtxRangeSel; //! hist. of ntracklets for physics selection + trigger name + pileup + with-vertex-contrib-range selected events
172 TH1F* fHistNtrUnCorrPSTrigPileUpVtxRangeCentrSel; //! hist. of ntracklets for physics selection + trigger name + pileup + with-vertex-contrib-range + centrality selected events
173 TH1F* fHistNtrUnCorrEvSel; //! hist. of ntracklets for selected events
174 TH1F* fHistNtrUnCorrEvWithCand; //! hist. of ntracklets for evnts with a candidate
175 TH1F* fHistNtrUnCorrEvWithD;//! hist. of ntracklets for evnts with a candidate in D mass peak
176 TH1F* fHistNtrCorrPSSel; //! hist. of ntracklets for physics selection only selected events
177 TH1F* fHistNtrCorrEvSel; //! hist. of ntracklets for selected events
178 TH1F* fHistNtrCorrEvWithCand; //! hist. of ntracklets for evnts with a candidate
179 TH1F* fHistNtrCorrEvWithD;//! hist. of ntracklets for evnts with a candidate in D mass peak
182 TH3F *fPtVsMassVsMult; //! hist. of Pt vs Mult vs. mass (
183 TH3F *fPtVsMassVsMultNoPid; //! hist. of Pt vs Mult vs. mass (no pid)
184 TH3F *fPtVsMassVsMultUncorr; //! hist. of Pt vs Mult vs. mass (raw mult)
185 TH3F *fPtVsMassVsMultPart; //! hist. of Pt vs Mult vs. mass (particle)
186 TH3F *fPtVsMassVsMultAntiPart; //! hist. of Pt vs Mult vs. mass (antiparticle)
187 TH3F *fPtVsMassVsMultMC; //! hist. of Pt vs Mult vs. mass (MC true candidates before reconstruction)
189 THnSparseF *fHistMassPtImpPar[5];//! histograms for impact paramter studies
191 Double_t fUpmasslimit; //upper inv mass limit for histos
192 Double_t fLowmasslimit; //lower inv mass limit for histos
193 Int_t fNMassBins; // nbins for invariant mass histos
195 AliRDHFCuts *fRDCutsAnalysis; // Cuts for Analysis
196 AliNormalizationCounter *fCounter; //!Counter for normalization
197 AliNormalizationCounter *fCounterU; //!Counter for normalization, uncorr mult
199 Bool_t fDoImpPar; //swicth for D impact parameter THnSparse
200 Int_t fNImpParBins; // nunber of bins in impact parameter histos
201 Double_t fLowerImpPar; // lower limit in impact parameter (um)
202 Double_t fHigherImpPar; // higher limit in impact parameter (um)
204 Bool_t fReadMC; //flag for access to MC
205 Int_t fMCOption; // 0=keep all cand, 1=keep only signal, 2= keep only back
206 Bool_t fisPPbData; // flag to run on pPb data (differen histogram bining)
207 Bool_t fUseBit; // flag to use bitmask
208 Bool_t fSubtractTrackletsFromDau; // flag for subtracting D meson daughter contribution to N of tracklets
209 Bool_t fKeepCorrPlots; // flag to look at the correlation of different estimators (eta ranges)
211 Bool_t fUseNchWeight; // weight on the MC on the generated multiplicity
212 TH1F* fHistoMCNch; // weight histogram for the MC on the generated multiplicity
213 TH1F* fHistoMeasNch; //! weight histogram on the true measured multiplicity
215 TProfile* fMultEstimatorAvg[4]; //TProfile with mult vs. Z per period
216 Double_t fRefMult; // refrence multiplcity (period b)
217 Int_t fPdgMeson; // pdg code of analyzed meson
219 Int_t fMultiplicityEstimator; // Definition of the multiplicity estimator: kNtrk10=0, kNtrk10to16=1, kVZERO=2
220 Int_t fMCPrimariesEstimator; // Definition of the primaries estimator eta range: |eta|<1.0=0, -1.6<|eta|<1.0=1, VZEROrange=2
222 ClassDef(AliAnalysisTaskSEDvsMultiplicity,13); // D vs. mult task