Correct use of ROOT_INCLUDE_DIR
[u/mrichter/AliRoot.git] / THerwig / AliGenHerwig.h
... / ...
CommitLineData
1#ifndef ALIGENHERWIG_H
2#define ALIGENHERWIG_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 HERWIG as an external generator
9// The main HERWIG options are accessable for the user through this interface.
10// Author Andreas Morsch
11// andreas.morsch@cern.ch
12
13#include "AliGenMC.h"
14#include <TString.h>
15#include <TArrayI.h>
16#include <AliRndm.h>
17#include <AliStructFuncType.h>
18
19class THerwig6;
20class AliGenHerwigEventHeader;
21class TArrayI;
22class TParticle;
23class TClonesArray;
24
25
26class AliGenHerwig : public AliGenMC
27
28{
29 enum {kNoTrigger, kHardProcesses, kDirectPhotons};
30 enum {kHeJets = 1500, kHeDirectGamma = 1800};
31 public:
32 AliGenHerwig();
33 AliGenHerwig(Int_t npart);
34 virtual ~AliGenHerwig();
35 virtual void Generate();
36 virtual void Init();
37 virtual void InitJimmy();
38 // set centre of mass energy
39 virtual void SetBeamMomenta(Float_t p1=7000., Float_t p2=7000.)
40 {fMomentum1 = p1; fMomentum2 = p2;}
41 virtual void SetProcess(Int_t proc) {fProcess = proc;}
42 virtual void KeepFullEvent();
43 virtual void SetDecaysOff(Int_t flag=1) {fDecaysOff = flag;}
44 virtual void SetTrigger(Int_t flag=kNoTrigger) {fTrigger = flag;}
45 virtual void SetFlavor(Int_t flag=0) {fFlavor = flag;}
46 virtual void SetSelectAll(Int_t flag=0) {fSelectAll = flag;}
47 virtual void SetStrucFunc(StrucFunc_t func = kCTEQ5L)
48 {fStrucFunc = func;}
49 virtual void SetPtHardMin(Double_t pt) {fPtHardMin=pt;}
50 virtual void SetPtHardMax(Double_t pt) {fPtHardMax=pt;}
51 virtual void SetPtRMS(Double_t pt) {fPtRMS=pt;}
52 virtual void SetMaxPr(Int_t i) {fMaxPr=i;}
53 virtual void SetMaxErrors(Int_t i) {fMaxErrors=i;}
54 virtual void SetSeed(UInt_t seed) {GetRandom()->SetSeed(seed);}
55 virtual void FinishRun();
56 virtual void FinishRunJimmy();
57 virtual void SetEnSoft(Double_t e) {fEnSoft=e;}
58
59 virtual void SetHardProcessFile(TString filename) {fFileName=TString(filename);};
60 virtual Bool_t CheckParton(const TParticle* parton1, const TParticle* parton2);
61
62 virtual void GetPartonEtaRange(Float_t& etamin, Float_t& etamax) const
63 {etamin = fEtaMinParton; etamax = fEtaMaxParton;}
64 virtual void GetPartonPhiRange(Float_t& phimin, Float_t& phimax) const
65 {phimin = fPhiMinParton*180./TMath::Pi(); phimax = fPhiMaxParton*180/TMath::Pi();}
66 virtual void GetGammaEtaRange(Float_t& etamin, Float_t& etamax) const
67 {etamin = fEtaMinGamma; etamax = fEtaMaxGamma;}
68 virtual void GetGammaPhiRange(Float_t& phimin, Float_t& phimax) const
69 {phimin = fPhiMinGamma*180./TMath::Pi(); phimax = fPhiMaxGamma*180./TMath::Pi();}
70
71 virtual void SetPartonEtaRange(Float_t etamin = -20., Float_t etamax = 20.)
72 {fEtaMinParton = etamin; fEtaMaxParton = etamax;}
73 // Phi range for jet trigger
74 virtual void SetPartonPhiRange(Float_t phimin = 0., Float_t phimax = 360.)
75 {fPhiMinParton = TMath::Pi()*phimin/180.; fPhiMaxParton = TMath::Pi()*phimax/180.;}
76 // Eta range for gamma trigger
77 virtual void SetGammaEtaRange(Float_t etamin = -20., Float_t etamax = 20.)
78 {fEtaMinGamma = etamin; fEtaMaxGamma = etamax;}
79 // Phi range for gamma trigger
80 virtual void SetGammaPhiRange(Float_t phimin = 0., Float_t phimax = 360.)
81 {fPhiMinGamma = TMath::Pi()*phimin/180.; fPhiMaxGamma = TMath::Pi()*phimax/180.;}
82
83 protected:
84 Bool_t SelectFlavor(Int_t pid) const;
85 void MakeHeader();
86 protected:
87 TString fAutPDF; // PDF group
88 Int_t fModPDF; // PDF set
89 StrucFunc_t fStrucFunc; // Structure Function
90 Int_t fKeep; // Flag to keep full event information
91 Int_t fDecaysOff; // Flag to turn off decays of pi0, K_s, D, Lambda, sigma
92 Int_t fTrigger; // Trigger type
93 Int_t fSelectAll; // Flag to write the full event
94 Int_t fFlavor; // Selected particle flavor 4: charm+beauty 5: beauty
95 Float_t fMomentum1; // Momentum of projectile
96 Float_t fMomentum2; // Momentum of target
97 Float_t fKineBias; // Bias from kinematic selection
98 Int_t fTrials; // Number of trials
99 Float_t fXsection; // Cross-section
100 THerwig6 *fHerwig; // Herwig
101 Int_t fProcess; // Process number
102 Double_t fPtHardMin; // higher pT-hard cut
103 Double_t fPtHardMax; // lower pT-hard cut
104 Double_t fPtRMS; // intrinsic pt of incoming hadrons
105 Int_t fMaxPr; // maximum number of events to print out
106 Int_t fMaxErrors; // maximum number of errors allowed
107 Double_t fEnSoft; // change on soft energy distribution
108 TString fFileName; //!Name of file to read from hard scattering
109 Float_t fEtaMinParton; //Minimum eta of parton shower
110 Float_t fEtaMaxParton; //Maximum eta of parton shower
111 Float_t fPhiMinParton; //Minimum phi of parton shower
112 Float_t fPhiMaxParton; //Maximum phi of parton shower
113 Float_t fEtaMinGamma; // Minimum eta of triggered gamma
114 Float_t fEtaMaxGamma; // Maximum eta of triggered gamma
115 Float_t fPhiMinGamma; // Minimum phi of triggered gamma
116 Float_t fPhiMaxGamma; // Maximum phi of triggered gamma
117 AliGenHerwigEventHeader* fHeader; //! Event header
118 private:
119 AliGenHerwig(const AliGenHerwig &Herwig);
120 AliGenHerwig & operator=(const AliGenHerwig & rhs);
121
122 // check if particle is selected as parent particle
123 Bool_t ParentSelected(Int_t ip);
124 // check if particle is selected as child particle
125 Bool_t ChildSelected(Int_t ip);
126 // adjust the weight from kinematic cuts
127 void AdjustWeights();
128 // check seleted daughters
129 Bool_t DaughtersSelection(const TParticle* iparticle, const TClonesArray* particles);
130 // check if stable
131 Bool_t Stable(const TParticle* particle) const;
132
133 void InitPDF();
134
135 ClassDef(AliGenHerwig,2) // AliGenerator interface to Herwig
136};
137#endif
138
139
140
141
142