FinishRun added
[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.;}
9dcc0beb 38 virtual void SetImpactParameterRange(Float_t bmin=0., Float_t bmax=1.)
1a52e0ed 39 {fMinImpactParam=bmin; fMaxImpactParam=bmax;}
e67e0fb7 40 virtual void SetProcess(DpmProcess_t iproc) {fProcess = iproc;}
1a52e0ed 41 virtual void SetCentral(Int_t icentr=-2) {fICentr = icentr;}
1a52e0ed 42 virtual void SetFlavor(Int_t flag=0) {fFlavor = flag;}
1a52e0ed 43 virtual void SetSelectAll(Int_t flag=0) {fSelectAll = flag;}
44 virtual void SetSpectators(Int_t spects=1) {fSpectators = spects;}
45 virtual void SetBoostLHC(Int_t flag=0) {fLHC = flag;}
f97d4a8e 46 virtual void SetPi0Decay(Int_t iPi0) {fPi0Decay = iPi0;}
1a52e0ed 47 virtual Float_t GetEnergyCMS() {return fEnergyCMS;}
48 virtual void GetProjectile(Int_t& a, Int_t& z)
49 {a = fAProjectile; z = fZProjectile;}
50 virtual void GetTarget(Int_t& a, Int_t& z)
51 {a = fATarget; z = fZTarget;}
52 virtual void GetImpactParameterRange(Float_t& bmin, Float_t& bmax)
53 {bmin = fMinImpactParam; bmax = fMaxImpactParam;}
ba758f5a 54 virtual Int_t GetSpectators() {return fSpectators;}
55 virtual Int_t GetFlavor() {return fFlavor;}
1a52e0ed 56
1a52e0ed 57 virtual void SetGenImpPar(Float_t bValue) {fGenImpPar=bValue;}
58 virtual Float_t GetGenImpPar() {return fGenImpPar;}
59
1a52e0ed 60 AliGenDPMjet & operator=(const AliGenDPMjet & rhs);
9dcc0beb 61 void AddHeader(AliGenEventHeader* header);
1a52e0ed 62
63 protected:
64 Bool_t SelectFlavor(Int_t pid);
65 void MakeHeader();
66
67 protected:
1a52e0ed 68 Float_t fBeamEn; // beam energy
69 Float_t fEnergyCMS; // Centre of mass energy
70 Float_t fMinImpactParam; // minimum impact parameter
71 Float_t fMaxImpactParam; // maximum impact parameter
72 Int_t fICentr; // Flag to force central production
1a52e0ed 73 Int_t fSelectAll; // Flag to write the full event
74 Int_t fFlavor; // Selected particle flavor 4: charm+beauty 5: beauty
75 Int_t fTrials; // Number of trials
1a52e0ed 76 Int_t fSpectators; // put spectators on stack
77 Int_t fSpecn; // Num. of spectator neutrons
78 Int_t fSpecp; // Num. of spectator protons
79 TDPMjet *fDPMjet; // DPMjet
1a52e0ed 80 Int_t fNoGammas; // Don't write gammas if flag "on"
81 Int_t fLHC; // Assume LHC as lab frame
f97d4a8e 82 Int_t fPi0Decay; // Flag for pi0 decays
1a52e0ed 83 Float_t fGenImpPar; // GeneratedImpactParameter
e67e0fb7 84 DpmProcess_t fProcess; // Process type
1a52e0ed 85
86 private:
87 // adjust the weight from kinematic cuts
88 void AdjustWeights();
89 // check seleted daughters
90 Bool_t DaughtersSelection(TParticle* iparticle);
91 // check if stable
92 Bool_t Stable(TParticle* particle);
93
94 ClassDef(AliGenDPMjet,1) // AliGenerator interface to DPMJET
95};
96#endif
97
98
99
100
101