AOD includes added.

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

diff --git a/STEER/AliMultiplicity.cxx b/STEER/AliMultiplicity.cxx
index 14f32292974625921e50cad6bac47332d271afeb..1fa8f9a2e966310c6b7f9f7650bb164eba77498d 100644 (file)
--- a/STEER/AliMultiplicity.cxx
+++ b/STEER/AliMultiplicity.cxx
@@ -1,3 +1,4 @@
+#include <string.h>
 #include "AliMultiplicity.h"
 
 ClassImp(AliMultiplicity)
@@ -9,6 +10,7 @@ AliMultiplicity::AliMultiplicity():
   fTh(0),
   fPhi(0),
   fDeltPhi(0),
+  fLabels(0),
   fNsingle(0),
   fThsingle(0),
   fPhisingle(0)
@@ -18,12 +20,13 @@ AliMultiplicity::AliMultiplicity():
 }
 
 //______________________________________________________________________
-AliMultiplicity::AliMultiplicity(Int_t ntr, Float_t *t,  Float_t *ph, Float_t *df, Int_t ns, Float_t *ts, Float_t *ps):
+AliMultiplicity::AliMultiplicity(Int_t ntr, Float_t *t,  Float_t *ph, Float_t *df, Int_t *labels, Int_t ns, Float_t *ts, Float_t *ps):
   TObject(),
   fNtracks(ntr),
   fTh(0),
   fPhi(0),
   fDeltPhi(0),
+  fLabels(0),
   fNsingle(ns),
   fThsingle(0),
   fPhisingle(0)
@@ -33,10 +36,12 @@ AliMultiplicity::AliMultiplicity(Int_t ntr, Float_t *t,  Float_t *ph, Float_t *d
     fTh = new Float_t [ntr];
     fPhi = new Float_t [ntr];
     fDeltPhi = new Float_t [ntr];
+    fLabels = new Int_t[ntr];
     for(Int_t i=0;i<fNtracks;i++){
       fTh[i]=t[i];
       fPhi[i]=ph[i];
       fDeltPhi[i]=df[i];
+      fLabels[i] = labels[i];
     }
   }
   if(ns>0){
@@ -56,6 +61,7 @@ AliMultiplicity::AliMultiplicity(const AliMultiplicity& m):
   fTh(0),
   fPhi(0),
   fDeltPhi(0),
+  fLabels(0),
   fNsingle(m.fNsingle),
   fThsingle(0),
   fPhisingle(0)
@@ -75,6 +81,7 @@ AliMultiplicity &AliMultiplicity::operator=(const AliMultiplicity& m){
   if (fTh)delete [] fTh;
   if(fPhi)delete [] fPhi;
   if(fDeltPhi)delete [] fDeltPhi;
+  if(fLabels)delete [] fLabels;
   if(fThsingle)delete [] fThsingle;
   if(fPhisingle)delete [] fPhisingle;
   Duplicate(m);
@@ -90,11 +97,13 @@ void AliMultiplicity::Duplicate(const AliMultiplicity& m){
     fTh = new Float_t [fNtracks];
     fPhi = new Float_t [fNtracks];
     fDeltPhi = new Float_t [fNtracks];
+    fLabels = new Int_t[fNtracks];
   }
   else {
     fTh = 0;
     fPhi = 0;
     fDeltPhi = 0;
+    fLabels = 0;
   }
   fNsingle = m.fNsingle;
   if(fNsingle>0){
@@ -108,6 +117,7 @@ void AliMultiplicity::Duplicate(const AliMultiplicity& m){
   memcpy(fTh,m.fTh,fNtracks*sizeof(Float_t));
   memcpy(fPhi,m.fPhi,fNtracks*sizeof(Float_t));
   memcpy(fDeltPhi,m.fDeltPhi,fNtracks*sizeof(Float_t));
+  memcpy(fLabels,m.fLabels,fNtracks*sizeof(Int_t));
   memcpy(fThsingle,m.fThsingle,fNsingle*sizeof(Float_t));
   memcpy(fPhisingle,m.fPhisingle,fNsingle*sizeof(Float_t));
 }
@@ -118,6 +128,7 @@ AliMultiplicity::~AliMultiplicity(){
   if (fTh)delete [] fTh;
   if(fPhi)delete [] fPhi;
   if(fDeltPhi)delete [] fDeltPhi;
+  if(fLabels)delete [] fLabels;
   if(fThsingle)delete [] fThsingle;
   if(fPhisingle)delete [] fPhisingle;
 }