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