As part of the efforts of the combined PID subgroup we are continuing to develop...

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

diff --git a/STEER/AliAODPid.cxx b/STEER/AliAODPid.cxx
index b359af97478cd0063ff7ce05f8bfcf80080a35a7..1f462715e428c2f33be18af0cad63935e554699b 100644 (file)
--- a/STEER/AliAODPid.cxx
+++ b/STEER/AliAODPid.cxx
@@ -39,8 +39,12 @@ AliAODPid::AliAODPid():
     fHMPIDsignal(0)
 {
   // 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<kSPECIES; i++) fIntTime[i]   = 0; 
+    for(Int_t i=0; i<5; i++) fHMPIDprobs[i] = 0.;
+    for(Int_t i=0; i<3; i++) fEMCALPosition[i]    = 0.;
+    for(Int_t i=0; i<5; i++) fTOFpidResolution[i] = 0.;
+    for(Int_t i=0; i<6; i++) fTRDmomentum[i]      = 0.;
+    for(Int_t i=0; i<3; i++) fEMCALMomentum[i]    = 0.;
   
 }
 
@@ -69,11 +73,14 @@ 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<5; i++) fHMPIDprobs[i] = pid.fHMPIDprobs[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];
 }
 
 //______________________________________________________________________________
@@ -97,11 +104,13 @@ 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<5; i++) fHMPIDprobs[i] = pid.fHMPIDprobs[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;
@@ -142,3 +151,31 @@ 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];
+}
+
+//______________________________________________________________________________
+void AliAODPid::SetHMPIDprobs(Double_t hmpPid[5]) 
+{
+  //
+  // Set the HMPID PID probablities that are read from ESD
+  //  
+  for(Int_t i = 0; i < 5; i++ ) fHMPIDprobs[i] =  hmpPid[i];
+}
+//______________________________________________________________________________
+void AliAODPid::GetHMPIDprobs(Double_t *p) 
+{
+  //
+  // Set the HMPID PID probablities that are read from ESD
+  //  
+  for(Int_t i = 0; i < AliPID::kSPECIES; i++ ) p[i] =  fHMPIDprobs[i];
+}