new functions, mc analysis can be performed also for pi0 and k0. eta of seed can...

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

diff --git a/STEER/AliAODPid.cxx b/STEER/AliAODPid.cxx
index c5eb933367c857998b80bcf2111df0d75b8f576a..41a5e2a4f65b5f8b7b4b0525170974de3ce070a8 100644 (file)
--- a/STEER/AliAODPid.cxx
+++ b/STEER/AliAODPid.cxx
@@ -31,6 +31,8 @@ ClassImp(AliAODPid)
 AliAODPid::AliAODPid():
     fITSsignal(0), 
     fTPCsignal(0),
+    fTPCsignalN(0),
+    fTPCmomentum(0),
     fTRDnSlices(0),
     fTRDslices(0x0),
     fTOFesdsignal(0),
@@ -39,6 +41,7 @@ AliAODPid::AliAODPid():
   // default constructor
     for(Int_t i=0; i<kSPECIES; i++) fIntTime[i]=0; 
     for(Int_t i=0; i<3; i++) fEMCALPosition[i] = 0.;
+    for(Int_t i=0; i<5; i++) fTOFpidResolution[i] = 0.;
   
 }
 
@@ -56,6 +59,8 @@ AliAODPid::AliAODPid(const AliAODPid& pid) :
   TObject(pid),
   fITSsignal(pid.fITSsignal), 
   fTPCsignal(pid.fTPCsignal),
+  fTPCsignalN(pid.fTPCsignalN),
+  fTPCmomentum(pid.fTPCmomentum),
   fTRDnSlices(pid.fTRDnSlices),
   fTRDslices(0x0),
   fTOFesdsignal(pid.fTOFesdsignal),
@@ -65,7 +70,13 @@ AliAODPid::AliAODPid(const AliAODPid& pid) :
     fTRDslices = new Double32_t[fTRDnSlices];
     for(Int_t i=0; i< fTRDnSlices; i++) fTRDslices[i]=pid.fTRDslices[i];
     for(Int_t i=0; i<kSPECIES; i++) fIntTime[i]=pid.fIntTime[i];
-    for(Int_t i=0; i<3; i++) fEMCALPosition[i]=pid.fEMCALPosition[i];
+    for(Int_t i=0; i<3; i++) {
+      fEMCALPosition[i]=pid.fEMCALPosition[i];
+      fEMCALMomentum[i]=pid.fEMCALMomentum[i];
+    }
+    for(Int_t i=0; i<6; i++) fTRDmomentum[i]=pid.fTRDmomentum[i];
+
+    for(Int_t i=0; i<5; i++) fTOFpidResolution[i]=pid.fTOFpidResolution[i];
 }
 
 //______________________________________________________________________________
@@ -89,13 +100,18 @@ AliAODPid& AliAODPid::operator=(const AliAODPid& pid)
     fTOFesdsignal=pid.fTOFesdsignal;
     fHMPIDsignal=pid.fHMPIDsignal;
     for(Int_t i=0; i<kSPECIES; i++) fIntTime[i]=pid.fIntTime[i];
-    for(Int_t i=0; i<3; i++) fEMCALPosition[i]=pid.fEMCALPosition[i];
+    for(Int_t i=0; i<6; i++) fTRDmomentum[i]=pid.fTRDmomentum[i];
+    for(Int_t i=0; i<3; i++) {
+      fEMCALPosition[i]=pid.fEMCALPosition[i];
+      fEMCALMomentum[i]=pid.fEMCALMomentum[i];
+    }
+    for (Int_t i=0; i<5; i++) fTOFpidResolution[i]=pid.fTOFpidResolution[i];
   }
 
   return *this;
 }
 //_______________________________________________________________________________
-void AliAODPid::GetIntegratedTimes(Double_t timeint[kSPECIES])
+void AliAODPid::GetIntegratedTimes(Double_t timeint[kSPECIES]) const
 {
  // Returns the array with integrated times for each particle hypothesis
 for(Int_t i=0; i<kSPECIES; i++) timeint[i]=fIntTime[i];
@@ -107,7 +123,7 @@ void AliAODPid::SetIntegratedTimes(Double_t timeint[kSPECIES])
 for(Int_t i=0; i<kSPECIES; i++) fIntTime[i]=timeint[i];
 }
 //_______________________________________________________________________________
-void AliAODPid::GetEMCALPosition(Double_t emcalpos[3])
+void AliAODPid::GetEMCALPosition(Double_t emcalpos[3]) const
 {
  // Returns the array with extrapolated track position at the EMCAL surface
   for(Int_t i=0; i<3; i++) emcalpos[i]=fEMCALPosition[i];
@@ -119,7 +135,7 @@ void AliAODPid::SetEMCALPosition(Double_t emcpos[3])
   for(Int_t i=0; i<3; i++) fEMCALPosition[i]=emcpos[i];
 }
 //_______________________________________________________________________________
-void AliAODPid::GetEMCALMomentum(Double_t emcalmom[3])
+void AliAODPid::GetEMCALMomentum(Double_t emcalmom[3]) const
 {
  // Returns the array with extrapolated track momentum at the EMCAL surface
   for(Int_t i=0; i<3; i++) emcalmom[i]=fEMCALMomentum[i];
@@ -130,3 +146,15 @@ void AliAODPid::SetEMCALMomentum(Double_t emcmom[3])
  // Sets the array with extrapolated track momentum at the EMCAL surface
   for(Int_t i=0; i<3; i++) fEMCALMomentum[i]=emcmom[i];
 }
+//______________________________________________________________________________
+void AliAODPid::SetTOFpidResolution(Double_t tofPIDres[5])
+{
+  for (Int_t i=0; i<5; i++) fTOFpidResolution[i]=tofPIDres[i];
+
+}
+//______________________________________________________________________________
+void AliAODPid::GetTOFpidResolution(Double_t tofRes[5]) const
+{
+  for (Int_t i=0; i<5; i++) tofRes[i]=fTOFpidResolution[i];
+}
+