]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - STEER/AliESDv0.cxx
Fix fixed-string length bug
[u/mrichter/AliRoot.git] / STEER / AliESDv0.cxx
index d72d5bd8f7bf2721a14c842c319d5a9059306cb5..62b6c677f16de9e9c661d33129d2123c4e43ef9d 100644 (file)
 
 ClassImp(AliESDv0)
 
-AliESDV0Params  AliESDv0::fgkParams;
+const AliESDV0Params  AliESDv0::fgkParams;
 
 AliESDv0::AliESDv0() :
   TObject(),
-  fOnFlyStatus(kFALSE),
-  fPdgCode(kK0Short),
-  fEffMass(TDatabasePDG::Instance()->GetParticle(kK0Short)->Mass()),
-  fDcaV0Daughters(0),
-  fPointAngle(0),
-  fChi2V0(31.),
-  fNidx(-1),
   fParamN(),
-  fPidx(-1),
   fParamP(),
-  fRr(-1),
-  fStatus(0),
+  fEffMass(TDatabasePDG::Instance()->GetParticle(kK0Short)->Mass()),
+  fDcaV0Daughters(0),
+  fChi2V0(0.),
+  fRr(0),
   fDistSigma(0),
   fChi2Before(0),
-  fNBefore(0),
   fChi2After(0),
-  fNAfter(0),
   fPointAngleFi(0),
-  fPointAngleTh(0)
+  fPointAngleTh(0),
+  fPointAngle(0),
+  fPdgCode(kK0Short),
+  fNidx(0),
+  fPidx(0),
+  fStatus(0),
+  fNBefore(0),
+  fNAfter(0),
+  fOnFlyStatus(kFALSE)
 {
   //--------------------------------------------------------------------
   // Default constructor  (K0s)
@@ -83,25 +83,25 @@ AliESDv0::AliESDv0() :
 
 AliESDv0::AliESDv0(const AliESDv0& v0) :
   TObject(v0),
-  fOnFlyStatus(v0.fOnFlyStatus),
-  fPdgCode(v0.fPdgCode),
+  fParamN(v0.fParamN),
+  fParamP(v0.fParamP),
   fEffMass(v0.fEffMass),
   fDcaV0Daughters(v0.fDcaV0Daughters),
-  fPointAngle(v0.fPointAngle),
   fChi2V0(v0.fChi2V0),
-  fNidx(v0.fNidx),
-  fParamN(v0.fParamN),
-  fPidx(v0.fPidx),
-  fParamP(v0.fParamP),
   fRr(v0.fRr),
-  fStatus(v0.fStatus),
   fDistSigma(v0.fDistSigma),
   fChi2Before(v0.fChi2Before),
-  fNBefore(v0.fNBefore),
   fChi2After(v0.fChi2After),
-  fNAfter(v0.fNAfter),
   fPointAngleFi(v0.fPointAngleFi),
-  fPointAngleTh(v0.fPointAngleTh)
+  fPointAngleTh(v0.fPointAngleTh),
+  fPointAngle(v0.fPointAngle),
+  fPdgCode(v0.fPdgCode),
+  fNidx(v0.fNidx),
+  fPidx(v0.fPidx),
+  fStatus(v0.fStatus),
+  fNBefore(v0.fNBefore),
+  fNAfter(v0.fNAfter),
+  fOnFlyStatus(v0.fOnFlyStatus)
 {
   //--------------------------------------------------------------------
   // The copy constructor
@@ -129,28 +129,29 @@ AliESDv0::AliESDv0(const AliESDv0& v0) :
   for (Int_t i=0;i<4;i++){fCausality[i]=v0.fCausality[i];}
 }
 
+
 AliESDv0::AliESDv0(const AliExternalTrackParam &t1, Int_t i1,
                    const AliExternalTrackParam &t2, Int_t i2) :
   TObject(),
-  fOnFlyStatus(kFALSE),
-  fPdgCode(kK0Short),
+  fParamN(t1),
+  fParamP(t2),
   fEffMass(TDatabasePDG::Instance()->GetParticle(kK0Short)->Mass()),
   fDcaV0Daughters(0),
+  fChi2V0(0.),
+  fRr(0),
+  fDistSigma(0),
+  fChi2Before(0),
+  fChi2After(0),
+  fPointAngleFi(0),
+  fPointAngleTh(0),
   fPointAngle(0),
-  fChi2V0(31.),
+  fPdgCode(kK0Short),
   fNidx(i1),
-  fParamN(t1),
   fPidx(i2),
-  fParamP(t2),
-  fRr(-1),
   fStatus(0),
-  fDistSigma(0),
-  fChi2Before(0),
   fNBefore(0),
-  fChi2After(0),
   fNAfter(0),
-  fPointAngleFi(0),
-  fPointAngleTh(0)
+  fOnFlyStatus(kFALSE)
 {
   //--------------------------------------------------------------------
   // Main constructor  (K0s)
@@ -194,6 +195,71 @@ AliESDv0::AliESDv0(const AliExternalTrackParam &t1, Int_t i1,
   for (Int_t i=0;i<4;i++){fCausality[i]=0;}
 }
 
+AliESDv0& AliESDv0::operator=(const AliESDv0 &v0){
+
+
+  //--------------------------------------------------------------------
+  // The assingment operator
+  //--------------------------------------------------------------------
+
+  if(this==&v0)return *this;
+  TObject::operator=(v0);
+  fParamN  = v0.fParamN;
+  fParamP  = v0.fParamP;
+  fEffMass = v0.fEffMass;
+  fDcaV0Daughters = v0.fDcaV0Daughters;
+  fChi2V0 = v0.fChi2V0;
+  fRr = v0.fRr;
+  fDistSigma    = v0.fDistSigma;
+  fChi2Before   = v0.fChi2Before;
+  fChi2After    = v0.fChi2After;
+  fPointAngleFi = v0.fPointAngleFi;
+  fPointAngleTh = v0.fPointAngleTh;
+  fPointAngle   = v0.fPointAngle;
+  fPdgCode      = v0.fPdgCode;
+  fNidx         = v0.fNidx;
+  fPidx         = v0.fPidx;
+  fStatus       = v0.fStatus;
+  fNBefore      = v0.fNBefore;
+  fNAfter       = v0.fNAfter;
+  fOnFlyStatus  = v0.fOnFlyStatus;
+
+  for (int i=0; i<3; i++) {
+    fPos[i]  = v0.fPos[i];
+    fNmom[i] = v0.fNmom[i];
+    fPmom[i] = v0.fPmom[i];
+  }
+  for (int i=0; i<6; i++) {
+    fPosCov[i]  = v0.fPosCov[i];
+  }
+  for (Int_t i=0; i<2; i++) {
+    fNormDCAPrim[i]=v0.fNormDCAPrim[i];
+  }
+  for (Int_t i=0;i<6;i++){
+      fClusters[0][i]=v0.fClusters[0][i]; 
+      fClusters[1][i]=v0.fClusters[1][i];
+  }
+  for (Int_t i=0;i<3;i++){
+      fAngle[i]=v0.fAngle[i];
+  }
+  for (Int_t i=0;i<4;i++){fCausality[i]=v0.fCausality[i];}
+
+  return *this;
+}
+
+void AliESDv0::Copy(TObject& obj) const {
+
+   // this overwrites the virtual TOBject::Copy()
+  // to allow run time copying without casting
+  // in AliESDEvent
+
+  if(this==&obj)return;
+  AliESDv0 *robj = dynamic_cast<AliESDv0*>(&obj);
+  if(!robj)return; // not an aliesesv0
+  *robj = *this;
+}
+
+
 AliESDv0::~AliESDv0(){
   //--------------------------------------------------------------------
   // Empty destructor
@@ -317,6 +383,12 @@ Float_t AliESDv0::GetV0CosineOfPointingAngle(Double_t& refPointX, Double_t& refP
 
 // **** The following functions need to be revised
 
+void AliESDv0::GetPosCov(Double_t cov[6]) const {
+
+  for (Int_t i=0; i<6; ++i) cov[i] = fPosCov[i];
+
+}
+
 Double_t AliESDv0::GetSigmaY(){
   //
   // return sigmay in y  at vertex position  using covariance matrix 
@@ -544,3 +616,38 @@ void  AliESDv0::SetClusters(Int_t *clp, Int_t *clm)
   for (Int_t i=0;i<6;i++) fClusters[0][i] = clp[i]; 
   for (Int_t i=0;i<6;i++) fClusters[1][i] = clm[i]; 
 }
+
+Float_t AliESDv0::GetEffMass(UInt_t p1, UInt_t p2){
+  //
+  // calculate effective mass
+  //
+  const Float_t kpmass[5] = {5.10000000000000037e-04,1.05660000000000004e-01,1.39570000000000000e-01,
+                     4.93599999999999983e-01, 9.38270000000000048e-01};
+  if (p1>4) return -1;
+  if (p2>4) return -1;
+  Float_t mass1 = kpmass[p1]; 
+  Float_t mass2 = kpmass[p2];   
+  Double_t *m1 = fPmom;
+  Double_t *m2 = fNmom;
+  //
+  //if (fRP[p1]+fRM[p2]<fRP[p2]+fRM[p1]){
+  //  m1 = fPM;
+  //  m2 = fPP;
+  //}
+  //
+  Float_t e1    = TMath::Sqrt(mass1*mass1+
+                              m1[0]*m1[0]+
+                              m1[1]*m1[1]+
+                              m1[2]*m1[2]);
+  Float_t e2    = TMath::Sqrt(mass2*mass2+
+                              m2[0]*m2[0]+
+                              m2[1]*m2[1]+
+                              m2[2]*m2[2]);  
+  Float_t mass =  
+    (m2[0]+m1[0])*(m2[0]+m1[0])+
+    (m2[1]+m1[1])*(m2[1]+m1[1])+
+    (m2[2]+m1[2])*(m2[2]+m1[2]);
+  
+  mass = TMath::Sqrt((e1+e2)*(e1+e2)-mass);
+  return mass;
+}