Response of the 4 towers for the 2 neutron calorimeters since,

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

diff --git a/STEER/AliPID.cxx b/STEER/AliPID.cxx
index 50bd0de35db9fb2e4c9f269465653fbc4b00cd98..4238ac1080ec918c90b856f523a506cc48749e05 100644 (file)
--- a/STEER/AliPID.cxx
+++ b/STEER/AliPID.cxx
@@ -158,6 +158,19 @@ AliPID::AliPID(const AliPID& pid) :
     fProbDensity[i] = pid.fProbDensity[i];
 }
 
+//_______________________________________________________________________
+void AliPID::SetProbabilities(const Double_t* probDensity, Bool_t charged) 
+{
+  //
+  // Set the probability densities
+  //
+  for (Int_t i = 0; i < kSPECIES; i++) 
+    fProbDensity[i] = probDensity[i];
+
+  for (Int_t i = kSPECIES; i < kSPECIESN; i++) 
+    fProbDensity[i] = ((charged) ? 0 : probDensity[i]);
+}
+
 //_______________________________________________________________________
 AliPID& AliPID::operator = (const AliPID& pid)
 {
@@ -179,7 +192,7 @@ void AliPID::Init()
   // Initialise only once... 
   if(!fgkParticleMass[0]) 
     for (Int_t i = 0; i < kSPECIESN; i++) 
-      fgkParticleMass[i] = M(fgkParticleCode[i]);
+      fgkParticleMass[i] = M(i);
 }
 
 //_____________________________________________________________________________