First implementation of EMCAL trigger QA from Nicola Arbor
[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;}
45 virtual void SetPtHardMin(Float_t ptmin) {fPtHardMin = ptmin;}
46 virtual void SetPtHardMax(Float_t ptmax) {fPtHardMax = ptmax;}
47 virtual void SetPtJet(Float_t ptmin) {fPtMinJet = ptmin;}
48 virtual void SetSimpleJets(Int_t flag=0) {fSimpleJet = flag;}
49 virtual void SetNoGammas(Int_t flag=0) {fNoGammas = flag;}
50 virtual void SetJetEtaRange(Float_t etamin = -20., Float_t etamax = 20.)
51 {fEtaMinJet = etamin; fEtaMaxJet = etamax;}
52 virtual void SetJetPhiRange(Float_t phimin = -180., Float_t phimax = 180.)
53 {fPhiMinJet = TMath::Pi()*phimin/180.; fPhiMaxJet = TMath::Pi()*phimax/180.;}
54 virtual void SetBoostLHC(Int_t flag = 0) {fLHC = flag;}
55 virtual void SetRandomPz(Bool_t flag = 0) {fRandomPz = flag;}
56 virtual void SwitchOffHeavyQuarks(Bool_t flag = kTRUE) {fNoHeavyQuarks = flag;}
a004b331 57 virtual void SetIsoft(Int_t i) {fIsoft = i; }
58 virtual void SetNtMax(Int_t max) {fNtMax = max;}
59 virtual void SetIpop(Int_t pop) {fIpop = pop;}
60 virtual void SetXmu(Float_t m) {fXmu = m; }
61 virtual void SetAlpha(Float_t alpha) {fAlpha = alpha; }
62 virtual void SetStringFrag(Float_t a, Float_t b) {fStringA = a; fStringB = b;}
0119ef9a 63
64 // Getters
65 virtual TString GetReferenceFrame() const {return fFrame;}
66 virtual void GetImpactParameterRange(Float_t& bmin, Float_t& bmax) const
67 {bmin = fMinImpactParam; bmax = fMaxImpactParam;}
68 virtual Int_t GetJetQuenching() const {return fQuench;}
69 virtual Int_t GetShadowing() const {return fShadowing;}
70 virtual Int_t GetTrigger() const {return fTrigger;}
71 virtual Int_t GetFlavor() const {return fFlavor;}
72 virtual Int_t GetRadiation() const {return fRadiation;}
73 virtual Int_t GetSpectators() const {return fSpectators;}
74 virtual Float_t GetPtHardMin() const {return fPtHardMin;}
75 virtual Float_t GetPtHardMax() const {return fPtHardMax;}
76 virtual Float_t GetPtJet() const {return fPtMinJet;}
77 virtual void GetJetEtaRange(Float_t& etamin, Float_t& etamax) const
78 {etamin = fEtaMinJet; etamax = fEtaMaxJet;}
79 virtual void GetJetPhiRange(Float_t& phimin, Float_t& phimax) const
80 {phimin = fPhiMinJet*180./TMath::Pi(); phimax = fPhiMaxJet*180./TMath::Pi();}
81 const TClonesArray *GetParticles() const { return &fParticles; }
82
83 // Physics Routines
84 virtual Bool_t ProvidesCollisionGeometry() const {return kTRUE;}
85 virtual void EvaluateCrossSections();
86 virtual TGraph* CrossSection() {return fDsigmaDb;}
87 virtual TGraph* BinaryCollisions() {return fDnDb;}
88 virtual Bool_t CheckTrigger();
89
90 protected:
91 Bool_t SelectFlavor(Int_t pid);
92 void MakeHeader();
93
94 TString fFrame; // Reference frame
95 Float_t fMinImpactParam; // minimum impact parameter
96 Float_t fMaxImpactParam; // maximum impact parameter
97 Int_t fKeep; // Flag to keep full event information
98 Int_t fQuench; // Flag to switch on jet quenching
99 Int_t fShadowing; // Flag to switch on nuclear effects on parton distribution function
100 Int_t fDecaysOff; // Flag to turn off decays of pi0, K_s, D, Lambda, sigma
101 Int_t fTrigger; // Trigger type
102 Int_t fEvaluate; // Evaluate total and partial cross-sections
103 Int_t fSelectAll; // Flag to write the full event
104 Int_t fFlavor; // Selected particle flavor 4: charm+beauty 5: beauty
105 Float_t fKineBias; // Bias from kinematic selection
106 Int_t fTrials; // Number of trials
107 Float_t fXsection; // Cross-section
108 TAmpt *fAmpt; //!Ampt
109 Float_t fPtHardMin; // lower pT-hard cut
110 Float_t fPtHardMax; // higher pT-hard cut
111 Int_t fSpectators; // put spectators on stack
112 TGraph* fDsigmaDb; // dSigma/db for the system
113 TGraph* fDnDb; // dNBinaryCollisions/db
114 Float_t fPtMinJet; // Minimum Pt of triggered Jet
115 Float_t fEtaMinJet; // Minimum eta of triggered Jet
116 Float_t fEtaMaxJet; // Maximum eta of triggered Jet
117 Float_t fPhiMinJet; // At least one of triggered Jets must be in this
118 Float_t fPhiMaxJet; // phi range
119 Int_t fRadiation; // Flag to switch on/off initial and final state radiation
120 Int_t fSimpleJet; // Flag to produce simple tiggered jet topology
121 Int_t fNoGammas; // Don't write gammas if flag "on"
122 Int_t fProjectileSpecn; // Num. of spectator neutrons from projectile nucleus
123 Int_t fProjectileSpecp; // Num. of spectator protons from projectile nucleus
124 Int_t fTargetSpecn; // Num. of spectator neutrons from target nucleus
125 Int_t fTargetSpecp; // Num. of spectator protons from target nucleus
126 Int_t fLHC; // Assume LHC as lab frame
127 Bool_t fRandomPz; // Randomise sign of pz event by event
128 Bool_t fNoHeavyQuarks; // If true no heavy quarks are produced
0119ef9a 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
d60fac1e 137 Bool_t fDecay; // decay "long-lived" particles
0119ef9a 138
139 private:
140 AliGenAmpt(const AliGenAmpt &Ampt);
141 AliGenAmpt & operator=(const AliGenAmpt & rhs);
142
143 // adjust the weight from kinematic cuts
144 void AdjustWeights();
145 // check seleted daughters
146 Bool_t DaughtersSelection(TParticle* iparticle);
147 // check if stable
148 Bool_t Stable(TParticle* particle) const;
149
cdfa7be6 150 ClassDef(AliGenAmpt, 4) // AliGenerator interface to Ampt
0119ef9a 151};
152#endif