]>
Commit | Line | Data |
---|---|---|
a65a7e70 | 1 | /************************************************************************** |
2 | * Copyright(c) 2007-2009, ALICE Experiment at CERN, All rights reserved. * | |
3 | * * | |
4 | * Author: The ALICE Off-line Project. * | |
5 | * Contributors are mentioned in the code where appropriate. * | |
6 | * * | |
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 | **************************************************************************/ | |
15 | /* $Id$ */ | |
16 | ||
17 | //------------------------------------------------------------------------- | |
18 | // Implementation of the on-the-fly v0 finder for ITS tracker MI | |
19 | // Origin: Marian Ivanov, CERN, Marian.Ivanov@cern.ch | |
20 | // Extraction from AliITStrackerMI: Andrea Dainese | |
21 | // Current support and development: | |
22 | //------------------------------------------------------------------------- | |
23 | ||
24 | #include <TMatrixD.h> | |
25 | #include <TTree.h> | |
26 | #include <TString.h> | |
27 | #include <TRandom.h> | |
28 | #include <TTreeStream.h> | |
29 | ||
30 | ||
31 | #include "AliLog.h" | |
32 | #include "AliESDVertex.h" | |
33 | #include "AliESDEvent.h" | |
34 | #include "AliESDtrack.h" | |
35 | #include "AliESDV0Params.h" | |
36 | #include "AliV0.h" | |
37 | #include "AliHelix.h" | |
38 | #include "AliITSRecPoint.h" | |
39 | #include "AliITSReconstructor.h" | |
40 | #include "AliITStrackerMI.h" | |
41 | #include "AliITSV0Finder.h" | |
42 | #include "AliKFParticle.h" | |
43 | #include "AliKFVertex.h" | |
44 | ||
45 | ClassImp(AliITSV0Finder) | |
46 | ||
47 | ||
48 | AliITSV0Finder::AliITSV0Finder(): | |
49 | fDebugStreamer(0) | |
50 | { | |
51 | //Default constructor | |
52 | ||
53 | if (AliITSReconstructor::GetRecoParam()->GetESDV0Params()->StreamLevel()>0) | |
54 | fDebugStreamer = new TTreeSRedirector("ITSdebug.root"); | |
55 | } | |
56 | ||
57 | //------------------------------------------------------------------------ | |
58 | void AliITSV0Finder::UpdateTPCV0(const AliESDEvent *event, | |
59 | AliITStrackerMI *tracker) | |
60 | { | |
61 | // | |
62 | //try to update, or reject TPC V0s | |
63 | // | |
64 | const Float_t kMinTgl2= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinTgl2(); | |
65 | ||
66 | TObjArray *trackHypothesys = tracker->GetTrackHypothesys(); | |
67 | ||
68 | Int_t nv0s = event->GetNumberOfV0s(); | |
69 | Int_t nitstracks = trackHypothesys->GetEntriesFast(); | |
70 | ||
71 | for (Int_t i=0;i<nv0s;i++){ | |
72 | AliESDv0 * vertex = event->GetV0(i); | |
73 | Int_t ip = vertex->GetIndex(0); | |
74 | Int_t im = vertex->GetIndex(1); | |
75 | // | |
76 | TObjArray * arrayp = (ip<nitstracks) ? (TObjArray*)trackHypothesys->At(ip):0; | |
77 | TObjArray * arraym = (im<nitstracks) ? (TObjArray*)trackHypothesys->At(im):0; | |
78 | AliITStrackMI * trackp = (arrayp!=0) ? (AliITStrackMI*)arrayp->At(0):0; | |
79 | AliITStrackMI * trackm = (arraym!=0) ? (AliITStrackMI*)arraym->At(0):0; | |
80 | // | |
81 | // | |
82 | if (trackp){ | |
83 | if (trackp->GetNumberOfClusters()+trackp->GetNDeadZone()>5.5){ | |
84 | if (trackp->GetConstrain()&&trackp->GetChi2MIP(0)<3) vertex->SetStatus(-100); | |
85 | if (!trackp->GetConstrain()&&trackp->GetChi2MIP(0)<2) vertex->SetStatus(-100); | |
86 | } | |
87 | } | |
88 | ||
89 | if (trackm){ | |
90 | if (trackm->GetNumberOfClusters()+trackm->GetNDeadZone()>5.5){ | |
91 | if (trackm->GetConstrain()&&trackm->GetChi2MIP(0)<3) vertex->SetStatus(-100); | |
92 | if (!trackm->GetConstrain()&&trackm->GetChi2MIP(0)<2) vertex->SetStatus(-100); | |
93 | } | |
94 | } | |
95 | if (vertex->GetStatus()==-100) continue; | |
96 | // | |
97 | Double_t xrp[3]; vertex->GetXYZ(xrp[0],xrp[1],xrp[2]); //I.B. | |
98 | Int_t clayer = tracker->GetNearestLayer(xrp); //I.B. | |
99 | vertex->SetNBefore(clayer); // | |
100 | vertex->SetChi2Before(9*clayer); // | |
101 | vertex->SetNAfter(6-clayer); // | |
102 | vertex->SetChi2After(0); // | |
103 | // | |
104 | if (clayer >1 ){ // calculate chi2 before vertex | |
105 | Float_t chi2p = 0, chi2m=0; | |
106 | // | |
107 | if (trackp){ | |
108 | for (Int_t ilayer=0;ilayer<clayer;ilayer++){ | |
109 | if (trackp->GetClIndex(ilayer)>=0){ | |
110 | chi2p+=trackp->GetDy(ilayer)*trackp->GetDy(ilayer)/(trackp->GetSigmaY(ilayer)*trackp->GetSigmaY(ilayer))+ | |
111 | trackp->GetDz(ilayer)*trackp->GetDz(ilayer)/(trackp->GetSigmaZ(ilayer)*trackp->GetSigmaZ(ilayer)); | |
112 | } | |
113 | else{ | |
114 | chi2p+=9; | |
115 | } | |
116 | } | |
117 | }else{ | |
118 | chi2p = 9*clayer; | |
119 | } | |
120 | // | |
121 | if (trackm){ | |
122 | for (Int_t ilayer=0;ilayer<clayer;ilayer++){ | |
123 | if (trackm->GetClIndex(ilayer)>=0){ | |
124 | chi2m+=trackm->GetDy(ilayer)*trackm->GetDy(ilayer)/(trackm->GetSigmaY(ilayer)*trackm->GetSigmaY(ilayer))+ | |
125 | trackm->GetDz(ilayer)*trackm->GetDz(ilayer)/(trackm->GetSigmaZ(ilayer)*trackm->GetSigmaZ(ilayer)); | |
126 | } | |
127 | else{ | |
128 | chi2m+=9; | |
129 | } | |
130 | } | |
131 | }else{ | |
132 | chi2m = 9*clayer; | |
133 | } | |
134 | vertex->SetChi2Before(TMath::Min(chi2p,chi2m)); | |
135 | if (TMath::Min(chi2p,chi2m)/Float_t(clayer)<4) vertex->SetStatus(-10); // track exist before vertex | |
136 | } | |
137 | ||
138 | if (clayer < 5 ){ // calculate chi2 after vertex | |
139 | Float_t chi2p = 0, chi2m=0; | |
140 | // | |
141 | if (trackp&&TMath::Abs(trackp->GetTgl())<kMinTgl2){ | |
142 | for (Int_t ilayer=clayer;ilayer<6;ilayer++){ | |
143 | if (trackp->GetClIndex(ilayer)>=0){ | |
144 | chi2p+=trackp->GetDy(ilayer)*trackp->GetDy(ilayer)/(trackp->GetSigmaY(ilayer)*trackp->GetSigmaY(ilayer))+ | |
145 | trackp->GetDz(ilayer)*trackp->GetDz(ilayer)/(trackp->GetSigmaZ(ilayer)*trackp->GetSigmaZ(ilayer)); | |
146 | } | |
147 | else{ | |
148 | chi2p+=9; | |
149 | } | |
150 | } | |
151 | }else{ | |
152 | chi2p = 0; | |
153 | } | |
154 | // | |
155 | if (trackm&&TMath::Abs(trackm->GetTgl())<kMinTgl2){ | |
156 | for (Int_t ilayer=clayer;ilayer<6;ilayer++){ | |
157 | if (trackm->GetClIndex(ilayer)>=0){ | |
158 | chi2m+=trackm->GetDy(ilayer)*trackm->GetDy(ilayer)/(trackm->GetSigmaY(ilayer)*trackm->GetSigmaY(ilayer))+ | |
159 | trackm->GetDz(ilayer)*trackm->GetDz(ilayer)/(trackm->GetSigmaZ(ilayer)*trackm->GetSigmaZ(ilayer)); | |
160 | } | |
161 | else{ | |
162 | chi2m+=9; | |
163 | } | |
164 | } | |
165 | }else{ | |
166 | chi2m = 0; | |
167 | } | |
168 | vertex->SetChi2After(TMath::Max(chi2p,chi2m)); | |
169 | if (TMath::Max(chi2m,chi2p)/Float_t(6-clayer)>9) vertex->SetStatus(-20); // track not found in ITS | |
170 | } | |
171 | } | |
172 | // | |
173 | } | |
174 | //------------------------------------------------------------------------ | |
175 | void AliITSV0Finder::FindV02(AliESDEvent *event, | |
176 | AliITStrackerMI *tracker) | |
177 | { | |
178 | // | |
179 | // V0 finder | |
180 | // | |
181 | // Cuts on DCA - R dependent | |
182 | // max distance DCA between 2 tracks cut | |
183 | // maxDist = TMath::Min(kMaxDist,kMaxDist0+pvertex->GetRr()*kMaxDist); | |
184 | // | |
185 | const Bool_t kCheckPropagate = kFALSE; | |
186 | const Float_t kMaxDist0 = AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMaxDist0(); | |
187 | const Float_t kMaxDist1 = AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMaxDist1(); | |
188 | const Float_t kMaxDist = AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMaxDist(); | |
189 | const Float_t kMinPointAngle = AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinPointAngle(); | |
190 | const Float_t kMinPointAngle2 = AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinPointAngle2(); | |
191 | const Float_t kMinR = AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinR(); | |
192 | const Float_t kMaxR = AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMaxR(); | |
193 | const Float_t kMinPABestConst = AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinPABestConst(); | |
194 | const Float_t kMaxRBestConst = AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMaxRBestConst(); | |
195 | const Float_t kCausality0Cut = AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetCausality0Cut(); | |
196 | const Float_t kLikelihood01Cut = AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetLikelihood01Cut(); | |
197 | const Float_t kLikelihood1Cut = AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetLikelihood1Cut(); | |
198 | const Float_t kCombinedCut = AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetCombinedCut(); | |
199 | const Float_t kMinClFullTrk= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinClFullTrk(); | |
200 | const Float_t kMinTgl0= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinTgl0(); | |
201 | const Float_t kMinRTgl0= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinRTgl0(); | |
202 | ||
203 | const Float_t kMinNormDistForbTgl0= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinNormDistForbTgl0(); | |
204 | const Float_t kMinClForb0= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinClForb0(); | |
205 | const Float_t kMinNormDistForb1= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinNormDistForb1(); | |
206 | const Float_t kMinNormDistForb2= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinNormDistForb2(); | |
207 | const Float_t kMinNormDistForb3= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinNormDistForb3(); | |
208 | const Float_t kMinNormDistForb4= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinNormDistForb4(); | |
209 | const Float_t kMinNormDistForb5= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinNormDistForb5(); | |
210 | const Float_t kMinNormDistForbProt= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinNormDistForbProt(); | |
211 | const Float_t kMaxPidProbPionForb= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMaxPidProbPionForb(); | |
212 | ||
213 | const Float_t kMinRTPCdensity= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinRTPCdensity(); | |
214 | const Float_t kMaxRTPCdensity0= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMaxRTPCdensity0(); | |
215 | const Float_t kMaxRTPCdensity10= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMaxRTPCdensity10(); | |
216 | const Float_t kMaxRTPCdensity20= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMaxRTPCdensity20(); | |
217 | const Float_t kMaxRTPCdensity30= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMaxRTPCdensity30(); | |
218 | ||
219 | ||
220 | const Float_t kMinTPCdensity= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinTPCdensity(); | |
221 | const Float_t kMinTgl1= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinTgl1(); | |
222 | ||
223 | const Float_t kMinchi2before0= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinchi2before0(); | |
224 | const Float_t kMinchi2before1= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinchi2before1(); | |
225 | const Float_t kMinchi2after0= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinchi2after0(); | |
226 | const Float_t kMinchi2after1= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMinchi2after1(); | |
227 | const Float_t kAddchi2SharedCl= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetAddchi2SharedCl(); | |
228 | const Float_t kAddchi2NegCl0= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetAddchi2NegCl0(); | |
229 | const Float_t kAddchi2NegCl1= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetAddchi2NegCl1(); | |
230 | ||
231 | const Float_t kSigp0Par0= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetSigp0Par0(); | |
232 | const Float_t kSigp0Par1= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetSigp0Par1(); | |
233 | const Float_t kSigp0Par2= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetSigp0Par2(); | |
234 | ||
235 | const Float_t kSigpPar0= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetSigpPar0(); | |
236 | const Float_t kSigpPar1= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetSigpPar1(); | |
237 | const Float_t kSigpPar2= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetSigpPar2(); | |
238 | const Float_t kMaxDcaLh0= AliITSReconstructor::GetRecoParam()->GetESDV0Params()->GetMaxDcaLh0(); | |
239 | ||
240 | ||
241 | TObjArray *trackHypothesys = tracker->GetTrackHypothesys(); | |
242 | TObjArray *bestHypothesys = tracker->GetBestHypothesys(); | |
243 | TObjArray *originalTracks = tracker->GetOriginal(); | |
244 | // | |
245 | // | |
246 | TTreeSRedirector &cstream = *(tracker->GetDebugStreamer()); | |
247 | Int_t ntracks = event->GetNumberOfTracks(); | |
248 | Int_t nitstracks = originalTracks->GetEntriesFast(); | |
249 | originalTracks->Expand(ntracks); | |
250 | trackHypothesys->Expand(ntracks); | |
251 | bestHypothesys->Expand(ntracks); | |
252 | // | |
253 | AliHelix * helixes = new AliHelix[ntracks+2]; | |
254 | TObjArray trackarray(ntracks+2); //array with tracks - with vertex constrain | |
255 | TObjArray trackarrayc(ntracks+2); //array of "best tracks" - without vertex constrain | |
256 | TObjArray trackarrayl(ntracks+2); //array of "longest tracks" - without vertex constrain | |
257 | //Change from Bool_t to Int_t for optimization | |
258 | // Int_t forbN=0; | |
259 | // Int_t * forbidden = new Int_t [ntracks+2]; | |
260 | Bool_t * forbidden = new Bool_t [ntracks+2]; | |
261 | Int_t *itsmap = new Int_t [ntracks+2]; | |
262 | Float_t *dist = new Float_t[ntracks+2]; | |
263 | Float_t *normdist0 = new Float_t[ntracks+2]; | |
264 | Float_t *normdist1 = new Float_t[ntracks+2]; | |
265 | Float_t *normdist = new Float_t[ntracks+2]; | |
266 | Float_t *norm = new Float_t[ntracks+2]; | |
267 | Float_t *maxr = new Float_t[ntracks+2]; | |
268 | Float_t *minr = new Float_t[ntracks+2]; | |
269 | Float_t *minPointAngle= new Float_t[ntracks+2]; | |
270 | // | |
271 | AliV0 *pvertex = new AliV0; | |
272 | AliITStrackMI * dummy= new AliITStrackMI; | |
273 | dummy->SetLabel(0); | |
274 | AliITStrackMI trackat0; //temporary track for DCA calculation | |
275 | // | |
2942f542 | 276 | Float_t primvertex[3]={static_cast<Float_t>(tracker->GetX()),static_cast<Float_t>(tracker->GetY()),static_cast<Float_t>(tracker->GetZ())}; |
a65a7e70 | 277 | // |
278 | // make ITS - ESD map | |
279 | // | |
280 | for (Int_t itrack=0;itrack<ntracks+2;itrack++) { | |
281 | itsmap[itrack] = -1; | |
282 | // forbidden[itrack] = 0; | |
283 | forbidden[itrack] = kFALSE; | |
284 | maxr[itrack] = kMaxR; | |
285 | minr[itrack] = kMinR; | |
286 | minPointAngle[itrack] = kMinPointAngle; | |
287 | } | |
288 | for (Int_t itrack=0;itrack<nitstracks;itrack++){ | |
289 | AliITStrackMI * original = (AliITStrackMI*)(originalTracks->At(itrack)); | |
290 | Int_t esdindex = original->GetESDtrack()->GetID(); | |
291 | itsmap[esdindex] = itrack; | |
292 | } | |
293 | // | |
294 | // create ITS tracks from ESD tracks if not done before | |
295 | // | |
296 | for (Int_t itrack=0;itrack<ntracks;itrack++){ | |
297 | if (itsmap[itrack]>=0) continue; | |
298 | AliITStrackMI * tpctrack = new AliITStrackMI(*(event->GetTrack(itrack))); | |
299 | //tpctrack->fD[0] = tpctrack->GetD(GetX(),GetY()); | |
300 | //tpctrack->fD[1] = tpctrack->GetZat(GetX())-GetZ(); | |
301 | tpctrack->GetDZ(tracker->GetX(),tracker->GetY(),tracker->GetZ(),tpctrack->GetDP()); //I.B. | |
302 | if (tpctrack->GetD(0)<20 && tpctrack->GetD(1)<20){ | |
303 | // tracks which can reach inner part of ITS | |
304 | // propagate track to outer its volume - with correction for material | |
305 | AliITStrackerMI::CorrectForTPCtoITSDeadZoneMaterial(tpctrack); | |
306 | } | |
307 | itsmap[itrack] = nitstracks; | |
308 | originalTracks->AddAt(tpctrack,nitstracks); | |
309 | nitstracks++; | |
310 | } | |
311 | // | |
312 | // fill temporary arrays | |
313 | // | |
314 | for (Int_t itrack=0;itrack<ntracks;itrack++){ | |
315 | AliESDtrack * esdtrack = event->GetTrack(itrack); | |
316 | Int_t itsindex = itsmap[itrack]; | |
317 | AliITStrackMI *original = (AliITStrackMI*)originalTracks->At(itsmap[itrack]); | |
318 | if (!original) continue; | |
319 | AliITStrackMI *bestConst = 0; | |
320 | AliITStrackMI *bestLong = 0; | |
321 | AliITStrackMI *best = 0; | |
322 | // | |
323 | // | |
324 | TObjArray * array = (TObjArray*) trackHypothesys->At(itsindex); | |
325 | Int_t hentries = (array==0) ? 0 : array->GetEntriesFast(); | |
326 | // Get best track with vertex constrain | |
327 | for (Int_t ih=0;ih<hentries;ih++){ | |
328 | AliITStrackMI * trackh = (AliITStrackMI*)array->At(ih); | |
329 | if (!trackh->GetConstrain()) continue; | |
330 | if (!bestConst) bestConst = trackh; | |
331 | if (trackh->GetNumberOfClusters()>kMinClFullTrk ){ | |
332 | bestConst = trackh; // full track - with minimal chi2 | |
333 | break; | |
334 | } | |
335 | if (trackh->GetNumberOfClusters()+trackh->GetNDeadZone()<=bestConst->GetNumberOfClusters()+bestConst->GetNDeadZone()) continue; | |
336 | bestConst = trackh; | |
337 | break; | |
338 | } | |
339 | // Get best long track without vertex constrain and best track without vertex constrain | |
340 | for (Int_t ih=0;ih<hentries;ih++){ | |
341 | AliITStrackMI * trackh = (AliITStrackMI*)array->At(ih); | |
342 | if (trackh->GetConstrain()) continue; | |
343 | if (!best) best = trackh; | |
344 | if (!bestLong) bestLong = trackh; | |
345 | if (trackh->GetNumberOfClusters()>kMinClFullTrk){ | |
346 | bestLong = trackh; // full track - with minimal chi2 | |
347 | break; | |
348 | } | |
349 | if (trackh->GetNumberOfClusters()+trackh->GetNDeadZone()<=bestLong->GetNumberOfClusters()+bestLong->GetNDeadZone()) continue; | |
350 | bestLong = trackh; | |
351 | } | |
352 | if (!best) { | |
353 | best = original; | |
354 | bestLong = original; | |
355 | } | |
356 | //I.B. trackat0 = *bestLong; | |
357 | new (&trackat0) AliITStrackMI(*bestLong); | |
358 | Double_t xx,yy,zz,alpha; | |
359 | if (!bestLong->GetGlobalXYZat(bestLong->GetX(),xx,yy,zz)) continue; | |
360 | ||
361 | ||
362 | alpha = TMath::ATan2(yy,xx); | |
363 | // if (!trackat0.Propagate(alpha,0)) continue; | |
364 | // trackat0.Propagate(alpha,0); //PH The check on the return value is temporarily disabled (bug 45751) | |
365 | if(!trackat0.Propagate(alpha,0) && kCheckPropagate)continue; | |
366 | // calculate normalized distances to the vertex | |
367 | // | |
368 | Float_t ptfac = (1.+100.*TMath::Abs(trackat0.GetC())); | |
369 | if ( bestLong->GetNumberOfClusters()>3 ){ | |
370 | dist[itsindex] = trackat0.GetY(); | |
371 | norm[itsindex] = ptfac*TMath::Sqrt(trackat0.GetSigmaY2()); | |
372 | normdist0[itsindex] = TMath::Abs(trackat0.GetY()/norm[itsindex]); | |
373 | normdist1[itsindex] = TMath::Abs((trackat0.GetZ()-primvertex[2])/(ptfac*TMath::Sqrt(trackat0.GetSigmaZ2()))); | |
374 | normdist[itsindex] = TMath::Sqrt(normdist0[itsindex]*normdist0[itsindex]+normdist1[itsindex]*normdist1[itsindex]); | |
375 | if (!bestConst){ | |
376 | if (bestLong->GetNumberOfClusters()+bestLong->GetNDeadZone()<6) normdist[itsindex]*=2.; | |
377 | if (bestLong->GetNumberOfClusters()+bestLong->GetNDeadZone()<5) normdist[itsindex]*=2.; | |
378 | if (bestLong->GetNumberOfClusters()+bestLong->GetNDeadZone()<4) normdist[itsindex]*=2.; | |
379 | }else{ | |
380 | if (bestConst->GetNumberOfClusters()+bestConst->GetNDeadZone()<6) normdist[itsindex]*=1.5; | |
381 | if (bestConst->GetNumberOfClusters()+bestConst->GetNDeadZone()<5) normdist[itsindex]*=1.5; | |
382 | } | |
383 | } | |
384 | else{ | |
385 | if (bestConst&&bestConst->GetNumberOfClusters()+bestConst->GetNDeadZone()>4.5){ | |
386 | dist[itsindex] = bestConst->GetD(0); | |
387 | norm[itsindex] = bestConst->GetDnorm(0); | |
388 | normdist0[itsindex] = TMath::Abs(bestConst->GetD(0)/norm[itsindex]); | |
389 | normdist1[itsindex] = TMath::Abs(bestConst->GetD(0)/norm[itsindex]); | |
390 | normdist[itsindex] = TMath::Sqrt(normdist0[itsindex]*normdist0[itsindex]+normdist1[itsindex]*normdist1[itsindex]); | |
391 | }else{ | |
392 | dist[itsindex] = trackat0.GetY(); | |
393 | norm[itsindex] = ptfac*TMath::Sqrt(trackat0.GetSigmaY2()); | |
394 | normdist0[itsindex] = TMath::Abs(trackat0.GetY()/norm[itsindex]); | |
395 | normdist1[itsindex] = TMath::Abs((trackat0.GetZ()-primvertex[2])/(ptfac*TMath::Sqrt(trackat0.GetSigmaZ2()))); | |
396 | normdist[itsindex] = TMath::Sqrt(normdist0[itsindex]*normdist0[itsindex]+normdist1[itsindex]*normdist1[itsindex]); | |
397 | if (TMath::Abs(trackat0.GetTgl())>kMinTgl0){ | |
398 | if (normdist[itsindex]<kMinNormDistForbTgl0){ | |
399 | // forbN=1; | |
400 | // forbidden[itsindex]+=1<<forbN; | |
401 | forbidden[itsindex]=kTRUE; | |
402 | } | |
403 | if (normdist[itsindex]>kMinNormDistForbTgl0) { | |
404 | minr[itsindex] = TMath::Max(kMinRTgl0,minr[itsindex]); | |
405 | } | |
406 | } | |
407 | } | |
408 | } | |
409 | ||
410 | ||
411 | // | |
412 | //----------------------------------------------------------- | |
413 | //Forbid primary track candidates - | |
414 | // | |
415 | //treetr->SetAlias("forbidden0","Tr0.fN<4&&Tr1.fN+Tr1.fNDeadZone>4.5"); | |
416 | //treetr->SetAlias("forbidden1","ND<3&&Tr1.fN+Tr1.fNDeadZone>5.5"); | |
417 | //treetr->SetAlias("forbidden2","ND<2&&Tr1.fClIndex[0]>0&&Tr1.fClIndex[0]>0"); | |
418 | //treetr->SetAlias("forbidden3","ND<1&&Tr1.fClIndex[0]>0"); | |
419 | //treetr->SetAlias("forbidden4","ND<4&&Tr1.fNormChi2[0]<2"); | |
420 | //treetr->SetAlias("forbidden5","ND<5&&Tr1.fNormChi2[0]<1"); | |
421 | //----------------------------------------------------------- | |
422 | if (bestConst){ | |
423 | if (bestLong->GetNumberOfClusters()<4 && bestConst->GetNumberOfClusters()+bestConst->GetNDeadZone()>kMinClForb0){ | |
424 | // forbN=2; | |
425 | // forbidden[itsindex]+=1<<forbN; | |
426 | forbidden[itsindex]=kTRUE; | |
427 | } | |
428 | if (normdist[itsindex]<kMinNormDistForb1 && bestConst->GetNumberOfClusters()+bestConst->GetNDeadZone()>5.5){ | |
429 | // forbN=3; | |
430 | // forbidden[itsindex]+=1<<forbN; | |
431 | forbidden[itsindex]=kTRUE; | |
432 | } | |
433 | if (normdist[itsindex]<kMinNormDistForb2 && bestConst->GetClIndex(0)>=0 && bestConst->GetClIndex(1)>=0 ){ | |
434 | // forbN=4; | |
435 | // forbidden[itsindex]+=1<<forbN; | |
436 | forbidden[itsindex]=kTRUE; | |
437 | } | |
438 | if (normdist[itsindex]<kMinNormDistForb3 && bestConst->GetClIndex(0)>=0){ | |
439 | // forbN=5; | |
440 | // forbidden[itsindex]+=1<<forbN; | |
441 | forbidden[itsindex]=kTRUE; | |
442 | } | |
443 | if (normdist[itsindex]<kMinNormDistForb4 && bestConst->GetNormChi2(0)<2){ | |
444 | // forbN=6; | |
445 | // forbidden[itsindex]+=1<<forbN; | |
446 | forbidden[itsindex]=kTRUE; | |
447 | } | |
448 | if (normdist[itsindex]<kMinNormDistForb5 && bestConst->GetNormChi2(0)<1){ | |
449 | // forbN=7; | |
450 | // forbidden[itsindex]+=1<<forbN; | |
451 | forbidden[itsindex]=kTRUE; | |
452 | } | |
453 | if (bestConst->GetNormChi2(0)<2.5) { | |
454 | minPointAngle[itsindex]= kMinPABestConst; | |
455 | maxr[itsindex] = kMaxRBestConst; | |
456 | } | |
457 | } | |
458 | // | |
459 | //forbid daughter kink candidates | |
460 | // | |
461 | if (esdtrack->GetKinkIndex(0)>0) forbidden[itsindex] = kTRUE; | |
462 | Bool_t isElectron = kTRUE; | |
463 | Bool_t isProton = kTRUE; | |
464 | Double_t pid[5]; | |
465 | esdtrack->GetESDpid(pid); | |
466 | for (Int_t i=1;i<5;i++){ | |
467 | if (pid[0]<pid[i]) isElectron= kFALSE; | |
468 | if (pid[4]<pid[i]) isProton= kFALSE; | |
469 | } | |
470 | if (isElectron){ | |
471 | forbidden[itsindex]=kFALSE; | |
472 | normdist[itsindex]*=-1; | |
473 | } | |
474 | if (isProton){ | |
475 | if (normdist[itsindex]>kMinNormDistForbProt) forbidden[itsindex]=kFALSE; | |
476 | normdist[itsindex]*=-1; | |
477 | } | |
478 | ||
479 | // We allow all tracks that are not pions | |
480 | if( (pid[1]+pid[2])< kMaxPidProbPionForb ){ | |
481 | forbidden[itsindex]=kFALSE; | |
482 | } | |
483 | ||
484 | // | |
485 | // Causality cuts in TPC volume | |
486 | // | |
487 | if (esdtrack->GetTPCdensity(0,10) >kMinTPCdensity) maxr[itsindex] = TMath::Min(Float_t(kMaxRTPCdensity0),maxr[itsindex]); | |
488 | if (esdtrack->GetTPCdensity(10,30)>kMinTPCdensity) maxr[itsindex] = TMath::Min(Float_t(kMaxRTPCdensity10),maxr[itsindex]); | |
489 | if (esdtrack->GetTPCdensity(20,40)>kMinTPCdensity) maxr[itsindex] = TMath::Min(Float_t(kMaxRTPCdensity20),maxr[itsindex]); | |
490 | if (esdtrack->GetTPCdensity(30,50)>kMinTPCdensity) maxr[itsindex] = TMath::Min(Float_t(kMaxRTPCdensity30),maxr[itsindex]); | |
491 | // | |
492 | if (esdtrack->GetTPCdensity(0,60)<0.4&&bestLong->GetNumberOfClusters()<3) minr[itsindex]=kMinRTPCdensity; | |
493 | // | |
494 | // | |
495 | ||
496 | if (AliITSReconstructor::GetRecoParam()->GetESDV0Params()->StreamLevel()>0){ | |
497 | cstream<<"Track"<< | |
498 | "Tr0.="<<best<< | |
499 | "Tr1.="<<((bestConst)? bestConst:dummy)<< | |
500 | "Tr2.="<<bestLong<< | |
501 | "Tr3.="<<&trackat0<< | |
502 | "Esd.="<<esdtrack<< | |
503 | "Dist="<<dist[itsindex]<< | |
504 | "ND0="<<normdist0[itsindex]<< | |
505 | "ND1="<<normdist1[itsindex]<< | |
506 | "ND="<<normdist[itsindex]<< | |
507 | "Pz="<<primvertex[2]<< | |
508 | "Forbid="<<forbidden[itsindex]<< | |
509 | "\n"; | |
510 | // | |
511 | } | |
512 | trackarray.AddAt(best,itsindex); | |
513 | trackarrayc.AddAt(bestConst,itsindex); | |
514 | trackarrayl.AddAt(bestLong,itsindex); | |
515 | new (&helixes[itsindex]) AliHelix(*best); | |
516 | } | |
517 | // | |
518 | // | |
519 | // | |
520 | // first iterration of V0 finder | |
521 | // | |
522 | // AM Comment out for optimization | |
523 | // Int_t rejectBase=0; | |
524 | // Int_t cutN=0; | |
525 | ||
526 | for (Int_t iesd0=0;iesd0<ntracks;iesd0++){ | |
527 | Int_t itrack0 = itsmap[iesd0]; | |
528 | //-AM comment for optimization and store the forbidden value in the debug streamer | |
529 | if (forbidden[itrack0]) continue; | |
530 | AliITStrackMI * btrack0 = (AliITStrackMI*)trackarray.At(itrack0); | |
531 | if (!btrack0) continue; | |
532 | AliITStrackMI *trackc0 = (AliITStrackMI*)trackarrayc.At(itrack0); | |
533 | // | |
534 | for (Int_t iesd1=iesd0+1;iesd1<ntracks;iesd1++){ | |
535 | Int_t itrack1 = itsmap[iesd1]; | |
536 | //-AM comment for optimization and store the forbidden value in the debug streamer | |
537 | if (forbidden[itrack1]) continue; | |
538 | ||
539 | AliITStrackMI * btrack1 = (AliITStrackMI*)trackarray.At(itrack1); | |
540 | if (!btrack1) continue; | |
541 | ||
542 | if ( (btrack0->GetSign()==btrack1->GetSign()) && !AliITSReconstructor::GetRecoParam()->GetStoreLikeSignV0s()) continue; | |
543 | ||
544 | Bool_t isGold = kFALSE; | |
545 | if (TMath::Abs(TMath::Abs(btrack0->GetLabel())-TMath::Abs(btrack1->GetLabel()))==1){ | |
546 | isGold = kTRUE; | |
547 | } | |
548 | // rejectBase=0; | |
549 | AliITStrackMI *trackc1 = (AliITStrackMI*)trackarrayc.At(itrack1); | |
550 | AliHelix &h1 = helixes[itrack0]; | |
551 | AliHelix &h2 = helixes[itrack1]; | |
552 | // | |
553 | // find linear distance | |
554 | Double_t rmin =0; | |
555 | // | |
556 | // | |
557 | // | |
558 | Double_t phase[2][2]={{0.,0.},{0.,0.}}; | |
559 | Double_t radius[2]={0.,0.}; | |
560 | Int_t points = h1.GetRPHIintersections(h2, phase, radius); | |
561 | if (points==0) { | |
562 | // cutN=1; | |
563 | // rejectBase+=1<<cutN; | |
564 | continue; | |
565 | } | |
566 | Double_t delta[2]={1000000,1000000}; | |
567 | rmin = radius[0]; | |
568 | h1.ParabolicDCA(h2,phase[0][0],phase[0][1],radius[0],delta[0]); | |
569 | if (points==2){ | |
570 | if (radius[1]<rmin) rmin = radius[1]; | |
571 | h1.ParabolicDCA(h2,phase[1][0],phase[1][1],radius[1],delta[1]); | |
572 | } | |
573 | rmin = TMath::Sqrt(rmin); | |
574 | Double_t distance = 0; | |
575 | Double_t radiusC = 0; | |
576 | Int_t iphase = 0; | |
577 | if (points==1 || delta[0]<delta[1]){ | |
578 | distance = TMath::Sqrt(delta[0]); | |
579 | radiusC = TMath::Sqrt(radius[0]); | |
580 | }else{ | |
581 | distance = TMath::Sqrt(delta[1]); | |
582 | radiusC = TMath::Sqrt(radius[1]); | |
583 | iphase=1; | |
584 | } | |
585 | if (radiusC<TMath::Max(minr[itrack0],minr[itrack1])){ | |
586 | // cutN=2; | |
587 | //rejectBase+=1<<cutN; | |
588 | continue; | |
589 | } | |
590 | if (radiusC>TMath::Min(maxr[itrack0],maxr[itrack1])){ | |
591 | // cutN=3; | |
592 | // rejectBase+=1<<cutN; | |
593 | continue; | |
594 | } | |
595 | Float_t maxDist = TMath::Min(kMaxDist,Float_t(kMaxDist0+radiusC*kMaxDist1)); | |
596 | if (distance>maxDist){ | |
597 | // cutN=4; | |
598 | // rejectBase+=1<<cutN; | |
599 | continue; | |
600 | } | |
601 | Float_t pointAngle = h1.GetPointAngle(h2,phase[iphase],primvertex); | |
602 | if (pointAngle<TMath::Max(minPointAngle[itrack0],minPointAngle[itrack1])) { | |
603 | // cutN=5; | |
604 | // rejectBase+=1<<cutN; | |
605 | continue; | |
606 | } | |
607 | // | |
608 | // | |
609 | // Double_t distance = TestV0(h1,h2,pvertex,rmin); | |
610 | // | |
611 | // if (distance>maxDist) continue; | |
612 | // if (pvertex->GetRr()<kMinR) continue; | |
613 | // if (pvertex->GetRr()>kMaxR) continue; | |
614 | AliITStrackMI * track0=btrack0; | |
615 | AliITStrackMI * track1=btrack1; | |
616 | // if (pvertex->GetRr()<3.5){ | |
617 | if (radiusC<3.5){ | |
618 | //use longest tracks inside the pipe | |
619 | track0 = (AliITStrackMI*)trackarrayl.At(itrack0); | |
620 | track1 = (AliITStrackMI*)trackarrayl.At(itrack1); | |
621 | } | |
622 | // | |
623 | // | |
624 | ||
625 | ||
626 | ||
627 | pvertex->SetParamN(*track0); | |
628 | pvertex->SetParamP(*track1); | |
629 | pvertex->Update(primvertex); | |
630 | pvertex->SetClusters(track0->ClIndex(),track1->ClIndex()); // register clusters | |
631 | ||
632 | // Define gamma, K0, lambda and lambda_bar from the decay particles and calculate the chi2 | |
633 | AliKFVertex vertexKF; | |
634 | ||
635 | vertexKF.X() = tracker->GetX(); | |
636 | vertexKF.Y() = tracker->GetY(); | |
637 | vertexKF.Z() = tracker->GetZ(); | |
638 | vertexKF.Covariance(0,0) = tracker->GetSigmaX()*tracker->GetSigmaX(); | |
aa2f9f33 | 639 | vertexKF.Covariance(1,1) = tracker->GetSigmaY()*tracker->GetSigmaY(); |
a65a7e70 | 640 | vertexKF.Covariance(2,2) = tracker->GetSigmaZ()*tracker->GetSigmaZ(); |
641 | ||
642 | AliKFParticle elecKF( *(pvertex->GetParamN()) ,11); | |
643 | AliKFParticle posiKF( *(pvertex->GetParamP()) ,-11); | |
644 | AliKFParticle pipKF( *(pvertex->GetParamN()) , 211); | |
645 | AliKFParticle pinKF( *(pvertex->GetParamP()) ,-211); | |
646 | AliKFParticle protKF( *(pvertex->GetParamP()) ,2212); | |
647 | AliKFParticle aproKF ( *(pvertex->GetParamN()) ,-2212); | |
648 | ||
649 | ||
650 | // Gamma | |
651 | AliKFParticle gamKF(elecKF,posiKF); | |
652 | gamKF.SetProductionVertex(vertexKF); | |
653 | ||
654 | Float_t gamKFchi2 = 1000; | |
655 | if ( gamKF.GetNDF()!=0 ){ | |
656 | gamKFchi2 = gamKF.GetChi2()/gamKF.GetNDF(); | |
657 | } | |
658 | ||
659 | Float_t massG=0.; | |
660 | Float_t sigmaMG=0.001; | |
661 | gamKF.SetMassConstraint(massG,sigmaMG); | |
662 | ||
663 | Float_t gamKFchi2C = 1000; | |
664 | if ( gamKF.GetNDF()!=0 ){ | |
665 | gamKFchi2C = gamKF.GetChi2()/gamKF.GetNDF(); | |
666 | } | |
667 | ||
668 | //K0 short | |
669 | AliKFParticle k0KF(pipKF,pinKF); | |
670 | k0KF.SetProductionVertex(vertexKF); | |
671 | ||
672 | Float_t k0KFchi2 = 1000; | |
673 | if ( k0KF.GetNDF()!=0 ){ | |
674 | k0KFchi2 = k0KF.GetChi2()/k0KF.GetNDF(); | |
675 | } | |
676 | ||
677 | //Lambda | |
678 | AliKFParticle lambdaKF(protKF,pinKF); | |
679 | lambdaKF.SetProductionVertex(vertexKF); | |
680 | ||
681 | Float_t lambdaKFchi2 = 1000; | |
682 | if ( lambdaKF.GetNDF()!=0 ){ | |
683 | lambdaKFchi2 = lambdaKF.GetChi2()/lambdaKF.GetNDF(); | |
684 | } | |
685 | ||
686 | //Lambda_bar | |
687 | AliKFParticle alambKF(aproKF,pipKF); | |
688 | alambKF.SetProductionVertex(vertexKF); | |
689 | ||
690 | Float_t alambKFchi2 = 1000; | |
691 | if ( alambKF.GetNDF()!=0 ){ | |
692 | alambKFchi2 = alambKF.GetChi2()/alambKF.GetNDF(); | |
693 | } | |
694 | ||
695 | ||
696 | ||
697 | ||
698 | ||
699 | if (pvertex->GetRr()<kMinR){ | |
700 | // cutN=6; | |
701 | // rejectBase+=1<<cutN; | |
702 | continue; | |
703 | } | |
704 | if (pvertex->GetRr()>kMaxR){ | |
705 | // cutN=7; | |
706 | // rejectBase+=1<<cutN; | |
707 | continue; | |
708 | } | |
709 | if (pvertex->GetV0CosineOfPointingAngle()<kMinPointAngle){ | |
710 | // cutN=8; | |
711 | // rejectBase+=1<<cutN; | |
712 | continue; | |
713 | } | |
714 | //Bo: if (pvertex->GetDist2()>maxDist) continue; | |
715 | ||
716 | if (pvertex->GetDcaV0Daughters()>maxDist){ | |
717 | // cutN=9; | |
718 | // rejectBase+=1<<cutN; | |
719 | continue; | |
720 | } | |
721 | //Bo: pvertex->SetLab(0,track0->GetLabel()); | |
722 | //Bo: pvertex->SetLab(1,track1->GetLabel()); | |
723 | pvertex->SetIndex(0,track0->GetESDtrack()->GetID()); | |
724 | pvertex->SetIndex(1,track1->GetESDtrack()->GetID()); | |
725 | // | |
726 | AliITStrackMI * htrackc0 = trackc0 ? trackc0:dummy; | |
727 | AliITStrackMI * htrackc1 = trackc1 ? trackc1:dummy; | |
728 | ||
729 | // | |
730 | // | |
731 | TObjArray * array0b = (TObjArray*)bestHypothesys->At(itrack0); | |
732 | if (!array0b&&pvertex->GetRr()<40 && TMath::Abs(track0->GetTgl())<kMinTgl1) { | |
733 | tracker->SetCurrentEsdTrack(itrack0); | |
734 | tracker->FollowProlongationTree((AliITStrackMI*)originalTracks->At(itrack0),itrack0, kFALSE); | |
735 | } | |
736 | TObjArray * array1b = (TObjArray*)bestHypothesys->At(itrack1); | |
737 | if (!array1b&&pvertex->GetRr()<40 && TMath::Abs(track1->GetTgl())<kMinTgl1) { | |
738 | tracker->SetCurrentEsdTrack(itrack1); | |
739 | tracker->FollowProlongationTree((AliITStrackMI*)originalTracks->At(itrack1),itrack1, kFALSE); | |
740 | } | |
741 | // | |
742 | AliITStrackMI * track0b = (AliITStrackMI*)originalTracks->At(itrack0); | |
743 | AliITStrackMI * track1b = (AliITStrackMI*)originalTracks->At(itrack1); | |
744 | AliITStrackMI * track0l = (AliITStrackMI*)originalTracks->At(itrack0); | |
745 | AliITStrackMI * track1l = (AliITStrackMI*)originalTracks->At(itrack1); | |
746 | ||
747 | Float_t minchi2before0=kMinchi2before0; | |
748 | Float_t minchi2before1=kMinchi2before1; | |
749 | Float_t minchi2after0 =kMinchi2after0; | |
750 | Float_t minchi2after1 =kMinchi2after1; | |
751 | Double_t xrp[3]; pvertex->GetXYZ(xrp[0],xrp[1],xrp[2]); //I.B. | |
752 | Int_t maxLayer = tracker->GetNearestLayer(xrp); //I.B. | |
753 | ||
754 | if (array0b) for (Int_t i=0;i<5;i++){ | |
755 | // best track after vertex | |
756 | AliITStrackMI * btrack = (AliITStrackMI*)array0b->At(i); | |
757 | if (!btrack) continue; | |
758 | if (btrack->GetNumberOfClusters()>track0l->GetNumberOfClusters()) track0l = btrack; | |
759 | // if (btrack->fX<pvertex->GetRr()-2.-0.5/(0.1+pvertex->GetAnglep()[2])) { | |
760 | if (btrack->GetX()<pvertex->GetRr()-2.) { | |
761 | if ( (maxLayer>i+2|| (i==0)) && btrack->GetNumberOfClusters()==(6-i)&&i<3){ | |
762 | Float_t sumchi2= 0; | |
763 | Float_t sumn = 0; | |
764 | if (maxLayer<3){ // take prim vertex as additional measurement | |
765 | if (normdist[itrack0]>0 && htrackc0){ | |
766 | sumchi2 += TMath::Min((3.-maxLayer)*normdist[itrack0]*normdist[itrack0],16.); | |
767 | }else{ | |
768 | sumchi2 += TMath::Min((3.-maxLayer)*(3*normdist[itrack0]*normdist[itrack0]+3.),16.); | |
769 | } | |
770 | sumn += 3-maxLayer; | |
771 | } | |
772 | for (Int_t ilayer=i;ilayer<maxLayer;ilayer++){ | |
773 | sumn+=1.; | |
774 | if (btrack->GetClIndex(ilayer)<0){ | |
775 | sumchi2+=kAddchi2NegCl0; | |
776 | continue; | |
777 | }else{ | |
778 | Int_t c=( btrack->GetClIndex(ilayer) & 0x0fffffff); | |
779 | for (Int_t itrack=0;itrack<4;itrack++){ | |
780 | AliITStrackerMI::AliITSlayer &layer=tracker->GetLayer(ilayer); | |
781 | if (layer.GetClusterTracks(itrack,c)>=0 && layer.GetClusterTracks(itrack,c)!=itrack0){ | |
782 | sumchi2+=kAddchi2SharedCl; //shared cluster | |
783 | break; | |
784 | } | |
785 | } | |
786 | sumchi2+=btrack->GetDy(ilayer)*btrack->GetDy(ilayer)/(btrack->GetSigmaY(ilayer)*btrack->GetSigmaY(ilayer)); | |
787 | sumchi2+=btrack->GetDz(ilayer)*btrack->GetDz(ilayer)/(btrack->GetSigmaZ(ilayer)*btrack->GetSigmaZ(ilayer)); | |
788 | } | |
789 | } | |
790 | sumchi2/=sumn; | |
791 | if (sumchi2<minchi2before0) minchi2before0=sumchi2; | |
792 | } | |
793 | continue; //safety space - Geo manager will give exact layer | |
794 | } | |
795 | track0b = btrack; | |
796 | minchi2after0 = btrack->GetNormChi2(i); | |
797 | break; | |
798 | } | |
799 | if (array1b) for (Int_t i=0;i<5;i++){ | |
800 | // best track after vertex | |
801 | AliITStrackMI * btrack = (AliITStrackMI*)array1b->At(i); | |
802 | if (!btrack) continue; | |
803 | if (btrack->GetNumberOfClusters()>track1l->GetNumberOfClusters()) track1l = btrack; | |
804 | // if (btrack->fX<pvertex->GetRr()-2-0.5/(0.1+pvertex->GetAnglep()[2])){ | |
805 | if (btrack->GetX()<pvertex->GetRr()-2){ | |
806 | if ((maxLayer>i+2 || (i==0))&&btrack->GetNumberOfClusters()==(6-i)&&(i<3)){ | |
807 | Float_t sumchi2= 0; | |
808 | Float_t sumn = 0; | |
809 | if (maxLayer<3){ // take prim vertex as additional measurement | |
810 | if (normdist[itrack1]>0 && htrackc1){ | |
811 | sumchi2 += TMath::Min((3.-maxLayer)*normdist[itrack1]*normdist[itrack1],16.); | |
812 | }else{ | |
813 | sumchi2 += TMath::Min((3.-maxLayer)*(3*normdist[itrack1]*normdist[itrack1]+3.),16.); | |
814 | } | |
815 | sumn += 3-maxLayer; | |
816 | } | |
817 | for (Int_t ilayer=i;ilayer<maxLayer;ilayer++){ | |
818 | sumn+=1.; | |
819 | if (btrack->GetClIndex(ilayer)<0){ | |
820 | sumchi2+=kAddchi2NegCl1; | |
821 | continue; | |
822 | }else{ | |
823 | Int_t c=( btrack->GetClIndex(ilayer) & 0x0fffffff); | |
824 | for (Int_t itrack=0;itrack<4;itrack++){ | |
825 | AliITStrackerMI::AliITSlayer &layer=tracker->GetLayer(ilayer); | |
826 | if (layer.GetClusterTracks(itrack,c)>=0 && layer.GetClusterTracks(itrack,c)!=itrack1){ | |
827 | sumchi2+=kAddchi2SharedCl; //shared cluster | |
828 | break; | |
829 | } | |
830 | } | |
831 | sumchi2+=btrack->GetDy(ilayer)*btrack->GetDy(ilayer)/(btrack->GetSigmaY(ilayer)*btrack->GetSigmaY(ilayer)); | |
832 | sumchi2+=btrack->GetDz(ilayer)*btrack->GetDz(ilayer)/(btrack->GetSigmaZ(ilayer)*btrack->GetSigmaZ(ilayer)); | |
833 | } | |
834 | } | |
835 | sumchi2/=sumn; | |
836 | if (sumchi2<minchi2before1) minchi2before1=sumchi2; | |
837 | } | |
838 | continue; //safety space - Geo manager will give exact layer | |
839 | } | |
840 | track1b = btrack; | |
841 | minchi2after1 = btrack->GetNormChi2(i); | |
842 | break; | |
843 | } | |
844 | // | |
845 | // position resolution - used for DCA cut | |
846 | Float_t sigmad = track0b->GetSigmaY2()+track0b->GetSigmaZ2()+track1b->GetSigmaY2()+track1b->GetSigmaZ2()+ | |
847 | (track0b->GetX()-pvertex->GetRr())*(track0b->GetX()-pvertex->GetRr())*(track0b->GetSigmaSnp2()+track0b->GetSigmaTgl2())+ | |
848 | (track1b->GetX()-pvertex->GetRr())*(track1b->GetX()-pvertex->GetRr())*(track1b->GetSigmaSnp2()+track1b->GetSigmaTgl2()); | |
849 | sigmad =TMath::Sqrt(sigmad)+0.04; | |
850 | if (pvertex->GetRr()>50){ | |
851 | Double_t cov0[15],cov1[15]; | |
852 | track0b->GetESDtrack()->GetInnerExternalCovariance(cov0); | |
853 | track1b->GetESDtrack()->GetInnerExternalCovariance(cov1); | |
854 | sigmad = cov0[0]+cov0[2]+cov1[0]+cov1[2]+ | |
855 | (80.-pvertex->GetRr())*(80.-pvertex->GetRr())*(cov0[5]+cov0[9])+ | |
856 | (80.-pvertex->GetRr())*(80.-pvertex->GetRr())*(cov1[5]+cov1[9]); | |
857 | sigmad =TMath::Sqrt(sigmad)+0.05; | |
858 | } | |
859 | // | |
860 | AliV0 vertex2; | |
861 | vertex2.SetParamN(*track0b); | |
862 | vertex2.SetParamP(*track1b); | |
863 | vertex2.Update(primvertex); | |
864 | //Bo: if (vertex2.GetDist2()<=pvertex->GetDist2()&&(vertex2.GetV0CosineOfPointingAngle()>=pvertex->GetV0CosineOfPointingAngle())){ | |
865 | if (vertex2.GetDcaV0Daughters()<=pvertex->GetDcaV0Daughters()&&(vertex2.GetV0CosineOfPointingAngle()>=pvertex->GetV0CosineOfPointingAngle())){ | |
866 | pvertex->SetParamN(*track0b); | |
867 | pvertex->SetParamP(*track1b); | |
868 | pvertex->Update(primvertex); | |
869 | pvertex->SetClusters(track0b->ClIndex(),track1b->ClIndex()); // register clusters | |
870 | pvertex->SetIndex(0,track0->GetESDtrack()->GetID()); | |
871 | pvertex->SetIndex(1,track1->GetESDtrack()->GetID()); | |
872 | } | |
873 | pvertex->SetDistSigma(sigmad); | |
874 | //Bo: pvertex->SetDistNorm(pvertex->GetDist2()/sigmad); | |
875 | pvertex->SetNormDCAPrim(normdist[itrack0],normdist[itrack1]); | |
876 | // | |
877 | // define likelihhod and causalities | |
878 | // | |
879 | Float_t pa0=1, pa1=1, pb0=0.26, pb1=0.26; | |
880 | if (maxLayer<1){ | |
881 | Float_t fnorm0 = normdist[itrack0]; | |
882 | if (fnorm0<0) fnorm0*=-3; | |
883 | Float_t fnorm1 = normdist[itrack1]; | |
884 | if (fnorm1<0) fnorm1*=-3; | |
885 | if ((pvertex->GetAnglep()[2]>0.1) || ( (pvertex->GetRr()<10.5)&& pvertex->GetAnglep()[2]>0.05 ) || (pvertex->GetRr()<3)){ | |
886 | pb0 = TMath::Exp(-TMath::Min(fnorm0,Float_t(16.))/12.); | |
887 | pb1 = TMath::Exp(-TMath::Min(fnorm1,Float_t(16.))/12.); | |
888 | } | |
889 | pvertex->SetChi2Before(normdist[itrack0]); | |
890 | pvertex->SetChi2After(normdist[itrack1]); | |
891 | pvertex->SetNAfter(0); | |
892 | pvertex->SetNBefore(0); | |
893 | }else{ | |
894 | pvertex->SetChi2Before(minchi2before0); | |
895 | pvertex->SetChi2After(minchi2before1); | |
896 | if (pvertex->GetAnglep()[2]>0.1 || ( pvertex->GetRr()<10.5 && pvertex->GetAnglep()[2]>0.05) || pvertex->GetRr()<3){ | |
897 | pb0 = TMath::Exp(-TMath::Min(minchi2before0,Float_t(16))/12.); | |
898 | pb1 = TMath::Exp(-TMath::Min(minchi2before1,Float_t(16))/12.); | |
899 | } | |
900 | pvertex->SetNAfter(maxLayer); | |
901 | pvertex->SetNBefore(maxLayer); | |
902 | } | |
903 | if (pvertex->GetRr()<90){ | |
904 | pa0 *= TMath::Min(track0->GetESDtrack()->GetTPCdensity(0,60),Double_t(1.)); | |
905 | pa1 *= TMath::Min(track1->GetESDtrack()->GetTPCdensity(0,60),Double_t(1.)); | |
906 | } | |
907 | if (pvertex->GetRr()<20){ | |
908 | pa0 *= (0.2+TMath::Exp(-TMath::Min(minchi2after0,Float_t(16))/8.))/1.2; | |
909 | pa1 *= (0.2+TMath::Exp(-TMath::Min(minchi2after1,Float_t(16))/8.))/1.2; | |
910 | } | |
911 | // | |
912 | pvertex->SetCausality(pb0,pb1,pa0,pa1); | |
913 | // | |
914 | // Likelihood calculations - apply cuts | |
915 | // | |
916 | ||
917 | // AM - Comment out for optimization and store the v0OK value | |
918 | // Int_t v0OK = 0; | |
919 | // Int_t cutOKN=0; | |
920 | Bool_t v0OK = kTRUE; | |
921 | ||
922 | ||
923 | Float_t p12 = pvertex->GetParamP()->GetParameter()[4]*pvertex->GetParamP()->GetParameter()[4]; | |
924 | p12 += pvertex->GetParamN()->GetParameter()[4]*pvertex->GetParamN()->GetParameter()[4]; | |
925 | p12 = TMath::Sqrt(p12); // "mean" momenta | |
926 | ||
927 | Float_t sigmap0 = kSigp0Par0+kSigp0Par1/(kSigp0Par2+pvertex->GetRr()); | |
928 | Float_t sigmap = kSigpPar0*sigmap0*(kSigpPar1+kSigpPar2*p12); // "resolution: of point angle - as a function of radius and momenta | |
929 | ||
930 | ||
931 | Float_t causalityA = (1.0-pvertex->GetCausalityP()[0])*(1.0-pvertex->GetCausalityP()[1]); | |
932 | Float_t causalityB = TMath::Sqrt(TMath::Min(pvertex->GetCausalityP()[2],Double_t(0.7))* | |
933 | TMath::Min(pvertex->GetCausalityP()[3],Double_t(0.7))); | |
934 | // | |
935 | //Bo: Float_t likelihood0 = (TMath::Exp(-pvertex->GetDistNorm())+0.1) *(pvertex->GetDist2()<0.5)*(pvertex->GetDistNorm()<5); | |
936 | Float_t lDistNorm = pvertex->GetDcaV0Daughters()/pvertex->GetDistSigma(); | |
937 | Float_t likelihood0 = (TMath::Exp(-lDistNorm)+0.1) *(pvertex->GetDcaV0Daughters()<kMaxDcaLh0)*(lDistNorm<5); | |
938 | ||
939 | Float_t likelihood1 = TMath::Exp(-(1.0001-pvertex->GetV0CosineOfPointingAngle())/sigmap)+ | |
940 | 0.4*TMath::Exp(-(1.0001-pvertex->GetV0CosineOfPointingAngle())/(4.*sigmap))+ | |
941 | 0.4*TMath::Exp(-(1.0001-pvertex->GetV0CosineOfPointingAngle())/(8.*sigmap))+ | |
942 | 0.1*TMath::Exp(-(1.0001-pvertex->GetV0CosineOfPointingAngle())/0.01); | |
943 | // | |
944 | if (causalityA<kCausality0Cut){ | |
945 | // cutOKN=1; | |
946 | // v0OK += 1<<cutOKN; | |
947 | v0OK = kFALSE; | |
948 | } | |
949 | if (TMath::Sqrt(likelihood0*likelihood1)<kLikelihood01Cut){ | |
950 | // cutOKN=2; | |
951 | // v0OK += 1<<cutOKN; | |
952 | v0OK = kFALSE; | |
953 | } | |
954 | if (likelihood1<kLikelihood1Cut){ | |
955 | // cutOKN=3; | |
956 | // v0OK += 1<<cutOKN; | |
957 | v0OK = kFALSE; | |
958 | } | |
959 | if (TMath::Power(likelihood0*likelihood1*causalityB,0.33)<kCombinedCut){ | |
960 | // cutOKN=4; | |
961 | // v0OK += 1<<cutOKN; | |
962 | v0OK = kFALSE; | |
963 | } | |
964 | // | |
965 | // | |
966 | if (AliITSReconstructor::GetRecoParam()->GetESDV0Params()->StreamLevel()>0){ | |
967 | Bool_t gold = TMath::Abs(TMath::Abs(track0->GetLabel())-TMath::Abs(track1->GetLabel()))==1; | |
968 | cstream<<"It0"<< | |
969 | "Tr0.="<<track0<< //best without constrain | |
970 | "Tr1.="<<track1<< //best without constrain | |
971 | "posiKF.="<<&posiKF<< //KF from track0 | |
972 | "elecKF.="<<&elecKF<< //KF from track1 | |
973 | "Tr0B.="<<track0b<< //best without constrain after vertex | |
974 | "Tr1B.="<<track1b<< //best without constrain after vertex | |
975 | "Tr0C.="<<htrackc0<< //best with constrain if exist | |
976 | "Tr1C.="<<htrackc1<< //best with constrain if exist | |
977 | "Tr0L.="<<track0l<< //longest best | |
978 | "Tr1L.="<<track1l<< //longest best | |
979 | "Esd0.="<<track0->GetESDtrack()<< // esd track0 params | |
980 | "Esd1.="<<track1->GetESDtrack()<< // esd track1 params | |
981 | "V0.="<<pvertex<< //vertex properties | |
982 | "V0b.="<<&vertex2<< //vertex properties at "best" track | |
983 | "gamKF.="<<&gamKF<< //KF from pvertex | |
984 | "k0KF.="<<&k0KF<< //KF from pvertex | |
985 | "lambdaKF.="<<&lambdaKF<< //KF from pvertex | |
986 | "alambKF.="<<&lambdaKF<< //KF from pvertex | |
987 | "gamKFchi2="<<gamKFchi2<< //Normalized chi2 from KF assuming gamma momther | |
988 | "gamKFchi2C="<<gamKFchi2C<< //Normalized chi2 from KF assuming gamma mother+mass constraint | |
989 | "k0KFchi2="<<k0KFchi2<< //Normalized chi2 from KF assuming K0 mother | |
990 | "lambdaKFchi2="<<lambdaKFchi2<< //Normalized chi2 from KF assuming Lambda mother | |
991 | "alambKFchi2="<<alambKFchi2<< //Normalized chi2 from KF assuming lambda_bar mother | |
992 | "ND0="<<normdist[itrack0]<< //normalize distance for track0 | |
993 | "ND1="<<normdist[itrack1]<< //normalize distance for track1 | |
994 | "Gold="<<gold<< // | |
995 | // "RejectBase="<<rejectBase<< //rejection in First itteration | |
996 | "OK="<<v0OK<< | |
997 | "rmin="<<rmin<< | |
998 | "sigmad="<<sigmad<< | |
999 | "Forbid0="<<forbidden[itrack0]<< | |
1000 | "Forbid1="<<forbidden[itrack1]<< | |
1001 | "\n"; | |
1002 | } | |
1003 | // if (rejectBase>0) continue; | |
1004 | //if (forbidden[itrack0]>0 ||forbidden[itrack1]>0) continue; | |
1005 | // | |
1006 | pvertex->SetStatus(0); | |
1007 | // if (rejectBase) { | |
1008 | // pvertex->SetStatus(-100); | |
1009 | //} | |
1010 | if (pvertex->GetV0CosineOfPointingAngle()>kMinPointAngle2) { | |
1011 | //Bo: pvertex->SetESDindexes(track0->GetESDtrack()->GetID(),track1->GetESDtrack()->GetID()); | |
1012 | pvertex->SetIndex(0,track0->GetESDtrack()->GetID());//Bo: consistency 0 for neg | |
1013 | pvertex->SetIndex(1,track1->GetESDtrack()->GetID());//Bo: consistency 1 for pos | |
1014 | // if (v0OK==0){ | |
1015 | if (v0OK){ | |
1016 | // AliV0vertex vertexjuri(*track0,*track1); | |
1017 | // vertexjuri.SetESDindexes(track0->fESDtrack->GetID(),track1->fESDtrack->GetID()); | |
1018 | // event->AddV0(&vertexjuri); | |
1019 | pvertex->SetStatus(100); | |
1020 | } | |
1021 | pvertex->SetOnFlyStatus(kTRUE); | |
1022 | pvertex->ChangeMassHypothesis(kK0Short); | |
1023 | event->AddV0(pvertex); | |
1024 | } | |
1025 | } | |
1026 | } | |
1027 | // | |
1028 | // | |
1029 | // delete temporary arrays | |
1030 | // | |
1031 | delete[] forbidden; | |
1032 | delete[] minPointAngle; | |
1033 | delete[] maxr; | |
1034 | delete[] minr; | |
1035 | delete[] norm; | |
1036 | delete[] normdist; | |
1037 | delete[] normdist1; | |
1038 | delete[] normdist0; | |
1039 | delete[] dist; | |
1040 | delete[] itsmap; | |
1041 | delete[] helixes; | |
1042 | delete pvertex; | |
1043 | delete dummy; | |
1044 | } | |
1045 | //------------------------------------------------------------------------ | |
1046 | void AliITSV0Finder::RefitV02(const AliESDEvent *event, | |
1047 | AliITStrackerMI *tracker) | |
1048 | { | |
1049 | // | |
1050 | //try to refit V0s in the third path of the reconstruction | |
1051 | // | |
1052 | TTreeSRedirector &cstream = *(tracker->GetDebugStreamer()); | |
1053 | // | |
1054 | Int_t nv0s = event->GetNumberOfV0s(); | |
2942f542 | 1055 | Float_t primvertex[3]={static_cast<Float_t>(tracker->GetX()),static_cast<Float_t>(tracker->GetY()),static_cast<Float_t>(tracker->GetZ())}; |
a65a7e70 | 1056 | AliV0 v0temp; |
1057 | for (Int_t iv0 = 0; iv0<nv0s;iv0++){ | |
1058 | AliV0 * v0mi = (AliV0*)event->GetV0(iv0); | |
1059 | if (!v0mi) continue; | |
1060 | Int_t itrack0 = v0mi->GetIndex(0); | |
1061 | Int_t itrack1 = v0mi->GetIndex(1); | |
1062 | AliESDtrack *esd0 = event->GetTrack(itrack0); | |
1063 | AliESDtrack *esd1 = event->GetTrack(itrack1); | |
1064 | if (!esd0||!esd1) continue; | |
1065 | AliITStrackMI tpc0(*esd0); | |
1066 | AliITStrackMI tpc1(*esd1); | |
1067 | Double_t x,y,z; v0mi->GetXYZ(x,y,z); //I.B. | |
1068 | Double_t alpha =TMath::ATan2(y,x); //I.B. | |
1069 | if (v0mi->GetRr()>85){ | |
1070 | if (tpc0.Propagate(alpha,v0mi->GetRr())&&tpc1.Propagate(alpha,v0mi->GetRr())){ | |
1071 | v0temp.SetParamN(tpc0); | |
1072 | v0temp.SetParamP(tpc1); | |
1073 | v0temp.Update(primvertex); | |
1074 | if (AliITSReconstructor::GetRecoParam()->GetESDV0Params()->StreamLevel()>0) cstream<<"Refit"<< | |
1075 | "V0.="<<v0mi<< | |
1076 | "V0refit.="<<&v0temp<< | |
1077 | "Tr0.="<<&tpc0<< | |
1078 | "Tr1.="<<&tpc1<< | |
1079 | "\n"; | |
1080 | //Bo: if (v0temp.GetDist2()<v0mi->GetDist2() || v0temp.GetV0CosineOfPointingAngle()>v0mi->GetV0CosineOfPointingAngle()){ | |
1081 | if (v0temp.GetDcaV0Daughters()<v0mi->GetDcaV0Daughters() || v0temp.GetV0CosineOfPointingAngle()>v0mi->GetV0CosineOfPointingAngle()){ | |
1082 | v0mi->SetParamN(tpc0); | |
1083 | v0mi->SetParamP(tpc1); | |
1084 | v0mi->Update(primvertex); | |
1085 | } | |
1086 | } | |
1087 | continue; | |
1088 | } | |
1089 | if (v0mi->GetRr()>35){ | |
1090 | AliITStrackerMI::CorrectForTPCtoITSDeadZoneMaterial(&tpc0); | |
1091 | AliITStrackerMI::CorrectForTPCtoITSDeadZoneMaterial(&tpc1); | |
1092 | if (tpc0.Propagate(alpha,v0mi->GetRr())&&tpc1.Propagate(alpha,v0mi->GetRr())){ | |
1093 | v0temp.SetParamN(tpc0); | |
1094 | v0temp.SetParamP(tpc1); | |
1095 | v0temp.Update(primvertex); | |
1096 | if (AliITSReconstructor::GetRecoParam()->GetESDV0Params()->StreamLevel()>0) cstream<<"Refit"<< | |
1097 | "V0.="<<v0mi<< | |
1098 | "V0refit.="<<&v0temp<< | |
1099 | "Tr0.="<<&tpc0<< | |
1100 | "Tr1.="<<&tpc1<< | |
1101 | "\n"; | |
1102 | //Bo: if (v0temp.GetDist2()<v0mi->GetDist2() || v0temp.GetV0CosineOfPointingAngle()>v0mi->GetV0CosineOfPointingAngle()){ | |
1103 | if (v0temp.GetDcaV0Daughters()<v0mi->GetDcaV0Daughters() || v0temp.GetV0CosineOfPointingAngle()>v0mi->GetV0CosineOfPointingAngle()){ | |
1104 | v0mi->SetParamN(tpc0); | |
1105 | v0mi->SetParamP(tpc1); | |
1106 | v0mi->Update(primvertex); | |
1107 | } | |
1108 | } | |
1109 | continue; | |
1110 | } | |
1111 | AliITStrackerMI::CorrectForTPCtoITSDeadZoneMaterial(&tpc0); | |
1112 | AliITStrackerMI::CorrectForTPCtoITSDeadZoneMaterial(&tpc1); | |
1113 | // if (tpc0.Propagate(alpha,v0mi->GetRr())&&tpc1.Propagate(alpha,v0mi->GetRr())){ | |
1114 | if (tracker->RefitAt(v0mi->GetRr(),&tpc0, v0mi->GetClusters(0)) && | |
1115 | tracker->RefitAt(v0mi->GetRr(),&tpc1, v0mi->GetClusters(1))){ | |
1116 | v0temp.SetParamN(tpc0); | |
1117 | v0temp.SetParamP(tpc1); | |
1118 | v0temp.Update(primvertex); | |
1119 | if (AliITSReconstructor::GetRecoParam()->GetESDV0Params()->StreamLevel()>0) cstream<<"Refit"<< | |
1120 | "V0.="<<v0mi<< | |
1121 | "V0refit.="<<&v0temp<< | |
1122 | "Tr0.="<<&tpc0<< | |
1123 | "Tr1.="<<&tpc1<< | |
1124 | "\n"; | |
1125 | //Bo: if (v0temp.GetDist2()<v0mi->GetDist2() || v0temp.GetV0CosineOfPointingAngle()>v0mi->GetV0CosineOfPointingAngle()){ | |
1126 | if (v0temp.GetDcaV0Daughters()<v0mi->GetDcaV0Daughters() || v0temp.GetV0CosineOfPointingAngle()>v0mi->GetV0CosineOfPointingAngle()){ | |
1127 | v0mi->SetParamN(tpc0); | |
1128 | v0mi->SetParamP(tpc1); | |
1129 | v0mi->Update(primvertex); | |
1130 | } | |
1131 | } | |
1132 | } | |
1133 | } | |
1134 | //------------------------------------------------------------------------ | |
1135 | ||
1136 | ||
1137 | AliITSV0Finder::~AliITSV0Finder() | |
1138 | { | |
1139 | // | |
1140 | //destructor | |
1141 | // | |
1142 | if (fDebugStreamer) { | |
1143 | //fDebugStreamer->Close(); | |
1144 | delete fDebugStreamer; | |
1145 | } | |
1146 | ||
1147 | } |