]> git.uio.no Git - u/mrichter/AliRoot.git/blame - PHOS/AliPHOSTrackSegmentMakerv1.cxx
Bug fix for CPV-EMC distance
[u/mrichter/AliRoot.git] / PHOS / AliPHOSTrackSegmentMakerv1.cxx
CommitLineData
d15a28e7 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 **************************************************************************/
b2a60966 15/* $Id$ */
702ab87e 16
17/* History of cvs commits:
18 *
19 * $Log$
78881c42 20 * Revision 1.87 2007/05/04 14:49:29 policheh
21 * AliPHOSRecPoint inheritance from AliCluster
22 *
9ee9f78d 23 * Revision 1.86 2007/04/02 15:00:16 cvetan
24 * No more calls to gAlice in the reconstruction
25 *
999f9a8f 26 * Revision 1.85 2007/03/28 19:18:15 kharlov
27 * RecPoints recalculation in TSM removed
28 *
8c1fb709 29 * Revision 1.84 2007/03/07 07:01:21 hristov
30 * Fixing copy/paste erro. Additional protections
31 *
b89c16b2 32 * Revision 1.83 2007/03/06 21:07:37 kharlov
33 * DP: xz CPV-EMC distance filled to TS
34 *
d7ddbb70 35 * Revision 1.82 2007/03/06 06:54:48 kharlov
36 * DP:Calculation of cluster properties dep. on vertex added
37 *
e84d3def 38 * Revision 1.81 2007/02/05 10:02:40 kharlov
39 * Module numbering is corrected
40 *
a2ee5b34 41 * Revision 1.80 2006/08/28 10:01:56 kharlov
42 * Effective C++ warnings fixed (Timur Pocheptsov)
43 *
3663622c 44 * Revision 1.79 2006/04/25 12:41:15 hristov
45 * Moving non-persistent data to AliESDfriend (Yu.Belikov)
46 *
15e85efa 47 * Revision 1.78 2005/11/18 13:04:51 hristov
48 * Bug fix
49 *
a494460a 50 * Revision 1.77 2005/11/17 23:34:36 hristov
51 * Corrected logics
52 *
f74a631d 53 * Revision 1.76 2005/11/17 22:29:12 hristov
54 * Faster version, no attempt to match tracks outside the PHOS acceptance
55 *
0d9aa319 56 * Revision 1.75 2005/11/17 12:35:27 hristov
57 * Use references instead of objects. Avoid to create objects when they are not really needed
58 *
7b51037f 59 * Revision 1.74 2005/07/08 14:01:36 hristov
60 * Tracking in non-uniform nmagnetic field (Yu.Belikov)
61 *
c84a5e9e 62 * Revision 1.73 2005/05/28 14:19:05 schutz
63 * Compilation warnings fixed by T.P.
64 *
702ab87e 65 */
66
d15a28e7 67//_________________________________________________________________________
b2a60966 68// Implementation version 1 of algorithm class to construct PHOS track segments
f035f6ce 69// Track segment for PHOS is list of
194f9939 70// EMC RecPoint + (possibly) CPV RecPoint
a4e98857 71// To find TrackSegments we do the following:
72// for each EMC RecPoints we look at
194f9939 73// CPV RecPoints in the radious fRcpv.
a4e98857 74// If there is such a CPV RecPoint,
194f9939 75// we make "Link" it is just indexes of EMC and CPV RecPoint and distance
a4e98857 76// between them in the PHOS plane.
77// Then we sort "Links" and starting from the
78// least "Link" pointing to the unassined EMC and CPV RecPoints assing them to
79// new TrackSegment.
194f9939 80// If there is no CPV RecPoint we make TrackSegment
a4e98857 81// consisting from EMC alone. There is no TrackSegments without EMC RecPoint.
f444a19f 82//// In principle this class should be called from AliPHOSReconstructor, but
a4e98857 83// one can use it as well in standalone mode.
84// Use case:
fc12304f 85// root [0] AliPHOSTrackSegmentMakerv1 * t = new AliPHOSTrackSegmentMaker("galice.root", "tracksegmentsname", "recpointsname")
a4e98857 86// Warning in <TDatabasePDG::TDatabasePDG>: object already instantiated
fc12304f 87// // reads gAlice from header file "galice.root", uses recpoints stored in the branch names "recpointsname" (default = "Default")
88// // and saves recpoints in branch named "tracksegmentsname" (default = "recpointsname")
a4e98857 89// root [1] t->ExecuteTask()
a4e98857 90// root [3] t->SetTrackSegmentsBranch("max distance 5 cm")
91// root [4] t->ExecuteTask("deb all time")
f035f6ce 92//
fc12304f 93//*-- Author: Dmitri Peressounko (RRC Ki & SUBATECH) & Yves Schutz (SUBATECH)
b2a60966 94//
d15a28e7 95
96// --- ROOT system ---
2731cd1e 97#include "TTree.h"
2731cd1e 98#include "TBenchmark.h"
d15a28e7 99
21cd0c07 100// --- Standard library ---
194f9939 101#include "Riostream.h"
d15a28e7 102// --- AliRoot header files ---
e957fea8 103#include "AliPHOSGeometry.h"
d15a28e7 104#include "AliPHOSTrackSegmentMakerv1.h"
105#include "AliPHOSTrackSegment.h"
106#include "AliPHOSLink.h"
7b7c1533 107#include "AliPHOSGetter.h"
aa0b9641 108#include "AliESD.h"
109#include "AliESDtrack.h"
d15a28e7 110
111ClassImp( AliPHOSTrackSegmentMakerv1)
112
113
114//____________________________________________________________________________
3663622c 115AliPHOSTrackSegmentMakerv1::AliPHOSTrackSegmentMakerv1() :
116 AliPHOSTrackSegmentMaker(),
117 fDefaultInit(kTRUE),
118 fWrite(kFALSE),
119 fNTrackSegments(0),
120 fRcpv(0.f),
121 fRtpc(0.f),
122 fLinkUpArray(0),
123 fEmcFirst(0),
124 fEmcLast(0),
125 fCpvFirst(0),
126 fCpvLast(0),
127 fModule(0),
128 fTrackSegmentsInRun(0)
129
d15a28e7 130{
7b7c1533 131 // default ctor (to be used mainly by Streamer)
8d0f3f77 132 InitParameters() ;
d15a28e7 133}
7b7c1533 134
9f616d61 135//____________________________________________________________________________
3663622c 136AliPHOSTrackSegmentMakerv1::AliPHOSTrackSegmentMakerv1(const TString & alirunFileName, const TString & eventFolderName) :
137 AliPHOSTrackSegmentMaker(alirunFileName, eventFolderName),
138 fDefaultInit(kFALSE),
139 fWrite(kFALSE),
140 fNTrackSegments(0),
141 fRcpv(0.f),
142 fRtpc(0.f),
143 fLinkUpArray(0),
144 fEmcFirst(0),
145 fEmcLast(0),
146 fCpvFirst(0),
147 fCpvLast(0),
148 fModule(0),
149 fTrackSegmentsInRun(0)
2731cd1e 150{
151 // ctor
8d0f3f77 152 InitParameters() ;
7b7c1533 153 Init() ;
aa0b9641 154 fESD = 0;
2731cd1e 155}
98cbd830 156
3663622c 157
158AliPHOSTrackSegmentMakerv1::AliPHOSTrackSegmentMakerv1(const AliPHOSTrackSegmentMakerv1 & tsm) :
159 AliPHOSTrackSegmentMaker(tsm),
160 fDefaultInit(kFALSE),
161 fWrite(kFALSE),
162 fNTrackSegments(0),
163 fRcpv(0.f),
164 fRtpc(0.f),
165 fLinkUpArray(0),
166 fEmcFirst(0),
167 fEmcLast(0),
168 fCpvFirst(0),
169 fCpvLast(0),
170 fModule(0),
171 fTrackSegmentsInRun(0)
172{
173 // cpy ctor: no implementation yet
174 // requested by the Coding Convention
175 Fatal("cpy ctor", "not implemented") ;
176}
177
178
2731cd1e 179//____________________________________________________________________________
180 AliPHOSTrackSegmentMakerv1::~AliPHOSTrackSegmentMakerv1()
181{
182 // dtor
92f521a9 183 // fDefaultInit = kTRUE if TrackSegmentMaker created by default ctor (to get just the parameters)
88cb7938 184 if (!fDefaultInit)
185 delete fLinkUpArray ;
d15a28e7 186}
9f616d61 187
8d0f3f77 188
fc12304f 189//____________________________________________________________________________
190const TString AliPHOSTrackSegmentMakerv1::BranchName() const
191{
88cb7938 192
193 return GetName() ;
fc12304f 194}
195
d15a28e7 196//____________________________________________________________________________
2731cd1e 197void AliPHOSTrackSegmentMakerv1::FillOneModule()
9f616d61 198{
f035f6ce 199 // Finds first and last indexes between which
200 // clusters from one PHOS module are
88cb7938 201
202 AliPHOSGetter * gime = AliPHOSGetter::Instance() ;
fc12304f 203
fbf811ec 204 TObjArray * emcRecPoints = gime->EmcRecPoints() ;
205 TObjArray * cpvRecPoints = gime->CpvRecPoints() ;
9688c1dd 206
2731cd1e 207 //First EMC clusters
7b7c1533 208 Int_t totalEmc = emcRecPoints->GetEntriesFast() ;
2731cd1e 209 for(fEmcFirst = fEmcLast; (fEmcLast < totalEmc) &&
29b077b5 210 ((dynamic_cast<AliPHOSRecPoint *>(emcRecPoints->At(fEmcLast)))->GetPHOSMod() == fModule );
2731cd1e 211 fEmcLast ++) ;
212
2731cd1e 213 //Now CPV clusters
7b7c1533 214 Int_t totalCpv = cpvRecPoints->GetEntriesFast() ;
6ad0bfa0 215
2731cd1e 216 for(fCpvFirst = fCpvLast; (fCpvLast < totalCpv) &&
88cb7938 217 ((dynamic_cast<AliPHOSRecPoint *>(cpvRecPoints->At(fCpvLast)))->GetPHOSMod() == fModule );
218 fCpvLast ++) ;
9688c1dd 219
d15a28e7 220}
7b7c1533 221
d15a28e7 222//____________________________________________________________________________
e84d3def 223void AliPHOSTrackSegmentMakerv1::GetDistanceInPHOSPlane(AliPHOSEmcRecPoint * emcClu,
224 AliPHOSCpvRecPoint * cpvClu,
225 Int_t &trackindex,
226 Float_t &dx, Float_t &dz) const
d15a28e7 227{
194f9939 228 // Calculates the distance between the EMC RecPoint and the CPV RecPoint
a4e98857 229 // Clusters are sorted in "rows" and "columns" of width 1 cm
f035f6ce 230
e84d3def 231// Float_t delta = 1 ; // Width of the rows in sorting of RecPoints (in cm)
232// // if you change this value, change it as well in xxxRecPoint::Compare()
743cb288 233 Float_t distance2Track = fRtpc ;
194f9939 234
235 trackindex = -1 ; // closest track within fRCpv
236
aa0b9641 237 TVector3 vecEmc ; // Local position of EMC recpoint
e84d3def 238 TVector3 vecP ; // Momentum direction at CPV plain
239 TVector3 vecPloc ; // Momentum direction at CPV plain
2731cd1e 240
194f9939 241 //toofar = kTRUE ;
e84d3def 242 if(emcClu->GetPHOSMod() != cpvClu->GetPHOSMod()){
243 dx=999. ;
244 dz=999. ;
245 return ;
246 }
247
248 emcClu->GetLocalPosition(vecEmc) ;
2731cd1e 249
e84d3def 250 AliPHOSGetter * gime = AliPHOSGetter::Instance() ;
251 const AliPHOSGeometry * geom = gime->PHOSGeometry() ;
194f9939 252
e84d3def 253 Double_t xCPV,zCPV ; //EMC-projected coordinates of CPV cluster
254 TVector3 cpvGlobal; // Global position of the CPV recpoint
9ee9f78d 255 geom->GetGlobalPHOS((AliPHOSRecPoint*)cpvClu,cpvGlobal);
e84d3def 256 Double_t vtxCPV[3]={cpvGlobal.X(),cpvGlobal.Y(),cpvGlobal.Z()} ;
78881c42 257 Int_t dummyMod ;
e84d3def 258
259 if (fESD == 0x0) {
260 //if no track information available, assume straight line from IP to emcal
e84d3def 261 geom->ImpactOnEmc(vtxCPV,cpvGlobal.Theta(),cpvGlobal.Phi(),dummyMod,xCPV,zCPV) ;
262 dx=xCPV - vecEmc.X() ;
263 dz=zCPV - vecEmc.Z() ;
264 return ;
265 }
266
267 //if there is ESD try to correct distance using TPC information on particle direct in CPV
268 if (fESD != 0x0) {
0d9aa319 269
e84d3def 270 Double_t rCPV = cpvGlobal.Pt() ;// Radius from IP to current point
271
272 // Extrapolate the global track direction if any to CPV and find the closest track
273 Int_t nTracks = fESD->GetNumberOfTracks();
274 Int_t iClosestTrack = -1;
275 Double_t minDistance = 1.e6;
276 TVector3 inPHOS ;
277
278 AliESDtrack *track;
279 Double_t xyz[3] ;
280 Double_t pxyz[3];
281 for (Int_t iTrack=0; iTrack<nTracks; iTrack++) {
282 track = fESD->GetTrack(iTrack);
283 if (!track->GetXYZAt(rCPV, fESD->GetMagneticField(), xyz))
23904d16 284 continue; //track coord on the cylinder of PHOS radius
e84d3def 285 if ((TMath::Abs(xyz[0])+TMath::Abs(xyz[1])+TMath::Abs(xyz[2]))<=0)
23904d16 286 continue;
e84d3def 287 //Check if this track hits PHOS
288 inPHOS.SetXYZ(xyz[0],xyz[1],xyz[2]);
289 distance2Track = inPHOS.Angle(cpvGlobal) ;
290 // Find the closest track to the CPV recpoint
291 if (distance2Track < minDistance) {
292 minDistance = distance2Track;
293 iClosestTrack = iTrack;
aa0b9641 294 }
e84d3def 295 }
aa0b9641 296
e84d3def 297 if (iClosestTrack != -1) {
298 track = fESD->GetTrack(iClosestTrack);
299 if (track->GetPxPyPzAt(rCPV, fESD->GetMagneticField(), pxyz)) { // track momentum ibid.
300 vecP.SetXYZ(pxyz[0],pxyz[1],pxyz[2]);
301 Int_t dummyMod ;
302 geom->ImpactOnEmc(vtxCPV,vecP.Theta(),vecP.Phi(),dummyMod,xCPV,zCPV) ;
aa0b9641 303 }
e84d3def 304 }
194f9939 305
e84d3def 306 if(minDistance < fRtpc ){
307 trackindex = iClosestTrack ;
aa0b9641 308 }
bfc17d18 309 }
e84d3def 310 if(trackindex!=-1){
78881c42 311 // If the closest global track is found, calculate EMC-CPV distance from it
e84d3def 312 dx=xCPV - vecEmc.X() ;
313 dz=zCPV - vecEmc.Z() ;
314 }
315 else{
78881c42 316 // If no global track was found, just take the nearest CPV point
317 geom->ImpactOnEmc(vtxCPV,cpvGlobal.Theta(),cpvGlobal.Phi(),dummyMod,xCPV,zCPV) ;
318 dx=xCPV - vecEmc.X() ;
319 dz=zCPV - vecEmc.Z() ;
e84d3def 320 }
321 return ;
d15a28e7 322}
7b7c1533 323//____________________________________________________________________________
324void AliPHOSTrackSegmentMakerv1::Init()
325{
326 // Make all memory allocations that are not possible in default constructor
327
adcca1e6 328 AliPHOSGetter* gime = AliPHOSGetter::Instance();
329 if(!gime)
330 gime = AliPHOSGetter::Instance(GetTitle(), fEventFolderName.Data());
fbf811ec 331
7b7c1533 332 fLinkUpArray = new TClonesArray("AliPHOSLink", 1000);
88cb7938 333 if ( !gime->TrackSegmentMaker() ) {
334 gime->PostTrackSegmentMaker(this);
335 }
7b7c1533 336}
337
8d0f3f77 338//____________________________________________________________________________
339void AliPHOSTrackSegmentMakerv1::InitParameters()
340{
e957fea8 341 //Initializes parameters
743cb288 342 fRcpv = 10. ;
343 fRtpc = 4. ;
8d0f3f77 344 fEmcFirst = 0 ;
345 fEmcLast = 0 ;
346 fCpvFirst = 0 ;
347 fCpvLast = 0 ;
348 fLinkUpArray = 0 ;
adcca1e6 349 fWrite = kTRUE ;
88cb7938 350 fTrackSegmentsInRun = 0 ;
eabde521 351 SetEventRange(0,-1) ;
8d0f3f77 352}
353
354
d15a28e7 355//____________________________________________________________________________
baef0810 356void AliPHOSTrackSegmentMakerv1::MakeLinks()const
d15a28e7 357{
194f9939 358 // Finds distances (links) between all EMC and CPV clusters,
fbf811ec 359 // which are not further apart from each other than fRcpv
f035f6ce 360 // and sort them in accordance with this distance
9688c1dd 361
88cb7938 362 AliPHOSGetter * gime = AliPHOSGetter::Instance() ;
fbf811ec 363 TObjArray * emcRecPoints = gime->EmcRecPoints() ;
364 TObjArray * cpvRecPoints = gime->CpvRecPoints() ;
7b7c1533 365
2731cd1e 366 fLinkUpArray->Clear() ;
2731cd1e 367
2bb500e5 368 AliPHOSCpvRecPoint * cpv ;
92862013 369 AliPHOSEmcRecPoint * emcclu ;
28c3a259 370
d15a28e7 371 Int_t iLinkUp = 0 ;
372
28c3a259 373 Int_t iEmcRP;
2731cd1e 374 for(iEmcRP = fEmcFirst; iEmcRP < fEmcLast; iEmcRP++ ) {
29b077b5 375 emcclu = dynamic_cast<AliPHOSEmcRecPoint *>(emcRecPoints->At(iEmcRP)) ;
2731cd1e 376
194f9939 377 //Bool_t toofar ;
2731cd1e 378 Int_t iCpv = 0 ;
379 for(iCpv = fCpvFirst; iCpv < fCpvLast;iCpv++ ) {
28c3a259 380
2bb500e5 381 cpv = dynamic_cast<AliPHOSCpvRecPoint *>(cpvRecPoints->At(iCpv)) ;
194f9939 382 Int_t track = -1 ;
e84d3def 383 Float_t dx,dz ;
384 GetDistanceInPHOSPlane(emcclu, cpv, track,dx,dz) ;
385 if(TMath::Sqrt(dx*dx+dz*dz) < fRcpv ){
386 new ((*fLinkUpArray)[iLinkUp++]) AliPHOSLink(dx, dz, iEmcRP, iCpv, track) ;
28c3a259 387 }
d15a28e7 388 }
28c3a259 389 }
d15a28e7 390
9688c1dd 391 fLinkUpArray->Sort() ; //first links with smallest distances
d15a28e7 392}
28c3a259 393
d15a28e7 394//____________________________________________________________________________
2731cd1e 395void AliPHOSTrackSegmentMakerv1::MakePairs()
6ad0bfa0 396{
f035f6ce 397 // Using the previously made list of "links", we found the smallest link - i.e.
a4e98857 398 // link with the least distance between EMC and CPV and pointing to still
f035f6ce 399 // unassigned RecParticles. We assign these RecPoints to TrackSegment and
400 // remove them from the list of "unassigned".
88cb7938 401
402 AliPHOSGetter * gime = AliPHOSGetter::Instance() ;
403
fbf811ec 404 TObjArray * emcRecPoints = gime->EmcRecPoints() ;
405 TObjArray * cpvRecPoints = gime->CpvRecPoints() ;
88cb7938 406 TClonesArray * trackSegments = gime->TrackSegments();
9688c1dd 407
01a599c9 408 //Make arrays to mark clusters already chosen
2731cd1e 409 Int_t * emcExist = 0;
410 if(fEmcLast > fEmcFirst)
411 emcExist = new Int_t[fEmcLast-fEmcFirst] ;
412
413 Int_t index;
414 for(index = 0; index <fEmcLast-fEmcFirst; index ++)
415 emcExist[index] = 1 ;
416
417 Bool_t * cpvExist = 0;
418 if(fCpvLast > fCpvFirst)
419 cpvExist = new Bool_t[fCpvLast-fCpvFirst] ;
420 for(index = 0; index <fCpvLast-fCpvFirst; index ++)
421 cpvExist[index] = kTRUE ;
422
2731cd1e 423
424 // Finds the smallest links and makes pairs of CPV and EMC clusters with smallest distance
2731cd1e 425 TIter nextUp(fLinkUpArray) ;
d15a28e7 426
d15a28e7 427 AliPHOSLink * linkUp ;
9688c1dd 428
2bb500e5 429 AliPHOSCpvRecPoint * nullpointer = 0 ;
9688c1dd 430
29b077b5 431 while ( (linkUp = static_cast<AliPHOSLink *>(nextUp()) ) ){
9688c1dd 432
194f9939 433 if(emcExist[linkUp->GetEmc()-fEmcFirst] != -1){
d15a28e7 434
194f9939 435 if(cpvExist[linkUp->GetCpv()-fCpvFirst]){ //CPV still exist
d7ddbb70 436 Float_t dx,dz ;
437 linkUp->GetXZ(dx,dz) ;
194f9939 438 new ((* trackSegments)[fNTrackSegments])
439 AliPHOSTrackSegment(dynamic_cast<AliPHOSEmcRecPoint *>(emcRecPoints->At(linkUp->GetEmc())) ,
440 dynamic_cast<AliPHOSCpvRecPoint *>(cpvRecPoints->At(linkUp->GetCpv())) ,
d7ddbb70 441 linkUp->GetTrack(),dx,dz) ;
194f9939 442
88cb7938 443 (dynamic_cast<AliPHOSTrackSegment *>(trackSegments->At(fNTrackSegments)))->SetIndexInList(fNTrackSegments);
444 fNTrackSegments++ ;
88cb7938 445 emcExist[linkUp->GetEmc()-fEmcFirst] = -1 ; //Mark emc that Cpv was found
446 //mark CPV recpoint as already used
194f9939 447 cpvExist[linkUp->GetCpv()-fCpvFirst] = kFALSE ;
448 } //if CpvUp still exist
28c3a259 449 }
88cb7938 450 }
28c3a259 451
194f9939 452 //look through emc recPoints left without CPV
2731cd1e 453 if(emcExist){ //if there is emc rec point
454 Int_t iEmcRP ;
455 for(iEmcRP = 0; iEmcRP < fEmcLast-fEmcFirst ; iEmcRP++ ){
456 if(emcExist[iEmcRP] > 0 ){
88cb7938 457 new ((*trackSegments)[fNTrackSegments])
458 AliPHOSTrackSegment(dynamic_cast<AliPHOSEmcRecPoint *>(emcRecPoints->At(iEmcRP+fEmcFirst)),
459 nullpointer) ;
460 (dynamic_cast<AliPHOSTrackSegment *>(trackSegments->At(fNTrackSegments)))->SetIndexInList(fNTrackSegments);
461 fNTrackSegments++;
2731cd1e 462 }
d15a28e7 463 }
d15a28e7 464 }
780a31c1 465 delete [] emcExist ;
466 delete [] cpvExist ;
d15a28e7 467}
468
469//____________________________________________________________________________
eabde521 470void AliPHOSTrackSegmentMakerv1::Exec(Option_t *option)
d15a28e7 471{
eabde521 472 // Steering method to perform track segment construction for events
473 // in the range from fFirstEvent to fLastEvent.
474 // This range is optionally set by SetEventRange().
475 // if fLastEvent=-1 (by default), then process events until the end.
88cb7938 476
2731cd1e 477 if(strstr(option,"tim"))
b3f97575 478 gBenchmark->Start("PHOSTSMaker");
7b7c1533 479
480 if(strstr(option,"print")) {
88cb7938 481 Print() ;
7b7c1533 482 return ;
483 }
88cb7938 484
adcca1e6 485 AliPHOSGetter * gime = AliPHOSGetter::Instance() ;
fb43ada4 486
55ea5766 487 const AliPHOSGeometry * geom = gime->PHOSGeometry() ;
7b7c1533 488
0588c896 489 if (fLastEvent == -1)
490 fLastEvent = gime->MaxEvent() - 1 ;
491 else
492 fLastEvent = TMath::Min(fFirstEvent,gime->MaxEvent());
eabde521 493 Int_t nEvents = fLastEvent - fFirstEvent + 1;
01a599c9 494
fb43ada4 495 Int_t ievent ;
496 for (ievent = fFirstEvent; ievent <= fLastEvent; ievent++) {
e84d3def 497 gime->Event(ievent,"DR") ;
0588c896 498 //Make some initializations
55ea5766 499 fNTrackSegments = 0 ;
500 fEmcFirst = 0 ;
501 fEmcLast = 0 ;
502 fCpvFirst = 0 ;
503 fCpvLast = 0 ;
88cb7938 504
505 gime->TrackSegments()->Clear();
55ea5766 506
9688c1dd 507 // if(!ReadRecPoints(ievent)) continue; //reads RecPoints for event ievent
508
88cb7938 509 for(fModule = 1; fModule <= geom->GetNModules() ; fModule++ ) {
2731cd1e 510 FillOneModule() ;
2731cd1e 511 MakeLinks() ;
2731cd1e 512 MakePairs() ;
2731cd1e 513 }
28c3a259 514
90cceaf6 515 WriteTrackSegments() ;
7b7c1533 516
2731cd1e 517 if(strstr(option,"deb"))
88cb7938 518 PrintTrackSegments(option);
94de8339 519
520 //increment the total number of track segments per run
88cb7938 521 fTrackSegmentsInRun += gime->TrackSegments()->GetEntriesFast() ;
2731cd1e 522 }
88cb7938 523
2731cd1e 524 if(strstr(option,"tim")){
525 gBenchmark->Stop("PHOSTSMaker");
21cd0c07 526 Info("Exec", "took %f seconds for making TS %f seconds per event",
88cb7938 527 gBenchmark->GetCpuTime("PHOSTSMaker"),
eabde521 528 gBenchmark->GetCpuTime("PHOSTSMaker")/nEvents) ;
88cb7938 529 }
adcca1e6 530 if(fWrite) //do not unload in "on flight" mode
531 Unload();
e84d3def 532}
88cb7938 533//____________________________________________________________________________
534void AliPHOSTrackSegmentMakerv1::Unload()
535{
e957fea8 536 // Unloads the task from the folder
88cb7938 537 AliPHOSGetter * gime = AliPHOSGetter::Instance() ;
538 gime->PhosLoader()->UnloadRecPoints() ;
539 gime->PhosLoader()->UnloadTracks() ;
d15a28e7 540}
7b7c1533 541
d15a28e7 542//____________________________________________________________________________
702ab87e 543void AliPHOSTrackSegmentMakerv1::Print(const Option_t *)const
a4e98857 544{
baef0810 545 // Print TrackSegmentMaker parameters
546
21cd0c07 547 TString message("") ;
7b7c1533 548 if( strcmp(GetName(), "") != 0 ) {
21cd0c07 549 message = "\n======== AliPHOSTrackSegmentMakerv1 ========\n" ;
550 message += "Making Track segments\n" ;
551 message += "with parameters:\n" ;
194f9939 552 message += " Maximal EMC - CPV distance (cm) %f\n" ;
21cd0c07 553 message += "============================================\n" ;
554 Info("Print", message.Data(),fRcpv) ;
2731cd1e 555 }
556 else
21cd0c07 557 Info("Print", "AliPHOSTrackSegmentMakerv1 not initialized ") ;
d15a28e7 558}
7b7c1533 559
d15a28e7 560//____________________________________________________________________________
90cceaf6 561void AliPHOSTrackSegmentMakerv1::WriteTrackSegments()
a4e98857 562{
f035f6ce 563 // Writes found TrackSegments to TreeR. Creates branches
564 // "PHOSTS" and "AliPHOSTrackSegmentMaker" with the same title.
565 // In the former branch found TrackSegments are stored, while
566 // in the latter all parameters, with which TS were made.
567 // ROOT does not allow overwriting existing branches, therefore
a4e98857 568 // first we check, if branches with the same title already exist.
f035f6ce 569 // If yes - exits without writing.
88cb7938 570
571 AliPHOSGetter *gime = AliPHOSGetter::Instance() ;
55ea5766 572
fbf811ec 573 TClonesArray * trackSegments = gime->TrackSegments() ;
55ea5766 574 trackSegments->Expand(trackSegments->GetEntriesFast()) ;
8d0f3f77 575
adcca1e6 576 if(fWrite){ //We write TreeT
577 TTree * treeT = gime->TreeT();
578
579 //First TS
580 Int_t bufferSize = 32000 ;
581 TBranch * tsBranch = treeT->Branch("PHOSTS",&trackSegments,bufferSize);
582 tsBranch->Fill() ;
583
584 gime->WriteTracks("OVERWRITE");
585 gime->WriteTrackSegmentMaker("OVERWRITE");
586 }
2731cd1e 587}
98cbd830 588
98cbd830 589
2731cd1e 590//____________________________________________________________________________
a4e98857 591void AliPHOSTrackSegmentMakerv1::PrintTrackSegments(Option_t * option)
592{
f035f6ce 593 // option deb - prints # of found TrackSegments
594 // option deb all - prints as well indexed of found RecParticles assigned to the TS
9688c1dd 595
88cb7938 596 TClonesArray * trackSegments = AliPHOSGetter::Instance()->TrackSegments() ;
21cd0c07 597
88cb7938 598 Info("PrintTrackSegments", "Results from TrackSegmentMaker:") ;
999f9a8f 599 printf("nevent: %d\n", AliPHOSGetter::Instance()->EventNumber()) ;
88cb7938 600 printf(" Found %d TrackSegments\n", trackSegments->GetEntriesFast() );
601
2731cd1e 602 if(strstr(option,"all")) { // printing found TS
88cb7938 603 printf("TrackSegment # EMC RP# CPV RP#\n") ;
2731cd1e 604 Int_t index;
7b7c1533 605 for (index = 0 ; index <trackSegments->GetEntriesFast() ; index++) {
606 AliPHOSTrackSegment * ts = (AliPHOSTrackSegment * )trackSegments->At(index) ;
88cb7938 607 printf(" %d %d %d \n", ts->GetIndexInList(), ts->GetEmcIndex(), ts->GetCpvIndex() ) ;
2731cd1e 608 }
d15a28e7 609 }
2731cd1e 610}