Merging correctly the changes with the previous code clean-up
[u/mrichter/AliRoot.git] / THijing / AliGenHijing.h
CommitLineData
36b81802 1#ifndef ALIGENHIJING_H
2#define ALIGENHIJING_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// Generator using HIJING as an external generator
9// The main HIJING options are accessable for the user through this interface.
10// andreas.morsch@cern.ch
11
12#include "AliGenMC.h"
13#include <TString.h>
14#include <TArrayI.h>
15
16class THijing;
17class TArrayI;
18class TParticle;
19class TClonesArray;
20class TGraph;
21
22class AliGenHijing : public AliGenMC
23{
24 enum {kNoTrigger, kHardProcesses, kDirectPhotons};
25
26 public:
27 AliGenHijing();
28 AliGenHijing(Int_t npart);
29 AliGenHijing(const AliGenHijing &Hijing);
30 virtual ~AliGenHijing();
31 virtual void Generate();
32 virtual void Init();
33 // set centre of mass energy
34 virtual void SetEnergyCMS(Float_t energy=5500) {fEnergyCMS=energy;}
35 virtual void SetReferenceFrame(TString frame="CMS")
36 {fFrame=frame;}
37 virtual void SetProjectile(TString proj="A", Int_t a=208, Int_t z=82)
38 {fProjectile = proj; fAProjectile = a; fZProjectile = z;}
39 virtual void SetTarget(TString tar="A", Int_t a=208, Int_t z=82)
40 {fTarget = tar; fATarget = a; fZTarget = z;}
41 virtual void SetImpactParameterRange(Float_t bmin = 0, Float_t bmax = 15.)
42 {fMinImpactParam=bmin; fMaxImpactParam=bmax;}
43 virtual void KeepFullEvent();
44 virtual void SetJetQuenching(Int_t flag=1) {fQuench = flag;}
45 virtual void SetShadowing(Int_t flag=1) {fShadowing = flag;}
46 virtual void SetDecaysOff(Int_t flag=1) {fDecaysOff = flag;}
47 virtual void SetTrigger(Int_t flag=kNoTrigger) {fTrigger = flag;}
48 virtual void SetFlavor(Int_t flag=0) {fFlavor = flag;}
49 virtual void SetEvaluate(Int_t flag=0) {fEvaluate = flag;}
50 virtual void SetSelectAll(Int_t flag=0) {fSelectAll = flag;}
51 virtual void SetRadiation(Int_t flag=3) {fRadiation = flag;}
52 virtual void SetSpectators(Int_t spects=1) {fSpectators = spects;}
53 virtual void SetPtJet(Float_t ptmin) {fPtMinJet = ptmin;}
54 virtual void SetSimpleJets(Int_t flag=0) {fSimpleJet = flag;}
55 virtual void SetNoGammas(Int_t flag=0) {fNoGammas = flag;}
56
57 virtual void SetJetEtaRange(Float_t etamin = -20., Float_t etamax = 20.)
58 {fEtaMinJet = etamin; fEtaMaxJet = etamax;}
59 virtual void SetJetPhiRange(Float_t phimin = -180., Float_t phimax = 180.)
60 {fPhiMinJet = TMath::Pi()*phimin/180.; fPhiMaxJet = TMath::Pi()*phimax/180.;}
61 virtual void SetBoostLHC(Int_t flag = 0) {fLHC = flag;}
62// Getters
63 virtual Float_t GetEnergyCMS() {return fEnergyCMS;}
64 virtual TString GetReferenceFrame() {return fFrame;}
65 virtual void GetProjectile(TString& tar, Int_t& a, Int_t& z)
66 {tar = fProjectile; a = fAProjectile; z = fZProjectile;}
67 virtual void GetTarget(TString& tar, Int_t& a, Int_t& z)
68 {tar = fTarget; a = fATarget; z = fZTarget;}
69 virtual void GetImpactParameterRange(Float_t& bmin, Float_t& bmax)
70 {bmin = fMinImpactParam; bmax = fMaxImpactParam;}
71 virtual Int_t GetJetQuenching() {return fQuench;}
72 virtual Int_t GetShadowing() {return fShadowing;}
73 virtual Int_t GetTrigger(Int_t flag=kNoTrigger) {return fTrigger;}
74 virtual Int_t GetFlavor(Int_t flag=0) {return fFlavor;}
75 virtual Int_t GetRadiation(Int_t flag=3) {return fRadiation;}
76 virtual Int_t GetSpectators(Int_t spects=1) {return fSpectators;}
77 virtual Float_t GetPtJet(Float_t ptmin) {return fPtMinJet;}
78 virtual void GetJetEtaRange(Float_t& etamin, Float_t& etamax)
79 {etamin = fEtaMinJet; etamax = fEtaMaxJet;}
80 virtual void GetJetPhiRange(Float_t& phimin, Float_t& phimax)
81 {phimin = fPhiMinJet*180./TMath::Pi(); phimax = fPhiMaxJet*180./TMath::Pi();}
82
83
84// Physics Routines
85 virtual Bool_t ProvidesCollisionGeometry() {return kTRUE;}
86 virtual AliCollisionGeometry* CollisionGeometry() {return fCollisionGeometry;}
87 virtual void EvaluateCrossSections();
88 virtual void Boost();
89 virtual TGraph* CrossSection() {return fDsigmaDb;}
90 virtual TGraph* BinaryCollisions() {return fDnDb;}
91 virtual Bool_t CheckTrigger();
92//
93 AliGenHijing & operator=(const AliGenHijing & rhs);
94 protected:
95 Bool_t SelectFlavor(Int_t pid);
96 void MakeHeader();
97
98 protected:
99 TString fFrame; // Reference frame
100 TString fProjectile; // Projectile
101 TString fTarget; // Target
102 Int_t fAProjectile; // Projectile A
103 Int_t fZProjectile; // Projectile Z
104 Int_t fATarget; // Target A
105 Int_t fZTarget; // Target Z
106 Float_t fMinImpactParam; // minimum impact parameter
107 Float_t fMaxImpactParam; // maximum impact parameter
108 Int_t fKeep; // Flag to keep full event information
109 Int_t fQuench; // Flag to switch on jet quenching
110 Int_t fShadowing; // Flag to switch on nuclear effects on parton distribution function
111 Int_t fDecaysOff; // Flag to turn off decays of pi0, K_s, D, Lambda, sigma
112 Int_t fTrigger; // Trigger type
113 Int_t fEvaluate; // Evaluate total and partial cross-sections
114 Int_t fSelectAll; // Flag to write the full event
115 Int_t fFlavor; // Selected particle flavor 4: charm+beauty 5: beauty
116 Float_t fEnergyCMS; // Centre of mass energy
117 Float_t fKineBias; // Bias from kinematic selection
118 Int_t fTrials; // Number of trials
119 TArrayI fParentSelect; // Parent particles to be selected
120 TArrayI fChildSelect; // Decay products to be selected
121 Float_t fXsection; // Cross-section
122 THijing *fHijing; // Hijing
123 Float_t fPtHardMin; // lower pT-hard cut
124 Float_t fPtHardMax; // higher pT-hard cut
125 Int_t fSpectators; // put spectators on stack
126 TGraph* fDsigmaDb; // dSigma/db for the system
127 TGraph* fDnDb; // dNBinaryCollisions/db
128 Float_t fPtMinJet; // Minimum Pt of triggered Jet
129 Float_t fEtaMinJet; // Minimum eta of triggered Jet
130 Float_t fEtaMaxJet; // Maximum eta of triggered Jet
131 Float_t fPhiMinJet; // At least one of triggered Jets must be in this
132 Float_t fPhiMaxJet; // phi range
133 Int_t fRadiation; // Flag to switch on/off initial and final state radiation
134 Int_t fSimpleJet; // Flag to produce simple tiggered jet topology
135 Int_t fNoGammas; // Don't write gammas if flag "on"
136
137// ZDC proposal (by Chiara) to store num. of SPECTATORS protons and neutrons
138 Int_t fSpecn; // Num. of spectator neutrons
139 Int_t fSpecp; // Num. of spectator protons
140 Int_t fLHC; // Assume LHC as lab frame
141 TClonesArray* fParticles; // Particle List
142
143 private:
144 // adjust the weight from kinematic cuts
145 void AdjustWeights();
146 // check seleted daughters
147 Bool_t DaughtersSelection(TParticle* iparticle);
148 // check if stable
149 Bool_t Stable(TParticle* particle);
150
151 ClassDef(AliGenHijing,4) // AliGenerator interface to Hijing
152};
153#endif
154
155
156
157
158