]>
Commit | Line | Data |
---|---|---|
4c039060 | 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 | ||
f531a546 | 16 | // $Id$ |
4c039060 | 17 | |
959fbac5 | 18 | /////////////////////////////////////////////////////////////////////////// |
19 | // Class AliVertex | |
20 | // Creation and investigation of an AliVertex. | |
21 | // An AliVertex can be constructed by adding AliTracks and/or AliJets. | |
22 | // | |
23 | // Note : Also (secondary) vertices can be added to a vertex. | |
24 | // | |
f34f4acb | 25 | // To provide maximal flexibility to the user, two modes of vertex storage |
26 | // are provided by means of the memberfunction SetVertexCopy(). | |
27 | // The same holds for the storage of jets via SetJetCopy(). | |
28 | // | |
29 | // a) SetVertexCopy(0) (which is the default). | |
30 | // Only the pointers of the 'added' vertices are stored. | |
31 | // This mode is typically used by making vertex studies based on a fixed list | |
32 | // of vertices which stays under user control or is contained for instance | |
33 | // in an AliEvent. | |
34 | // In this way the AliVertex just represents a 'logical structure' for the | |
35 | // physics analysis which can be embedded in e.g. an AliEvent or AliVertex. | |
35044448 | 36 | // |
37 | // Note : | |
f34f4acb | 38 | // Modifications made to the original vertices also affect the AliVertex objects |
35044448 | 39 | // which are stored. |
40 | // | |
f34f4acb | 41 | // b) SetVertexCopy(1). |
42 | // Of every 'added' vertex a private copy will be made of which the pointer | |
43 | // will be stored. | |
44 | // In this way the AliVertex represents an entity on its own and modifications | |
45 | // made to the original vertices do not affect the AliVertex objects which are | |
46 | // stored. | |
47 | // This mode will allow 'adding' many different AliVertex objects by | |
48 | // creating only one AliVertex instance in the main programme and using the | |
35044448 | 49 | // AliVertex::Reset, AliVertex::AddTrack and parameter setting memberfunctions. |
f34f4acb | 50 | // |
8e8e6c7f | 51 | // See also the documentation provided for the memberfunction SetOwner(). |
52 | // | |
959fbac5 | 53 | // Coding example to make 3 vertices v1, v2 and v3. |
54 | // ------------------------------------------------ | |
55 | // v1 contains the tracks 1,2,3 and 4 | |
f34f4acb | 56 | // v2 contains many different tracks |
959fbac5 | 57 | // v3 contains the jets 1 and 2 |
58 | // | |
f34f4acb | 59 | // AliTrack t1,t2,t3,t4; |
959fbac5 | 60 | // ... |
61 | // ... // code to fill the track data | |
62 | // ... | |
63 | // | |
64 | // AliJet j1,j2; | |
65 | // ... | |
66 | // ... // code to fill the jet data | |
67 | // ... | |
68 | // | |
f34f4acb | 69 | // AliVertex v1; |
70 | // v1.SetVertexCopy(1); | |
959fbac5 | 71 | // |
f531a546 | 72 | // v1.AddTrack(t1); |
73 | // v1.AddTrack(t2); | |
74 | // v1.AddTrack(t3); | |
75 | // v1.AddTrack(t4); | |
959fbac5 | 76 | // |
77 | // Float_t r1[3]={2.4,0.1,-8.5}; | |
78 | // v1.SetPosition(r1,"car"); | |
79 | // | |
f34f4acb | 80 | // AliVertex v2; |
81 | // v2.SetTrackCopy(1); | |
82 | // | |
83 | // AliTrack* tx=new AliTrack(); | |
84 | // for (Int_t i=0; i<10; i++) | |
85 | // { | |
86 | // ... | |
87 | // ... // code to fill the track data | |
88 | // ... | |
89 | // v2.AddTrack(tx); | |
90 | // tx->Reset(); | |
91 | // } | |
959fbac5 | 92 | // |
93 | // Float_t r2[3]={1.6,-3.2,5.7}; | |
94 | // v2.SetPosition(r2,"car"); | |
95 | // | |
96 | // AliVertex v3; | |
97 | // | |
f531a546 | 98 | // v3.AddJet(j1); |
99 | // v3.AddJet(j2); | |
959fbac5 | 100 | // |
101 | // Float_t r3[3]={6.2,4.8,1.3}; | |
102 | // v3.SetPosition(r3,"car"); | |
103 | // | |
84bb7c66 | 104 | // v1.Data("sph"); |
959fbac5 | 105 | // v2.ListAll(); |
106 | // v3.List("cyl"); | |
107 | // | |
108 | // Float_t e1=v1.GetEnergy(); | |
109 | // Ali3Vector p1=v1.Get3Momentum(); | |
110 | // Float_t loc[3]; | |
111 | // v1.GetPosition(loc,"sph"); | |
112 | // AliPosition r=v2.GetPosition(); | |
84bb7c66 | 113 | // r.Data(); |
959fbac5 | 114 | // Int_t nt=v2.GetNtracks(); |
115 | // AliTrack* tv=v2.GetTrack(1); // Access track number 1 of Vertex v2 | |
116 | // | |
117 | // Specify the vertices v2 and v3 as secondary vertices of v1 | |
118 | // | |
f531a546 | 119 | // v1.AddVertex(v2); |
120 | // v1.AddVertex(v3); | |
959fbac5 | 121 | // |
122 | // v1.List(); | |
123 | // | |
124 | // Int_t nv=v1.GetNvtx(); | |
125 | // AliVertex* vx=v1.GetVertex(1); // Access 1st secondary vertex of v1 | |
126 | // Float_t e=vx->GetEnergy(); | |
127 | // | |
128 | // Float_t M=v1.GetInvmass(); | |
129 | // | |
130 | // Reconstruct Vertex v1 from scratch | |
131 | // | |
132 | // v1.Reset(); | |
133 | // v1.SetNvmax(25); // Increase initial no. of sec. vertices | |
f531a546 | 134 | // v1.AddTrack(t3); |
f34f4acb | 135 | // v1.AddTrack(t4); |
f531a546 | 136 | // v1.AddJet(j2); |
959fbac5 | 137 | // Float_t pos[3]={7,9,4}; |
138 | // v1.SetPosition(pos,"car"); | |
139 | // | |
140 | // Note : All quantities are in GeV, GeV/c or GeV/c**2 | |
141 | // | |
142 | //--- Author: Nick van Eijndhoven 04-apr-1998 UU-SAP Utrecht | |
f531a546 | 143 | //- Modified: NvE $Date$ UU-SAP Utrecht |
959fbac5 | 144 | /////////////////////////////////////////////////////////////////////////// |
145 | ||
d88f97cc | 146 | #include "AliVertex.h" |
c72198f1 | 147 | #include "Riostream.h" |
d88f97cc | 148 | |
149 | ClassImp(AliVertex) // Class implementation to enable ROOT I/O | |
150 | ||
c72198f1 | 151 | AliVertex::AliVertex() : AliJet(),AliPosition() |
d88f97cc | 152 | { |
959fbac5 | 153 | // Default constructor. |
154 | // All variables initialised to 0. | |
155 | // Initial maximum number of tracks is set to the default value. | |
156 | // Initial maximum number of sec. vertices is set to the default value. | |
6516b62d | 157 | Init(); |
d88f97cc | 158 | Reset(); |
d88f97cc | 159 | SetNvmax(); |
f34f4acb | 160 | SetNjmax(); |
d88f97cc | 161 | } |
162 | /////////////////////////////////////////////////////////////////////////// | |
6516b62d | 163 | void AliVertex::Init() |
d88f97cc | 164 | { |
6516b62d | 165 | // Initialisation of pointers etc... |
d88f97cc | 166 | fNvmax=0; |
167 | fVertices=0; | |
29beb80d | 168 | fConnects=0; |
f34f4acb | 169 | fVertexCopy=0; |
170 | fNjmax=0; | |
171 | fJets=0; | |
6516b62d | 172 | fJetTracks=0; |
f34f4acb | 173 | fJetCopy=0; |
a37bb40e | 174 | fLines=0; |
6516b62d | 175 | } |
176 | /////////////////////////////////////////////////////////////////////////// | |
c72198f1 | 177 | AliVertex::AliVertex(Int_t n) : AliJet(n),AliPosition() |
6516b62d | 178 | { |
179 | // Create a vertex to hold initially a maximum of n tracks | |
180 | // All variables initialised to 0 | |
c72198f1 | 181 | if (n<=0) |
d88f97cc | 182 | { |
c72198f1 | 183 | cout << " *** This AliJet initialisation was invoked via the AliVertex ctor." << endl; |
d88f97cc | 184 | } |
c72198f1 | 185 | Init(); |
186 | Reset(); | |
d88f97cc | 187 | SetNvmax(); |
f34f4acb | 188 | SetNjmax(); |
d88f97cc | 189 | } |
190 | /////////////////////////////////////////////////////////////////////////// | |
191 | AliVertex::~AliVertex() | |
192 | { | |
193 | // Default destructor | |
f34f4acb | 194 | if (fVertices) |
195 | { | |
f34f4acb | 196 | delete fVertices; |
197 | fVertices=0; | |
198 | } | |
29beb80d | 199 | if (fConnects) |
200 | { | |
29beb80d | 201 | delete fConnects; |
202 | fConnects=0; | |
203 | } | |
f34f4acb | 204 | if (fJets) |
205 | { | |
f34f4acb | 206 | delete fJets; |
207 | fJets=0; | |
208 | } | |
6516b62d | 209 | if (fJetTracks) |
210 | { | |
211 | delete fJetTracks; | |
212 | fJetTracks=0; | |
213 | } | |
a37bb40e | 214 | if (fLines) |
215 | { | |
216 | delete fLines; | |
217 | fLines=0; | |
218 | } | |
6516b62d | 219 | } |
220 | /////////////////////////////////////////////////////////////////////////// | |
8e8e6c7f | 221 | void AliVertex::SetOwner(Bool_t own) |
222 | { | |
223 | // Set ownership of all added objects. | |
224 | // The default parameter is own=kTRUE. | |
225 | // | |
226 | // Invokation of this memberfunction also sets all the copy modes | |
227 | // (e.g. TrackCopy & co.) according to the value of own. | |
228 | // | |
229 | // This function (with own=kTRUE) is particularly useful when reading data | |
230 | // from a tree/file, since Reset() will then actually remove all the | |
231 | // added objects from memory irrespective of the copy mode settings | |
232 | // during the tree/file creation process. In this way it provides a nice way | |
233 | // of preventing possible memory leaks in the reading/analysis process. | |
234 | // | |
235 | // In addition this memberfunction can also be used as a shortcut to set all | |
236 | // copy modes in one go during a tree/file creation process. | |
237 | // However, in this case the user has to take care to only set/change the | |
238 | // ownership (and copy mode) for empty objects (e.g. newly created objects | |
239 | // or after invokation of the Reset() memberfunction) otherwise it will | |
240 | // very likely result in inconsistent destructor behaviour. | |
241 | ||
242 | Int_t mode=1; | |
243 | if (!own) mode=0; | |
244 | if (fVertices) fVertices->SetOwner(own); | |
245 | fVertexCopy=mode; | |
246 | if (fJets) fJets->SetOwner(own); | |
247 | fJetCopy=mode; | |
248 | ||
249 | AliJet::SetOwner(own); | |
250 | } | |
251 | /////////////////////////////////////////////////////////////////////////// | |
c72198f1 | 252 | AliVertex::AliVertex(AliVertex& v) : AliJet(v.fNtinit),AliPosition(v) |
6516b62d | 253 | { |
254 | // Copy constructor | |
255 | Init(); | |
c72198f1 | 256 | fNvtx=0; |
257 | fNjets=0; | |
258 | SetNvmax(v.fNvmax); | |
259 | SetNjmax(v.fNjmax); | |
6516b62d | 260 | SetTrackCopy(v.GetTrackCopy()); |
261 | SetVertexCopy(v.GetVertexCopy()); | |
262 | SetJetCopy(v.GetJetCopy()); | |
263 | SetId(v.GetId()); | |
6516b62d | 264 | |
265 | // Copy all tracks except the ones coming from jets | |
266 | AliTrack* tx=0; | |
267 | Int_t jetflag=0,connect=0; | |
268 | AliTrack* tx2=0; | |
269 | for (Int_t it=1; it<=v.GetNtracks(); it++) | |
270 | { | |
271 | tx=v.GetTrack(it); | |
272 | if (tx) | |
273 | { | |
274 | jetflag=v.IsJetTrack(tx); | |
275 | connect=v.IsConnectTrack(tx); | |
276 | ||
277 | if (!jetflag && !connect) AddTrack(tx); | |
278 | ||
279 | if (connect) | |
280 | { | |
281 | if (!fConnects) | |
282 | { | |
283 | fConnects=new TObjArray(fNvmax); | |
284 | if (!fTrackCopy) fConnects->SetOwner(); | |
285 | } | |
286 | tx2=new AliTrack(*tx); | |
287 | fConnects->Add(tx2); | |
288 | AddTrack(tx2,0); | |
289 | } | |
290 | } | |
291 | } | |
292 | ||
293 | // Copy all the (secondary) vertices without re-creating connecting tracks | |
294 | // The connecting tracks have already been copied above | |
295 | AliVertex* vx=0; | |
296 | for (Int_t iv=1; iv<=v.GetNvertices(); iv++) | |
297 | { | |
298 | vx=v.GetVertex(iv); | |
299 | if (vx) AddVertex(vx,0); | |
300 | } | |
301 | ||
302 | // Copy all the jets including the jet tracks for these jets for which | |
303 | // this was also the case in the original vertex | |
304 | AliJet* jx=0; | |
305 | for (Int_t ij=1; ij<=v.GetNjets(); ij++) | |
306 | { | |
307 | jx=v.GetJet(ij); | |
308 | if (jx) | |
309 | { | |
310 | jetflag=0; | |
311 | if (jx->GetNtracks()) | |
312 | { | |
313 | tx=jx->GetTrack(1); | |
314 | if (tx) | |
315 | { | |
316 | jetflag=v.IsJetTrack(tx); | |
317 | } | |
318 | } | |
319 | AddJet(jx,jetflag); | |
320 | } | |
321 | } | |
d88f97cc | 322 | } |
323 | /////////////////////////////////////////////////////////////////////////// | |
324 | void AliVertex::SetNvmax(Int_t n) | |
325 | { | |
326 | // Set the initial maximum number of (secondary) vertices | |
327 | if (n > 0) | |
328 | { | |
329 | fNvmax=n; | |
330 | } | |
331 | else | |
332 | { | |
333 | fNvmax=1; | |
334 | } | |
f34f4acb | 335 | if (fVertices) |
336 | { | |
f34f4acb | 337 | delete fVertices; |
338 | fVertices=0; | |
339 | } | |
340 | } | |
341 | /////////////////////////////////////////////////////////////////////////// | |
342 | void AliVertex::SetNjmax(Int_t n) | |
343 | { | |
344 | // Set the initial maximum number of jets | |
345 | if (n > 0) | |
346 | { | |
347 | fNjmax=n; | |
348 | } | |
349 | else | |
350 | { | |
351 | fNjmax=1; | |
352 | } | |
353 | if (fJets) | |
354 | { | |
f34f4acb | 355 | delete fJets; |
356 | fJets=0; | |
357 | } | |
d88f97cc | 358 | } |
359 | /////////////////////////////////////////////////////////////////////////// | |
360 | void AliVertex::Reset() | |
361 | { | |
7849a8ab | 362 | // Reset all variables to 0 and reset all stored vertex and jet lists. |
d88f97cc | 363 | // The max. number of tracks is set to the initial value again |
364 | // The max. number of vertices is set to the default value again | |
7849a8ab | 365 | // The max. number of jets is set to the default value again |
d88f97cc | 366 | |
367 | AliJet::Reset(); | |
368 | ||
f531a546 | 369 | Double_t a[3]={0,0,0}; |
370 | SetPosition(a,"sph"); | |
371 | SetPositionErrors(a,"car"); | |
372 | ||
d88f97cc | 373 | fNvtx=0; |
374 | if (fNvmax>0) SetNvmax(fNvmax); | |
29beb80d | 375 | if (fConnects) |
376 | { | |
29beb80d | 377 | delete fConnects; |
378 | fConnects=0; | |
379 | } | |
f34f4acb | 380 | |
381 | fNjets=0; | |
382 | if (fNjmax>0) SetNjmax(fNjmax); | |
6516b62d | 383 | if (fJetTracks) |
384 | { | |
385 | delete fJetTracks; | |
386 | fJetTracks=0; | |
387 | } | |
a37bb40e | 388 | |
389 | if (fLines) | |
390 | { | |
391 | delete fLines; | |
392 | fLines=0; | |
393 | } | |
d88f97cc | 394 | } |
395 | /////////////////////////////////////////////////////////////////////////// | |
7849a8ab | 396 | void AliVertex::ResetVertices() |
397 | { | |
398 | // Reset the stored vertex list and delete all connecting tracks which | |
399 | // were generated automatically via connect=1 in AddVertex(). | |
400 | // The max. number of vertices is set to the default value again. | |
35044448 | 401 | // All physics quantities are updated according to the removal of the |
402 | // connecting tracks. | |
7849a8ab | 403 | AliTrack* t; |
404 | if (fConnects) | |
405 | { | |
406 | for (Int_t i=0; i<=fConnects->GetLast(); i++) | |
407 | { | |
408 | t=(AliTrack*)fConnects->At(i); | |
409 | AliTrack* test=(AliTrack*)fTracks->Remove(t); | |
35044448 | 410 | if (test) |
411 | { | |
412 | fNtrk--; | |
413 | (Ali4Vector&)(*this)-=(Ali4Vector&)(*t); | |
414 | fQ-=t->GetCharge(); | |
6516b62d | 415 | if (fTrackCopy) delete t; |
35044448 | 416 | } |
7849a8ab | 417 | } |
418 | fTracks->Compress(); | |
419 | } | |
420 | ||
7849a8ab | 421 | fNvtx=0; |
422 | if (fNvmax>0) SetNvmax(fNvmax); | |
423 | if (fConnects) | |
424 | { | |
7849a8ab | 425 | delete fConnects; |
426 | fConnects=0; | |
427 | } | |
428 | } | |
429 | /////////////////////////////////////////////////////////////////////////// | |
f34f4acb | 430 | void AliVertex::AddJet(AliJet& j,Int_t tracks) |
d88f97cc | 431 | { |
f34f4acb | 432 | // Add a jet (and its tracks) to the vertex |
433 | // In case the maximum number of jets has been reached, | |
434 | // the array space will be extended automatically | |
435 | // | |
436 | // Note : By default the tracks of the jet are added to the current (primary) | |
437 | // vertex. | |
438 | // The automatic addition of the tracks of the jet can be suppressed | |
439 | // by specifying tracks=0. In this case only the AliJet object will | |
440 | // be stored according to the mode specified by SetJetCopy(). | |
441 | // The latter will enable jet studies based on a fixed list of tracks | |
442 | // as contained e.g. in an AliVertex or AliEvent. | |
6516b62d | 443 | if (!fJets) |
444 | { | |
445 | fJets=new TObjArray(fNjmax); | |
446 | if (fJetCopy) fJets->SetOwner(); | |
447 | } | |
f34f4acb | 448 | if (fNjets == fNjmax) // Check if maximum jet number is reached |
d88f97cc | 449 | { |
f34f4acb | 450 | fNjmax++; |
451 | fJets->Expand(fNjmax); | |
452 | } | |
453 | ||
454 | // Add the jet to the list | |
6516b62d | 455 | AliJet* jx=&j; |
456 | if (fJetCopy) jx=new AliJet(j); | |
457 | ||
458 | if (jx) | |
f34f4acb | 459 | { |
6516b62d | 460 | fNjets++; |
461 | fJets->Add(jx); | |
f34f4acb | 462 | } |
463 | ||
464 | // Add the tracks of the jet to this vertex | |
465 | if (tracks) | |
466 | { | |
6516b62d | 467 | if (!fJetTracks) |
468 | { | |
469 | fJetTracks=new TObjArray(); | |
470 | } | |
471 | Int_t copy=1-(jx->GetTrackCopy()); | |
f34f4acb | 472 | AliTrack* tj; |
6516b62d | 473 | for (Int_t i=1; i<=jx->GetNtracks(); i++) |
f34f4acb | 474 | { |
6516b62d | 475 | tj=jx->GetTrack(i); |
476 | if (tj) | |
477 | { | |
478 | AddTrack(tj,copy); | |
479 | fJetTracks->Add(tj); | |
480 | } | |
f34f4acb | 481 | } |
d88f97cc | 482 | } |
483 | } | |
484 | /////////////////////////////////////////////////////////////////////////// | |
f531a546 | 485 | void AliVertex::AddVertex(AliVertex& v,Int_t connect) |
d88f97cc | 486 | { |
487 | // Add a (secondary) vertex to the current vertex. | |
488 | // In case the maximum number of (secondary) vertices has been reached, | |
489 | // the array space will be extended automatically | |
490 | // | |
29beb80d | 491 | // Note : By default the 4-momentum and charge of the current (primary) vertex |
492 | // are updated by automatically creating the track connecting | |
493 | // both vertices. The track parameters are taken from the | |
494 | // 4-momentum and charge of the secondary vertex. | |
495 | // The automatic creation of the connecting track and updating | |
496 | // of the (primary) vertex 4-momentum and charge can be suppressed | |
497 | // by specifying connect=0. In this case, however, the user | |
498 | // has to introduce the connecting track lateron by hand | |
499 | // explicitly in order to match the kinematics and charge. | |
d88f97cc | 500 | // |
6516b62d | 501 | if (!fVertices) |
502 | { | |
503 | fVertices=new TObjArray(fNvmax); | |
504 | if (fVertexCopy) fVertices->SetOwner(); | |
505 | } | |
d88f97cc | 506 | if (fNvtx == fNvmax) // Check if maximum vertex number is reached |
507 | { | |
508 | fNvmax++; | |
509 | fVertices->Expand(fNvmax); | |
510 | } | |
29beb80d | 511 | |
512 | // Add the linked (secondary) vertex to the list | |
6516b62d | 513 | AliVertex* vx=&v; |
514 | if (fVertexCopy) vx=new AliVertex(v); | |
515 | ||
516 | if (vx) | |
f34f4acb | 517 | { |
6516b62d | 518 | fNvtx++; |
519 | fVertices->Add(vx); | |
520 | } | |
29beb80d | 521 | |
522 | // Create connecting track and update 4-momentum and charge for current vertex | |
523 | if (connect) | |
524 | { | |
43bfa5be | 525 | AliTrack* t=new AliTrack(); |
6516b62d | 526 | t->SetBeginPoint(GetPosition()); |
527 | t->SetEndPoint(v.GetPosition()); | |
528 | t->SetCharge(v.GetCharge()); | |
529 | t->Set4Momentum((Ali4Vector&)v); | |
29beb80d | 530 | |
6516b62d | 531 | AddTrack(t,0); |
29beb80d | 532 | |
6516b62d | 533 | if (!fConnects) |
534 | { | |
535 | fConnects=new TObjArray(fNvmax); | |
536 | if (!fTrackCopy) fConnects->SetOwner(); | |
537 | } | |
29beb80d | 538 | fConnects->Add(t); |
539 | } | |
d88f97cc | 540 | } |
541 | /////////////////////////////////////////////////////////////////////////// | |
84bb7c66 | 542 | void AliVertex::Data(TString f) |
d88f97cc | 543 | { |
544 | // Provide vertex information within the coordinate frame f | |
387a745b | 545 | cout << " *AliVertex::Data* Name : " << GetName() << " Title : " << GetTitle() << endl; |
546 | cout << " Id : " << fUserId << " Invmass : " << GetInvmass() | |
d88f97cc | 547 | << " Charge : " << GetCharge() << " Momentum : " << GetMomentum() |
f34f4acb | 548 | << " Ntracks : " << GetNtracks() << " Nvertices : " << fNvtx |
549 | << " Njets : " << fNjets << endl; | |
d88f97cc | 550 | cout << " "; |
84bb7c66 | 551 | Ali4Vector::Data(f); |
d88f97cc | 552 | cout << " Position"; |
84bb7c66 | 553 | AliPosition::Data(f); |
d88f97cc | 554 | } |
555 | /////////////////////////////////////////////////////////////////////////// | |
556 | void AliVertex::List(TString f) | |
557 | { | |
558 | // Provide primary track and sec. vertex information within the coordinate frame f | |
559 | ||
84bb7c66 | 560 | Data(f); // Information of the current vertex |
d88f97cc | 561 | |
562 | // The tracks of this vertex | |
563 | AliTrack* t; | |
564 | for (Int_t it=1; it<=GetNtracks(); it++) | |
565 | { | |
566 | t=GetTrack(it); | |
567 | if (t) | |
568 | { | |
569 | cout << " ---Track no. " << it << endl; | |
570 | cout << " "; | |
84bb7c66 | 571 | t->Data(f); |
d88f97cc | 572 | } |
573 | else | |
574 | { | |
575 | cout << " *AliVertex::List* Error : No track present." << endl; | |
576 | } | |
577 | } | |
578 | ||
579 | // The secondary vertices of this vertex | |
580 | AliVertex* v; | |
581 | for (Int_t iv=1; iv<=GetNvertices(); iv++) | |
582 | { | |
583 | v=GetVertex(iv); | |
584 | if (v) | |
585 | { | |
586 | cout << " ---Level 1 sec. vertex no. " << iv << endl; | |
587 | cout << " "; | |
84bb7c66 | 588 | v->Data(f); |
d88f97cc | 589 | } |
590 | else | |
591 | { | |
592 | cout << " *AliVertex::List* Error : No sec. vertex present." << endl; | |
593 | } | |
594 | } | |
595 | } | |
596 | /////////////////////////////////////////////////////////////////////////// | |
597 | void AliVertex::ListAll(TString f) | |
598 | { | |
599 | // Provide complete (sec) vertex and (decay) track info within the coordinate frame f | |
600 | ||
84bb7c66 | 601 | Data(f); // Information of the current vertex |
d88f97cc | 602 | |
603 | // The tracks of this vertex | |
604 | AliTrack* t; | |
605 | for (Int_t it=1; it<=GetNtracks(); it++) | |
606 | { | |
607 | t=GetTrack(it); | |
608 | if (t) | |
609 | { | |
610 | cout << " ---Track no. " << it << endl; | |
611 | cout << " "; | |
612 | t->ListAll(f); | |
613 | } | |
614 | else | |
615 | { | |
616 | cout << " *AliVertex::ListAll* Error : No track present." << endl; | |
617 | } | |
618 | } | |
619 | ||
620 | AliVertex* v=this; | |
1c01b4f8 | 621 | Dumps(v,1,f); // Information of all sec. vertices |
d88f97cc | 622 | } |
623 | ////////////////////////////////////////////////////////////////////////// | |
1c01b4f8 | 624 | void AliVertex::Dumps(AliVertex* v,Int_t n,TString f) |
d88f97cc | 625 | { |
626 | // Recursively provide the info of all secondary vertices of this vertex | |
627 | AliVertex* vs; | |
628 | for (Int_t iv=1; iv<=v->GetNvertices(); iv++) | |
629 | { | |
630 | vs=v->GetVertex(iv); | |
631 | if (vs) | |
632 | { | |
633 | cout << " ---Level " << n << " sec. vertex no. " << iv << endl; | |
634 | cout << " "; | |
84bb7c66 | 635 | vs->Data(f); |
d88f97cc | 636 | |
637 | // The tracks of this vertex | |
638 | AliTrack* t; | |
639 | for (Int_t it=1; it<=vs->GetNtracks(); it++) | |
640 | { | |
641 | t=vs->GetTrack(it); | |
642 | if (t) | |
643 | { | |
644 | cout << " ---Track no. " << it << endl; | |
645 | cout << " "; | |
646 | t->ListAll(f); | |
647 | } | |
648 | else | |
649 | { | |
1c01b4f8 | 650 | cout << " *AliVertex::Dumps* Error : No track present." << endl; |
d88f97cc | 651 | } |
652 | } | |
653 | ||
654 | // Go for next sec. vertex level of this sec. vertex recursively | |
1c01b4f8 | 655 | Dumps(vs,n+1,f); |
d88f97cc | 656 | } |
657 | else | |
658 | { | |
1c01b4f8 | 659 | cout << " *AliVertex::Dumps* Error : No sec. vertex present." << endl; |
d88f97cc | 660 | } |
661 | } | |
662 | } | |
663 | ////////////////////////////////////////////////////////////////////////// | |
664 | Int_t AliVertex::GetNvertices() | |
665 | { | |
666 | // Return the current number of (secondary) vertices | |
667 | return fNvtx; | |
668 | } | |
669 | /////////////////////////////////////////////////////////////////////////// | |
670 | AliVertex* AliVertex::GetVertex(Int_t i) | |
671 | { | |
672 | // Return the i-th (secondary) vertex of the current vertex | |
f34f4acb | 673 | if (!fVertices) |
674 | { | |
675 | cout << " *AliVertex*::GetVertex* No (secondary) vertices present." << endl; | |
676 | return 0; | |
677 | } | |
678 | else | |
679 | { | |
680 | if (i<=0 || i>fNvtx) | |
681 | { | |
682 | cout << " *AliVertex*::GetVertex* Invalid argument i : " << i | |
683 | << " Nvtx = " << fNvtx << endl; | |
684 | return 0; | |
685 | } | |
686 | else | |
687 | { | |
688 | return (AliVertex*)fVertices->At(i-1); | |
689 | } | |
690 | } | |
691 | } | |
692 | /////////////////////////////////////////////////////////////////////////// | |
43bfa5be | 693 | AliVertex* AliVertex::GetIdVertex(Int_t id) |
694 | { | |
695 | // Return the (sec.) vertex with user identifier "id" | |
696 | AliVertex* vx=0; | |
697 | AliVertex* v=0; | |
698 | if (!fVertices) | |
699 | { | |
700 | cout << " *AliVertex*::GetIdVertex* No (secondary) vertices present." << endl; | |
701 | return 0; | |
702 | } | |
703 | else | |
704 | { | |
705 | for (Int_t i=0; i<fNvtx; i++) | |
706 | { | |
707 | vx=(AliVertex*)fVertices->At(i); | |
708 | if (id == vx->GetId()) v=vx; | |
709 | } | |
710 | return v; | |
711 | } | |
712 | } | |
713 | /////////////////////////////////////////////////////////////////////////// | |
f34f4acb | 714 | void AliVertex::SetVertexCopy(Int_t j) |
715 | { | |
716 | // (De)activate the creation of private copies of the added vertices. | |
717 | // j=0 ==> No private copies are made; pointers of original vertices are stored. | |
718 | // j=1 ==> Private copies of the vertices are made and these pointers are stored. | |
719 | // | |
720 | // Note : Once the storage contains pointer(s) to AliVertex objects one cannot | |
721 | // change the VertexCopy mode anymore. | |
722 | // To change the VertexCopy mode for an existing AliVertex containing | |
723 | // vertices one first has to invoke Reset(). | |
724 | if (!fVertices) | |
725 | { | |
726 | if (j==0 || j==1) | |
727 | { | |
728 | fVertexCopy=j; | |
729 | } | |
730 | else | |
731 | { | |
732 | cout << "*AliVertex::SetVertexCopy* Invalid argument : " << j << endl; | |
733 | } | |
734 | } | |
735 | else | |
736 | { | |
737 | cout << "*AliVertex::SetVertexCopy* Storage already contained vertices." | |
738 | << " ==> VertexCopy mode not changed." << endl; | |
739 | } | |
740 | } | |
741 | /////////////////////////////////////////////////////////////////////////// | |
742 | Int_t AliVertex::GetVertexCopy() | |
743 | { | |
744 | // Provide value of the VertexCopy mode. | |
745 | // 0 ==> No private copies are made; pointers of original vertices are stored. | |
746 | // 1 ==> Private copies of the vertices are made and these pointers are stored. | |
747 | return fVertexCopy; | |
748 | } | |
749 | /////////////////////////////////////////////////////////////////////////// | |
750 | Int_t AliVertex::GetNjets() | |
751 | { | |
752 | // Return the current number of jets | |
753 | return fNjets; | |
754 | } | |
755 | /////////////////////////////////////////////////////////////////////////// | |
756 | AliJet* AliVertex::GetJet(Int_t i) | |
757 | { | |
758 | // Return the i-th jet of the current vertex | |
759 | if (!fJets) | |
760 | { | |
761 | cout << " *AliVertex*::GetJet* No jets present." << endl; | |
762 | return 0; | |
763 | } | |
764 | else | |
765 | { | |
766 | if (i<=0 || i>fNjets) | |
767 | { | |
768 | cout << " *AliVertex*::GetJet* Invalid argument i : " << i | |
769 | << " Njets = " << fNjets << endl; | |
770 | return 0; | |
771 | } | |
772 | else | |
773 | { | |
774 | return (AliJet*)fJets->At(i-1); | |
775 | } | |
776 | } | |
777 | } | |
778 | /////////////////////////////////////////////////////////////////////////// | |
43bfa5be | 779 | AliJet* AliVertex::GetIdJet(Int_t id) |
780 | { | |
781 | // Return the jet with user identifier "id" | |
782 | AliJet* jx=0; | |
783 | AliJet* j=0; | |
784 | if (!fJets) | |
785 | { | |
786 | cout << " *AliVertex*::GetIdJet* No jets present." << endl; | |
787 | return 0; | |
788 | } | |
789 | else | |
790 | { | |
791 | for (Int_t i=0; i<fNjets; i++) | |
792 | { | |
793 | jx=(AliJet*)fJets->At(i); | |
794 | if (id == jx->GetId()) j=jx; | |
795 | } | |
796 | return j; | |
797 | } | |
798 | } | |
799 | /////////////////////////////////////////////////////////////////////////// | |
f34f4acb | 800 | void AliVertex::SetJetCopy(Int_t j) |
801 | { | |
802 | // (De)activate the creation of private copies of the added jets. | |
803 | // j=0 ==> No private copies are made; pointers of original jets are stored. | |
804 | // j=1 ==> Private copies of the jets are made and these pointers are stored. | |
805 | // | |
806 | // Note : Once the storage contains pointer(s) to AliJet objects one cannot | |
807 | // change the JetCopy mode anymore. | |
808 | // To change the JetCopy mode for an existing AliVertex containing | |
809 | // jets one first has to invoke Reset(). | |
810 | if (!fJets) | |
811 | { | |
812 | if (j==0 || j==1) | |
813 | { | |
814 | fJetCopy=j; | |
815 | } | |
816 | else | |
817 | { | |
818 | cout << "*AliVertex::SetJetCopy* Invalid argument : " << j << endl; | |
819 | } | |
820 | } | |
821 | else | |
822 | { | |
823 | cout << "*AliVertex::SetJetCopy* Storage already contained jets." | |
824 | << " ==> JetCopy mode not changed." << endl; | |
825 | } | |
826 | } | |
827 | /////////////////////////////////////////////////////////////////////////// | |
828 | Int_t AliVertex::GetJetCopy() | |
829 | { | |
830 | // Provide value of the JetCopy mode. | |
831 | // 0 ==> No private copies are made; pointers of original jets are stored. | |
832 | // 1 ==> Private copies of the jets are made and these pointers are stored. | |
833 | return fJetCopy; | |
d88f97cc | 834 | } |
835 | /////////////////////////////////////////////////////////////////////////// | |
6516b62d | 836 | Int_t AliVertex::IsConnectTrack(AliTrack* t) |
837 | { | |
838 | // Indicate whether a track from the tracklist was created via the | |
839 | // connection of a (secondary) vertex or not. | |
840 | // In case the track was the result of (secondary) vertex addition the | |
841 | // return value is 1, otherwise the value 0 will be returned. | |
842 | Int_t connect=0; | |
843 | if (fConnects) | |
844 | { | |
845 | if (fConnects->FindObject(t)) connect=1; | |
846 | } | |
847 | return connect; | |
848 | } | |
849 | /////////////////////////////////////////////////////////////////////////// | |
850 | Int_t AliVertex::IsJetTrack(AliTrack* t) | |
851 | { | |
852 | // Indicate whether a track from the tracklist was created via the | |
853 | // addition of a jet or not. | |
854 | // In case the track was the result of jet addition the return value is 1, | |
855 | // otherwise the value 0 will be returned. | |
856 | Int_t jetflag=0; | |
857 | if (fJetTracks) | |
858 | { | |
859 | if (fJetTracks->FindObject(t)) jetflag=1; | |
860 | } | |
861 | return jetflag; | |
862 | } | |
863 | /////////////////////////////////////////////////////////////////////////// | |
a37bb40e | 864 | void AliVertex::Draw(Int_t secs,Int_t cons,Int_t jets) |
865 | { | |
866 | // 3-Dimensional visualisation of an AliVertex with its attributes. | |
867 | // The displayed tracklength is proportional to the momentum of the track. | |
868 | // | |
869 | // Color conventions : | |
870 | // ------------------- | |
871 | // positive track : red | |
872 | // neutral track : green | |
873 | // negative track : blue | |
874 | // jet-track : magenta (if explicit marking selected) | |
875 | // | |
1c01b4f8 | 876 | // secs = 1 --> Draw secondary vertices. (Default) |
a37bb40e | 877 | // 0 --> Don't draw secondary vertices. |
878 | // | |
1c01b4f8 | 879 | // cons = 1 --> Draw (auto generated) connecting tracks. (Default) |
a37bb40e | 880 | // 0 --> Don't draw (auto generated) connecting tracks. |
881 | // | |
882 | // jets = 1 --> Mark tracks belonging to jets. | |
1c01b4f8 | 883 | // 0 --> Don't mark jet-tracks. (Default) |
a37bb40e | 884 | // |
885 | // Notes : | |
886 | // ------- | |
887 | // Auto generated connecting tracks will be drawn as thin lines. | |
888 | // Tracks belonging to jets will be marked as somewhat thinner magenta lines. | |
889 | // This memberfunction is used recursively. | |
890 | // | |
891 | Double_t vec[3]={0,0,0}; | |
892 | AliTrack* tx=0; | |
893 | AliVertex* vx=0; | |
c72198f1 | 894 | AliPosition* r=0; |
a37bb40e | 895 | Ali3Vector p; |
1c01b4f8 | 896 | Float_t charge; |
a37bb40e | 897 | |
c72198f1 | 898 | AliPosition dummy; |
899 | ||
a37bb40e | 900 | if (fLines) delete fLines; |
901 | fLines=new TObjArray(); | |
902 | fLines->SetOwner(); | |
903 | ||
904 | Int_t ntk=GetNtracks(); | |
905 | for (Int_t jtk=1; jtk<=ntk; jtk++) | |
906 | { | |
907 | tx=GetTrack(jtk); | |
908 | ||
909 | if (!tx) continue; | |
910 | ||
1c01b4f8 | 911 | charge=tx->GetCharge(); |
a37bb40e | 912 | |
913 | TPolyLine3D* line=new TPolyLine3D(); | |
914 | fLines->Add(line); | |
915 | ||
916 | if (IsConnectTrack(tx)) | |
917 | { | |
918 | if (cons==1) | |
919 | { | |
920 | r=tx->GetBeginPoint(); | |
c72198f1 | 921 | if (!r) r=&dummy; |
922 | r->GetPosition(vec,"car"); | |
a37bb40e | 923 | line->SetNextPoint(vec[0],vec[1],vec[2]); |
924 | r=tx->GetEndPoint(); | |
c72198f1 | 925 | if (!r) r=&dummy; |
926 | r->GetPosition(vec,"car"); | |
a37bb40e | 927 | line->SetNextPoint(vec[0],vec[1],vec[2]); |
928 | line->SetLineWidth(1); | |
929 | } | |
930 | } | |
931 | else | |
932 | { | |
933 | r=tx->GetClosestPoint(); | |
c72198f1 | 934 | if (!r) r=&dummy; |
935 | r->GetPosition(vec,"car"); | |
a37bb40e | 936 | line->SetNextPoint(vec[0],vec[1],vec[2]); |
937 | p=tx->Get3Momentum(); | |
c72198f1 | 938 | p=p+(*r); |
a37bb40e | 939 | p.GetVector(vec,"car"); |
940 | line->SetNextPoint(vec[0],vec[1],vec[2]); | |
941 | line->SetLineWidth(3); | |
942 | } | |
943 | ||
1c01b4f8 | 944 | line->SetLineColor(kGreen); // Neutral track |
945 | if (charge>0) line->SetLineColor(kRed); // Positive track | |
946 | if (charge<0) line->SetLineColor(kBlue); // Negative track | |
a37bb40e | 947 | |
948 | // Mark tracks belonging to jets | |
949 | if (IsJetTrack(tx)) | |
950 | { | |
951 | if (jets==1) | |
952 | { | |
953 | line->SetLineWidth(2); | |
954 | line->SetLineColor(kMagenta); | |
955 | } | |
956 | } | |
957 | ||
958 | line->Draw(); | |
959 | } | |
960 | ||
961 | // Go for secondary vertices if selected | |
962 | if (secs==1) | |
963 | { | |
964 | Int_t nvtx=GetNvertices(); | |
965 | for (Int_t jvtx=1; jvtx<=nvtx; jvtx++) | |
966 | { | |
967 | vx=GetVertex(jvtx); | |
968 | if (vx) vx->Draw(secs,cons,jets); | |
969 | } | |
970 | } | |
971 | } | |
972 | /////////////////////////////////////////////////////////////////////////// |