1 /**************************************************************************
2 * Copyright(c) 1998-2009, 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 //-----------------------------------------------------------------------
17 // Class for HF corrections as a function of many variables and step
18 // Author : C. Zampolli, CERN
19 // D. Caffarri, Univ & INFN Padova caffarri@pd.infn.it
20 // Base class for HF Unfolding - agrelli@uu.nl
21 //-----------------------------------------------------------------------
23 #include "AliAODRecoDecayHF2Prong.h"
24 #include "AliAODMCParticle.h"
25 #include "AliAODEvent.h"
26 #include "TClonesArray.h"
27 #include "AliCFVertexingHF.h"
28 #include "AliESDtrack.h"
29 #include "TDatabasePDG.h"
31 #include "AliCFVertexingHF2Prong.h"
32 #include "AliCFContainer.h"
33 #include "AliCFTaskVertexingHF.h"
35 ClassImp(AliCFVertexingHF2Prong)
38 //_________________________________________
39 AliCFVertexingHF2Prong::AliCFVertexingHF2Prong(TClonesArray *mcArray, UShort_t originDselection):
40 AliCFVertexingHF(mcArray, originDselection)
47 fPtAccCut=new Float_t[fProngs];
48 fEtaAccCut=new Float_t[fProngs];
49 for(Int_t iP=0; iP<fProngs; iP++){
57 //_____________________________________
58 AliCFVertexingHF2Prong& AliCFVertexingHF2Prong::operator=(const AliCFVertexingHF2Prong& c)
65 AliCFVertexingHF::operator=(c);
70 //__________________________________________
71 Bool_t AliCFVertexingHF2Prong::SetRecoCandidateParam(AliAODRecoDecayHF *recoCand)
74 // setting the recontructed candidate
77 Bool_t bSignAssoc = kFALSE;
78 fRecoCandidate = recoCand;
79 if (!fRecoCandidate) {
80 AliError("fRecoCandidate not found, problem in assignement\n");
84 if (fRecoCandidate->GetPrimaryVtx()) AliDebug(3,"fReco Candidate has a pointer to PrimVtx\n");
85 if (recoCand->GetPrimaryVtx()) AliDebug(3,"Reco Cand has a pointer to PrimVtx\n");
88 Int_t pdgDgD0toKpi[2]={321,211};
89 Int_t nentries = fmcArray->GetEntriesFast();
91 AliDebug(3,Form("nentries = %d\n", nentries));
93 Int_t mcLabel = fRecoCandidate->MatchToMC(pdgCand,fmcArray,2,pdgDgD0toKpi);
94 if (mcLabel == -1) return bSignAssoc;
96 if (fRecoCandidate->NumberOfFakeDaughters()>0){
97 fFake = 0; // fake candidate
98 if (fFakeSelection==1) return bSignAssoc;
100 if (fRecoCandidate->NumberOfFakeDaughters()==0){
101 fFake = 2; // non-fake candidate
102 if (fFakeSelection==2) return bSignAssoc;
106 fmcPartCandidate = dynamic_cast<AliAODMCParticle*>(fmcArray->At(fmcLabel));
107 if (!fmcPartCandidate){
108 AliDebug(3,"No part candidate");
115 //______________________________________________
116 Bool_t AliCFVertexingHF2Prong::GetGeneratedValuesFromMCParticle(Double_t* vectorMC)
119 // collecting all the necessary info (pt, y, cosThetaStar, ptPi, ptKa, cT) from MC particle
122 Bool_t bGenValues = kFALSE;
123 Double_t vtx1[3] = {0,0,0}; // primary vertex
124 Double_t vtx2daughter0[3] = {0,0,0}; // secondary vertex from daughter 0
125 Double_t vtx2daughter1[3] = {0,0,0}; // secondary vertex from daughter 1
126 fmcPartCandidate->XvYvZv(vtx1); // cm
128 Int_t daughter0 = fmcPartCandidate->GetDaughter(0);
129 Int_t daughter1 = fmcPartCandidate->GetDaughter(1);
130 AliAODMCParticle* mcPartDaughter0 = dynamic_cast<AliAODMCParticle*>(fmcArray->At(daughter0));
131 AliAODMCParticle* mcPartDaughter1 = dynamic_cast<AliAODMCParticle*>(fmcArray->At(daughter1));
132 if(!mcPartDaughter0 || !mcPartDaughter1) return bGenValues;
134 // getting vertex from daughters
135 mcPartDaughter0->XvYvZv(vtx2daughter0); // cm
136 mcPartDaughter1->XvYvZv(vtx2daughter1); //cm
137 if (TMath::Abs(vtx2daughter0[0] - vtx2daughter1[0]) > 1E-5 || TMath::Abs(vtx2daughter0[1] - vtx2daughter1[1]) > 1E-5 || TMath::Abs(vtx2daughter0[2] - vtx2daughter1[2])>1E-5) {
138 AliError("Daughters have different secondary vertex, skipping the track");
144 // always instantiate the AliAODRecoDecay with the positive daughter first, the negative second
145 AliAODMCParticle* positiveDaugh = mcPartDaughter0;
146 AliAODMCParticle* negativeDaugh = mcPartDaughter1;
147 if (mcPartDaughter0->GetPdgCode()<0 && mcPartDaughter1->GetPdgCode()>0){
148 // inverting in case the positive daughter is the second one
149 positiveDaugh = mcPartDaughter1;
150 negativeDaugh = mcPartDaughter0;
152 // getting the momentum from the daughters
153 Double_t px[2] = {positiveDaugh->Px(), negativeDaugh->Px()};
154 Double_t py[2] = {positiveDaugh->Py(), negativeDaugh->Py()};
155 Double_t pz[2] = {positiveDaugh->Pz(), negativeDaugh->Pz()};
157 Double_t d0[2] = {0.,0.};
159 AliAODRecoDecayHF* decay = new AliAODRecoDecayHF(vtx1,vtx2daughter0,nprongs,charge,px,py,pz,d0);
161 Double_t cosThetaStar = 0.;
162 Double_t cosThetaStarD0 = 0.;
163 Double_t cosThetaStarD0bar = 0.;
164 cosThetaStarD0 = decay->CosThetaStar(1,421,211,321);
165 cosThetaStarD0bar = decay->CosThetaStar(0,421,321,211);
166 if (fmcPartCandidate->GetPdgCode() == 421){ // D0
167 AliDebug(3, Form("D0, with pdgprong0 = %d, pdgprong1 = %d",mcPartDaughter0->GetPdgCode(),mcPartDaughter1->GetPdgCode()));
168 cosThetaStar = cosThetaStarD0;
170 else if (fmcPartCandidate->GetPdgCode() == -421){ // D0bar{
171 AliDebug(3, Form("D0bar, with pdgprong0 = %d, pdgprong1 = %d",mcPartDaughter0->GetPdgCode(),mcPartDaughter1->GetPdgCode()));
172 cosThetaStar = cosThetaStarD0bar;
175 AliWarning("There are problems!! particle was expected to be either a D0 or a D0bar, check...");
178 if (cosThetaStar < -1 || cosThetaStar > 1) {
179 AliWarning("Invalid value for cosine Theta star, returning");
184 Double_t cT = decay->Ct(421);
185 // get the pT of the daughters
189 if (TMath::Abs(mcPartDaughter0->GetPdgCode()) == 211) {
190 pTpi = mcPartDaughter0->Pt();
191 pTK = mcPartDaughter1->Pt();
194 pTpi = mcPartDaughter1->Pt();
195 pTK = mcPartDaughter0->Pt();
198 switch (fConfiguration){
199 case AliCFTaskVertexingHF::kSnail:
200 vectorMC[0] = fmcPartCandidate->Pt();
201 vectorMC[1] = fmcPartCandidate->Y() ;
202 vectorMC[2] = cosThetaStar ;
205 vectorMC[5] = cT*1.E4 ; // in micron
206 vectorMC[6] = 0.; // dummy value for dca, meaningless in MC
207 vectorMC[7] = -80000.; // dummy value for d0pixd0K, meaningless in MC, in micron^2
208 vectorMC[8] = 1.01; // dummy value for cosPointing, meaningless in MC
209 vectorMC[9] = fmcPartCandidate->Phi();
210 vectorMC[10] = fzMCVertex; // z of reconstructed of primary vertex
211 vectorMC[11] = fCentValue; //reconstructed centrality
212 vectorMC[12] = 1.; // fake: always filling with 1 at MC level
213 vectorMC[13] = 1.01; // dummy value for cosPointingXY multiplicity
214 vectorMC[14] = 0.; // dummy value for NormalizedDecayLengthXY multiplicity
215 vectorMC[15] = fMultiplicity; // reconstructed multiplicity
217 case AliCFTaskVertexingHF::kCheetah:
218 vectorMC[0] = fmcPartCandidate->Pt();
219 vectorMC[1] = fmcPartCandidate->Y() ;
220 vectorMC[2] = cT*1.E4; // in micron
221 vectorMC[3] = fmcPartCandidate->Phi();
222 vectorMC[4] = fzMCVertex;
223 vectorMC[5] = fCentValue; // dummy value for dca, meaningless in MC
224 vectorMC[6] = 1. ; // fake: always filling with 1 at MC level
225 vectorMC[7] = fMultiplicity; // dummy value for d0pi, meaningless in MC, in micron
232 //____________________________________________
233 Bool_t AliCFVertexingHF2Prong::GetRecoValuesFromCandidate(Double_t *vectorReco) const
236 // Getting the reconstructed values from the candidate
239 Bool_t bFillRecoValues=kFALSE;
241 AliAODRecoDecayHF2Prong *d0toKpi = (AliAODRecoDecayHF2Prong*)fRecoCandidate;
243 if (d0toKpi->GetPrimaryVtx())AliDebug(3,"d0toKpi has primary vtx\n");
244 if (fRecoCandidate->GetPrimaryVtx())AliDebug(3,"fRecoCandidate has primary vtx\n");
246 Double_t pt = d0toKpi->Pt();
247 Double_t rapidity = d0toKpi->YD0();
249 Double_t cosThetaStar = 9999.;
252 Double_t dca = d0toKpi->GetDCA();
255 Double_t d0xd0 = d0toKpi->Prodd0d0();
256 Double_t cosPointingAngle = d0toKpi->CosPointingAngle();
257 Double_t phi = d0toKpi->Phi();
258 Int_t pdgCode = fmcPartCandidate->GetPdgCode();
259 Double_t cosPointingAngleXY = d0toKpi->CosPointingAngleXY();
260 Double_t normDecayLengthXY = d0toKpi->NormalizedDecayLengthXY();
263 cosThetaStar = d0toKpi->CosThetaStarD0();
264 pTpi = d0toKpi->PtProng(0);
265 pTK = d0toKpi->PtProng(1);
266 d0pi = d0toKpi->Getd0Prong(0);
267 d0K = d0toKpi->Getd0Prong(1);
268 invMass=d0toKpi->InvMassD0();
271 cosThetaStar = d0toKpi->CosThetaStarD0bar();
272 pTpi = d0toKpi->PtProng(1);
273 pTK = d0toKpi->PtProng(0);
274 d0pi = d0toKpi->Getd0Prong(1);
275 d0K = d0toKpi->Getd0Prong(0);
276 invMass= d0toKpi->InvMassD0bar();
279 Double_t cT = d0toKpi->CtD0();
281 switch (fConfiguration){
282 case AliCFTaskVertexingHF::kSnail:
284 vectorReco[1] = rapidity;
285 vectorReco[2] = cosThetaStar;
286 vectorReco[3] = pTpi;
288 vectorReco[5] = cT*1.E4; // in micron
289 vectorReco[6] = dca*1.E4; // in micron
290 vectorReco[7] = d0xd0*1.E8; // in micron^2
291 vectorReco[8] = cosPointingAngle;
293 vectorReco[10] = fzPrimVertex; // z of reconstructed of primary vertex
294 vectorReco[11] = fCentValue; //reconstructed centrality
295 vectorReco[12] = fFake; // whether the reconstructed candidate was a fake (fFake = 0) or not (fFake = 2)
296 vectorReco[13] = cosPointingAngleXY;
297 vectorReco[14] = normDecayLengthXY; // in cm
298 vectorReco[15] = fMultiplicity; // reconstructed multiplicity
300 case AliCFTaskVertexingHF::kCheetah:
302 vectorReco[1] = rapidity ;
303 vectorReco[2] = cT*1.E4; // in micron
305 vectorReco[4] = fzPrimVertex;
306 vectorReco[5] = fCentValue;
307 vectorReco[6] = fFake ;
308 vectorReco[7] = fMultiplicity;
312 bFillRecoValues = kTRUE;
314 return bFillRecoValues;
317 //_____________________________________________________________
318 Bool_t AliCFVertexingHF2Prong::CheckMCChannelDecay() const
321 // checking the MC decay channel
323 Bool_t checkCD = kFALSE;
324 Int_t daughter0 = fmcPartCandidate->GetDaughter(0);
325 Int_t daughter1 = fmcPartCandidate->GetDaughter(1);
326 AliAODMCParticle* mcPartDaughter0 = dynamic_cast<AliAODMCParticle*>(fmcArray->At(daughter0));
327 AliAODMCParticle* mcPartDaughter1 = dynamic_cast<AliAODMCParticle*>(fmcArray->At(daughter1));
329 if (!mcPartDaughter0 || !mcPartDaughter1) {
330 AliDebug (2,"Problems in the MC Daughters\n");
334 if (!(TMath::Abs(mcPartDaughter0->GetPdgCode())==321 &&
335 TMath::Abs(mcPartDaughter1->GetPdgCode())==211) &&
336 !(TMath::Abs(mcPartDaughter0->GetPdgCode())==211 &&
337 TMath::Abs(mcPartDaughter1->GetPdgCode())==321)) {
338 AliDebug(2, "The D0 MC doesn't come from a Kpi decay, skipping!!");