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"
35 class AliAnalysisTaskSEDvsMultiplicity : public AliAnalysisTaskSE
39 AliAnalysisTaskSEDvsMultiplicity();
40 AliAnalysisTaskSEDvsMultiplicity(const char *name, Int_t pdgMeson, AliRDHFCuts* cuts);
41 virtual ~AliAnalysisTaskSEDvsMultiplicity();
44 void SetMassLimits(Double_t lowlimit, Double_t uplimit);
45 void SetMassLimits(Int_t pdg, Double_t range);
46 Double_t GetUpperMassLimit() const {return fUpmasslimit;}
47 Double_t GetLowerMassLimit() const {return fLowmasslimit;}
48 void SetNMassBins(Int_t nbins){fNMassBins=nbins;}
49 Int_t GetNMassBins() const{return fNMassBins;}
51 void SetImpactParameterBinning(Int_t nbins, Double_t dmin, Double_t dmax){
57 void SetReadMC(Bool_t readMC=kTRUE){fReadMC=readMC;}
58 void SetUseBit(Bool_t use=kTRUE){fUseBit=use;}
59 void SetDoImpactParameterHistos(Bool_t doImp=kTRUE){fDoImpPar=doImp;}
62 void SetMultiplVsZProfileLHC10b(TProfile* hprof){
63 if(fMultEstimatorAvg[0]) delete fMultEstimatorAvg[0];
64 fMultEstimatorAvg[0]=new TProfile(*hprof);
66 void SetMultiplVsZProfileLHC10c(TProfile* hprof){
67 if(fMultEstimatorAvg[1]) delete fMultEstimatorAvg[1];
68 fMultEstimatorAvg[1]=new TProfile(*hprof);
70 void SetMultiplVsZProfileLHC10d(TProfile* hprof){
71 if(fMultEstimatorAvg[2]) delete fMultEstimatorAvg[2];
72 fMultEstimatorAvg[2]=new TProfile(*hprof);
74 void SetMultiplVsZProfileLHC10e(TProfile* hprof){
75 if(fMultEstimatorAvg[3]) delete fMultEstimatorAvg[3];
76 fMultEstimatorAvg[3]=new TProfile(*hprof);
80 Int_t CheckOrigin(TClonesArray* arrayMC, AliAODMCParticle *mcPartCandidate) const;
82 // Implementation of interface methods
83 virtual void UserCreateOutputObjects();
85 virtual void LocalInit() {Init();}
86 virtual void UserExec(Option_t *option);
87 virtual void Terminate(Option_t *option);
91 AliAnalysisTaskSEDvsMultiplicity(const AliAnalysisTaskSEDvsMultiplicity &source);
92 AliAnalysisTaskSEDvsMultiplicity& operator=(const AliAnalysisTaskSEDvsMultiplicity& source);
94 TProfile* GetEstimatorHistogram(const AliVEvent *event);
95 void CreateImpactParameterHistos();
98 TList *fOutput; //! list send on output slot 1
99 TList *fListCuts; //list of cuts
100 TList *fOutputCounters; //! list send on output slot 3
102 TH1F *fHistNEvents; //!hist. for No. of events
104 TH2F* fHistNtrEta16vsNtrEta1; //!hist. for Ntracklets in eta<1.6 vs. eta<1.
105 TH2F* fHistNtrCorrEta1vsNtrRawEta1; //!hist. for Ntracklets in eta<1 with and w/o corrections
106 TH2F* fHistNtrVsZvtx; //! hist of ntracklets vs Zvertex
107 TH2F* fHistNtrCorrVsZvtx; //! hist of ntracklets vs Zvertex
109 TH1F* fHistNtrCorrEvSel; //! hist. of ntracklets for selected events
110 TH1F* fHistNtrCorrEvWithCand; //! hist. of ntracklets for evnts with a candidate
111 TH1F* fHistNtrCorrEvWithD;//! hist. of ntracklets for evnts with a candidate in D mass peak
113 TH3F *fPtVsMassVsMult; //! hist. of Pt vs Mult vs. mass (
114 TH3F *fPtVsMassVsMultNoPid; //! hist. of Pt vs Mult vs. mass (no pid)
115 TH3F *fPtVsMassVsMultUncorr; //! hist. of Pt vs Mult vs. mass (raw mult)
116 TH3F *fPtVsMassVsMultPart; //! hist. of Pt vs Mult vs. mass (particle)
117 TH3F *fPtVsMassVsMultAntiPart; //! hist. of Pt vs Mult vs. mass (antiparticle)
119 THnSparseF *fHistMassPtImpPar[5];//! histograms for impact paramter studies
121 Double_t fUpmasslimit; //upper inv mass limit for histos
122 Double_t fLowmasslimit; //lower inv mass limit for histos
123 Int_t fNMassBins; // nbins for invariant mass histos
125 AliRDHFCuts *fRDCutsAnalysis; // Cuts for Analysis
126 AliNormalizationCounter *fCounter; //!Counter for normalization
127 AliNormalizationCounter *fCounterU; //!Counter for normalization, uncorr mult
129 Bool_t fDoImpPar; //swicth for D impact parameter THnSparse
130 Int_t fNImpParBins; // nunber of bins in impact parameter histos
131 Double_t fLowerImpPar; // lower limit in impact parameter (um)
132 Double_t fHigherImpPar; // higher limit in impact parameter (um)
134 Bool_t fReadMC; //flag for access to MC
135 Int_t fMCOption; // 0=keep all cand, 1=keep only signal, 2= keep only back
136 Bool_t fUseBit; // flag to use bitmask
138 TProfile* fMultEstimatorAvg[4]; // TProfile with mult vs. Z per period
139 Double_t fRefMult; // refrence multiplcity (period b)
140 Int_t fPdgMeson; // pdg code of analyzed meson
143 ClassDef(AliAnalysisTaskSEDvsMultiplicity,1); // AliAnalysisTaskSE for the MC association of heavy-flavour decay candidates