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