Changes suggested by Effective C++ (F.Carminati)
[u/mrichter/AliRoot.git] / STEER / AliESDv0.cxx
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
16 /* $Id$ */
17
18 //-------------------------------------------------------------------------
19 //               Implementation of the ESD V0 vertex class
20 //            This class is part of the Event Data Summary
21 //            set of classes and contains information about
22 //            V0 kind vertexes generated by a neutral particle
23 //     Origin: Iouri Belikov, IReS, Strasbourg, Jouri.Belikov@cern.ch
24 //-------------------------------------------------------------------------
25
26 #include <Riostream.h>
27 #include <TMath.h>
28 #include <TDatabasePDG.h>
29 #include <TPDGCode.h>
30 #include <TParticlePDG.h>
31
32 #include "AliLog.h"
33 #include "AliESDv0.h"
34
35 ClassImp(AliESDv0)
36
37 AliESDv0::AliESDv0() :
38   TObject(),
39   fPdgCode(kK0Short),
40   fEffMass(TDatabasePDG::Instance()->GetParticle(kK0Short)->Mass()),
41   fChi2(1.e+33),
42   fNidx(0),
43   fPidx(0)
44 {
45   //--------------------------------------------------------------------
46   // Default constructor  (K0s)
47   //--------------------------------------------------------------------
48   fPos[0]=fPos[1]=fPos[2]=0.;
49   fPosCov[0]=fPosCov[1]=fPosCov[2]=fPosCov[3]=fPosCov[4]=fPosCov[5]=0.;
50 }
51
52 Double_t AliESDv0::ChangeMassHypothesis(Int_t code) {
53   //--------------------------------------------------------------------
54   // This function changes the mass hypothesis for this V0
55   // and returns the "kinematical quality" of this hypothesis 
56   //--------------------------------------------------------------------
57   Double_t nmass=0.13957, pmass=0.13957, mass=0.49767, ps=0.206;
58
59   fPdgCode=code;
60
61   switch (code) {
62   case kLambda0:
63     nmass=0.13957; pmass=0.93827; mass=1.1157; ps=0.101; break;
64   case kLambda0Bar:
65     pmass=0.13957; nmass=0.93827; mass=1.1157; ps=0.101; break;
66   case kK0Short: 
67     break;
68   default:
69     AliError("invalide PDG code ! Assuming K0s...");
70     fPdgCode=kK0Short;
71     break;
72   }
73
74   Double_t pxn=fNmom[0], pyn=fNmom[1], pzn=fNmom[2]; 
75   Double_t pxp=fPmom[0], pyp=fPmom[1], pzp=fPmom[2];
76
77   Double_t en=TMath::Sqrt(nmass*nmass + pxn*pxn + pyn*pyn + pzn*pzn);
78   Double_t ep=TMath::Sqrt(pmass*pmass + pxp*pxp + pyp*pyp + pzp*pzp);
79   Double_t pxl=pxn+pxp, pyl=pyn+pyp, pzl=pzn+pzp;
80   Double_t pl=TMath::Sqrt(pxl*pxl + pyl*pyl + pzl*pzl);
81
82   fEffMass=TMath::Sqrt((en+ep)*(en+ep)-pl*pl);
83
84   Double_t beta=pl/(en+ep);
85   Double_t pln=(pxn*pxl + pyn*pyl + pzn*pzl)/pl;
86   Double_t plp=(pxp*pxl + pyp*pyl + pzp*pzl)/pl;
87
88   Double_t pt2=pxp*pxp + pyp*pyp + pzp*pzp - plp*plp;
89
90   Double_t a=(plp-pln)/(plp+pln);
91   a -= (pmass*pmass-nmass*nmass)/(mass*mass);
92   a = 0.25*beta*beta*mass*mass*a*a + pt2;
93
94   return (a - ps*ps);
95   
96 }
97
98 void AliESDv0::GetPxPyPz(Double_t &px, Double_t &py, Double_t &pz) const {
99   //--------------------------------------------------------------------
100   // This function returns V0's momentum (global)
101   //--------------------------------------------------------------------
102   px=fNmom[0]+fPmom[0]; 
103   py=fNmom[1]+fPmom[1]; 
104   pz=fNmom[2]+fPmom[2]; 
105 }
106
107 void AliESDv0::GetXYZ(Double_t &x, Double_t &y, Double_t &z) const {
108   //--------------------------------------------------------------------
109   // This function returns V0's position (global)
110   //--------------------------------------------------------------------
111   x=fPos[0]; 
112   y=fPos[1]; 
113   z=fPos[2]; 
114 }
115
116 Double_t AliESDv0::GetD(Double_t x0, Double_t y0, Double_t z0) const {
117   //--------------------------------------------------------------------
118   // This function returns V0's impact parameter
119   //--------------------------------------------------------------------
120   Double_t x=fPos[0],y=fPos[1],z=fPos[2];
121   Double_t px=fNmom[0]+fPmom[0];
122   Double_t py=fNmom[1]+fPmom[1];
123   Double_t pz=fNmom[2]+fPmom[2];
124
125   Double_t dx=(y0-y)*pz - (z0-z)*py; 
126   Double_t dy=(x0-x)*pz - (z0-z)*px;
127   Double_t dz=(x0-x)*py - (y0-y)*px;
128   Double_t d=TMath::Sqrt((dx*dx+dy*dy+dz*dz)/(px*px+py*py+pz*pz));
129   return d;
130 }