Updated code/macros to be compliant with the current HEAD (from Y. Belikov)

[u/mrichter/AliRoot.git] / ITS / AliV0vertex.cxx

diff --git a/ITS/AliV0vertex.cxx b/ITS/AliV0vertex.cxx
index ca305e32c692878a60a92be6d17d1100a900ffd2..d675175839d082c8ae96caa7e3d09b709c54fe57 100644 (file)
--- a/ITS/AliV0vertex.cxx
+++ b/ITS/AliV0vertex.cxx
@@ -85,35 +85,96 @@ AliV0vertex::AliV0vertex(const AliITStrackV2 &n, const AliITStrackV2 &p) {
 
   fChi2=7.;   
 }
 
   fChi2=7.;   
 }

-
-void AliV0vertex::ChangeMassHypothesis(Int_t code) {
+/*
+Double_t AliV0vertex::ChangeMassHypothesis(Int_t code) {

   //--------------------------------------------------------------------
   // This function changes the mass hypothesis for this V0
   //--------------------------------------------------------------------
   // This function changes the mass hypothesis for this V0

+  // and returns the "kinematical quality" of this hypothesis 

   //--------------------------------------------------------------------
   //--------------------------------------------------------------------

-  Double_t nmass, pmass;
+  Double_t nmass=0.13957, pmass=0.13957, mass=0.49767, des=0;
+
+  fPdgCode=code;

 
   switch (code) {
   case kLambda0:
 
   switch (code) {
   case kLambda0:

-    nmass=0.13957; pmass=0.93827; break;
+    nmass=0.13957; pmass=0.93827; mass=1.1157; des=0.9437-0.1723; break;

   case kLambda0Bar:
   case kLambda0Bar:

-    pmass=0.13957; nmass=0.93827; break;
+    pmass=0.13957; nmass=0.93827; mass=1.1157; des=0.1723-0.9437; break;

   case kK0Short: 
   case kK0Short: 

-    nmass=0.13957; pmass=0.13957; break;
+    break;

   default:
     cerr<<"AliV0vertex::ChangeMassHypothesis: ";
     cerr<<"invalide PDG code ! Assuming K0s...\n";
   default:
     cerr<<"AliV0vertex::ChangeMassHypothesis: ";
     cerr<<"invalide PDG code ! Assuming K0s...\n";

-    nmass=0.13957; pmass=0.13957; break;
+    fPdgCode=kK0Short;
+    break;

   }
 
   }
 

-  Double_t px1=fNmom[0], py1=fNmom[1], pz1=fNmom[2]; 
-  Double_t px2=fPmom[0], py2=fPmom[1], pz2=fPmom[2];
+  Double_t pxn=fNmom[0], pyn=fNmom[1], pzn=fNmom[2]; 
+  Double_t pxp=fPmom[0], pyp=fPmom[1], pzp=fPmom[2];

 
 

-  Double_t e1=TMath::Sqrt(nmass*nmass + px1*px1 + py1*py1 + pz1*pz1);
-  Double_t e2=TMath::Sqrt(pmass*pmass + px2*px2 + py2*py2 + pz2*pz2);
-  fEffMass=TMath::Sqrt((e1+e2)*(e1+e2)-
-    (px1+px2)*(px1+px2)-(py1+py2)*(py1+py2)-(pz1+pz2)*(pz1+pz2));
+  Double_t en=TMath::Sqrt(nmass*nmass + pxn*pxn + pyn*pyn + pzn*pzn);
+  Double_t ep=TMath::Sqrt(pmass*pmass + pxp*pxp + pyp*pyp + pzp*pzp);
+  Double_t pxl=pxn+pxp, pyl=pyn+pyp, pzl=pzn+pzp;
+  Double_t pl=TMath::Sqrt(pxl*pxl + pyl*pyl + pzl*pzl);
+
+  fEffMass=TMath::Sqrt((en+ep)*(en+ep)-pl*pl);
+
+  Double_t gamma=(en+ep)/mass, betagamma=pl/mass;
+  Double_t pln=(pxn*pxl + pyn*pyl + pzn*pzl)/pl;
+  Double_t plp=(pxp*pxl + pyp*pyl + pzp*pzl)/pl;
+  Double_t plps=gamma*plp - betagamma*ep;
+
+  Double_t diff=2*gamma*plps + betagamma*des;
+
+  return (plp-pln-diff);

   
   

+}
+*/
+Double_t AliV0vertex::ChangeMassHypothesis(Int_t code) {
+  //--------------------------------------------------------------------
+  // This function changes the mass hypothesis for this V0
+  // and returns the "kinematical quality" of this hypothesis 
+  //--------------------------------------------------------------------
+  Double_t nmass=0.13957, pmass=0.13957, mass=0.49767, ps=0.206;
+

   fPdgCode=code;
   fPdgCode=code;

+
+  switch (code) {
+  case kLambda0:
+    nmass=0.13957; pmass=0.93827; mass=1.1157; ps=0.101; break;
+  case kLambda0Bar:
+    pmass=0.13957; nmass=0.93827; mass=1.1157; ps=0.101; break;
+  case kK0Short: 
+    break;
+  default:
+    cerr<<"AliV0vertex::ChangeMassHypothesis: ";
+    cerr<<"invalide PDG code ! Assuming K0s...\n";
+    fPdgCode=kK0Short;
+    break;
+  }
+
+  Double_t pxn=fNmom[0], pyn=fNmom[1], pzn=fNmom[2]; 
+  Double_t pxp=fPmom[0], pyp=fPmom[1], pzp=fPmom[2];
+
+  Double_t en=TMath::Sqrt(nmass*nmass + pxn*pxn + pyn*pyn + pzn*pzn);
+  Double_t ep=TMath::Sqrt(pmass*pmass + pxp*pxp + pyp*pyp + pzp*pzp);
+  Double_t pxl=pxn+pxp, pyl=pyn+pyp, pzl=pzn+pzp;
+  Double_t pl=TMath::Sqrt(pxl*pxl + pyl*pyl + pzl*pzl);
+
+  fEffMass=TMath::Sqrt((en+ep)*(en+ep)-pl*pl);
+
+  Double_t beta=pl/(en+ep);
+  Double_t pln=(pxn*pxl + pyn*pyl + pzn*pzl)/pl;
+  Double_t plp=(pxp*pxl + pyp*pyl + pzp*pzl)/pl;
+
+  Double_t pt2=pxp*pxp + pyp*pyp + pzp*pzp - plp*plp;
+
+  Double_t a=(plp-pln)/(plp+pln);
+  a -= (pmass*pmass-nmass*nmass)/(mass*mass);
+  a = 0.25*beta*beta*mass*mass*a*a + pt2;
+
+  return (a - ps*ps);
+  

 }
 
 void AliV0vertex::GetPxPyPz(Double_t &px, Double_t &py, Double_t &pz) const {
 }
 
 void AliV0vertex::GetPxPyPz(Double_t &px, Double_t &py, Double_t &pz) const {