event listing removed
[u/mrichter/AliRoot.git] / THerwig / AliGenHerwig.h
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
19 class THerwig6;
20 class AliGenHerwigEventHeader;
21 class TArrayI;
22 class TParticle;
23 class TClonesArray;
24
25
26 class 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    FinishRun();
55     virtual void    FinishRunJimmy();
56     virtual void    SetEnSoft(Double_t e) {fEnSoft=e;}
57
58     virtual void    SetHardProcessFile(TString filename) {fFileName=TString(filename);};
59     virtual Bool_t CheckParton(const TParticle* parton1, const TParticle* parton2);
60
61     virtual void         GetPartonEtaRange(Float_t& etamin, Float_t& etamax) const
62         {etamin = fEtaMinParton; etamax = fEtaMaxParton;}
63     virtual void         GetPartonPhiRange(Float_t& phimin, Float_t& phimax) const
64         {phimin = fPhiMinParton*180./TMath::Pi(); phimax = fPhiMaxParton*180/TMath::Pi();}
65     virtual void         GetGammaEtaRange(Float_t& etamin, Float_t& etamax) const
66         {etamin = fEtaMinGamma; etamax = fEtaMaxGamma;}
67     virtual void         GetGammaPhiRange(Float_t& phimin, Float_t& phimax) const
68         {phimin = fPhiMinGamma*180./TMath::Pi(); phimax = fPhiMaxGamma*180./TMath::Pi();}
69
70     virtual void    SetPartonEtaRange(Float_t etamin = -20., Float_t etamax = 20.)
71         {fEtaMinParton = etamin; fEtaMaxParton = etamax;}
72     // Phi range for jet trigger
73     virtual void    SetPartonPhiRange(Float_t phimin = 0., Float_t phimax = 360.)
74         {fPhiMinParton = TMath::Pi()*phimin/180.; fPhiMaxParton = TMath::Pi()*phimax/180.;}
75     // Eta range for gamma trigger 
76     virtual void    SetGammaEtaRange(Float_t etamin = -20., Float_t etamax = 20.)
77         {fEtaMinGamma = etamin; fEtaMaxGamma = etamax;}
78     // Phi range for gamma trigger
79     virtual void    SetGammaPhiRange(Float_t phimin = 0., Float_t phimax = 360.)
80         {fPhiMinGamma = TMath::Pi()*phimin/180.; fPhiMaxGamma = TMath::Pi()*phimax/180.;}
81
82  protected:
83     Bool_t SelectFlavor(Int_t pid) const;
84     void   MakeHeader();    
85  protected:
86     TString     fAutPDF;         // PDF group
87     Int_t       fModPDF;         // PDF set
88     StrucFunc_t fStrucFunc;      // Structure Function
89     Int_t       fKeep;           // Flag to keep full event information
90     Int_t       fDecaysOff;      // Flag to turn off decays of pi0, K_s, D, Lambda, sigma
91     Int_t       fTrigger;        // Trigger type
92     Int_t       fSelectAll;      // Flag to write the full event
93     Int_t       fFlavor;         // Selected particle flavor 4: charm+beauty 5: beauty
94     Float_t     fMomentum1;      // Momentum of projectile
95     Float_t     fMomentum2;      // Momentum of target
96     Float_t     fKineBias;       // Bias from kinematic selection
97     Int_t       fTrials;         // Number of trials
98     Float_t     fXsection;       // Cross-section
99     THerwig6    *fHerwig;        // Herwig
100     Int_t       fProcess;        // Process number
101     Double_t    fPtHardMin;      // higher pT-hard cut
102     Double_t    fPtHardMax;      // lower pT-hard cut
103     Double_t    fPtRMS;          // intrinsic pt of incoming hadrons
104     Int_t       fMaxPr;          // maximum number of events to print out
105     Int_t       fMaxErrors;      // maximum number of errors allowed
106     Double_t    fEnSoft;          // change on soft energy distribution
107     TString     fFileName;       //!Name of file to read from hard scattering
108     Float_t     fEtaMinParton;         //Minimum eta of parton shower
109     Float_t     fEtaMaxParton;         //Maximum eta of parton shower
110     Float_t     fPhiMinParton;         //Minimum phi of parton shower
111     Float_t     fPhiMaxParton;         //Maximum phi of parton shower
112     Float_t     fEtaMinGamma;       // Minimum eta of triggered gamma
113     Float_t     fEtaMaxGamma;       // Maximum eta of triggered gamma
114     Float_t     fPhiMinGamma;       // Minimum phi of triggered gamma
115     Float_t     fPhiMaxGamma;       // Maximum phi of triggered gamma
116     AliGenHerwigEventHeader* fHeader;  //! Event header
117  private:
118     AliGenHerwig(const AliGenHerwig &Herwig);
119     AliGenHerwig &  operator=(const AliGenHerwig & rhs);
120
121     // check if particle is selected as parent particle
122     Bool_t ParentSelected(Int_t ip);
123     // check if particle is selected as child particle
124     Bool_t ChildSelected(Int_t ip);
125     // adjust the weight from kinematic cuts
126     void   AdjustWeights();
127     // check seleted daughters
128     Bool_t DaughtersSelection(const TParticle* iparticle, const TClonesArray* particles);
129     // check if stable
130     Bool_t Stable(const TParticle*  particle) const;
131
132     void InitPDF();
133
134     ClassDef(AliGenHerwig,2) // AliGenerator interface to Herwig
135 };
136 #endif
137
138
139
140
141