fix compilation
[u/mrichter/AliRoot.git] / TDPMjet / AliGenDPMjet.h
CommitLineData
1a52e0ed 1#ifndef ALIGENDPMJET_H
2#define ALIGENDPMJET_H
3/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4 * See cxx source for full Copyright notice */
5
6// Generator using DPMJET as an external generator
7// The main DPMJET options are accessable for the user through this interface.
8
9#include "AliGenMC.h"
e67e0fb7 10#include "TDPMjet.h"
1a52e0ed 11#include <TString.h>
12#include <TArrayI.h>
13
14class TDPMjet;
15class TArrayI;
16class TParticle;
17class TClonesArray;
18class TGraph;
9dcc0beb 19class AliGenEventHeader;
20class AliStack;
21class AliRunLoader;
22class AliGenDPMjetEventHeader;
23
24
1a52e0ed 25
26class AliGenDPMjet : public AliGenMC
27{
28
29 public:
30 AliGenDPMjet();
31 AliGenDPMjet(Int_t npart);
ba758f5a 32 AliGenDPMjet(const AliGenDPMjet &Dpmjet);
1a52e0ed 33 virtual ~AliGenDPMjet();
34 virtual void Generate();
35 virtual void Init();
4e5c2fd7 36 virtual void FinishRun();
e67e0fb7 37 virtual void SetEnergyCMS(Float_t energy = 14000.) {fEnergyCMS = energy; fBeamEn = energy / 2.;}
5fed2e29 38 virtual void SetpBeamEnergy(Float_t benergy = 14000.) {fBeamEn = benergy;}
9dcc0beb 39 virtual void SetImpactParameterRange(Float_t bmin=0., Float_t bmax=1.)
1a52e0ed 40 {fMinImpactParam=bmin; fMaxImpactParam=bmax;}
e67e0fb7 41 virtual void SetProcess(DpmProcess_t iproc) {fProcess = iproc;}
1a52e0ed 42 virtual void SetCentral(Int_t icentr=-2) {fICentr = icentr;}
1a52e0ed 43 virtual void SetFlavor(Int_t flag=0) {fFlavor = flag;}
1a52e0ed 44 virtual void SetSelectAll(Int_t flag=0) {fSelectAll = flag;}
45 virtual void SetSpectators(Int_t spects=1) {fSpectators = spects;}
46 virtual void SetBoostLHC(Int_t flag=0) {fLHC = flag;}
f97d4a8e 47 virtual void SetPi0Decay(Int_t iPi0) {fPi0Decay = iPi0;}
717765ce 48 virtual void SetDecayAll(Int_t iDec) {fDecayAll = iDec;}
1a52e0ed 49 virtual void GetImpactParameterRange(Float_t& bmin, Float_t& bmax)
50 {bmin = fMinImpactParam; bmax = fMaxImpactParam;}
ba758f5a 51 virtual Int_t GetSpectators() {return fSpectators;}
52 virtual Int_t GetFlavor() {return fFlavor;}
1a52e0ed 53
1a52e0ed 54 virtual void SetGenImpPar(Float_t bValue) {fGenImpPar=bValue;}
55 virtual Float_t GetGenImpPar() {return fGenImpPar;}
3b50c39e 56 virtual void SetTriggerChargedMultiplicity(Int_t multiplicity,
57 Float_t etamax = 0, Float_t ptmin = -1.)
58 {fTriggerMultiplicity = multiplicity; fTriggerMultiplicityEta = etamax;
59 fTriggerMultiplicityPtMin = ptmin;}
60
1a52e0ed 61 AliGenDPMjet & operator=(const AliGenDPMjet & rhs);
9dcc0beb 62 void AddHeader(AliGenEventHeader* header);
1a52e0ed 63
19896767 64 void SetTuneForDiff(Bool_t a=kTRUE) {fkTuneForDiff=a;}
65
5fed2e29 66 virtual void SetFragmentProd(Bool_t val) {fFragmentation = val;}
67
1a52e0ed 68 protected:
69 Bool_t SelectFlavor(Int_t pid);
70 void MakeHeader();
71
72 protected:
1a52e0ed 73 Float_t fBeamEn; // beam energy
1a52e0ed 74 Float_t fMinImpactParam; // minimum impact parameter
75 Float_t fMaxImpactParam; // maximum impact parameter
76 Int_t fICentr; // Flag to force central production
1a52e0ed 77 Int_t fSelectAll; // Flag to write the full event
78 Int_t fFlavor; // Selected particle flavor 4: charm+beauty 5: beauty
79 Int_t fTrials; // Number of trials
1a52e0ed 80 Int_t fSpectators; // put spectators on stack
81 Int_t fSpecn; // Num. of spectator neutrons
82 Int_t fSpecp; // Num. of spectator protons
83 TDPMjet *fDPMjet; // DPMjet
1a52e0ed 84 Int_t fNoGammas; // Don't write gammas if flag "on"
85 Int_t fLHC; // Assume LHC as lab frame
f97d4a8e 86 Int_t fPi0Decay; // Flag for pi0 decays
717765ce 87 Int_t fDecayAll; // Flag to switch on long-lived particle decays
1a52e0ed 88 Float_t fGenImpPar; // GeneratedImpactParameter
e67e0fb7 89 DpmProcess_t fProcess; // Process type
3b50c39e 90 // Multiplicity Trigger
91 Int_t fTriggerMultiplicity; // Triggered multiplicity
92 Float_t fTriggerMultiplicityEta; // Triggered multiplicity eta cut
93 Float_t fTriggerMultiplicityPtMin; // Triggered multiplicity min pt
94
19896767 95 Bool_t fkTuneForDiff; // Phojet tune
96 Int_t fProcDiff;
5fed2e29 97
98 Bool_t fFragmentation; // Allows evaporation and fragments production
19896767 99
1a52e0ed 100 private:
101 // adjust the weight from kinematic cuts
102 void AdjustWeights();
103 // check seleted daughters
104 Bool_t DaughtersSelection(TParticle* iparticle);
105 // check if stable
106 Bool_t Stable(TParticle* particle);
19896767 107
108 Bool_t CheckDiffraction();
c5dfa3e4 109 Bool_t GetWeightsDiffraction(Double_t M, Double_t &Mmin, Double_t &Mmax,
110 Double_t &wSD, Double_t &wDD, Double_t &wND);
19896767 111
5fed2e29 112 ClassDef(AliGenDPMjet,5) // AliGenerator interface to DPMJET
1a52e0ed 113};
114#endif
115
116
117
118
119