1 /**************************************************************************
2 * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4 * Author: The ALICE Off-line Project. *
5 * Contributors are mentioned in the code where appropriate. *
7 * Permission to use, copy, modify and distribute this software and its *
8 * documentation strictly for non-commercial purposes is hereby granted *
9 * without fee, provided that the above copyright notice appears in all *
10 * copies and that both the copyright notice and this permission notice *
11 * appear in the supporting documentation. The authors make no claims *
12 * about the suitability of this software for any purpose. It is *
13 * provided "as is" without express or implied warranty. *
14 **************************************************************************/
16 // QA class of Heavy Flavor quark and fragmeted/decayed particles
17 // -Check kinematics of Heavy Quarks/hadrons, and decayed leptons
19 // decay lepton kinematics w/wo acceptance
20 // heavy hadron decay length, electron pT fraction carried from decay
21 // -Check yield of Heavy Quarks/hadrons
22 // Number of produced heavy quark
23 // Number of produced hadron of given pdg code
30 //#include <TObject.h>
39 class AliGenEventHeader;
41 class AliAODMCParticle;
42 class AliHFEcollection;
43 class TTreeSRedirector;
45 //________________________________________________________________
46 class AliHFEmcQA: public TObject {
49 enum heavyType {kCharm=4, kBeauty=5, kOthers=6, kElectronPDG=11};
50 enum qType {kQuark, kantiQuark, kHadron, keHadron, kDeHadron, kElectron, kElectron2nd};
51 enum SourceType {kDirectCharm=1, kDirectBeauty=2, kBeautyCharm=3, kGamma=4, kPi0=5, kElse=6, kMisID=7, kEta=8, kOmega=9, kPhi=10, kEtaPrime=11, kRho0=12, kGammaPi0=13, kGammaEta=14, kGammaOmega=15, kGammaPhi=16, kGammaEtaPrime=17, kGammaRho0=18, kJpsi=19, kB2Jpsi=20, kKe3=21, kGammaB2M=22, kGammaD2M=23, kGammaEta2Pi0=24, kB2M=25, kD2M=26, kEta2Pi0=27, kScdryM=28};
53 kPairCreationFromq, kPairCreationFromg, kFlavourExitation, kGluonSplitting, kInitialPartonShower, kLightQuarkShower
64 AliHFEmcQA(const AliHFEmcQA &p); // copy constructor
65 AliHFEmcQA &operator=(const AliHFEmcQA &); // assignment operator
67 virtual ~AliHFEmcQA();
69 TList *GetList() const { return fQAhistos; };
70 void PostAnalyze() const;
71 void CreatDefaultHistograms(TList * const qaList); // create default histograms
72 void CreateHistograms(const Int_t kquark); // create histograms for mc qa analysis
73 void SetMCEvent(AliMCEvent* const mcEvent){fMCEvent = mcEvent;}
74 void SetGenEventHeader(AliGenEventHeader* const mcHeader){fMCHeader=mcHeader;} // set stack pointer
75 void SetMCArray(TClonesArray* const mcarry){fMCArray=mcarry;} // set mcarray pointer
78 void GetQuarkKine(TParticle *part, Int_t iTrack, const Int_t kquark); // get heavy quark kinematics distribution
79 void GetHadronKine(TParticle *part, const Int_t kquark); // get heavy hadron kinematics distribution
80 void GetDecayedKine(TParticle *part, const Int_t kquark, const Int_t kdecayed); // get decay electron kinematics distribution
81 void GetDecayedKine(AliAODMCParticle *mcpart, const Int_t kquark, Int_t kdecayed); // get decay electron kinematics for AOD
82 void GetMesonKine(); // get meson and its decay electron pt spectra
83 void EndOfEventAna(const Int_t kquark); // run analysis which should be done at the end of the event loop
84 Int_t GetSource(const TParticle * const mcpart) const; // return source id
85 Int_t GetElecSource(const AliVParticle * const mctrack) const;
86 Int_t GetElecSource(TParticle * const mcpart) const; // return electron source id
87 Int_t GetElecSource(const AliAODMCParticle * const mcpart) const;
88 Int_t GetSource(const AliVParticle * const mcpart) const; // return electron source id for AOD
89 Double_t GetWeightFactor(AliMCParticle *mctrack, const Int_t iBgLevel); // return best/lower/upper weighting factor for electron's mother meson
90 Double_t GetWeightFactor(const AliAODMCParticle * const mcpart, const Int_t iBgLevel);
91 Int_t GetWeightCentralityBin(const Float_t percentile) const; //translate the centrality percentile into the centrality bin of the reference weighting histograms for electron background
92 void EnableDebugStreamer() { fIsDebugStreamerON = kTRUE;};
94 void SetBackgroundWeightFactor(Double_t *elecBackgroundFactor, Double_t *binLimit);
95 void SetContainerStep(Int_t containerStep) { fContainerStep = containerStep;};
96 void SetHFEImpactParameters(Double_t hfeimpactR, Double_t hfeimpactnsigmaR) {fHfeImpactR = hfeimpactR; fHfeImpactnsigmaR = hfeimpactnsigmaR; };
97 void SetTrkKine(Double_t pt, Double_t eta, Double_t phi) {fRecPt = pt; fRecEta = eta; fRecPhi = phi;};
98 void SetITSInfo(Double_t ilyrhit, Double_t ilyrstat) { fLyrhit = ilyrhit; fLyrstat = ilyrstat;};
100 void SetCentrality(Int_t centrality) { fCentrality = centrality; };
101 void SetPercentrality(Int_t percentrality) { fPerCentrality = percentrality; };//centrality percentile
102 void SetPbPb() { fIsPbPb = kTRUE; };
103 void SetPP() { fIsPbPb = kFALSE; };
104 void SetPPMultiBin() { fIsppMultiBin = kFALSE; };
105 Bool_t IsPbPb() const { return fIsPbPb; };
106 Bool_t IsPP() const { return !fIsPbPb; };
107 Bool_t IsPPMultiBin() const { return fIsppMultiBin; };
110 Int_t GetMother(const AliVParticle * const track) const;
111 void IdentifyMother(Int_t motherlabel, Int_t &motherpdg, Int_t &grandmotherlabel); //
112 void HardScattering(const Int_t kquark, Int_t &motherID, Int_t &mothertype, Int_t &motherlabel); // check if the quark is produced from hard scattering
113 void ReportStrangeness(Int_t &motherID, Int_t &mothertype, Int_t &motherlabel); // report if the quark production process is unknown
114 Bool_t IsFromInitialShower(Int_t inputmotherlabel, Int_t &motherID, Int_t &mothertype, Int_t &motherlabel); // check if the quark is produced from initial parton shower
115 Bool_t IsFromFinalParton(Int_t inputmotherlabel, Int_t &motherID, Int_t &mothertype, Int_t &motherlabel); // check if the quark is produced from final parton shower
117 AliMCEvent* fMCEvent; // mcevent pointer
118 AliGenEventHeader* fMCHeader; // mcheader pointer
119 TClonesArray *fMCArray; // mc array pointer
121 static const Int_t fgkGluon=21; // gluon pdg code
122 static const Int_t fgkMaxGener=10; // ancester level wanted to be checked
123 static const Int_t fgkMaxIter=100; // number of iteration to find out matching particle
124 static const Int_t fgkqType=7; // number of particle type to be checked
125 static const Int_t fgkEtaRanges=3; // cuts for different eta ranges
128 TH1F *fPdgCode; // histogram to store particle pdg code
129 TH1F *fPt; // histogram to store pt
130 TH1F *fY; // histogram to store rapidity
131 TH1F *fEta; // histogram to store eta
139 // default constructor
141 AliHists(const AliHists & p)
142 : fPdgCode(p.fPdgCode)
149 AliHists &operator=(const AliHists &)
151 // assignment operator, not yet implemented
154 void FillList(TList *l) const;
156 struct AliHistsComm {
157 TH1F *fNq; // histogram to store number of quark
158 TH1F *fProcessID; // histogram to store process id
159 TH2F *fePtRatio; // fraction of electron pT from D or B hadron
160 TH2F *fPtCorr; // pt correlation between e and direct D or B
161 TH2F *fPtCorrDp; // pt correlation between e and direct D+
162 TH2F *fPtCorrD0; // pt correlation between e and direct D0
163 TH2F *fPtCorrDrest; // pt correlation between e and direct D rest
164 TH2F *fDePtRatio; // fraction of D electron pT from B hadron
165 TH2F *feDistance; // distance between electron production point to mother particle
166 TH2F *fDeDistance; // distance between D electron production point to mother particle
168 TH2F *fPtCorrDinein; // pt correlation between e and direct D or B
169 TH2F *fPtCorrDineout; // pt correlation between e and direct D or B
170 TH2F *fPtCorrDoutein; // pt correlation between e and direct D or B
171 TH2F *fPtCorrDouteout; // pt correlation between e and direct D or B
172 TH2F *fPtCorrDpDinein; // pt correlation between e and direct D+
173 TH2F *fPtCorrDpDineout; // pt correlation between e and direct D+
174 TH2F *fPtCorrDpDoutein; // pt correlation between e and direct D+
175 TH2F *fPtCorrDpDouteout; // pt correlation between e and direct D+
176 TH2F *fPtCorrD0Dinein; // pt correlation between e and direct D0
177 TH2F *fPtCorrD0Dineout; // pt correlation between e and direct D0
178 TH2F *fPtCorrD0Doutein; // pt correlation between e and direct D0
179 TH2F *fPtCorrD0Douteout; // pt correlation between e and direct D0
180 TH2F *fPtCorrDrestDinein; // pt correlation between e and direct D rest
181 TH2F *fPtCorrDrestDineout; // pt correlation between e and direct D rest
182 TH2F *fPtCorrDrestDoutein; // pt correlation between e and direct D rest
183 TH2F *fPtCorrDrestDouteout; // pt correlation between e and direct D rest
185 TH2F *fEtaCorrD; // eta correlation between e and direct D
186 TH2F *fEtaCorrDp; // eta correlation between e and direct D
187 TH2F *fEtaCorrD0; // eta correlation between e and direct D
188 TH2F *fEtaCorrDrest; // eta correlation between e and direct D
189 TH2F *fEtaCorrGD; // eta correlation between e and direct D
190 TH2F *fEtaCorrGDp; // eta correlation between e and direct D
191 TH2F *fEtaCorrGD0; // eta correlation between e and direct D
192 TH2F *fEtaCorrGDrest; // eta correlation between e and direct D
194 TH2F *fEtaCorrB; // eta correlation between e and B
195 TH2F *fEtaCorrGB; // eta correlation between e and B
196 TH2F *fPtCorrBinein; // eta correlation between e and B
197 TH2F *fPtCorrBineout; // eta correlation between e and B
198 TH2F *fPtCorrBoutein; // eta correlation between e and B
199 TH2F *fPtCorrBouteout; // eta correlation between e and B
220 , fPtCorrDpDouteout()
224 , fPtCorrD0Douteout()
225 , fPtCorrDrestDinein()
226 , fPtCorrDrestDineout()
227 , fPtCorrDrestDoutein()
228 , fPtCorrDrestDouteout()
246 // default constructor
248 AliHistsComm(const AliHistsComm & p)
250 , fProcessID(p.fProcessID)
251 , fePtRatio(p.fePtRatio)
253 , fPtCorrDp(p.fPtCorrDp)
254 , fPtCorrD0(p.fPtCorrD0)
255 , fPtCorrDrest(p.fPtCorrDrest)
256 , fDePtRatio(p.fDePtRatio)
257 , feDistance(p.feDistance)
258 , fDeDistance(p.fDeDistance)
260 , fPtCorrDinein(p.fPtCorrDinein)
261 , fPtCorrDineout(p.fPtCorrDineout)
262 , fPtCorrDoutein(p.fPtCorrDoutein)
263 , fPtCorrDouteout(p.fPtCorrDouteout)
264 , fPtCorrDpDinein(p.fPtCorrDpDinein)
265 , fPtCorrDpDineout(p.fPtCorrDpDineout)
266 , fPtCorrDpDoutein(p.fPtCorrDpDoutein)
267 , fPtCorrDpDouteout(p.fPtCorrDpDouteout)
268 , fPtCorrD0Dinein(p.fPtCorrD0Dinein)
269 , fPtCorrD0Dineout(p.fPtCorrD0Dineout)
270 , fPtCorrD0Doutein(p.fPtCorrD0Doutein)
271 , fPtCorrD0Douteout(p.fPtCorrD0Douteout)
272 , fPtCorrDrestDinein(p.fPtCorrDrestDinein)
273 , fPtCorrDrestDineout(p.fPtCorrDrestDineout)
274 , fPtCorrDrestDoutein(p.fPtCorrDrestDoutein)
275 , fPtCorrDrestDouteout(p.fPtCorrDrestDouteout)
277 , fEtaCorrD(p.fEtaCorrD)
278 , fEtaCorrDp(p.fEtaCorrDp)
279 , fEtaCorrD0(p.fEtaCorrD0)
280 , fEtaCorrDrest(p.fEtaCorrDrest)
281 , fEtaCorrGD(p.fEtaCorrGD)
282 , fEtaCorrGDp(p.fEtaCorrGDp)
283 , fEtaCorrGD0(p.fEtaCorrGD0)
284 , fEtaCorrGDrest(p.fEtaCorrGDrest)
286 , fEtaCorrB(p.fEtaCorrB)
287 , fEtaCorrGB(p.fEtaCorrGB)
288 , fPtCorrBinein(p.fPtCorrBinein)
289 , fPtCorrBineout(p.fPtCorrBineout)
290 , fPtCorrBoutein(p.fPtCorrBoutein)
291 , fPtCorrBouteout(p.fPtCorrBouteout)
295 AliHistsComm &operator=(const AliHistsComm &)
297 // assignment operator, not yet implemented
300 void FillList(TList *l) const;
303 AliHists fHist[3][7][6][11]; // struct of histograms to store kinematics of given particles
304 AliHistsComm fHistComm[2][6]; // struct of additional histograms of given particles
305 TH2F *fhD[9]; // D meson pt,Y spectra
307 TList *fQAhistos; // Container for QA histos
308 AliHFEcollection *fMCQACollection; //! Tasks own QA collection
309 TParticle *fHeavyQuark[50]; //! store pointer of heavy flavour quark
310 Int_t fIsHeavy[2]; // count of heavy flavour
311 Int_t fNparents; // number of heavy hadrons to be considered
312 Int_t fParentSelect[2][7]; // heavy hadron species
314 Double_t fElecBackgroundFactor[kBgLevels][kCentBins][kElecBgSpecies][kBgPtBins]; // Electron background factors
315 Double_t fBinLimit[kBgPtBins+1]; // Electron background bins
318 Int_t fCentrality; // Centrality
319 Int_t fPerCentrality; // Centrality percentile
320 Bool_t fIsPbPb; // Analysis Type: pp or PbPb
321 Bool_t fIsppMultiBin; // pp multiplicity bin analysis
322 Int_t fContainerStep; // step the weighting factor called
323 Bool_t fIsDebugStreamerON; // check if the debugstreamer is on
325 Double_t fRecPt; //reconstructed pt
326 Double_t fRecEta; //reconstructed eta
327 Double_t fRecPhi; //reconstructed phi
328 Double_t fLyrhit; //its layer hit
329 Double_t fLyrstat; //its layer status
331 Double_t fHfeImpactR; //absolute impact parameter R
332 Double_t fHfeImpactnsigmaR; //absolute impact parameter sigma R
334 TTreeSRedirector *fTreeStream; //! TreeStream
336 ClassDef(AliHFEmcQA,1);