]>
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 AliJet | |
20 | // Creation and investigation of a jet of particle tracks. | |
21 | // An AliJet can be constructed by adding AliTracks. | |
22 | // | |
c1f25d1d | 23 | // To provide maximal flexibility to the user, two modes of track storage |
24 | // are provided by means of the memberfunction SetTrackCopy(). | |
25 | // | |
26 | // a) SetTrackCopy(0) (which is the default). | |
27 | // Only the pointers of the 'added' tracks are stored. | |
28 | // This mode is typically used by making jet studies based on a fixed list | |
29 | // of tracks which stays under user control or is contained for instance | |
30 | // in an AliEvent. | |
31 | // In this way the AliJet just represents a 'logical structure' for the | |
32 | // physics analysis which can be embedded in e.g. an AliEvent or AliVertex. | |
35044448 | 33 | // |
34 | // Note : | |
c1f25d1d | 35 | // Modifications made to the original tracks also affect the AliTrack objects |
36 | // which are stored in the AliJet. | |
35044448 | 37 | // |
c1f25d1d | 38 | // b) SetTrackCopy(1). |
39 | // Of every 'added' track a private copy will be made of which the pointer | |
40 | // will be stored. | |
41 | // In this way the AliJet represents an entity on its own and modifications | |
42 | // made to the original tracks do not affect the AliTrack objects which are | |
43 | // stored in the AliJet. | |
44 | // This mode will allow 'adding' many different AliTracks into an AliJet by | |
45 | // creating only one AliTrack instance in the main programme and using the | |
46 | // AliTrack::Reset() and AliTrack parameter setting memberfunctions. | |
47 | // | |
8e8e6c7f | 48 | // See also the documentation provided for the memberfunction SetOwner(). |
49 | // | |
959fbac5 | 50 | // Coding example to make 2 jets j1 and j2. |
51 | // ---------------------------------------- | |
c1f25d1d | 52 | // j1 contains the AliTracks t1 and t2 |
53 | // j2 contains 10 different AliTracks via tx | |
959fbac5 | 54 | // |
c1f25d1d | 55 | // AliTrack t1,t2; |
959fbac5 | 56 | // ... |
57 | // ... // code to fill the AliTrack data | |
58 | // ... | |
c1f25d1d | 59 | // AliJet j1(); |
f531a546 | 60 | // j1.AddTrack(t1); |
61 | // j1.AddTrack(t2); | |
c1f25d1d | 62 | // |
63 | // AliJet j2(); | |
64 | // j2.SetTrackCopy(1); | |
65 | // AliTrack* tx=new AliTrack(); | |
66 | // for (Int_t i=0; i<10; i++) | |
67 | // { | |
68 | // ... | |
69 | // ... // code to set momentum etc... of the track tx | |
70 | // ... | |
71 | // j2.AddTrack(tx); | |
72 | // tx->Reset(); | |
73 | // } | |
959fbac5 | 74 | // |
84bb7c66 | 75 | // j1.Data(); |
76 | // j2.Data("sph"); | |
959fbac5 | 77 | // |
78 | // Float_t e1=j1.GetEnergy(); | |
79 | // Float_t pnorm=j1->GetMomentum(); | |
80 | // Ali3Vector p=j1->Get3Momentum(); | |
81 | // Float_t m=j1.GetInvmass(); | |
82 | // Int_t ntk=j1.GetNtracks(); | |
83 | // AliTrack* tj=j1.GetTrack(1); | |
84 | // | |
c1f25d1d | 85 | // delete tx; |
86 | // | |
959fbac5 | 87 | // Note : All quantities are in GeV, GeV/c or GeV/c**2 |
88 | // | |
89 | //--- Author: Nick van Eijndhoven 10-jul-1997 UU-SAP Utrecht | |
f531a546 | 90 | //- Modified: NvE $Date$ UU-SAP Utrecht |
959fbac5 | 91 | /////////////////////////////////////////////////////////////////////////// |
92 | ||
d88f97cc | 93 | #include "AliJet.h" |
c72198f1 | 94 | #include "Riostream.h" |
d88f97cc | 95 | |
96 | ClassImp(AliJet) // Class implementation to enable ROOT I/O | |
97 | ||
387a745b | 98 | AliJet::AliJet() : TNamed(),Ali4Vector() |
d88f97cc | 99 | { |
100 | // Default constructor | |
101 | // All variables initialised to 0 | |
102 | // Initial maximum number of tracks is set to the default value | |
6516b62d | 103 | Init(); |
104 | Reset(); | |
105 | SetNtinit(); | |
106 | } | |
107 | /////////////////////////////////////////////////////////////////////////// | |
108 | void AliJet::Init() | |
109 | { | |
110 | // Initialisation of pointers etc... | |
d88f97cc | 111 | fTracks=0; |
112 | fNtinit=0; | |
c1f25d1d | 113 | fTrackCopy=0; |
413d0114 | 114 | fRef=0; |
ea0b5b7f | 115 | fSelected=0; |
d88f97cc | 116 | } |
117 | /////////////////////////////////////////////////////////////////////////// | |
387a745b | 118 | AliJet::AliJet(Int_t n) : TNamed(),Ali4Vector() |
d88f97cc | 119 | { |
120 | // Create a jet to hold initially a maximum of n tracks | |
121 | // All variables initialised to 0 | |
6516b62d | 122 | Init(); |
d88f97cc | 123 | Reset(); |
124 | if (n > 0) | |
125 | { | |
126 | SetNtinit(n); | |
127 | } | |
128 | else | |
129 | { | |
130 | cout << endl; | |
131 | cout << " *AliJet* Initial max. number of tracks entered : " << n << endl; | |
132 | cout << " This is invalid. Default initial maximum will be used." << endl; | |
133 | cout << endl; | |
134 | SetNtinit(); | |
135 | } | |
136 | } | |
137 | /////////////////////////////////////////////////////////////////////////// | |
138 | AliJet::~AliJet() | |
139 | { | |
140 | // Default destructor | |
c1f25d1d | 141 | if (fTracks) |
142 | { | |
c1f25d1d | 143 | delete fTracks; |
144 | fTracks=0; | |
145 | } | |
413d0114 | 146 | if (fRef) |
147 | { | |
148 | delete fRef; | |
149 | fRef=0; | |
150 | } | |
ea0b5b7f | 151 | if (fSelected) |
152 | { | |
153 | delete fSelected; | |
154 | fSelected=0; | |
155 | } | |
d88f97cc | 156 | } |
157 | /////////////////////////////////////////////////////////////////////////// | |
8e8e6c7f | 158 | void AliJet::SetOwner(Bool_t own) |
159 | { | |
160 | // Set ownership of all added objects. | |
161 | // The default parameter is own=kTRUE. | |
162 | // | |
163 | // Invokation of this memberfunction also sets all the copy modes | |
164 | // (e.g. TrackCopy & co.) according to the value of own. | |
165 | // | |
166 | // This function (with own=kTRUE) is particularly useful when reading data | |
167 | // from a tree/file, since Reset() will then actually remove all the | |
168 | // added objects from memory irrespective of the copy mode settings | |
169 | // during the tree/file creation process. In this way it provides a nice way | |
170 | // of preventing possible memory leaks in the reading/analysis process. | |
171 | // | |
172 | // In addition this memberfunction can also be used as a shortcut to set all | |
173 | // copy modes in one go during a tree/file creation process. | |
174 | // However, in this case the user has to take care to only set/change the | |
175 | // ownership (and copy mode) for empty objects (e.g. newly created objects | |
176 | // or after invokation of the Reset() memberfunction) otherwise it will | |
177 | // very likely result in inconsistent destructor behaviour. | |
178 | ||
179 | Int_t mode=1; | |
180 | if (!own) mode=0; | |
181 | if (fTracks) fTracks->SetOwner(own); | |
182 | fTrackCopy=mode; | |
183 | } | |
184 | /////////////////////////////////////////////////////////////////////////// | |
261c0caf | 185 | AliJet::AliJet(const AliJet& j) : TNamed(j),Ali4Vector(j) |
6516b62d | 186 | { |
187 | // Copy constructor | |
c72198f1 | 188 | fNtinit=j.fNtinit; |
189 | fNtmax=j.fNtmax; | |
190 | fQ=j.fQ; | |
191 | fNtrk=j.fNtrk; | |
192 | fTrackCopy=j.fTrackCopy; | |
193 | fUserId=j.fUserId; | |
413d0114 | 194 | if (j.fRef) fRef=new AliPositionObj(*(j.fRef)); |
c72198f1 | 195 | |
ea0b5b7f | 196 | fSelected=0; |
197 | ||
c72198f1 | 198 | fTracks=0; |
199 | if (fNtrk) | |
200 | { | |
201 | fTracks=new TObjArray(fNtmax); | |
202 | if (fTrackCopy) fTracks->SetOwner(); | |
203 | } | |
6516b62d | 204 | |
c72198f1 | 205 | for (Int_t i=1; i<=fNtrk; i++) |
6516b62d | 206 | { |
c72198f1 | 207 | AliTrack* tx=j.GetTrack(i); |
208 | if (fTrackCopy) | |
209 | { | |
5f25234b | 210 | fTracks->Add(tx->Clone()); |
c72198f1 | 211 | } |
212 | else | |
213 | { | |
214 | fTracks->Add(tx); | |
215 | } | |
6516b62d | 216 | } |
217 | } | |
218 | /////////////////////////////////////////////////////////////////////////// | |
d88f97cc | 219 | void AliJet::SetNtinit(Int_t n) |
220 | { | |
221 | // Set the initial maximum number of tracks for this jet | |
222 | fNtinit=n; | |
223 | fNtmax=n; | |
f4d1f676 | 224 | |
79830a7e | 225 | if (fTracks) |
c1f25d1d | 226 | { |
c1f25d1d | 227 | delete fTracks; |
228 | fTracks=0; | |
229 | } | |
413d0114 | 230 | if (fRef) |
231 | { | |
232 | delete fRef; | |
233 | fRef=0; | |
234 | } | |
d88f97cc | 235 | } |
236 | /////////////////////////////////////////////////////////////////////////// | |
237 | void AliJet::Reset() | |
238 | { | |
239 | // Reset all variables to 0 | |
240 | // The max. number of tracks is set to the initial value again | |
241 | fNtrk=0; | |
242 | fQ=0; | |
43bfa5be | 243 | fUserId=0; |
d88f97cc | 244 | Double_t a[4]={0,0,0,0}; |
245 | SetVector(a,"sph"); | |
246 | if (fNtinit > 0) SetNtinit(fNtinit); | |
247 | } | |
248 | /////////////////////////////////////////////////////////////////////////// | |
1fbffa23 | 249 | void AliJet::AddTrack(AliTrack& t) |
d88f97cc | 250 | { |
35044448 | 251 | // Add a track to the jet. |
d88f97cc | 252 | // In case the maximum number of tracks has been reached |
253 | // space will be extended to hold an additional amount of tracks as | |
35044448 | 254 | // was initially reserved. |
1fbffa23 | 255 | // See SetTrackCopy() to tailor the functionality of the stored structures. |
5f25234b | 256 | // |
1f241680 | 257 | // Notes : |
258 | // ------- | |
5f25234b | 259 | // In case a private copy is made, this is performed via the Clone() memberfunction. |
260 | // All AliTrack and derived classes have the default TObject::Clone() memberfunction. | |
261 | // However, derived classes generally contain an internal data structure which may | |
262 | // include pointers to other objects. Therefore it is recommended to provide | |
263 | // for all derived classes a specific copy constructor and override the default Clone() | |
264 | // memberfunction using this copy constructor. | |
265 | // An example for this may be seen from AliTrack. | |
1f241680 | 266 | // |
267 | // In case NO private copy is made, a check will be performed if this | |
268 | // specific track is already present in the jet. | |
269 | // If this is the case, no action is performed to prevent multiple | |
270 | // additions of the same track. | |
271 | ||
5f25234b | 272 | |
1fbffa23 | 273 | AddTrack(t,1); |
274 | } | |
275 | /////////////////////////////////////////////////////////////////////////// | |
276 | void AliJet::AddTrack(AliTrack& t,Int_t copy) | |
277 | { | |
278 | // Internal memberfunction to actually add a track to the jet. | |
279 | // In case the maximum number of tracks has been reached | |
280 | // space will be extended to hold an additional amount of tracks as | |
281 | // was initially reserved. | |
282 | // | |
283 | // If copy=0 NO copy of the track will be made, irrespective of the setting | |
284 | // of the TrackCopy flag. | |
285 | // This allows a proper treatment of automatically generated connecting | |
286 | // tracks between vertices. | |
5f25234b | 287 | // |
1f241680 | 288 | // In case NO copy of the track is made, a check will be performed if this |
289 | // specific track is already present in the jet. | |
290 | // If this is the case, no action is performed to prevent multiple | |
291 | // additions of the same track. | |
292 | // | |
5f25234b | 293 | // Note : |
294 | // In case a private copy is made, this is performed via the Clone() memberfunction. | |
295 | ||
6516b62d | 296 | if (!fTracks) |
297 | { | |
298 | fTracks=new TObjArray(fNtmax); | |
299 | if (fTrackCopy) fTracks->SetOwner(); | |
300 | } | |
1f241680 | 301 | else if (!fTrackCopy || !copy) // Check if this track is already present |
302 | { | |
303 | for (Int_t i=0; i<fNtrk; i++) | |
304 | { | |
305 | AliTrack* tx=(AliTrack*)fTracks->At(i); | |
306 | if (tx == &t) return; | |
307 | } | |
308 | } | |
309 | ||
d88f97cc | 310 | if (fNtrk == fNtmax) // Check if maximum track number is reached |
311 | { | |
312 | fNtmax+=fNtinit; | |
313 | fTracks->Expand(fNtmax); | |
314 | } | |
315 | ||
316 | // Add the track to this jet | |
317 | fNtrk++; | |
6516b62d | 318 | if (fTrackCopy && copy) |
c1f25d1d | 319 | { |
5f25234b | 320 | fTracks->Add(t.Clone()); |
c1f25d1d | 321 | } |
322 | else | |
323 | { | |
324 | fTracks->Add(&t); | |
325 | } | |
35044448 | 326 | |
d88f97cc | 327 | (*this)+=(Ali4Vector&)t; |
328 | fQ+=t.GetCharge(); | |
35044448 | 329 | |
d88f97cc | 330 | } |
331 | /////////////////////////////////////////////////////////////////////////// | |
1f241680 | 332 | void AliJet::Data(TString f,TString u) |
d88f97cc | 333 | { |
334 | // Provide jet information within the coordinate frame f | |
1f241680 | 335 | // |
336 | // The string argument "u" allows to choose between different angular units | |
337 | // in case e.g. a spherical frame is selected. | |
338 | // u = "rad" : angles provided in radians | |
339 | // "deg" : angles provided in degrees | |
340 | // | |
341 | // The defaults are f="car" and u="rad". | |
342 | ||
5f25234b | 343 | const char* name=GetName(); |
344 | const char* title=GetTitle(); | |
345 | ||
346 | cout << " *AliJet::Data*"; | |
347 | if (strlen(name)) cout << " Name : " << GetName(); | |
348 | if (strlen(title)) cout << " Title : " << GetTitle(); | |
349 | cout << endl; | |
387a745b | 350 | cout << " Id : " << fUserId << " Invmass : " << GetInvmass() << " Charge : " << fQ |
ea0b5b7f | 351 | << " Momentum : " << GetMomentum() << endl; |
352 | ||
353 | ShowTracks(0); | |
c72198f1 | 354 | |
1f241680 | 355 | Ali4Vector::Data(f,u); |
d88f97cc | 356 | } |
357 | /////////////////////////////////////////////////////////////////////////// | |
1f241680 | 358 | void AliJet::List(TString f,TString u) |
d88f97cc | 359 | { |
360 | // Provide jet and primary track information within the coordinate frame f | |
1f241680 | 361 | // |
362 | // The string argument "u" allows to choose between different angular units | |
363 | // in case e.g. a spherical frame is selected. | |
364 | // u = "rad" : angles provided in radians | |
365 | // "deg" : angles provided in degrees | |
366 | // | |
367 | // The defaults are f="car" and u="rad". | |
d88f97cc | 368 | |
1f241680 | 369 | Data(f,u); // Information of the current jet |
413d0114 | 370 | if (fRef) { cout << " Ref-point :"; fRef->Data(f,u); } |
d88f97cc | 371 | |
372 | // The tracks of this jet | |
373 | AliTrack* t; | |
374 | for (Int_t it=1; it<=fNtrk; it++) | |
375 | { | |
376 | t=GetTrack(it); | |
377 | if (t) | |
378 | { | |
379 | cout << " ---Track no. " << it << endl; | |
380 | cout << " "; | |
1f241680 | 381 | t->Data(f,u); |
d88f97cc | 382 | } |
383 | else | |
384 | { | |
385 | cout << " *AliJet::List* Error : No track present." << endl; | |
386 | } | |
387 | } | |
388 | } | |
389 | /////////////////////////////////////////////////////////////////////////// | |
1f241680 | 390 | void AliJet::ListAll(TString f,TString u) |
d88f97cc | 391 | { |
392 | // Provide jet and prim.+sec. track information within the coordinate frame f | |
1f241680 | 393 | // |
394 | // The string argument "u" allows to choose between different angular units | |
395 | // in case e.g. a spherical frame is selected. | |
396 | // u = "rad" : angles provided in radians | |
397 | // "deg" : angles provided in degrees | |
398 | // | |
399 | // The defaults are f="car" and u="rad". | |
d88f97cc | 400 | |
1f241680 | 401 | Data(f,u); // Information of the current jet |
413d0114 | 402 | if (fRef) { cout << " Ref-point :"; fRef->Data(f,u); } |
d88f97cc | 403 | |
404 | // The tracks of this jet | |
405 | AliTrack* t; | |
406 | for (Int_t it=1; it<=fNtrk; it++) | |
407 | { | |
408 | t=GetTrack(it); | |
409 | if (t) | |
410 | { | |
411 | cout << " ---Track no. " << it << endl; | |
412 | cout << " "; | |
1f241680 | 413 | t->ListAll(f,u); |
d88f97cc | 414 | } |
415 | else | |
416 | { | |
417 | cout << " *AliJet::List* Error : No track present." << endl; | |
418 | } | |
419 | } | |
420 | } | |
421 | /////////////////////////////////////////////////////////////////////////// | |
325b076c | 422 | Int_t AliJet::GetNtracks(Int_t idmode,Int_t chmode,Int_t pcode) |
d88f97cc | 423 | { |
325b076c | 424 | // Provide the number of user selected tracks in this jet based on the |
425 | // idmode, chmode and pcode selections as specified by the user. | |
426 | // For specification of the selection parameters see GetTracks(). | |
427 | // The default parameters correspond to no selection, which implies | |
428 | // that invokation of GetNtracks() just returns the total number of | |
429 | // tracks registered in this jet. | |
430 | // | |
431 | // Note : In case certain selections are specified, this function | |
432 | // invokes GetTracks(idmode,chmode,pcode) to determine the | |
433 | // number of tracks corresponding to the selections. | |
434 | // When the jet contains a large number of tracks, invokation | |
435 | // of GetTracks(idmode,chmode,pcode) and subsequently invoking | |
436 | // GetEntries() for the resulting TObjArray* might be slightly | |
437 | // faster. | |
438 | ||
439 | Int_t n=0; | |
440 | if (idmode==0 && chmode==2 && pcode==0) | |
441 | { | |
442 | return fNtrk; | |
443 | } | |
444 | else | |
445 | { | |
446 | TObjArray* arr=GetTracks(idmode,chmode,pcode); | |
447 | n=arr->GetEntries(); | |
448 | return n; | |
449 | } | |
d88f97cc | 450 | } |
451 | /////////////////////////////////////////////////////////////////////////// | |
452 | Double_t AliJet::GetEnergy() | |
453 | { | |
454 | // Return the total energy of the jet | |
455 | return GetScalar(); | |
456 | } | |
457 | /////////////////////////////////////////////////////////////////////////// | |
458 | Double_t AliJet::GetMomentum() | |
459 | { | |
460 | // Return the value of the total jet 3-momentum | |
d071d629 | 461 | // The error can be obtained by invoking GetResultError() after |
462 | // invokation of GetMomentum(). | |
463 | Double_t norm=fV.GetNorm(); | |
464 | fDresult=fV.GetResultError(); | |
465 | return norm; | |
d88f97cc | 466 | } |
467 | /////////////////////////////////////////////////////////////////////////// | |
261c0caf | 468 | Ali3Vector AliJet::Get3Momentum() const |
d88f97cc | 469 | { |
470 | // Return the the total jet 3-momentum | |
471 | Ali3Vector p=Get3Vector(); | |
472 | return p; | |
473 | } | |
474 | /////////////////////////////////////////////////////////////////////////// | |
475 | Double_t AliJet::GetInvmass() | |
476 | { | |
477 | // Return the invariant mass of the jet | |
478 | Double_t m2=Dot(*this); | |
479 | if (m2>0) | |
480 | { | |
481 | return sqrt(m2); | |
482 | } | |
483 | else | |
484 | { | |
485 | return 0; | |
486 | } | |
487 | } | |
488 | /////////////////////////////////////////////////////////////////////////// | |
261c0caf | 489 | Float_t AliJet::GetCharge() const |
d88f97cc | 490 | { |
491 | // Return the total charge of the jet | |
492 | return fQ; | |
493 | } | |
494 | /////////////////////////////////////////////////////////////////////////// | |
261c0caf | 495 | AliTrack* AliJet::GetTrack(Int_t i) const |
d88f97cc | 496 | { |
497 | // Return the i-th track of this jet | |
ea0b5b7f | 498 | |
499 | if (!fTracks) return 0; | |
500 | ||
501 | if (i<=0 || i>fNtrk) | |
f34f4acb | 502 | { |
ea0b5b7f | 503 | cout << " *AliJet*::GetTrack* Invalid argument i : " << i |
504 | << " Ntrk = " << fNtrk << endl; | |
f34f4acb | 505 | return 0; |
506 | } | |
507 | else | |
508 | { | |
ea0b5b7f | 509 | return (AliTrack*)fTracks->At(i-1); |
f34f4acb | 510 | } |
d88f97cc | 511 | } |
512 | /////////////////////////////////////////////////////////////////////////// | |
261c0caf | 513 | AliTrack* AliJet::GetIdTrack(Int_t id) const |
43bfa5be | 514 | { |
515 | // Return the track with user identifier "id" of this jet | |
4f368c8c | 516 | if (!fTracks) return 0; |
517 | ||
43bfa5be | 518 | AliTrack* tx=0; |
4f368c8c | 519 | for (Int_t i=0; i<fNtrk; i++) |
43bfa5be | 520 | { |
4f368c8c | 521 | tx=(AliTrack*)fTracks->At(i); |
522 | if (id == tx->GetId()) return tx; | |
43bfa5be | 523 | } |
4f368c8c | 524 | return 0; // No matching id found |
43bfa5be | 525 | } |
526 | /////////////////////////////////////////////////////////////////////////// | |
ea0b5b7f | 527 | TObjArray* AliJet::GetTracks(Int_t idmode,Int_t chmode,Int_t pcode) |
528 | { | |
529 | // Provide references to user selected tracks based on the idmode, chmode | |
530 | // and pcode selections as specified by the user. | |
531 | // | |
532 | // The following selection combinations are available : | |
533 | // ---------------------------------------------------- | |
534 | // idmode = -1 ==> Select tracks with negative user identifier "id" | |
535 | // 0 ==> No selection on user identifier | |
536 | // 1 ==> Select tracks with positive user identifier "id" | |
537 | // | |
538 | // chmode = -1 ==> Select tracks with negative charge | |
539 | // 0 ==> Select neutral tracks | |
540 | // 1 ==> Select tracks with positive charge | |
541 | // 2 ==> No selection on charge | |
325b076c | 542 | // 3 ==> Select all charged tracks |
ea0b5b7f | 543 | // |
544 | // pcode = 0 ==> No selection on particle code | |
545 | // X ==> Select tracks with particle code +X or -X | |
546 | // This allows selection of both particles and anti-particles | |
547 | // in case of PDG particle codes. | |
548 | // Selection of either particles or anti-particles can be | |
549 | // obtained in combination with the "chmode" selector. | |
550 | // | |
551 | // Examples : | |
552 | // ---------- | |
553 | // idmode=-1 chmode=0 pcode=0 : Selection of all neutral tracks with negative id. | |
554 | // idmode=0 chmode=2 pcode=211 : Selection of all charged pions (PDG convention). | |
555 | // idmode=0 chmode=1 pcode=321 : Selection of all positive kaons (PDG convention). | |
556 | // | |
557 | // The default values are idmode=0 chmode=2 pcode=0 (i.e. no selections applied). | |
558 | // | |
559 | // Notes : | |
560 | // ------- | |
561 | // 1) In case the user has labeled simulated tracks with negative id and | |
562 | // reconstructed tracks with positive id, this memberfunction provides | |
563 | // easy access to either all simulated or reconstructed tracks. | |
564 | // 2) Subsequent invokations of this memberfunction with e.g. chmode=-1 and chmode=1 | |
565 | // provides a convenient way to investigate particle pairs with opposite charge | |
566 | // (e.g. for invariant mass analysis). | |
567 | ||
568 | if (fSelected) | |
569 | { | |
570 | fSelected->Clear(); | |
571 | } | |
572 | else | |
573 | { | |
574 | fSelected=new TObjArray(); | |
575 | } | |
576 | ||
577 | if (!fTracks) return fSelected; | |
578 | ||
579 | AliTrack* tx=0; | |
580 | Int_t code=0; | |
581 | Int_t id=0; | |
582 | Float_t q=0; | |
583 | for (Int_t i=0; i<fNtrk; i++) | |
584 | { | |
585 | tx=(AliTrack*)fTracks->At(i); | |
586 | if (!tx) continue; | |
587 | ||
588 | code=tx->GetParticleCode(); | |
589 | if (pcode && abs(pcode)!=abs(code)) continue; | |
590 | ||
591 | id=tx->GetId(); | |
592 | if (idmode==-1 && id>=0) continue; | |
593 | if (idmode==1 && id<=0) continue; | |
594 | ||
595 | q=tx->GetCharge(); | |
596 | if (chmode==-1 && q>=0) continue; | |
597 | if (chmode==0 && fabs(q)>1e-10) continue; | |
598 | if (chmode==1 && q<=0) continue; | |
325b076c | 599 | if (chmode==3 && fabs(q)<1e-10) continue; |
ea0b5b7f | 600 | |
601 | fSelected->Add(tx); | |
602 | } | |
603 | ||
604 | return fSelected; | |
605 | } | |
606 | /////////////////////////////////////////////////////////////////////////// | |
325b076c | 607 | void AliJet::ShowTracks(Int_t mode) |
ea0b5b7f | 608 | { |
609 | // Provide an overview of the available tracks. | |
610 | // The argument mode determines the amount of information as follows : | |
611 | // mode = 0 ==> Only printout of the number of tracks | |
612 | // 1 ==> Provide a listing with 1 line of info for each track | |
613 | // | |
614 | // The default is mode=1. | |
615 | // | |
616 | Int_t ntk=GetNtracks(); | |
617 | if (ntk) | |
618 | { | |
619 | if (!mode) | |
620 | { | |
621 | cout << " There are " << ntk << " tracks available." << endl; | |
622 | } | |
623 | else | |
624 | { | |
625 | cout << " The following " << ntk << " tracks are available :" << endl; | |
626 | for (Int_t i=1; i<=ntk; i++) | |
627 | { | |
628 | AliTrack* tx=GetTrack(i); | |
629 | if (tx) | |
630 | { | |
631 | const char* name=tx->GetName(); | |
632 | const char* title=tx->GetTitle(); | |
633 | cout << " Track : " << i; | |
634 | cout << " Id : " << tx->GetId(); | |
635 | cout << " Q : " << tx->GetCharge() << " m : " << tx->GetMass() << " p : " << tx->GetMomentum(); | |
636 | if (strlen(name)) cout << " Name : " << name; | |
637 | if (strlen(title)) cout << " Title : " << title; | |
638 | cout << endl; | |
639 | } | |
640 | } | |
641 | } | |
642 | } | |
643 | else | |
644 | { | |
645 | cout << " No tracks are present." << endl; | |
646 | } | |
647 | } | |
648 | /////////////////////////////////////////////////////////////////////////// | |
d071d629 | 649 | Double_t AliJet::GetPt() |
650 | { | |
651 | // Provide trans. momentum value w.r.t. z-axis. | |
652 | // The error on the value can be obtained by GetResultError() | |
653 | // after invokation of GetPt(). | |
654 | Ali3Vector v; | |
655 | v=GetVecTrans(); | |
656 | Double_t norm=v.GetNorm(); | |
657 | fDresult=v.GetResultError(); | |
658 | ||
659 | return norm; | |
660 | } | |
661 | /////////////////////////////////////////////////////////////////////////// | |
662 | Double_t AliJet::GetPl() | |
663 | { | |
664 | // Provide long. momentum value w.r.t. z-axis. | |
8adaf597 | 665 | // Note : the returned value can also be negative. |
d071d629 | 666 | // The error on the value can be obtained by GetResultError() |
667 | // after invokation of GetPl(). | |
668 | Ali3Vector v; | |
669 | v=GetVecLong(); | |
8adaf597 | 670 | |
671 | Double_t pl=v.GetNorm(); | |
d071d629 | 672 | fDresult=v.GetResultError(); |
673 | ||
8adaf597 | 674 | Double_t a[3]; |
675 | v.GetVector(a,"sph"); | |
676 | if (cos(a[1])<0) pl=-pl; | |
677 | ||
678 | return pl; | |
d071d629 | 679 | } |
680 | /////////////////////////////////////////////////////////////////////////// | |
681 | Double_t AliJet::GetEt() | |
682 | { | |
683 | // Provide trans. energy value w.r.t. z-axis. | |
684 | // The error on the value can be obtained by GetResultError() | |
685 | // after invokation of GetEt(). | |
686 | Double_t et=GetScaTrans(); | |
687 | ||
688 | return et; | |
689 | } | |
690 | /////////////////////////////////////////////////////////////////////////// | |
691 | Double_t AliJet::GetEl() | |
692 | { | |
693 | // Provide long. energy value w.r.t. z-axis. | |
8adaf597 | 694 | // Note : the returned value can also be negative. |
d071d629 | 695 | // The error on the value can be obtained by GetResultError() |
696 | // after invokation of GetEl(). | |
697 | Double_t el=GetScaLong(); | |
698 | ||
699 | return el; | |
700 | } | |
701 | /////////////////////////////////////////////////////////////////////////// | |
702 | Double_t AliJet::GetMt() | |
703 | { | |
704 | // Provide transverse mass value w.r.t. z-axis. | |
705 | // The error on the value can be obtained by GetResultError() | |
706 | // after invokation of GetMt(). | |
707 | Double_t pt=GetPt(); | |
708 | Double_t dpt=GetResultError(); | |
709 | Double_t m=GetInvmass(); | |
710 | Double_t dm=GetResultError(); | |
711 | ||
712 | Double_t mt=sqrt(pt*pt+m*m); | |
713 | Double_t dmt2=0; | |
714 | if (mt) dmt2=(pow((pt*dpt),2)+pow((m*dm),2))/(mt*mt); | |
715 | ||
716 | fDresult=sqrt(dmt2); | |
717 | return mt; | |
718 | } | |
719 | /////////////////////////////////////////////////////////////////////////// | |
8adaf597 | 720 | Double_t AliJet::GetRapidity() |
721 | { | |
722 | // Provide rapidity value w.r.t. z-axis. | |
723 | // The error on the value can be obtained by GetResultError() | |
724 | // after invokation of GetRapidity(). | |
725 | // Note : Also GetPseudoRapidity() is available since this class is | |
726 | // derived from Ali4Vector. | |
727 | Double_t e=GetEnergy(); | |
728 | Double_t de=GetResultError(); | |
729 | Double_t pl=GetPl(); | |
730 | Double_t dpl=GetResultError(); | |
731 | Double_t sum=e+pl; | |
732 | Double_t dif=e-pl; | |
733 | ||
734 | Double_t y=9999,dy2=0; | |
735 | if (sum && dif) y=0.5*log(sum/dif); | |
736 | ||
737 | if (sum*dif) dy2=(1./(sum*dif))*(pow((pl*de),2)+pow((e*dpl),2)); | |
738 | ||
739 | fDresult=sqrt(dy2); | |
740 | return y; | |
741 | } | |
742 | /////////////////////////////////////////////////////////////////////////// | |
c1f25d1d | 743 | void AliJet::SetTrackCopy(Int_t j) |
744 | { | |
745 | // (De)activate the creation of private copies of the added tracks. | |
746 | // j=0 ==> No private copies are made; pointers of original tracks are stored. | |
747 | // j=1 ==> Private copies of the tracks are made and these pointers are stored. | |
748 | // | |
749 | // Note : Once the storage contains pointer(s) to AliTrack(s) one cannot | |
750 | // change the TrackCopy mode anymore. | |
751 | // To change the TrackCopy mode for an existing AliJet containing | |
752 | // tracks one first has to invoke Reset(). | |
753 | if (!fTracks) | |
754 | { | |
755 | if (j==0 || j==1) | |
756 | { | |
757 | fTrackCopy=j; | |
758 | } | |
759 | else | |
760 | { | |
761 | cout << "*AliJet::SetTrackCopy* Invalid argument : " << j << endl; | |
762 | } | |
763 | } | |
764 | else | |
765 | { | |
766 | cout << "*AliJet::SetTrackCopy* Storage already contained tracks." | |
767 | << " ==> TrackCopy mode not changed." << endl; | |
768 | } | |
769 | } | |
770 | /////////////////////////////////////////////////////////////////////////// | |
261c0caf | 771 | Int_t AliJet::GetTrackCopy() const |
c1f25d1d | 772 | { |
773 | // Provide value of the TrackCopy mode. | |
774 | // 0 ==> No private copies are made; pointers of original tracks are stored. | |
775 | // 1 ==> Private copies of the tracks are made and these pointers are stored. | |
776 | return fTrackCopy; | |
777 | } | |
778 | /////////////////////////////////////////////////////////////////////////// | |
43bfa5be | 779 | void AliJet::SetId(Int_t id) |
780 | { | |
781 | // Set a user defined identifier for this jet. | |
782 | fUserId=id; | |
783 | } | |
784 | /////////////////////////////////////////////////////////////////////////// | |
261c0caf | 785 | Int_t AliJet::GetId() const |
43bfa5be | 786 | { |
787 | // Provide the user defined identifier of this jet. | |
788 | return fUserId; | |
789 | } | |
790 | /////////////////////////////////////////////////////////////////////////// | |
413d0114 | 791 | void AliJet::SetReferencePoint(AliPosition& p) |
792 | { | |
793 | // Store the position of the jet reference-point. | |
794 | // The reference-point of a jet provides a means to define a generic | |
795 | // space-time location for the jet as a whole. | |
796 | // This doesn't have to be necessarily the location where all the constituent | |
797 | // tracks originate (e.g. a bundle of parallel tracks doesn't have such | |
798 | // a location). As such the meaning of this reference-point is different from | |
799 | // a normal vertex position and allows to provide complimentary information. | |
800 | // This reference point is the preferable point to start e.g. extrapolations | |
801 | // and investigate coincidences in space and/or time. | |
802 | if (fRef) delete fRef; | |
803 | fRef=new AliPositionObj(p); | |
804 | } | |
805 | /////////////////////////////////////////////////////////////////////////// | |
806 | AliPosition* AliJet::GetReferencePoint() | |
807 | { | |
808 | // Provide the position of the jet reference-point. | |
809 | // The reference-point of a jet provides a means to define a generic | |
810 | // space-time location for the jet as a whole. | |
811 | // This doesn't have to be necessarily the location where all the constituent | |
812 | // tracks originate (e.g. a bundle of parallel tracks doesn't have such | |
813 | // a location). As such the meaning of this reference-point is different from | |
814 | // a normal vertex position and allows to provide complimentary information. | |
815 | // This reference point is the preferable point to start e.g. extrapolations | |
816 | // and investigate coincidences in space and/or time. | |
817 | return fRef; | |
818 | } | |
819 | /////////////////////////////////////////////////////////////////////////// | |
e51b7d1a | 820 | TObjArray* AliJet::SortTracks(Int_t mode,TObjArray* tracks) |
821 | { | |
822 | // Order the references to an array of tracks by looping over the input array "tracks" | |
823 | // and checking the value of a certain observable. | |
824 | // The ordered array is returned as a TObjArray. | |
825 | // In case tracks=0 (default), the registered tracks of the current jet are used. | |
826 | // Note that the original track array is not modified. | |
827 | // Via the "mode" argument the user can specify the observable to be checked upon | |
828 | // and specify whether sorting should be performed in decreasing order (mode<0) | |
829 | // or in increasing order (mode>0). | |
830 | // | |
831 | // The convention for the observable selection is the following : | |
832 | // mode : 1 ==> Number of signals associated to the track | |
833 | // 2 ==> Track energy | |
834 | // 3 ==> Track momentum | |
835 | // 4 ==> Mass of the track | |
836 | // 5 ==> Transverse momentum of the track | |
837 | // 6 ==> Longitudinal momentum of the track | |
838 | // 7 ==> Transverse energy of the track | |
839 | // 8 ==> Longitudinal energy of the track | |
840 | // 9 ==> Transverse mass of the track | |
841 | // 10 ==> Track rapidity | |
842 | // 11 ==> Pseudo-rapidity of the track | |
843 | // | |
844 | // The default is mode=-1. | |
845 | // | |
846 | // Note : This sorting routine uses a common area in memory, which is used | |
847 | // by various other sorting facilities as well. | |
848 | // This means that the resulting sorted TObjArray may be overwritten | |
849 | // when another sorting is invoked. | |
850 | // To retain the sorted list of pointers, the user is advised to copy | |
851 | // the pointers contained in the returned TObjArray into a private | |
852 | // TObjArray instance. | |
853 | ||
854 | if (fSelected) | |
855 | { | |
856 | delete fSelected; | |
857 | fSelected=0; | |
858 | } | |
859 | ||
860 | if (!tracks) tracks=fTracks; | |
861 | ||
862 | if (abs(mode)>11 || !tracks) return fSelected; | |
863 | ||
864 | Int_t ntracks=tracks->GetEntries(); | |
865 | if (!ntracks) | |
866 | { | |
867 | return fSelected; | |
868 | } | |
869 | else | |
870 | { | |
871 | fSelected=new TObjArray(ntracks); | |
872 | } | |
873 | ||
874 | Double_t val1,val2; // Values of the observable to be tested upon | |
875 | ||
876 | Int_t nord=0; | |
877 | for (Int_t i=0; i<ntracks; i++) // Loop over all tracks of the array | |
878 | { | |
879 | AliTrack* tx=(AliTrack*)tracks->At(i); | |
880 | ||
881 | if (!tx) continue; | |
882 | ||
883 | if (nord == 0) // store the first track at the first ordered position | |
884 | { | |
885 | nord++; | |
886 | fSelected->AddAt(tx,nord-1); | |
887 | continue; | |
888 | } | |
889 | ||
890 | for (Int_t j=0; j<=nord; j++) // put track in the right ordered position | |
891 | { | |
892 | if (j == nord) // track has smallest (mode<0) or largest (mode>0) observable value seen so far | |
893 | { | |
894 | nord++; | |
895 | fSelected->AddAt(tx,j); // add track at the end | |
896 | break; // go for next track | |
897 | } | |
898 | ||
899 | switch (abs(mode)) | |
900 | { | |
901 | case 1: | |
902 | val1=tx->GetNsignals(); | |
903 | val2=((AliTrack*)fSelected->At(j))->GetNsignals(); | |
904 | break; | |
905 | case 2: | |
906 | val1=tx->GetEnergy(); | |
907 | val2=((AliTrack*)fSelected->At(j))->GetEnergy(); | |
908 | break; | |
909 | case 3: | |
910 | val1=tx->GetMomentum(); | |
911 | val2=((AliTrack*)fSelected->At(j))->GetMomentum(); | |
912 | break; | |
913 | case 4: | |
914 | val1=tx->GetMass(); | |
915 | val2=((AliTrack*)fSelected->At(j))->GetMass(); | |
916 | break; | |
917 | case 5: | |
918 | val1=tx->GetPt(); | |
919 | val2=((AliTrack*)fSelected->At(j))->GetPt(); | |
920 | break; | |
921 | case 6: | |
922 | val1=tx->GetPl(); | |
923 | val2=((AliTrack*)fSelected->At(j))->GetPl(); | |
924 | break; | |
925 | case 7: | |
926 | val1=tx->GetEt(); | |
927 | val2=((AliTrack*)fSelected->At(j))->GetEt(); | |
928 | break; | |
929 | case 8: | |
930 | val1=tx->GetEl(); | |
931 | val2=((AliTrack*)fSelected->At(j))->GetEl(); | |
932 | break; | |
933 | case 9: | |
934 | val1=tx->GetMt(); | |
935 | val2=((AliTrack*)fSelected->At(j))->GetMt(); | |
936 | break; | |
937 | case 10: | |
938 | val1=tx->GetRapidity(); | |
939 | val2=((AliTrack*)fSelected->At(j))->GetRapidity(); | |
940 | break; | |
941 | case 11: | |
942 | val1=tx->GetPseudoRapidity(); | |
943 | val2=((AliTrack*)fSelected->At(j))->GetPseudoRapidity(); | |
944 | break; | |
945 | } | |
946 | ||
f5b75967 | 947 | if (mode<0 && val1 <= val2) continue; |
948 | if (mode>0 && val1 >= val2) continue; | |
e51b7d1a | 949 | |
950 | nord++; | |
951 | for (Int_t k=nord-1; k>j; k--) // create empty position | |
952 | { | |
953 | fSelected->AddAt(fSelected->At(k-1),k); | |
954 | } | |
955 | fSelected->AddAt(tx,j); // put track at empty position | |
956 | break; // go for next track | |
957 | } | |
958 | } | |
959 | return fSelected; | |
960 | } | |
961 | /////////////////////////////////////////////////////////////////////////// | |
261c0caf | 962 | TObject* AliJet::Clone(const char* name) const |
5f25234b | 963 | { |
964 | // Make a deep copy of the current object and provide the pointer to the copy. | |
965 | // This memberfunction enables automatic creation of new objects of the | |
966 | // correct type depending on the object type, a feature which may be very useful | |
967 | // for containers when adding objects in case the container owns the objects. | |
968 | // This feature allows e.g. AliVertex to store either AliJet objects or | |
969 | // objects derived from AliJet via the AddJet memberfunction, provided | |
970 | // these derived classes also have a proper Clone memberfunction. | |
971 | ||
972 | AliJet* jet=new AliJet(*this); | |
973 | if (name) | |
974 | { | |
975 | if (strlen(name)) jet->SetName(name); | |
976 | } | |
977 | return jet; | |
978 | } | |
979 | /////////////////////////////////////////////////////////////////////////// |