Adding TAmpt (Constantin)
[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;}
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
61 // Getters
62 virtual TString GetReferenceFrame() const {return fFrame;}
63 virtual void GetImpactParameterRange(Float_t& bmin, Float_t& bmax) const
64 {bmin = fMinImpactParam; bmax = fMaxImpactParam;}
65 virtual Int_t GetJetQuenching() const {return fQuench;}
66 virtual Int_t GetShadowing() const {return fShadowing;}
67 virtual Int_t GetTrigger() const {return fTrigger;}
68 virtual Int_t GetFlavor() const {return fFlavor;}
69 virtual Int_t GetRadiation() const {return fRadiation;}
70 virtual Int_t GetSpectators() const {return fSpectators;}
71 virtual Float_t GetPtHardMin() const {return fPtHardMin;}
72 virtual Float_t GetPtHardMax() const {return fPtHardMax;}
73 virtual Float_t GetPtJet() const {return fPtMinJet;}
74 virtual void GetJetEtaRange(Float_t& etamin, Float_t& etamax) const
75 {etamin = fEtaMinJet; etamax = fEtaMaxJet;}
76 virtual void GetJetPhiRange(Float_t& phimin, Float_t& phimax) const
77 {phimin = fPhiMinJet*180./TMath::Pi(); phimax = fPhiMaxJet*180./TMath::Pi();}
78 const TClonesArray *GetParticles() const { return &fParticles; }
79
80 // Physics Routines
81 virtual Bool_t ProvidesCollisionGeometry() const {return kTRUE;}
82 virtual void EvaluateCrossSections();
83 virtual TGraph* CrossSection() {return fDsigmaDb;}
84 virtual TGraph* BinaryCollisions() {return fDnDb;}
85 virtual Bool_t CheckTrigger();
86
87 protected:
88 Bool_t SelectFlavor(Int_t pid);
89 void MakeHeader();
90
91 TString fFrame; // Reference frame
92 Float_t fMinImpactParam; // minimum impact parameter
93 Float_t fMaxImpactParam; // maximum impact parameter
94 Int_t fKeep; // Flag to keep full event information
95 Int_t fQuench; // Flag to switch on jet quenching
96 Int_t fShadowing; // Flag to switch on nuclear effects on parton distribution function
97 Int_t fDecaysOff; // Flag to turn off decays of pi0, K_s, D, Lambda, sigma
98 Int_t fTrigger; // Trigger type
99 Int_t fEvaluate; // Evaluate total and partial cross-sections
100 Int_t fSelectAll; // Flag to write the full event
101 Int_t fFlavor; // Selected particle flavor 4: charm+beauty 5: beauty
102 Float_t fKineBias; // Bias from kinematic selection
103 Int_t fTrials; // Number of trials
104 Float_t fXsection; // Cross-section
105 TAmpt *fAmpt; //!Ampt
106 Float_t fPtHardMin; // lower pT-hard cut
107 Float_t fPtHardMax; // higher pT-hard cut
108 Int_t fSpectators; // put spectators on stack
109 TGraph* fDsigmaDb; // dSigma/db for the system
110 TGraph* fDnDb; // dNBinaryCollisions/db
111 Float_t fPtMinJet; // Minimum Pt of triggered Jet
112 Float_t fEtaMinJet; // Minimum eta of triggered Jet
113 Float_t fEtaMaxJet; // Maximum eta of triggered Jet
114 Float_t fPhiMinJet; // At least one of triggered Jets must be in this
115 Float_t fPhiMaxJet; // phi range
116 Int_t fRadiation; // Flag to switch on/off initial and final state radiation
117 Int_t fSimpleJet; // Flag to produce simple tiggered jet topology
118 Int_t fNoGammas; // Don't write gammas if flag "on"
119 Int_t fProjectileSpecn; // Num. of spectator neutrons from projectile nucleus
120 Int_t fProjectileSpecp; // Num. of spectator protons from projectile nucleus
121 Int_t fTargetSpecn; // Num. of spectator neutrons from target nucleus
122 Int_t fTargetSpecp; // Num. of spectator protons from target nucleus
123 Int_t fLHC; // Assume LHC as lab frame
124 Bool_t fRandomPz; // Randomise sign of pz event by event
125 Bool_t fNoHeavyQuarks; // If true no heavy quarks are produced
126 Float_t fEventTime; // The event time
127 Int_t fIsoft; // ISOFT (D=1): select Default AMPT or String Melting
128 Int_t fNtMax; // NTMAX: number of timesteps (D=150)
129 Int_t fIpop; // (D=1,yes;0,no) flag for popcorn mechanism(netbaryon stopping)
130 Float_t fXmu; // parton screening mass in fm^(-1) (D=3.2264d0)
131 AliGenHijingEventHeader *fHeader; // header
132
133 private:
134 AliGenAmpt(const AliGenAmpt &Ampt);
135 AliGenAmpt & operator=(const AliGenAmpt & rhs);
136
137 // adjust the weight from kinematic cuts
138 void AdjustWeights();
139 // check seleted daughters
140 Bool_t DaughtersSelection(TParticle* iparticle);
141 // check if stable
142 Bool_t Stable(TParticle* particle) const;
143
144 ClassDef(AliGenAmpt, 1) // AliGenerator interface to Ampt
145};
146#endif