]>
Commit | Line | Data |
---|---|---|
1 | /************************************************************************** | |
2 | * Copyright(c) 1998-1999, 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 | /* History of cvs commits: | |
18 | * | |
19 | * $Log$ | |
20 | * Revision 1.93 2007/10/10 09:05:10 schutz | |
21 | * Changing name QualAss to QA | |
22 | * | |
23 | * Revision 1.92 2007/08/28 12:55:08 policheh | |
24 | * Loaders removed from the reconstruction code (C.Cheshkov) | |
25 | * | |
26 | * Revision 1.91 2007/08/07 14:12:03 kharlov | |
27 | * Quality assurance added (Yves Schutz) | |
28 | * | |
29 | * Revision 1.90 2007/07/11 13:43:30 hristov | |
30 | * New class AliESDEvent, backward compatibility with the old AliESD (Christian) | |
31 | * | |
32 | * Revision 1.89 2007/07/03 08:13:04 kharlov | |
33 | * Bug fix in CPV local coordinates | |
34 | * | |
35 | * Revision 1.88 2007/06/27 09:11:07 kharlov | |
36 | * Bug fix for CPV-EMC distance | |
37 | * | |
38 | * Revision 1.87 2007/05/04 14:49:29 policheh | |
39 | * AliPHOSRecPoint inheritance from AliCluster | |
40 | * | |
41 | * Revision 1.86 2007/04/02 15:00:16 cvetan | |
42 | * No more calls to gAlice in the reconstruction | |
43 | * | |
44 | * Revision 1.85 2007/03/28 19:18:15 kharlov | |
45 | * RecPoints recalculation in TSM removed | |
46 | * | |
47 | * Revision 1.84 2007/03/07 07:01:21 hristov | |
48 | * Fixing copy/paste erro. Additional protections | |
49 | * | |
50 | * Revision 1.83 2007/03/06 21:07:37 kharlov | |
51 | * DP: xz CPV-EMC distance filled to TS | |
52 | * | |
53 | * Revision 1.82 2007/03/06 06:54:48 kharlov | |
54 | * DP:Calculation of cluster properties dep. on vertex added | |
55 | * | |
56 | * Revision 1.81 2007/02/05 10:02:40 kharlov | |
57 | * Module numbering is corrected | |
58 | * | |
59 | * Revision 1.80 2006/08/28 10:01:56 kharlov | |
60 | * Effective C++ warnings fixed (Timur Pocheptsov) | |
61 | * | |
62 | * Revision 1.79 2006/04/25 12:41:15 hristov | |
63 | * Moving non-persistent data to AliESDfriend (Yu.Belikov) | |
64 | * | |
65 | * Revision 1.78 2005/11/18 13:04:51 hristov | |
66 | * Bug fix | |
67 | * | |
68 | * Revision 1.77 2005/11/17 23:34:36 hristov | |
69 | * Corrected logics | |
70 | * | |
71 | * Revision 1.76 2005/11/17 22:29:12 hristov | |
72 | * Faster version, no attempt to match tracks outside the PHOS acceptance | |
73 | * | |
74 | * Revision 1.75 2005/11/17 12:35:27 hristov | |
75 | * Use references instead of objects. Avoid to create objects when they are not really needed | |
76 | * | |
77 | * Revision 1.74 2005/07/08 14:01:36 hristov | |
78 | * Tracking in non-uniform nmagnetic field (Yu.Belikov) | |
79 | * | |
80 | * Revision 1.73 2005/05/28 14:19:05 schutz | |
81 | * Compilation warnings fixed by T.P. | |
82 | * | |
83 | */ | |
84 | ||
85 | //_________________________________________________________________________ | |
86 | // Implementation version 1 of algorithm class to construct PHOS track segments | |
87 | // Track segment for PHOS is list of | |
88 | // EMC RecPoint + (possibly) CPV RecPoint | |
89 | // To find TrackSegments we do the following: | |
90 | // for each EMC RecPoints we look at | |
91 | // CPV RecPoints in the radious fRcpv. | |
92 | // If there is such a CPV RecPoint, | |
93 | // we make "Link" it is just indexes of EMC and CPV RecPoint and distance | |
94 | // between them in the PHOS plane. | |
95 | // Then we sort "Links" and starting from the | |
96 | // least "Link" pointing to the unassined EMC and CPV RecPoints assing them to | |
97 | // new TrackSegment. | |
98 | // If there is no CPV RecPoint we make TrackSegment | |
99 | // consisting from EMC alone. There is no TrackSegments without EMC RecPoint. | |
100 | //// In principle this class should be called from AliPHOSReconstructor, but | |
101 | // one can use it as well in standalone mode. | |
102 | // Use case: | |
103 | // root [0] AliPHOSTrackSegmentMakerv1 * t = new AliPHOSTrackSegmentMaker("galice.root", "tracksegmentsname", "recpointsname") | |
104 | // Warning in <TDatabasePDG::TDatabasePDG>: object already instantiated | |
105 | // // reads gAlice from header file "galice.root", uses recpoints stored in the branch names "recpointsname" (default = "Default") | |
106 | // // and saves recpoints in branch named "tracksegmentsname" (default = "recpointsname") | |
107 | // root [1] t->ExecuteTask() | |
108 | // root [3] t->SetTrackSegmentsBranch("max distance 5 cm") | |
109 | // root [4] t->ExecuteTask("deb all time") | |
110 | // | |
111 | //*-- Author: Dmitri Peressounko (RRC Ki & SUBATECH) & Yves Schutz (SUBATECH) | |
112 | // | |
113 | ||
114 | // --- ROOT system --- | |
115 | #include "TVector3.h" | |
116 | #include "TTree.h" | |
117 | #include "TBenchmark.h" | |
118 | ||
119 | // --- Standard library --- | |
120 | #include "Riostream.h" | |
121 | // --- AliRoot header files --- | |
122 | #include "AliPHOSGeometry.h" | |
123 | #include "AliPHOSTrackSegmentMakerv1.h" | |
124 | #include "AliPHOSTrackSegment.h" | |
125 | #include "AliPHOSLink.h" | |
126 | #include "AliESDEvent.h" | |
127 | #include "AliESDtrack.h" | |
128 | #include "AliPHOSEmcRecPoint.h" | |
129 | #include "AliPHOSCpvRecPoint.h" | |
130 | #include "AliLog.h" | |
131 | #include "AliMagF.h" | |
132 | #include "AliMagF.h" | |
133 | #include "AliTracker.h" | |
134 | #include "AliGeomManager.h" | |
135 | #include "AliCluster.h" | |
136 | #include "AliKalmanTrack.h" | |
137 | #include "AliGlobalQADataMaker.h" | |
138 | #include "AliVParticle.h" | |
139 | ||
140 | ||
141 | ClassImp( AliPHOSTrackSegmentMakerv1) | |
142 | ||
143 | ||
144 | //____________________________________________________________________________ | |
145 | AliPHOSTrackSegmentMakerv1::AliPHOSTrackSegmentMakerv1() : | |
146 | AliPHOSTrackSegmentMaker(), | |
147 | fDefaultInit(kTRUE), | |
148 | fWrite(kFALSE), | |
149 | fNTrackSegments(0), | |
150 | fRcpv(0.f), | |
151 | fRtpc(0.f), | |
152 | fVtx(0.f,0.f,0.f), | |
153 | fLinkUpArray(0), | |
154 | fEmcFirst(0), | |
155 | fEmcLast(0), | |
156 | fCpvFirst(0), | |
157 | fCpvLast(0), | |
158 | fModule(0), | |
159 | fTrackSegments(NULL) | |
160 | { | |
161 | // default ctor (to be used mainly by Streamer) | |
162 | InitParameters() ; | |
163 | } | |
164 | ||
165 | //____________________________________________________________________________ | |
166 | AliPHOSTrackSegmentMakerv1::AliPHOSTrackSegmentMakerv1(AliPHOSGeometry *geom) : | |
167 | AliPHOSTrackSegmentMaker(geom), | |
168 | fDefaultInit(kFALSE), | |
169 | fWrite(kFALSE), | |
170 | fNTrackSegments(0), | |
171 | fRcpv(0.f), | |
172 | fRtpc(0.f), | |
173 | fVtx(0.f,0.f,0.f), | |
174 | fLinkUpArray(0), | |
175 | fEmcFirst(0), | |
176 | fEmcLast(0), | |
177 | fCpvFirst(0), | |
178 | fCpvLast(0), | |
179 | fModule(0), | |
180 | fTrackSegments(NULL) | |
181 | { | |
182 | // ctor | |
183 | InitParameters() ; | |
184 | Init() ; | |
185 | fESD = 0; | |
186 | } | |
187 | ||
188 | ||
189 | AliPHOSTrackSegmentMakerv1::AliPHOSTrackSegmentMakerv1(const AliPHOSTrackSegmentMakerv1 & tsm) : | |
190 | AliPHOSTrackSegmentMaker(tsm), | |
191 | fDefaultInit(kFALSE), | |
192 | fWrite(kFALSE), | |
193 | fNTrackSegments(0), | |
194 | fRcpv(0.f), | |
195 | fRtpc(0.f), | |
196 | fVtx(0.f,0.f,0.f), | |
197 | fLinkUpArray(0), | |
198 | fEmcFirst(0), | |
199 | fEmcLast(0), | |
200 | fCpvFirst(0), | |
201 | fCpvLast(0), | |
202 | fModule(0), | |
203 | fTrackSegments(NULL) | |
204 | { | |
205 | // cpy ctor: no implementation yet | |
206 | // requested by the Coding Convention | |
207 | Fatal("cpy ctor", "not implemented") ; | |
208 | } | |
209 | ||
210 | ||
211 | //____________________________________________________________________________ | |
212 | AliPHOSTrackSegmentMakerv1::~AliPHOSTrackSegmentMakerv1() | |
213 | { | |
214 | // dtor | |
215 | // fDefaultInit = kTRUE if TrackSegmentMaker created by default ctor (to get just the parameters) | |
216 | if (!fDefaultInit) | |
217 | delete fLinkUpArray ; | |
218 | if (fTrackSegments) { | |
219 | fTrackSegments->Delete(); | |
220 | delete fTrackSegments; | |
221 | } | |
222 | } | |
223 | ||
224 | //____________________________________________________________________________ | |
225 | void AliPHOSTrackSegmentMakerv1::FillOneModule() | |
226 | { | |
227 | // Finds first and last indexes between which | |
228 | // clusters from one PHOS module are | |
229 | ||
230 | //First EMC clusters | |
231 | Int_t totalEmc = fEMCRecPoints->GetEntriesFast() ; | |
232 | for(fEmcFirst = fEmcLast; (fEmcLast < totalEmc) && | |
233 | ((static_cast<AliPHOSRecPoint *>(fEMCRecPoints->At(fEmcLast)))->GetPHOSMod() == fModule ); | |
234 | fEmcLast ++) ; | |
235 | ||
236 | //Now CPV clusters | |
237 | Int_t totalCpv = fCPVRecPoints->GetEntriesFast() ; | |
238 | ||
239 | for(fCpvFirst = fCpvLast; (fCpvLast < totalCpv) && | |
240 | ((static_cast<AliPHOSRecPoint *>(fCPVRecPoints->At(fCpvLast)))->GetPHOSMod() == fModule ); | |
241 | fCpvLast ++) ; | |
242 | ||
243 | } | |
244 | ||
245 | //____________________________________________________________________________ | |
246 | void AliPHOSTrackSegmentMakerv1::GetDistanceInPHOSPlane(AliPHOSEmcRecPoint * emcClu, | |
247 | AliPHOSCpvRecPoint * cpvClu, | |
248 | Int_t &trackindex, | |
249 | Float_t &dx, Float_t &dz) const | |
250 | { | |
251 | // Calculates the distance between the EMC RecPoint and the CPV RecPoint | |
252 | // If no CPV, calculates the distance between the EMC RecPoint and the track | |
253 | // prolongation to the PHOS module plane. | |
254 | // Clusters are sorted in "rows" and "columns" of width 1 cm | |
255 | ||
256 | // Float_t delta = 1 ; // Width of the rows in sorting of RecPoints (in cm) | |
257 | // // if you change this value, change it as well in xxxRecPoint::Compare() | |
258 | ||
259 | trackindex = -1; | |
260 | dx = 999.; | |
261 | dz = 999.; | |
262 | ||
263 | if(!cpvClu) { | |
264 | ||
265 | if(!emcClu) { | |
266 | return; | |
267 | } | |
268 | ||
269 | // *** Start the matching | |
270 | Int_t nt=fESD->GetNumberOfTracks(); | |
271 | Int_t iPHOSMod = emcClu->GetPHOSMod() ; | |
272 | //Calculate actual distance to PHOS module | |
273 | TVector3 globaPos ; | |
274 | fGeom->Local2Global(iPHOSMod, 0.,0., globaPos) ; | |
275 | const Double_t rPHOS = globaPos.Pt() ; //Distance to center of PHOS module | |
276 | const Double_t kYmax = 72.+10. ; //Size of the module (with some reserve) in phi direction | |
277 | const Double_t kZmax = 64.+10. ; //Size of the module (with some reserve) in z direction | |
278 | const Double_t kAlpha0=330./180.*TMath::Pi() ; //First PHOS module angular direction | |
279 | const Double_t kAlpha= 20./180.*TMath::Pi() ; //PHOS module angular size | |
280 | Double_t minDistance = 1.e6; | |
281 | ||
282 | TVector3 vecEmc ; // Local position of EMC recpoint | |
283 | emcClu->GetLocalPosition(vecEmc) ; | |
284 | ||
285 | Double_t gposTrack[3] ; | |
286 | Double_t bz = AliTracker::GetBz() ; //B-Field for approximate matching | |
287 | Double_t b[3]; | |
288 | for (Int_t i=0; i<nt; i++) { | |
289 | AliESDtrack *esdTrack=fESD->GetTrack(i); | |
290 | ||
291 | // Skip the tracks having "wrong" status (has to be checked/tuned) | |
292 | ULong_t status = esdTrack->GetStatus(); | |
293 | if ((status & AliESDtrack::kTPCout) == 0) continue; | |
294 | // if ((status & AliESDtrack::kTRDout) == 0) continue; | |
295 | // if ((status & AliESDtrack::kTRDrefit) == 1) continue; | |
296 | ||
297 | //Continue extrapolation from TPC outer surface | |
298 | const AliExternalTrackParam *outerParam=esdTrack->GetOuterParam(); | |
299 | if (!outerParam) continue; | |
300 | AliExternalTrackParam t(*outerParam); | |
301 | ||
302 | t.GetBxByBz(b) ; | |
303 | //Direction to the current PHOS module | |
304 | Double_t phiMod=kAlpha0-kAlpha*iPHOSMod ; | |
305 | if(!t.Rotate(phiMod)) | |
306 | continue ; | |
307 | ||
308 | Double_t y; // Some tracks do not reach the PHOS | |
309 | if (!t.GetYAt(rPHOS,bz,y)) continue; // because of the bending | |
310 | ||
311 | Double_t z; | |
312 | if(!t.GetZAt(rPHOS,bz,z)) | |
313 | continue ; | |
314 | if (TMath::Abs(z) > kZmax) | |
315 | continue; // Some tracks miss the PHOS in Z | |
316 | if(TMath::Abs(y) < kYmax){ | |
317 | t.PropagateToBxByBz(rPHOS,b); // Propagate to the matching module | |
318 | //t.CorrectForMaterial(...); // Correct for the TOF material, if needed | |
319 | t.GetXYZ(gposTrack) ; | |
320 | TVector3 globalPositionTr(gposTrack) ; | |
321 | TVector3 localPositionTr ; | |
322 | fGeom->Global2Local(localPositionTr,globalPositionTr,iPHOSMod) ; | |
323 | Double_t ddx = vecEmc.X()-localPositionTr.X(); | |
324 | Double_t ddz = vecEmc.Z()-localPositionTr.Z(); | |
325 | Double_t d2 = ddx*ddx + ddz*ddz; | |
326 | if(d2 < minDistance) { | |
327 | dx = ddx ; | |
328 | dz = ddz ; | |
329 | trackindex=i; | |
330 | minDistance=d2 ; | |
331 | } | |
332 | } | |
333 | } //Scanned all tracks | |
334 | return ; | |
335 | } | |
336 | ||
337 | ||
338 | TVector3 emcGlobal; | |
339 | fGeom->GetGlobalPHOS((AliPHOSRecPoint*)emcClu,emcGlobal); | |
340 | ||
341 | // Radius from IP to current point | |
342 | Double_t rEMC = TMath::Abs(emcGlobal.Pt()); | |
343 | ||
344 | // Extrapolate the global track direction to EMC | |
345 | // and find the closest track | |
346 | ||
347 | Int_t nTracks = fESD->GetNumberOfTracks(); | |
348 | ||
349 | AliESDtrack *track; | |
350 | Double_t xyz[] = {-1,-1,-1}; | |
351 | Double_t pxyz[3]; | |
352 | Double_t zEMC,xEMC; | |
353 | Int_t module; | |
354 | TVector3 vecP; | |
355 | TVector3 locClu; | |
356 | ||
357 | Float_t minDistance = 1.e6; | |
358 | Float_t dr; | |
359 | ||
360 | for (Int_t iTrack=0; iTrack<nTracks; iTrack++) { | |
361 | track = fESD->GetTrack(iTrack); | |
362 | if (!track->GetXYZAt(rEMC, fESD->GetMagneticField(), xyz)) continue; | |
363 | ||
364 | AliDebug(1,Form("Event %d, iTrack: %d, (%.3f,%.3f,%.3f)", | |
365 | fESD->GetEventNumberInFile(),iTrack,xyz[0],xyz[1],xyz[2])); | |
366 | ||
367 | if (track->GetPxPyPzAt(rEMC,fESD->GetMagneticField(),pxyz)) { | |
368 | ||
369 | vecP.SetXYZ(pxyz[0],pxyz[1],pxyz[2]); | |
370 | fGeom->ImpactOnEmc(xyz,vecP.Theta(),vecP.Phi(),module,zEMC,xEMC) ; | |
371 | ||
372 | if(!module) continue; | |
373 | AliDebug(1,Form("\t\tTrack hit PHOS! Module: %d, (x,z)=(%.3f,%.3f)",module,xEMC,zEMC)); | |
374 | ||
375 | if(emcClu->GetPHOSMod() != module) continue; | |
376 | ||
377 | // match track to EMC cluster | |
378 | emcClu->GetLocalPosition(locClu); | |
379 | ||
380 | Float_t delta_x = xEMC - locClu.X(); | |
381 | Float_t delta_z = zEMC - locClu.Z(); | |
382 | dr = TMath::Sqrt(delta_x*delta_x + delta_z*delta_z); | |
383 | AliDebug(1,Form("\tMatch iTrack=%d: (dx,dz)=(%.3f,%.3f)",iTrack,delta_x,delta_z)); | |
384 | ||
385 | if(dr<minDistance) { | |
386 | trackindex = iTrack; | |
387 | minDistance = dr; | |
388 | dx = delta_x; | |
389 | dz = delta_z; | |
390 | } | |
391 | } | |
392 | ||
393 | } | |
394 | ||
395 | if(trackindex>=0) { | |
396 | AliDebug(1,Form("\t\tBest match for (xClu,zClu,eClu)=(%.3f,%.3f,%.3f): iTrack=%d, dR=%.3f", | |
397 | locClu.X(),locClu.Z(),emcClu->GetEnergy(), | |
398 | trackindex,TMath::Sqrt(dx*dx+dz*dz))); | |
399 | return; | |
400 | } | |
401 | ||
402 | Float_t distance2Track = fRtpc ; | |
403 | ||
404 | trackindex = -1 ; // closest track within fRCpv | |
405 | ||
406 | TVector3 vecEmc ; // Local position of EMC recpoint | |
407 | TVector3 vecPloc ; // Momentum direction at CPV plain | |
408 | ||
409 | //toofar = kTRUE ; | |
410 | if(emcClu->GetPHOSMod() != cpvClu->GetPHOSMod()){ | |
411 | dx=999. ; | |
412 | dz=999. ; | |
413 | return ; | |
414 | } | |
415 | ||
416 | emcClu->GetLocalPosition(vecEmc) ; | |
417 | ||
418 | Double_t xCPV=0,zCPV=0 ; //EMC-projected coordinates of CPV cluster | |
419 | TVector3 cpvGlobal; // Global position of the CPV recpoint | |
420 | fGeom->GetGlobalPHOS((AliPHOSRecPoint*)cpvClu,cpvGlobal); | |
421 | Double_t vtxCPV[3]={cpvGlobal.X(),cpvGlobal.Y(),cpvGlobal.Z()} ; | |
422 | Int_t dummyMod ; | |
423 | ||
424 | if (fESD == 0x0) { | |
425 | //if no track information available, assume straight line from IP to emcal | |
426 | fGeom->ImpactOnEmc(vtxCPV,cpvGlobal.Theta(),cpvGlobal.Phi(),dummyMod,zCPV,xCPV) ; | |
427 | dx=xCPV - vecEmc.X() ; | |
428 | dz=zCPV - vecEmc.Z() ; | |
429 | return ; | |
430 | } | |
431 | ||
432 | //if there is ESD try to correct distance using TPC information on particle direct in CPV | |
433 | if (fESD != 0x0) { | |
434 | ||
435 | Double_t rCPV = cpvGlobal.Pt() ;// Radius from IP to current point | |
436 | ||
437 | // Extrapolate the global track direction if any to CPV and find the closest track | |
438 | Int_t iClosestTrack = -1; | |
439 | TVector3 inPHOS ; | |
440 | ||
441 | for (Int_t iTrack=0; iTrack<nTracks; iTrack++) { | |
442 | track = fESD->GetTrack(iTrack); | |
443 | if (!track->GetXYZAt(rCPV, fESD->GetMagneticField(), xyz)) | |
444 | continue; //track coord on the cylinder of PHOS radius | |
445 | if ((TMath::Abs(xyz[0])+TMath::Abs(xyz[1])+TMath::Abs(xyz[2]))<=0) | |
446 | continue; | |
447 | //Check if this track hits PHOS | |
448 | inPHOS.SetXYZ(xyz[0],xyz[1],xyz[2]); | |
449 | distance2Track = inPHOS.Angle(cpvGlobal) ; | |
450 | // Find the closest track to the CPV recpoint | |
451 | if (distance2Track < minDistance) { | |
452 | minDistance = distance2Track; | |
453 | iClosestTrack = iTrack; | |
454 | } | |
455 | } | |
456 | ||
457 | if (iClosestTrack != -1) { | |
458 | track = fESD->GetTrack(iClosestTrack); | |
459 | if (track->GetPxPyPzAt(rCPV, fESD->GetMagneticField(), pxyz)) { // track momentum ibid. | |
460 | vecP.SetXYZ(pxyz[0],pxyz[1],pxyz[2]); | |
461 | fGeom->ImpactOnEmc(vtxCPV,vecP.Theta(),vecP.Phi(),dummyMod,zCPV,xCPV) ; | |
462 | } | |
463 | } | |
464 | ||
465 | if(minDistance < fRtpc ){ | |
466 | trackindex = iClosestTrack ; | |
467 | } | |
468 | } | |
469 | if(trackindex!=-1){ | |
470 | // If the closest global track is found, calculate EMC-CPV distance from it | |
471 | dx=xCPV - vecEmc.X() ; | |
472 | dz=zCPV - vecEmc.Z() ; | |
473 | } | |
474 | else{ | |
475 | // If no global track was found, just take the nearest CPV point | |
476 | fGeom->ImpactOnEmc(vtxCPV,cpvGlobal.Theta(),cpvGlobal.Phi(),dummyMod,zCPV,xCPV) ; | |
477 | dx=xCPV - vecEmc.X() ; | |
478 | dz=zCPV - vecEmc.Z() ; | |
479 | } | |
480 | return ; | |
481 | } | |
482 | //____________________________________________________________________________ | |
483 | void AliPHOSTrackSegmentMakerv1::Init() | |
484 | { | |
485 | // Make all memory allocations that are not possible in default constructor | |
486 | ||
487 | fLinkUpArray = new TClonesArray("AliPHOSLink", 1000); | |
488 | fTrackSegments = new TClonesArray("AliPHOSTrackSegment",100); | |
489 | fTrackSegments->SetName("TRACKS"); | |
490 | } | |
491 | ||
492 | //____________________________________________________________________________ | |
493 | void AliPHOSTrackSegmentMakerv1::InitParameters() | |
494 | { | |
495 | //Initializes parameters | |
496 | fRcpv = 10. ; | |
497 | fRtpc = 4. ; | |
498 | fEmcFirst = 0 ; | |
499 | fEmcLast = 0 ; | |
500 | fCpvFirst = 0 ; | |
501 | fCpvLast = 0 ; | |
502 | fLinkUpArray = 0 ; | |
503 | fWrite = kTRUE ; | |
504 | } | |
505 | ||
506 | ||
507 | //____________________________________________________________________________ | |
508 | void AliPHOSTrackSegmentMakerv1::MakeLinks()const | |
509 | { | |
510 | // Finds distances (links) between all EMC and CPV clusters, | |
511 | // which are not further apart from each other than fRcpv | |
512 | // and sort them in accordance with this distance | |
513 | ||
514 | fLinkUpArray->Clear() ; | |
515 | ||
516 | AliPHOSCpvRecPoint * cpv ; | |
517 | AliPHOSEmcRecPoint * emcclu ; | |
518 | ||
519 | Int_t iLinkUp = 0 ; | |
520 | ||
521 | Int_t iEmcRP; | |
522 | for(iEmcRP = fEmcFirst; iEmcRP < fEmcLast; iEmcRP++ ) { | |
523 | emcclu = dynamic_cast<AliPHOSEmcRecPoint *>(fEMCRecPoints->At(iEmcRP)) ; | |
524 | ||
525 | //Bool_t toofar ; | |
526 | Int_t iCpv = 0 ; | |
527 | for(iCpv = fCpvFirst; iCpv < fCpvLast;iCpv++ ) { | |
528 | ||
529 | cpv = dynamic_cast<AliPHOSCpvRecPoint *>(fCPVRecPoints->At(iCpv)) ; | |
530 | Int_t track = -1 ; | |
531 | Float_t dx,dz ; | |
532 | GetDistanceInPHOSPlane(emcclu, cpv, track,dx,dz) ; | |
533 | if(TMath::Sqrt(dx*dx+dz*dz) < fRcpv ){ | |
534 | new ((*fLinkUpArray)[iLinkUp++]) AliPHOSLink(dx, dz, iEmcRP, iCpv, track) ; | |
535 | } | |
536 | } | |
537 | } | |
538 | ||
539 | fLinkUpArray->Sort() ; //first links with smallest distances | |
540 | } | |
541 | ||
542 | //____________________________________________________________________________ | |
543 | void AliPHOSTrackSegmentMakerv1::MakePairs() | |
544 | { | |
545 | // Using the previously made list of "links", we found the smallest link - i.e. | |
546 | // link with the least distance between EMC and CPV and pointing to still | |
547 | // unassigned RecParticles. We assign these RecPoints to TrackSegment and | |
548 | // remove them from the list of "unassigned". | |
549 | ||
550 | //Make arrays to mark clusters already chosen | |
551 | Int_t index; | |
552 | ||
553 | Int_t * emcExist = 0; | |
554 | if(fEmcLast > fEmcFirst) { | |
555 | emcExist = new Int_t[fEmcLast-fEmcFirst] ; | |
556 | for(index = 0; index <fEmcLast-fEmcFirst; index ++) | |
557 | emcExist[index] = 1 ; | |
558 | } | |
559 | else | |
560 | return; | |
561 | ||
562 | Bool_t * cpvExist = 0; | |
563 | if(fCpvLast > fCpvFirst) { | |
564 | cpvExist = new Bool_t[fCpvLast-fCpvFirst] ; | |
565 | for(index = 0; index <fCpvLast-fCpvFirst; index ++) | |
566 | cpvExist[index] = kTRUE ; | |
567 | } | |
568 | ||
569 | // Finds the smallest links and makes pairs of CPV and EMC clusters with smallest distance | |
570 | TIter nextUp(fLinkUpArray) ; | |
571 | ||
572 | AliPHOSLink * linkUp ; | |
573 | ||
574 | AliPHOSCpvRecPoint * nullpointer = 0 ; | |
575 | ||
576 | while ( (linkUp = static_cast<AliPHOSLink *>(nextUp()) ) ){ | |
577 | ||
578 | if(emcExist[linkUp->GetEmc()-fEmcFirst] != -1){ | |
579 | ||
580 | //array cpvExist[] should be non-zero as far as linkUp exists | |
581 | //But Coverity requires to check it | |
582 | if(cpvExist && cpvExist[linkUp->GetCpv()-fCpvFirst]){ //CPV still exist | |
583 | Float_t dx,dz ; | |
584 | linkUp->GetXZ(dx,dz) ; | |
585 | new ((* fTrackSegments)[fNTrackSegments]) | |
586 | AliPHOSTrackSegment(static_cast<AliPHOSEmcRecPoint *>(fEMCRecPoints->At(linkUp->GetEmc())) , | |
587 | static_cast<AliPHOSCpvRecPoint *>(fCPVRecPoints->At(linkUp->GetCpv())) , | |
588 | linkUp->GetTrack(),dx,dz) ; | |
589 | ||
590 | (static_cast<AliPHOSTrackSegment *>(fTrackSegments->At(fNTrackSegments)))->SetIndexInList(fNTrackSegments); | |
591 | fNTrackSegments++ ; | |
592 | emcExist[linkUp->GetEmc()-fEmcFirst] = -1 ; //Mark emc that Cpv was found | |
593 | //mark CPV recpoint as already used | |
594 | cpvExist[linkUp->GetCpv()-fCpvFirst] = kFALSE ; | |
595 | } //if CpvUp still exist | |
596 | } | |
597 | } | |
598 | ||
599 | //look through emc recPoints left without CPV | |
600 | if(emcExist){ //if there is emc rec point | |
601 | Int_t iEmcRP ; | |
602 | for(iEmcRP = 0; iEmcRP < fEmcLast-fEmcFirst ; iEmcRP++ ){ | |
603 | if(emcExist[iEmcRP] > 0 ){ | |
604 | Int_t track = -1 ; | |
605 | Float_t dx,dz ; | |
606 | AliPHOSEmcRecPoint *emcclu = dynamic_cast<AliPHOSEmcRecPoint *>(fEMCRecPoints->At(iEmcRP+fEmcFirst)); | |
607 | GetDistanceInPHOSPlane(emcclu, 0, track,dx,dz); | |
608 | if(track<0) | |
609 | new ((*fTrackSegments)[fNTrackSegments]) AliPHOSTrackSegment(emcclu,nullpointer) ; | |
610 | else | |
611 | new ((*fTrackSegments)[fNTrackSegments]) AliPHOSTrackSegment(emcclu,0,track,dx,dz); | |
612 | (static_cast<AliPHOSTrackSegment *>(fTrackSegments->At(fNTrackSegments)))->SetIndexInList(fNTrackSegments); | |
613 | fNTrackSegments++; | |
614 | } | |
615 | } | |
616 | } | |
617 | delete [] emcExist ; | |
618 | if(cpvExist) | |
619 | delete [] cpvExist ; | |
620 | } | |
621 | ||
622 | //____________________________________________________________________________ | |
623 | void AliPHOSTrackSegmentMakerv1::Clusters2TrackSegments(Option_t *option) | |
624 | { | |
625 | // Steering method to perform track segment construction for the current event | |
626 | // Returns an array with the found track-segments. | |
627 | ||
628 | if(strstr(option,"tim")) | |
629 | gBenchmark->Start("PHOSTSMaker"); | |
630 | ||
631 | if(strstr(option,"print")) { | |
632 | Print() ; | |
633 | return ; | |
634 | } | |
635 | ||
636 | //Make some initializations | |
637 | fNTrackSegments = 0 ; | |
638 | fEmcFirst = 0 ; | |
639 | fEmcLast = 0 ; | |
640 | fCpvFirst = 0 ; | |
641 | fCpvLast = 0 ; | |
642 | ||
643 | fTrackSegments->Clear(); | |
644 | ||
645 | // if(!ReadRecPoints(ievent)) continue; //reads RecPoints for event ievent | |
646 | ||
647 | for(fModule = 1; fModule <= fGeom->GetNModules() ; fModule++ ) { | |
648 | FillOneModule() ; | |
649 | MakeLinks() ; | |
650 | MakePairs() ; | |
651 | } | |
652 | ||
653 | if(strstr(option,"deb")) | |
654 | PrintTrackSegments(option); | |
655 | ||
656 | if(strstr(option,"tim")){ | |
657 | gBenchmark->Stop("PHOSTSMaker"); | |
658 | Info("Exec", "took %f seconds for making TS", | |
659 | gBenchmark->GetCpuTime("PHOSTSMaker")); | |
660 | } | |
661 | } | |
662 | ||
663 | //____________________________________________________________________________ | |
664 | void AliPHOSTrackSegmentMakerv1::Print(const Option_t *)const | |
665 | { | |
666 | // Print TrackSegmentMaker parameters | |
667 | ||
668 | TString message("") ; | |
669 | if( strcmp(GetName(), "") != 0 ) { | |
670 | message = "\n======== AliPHOSTrackSegmentMakerv1 ========\n" ; | |
671 | message += "Making Track segments\n" ; | |
672 | message += "with parameters:\n" ; | |
673 | message += " Maximal EMC - CPV distance (cm) %f\n" ; | |
674 | message += "============================================\n" ; | |
675 | Info("Print", message.Data(),fRcpv) ; | |
676 | } | |
677 | else | |
678 | Info("Print", "AliPHOSTrackSegmentMakerv1 not initialized ") ; | |
679 | } | |
680 | ||
681 | //____________________________________________________________________________ | |
682 | void AliPHOSTrackSegmentMakerv1::PrintTrackSegments(Option_t * option) | |
683 | { | |
684 | // option deb - prints # of found TrackSegments | |
685 | // option deb all - prints as well indexed of found RecParticles assigned to the TS | |
686 | ||
687 | Info("PrintTrackSegments", "Results from TrackSegmentMaker:") ; | |
688 | printf(" Found %d TrackSegments\n", fTrackSegments->GetEntriesFast() ); | |
689 | ||
690 | if(strstr(option,"all")) { // printing found TS | |
691 | printf("TrackSegment # EMC RP# CPV RP#\n") ; | |
692 | Int_t index; | |
693 | for (index = 0 ; index <fTrackSegments->GetEntriesFast() ; index++) { | |
694 | AliPHOSTrackSegment * ts = (AliPHOSTrackSegment * )fTrackSegments->At(index) ; | |
695 | printf(" %d %d %d \n", ts->GetIndexInList(), ts->GetEmcIndex(), ts->GetCpvIndex() ) ; | |
696 | } | |
697 | } | |
698 | } | |
699 |