Fixing warnings

[u/mrichter/AliRoot.git] / ZDC / AliZDCDigitizer.h

diff --git a/ZDC/AliZDCDigitizer.h b/ZDC/AliZDCDigitizer.h
index 435a9920984c93596f9b2681224c7fb52deb4d12..42bb9a8491165a4718addbb68e11884f60e432bc 100644 (file)
--- a/ZDC/AliZDCDigitizer.h
+++ b/ZDC/AliZDCDigitizer.h
@@ -14,7 +14,7 @@
 #include "AliCDBStorage.h"
 #include "AliZDCPedestals.h"
 
-class AliRunDigitizer;
+class AliDigitizationInput;
 
 class AliZDC;
 class AliZDCHit;
@@ -24,11 +24,11 @@ class AliZDCDigitizer: public AliDigitizer {
 
 public:
   AliZDCDigitizer();
-  AliZDCDigitizer(AliRunDigitizer* manager);
+  AliZDCDigitizer(AliDigitizationInput* digInput);
   virtual ~AliZDCDigitizer();
    
   virtual Bool_t Init();
-  virtual void Exec(Option_t* option=0);    
+  virtual void Digitize(Option_t* option=0);    
 
   //  PM gain
   void    SetPMGain(Int_t det, Int_t pmDet, Int_t pmGain)
@@ -47,6 +47,9 @@ public:
   
   void    SetSpectators2Track() {fSpectators2Track=kTRUE;}
 
+  // Added for p-A simulations
+  void    SetpAsystem() {fIspASystem=kTRUE;}
+
 private:
 
   AliZDCDigitizer(const AliZDCDigitizer&);
@@ -71,7 +74,10 @@ private:
   
   Bool_t  fSpectators2Track;    // should digitizer track spectators
   Float_t fBeamEnergy;          // beam energy taken from GRP object
+  
+  // Added for p-A simulations
+  Bool_t  fIspASystem;         // true if collision system is p-A
        
-  ClassDef(AliZDCDigitizer, 12)     // digitizer for ZDC
+  ClassDef(AliZDCDigitizer, 13)     // digitizer for ZDC
 };    
 #endif