Beam position from CDB (Massimo)

[u/mrichter/AliRoot.git] / STEER / AliGenerator.h

diff --git a/STEER/AliGenerator.h b/STEER/AliGenerator.h
index 1dc7a7296619e6e38a09ded75b14156e27eaaf2b..7a84c6d5936f13f90269131aa15ef7b916c28084 100644 (file)
--- a/STEER/AliGenerator.h
+++ b/STEER/AliGenerator.h
@@ -18,9 +18,10 @@
 
 #include "AliRndm.h"
 
+class AliVertexGenerator;
 class AliCollisionGeometry;
+class AliGenEventHeader;
 class AliStack;
-
 class TGenerator;
 
 
@@ -35,10 +36,8 @@ class AliGenerator : public TNamed, public AliRndm
  public:
     AliGenerator();
     AliGenerator(Int_t npart);
-    AliGenerator(const AliGenerator &gen);
     virtual ~AliGenerator();
     virtual void Init();
-    void Copy(AliGenerator &gen) const;
     virtual void SetOrigin(Float_t ox, Float_t oy, Float_t oz);
     virtual void SetOrigin(const TLorentzVector &o);
     virtual void SetSigma(Float_t sx, Float_t sy, Float_t sz);
@@ -61,15 +60,18 @@ class AliGenerator : public TNamed, public AliRndm
     virtual void SetVertexSource(VertexSource_t source = kInternal) {fVertexSource = source;}
     virtual void SetVertex(Float_t vx, Float_t vy, Float_t vz)
        {fVertex[0] = vx; fVertex[1] = vy; fVertex[2] = vz;}
+    void SetVertexGenerator(AliVertexGenerator* vertexGenerator)
+      {fVertexGenerator = vertexGenerator; fVertexSource = kExternal; fVertexSmear = kPerEvent;}
+    void SetPileUpTimeWindow(Float_t pileUpTimeW) {fPileUpTimeWindow = pileUpTimeW;}
+    
     virtual void SetTrackingFlag(Int_t flag=1) {fTrackIt=flag;}
     void Vertex();
     void VertexExternal();
     virtual void VertexInternal();
     virtual void FinishRun();
-    virtual void SetMC(TGenerator *theMC) 
-       {if (!fgMCEvGen) fgMCEvGen =theMC;}
-
-    AliGenerator & operator=(const AliGenerator &gen);
+    virtual void SetMC(TGenerator *theMC) {fMCEvGen =theMC;}
+    virtual void AddHeader(AliGenEventHeader* /*header*/) {;}
+    virtual void SetContainer(AliGenerator* container) {fContainer = container;}
 
   // Getters
 
@@ -83,14 +85,14 @@ class AliGenerator : public TNamed, public AliRndm
   // Collision Geometry
     virtual Bool_t ProvidesCollisionGeometry() const {return kFALSE;}
     virtual Bool_t NeedsCollisionGeometry()    const {return kFALSE;}    
-    virtual AliCollisionGeometry* CollisionGeometry() {return fCollisionGeometry;}
+    virtual AliCollisionGeometry* CollisionGeometry() const {return fCollisionGeometry;}
     virtual void SetCollisionGeometry(AliCollisionGeometry* geom) {fCollisionGeometry = geom;}
  protected:
-    virtual  void  SetTrack(Int_t done, Int_t parent, Int_t pdg,
+    virtual  void  PushTrack(Int_t done, Int_t parent, Int_t pdg,
                                Float_t *pmom, Float_t *vpos, Float_t *polar,
                                Float_t tof, TMCProcess mech, Int_t &ntr,
                                Float_t weight = 1, Int_t is = 0);
-    virtual  void  SetTrack(Int_t done, Int_t parent, Int_t pdg,
+    virtual  void  PushTrack(Int_t done, Int_t parent, Int_t pdg,
                       Double_t px, Double_t py, Double_t pz, Double_t e,
                       Double_t vx, Double_t vy, Double_t vz, Double_t tof,
                       Double_t polx, Double_t poly, Double_t polz,
@@ -99,7 +101,7 @@ class AliGenerator : public TNamed, public AliRndm
     virtual void   SetHighWaterMark(Int_t nt);
     
  protected:
-    static  TGenerator* fgMCEvGen; //Pointer to the generator
+    TGenerator* fMCEvGen;      //!Pointer to the generator
     Float_t     fThetaMin;     //Minimum theta of generation in radians
     Float_t     fThetaMax;     //Maximum theta of generation in radians
     Float_t     fPhiMin;       //Minimum phi of generation in radians
@@ -119,13 +121,17 @@ class AliGenerator : public TNamed, public AliRndm
    //
     VertexSmear_t     fVertexSmear;  //Vertex Smearing mode
     VertexSource_t    fVertexSource; //Vertex source (internal/external)
-    Float_t     fCutVertexZ;    // Vertex cut in units of sigma_z
+    Float_t     fCutVertexZ;         //Vertex cut in units of sigma_z
+    Float_t     fPileUpTimeWindow;   //Time window for pile-up events
     Int_t       fTrackIt;    // if 1, Track final state particles 
+    AliVertexGenerator* fVertexGenerator;  //! Generator for the vertex
     TArrayF     fOrigin;     // Origin of event
     TArrayF     fOsigma;     // Sigma of the Origin of event
-    TArrayF     fVertex;     //! Vertex of current event
+    TArrayF     fVertex;        //! Vertex of current event
 
-    AliStack*   fStack;      //! Local pointer to stack
+    AliStack*   fStack;         //! Local pointer to stack
+    AliGenerator* fContainer;   //! Local pointer to container
+    
     AliCollisionGeometry* fCollisionGeometry; //!Collision geometry
     /*************************************************************************/
     enum {kThetaRange    = BIT(14),
@@ -135,6 +141,11 @@ class AliGenerator : public TNamed, public AliRndm
          kYRange        = BIT(18),
          kMomentumRange = BIT(19)     
     };
+
+ private:
+    AliGenerator(const AliGenerator &gen);
+    AliGenerator & operator=(const AliGenerator &gen);
+
     ClassDef(AliGenerator,3) // Base class for event generators
 };