from Attilio Tarantola:

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

diff --git a/ZDC/AliZDCDigitizer.h b/ZDC/AliZDCDigitizer.h
index 435a9920984c93596f9b2681224c7fb52deb4d12..23b392435e1a00480a4e616807efd4b62cfa7463 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)
@@ -46,12 +46,22 @@ public:
   AliZDCPedestals  *GetPedData() const; 
   
   void    SetSpectators2Track() {fSpectators2Track=kTRUE;}
+  void    SetBeamEnergy(Float_t beamEnergy) {fBeamEnergy = beamEnergy;}
+
+  // Added for p-A simulations
+  void    SetpAsystem() {fIspASystem=kTRUE;}
+
+  // Added for RELDIS
+  void    SetRELDISGenerator() {fIsRELDISgen=kTRUE;}
 
 private:
 
   AliZDCDigitizer(const AliZDCDigitizer&);
   AliZDCDigitizer& operator=(const AliZDCDigitizer&);
 
+  void    CalculatePMTGains();
+  void    ReadPMTGains();
+
   void    Fragmentation(Float_t impPar, Int_t specN, Int_t specP,
                         Int_t &freeSpecN, Int_t &freeSpecP) const;
   void    SpectatorSignal(Int_t SpecType, Int_t numEvents, 
@@ -70,8 +80,15 @@ private:
   AliZDCPedestals  *fPedData;     //! pedestal calibration data
   
   Bool_t  fSpectators2Track;    // should digitizer track spectators
-  Float_t fBeamEnergy;          // beam energy taken from GRP object
+  Float_t fBeamEnergy;          // beam energy
+  TString fBeamType;           // beam type
+  
+  // Added for p-A simulations
+  Bool_t  fIspASystem;         // true if collision system is p-A
+
+  // Added for RELDIS
+  Bool_t  fIsRELDISgen;        // treu if generator is RELDIS
        
-  ClassDef(AliZDCDigitizer, 12)     // digitizer for ZDC
+  ClassDef(AliZDCDigitizer, 14)     // digitizer for ZDC
 };    
 #endif