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