]>
Commit | Line | Data |
---|---|---|
0119ef9a | 1 | #ifndef ALIGENAMPT_H |
2 | #define ALIGENAMPT_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 | #include "AliGenMC.h" | |
9 | #include <TString.h> | |
10 | ||
11 | class TAmpt; | |
12 | class TParticle; | |
13 | class TClonesArray; | |
14 | class TGraph; | |
15 | class AliGenHijingEventHeader; | |
16 | ||
17 | class AliGenAmpt : public AliGenMC | |
18 | { | |
19 | enum {kNoTrigger, kHardProcesses, kDirectPhotons}; | |
20 | ||
21 | public: | |
22 | AliGenAmpt(); | |
23 | AliGenAmpt(Int_t npart); | |
24 | virtual ~AliGenAmpt(); | |
25 | ||
43c4c137 | 26 | virtual void SetDecay(Bool_t b) {fDecay = b;} |
0119ef9a | 27 | virtual TAmpt *Ampt() { return fAmpt; } |
28 | virtual void Generate(); | |
29 | virtual void Init(); | |
30 | virtual void SetEnergyCMS(Float_t energy=5500) {fEnergyCMS=energy;} | |
31 | virtual void SetReferenceFrame(TString frame="CMS") | |
32 | {fFrame=frame;} | |
33 | virtual void SetImpactParameterRange(Float_t bmin = 0, Float_t bmax = 15.) | |
34 | {fMinImpactParam=bmin; fMaxImpactParam=bmax;} | |
35 | virtual void KeepFullEvent() {fKeep =1; } | |
36 | virtual void SetJetQuenching(Int_t flag=1) {fQuench = flag;} | |
37 | virtual void SetShadowing(Int_t flag=1) {fShadowing = flag;} | |
38 | virtual void SetDecaysOff(Int_t flag=1) {fDecaysOff = flag;} | |
39 | virtual void SetTrigger(Int_t flag=kNoTrigger) {fTrigger = flag;} | |
40 | virtual void SetFlavor(Int_t flag=0) {fFlavor = flag;} | |
41 | virtual void SetEvaluate(Int_t flag=0) {fEvaluate = flag;} | |
42 | virtual void SetSelectAll(Int_t flag=0) {fSelectAll = flag;} | |
43 | virtual void SetRadiation(Int_t flag=3) {fRadiation = flag;} | |
44 | virtual void SetSpectators(Int_t spects=1) {fSpectators = spects;} | |
d12123e8 | 45 | virtual void SetDecayer(AliDecayer *decayer) {fDecayer = decayer;} |
0119ef9a | 46 | virtual void SetPtHardMin(Float_t ptmin) {fPtHardMin = ptmin;} |
47 | virtual void SetPtHardMax(Float_t ptmax) {fPtHardMax = ptmax;} | |
48 | virtual void SetPtJet(Float_t ptmin) {fPtMinJet = ptmin;} | |
49 | virtual void SetSimpleJets(Int_t flag=0) {fSimpleJet = flag;} | |
50 | virtual void SetNoGammas(Int_t flag=0) {fNoGammas = flag;} | |
51 | virtual void SetJetEtaRange(Float_t etamin = -20., Float_t etamax = 20.) | |
52 | {fEtaMinJet = etamin; fEtaMaxJet = etamax;} | |
53 | virtual void SetJetPhiRange(Float_t phimin = -180., Float_t phimax = 180.) | |
54 | {fPhiMinJet = TMath::Pi()*phimin/180.; fPhiMaxJet = TMath::Pi()*phimax/180.;} | |
55 | virtual void SetBoostLHC(Int_t flag = 0) {fLHC = flag;} | |
56 | virtual void SetRandomPz(Bool_t flag = 0) {fRandomPz = flag;} | |
57 | virtual void SwitchOffHeavyQuarks(Bool_t flag = kTRUE) {fNoHeavyQuarks = flag;} | |
a004b331 | 58 | virtual void SetIsoft(Int_t i) {fIsoft = i; } |
59 | virtual void SetNtMax(Int_t max) {fNtMax = max;} | |
60 | virtual void SetIpop(Int_t pop) {fIpop = pop;} | |
61 | virtual void SetXmu(Float_t m) {fXmu = m; } | |
62 | virtual void SetAlpha(Float_t alpha) {fAlpha = alpha; } | |
63 | virtual void SetStringFrag(Float_t a, Float_t b) {fStringA = a; fStringB = b;} | |
0119ef9a | 64 | |
65 | // Getters | |
66 | virtual TString GetReferenceFrame() const {return fFrame;} | |
67 | virtual void GetImpactParameterRange(Float_t& bmin, Float_t& bmax) const | |
68 | {bmin = fMinImpactParam; bmax = fMaxImpactParam;} | |
69 | virtual Int_t GetJetQuenching() const {return fQuench;} | |
70 | virtual Int_t GetShadowing() const {return fShadowing;} | |
71 | virtual Int_t GetTrigger() const {return fTrigger;} | |
72 | virtual Int_t GetFlavor() const {return fFlavor;} | |
73 | virtual Int_t GetRadiation() const {return fRadiation;} | |
74 | virtual Int_t GetSpectators() const {return fSpectators;} | |
75 | virtual Float_t GetPtHardMin() const {return fPtHardMin;} | |
76 | virtual Float_t GetPtHardMax() const {return fPtHardMax;} | |
77 | virtual Float_t GetPtJet() const {return fPtMinJet;} | |
78 | virtual void GetJetEtaRange(Float_t& etamin, Float_t& etamax) const | |
79 | {etamin = fEtaMinJet; etamax = fEtaMaxJet;} | |
80 | virtual void GetJetPhiRange(Float_t& phimin, Float_t& phimax) const | |
81 | {phimin = fPhiMinJet*180./TMath::Pi(); phimax = fPhiMaxJet*180./TMath::Pi();} | |
82 | const TClonesArray *GetParticles() const { return &fParticles; } | |
83 | ||
84 | // Physics Routines | |
85 | virtual Bool_t ProvidesCollisionGeometry() const {return kTRUE;} | |
86 | virtual void EvaluateCrossSections(); | |
87 | virtual TGraph* CrossSection() {return fDsigmaDb;} | |
88 | virtual TGraph* BinaryCollisions() {return fDnDb;} | |
89 | virtual Bool_t CheckTrigger(); | |
d12123e8 | 90 | virtual Bool_t IsThisAKnownParticle(TParticle *thisGuy); |
48beeea0 | 91 | virtual void SetRandomReactionPlane(Bool_t on=kTRUE){fRotating=on;}; |
0119ef9a | 92 | |
93 | protected: | |
94 | Bool_t SelectFlavor(Int_t pid); | |
95 | void MakeHeader(); | |
96 | ||
d12123e8 | 97 | AliDecayer *fDecayer; |
0119ef9a | 98 | TString fFrame; // Reference frame |
99 | Float_t fMinImpactParam; // minimum impact parameter | |
100 | Float_t fMaxImpactParam; // maximum impact parameter | |
101 | Int_t fKeep; // Flag to keep full event information | |
102 | Int_t fQuench; // Flag to switch on jet quenching | |
103 | Int_t fShadowing; // Flag to switch on nuclear effects on parton distribution function | |
104 | Int_t fDecaysOff; // Flag to turn off decays of pi0, K_s, D, Lambda, sigma | |
105 | Int_t fTrigger; // Trigger type | |
106 | Int_t fEvaluate; // Evaluate total and partial cross-sections | |
107 | Int_t fSelectAll; // Flag to write the full event | |
108 | Int_t fFlavor; // Selected particle flavor 4: charm+beauty 5: beauty | |
109 | Float_t fKineBias; // Bias from kinematic selection | |
110 | Int_t fTrials; // Number of trials | |
111 | Float_t fXsection; // Cross-section | |
112 | TAmpt *fAmpt; //!Ampt | |
113 | Float_t fPtHardMin; // lower pT-hard cut | |
114 | Float_t fPtHardMax; // higher pT-hard cut | |
115 | Int_t fSpectators; // put spectators on stack | |
116 | TGraph* fDsigmaDb; // dSigma/db for the system | |
117 | TGraph* fDnDb; // dNBinaryCollisions/db | |
118 | Float_t fPtMinJet; // Minimum Pt of triggered Jet | |
119 | Float_t fEtaMinJet; // Minimum eta of triggered Jet | |
120 | Float_t fEtaMaxJet; // Maximum eta of triggered Jet | |
121 | Float_t fPhiMinJet; // At least one of triggered Jets must be in this | |
122 | Float_t fPhiMaxJet; // phi range | |
123 | Int_t fRadiation; // Flag to switch on/off initial and final state radiation | |
124 | Int_t fSimpleJet; // Flag to produce simple tiggered jet topology | |
125 | Int_t fNoGammas; // Don't write gammas if flag "on" | |
126 | Int_t fProjectileSpecn; // Num. of spectator neutrons from projectile nucleus | |
127 | Int_t fProjectileSpecp; // Num. of spectator protons from projectile nucleus | |
128 | Int_t fTargetSpecn; // Num. of spectator neutrons from target nucleus | |
129 | Int_t fTargetSpecp; // Num. of spectator protons from target nucleus | |
130 | Int_t fLHC; // Assume LHC as lab frame | |
131 | Bool_t fRandomPz; // Randomise sign of pz event by event | |
132 | Bool_t fNoHeavyQuarks; // If true no heavy quarks are produced | |
0119ef9a | 133 | Int_t fIsoft; // ISOFT (D=1): select Default AMPT or String Melting |
134 | Int_t fNtMax; // NTMAX: number of timesteps (D=150) | |
135 | Int_t fIpop; // (D=1,yes;0,no) flag for popcorn mechanism(netbaryon stopping) | |
136 | Float_t fXmu; // parton screening mass in fm^(-1) (D=3.2264d0) | |
a004b331 | 137 | Float_t fAlpha; // alpha running (fixed) coupling |
138 | Float_t fStringA; // string frag parameter A | |
139 | Float_t fStringB; // string frag parameter B | |
d12123e8 | 140 | Float_t fEventTime; // The event time |
0119ef9a | 141 | AliGenHijingEventHeader *fHeader; // header |
d60fac1e | 142 | Bool_t fDecay; // decay "long-lived" particles |
48beeea0 | 143 | Bool_t fRotating; // Random reaction plane on/off |
144 | ||
0119ef9a | 145 | |
146 | private: | |
147 | AliGenAmpt(const AliGenAmpt &Ampt); | |
148 | AliGenAmpt & operator=(const AliGenAmpt & rhs); | |
149 | ||
150 | // adjust the weight from kinematic cuts | |
151 | void AdjustWeights(); | |
152 | // check seleted daughters | |
153 | Bool_t DaughtersSelection(TParticle* iparticle); | |
154 | // check if stable | |
155 | Bool_t Stable(TParticle* particle) const; | |
156 | ||
cdfa7be6 | 157 | ClassDef(AliGenAmpt, 4) // AliGenerator interface to Ampt |
0119ef9a | 158 | }; |
159 | #endif |