Changes from Svein Lindal

[u/mrichter/AliRoot.git] / PHOS / AliPHOSPID.h

diff --git a/PHOS/AliPHOSPID.h b/PHOS/AliPHOSPID.h
index f1e53515382941fac505caff709aee1f0b2b3fc2..aceca482cd82d7b0b14202b8e210a868d3ada906 100644 (file)
--- a/PHOS/AliPHOSPID.h
+++ b/PHOS/AliPHOSPID.h
@@ -69,6 +69,9 @@ class AliPHOSPID : public TObject {
 
   void SetESD(AliESDEvent *esd) { fESD = esd; }
 
+  void SetEnergyCorrectionOn(Bool_t on=kTRUE) {fEnergyCorrectionOn = on;}
+  Bool_t GetEnergyCorrectionOn() const  {return fEnergyCorrectionOn;}
+
   virtual const char * Version() const = 0;
 
 protected:
@@ -81,10 +84,12 @@ protected:
 
   TClonesArray *fTrackSegments;     //!Array with found track segments
   TClonesArray *fRecParticles;      //!Array with reconstructed particles (PID)
-
+  
+  Bool_t   fEnergyCorrectionOn;     // Do energy correction in GetCalibratedEnergy()
+  
 private: 
 
-  ClassDef(AliPHOSPID,6)  // Particle Identifier algorithm (base class)
+  ClassDef(AliPHOSPID,7)  // Particle Identifier algorithm (base class)
 
 } ;