]>
Commit | Line | Data |
---|---|---|
e36044d6 | 1 | #ifndef ALIGENMC_H |
2 | #define ALIGENMC_H | |
3 | /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. * | |
4 | * See cxx source for full Copyright notice */ | |
5 | ||
6 | /* $Id$ */ | |
7 | ||
8 | ||
62d0ae06 | 9 | // Base class for generators using external MC generators. |
10 | // For example AliGenPythia using Pythia. | |
11 | // Provides basic functionality: setting of kinematic cuts on | |
12 | // decay products and particle selection. | |
13 | // andreas.morsch@cern.ch | |
14 | ||
ea24f998 | 15 | #include<TClonesArray.h> |
3b945a60 | 16 | #include <TArrayI.h> |
3b945a60 | 17 | #include <TString.h> |
ea24f998 | 18 | class TParticle; |
e36044d6 | 19 | |
ac3faee4 | 20 | class AliGeometry; |
d97fb2bb | 21 | class AliGenEventHeader; |
ac3faee4 | 22 | #include "AliDecayer.h" |
23 | #include "AliGenerator.h" | |
e36044d6 | 24 | |
25 | class AliGenMC : public AliGenerator | |
26 | { | |
27 | public: | |
28 | AliGenMC(); | |
29 | AliGenMC(Int_t npart); | |
e36044d6 | 30 | virtual ~AliGenMC(); |
7a5ebd8f | 31 | virtual void Init(); |
a0609231 | 32 | virtual void SetForceDecay(Decay_t decay = kAll) {fForceDecay = decay;} |
93a2041b | 33 | virtual void SetCutOnChild(Int_t flag = 0) {fCutOnChild = flag;} |
e36044d6 | 34 | virtual void SetChildMomentumRange(Float_t pmin = 0, Float_t pmax = 1.e10) |
35 | {fChildPMin = pmin; fChildPMax = pmax;} | |
36 | virtual void SetChildPtRange(Float_t ptmin = 0, Float_t ptmax = 20.) | |
37 | {fChildPtMin = ptmin; fChildPtMax = ptmax;} | |
ab456baa | 38 | virtual void SetChildPhiRange(Float_t phimin = 0., Float_t phimax = 360.) |
e36044d6 | 39 | {fChildPhiMin = TMath::Pi()*phimin/180; |
40 | fChildPhiMax = TMath::Pi()*phimax/180;} | |
41 | virtual void SetChildThetaRange(Float_t thetamin = 0, Float_t thetamax = 180) | |
42 | {fChildThetaMin = TMath::Pi()*thetamin/180; | |
43 | fChildThetaMax = TMath::Pi()*thetamax/180;} | |
44 | virtual void SetChildYRange(Float_t ymin = -12, Float_t ymax = 12) | |
45 | {fChildYMin = ymin; | |
46 | fChildYMax = ymax;} | |
65de9f86 | 47 | virtual void SetMaximumLifetime(Float_t time = 1.e-15) {fMaxLifeTime = time;} |
48 | ||
49 | virtual void SetGeometryAcceptance(AliGeometry * GeometryAcceptance=0) {fGeometryAcceptance = GeometryAcceptance;} | |
50 | ||
51 | virtual void SetPdgCodeParticleforAcceptanceCut(Int_t PdgCodeParticleforAcceptanceCut=0) {fPdgCodeParticleforAcceptanceCut = PdgCodeParticleforAcceptanceCut;} | |
52 | ||
53 | virtual void SetNumberOfAcceptedParticles(Int_t NumberOfAcceptedParticles=2) {fNumberOfAcceptedParticles = NumberOfAcceptedParticles;} | |
3b945a60 | 54 | |
65de9f86 | 55 | virtual Bool_t CheckAcceptanceGeometry(Int_t np, TClonesArray* particles); |
36d770d9 | 56 | virtual void SetProjectile(TString proj="P", Int_t a = 1, Int_t z = 1) |
3b945a60 | 57 | {fProjectile = proj; fAProjectile = a; fZProjectile = z;} |
36d770d9 | 58 | virtual void SetTarget(TString tar="P", Int_t a = 1, Int_t z = 1) |
1c56e311 | 59 | {fTarget = tar; fATarget = a; fZTarget = z;} |
60 | virtual void SetCrossingAngle(Float_t phiX, Float_t phiY) {fXingAngleX = phiX; fXingAngleY = phiY;} | |
36d770d9 | 61 | virtual void Boost(); |
7eb751fe | 62 | virtual void BeamCrossAngle(); |
d97fb2bb | 63 | virtual void AddHeader(AliGenEventHeader* header); |
fc7e1b1c | 64 | |
e36044d6 | 65 | protected: |
66 | // check if particle is selected as parent particle | |
62d0ae06 | 67 | Bool_t ParentSelected(Int_t ip) const; |
e36044d6 | 68 | // check if particle is selected as child particle |
62d0ae06 | 69 | Bool_t ChildSelected(Int_t ip) const; |
e36044d6 | 70 | // all kinematic selection cuts go here |
62d0ae06 | 71 | Bool_t KinematicSelection(TParticle *particle, Int_t flag) const; |
72 | Int_t CheckPDGCode(Int_t pdgcode) const; | |
93a2041b | 73 | |
e36044d6 | 74 | protected: |
ea24f998 | 75 | TClonesArray fParticles; //!Particle List |
76 | TArrayI fParentSelect; //!Parent particles to be selected | |
77 | TArrayI fChildSelect; //!Decay products to be selected | |
78 | Int_t fCutOnChild; // Cuts on decay products (children) are enabled/disabled | |
79 | Float_t fChildPtMin; // Children minimum pT | |
80 | Float_t fChildPtMax; // Children maximum pT | |
81 | Float_t fChildPMin; // Children minimum p | |
82 | Float_t fChildPMax; // Children maximum p | |
83 | Float_t fChildPhiMin; // Children minimum phi | |
84 | Float_t fChildPhiMax; // Children maximum phi | |
85 | Float_t fChildThetaMin; // Children minimum theta | |
86 | Float_t fChildThetaMax; // Children maximum theta | |
87 | Float_t fChildYMin; // Children minimum y | |
88 | Float_t fChildYMax; // Children maximum y | |
89 | Float_t fXingAngleX; // Crossing angle X | |
90 | Float_t fXingAngleY; // Crossing angle Y | |
91 | Decay_t fForceDecay; // Decay channel forced | |
92 | Float_t fMaxLifeTime; // Maximum lifetime for unstable particles | |
ea24f998 | 93 | Double_t fDyBoost; // dy for boost into lab frame |
94 | AliGeometry* fGeometryAcceptance; // Geometry to which particles must be simulated | |
95 | Int_t fPdgCodeParticleforAcceptanceCut; // Abs(PDG Code) of the particle to which the GeometryAcceptance must be applied | |
96 | Int_t fNumberOfAcceptedParticles; // Number of accepted particles in GeometryAcceptance with the right Abs(PdgCode) | |
97 | Int_t fNprimaries; // Number of produced and stored particles | |
7eb751fe | 98 | |
93a2041b | 99 | private: |
100 | AliGenMC(const AliGenMC &MC); | |
101 | AliGenMC & operator=(const AliGenMC & rhs); | |
47fc6bd5 | 102 | |
e7c989e4 | 103 | ClassDef(AliGenMC,7) // AliGenerator implementation for generators using MC methods |
e36044d6 | 104 | }; |
105 | #endif | |
106 | ||
107 | ||
108 | ||
109 | ||
110 |