Class moved from AliGenCocktail.h
[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
fe4da5cc 8/////////////////////////////////////////////////////////
9// Manager and hits classes for set:MUON version 0 //
10/////////////////////////////////////////////////////////
11#include "AliGenerator.h"
12#include "TNamed.h"
13#include "TF1.h"
14#include "TArrayF.h"
15#include "TTree.h"
16#include "AliPythia.h"
17#include "TArrayI.h"
1578254f 18#include "TParticle.h"
fe4da5cc 19
20class AliGenPythia : public AliGenerator
21{
fe4da5cc 22 public:
23 AliGenPythia();
24 AliGenPythia(Int_t npart);
25 virtual ~AliGenPythia();
26 virtual void Generate();
27 virtual void Init();
28 // select process type
29 virtual void SetProcess(Process_t proc=charm) {fProcess=proc;}
30 // select structure function
31 virtual void SetStrucFunc(StrucFunc_t func=GRV_HO) {fStrucFunc=func;}
32 // select pt of hard scattering
33 virtual void SetPtHard(Float_t ptmin=0, Float_t ptmax=1.e10)
34 {fPtHardMin=ptmin; fPtHardMax=ptmax; }
35 // set centre of mass energy
36 virtual void SetEnergyCMS(Float_t energy=5500) {fEnergyCMS=energy;}
37 // force decay type
886b6f73 38 virtual void SetForceDecay(Decay_t decay=semimuonic) {fForceDecay=decay;}
fe4da5cc 39 // get cross section of process
40 virtual Float_t GetXsection() {return fXsection;}
a8228d85 41 // Check PDG code
42 virtual Int_t CheckPDGCode(Int_t pdgcode);
43
a0134e39 44 protected:
45 Process_t fProcess; // Process type
46 StrucFunc_t fStrucFunc; // Structure Function
47 Decay_t fForceDecay; // Decay channel are forced
48 Float_t fEnergyCMS; // Centre of mass energy
49 Float_t fKineBias; // Bias from kinematic selection
50 Int_t fTrials; // Number of trials
51 TArrayI fParentSelect; // Parent particles to be selected
52 TArrayI fChildSelect; // Decay products to be selected
53 Float_t fXsection; // Cross-section
54 AliPythia *fPythia; // Pythia
55 Float_t fPtHardMin; // lower pT-hard cut
56 Float_t fPtHardMax; // higher pT-hard cut
57
58 private:
59 // check if particle is selected as parent particle
60 Bool_t ParentSelected(Int_t ip);
61 // check if particle is selected as child particle
62 Bool_t ChildSelected(Int_t ip);
63 // all kinematic selection cuts go here
64 Bool_t KinematicSelection(TParticle *particle);
65 // adjust the weight from kinematic cuts
66 void AdjustWeights();
67
fe4da5cc 68 ClassDef(AliGenPythia,1)
69};
70#endif
71
72
73
74
75