MC calculation of Q vector for AMPT

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

diff --git a/ZDC/AliZDC.h b/ZDC/AliZDC.h
index 25edd9a4df38d30bc87d5a47104cdc63478aee7f..580723fc338975ff7960fe65d5aee3f3842d9b74 100644 (file)
--- a/ZDC/AliZDC.h
+++ b/ZDC/AliZDC.h
@@ -13,6 +13,7 @@
 
 #include "AliDetector.h"
 #include "AliZDCTrigger.h"
+#include "AliZDCChMap.h"
 
 class AliZDCPedestals;
 class AliZDCEnCalib;
@@ -35,7 +36,7 @@ public:
   virtual void  SetTreeAddress();
   virtual void  MakeBranch(Option_t* opt);
   virtual void  Hits2SDigits();
-  virtual AliDigitizer* CreateDigitizer(AliRunDigitizer* manager) const;
+  virtual AliDigitizer* CreateDigitizer(AliDigitizationInput* digInput) const;
   virtual void  Digits2Raw();
   virtual Bool_t Raw2SDigits(AliRawReader* rawReader);
   Int_t   Pedestal(Int_t Detector, Int_t Quadrant, Int_t Res) const;
@@ -45,13 +46,36 @@ public:
   // Switching off the shower development in ZDCs
   void  NoShower(){fNoShower=1;}
   void  Shower()  {fNoShower=0;}
-
+  
+  virtual void SetVCollSideCAperture(Float_t /*aperture*/) {}
+  virtual void SetVCollSideCApertureNeg(Float_t /*aperture*/) {}
+  virtual void SetVCollSideCCentre(Float_t /*centre*/) {}
+  
+  virtual void SetVCollSideAAperture(Float_t /*aperture*/) {}
+  virtual void SetVCollSideAApertureNeg(Float_t /*aperture*/) {}
+  virtual void SetVCollSideACentre(Float_t /*centre*/) {}
+  
+  virtual void SetTCDDAperturePos(Float_t /*aperture*/) {}
+  virtual void SetTCDDApertureNeg(Float_t /*aperture*/) {}
+  
+  virtual void SetTDIAperturePos(Float_t /*aperture*/) {}
+  virtual void SetTDIApertureNeg(Float_t /*aperture*/) {}
+  
+  virtual void SetLumiLength(Float_t /*length*/) {}
+  
+  virtual void SetYZNC(Float_t /*yZNC*/) {}
+  virtual void SetYZNA(Float_t /*yZNA*/) {}
+  virtual void SetYZPC(Float_t /*yZPC*/) {}
+  virtual void SetYZPA(Float_t /*yZPA*/) {}
 
   //Calibration methods 
   void    SetZDCCalibFName(const char *name);
   char*   GetZDCCalibFName() const {return (char*)fZDCCalibFName.Data();}
   AliZDCPedestals* GetPedCalib()   const  {return fPedCalib;}
   AliZDCEnCalib*   GetECalibData() const  {return fEnCalibData;}
+  
+  // Map from OCDB
+  AliZDCChMap*     GetChMap() const;
 
   // Trigger
   virtual AliTriggerDetector* CreateTriggerDetector() const
@@ -60,6 +84,12 @@ public:
   
   void  SetSpectatorsTrack() {fSpectatorTracked=0;}
   Int_t SpectatorsTracked() const {return fSpectatorTracked;}
+  void  SetBeamEnergy(Float_t beamEnergy) {fBeamEnergy = beamEnergy;}
+  void  SetpAsystem() {fIspASystem = kTRUE;}
+  void  SetRELDISGenerator() {fIsRELDISgen = kTRUE;}
+  
+  void  SetOnlyZEM()           {fOnlyZEM=kTRUE;}
+  void  SetMotherFinding()     {fFindMother=kTRUE;}
 
 private:
 
@@ -76,9 +106,15 @@ protected:
 
   TString          fZDCCalibFName;     // Name of the ZDC calibration data
  
-  Int_t fSpectatorTracked; // Are spectator tracked by generator?
+  Int_t   fSpectatorTracked; // Are spectator tracked by generator? 0=NO
+  Float_t fBeamEnergy;      // beam energy from generator (AliGenZDC + RELDIS)
+  Bool_t  fIspASystem;       // Configuring pA collisions (MC only)
+  Bool_t  fIsRELDISgen;             // Is RELDIS used as generator
+  
+  Bool_t fOnlyZEM;          // build only ZEM (no had. calorimeters!)
+  Bool_t fFindMother;       // look for particle mothers in the stack in StepManager
   
-  ClassDef(AliZDC,9)   // Zero Degree Calorimeter base class
+  ClassDef(AliZDC,14)          // Zero Degree Calorimeter base class
 };
  
 // Calibration