Added methods for finding a given module in the DDL map (F. Prino)
[u/mrichter/AliRoot.git] / STEER / AliAODVertex.cxx
CommitLineData
df9db588 1/**************************************************************************
2 * Copyright(c) 1998-2007, 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
16/* $Id$ */
17
18//-------------------------------------------------------------------------
19// AOD track base class
20// Base class for Analysis Object Data
21// Generic version
22// Author: Markus Oldenburg, CERN
23//-------------------------------------------------------------------------
24
25#include "AliAODVertex.h"
26
0657f082 27#include "AliAODTrack.h"
28
df9db588 29ClassImp(AliAODVertex)
30
31//______________________________________________________________________________
32AliAODVertex::AliAODVertex() :
33 TObject(),
0657f082 34 fChi2perNDF(-999.),
02153d58 35 fID(-1),
9333290e 36 fType(kUndef),
9f7c531f 37 fCovMatrix(NULL),
0f09edc3 38 fParent(),
9333290e 39 fDaughters()
40 {
df9db588 41 // default constructor
42
43 fPosition[0] = fPosition[1] = fPosition[2] = -999.;
44}
45
46//______________________________________________________________________________
47AliAODVertex::AliAODVertex(const Double_t position[3],
31fd97b2 48 const Double_t covMatrix[6],
0657f082 49 Double_t chi2perNDF,
df9db588 50 TObject *parent,
02153d58 51 Short_t id,
df9db588 52 Char_t vtype) :
53 TObject(),
0657f082 54 fChi2perNDF(chi2perNDF),
02153d58 55 fID(id),
9333290e 56 fType(vtype),
9f7c531f 57 fCovMatrix(NULL),
df9db588 58 fParent(parent),
9333290e 59 fDaughters()
df9db588 60{
61 // constructor
62
63 SetPosition(position);
64 if (covMatrix) SetCovMatrix(covMatrix);
65}
66
67//______________________________________________________________________________
68AliAODVertex::AliAODVertex(const Float_t position[3],
69 const Float_t covMatrix[6],
0657f082 70 Double_t chi2perNDF,
df9db588 71 TObject *parent,
02153d58 72 Short_t id,
df9db588 73 Char_t vtype) :
4d209fca 74
df9db588 75 TObject(),
0657f082 76 fChi2perNDF(chi2perNDF),
02153d58 77 fID(id),
9333290e 78 fType(vtype),
9f7c531f 79 fCovMatrix(NULL),
df9db588 80 fParent(parent),
9333290e 81 fDaughters()
df9db588 82{
83 // constructor
84
85 SetPosition(position);
86 if (covMatrix) SetCovMatrix(covMatrix);
87}
88
89//______________________________________________________________________________
90AliAODVertex::AliAODVertex(const Double_t position[3],
0657f082 91 Double_t chi2perNDF,
df9db588 92 Char_t vtype) :
93 TObject(),
0657f082 94 fChi2perNDF(chi2perNDF),
02153d58 95 fID(-1),
9333290e 96 fType(vtype),
9f7c531f 97 fCovMatrix(NULL),
0f09edc3 98 fParent(),
9333290e 99 fDaughters()
df9db588 100{
101 // constructor without covariance matrix
102
103 SetPosition(position);
104}
105
106//______________________________________________________________________________
107AliAODVertex::AliAODVertex(const Float_t position[3],
0657f082 108 Double_t chi2perNDF,
df9db588 109 Char_t vtype) :
110 TObject(),
0657f082 111 fChi2perNDF(chi2perNDF),
02153d58 112 fID(-1),
9333290e 113 fType(vtype),
9f7c531f 114 fCovMatrix(NULL),
0f09edc3 115 fParent(),
9333290e 116 fDaughters()
df9db588 117{
118 // constructor without covariance matrix
119
120 SetPosition(position);
121}
122
123//______________________________________________________________________________
124AliAODVertex::~AliAODVertex()
125{
126 // Destructor
127
128 delete fCovMatrix;
129}
130
131//______________________________________________________________________________
132AliAODVertex::AliAODVertex(const AliAODVertex& vtx) :
133 TObject(vtx),
0657f082 134 fChi2perNDF(vtx.fChi2perNDF),
02153d58 135 fID(vtx.fID),
9333290e 136 fType(vtx.fType),
9f7c531f 137 fCovMatrix(NULL),
df9db588 138 fParent(vtx.fParent),
9333290e 139 fDaughters(vtx.fDaughters)
df9db588 140{
141 // Copy constructor.
142
143 for (int i = 0; i < 3; i++)
144 fPosition[i] = vtx.fPosition[i];
145
5d62ce04 146 if (vtx.fCovMatrix) fCovMatrix=new AliAODRedCov<3>(*vtx.fCovMatrix);
df9db588 147}
148
149//______________________________________________________________________________
150AliAODVertex& AliAODVertex::operator=(const AliAODVertex& vtx)
151{
152 // Assignment operator
153 if (this != &vtx) {
154
155 // name and type
156 TObject::operator=(vtx);
157
158 //momentum
159 for (int i = 0; i < 3; i++)
160 fPosition[i] = vtx.fPosition[i];
161
0657f082 162 fChi2perNDF = vtx.fChi2perNDF;
02153d58 163 fID = vtx.fID;
9333290e 164 fType = vtx.fType;
9f7c531f 165
df9db588 166 //covariance matrix
167 delete fCovMatrix;
9f7c531f 168 fCovMatrix = NULL;
5d62ce04 169 if (vtx.fCovMatrix) fCovMatrix=new AliAODRedCov<3>(*vtx.fCovMatrix);
df9db588 170
171 //other stuff
df9db588 172 fParent = vtx.fParent;
173 fDaughters = vtx.fDaughters;
df9db588 174 }
175
176 return *this;
177}
178
179//______________________________________________________________________________
180template <class T> void AliAODVertex::GetSigmaXYZ(T sigma[3]) const
181{
182 // Return errors on vertex position in thrust frame
183
184 if(fCovMatrix) {
185 sigma[0]=fCovMatrix[3]; //GetCovXZ
186 sigma[1]=fCovMatrix[4]; //GetCovYZ
187 sigma[2]=fCovMatrix[5]; //GetCovZZ
188 } else
189 sigma[0]=sigma[1]=sigma[2]=-999.;
190
191 /*
192 for (int i = 0, j = 6; i < 3; i++) {
193 j -= i+1;
194 sigma[2-i] = fCovMatrix ? TMath::Sqrt(fCovMatrix[j]) : -999.;
195 }
196 */
197}
198
199//______________________________________________________________________________
0657f082 200Int_t AliAODVertex::GetNContributors() const
201{
202 // Returns the number of tracks used to fit this vertex.
203
204 Int_t cont = 0;
205
206 for (Int_t iDaug = 0; iDaug < GetNDaughters(); iDaug++) {
207 if (((AliAODTrack*)fDaughters.At(iDaug))->GetUsedForVtxFit()) cont++;
208 }
209
210 return cont;
211}
212
213//______________________________________________________________________________
df9db588 214Bool_t AliAODVertex::HasDaughter(TObject *daughter) const
215{
216 // Checks if the given daughter (particle) is part of this vertex.
217
218 TRefArrayIter iter(&fDaughters);
219 while (TObject *daugh = iter.Next()) {
220 if (daugh == daughter) return kTRUE;
221 }
222 return kFALSE;
223}
224
225//______________________________________________________________________________
226Double_t AliAODVertex::RotatedCovMatrixXX(Double_t phi, Double_t theta) const
227{
228 // XX term of covariance matrix after rotation by phi around z-axis
229 // and, then, by theta around new y-axis
230
231 if (!fCovMatrix) {
232 //AliFatal("Covariance matrix not set");
233 return -999.;
234 }
235
236 Double_t covMatrix[6];
237
238 GetCovMatrix(covMatrix);
239
240 Double_t cp = TMath::Cos(phi);
241 Double_t sp = TMath::Sin(phi);
242 Double_t ct = TMath::Cos(theta);
243 Double_t st = TMath::Sin(theta);
244 return
245 covMatrix[0]*cp*cp*ct*ct // GetCovXX
246 +covMatrix[1]*2.*cp*sp*ct*ct // GetCovXY
247 +covMatrix[3]*2.*cp*ct*st // GetCovXZ
248 +covMatrix[2]*sp*sp*ct*ct // GetCovYY
249 +covMatrix[4]*2.*sp*ct*st // GetCovYZ
250 +covMatrix[5]*st*st; // GetCovZZ
251}
252
253//______________________________________________________________________________
254Double_t AliAODVertex::RotatedCovMatrixXY(Double_t phi, Double_t theta) const
255{
256 // XY term of covariance matrix after rotation by phi around z-axis
257 // and, then, by theta around new y-axis
258
259 if (!fCovMatrix) {
260 //AliFatal("Covariance matrix not set");
261 return -999.;
262 }
263
264 Double_t covMatrix[6];
265
266 GetCovMatrix(covMatrix);
267
268 Double_t cp = TMath::Cos(phi);
269 Double_t sp = TMath::Sin(phi);
270 Double_t ct = TMath::Cos(theta);
271 Double_t st = TMath::Sin(theta);
272 return
273 -covMatrix[0]*cp*sp*ct // GetCovXX
274 +covMatrix[1]*ct*(cp*cp-sp*sp) // GetCovXY
275 -covMatrix[3]*sp*st // GetCovXZ
276 +covMatrix[2]*cp*sp*ct // GetCovYY
277 +covMatrix[4]*cp*st; // GetCovYZ
278}
279
280//______________________________________________________________________________
281Double_t AliAODVertex::RotatedCovMatrixXZ(Double_t phi, Double_t theta) const
282{
283 // XZ term of covariance matrix after rotation by phi around z-axis
284 // and, then, by theta around new y-axis
285
286 if (!fCovMatrix) {
287 //AliFatal("Covariance matrix not set");
288 return -999.;
289 }
290
291 Double_t covMatrix[6];
292
293 GetCovMatrix(covMatrix);
294
295 Double_t cp = TMath::Cos(phi);
296 Double_t sp = TMath::Sin(phi);
297 Double_t ct = TMath::Cos(theta);
298 Double_t st = TMath::Sin(theta);
299 return
300 -covMatrix[0]*cp*cp*ct*st // GetCovXX
301 -covMatrix[1]*2.*cp*sp*ct*st // GetCovXY
302 +covMatrix[3]*cp*(ct*ct-st*st) // GetCovXZ
303 -covMatrix[2]*sp*sp*ct*st // GetCovYY
304 +covMatrix[4]*sp*(ct*ct-st*st) // GetCovYZ
305 +covMatrix[5]*ct*st; // GetCovZZ
306}
307
308//______________________________________________________________________________
309Double_t AliAODVertex::RotatedCovMatrixYY(Double_t phi) const
310{
311 // YY term of covariance matrix after rotation by phi around z-axis
312 // and, then, by theta around new y-axis
313
314 if (!fCovMatrix) {
315 //AliFatal("Covariance matrix not set");
316 return -999.;
317 }
318
319 Double_t covMatrix[6];
320
321 GetCovMatrix(covMatrix);
322
323 Double_t cp = TMath::Cos(phi);
324 Double_t sp = TMath::Sin(phi);
325 return
326 covMatrix[0]*sp*sp // GetCovXX
327 -covMatrix[1]*2.*cp*sp // GetCovXY
328 +covMatrix[2]*cp*cp; // GetCovYY
329}
330
331//______________________________________________________________________________
332Double_t AliAODVertex::RotatedCovMatrixYZ(Double_t phi, Double_t theta) const
333{
334 // YZ term of covariance matrix after rotation by phi around z-axis
335 // and, then, by theta around new y-axis
336
337 if (!fCovMatrix) {
338 //AliFatal("Covariance matrix not set");
339 return -999.;
340 }
341
342 Double_t covMatrix[6];
343
344 GetCovMatrix(covMatrix);
345
346 Double_t cp = TMath::Cos(phi);
347 Double_t sp = TMath::Sin(phi);
348 Double_t ct = TMath::Cos(theta);
349 Double_t st = TMath::Sin(theta);
350 return
351 covMatrix[0]*cp*sp*st // GetCovXX
352 +covMatrix[1]*st*(sp*sp-cp*cp) // GetCovXY
353 -covMatrix[3]*sp*ct // GetCovXZ
354 -covMatrix[2]*cp*sp*st // GetCovYY
355 +covMatrix[4]*cp*ct; // GetCovYZ
356}
357
358//______________________________________________________________________________
359Double_t AliAODVertex::RotatedCovMatrixZZ(Double_t phi, Double_t theta) const
360{
361 // ZZ term of covariance matrix after rotation by phi around z-axis
362 // and, then, by theta around new y-axis
363
364 if (!fCovMatrix) {
365 //AliFatal("Covariance matrix not set");
366 return -999.;
367 }
368
369 Double_t covMatrix[6];
370
371 GetCovMatrix(covMatrix);
372
373 Double_t cp = TMath::Cos(phi);
374 Double_t sp = TMath::Sin(phi);
375 Double_t ct = TMath::Cos(theta);
376 Double_t st = TMath::Sin(theta);
377 return
378 covMatrix[0]*cp*cp*st*st // GetCovXX
379 +covMatrix[1]*2.*cp*sp*st*st // GetCovXY
380 -covMatrix[3]*2.*cp*ct*st // GetCovXZ
381 +covMatrix[2]*sp*sp*st*st // GetCovYY
382 -covMatrix[4]*2.*sp*sp*ct*st // GetCovYZ
383 +covMatrix[5]*ct*ct; // GetCovZZ
384}
385
386//______________________________________________________________________________
387Double_t AliAODVertex::DistanceToVertex(AliAODVertex *vtx) const
388{
389 // distance in 3D to another AliAODVertex
390
391 Double_t dx = GetX()-vtx->GetX();
392 Double_t dy = GetY()-vtx->GetY();
393 Double_t dz = GetZ()-vtx->GetZ();
394
395 return TMath::Sqrt(dx*dx+dy*dy+dz*dz);
396}
397
398//______________________________________________________________________________
399Double_t AliAODVertex::DistanceXYToVertex(AliAODVertex *vtx) const
400{
401 // distance in XY to another AliAODVertex
402
403 Double_t dx = GetX()-vtx->GetX();
404 Double_t dy = GetY()-vtx->GetY();
405
406 return TMath::Sqrt(dx*dx+dy*dy);
407}
408
409//______________________________________________________________________________
410Double_t AliAODVertex::ErrorDistanceToVertex(AliAODVertex *vtx) const
411{
412 // error on the distance in 3D to another AliAODVertex
413
414 Double_t phi,theta;
415 PhiAndThetaToVertex(vtx,phi,theta);
416 // error2 due to this vertex
417 Double_t error2 = RotatedCovMatrixXX(phi,theta);
418 // error2 due to vtx vertex
419 Double_t error2vtx = vtx->RotatedCovMatrixXX(phi,theta);
420
421 return TMath::Sqrt(error2+error2vtx);
422}
423
424//______________________________________________________________________________
425Double_t AliAODVertex::ErrorDistanceXYToVertex(AliAODVertex *vtx) const
426{
427 // error on the distance in XY to another AliAODVertex
428
429 Double_t phi,theta;
430 PhiAndThetaToVertex(vtx,phi,theta);
431 // error2 due to this vertex
432 Double_t error2 = RotatedCovMatrixXX(phi);
433 // error2 due to vtx vertex
434 Double_t error2vtx = vtx->RotatedCovMatrixXX(phi);
435
436 return TMath::Sqrt(error2+error2vtx);
437}
438
439//______________________________________________________________________________
440template <class T, class P>
441void AliAODVertex::PhiAndThetaToVertex(AliAODVertex *vtx, P &phi, T &theta) const
442{
443 // rotation angles around z-axis (phi) and around new y-axis (theta)
444 // with which vtx is seen (used by RotatedCovMatrix... methods)
445
b1a9edc8 446 phi = TMath::Pi()+TMath::ATan2(-vtx->GetY()+GetY(),-vtx->GetX()+GetX());
df9db588 447 Double_t vtxxphi = vtx->GetX()*TMath::Cos(phi)+vtx->GetY()*TMath::Sin(phi);
448 Double_t xphi = GetX()*TMath::Cos(phi)+GetY()*TMath::Sin(phi);
449 theta = TMath::ATan2(vtx->GetZ()-GetZ(),vtxxphi-xphi);
450}
451
452//______________________________________________________________________________
453void AliAODVertex::PrintIndices() const
454{
455 // Print indices of particles originating form this vertex
456
457 TRefArrayIter iter(&fDaughters);
458 while (TObject *daugh = iter.Next()) {
459 printf("Particle %p originates from this vertex.\n", daugh);
460 }
461}
462
463//______________________________________________________________________________
464void AliAODVertex::Print(Option_t* /*option*/) const
465{
466 // Print information of all data members
467
468 printf("Vertex position:\n");
469 printf(" x = %f\n", fPosition[0]);
470 printf(" y = %f\n", fPosition[1]);
471 printf(" z = %f\n", fPosition[2]);
472 printf(" parent particle: %p\n", fParent.GetObject());
473 printf(" origin of %d particles\n", fDaughters.GetEntriesFast());
474 printf(" vertex type %d\n", fType);
475
476 /*
477 if (fCovMatrix) {
478 printf("Covariance matrix:\n");
479 printf(" %12.10f %12.10f %12.10f\n %12.10f %12.10f %12.10f\n %12.10f %12.10f %12.10f\n",
480 fCovMatrix[0],
481 fCovMatrix[1],
482 fCovMatrix[3],
483 fCovMatrix[1],
484 fCovMatrix[2],
485 fCovMatrix[4],
486 fCovMatrix[3],
487 fCovMatrix[4],
488 fCovMatrix[5]);
489 } */
0657f082 490 printf(" Chi^2/NDF = %f\n", fChi2perNDF);
df9db588 491}
492