Overlaps corrected, new shape of sectors
[u/mrichter/AliRoot.git] / EVGEN / AliGenMC.h
index ea77e1b..4d147a7 100644 (file)
@@ -6,28 +6,36 @@
 /* $Id$ */
 
 
-#include "AliGenerator.h"
-#include "AliDecayer.h"
-#include <TArrayI.h>    
+// Base class for generators using external MC generators.
+// For example AliGenPythia using Pythia.
+// Provides basic functionality: setting of kinematic cuts on 
+// decay products and particle selection.
+// andreas.morsch@cern.ch
 
+#include<TClonesArray.h>
+#include <TArrayI.h>   
+#include <TString.h>
 class TParticle;
 
+class AliGeometry;
+class AliGenEventHeader;
+#include "AliGenerator.h"
+#include "AliDecayer.h"
+
 class AliGenMC : public AliGenerator
 {
  public:
     AliGenMC();
     AliGenMC(Int_t npart);
-    AliGenMC(const AliGenMC &MC);
     virtual ~AliGenMC();
-    void Init();
-    virtual void SetForceDecay(Decay_t decay = kDiMuon) {fForceDecay = decay;}
-    AliGenMC & operator=(const AliGenMC & rhs);
-        virtual void SetCutOnChild(Int_t flag = 0) {fCutOnChild = flag;}
+    virtual void Init();
+    virtual void SetForceDecay(Decay_t decay = kAll) {fForceDecay = decay;}
+    virtual void SetCutOnChild(Int_t flag = 0) {fCutOnChild = flag;}
     virtual void SetChildMomentumRange(Float_t pmin = 0, Float_t pmax = 1.e10)
        {fChildPMin = pmin; fChildPMax = pmax;}
     virtual void SetChildPtRange(Float_t ptmin = 0, Float_t ptmax = 20.)
        {fChildPtMin = ptmin; fChildPtMax = ptmax;}
-    virtual void SetChildPhiRange(Float_t phimin = -180., Float_t phimax = 180)
+    virtual void SetChildPhiRange(Float_t phimin = 0., Float_t phimax = 360.)
        {fChildPhiMin = TMath::Pi()*phimin/180;
        fChildPhiMax  = TMath::Pi()*phimax/180;}
     virtual void SetChildThetaRange(Float_t thetamin = 0, Float_t thetamax = 180)
@@ -36,33 +44,64 @@ class AliGenMC : public AliGenerator
     virtual void SetChildYRange(Float_t ymin = -12, Float_t ymax = 12)
        {fChildYMin = ymin;
        fChildYMax  = ymax;}
+    virtual void SetMaximumLifetime(Float_t time = 1.e-15) {fMaxLifeTime = time;}
+   
+    virtual void SetGeometryAcceptance(AliGeometry * GeometryAcceptance=0) {fGeometryAcceptance = GeometryAcceptance;}
+
+    virtual void SetPdgCodeParticleforAcceptanceCut(Int_t PdgCodeParticleforAcceptanceCut=0) {fPdgCodeParticleforAcceptanceCut = PdgCodeParticleforAcceptanceCut;}
+
+    virtual void SetNumberOfAcceptedParticles(Int_t NumberOfAcceptedParticles=2) {fNumberOfAcceptedParticles = NumberOfAcceptedParticles;}
+    
+    virtual Bool_t CheckAcceptanceGeometry(Int_t np, TClonesArray* particles);
+    virtual void   SetProjectile(TString proj="P", Int_t a = 1, Int_t z = 1)
+       {fProjectile = proj; fAProjectile = a; fZProjectile = z;}    
+    virtual void   SetTarget(TString tar="P", Int_t a = 1, Int_t z = 1)
+       {fTarget = tar; fATarget = a; fZTarget = z;}
+    virtual void   SetCrossingAngle(Float_t phiX, Float_t phiY) {fXingAngleX = phiX; fXingAngleY = phiY;}
+    virtual void Boost();
+    virtual void SetDyBoost(Double_t dyBoost = 0.) {fDyBoost = dyBoost;}
+    virtual void BeamCrossAngle();
+    virtual void AddHeader(AliGenEventHeader* header);
 
  protected:
     // check if particle is selected as parent particle
-    Bool_t ParentSelected(Int_t ip);
+    Bool_t ParentSelected(Int_t ip) const;
     // check if particle is selected as child particle
-    Bool_t ChildSelected(Int_t ip);
+    Bool_t ChildSelected(Int_t ip) const;
     // all kinematic selection cuts go here 
-    Bool_t KinematicSelection(TParticle *particle, Int_t flag);
-    Int_t  CheckPDGCode(Int_t pdgcode);
+    Bool_t KinematicSelection(const TParticle *particle, Int_t flag) const;
+    Int_t  CheckPDGCode(Int_t pdgcode) const;
 
  protected:
-    TArrayI     fParentSelect;  // Parent particles to be selected 
-    TArrayI     fChildSelect;   // Decay products to be selected
-    Int_t       fCutOnChild;    // Cuts on decay products (children)  are enabled/disabled
-    Float_t     fChildPtMin;    // Children minimum pT
-    Float_t     fChildPtMax;    // Children maximum pT
-    Float_t     fChildPMin;     // Children minimum p
-    Float_t     fChildPMax;     // Children maximum p
-    Float_t     fChildPhiMin;   // Children minimum phi
-    Float_t     fChildPhiMax;   // Children maximum phi
-    Float_t     fChildThetaMin; // Children minimum theta
-    Float_t     fChildThetaMax; // Children maximum theta
-    Float_t     fChildYMin;     // Children minimum y
-    Float_t     fChildYMax;     // Children maximum y
-    Decay_t     fForceDecay;    // Decay channel forced
-       
-    ClassDef(AliGenMC,1)       // AliGenerator implementation for generators using MC methods
+    TClonesArray fParticles;     //!Particle  List
+    TArrayI      fParentSelect;  //!Parent particles to be selected 
+    TArrayI      fChildSelect;   //!Decay products to be selected
+    Int_t        fCutOnChild;    // Cuts on decay products (children)  are enabled/disabled
+    Float_t      fChildPtMin;    // Children minimum pT
+    Float_t      fChildPtMax;    // Children maximum pT
+    Float_t      fChildPMin;     // Children minimum p
+    Float_t      fChildPMax;     // Children maximum p
+    Float_t      fChildPhiMin;   // Children minimum phi
+    Float_t      fChildPhiMax;   // Children maximum phi
+    Float_t      fChildThetaMin; // Children minimum theta
+    Float_t      fChildThetaMax; // Children maximum theta
+    Float_t      fChildYMin;     // Children minimum y
+    Float_t      fChildYMax;     // Children maximum y
+    Float_t      fXingAngleX;    // Crossing angle X
+    Float_t      fXingAngleY;    // Crossing angle Y    
+    Decay_t      fForceDecay;    // Decay channel forced
+    Float_t      fMaxLifeTime;   // Maximum lifetime for unstable particles
+    Double_t     fDyBoost;       // dy for boost into lab frame
+    AliGeometry* fGeometryAcceptance; // Geometry to which particles must be simulated
+    Int_t        fPdgCodeParticleforAcceptanceCut;  // Abs(PDG Code) of the particle to which the GeometryAcceptance must be applied
+    Int_t        fNumberOfAcceptedParticles;  // Number of accepted particles in GeometryAcceptance with the right Abs(PdgCode) 
+    Int_t        fNprimaries;                 // Number of produced and stored particles
+ private:
+    AliGenMC(const AliGenMC &MC);
+    AliGenMC & operator=(const AliGenMC & rhs);
+    
+    ClassDef(AliGenMC,7)       // AliGenerator implementation for generators using MC methods
 };
 #endif