Add new functionalities (Laurent)
[u/mrichter/AliRoot.git] / TAmpt / AliGenAmpt.h
CommitLineData
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
11class TAmpt;
12class TParticle;
13class TClonesArray;
14class TGraph;
15class AliGenHijingEventHeader;
16
17class 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