AliGenMC base class for AliGenParam and AliGenPythia commonalities.
[u/mrichter/AliRoot.git] / EVGEN / AliGenPythia.h
CommitLineData
a0134e39 1#ifndef ALIGENPYTHIA_H
2#define ALIGENPYTHIA_H
3da30618 3/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4 * See cxx source for full Copyright notice */
5
6/* $Id$ */
7
f87cfe57 8
fe4da5cc 9#include "AliGenerator.h"
f1a48a38 10#include "AliPythia.h"
11#include "AliDecayer.h"
f87cfe57 12#include <TArrayI.h>
13
14class AliPythia;
15class TParticle;
fe4da5cc 16
17class AliGenPythia : public AliGenerator
18{
fe4da5cc 19 public:
20 AliGenPythia();
21 AliGenPythia(Int_t npart);
f87cfe57 22 AliGenPythia(const AliGenPythia &Pythia);
fe4da5cc 23 virtual ~AliGenPythia();
24 virtual void Generate();
25 virtual void Init();
11dfaf8d 26 virtual void SetEventListRange(Int_t eventFirst=-1, Int_t eventLast=-1);
fe4da5cc 27 // select process type
f1a48a38 28 virtual void SetProcess(Process_t proc = kPyCharm) {fProcess = proc;}
fe4da5cc 29 // select structure function
f1a48a38 30 virtual void SetStrucFunc(StrucFunc_t func = kGRV_HO) {fStrucFunc = func;}
fe4da5cc 31 // select pt of hard scattering
f1a48a38 32 virtual void SetPtHard(Float_t ptmin = 0, Float_t ptmax = 1.e10)
33 {fPtHardMin = ptmin; fPtHardMax = ptmax; }
fe4da5cc 34 // set centre of mass energy
f1a48a38 35 virtual void SetEnergyCMS(Float_t energy = 5500) {fEnergyCMS = energy;}
fe4da5cc 36 // force decay type
f1a48a38 37 virtual void SetForceDecay(Decay_t decay = kSemiMuonic) {fForceDecay = decay;}
811826d8 38 // treat protons as inside nuclei
39 virtual void SetNuclei(Int_t a1, Int_t a2);
fe4da5cc 40 // get cross section of process
11dfaf8d 41 virtual Float_t GetXsection() {return fXsection;}
a8228d85 42 // Check PDG code
f1a48a38 43 virtual Int_t CheckPDGCode(Int_t pdgcode);
9faa0850 44 virtual void FinishRun();
45
f87cfe57 46 // Assignment Operator
47 AliGenPythia & operator=(const AliGenPythia & rhs);
a0134e39 48 protected:
49 Process_t fProcess; // Process type
50 StrucFunc_t fStrucFunc; // Structure Function
51 Decay_t fForceDecay; // Decay channel are forced
52 Float_t fEnergyCMS; // Centre of mass energy
53 Float_t fKineBias; // Bias from kinematic selection
54 Int_t fTrials; // Number of trials
55 TArrayI fParentSelect; // Parent particles to be selected
56 TArrayI fChildSelect; // Decay products to be selected
57 Float_t fXsection; // Cross-section
a8a6107b 58 AliPythia *fPythia; //! Pythia
a0134e39 59 Float_t fPtHardMin; // lower pT-hard cut
60 Float_t fPtHardMax; // higher pT-hard cut
811826d8 61 Int_t fNucA1; // mass number nucleus side 1
62 Int_t fNucA2; // mass number nucleus side 2
f1a48a38 63 Bool_t fFullEvent; // Write Full event if true
c78d7f8c 64 AliDecayer *fDecayer; // ! pointer to the decayer instance
11dfaf8d 65 Int_t fDebugEventFirst; // First event to debug
66 Int_t fDebugEventLast; // Last event to debug
a0134e39 67 private:
68 // check if particle is selected as parent particle
69 Bool_t ParentSelected(Int_t ip);
70 // check if particle is selected as child particle
71 Bool_t ChildSelected(Int_t ip);
72 // all kinematic selection cuts go here
73 Bool_t KinematicSelection(TParticle *particle);
74 // adjust the weight from kinematic cuts
75 void AdjustWeights();
76
f87cfe57 77 ClassDef(AliGenPythia,1) // AliGenerator interface to Pythia
fe4da5cc 78};
79#endif
80
81
82
83
84