[u/mrichter/AliRoot.git] / STEER / AliAODVertex.cxx

/**************************************************************************
 * Copyright(c) 1998-2007, ALICE Experiment at CERN, All rights reserved. *
 *                                                                        *
 * Author: The ALICE Off-line Project.                                    *
 * Contributors are mentioned in the code where appropriate.              *
 *                                                                        *
 * Permission to use, copy, modify and distribute this software and its   *
 * documentation strictly for non-commercial purposes is hereby granted   *
 * without fee, provided that the above copyright notice appears in all   *
 * copies and that both the copyright notice and this permission notice   *
 * appear in the supporting documentation. The authors make no claims     *
 * about the suitability of this software for any purpose. It is          *
 * provided "as is" without express or implied warranty.                  *
 **************************************************************************/

/* $Id$ */

//-------------------------------------------------------------------------
//     AOD track base class
//     Base class for Analysis Object Data
//     Generic version
//     Author: Markus Oldenburg, CERN
//-------------------------------------------------------------------------

#include "AliAODVertex.h"

#include "AliAODTrack.h"

ClassImp(AliAODVertex)

//______________________________________________________________________________
AliAODVertex::AliAODVertex() : 
  TObject(),
  fChi2perNDF(-999.),
  fID(-1),
  fType(kUndef),
  fCovMatrix(NULL),
  fParent(0x0),
  fDaughters()
  {
  // default constructor

  fPosition[0] = fPosition[1] = fPosition[2] = -999.;
}

//______________________________________________________________________________
AliAODVertex::AliAODVertex(const Double_t position[3], 
			   const Double_t covMatrix[6],
			   Double_t  chi2perNDF,
			   TObject  *parent,
			   Short_t id,
			   Char_t vtype) :
  TObject(),
  fChi2perNDF(chi2perNDF),
  fID(id),
  fType(vtype),
  fCovMatrix(NULL),
  fParent(parent),
  fDaughters()
{
  // constructor

  SetPosition(position);
  if (covMatrix) SetCovMatrix(covMatrix);
}

//______________________________________________________________________________
AliAODVertex::AliAODVertex(const Float_t position[3], 
			   const Float_t  covMatrix[6],
			   Double_t  chi2perNDF,
			   TObject  *parent,
			   Short_t id,
			   Char_t vtype) :

  TObject(),
  fChi2perNDF(chi2perNDF),
  fID(id),
  fType(vtype),
  fCovMatrix(NULL),
  fParent(parent),
  fDaughters()
{
  // constructor

  SetPosition(position);
  if (covMatrix) SetCovMatrix(covMatrix);
}

//______________________________________________________________________________
AliAODVertex::AliAODVertex(const Double_t position[3], 
			   Double_t  chi2perNDF,
			   Char_t vtype) :
  TObject(),
  fChi2perNDF(chi2perNDF),
  fID(-1),
  fType(vtype),
  fCovMatrix(NULL),
  fParent(0x0),
  fDaughters()
{
  // constructor without covariance matrix

  SetPosition(position);  
}

//______________________________________________________________________________
AliAODVertex::AliAODVertex(const Float_t position[3], 
			   Double_t  chi2perNDF,
			   Char_t vtype) :
  TObject(),
  fChi2perNDF(chi2perNDF),
  fID(-1),
  fType(vtype),
  fCovMatrix(NULL),
  fParent(0x0),
  fDaughters()
{
  // constructor without covariance matrix

  SetPosition(position);  
}

//______________________________________________________________________________
AliAODVertex::~AliAODVertex() 
{
  // Destructor

  delete fCovMatrix;
}

//______________________________________________________________________________
AliAODVertex::AliAODVertex(const AliAODVertex& vtx) :
  TObject(vtx),
  fChi2perNDF(vtx.fChi2perNDF),
  fID(vtx.fID),
  fType(vtx.fType),
  fCovMatrix(NULL),
  fParent(vtx.fParent),
  fDaughters(vtx.fDaughters)
{
  // Copy constructor.
  
  for (int i = 0; i < 3; i++) 
    fPosition[i] = vtx.fPosition[i];

  if (vtx.fCovMatrix) fCovMatrix=new AliAODRedCov<3>(*vtx.fCovMatrix);
}

//______________________________________________________________________________
AliAODVertex& AliAODVertex::operator=(const AliAODVertex& vtx) 
{
  // Assignment operator
  if (this != &vtx) {

    // name and type
    TObject::operator=(vtx);

    //momentum
    for (int i = 0; i < 3; i++) 
      fPosition[i] = vtx.fPosition[i];
    
    fChi2perNDF = vtx.fChi2perNDF;
    fID = vtx.fID;
    fType = vtx.fType;

    //covariance matrix
    delete fCovMatrix;
    fCovMatrix = NULL;   
    if (vtx.fCovMatrix) fCovMatrix=new AliAODRedCov<3>(*vtx.fCovMatrix);
    
    //other stuff
    fParent = vtx.fParent;
    fDaughters = vtx.fDaughters;
  }
  
  return *this;
}

//______________________________________________________________________________
template <class T> void AliAODVertex::GetSigmaXYZ(T sigma[3]) const
{
  // Return errors on vertex position in thrust frame
  
  if(fCovMatrix) {
    sigma[0]=fCovMatrix[3]; //GetCovXZ
    sigma[1]=fCovMatrix[4]; //GetCovYZ
    sigma[2]=fCovMatrix[5]; //GetCovZZ
  } else 
    sigma[0]=sigma[1]=sigma[2]=-999.;

  /*
  for (int i = 0, j = 6; i < 3; i++) {
    j -= i+1;
    sigma[2-i] = fCovMatrix ? TMath::Sqrt(fCovMatrix[j]) : -999.;
  }
  */
}

//______________________________________________________________________________
Int_t AliAODVertex::GetNContributors() const 
{
  // Returns the number of tracks used to fit this vertex.
  
  Int_t cont = 0;

  for (Int_t iDaug = 0; iDaug < GetNDaughters(); iDaug++) {
    if (((AliAODTrack*)fDaughters.At(iDaug))->GetUsedForVtxFit()) cont++;
  }

  return cont;
}

//______________________________________________________________________________
Bool_t AliAODVertex::HasDaughter(TObject *daughter) const 
{
  // Checks if the given daughter (particle) is part of this vertex.

  TRefArrayIter iter(&fDaughters);
  while (TObject *daugh = iter.Next()) {
    if (daugh == daughter) return kTRUE;
  }
  return kFALSE;
}

//______________________________________________________________________________
Double_t AliAODVertex::RotatedCovMatrixXX(Double_t phi, Double_t theta) const
{
  // XX term of covariance matrix after rotation by phi around z-axis
  // and, then, by theta around new y-axis

  if (!fCovMatrix) {
    //AliFatal("Covariance matrix not set");
    return -999.;
  }

  Double_t covMatrix[6];

  GetCovMatrix(covMatrix);

  Double_t cp = TMath::Cos(phi);
  Double_t sp = TMath::Sin(phi);
  Double_t ct = TMath::Cos(theta);
  Double_t st = TMath::Sin(theta);
  return
     covMatrix[0]*cp*cp*ct*ct  // GetCovXX
    +covMatrix[1]*2.*cp*sp*ct*ct  // GetCovXY
    +covMatrix[3]*2.*cp*ct*st  // GetCovXZ
    +covMatrix[2]*sp*sp*ct*ct  // GetCovYY
    +covMatrix[4]*2.*sp*ct*st  // GetCovYZ
    +covMatrix[5]*st*st;  // GetCovZZ
}

//______________________________________________________________________________
Double_t AliAODVertex::RotatedCovMatrixXY(Double_t phi, Double_t theta) const
{
  // XY term of covariance matrix after rotation by phi around z-axis
  // and, then, by theta around new y-axis

  if (!fCovMatrix) {
    //AliFatal("Covariance matrix not set");
    return -999.;
  }

  Double_t covMatrix[6];

  GetCovMatrix(covMatrix);

  Double_t cp = TMath::Cos(phi);
  Double_t sp = TMath::Sin(phi);
  Double_t ct = TMath::Cos(theta);
  Double_t st = TMath::Sin(theta);
  return 
    -covMatrix[0]*cp*sp*ct  // GetCovXX
    +covMatrix[1]*ct*(cp*cp-sp*sp)  // GetCovXY
    -covMatrix[3]*sp*st  // GetCovXZ
    +covMatrix[2]*cp*sp*ct  // GetCovYY
    +covMatrix[4]*cp*st;  // GetCovYZ
}

//______________________________________________________________________________
Double_t AliAODVertex::RotatedCovMatrixXZ(Double_t phi, Double_t theta) const
{
  // XZ term of covariance matrix after rotation by phi around z-axis
  // and, then, by theta around new y-axis

  if (!fCovMatrix) {
    //AliFatal("Covariance matrix not set");
    return -999.;
  }

  Double_t covMatrix[6];

  GetCovMatrix(covMatrix);

  Double_t cp = TMath::Cos(phi);
  Double_t sp = TMath::Sin(phi);
  Double_t ct = TMath::Cos(theta);
  Double_t st = TMath::Sin(theta);
  return 
    -covMatrix[0]*cp*cp*ct*st  // GetCovXX
    -covMatrix[1]*2.*cp*sp*ct*st  // GetCovXY
    +covMatrix[3]*cp*(ct*ct-st*st)  // GetCovXZ
    -covMatrix[2]*sp*sp*ct*st  // GetCovYY
    +covMatrix[4]*sp*(ct*ct-st*st)  // GetCovYZ
    +covMatrix[5]*ct*st;  // GetCovZZ
}

//______________________________________________________________________________
Double_t AliAODVertex::RotatedCovMatrixYY(Double_t phi) const
{
  // YY term of covariance matrix after rotation by phi around z-axis
  // and, then, by theta around new y-axis

  if (!fCovMatrix) {
    //AliFatal("Covariance matrix not set");
    return -999.;
  }

  Double_t covMatrix[6];

  GetCovMatrix(covMatrix);

  Double_t cp = TMath::Cos(phi);
  Double_t sp = TMath::Sin(phi);
  return
     covMatrix[0]*sp*sp  // GetCovXX
    -covMatrix[1]*2.*cp*sp  // GetCovXY
    +covMatrix[2]*cp*cp;  // GetCovYY
}

//______________________________________________________________________________
Double_t AliAODVertex::RotatedCovMatrixYZ(Double_t phi, Double_t theta) const
{
  // YZ term of covariance matrix after rotation by phi around z-axis
  // and, then, by theta around new y-axis

  if (!fCovMatrix) {
    //AliFatal("Covariance matrix not set");
    return -999.;
  }

  Double_t covMatrix[6];

  GetCovMatrix(covMatrix);

  Double_t cp = TMath::Cos(phi);
  Double_t sp = TMath::Sin(phi);
  Double_t ct = TMath::Cos(theta);
  Double_t st = TMath::Sin(theta);
  return 
     covMatrix[0]*cp*sp*st  // GetCovXX
    +covMatrix[1]*st*(sp*sp-cp*cp)  // GetCovXY
    -covMatrix[3]*sp*ct  // GetCovXZ
    -covMatrix[2]*cp*sp*st  // GetCovYY
    +covMatrix[4]*cp*ct;  // GetCovYZ
}

//______________________________________________________________________________
Double_t AliAODVertex::RotatedCovMatrixZZ(Double_t phi, Double_t theta) const
{
  // ZZ term of covariance matrix after rotation by phi around z-axis
  // and, then, by theta around new y-axis

  if (!fCovMatrix) {
    //AliFatal("Covariance matrix not set");
    return -999.;
  }

  Double_t covMatrix[6];

  GetCovMatrix(covMatrix);

  Double_t cp = TMath::Cos(phi);
  Double_t sp = TMath::Sin(phi);
  Double_t ct = TMath::Cos(theta);
  Double_t st = TMath::Sin(theta);
  return
     covMatrix[0]*cp*cp*st*st  // GetCovXX
    +covMatrix[1]*2.*cp*sp*st*st  // GetCovXY
    -covMatrix[3]*2.*cp*ct*st  // GetCovXZ
    +covMatrix[2]*sp*sp*st*st  // GetCovYY
    -covMatrix[4]*2.*sp*sp*ct*st  // GetCovYZ
    +covMatrix[5]*ct*ct;  // GetCovZZ
}

//______________________________________________________________________________
Double_t AliAODVertex::DistanceToVertex(AliAODVertex *vtx) const
{
  // distance in 3D to another AliAODVertex

  Double_t dx = GetX()-vtx->GetX();
  Double_t dy = GetY()-vtx->GetY();
  Double_t dz = GetZ()-vtx->GetZ();

  return TMath::Sqrt(dx*dx+dy*dy+dz*dz);
}

//______________________________________________________________________________
Double_t AliAODVertex::DistanceXYToVertex(AliAODVertex *vtx) const
{
  // distance in XY to another AliAODVertex

  Double_t dx = GetX()-vtx->GetX();
  Double_t dy = GetY()-vtx->GetY();

  return TMath::Sqrt(dx*dx+dy*dy);
}

//______________________________________________________________________________
Double_t AliAODVertex::ErrorDistanceToVertex(AliAODVertex *vtx) const
{
  // error on the distance in 3D to another AliAODVertex

  Double_t phi,theta;
  PhiAndThetaToVertex(vtx,phi,theta);
  // error2 due to this vertex
  Double_t error2 = RotatedCovMatrixXX(phi,theta);
  // error2 due to vtx vertex
  Double_t error2vtx = vtx->RotatedCovMatrixXX(phi,theta);

  return TMath::Sqrt(error2+error2vtx);
}

//______________________________________________________________________________
Double_t AliAODVertex::ErrorDistanceXYToVertex(AliAODVertex *vtx) const
{
  // error on the distance in XY to another AliAODVertex

  Double_t phi,theta;
  PhiAndThetaToVertex(vtx,phi,theta);
  // error2 due to this vertex
  Double_t error2 = RotatedCovMatrixXX(phi);
  // error2 due to vtx vertex
  Double_t error2vtx = vtx->RotatedCovMatrixXX(phi);

  return TMath::Sqrt(error2+error2vtx);
}

//______________________________________________________________________________
template <class T, class P> 
void AliAODVertex::PhiAndThetaToVertex(AliAODVertex *vtx, P &phi, T &theta) const
{
  // rotation angles around z-axis (phi) and around new y-axis (theta)
  // with which vtx is seen (used by RotatedCovMatrix... methods)

  phi = TMath::Pi()+TMath::ATan2(-vtx->GetY()+GetY(),-vtx->GetX()+GetX());
  Double_t vtxxphi = vtx->GetX()*TMath::Cos(phi)+vtx->GetY()*TMath::Sin(phi);
  Double_t xphi = GetX()*TMath::Cos(phi)+GetY()*TMath::Sin(phi);
  theta = TMath::ATan2(vtx->GetZ()-GetZ(),vtxxphi-xphi);
}

//______________________________________________________________________________
void AliAODVertex::PrintIndices() const 
{
  // Print indices of particles originating form this vertex

  TRefArrayIter iter(&fDaughters);
  while (TObject *daugh = iter.Next()) {
    printf("Particle %p originates from this vertex.\n", daugh);
  }
}

//______________________________________________________________________________
void AliAODVertex::Print(Option_t* /*option*/) const 
{
  // Print information of all data members

  printf("Vertex position:\n");
  printf("     x = %f\n", fPosition[0]);
  printf("     y = %f\n", fPosition[1]);
  printf("     z = %f\n", fPosition[2]);
  printf(" parent particle: %p\n", fParent.GetObject());
  printf(" origin of %d particles\n", fDaughters.GetEntriesFast());
  printf(" vertex type %d\n", fType);
  
  /*
  if (fCovMatrix) {
    printf("Covariance matrix:\n");
    printf(" %12.10f  %12.10f  %12.10f\n %12.10f  %12.10f  %12.10f\n %12.10f  %12.10f  %12.10f\n", 
	   fCovMatrix[0],
	   fCovMatrix[1],
	   fCovMatrix[3],
	   fCovMatrix[1],
	   fCovMatrix[2],
	   fCovMatrix[4],
	   fCovMatrix[3],
	   fCovMatrix[4],
	   fCovMatrix[5]); 
	   } */
  printf(" Chi^2/NDF = %f\n", fChi2perNDF);
}
Commit	Line	Data
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"
0657f082	28
df9db588	29	ClassImp(AliAODVertex)
	30
	31	//______________________________________________________________________________
	32	AliAODVertex::AliAODVertex() :
	33	TObject(),
0657f082	34	fChi2perNDF(-999.),
02153d58	35	fID(-1),
9333290e	36	fType(kUndef),
9f7c531f	37	fCovMatrix(NULL),
df9db588	38	fParent(0x0),
9333290e	39	fDaughters()
9333290e	40	{
df9db588	41	// default constructor
	42
	43	fPosition[0] = fPosition[1] = fPosition[2] = -999.;
	44	}
	45
	46	//______________________________________________________________________________
	47	AliAODVertex::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) :
df9db588	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	//______________________________________________________________________________
	68	AliAODVertex::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	//______________________________________________________________________________
	90	AliAODVertex::AliAODVertex(const Double_t position[3],
0657f082	91	Double_t chi2perNDF,
df9db588	92	Char_t vtype) :
df9db588	93	TObject(),
0657f082	94	fChi2perNDF(chi2perNDF),
02153d58	95	fID(-1),
9333290e	96	fType(vtype),
9f7c531f	97	fCovMatrix(NULL),
df9db588	98	fParent(0x0),
9333290e	99	fDaughters()
df9db588	100	{
	101	// constructor without covariance matrix
	102
	103	SetPosition(position);
	104	}
	105
	106	//______________________________________________________________________________
	107	AliAODVertex::AliAODVertex(const Float_t position[3],
0657f082	108	Double_t chi2perNDF,
df9db588	109	Char_t vtype) :
df9db588	110	TObject(),
0657f082	111	fChi2perNDF(chi2perNDF),
02153d58	112	fID(-1),
9333290e	113	fType(vtype),
9f7c531f	114	fCovMatrix(NULL),
df9db588	115	fParent(0x0),
9333290e	116	fDaughters()
df9db588	117	{
	118	// constructor without covariance matrix
	119
	120	SetPosition(position);
	121	}
	122
	123	//______________________________________________________________________________
	124	AliAODVertex::~AliAODVertex()
	125	{
	126	// Destructor
	127
	128	delete fCovMatrix;
	129	}
	130
	131	//______________________________________________________________________________
	132	AliAODVertex::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	//______________________________________________________________________________
	150	AliAODVertex& 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
df9db588	167	delete fCovMatrix;
9f7c531f	168	fCovMatrix = NULL;
5d62ce04	169	if (vtx.fCovMatrix) fCovMatrix=new AliAODRedCov<3>(*vtx.fCovMatrix);
df9db588	170
df9db588	171	//other stuff
df9db588	172	fParent = vtx.fParent;
df9db588	173	fDaughters = vtx.fDaughters;
df9db588	174	}
	175
	176	return *this;
	177	}
	178
	179	//______________________________________________________________________________
	180	template <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
0657f082	199	//______________________________________________________________________________
	200	Int_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
df9db588	213	//______________________________________________________________________________
	214	Bool_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	//______________________________________________________________________________
	226	Double_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]cpcpctct // GetCovXX
	246	+covMatrix[1]2.cpspct*ct // GetCovXY
	247	+covMatrix[3]2.cpctst // GetCovXZ
	248	+covMatrix[2]spspctct // GetCovYY
	249	+covMatrix[4]2.spctst // GetCovYZ
	250	+covMatrix[5]stst; // GetCovZZ
	251	}
	252
	253	//______________________________________________________________________________
	254	Double_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]cpsp*ct // GetCovXX
	274	+covMatrix[1]ct(cpcp-spsp) // GetCovXY
	275	-covMatrix[3]spst // GetCovXZ
	276	+covMatrix[2]cpsp*ct // GetCovYY
277	+covMatrix[4]cpst; // GetCovYZ
278	}
279
280	//______________________________________________________________________________
281	Double_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]cpcpctst // GetCovXX
301	-covMatrix[1]2.cpspct*st // GetCovXY
302	+covMatrix[3]cp(ctct-stst) // GetCovXZ
303	-covMatrix[2]spspctst // GetCovYY
304	+covMatrix[4]sp(ctct-stst) // GetCovYZ
305	+covMatrix[5]ctst; // GetCovZZ
306	}
307
308	//______________________________________________________________________________
309	Double_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]spsp // GetCovXX
327	-covMatrix[1]2.cp*sp // GetCovXY
328	+covMatrix[2]cpcp; // GetCovYY
329	}
330
331	//______________________________________________________________________________
332	Double_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]cpsp*st // GetCovXX
352	+covMatrix[1]st(spsp-cpcp) // GetCovXY
353	-covMatrix[3]spct // GetCovXZ
354	-covMatrix[2]cpsp*st // GetCovYY
355	+covMatrix[4]cpct; // GetCovYZ
356	}
357
358	//______________________________________________________________________________
359	Double_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]cpcpstst // GetCovXX
379	+covMatrix[1]2.cpspst*st // GetCovXY
380	-covMatrix[3]2.cpctst // GetCovXZ
381	+covMatrix[2]spspstst // GetCovYY
382	-covMatrix[4]2.spspct*st // GetCovYZ
383	+covMatrix[5]ctct; // GetCovZZ
384	}
385
386	//______________________________________________________________________________
387	Double_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(dxdx+dydy+dz*dz);
396	}
397
398	//______________________________________________________________________________
399	Double_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(dxdx+dydy);
407	}
408
409	//______________________________________________________________________________
410	Double_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	//______________________________________________________________________________
425	Double_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	//______________________________________________________________________________
440	template <class T, class P>
441	void 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	//______________________________________________________________________________
	453	void 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	//______________________________________________________________________________
	464	void 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	}
df9db588	492