fFilename !-ed
[u/mrichter/AliRoot.git] / EVGEN / AliGenParam.h
CommitLineData
34825e2b 1#ifndef ALIGENPARAM_H
2#define ALIGENPARAM_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#include "AliGenerator.h"
34825e2b 9#include "GenTypeDefs.h"
5c3fd7ea 10#include <TArrayI.h>
f87cfe57 11
12class AliPythia;
13class TParticle;
14class TF1;
15
fe4da5cc 16//-------------------------------------------------------------
fe4da5cc 17class AliGenParam : public AliGenerator
18{
34825e2b 19 public:
20 AliGenParam();
21 AliGenParam(Int_t npart, Param_t param);
22 AliGenParam(Int_t npart, Param_t param,
23 Double_t (*PtPara)(Double_t*, Double_t*),
24 Double_t (*YPara )(Double_t*, Double_t*),
f87cfe57 25 Int_t (*IpPara)() );
26 AliGenParam(const AliGenParam &Param);
27
34825e2b 28 virtual ~AliGenParam();
29 virtual void Generate();
30 virtual void Init();
31 // select particle type
32 virtual void SetParam(Param_t param=jpsi_p) {fParam=param;}
33 // force decay type
34 virtual void SetForceDecay(Decay_t decay=dimuon) {fForceDecay=decay;}
35 virtual void SetWeighting(Weighting_t flag=analog) {fAnalog=flag;}
36 virtual void SetCutOnChild(Int_t flag=0) {fCutOnChild=flag;}
37 virtual void SetChildMomentumRange(Float_t pmin=0, Float_t pmax=1.e10)
38 {fChildPMin = pmin; fChildPMax = pmax;}
39 virtual void SetChildPtRange(Float_t ptmin=0, Float_t ptmax=20.)
40 {fChildPtMin = ptmin; fChildPtMax = ptmax;}
41 virtual void SetChildPhiRange(Float_t phimin=-180., Float_t phimax=180)
42 {fChildPhiMin = TMath::Pi()*phimin/180;
43 fChildPhiMax = TMath::Pi()*phimax/180;}
44 virtual void SetChildThetaRange(Float_t thetamin=0, Float_t thetamax=180)
45 {fChildThetaMin = TMath::Pi()*thetamin/180;
46 fChildThetaMax = TMath::Pi()*thetamax/180;}
47 virtual void SetDeltaPt(Float_t delta=0.01) {fDeltaPt=delta;}
48
f87cfe57 49 AliGenParam & operator=(const AliGenParam & rhs);
34825e2b 50 protected:
51 Double_t (*fPtParaFunc)(Double_t*, Double_t*); //! Pointer to Pt parametrisation function
52 Double_t (*fYParaFunc )(Double_t*, Double_t*); //! Pointer to Y parametrisation function
53 Int_t (*fIpParaFunc )(); //! Pointer to particle type parametrisation function
54 TF1* fPtPara; // Transverse momentum parameterisation
55 TF1* fYPara; // Rapidity parameterisation
4d9d5cbd 56 Param_t fParam; // Parameterisation type
57 Float_t fdNdy0; // central multiplicity per event
58 Float_t fYWgt; // Y-weight
59 Float_t fPtWgt; // Pt-weight
60 Weighting_t fAnalog; // Flag for anolog or pt-weighted generation
61 Float_t fBias; // Biasing factor
62 Int_t fTrials; // Number of trials
63 Decay_t fForceDecay; // Decay channel forced
64 Int_t fCutOnChild; // Cuts on decay products (children) are enabled/disabled
65 Float_t fChildPtMin; // Children minimum pT
66 Float_t fChildPtMax; // Children maximum pT
67 Float_t fChildPMin; // Children minimum p
68 Float_t fChildPMax; // Children maximum p
69 Float_t fChildPhiMin; // Children minimum phi
70 Float_t fChildPhiMax; // Children maximum phi
71 Float_t fChildThetaMin;// Children minimum theta
72 Float_t fChildThetaMax;// Children maximum theta
34825e2b 73 Float_t fDeltaPt; // pT sampling in steps of fDeltaPt
74 TArrayI fChildSelect; // Children to be selected from decay products
75 AliPythia *fPythia; // Pointer to pythia object for decays
fe4da5cc 76 private:
77 // check if particle is selected as child
78 Bool_t ChildSelected(Int_t ip);
79 // all kinematic selection goes here
1578254f 80 Bool_t KinematicSelection(TParticle *particle);
34825e2b 81
b7601ac4 82 ClassDef(AliGenParam,1) // Generator using parameterised pt- and y-distribution
fe4da5cc 83};
84#endif
85
86
87
88
89
90
91
92
93
94