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 | |
6d4dd317 |
8 | // |
2ad38d56 |
9 | // Class to generate particles from using parametrized pT and y distributions. |
6d4dd317 |
10 | // Distributions are obtained from pointer to object of type AliGenLib. |
11 | // (For example AliGenMUONlib) |
12 | // |
13 | // andreas.morsch@cern.ch |
14 | // |
15 | |
fff02fee |
16 | #include "AliGenMC.h" |
f87cfe57 |
17 | |
18 | class AliPythia; |
19 | class TParticle; |
b22ee262 |
20 | class AliGenLib; |
f87cfe57 |
21 | class TF1; |
22 | |
34f60c01 |
23 | typedef enum { kAnalog, kNonAnalog} Weighting_t; |
fe4da5cc |
24 | //------------------------------------------------------------- |
fff02fee |
25 | class AliGenParam : public AliGenMC |
fe4da5cc |
26 | { |
6078e216 |
27 | public: |
34825e2b |
28 | AliGenParam(); |
f2c13e03 |
29 | AliGenParam(Int_t npart, const AliGenLib * Library, Int_t param, const char* tname = 0); |
4725e0d2 |
30 | AliGenParam(Int_t npart, Int_t param, const char* tname = 0, const char* name = 0); |
34f60c01 |
31 | AliGenParam(Int_t npart, Int_t param, |
75e0cc59 |
32 | Double_t (*PtPara)(const Double_t*, const Double_t*), |
33 | Double_t (*YPara )(const Double_t*, const Double_t*), |
6078e216 |
34 | Double_t (*V2Para)(const Double_t*, const Double_t*), |
65fb704d |
35 | Int_t (*IpPara)(TRandom*) ); |
f87cfe57 |
36 | |
34825e2b |
37 | virtual ~AliGenParam(); |
d5615b85 |
38 | virtual void GenerateN(Int_t ntimes); |
34825e2b |
39 | virtual void Generate(); |
40 | virtual void Init(); |
41 | // select particle type |
34f60c01 |
42 | virtual void SetParam(Int_t param) {fParam = param;} |
18e09c20 |
43 | //Setting the flag for Background transportation while using SetForceDecay() |
44 | void SetSelectAll(Bool_t selectall) {fSelectAll = selectall;} |
34825e2b |
45 | // force decay type |
34f60c01 |
46 | virtual void SetWeighting(Weighting_t flag = kAnalog) {fAnalog = flag;} |
34f60c01 |
47 | virtual void SetDeltaPt(Float_t delta=0.01) {fDeltaPt = delta;} |
a595093b |
48 | virtual void SetDecayer(AliDecayer* decayer) {fDecayer = decayer;} |
4ae1c9f0 |
49 | virtual void SetForceGammaConversion(Bool_t force=kTRUE) {fForceConv = force;} |
309923cb |
50 | virtual void SetKeepParent(Bool_t keep=kTRUE){fKeepParent= keep;} //Store parent even if it does not have childs within cuts |
51 | virtual void SetKeepIfOneChildSelected(Bool_t keep=kTRUE){fKeepIfOneChildSelected = keep;} //Accept parent and child even if other children are not within cut. |
52 | |
dc1d768c |
53 | virtual void Draw(const char * opt); |
2ad38d56 |
54 | TF1 * GetPt() { return fPtPara;} |
55 | TF1 * GetY() {return fYPara;} |
56 | Float_t GetRelativeArea(Float_t ptMin, Float_t ptMax, Float_t yMin, Float_t yMax, Float_t phiMin, Float_t phiMax); |
57 | |
4ae1c9f0 |
58 | static TVector3 OrthogonalVector(TVector3 &inVec); |
71443190 |
59 | static void RotateVector(Double_t *pin, Double_t *pout, Double_t costheta, Double_t sintheta, |
60 | Double_t cosphi, Double_t sinphi); |
71443190 |
61 | static double ScreenFunction1(double d); |
62 | static double ScreenFunction2(double d); |
63 | double RandomEnergyFraction(double Z, double E); |
64 | double RandomPolarAngle(); |
65 | double RandomMass(Double_t mh); |
66 | Int_t VirtualGammaPairProduction(TClonesArray *particles, Int_t nPart); |
4ae1c9f0 |
67 | Int_t ForceGammaConversion(TClonesArray *particles, Int_t nPart); |
e3300604 |
68 | virtual void SetSeed(UInt_t /*seed*/) {;} |
4ae1c9f0 |
69 | |
6078e216 |
70 | protected: |
75e0cc59 |
71 | Double_t (*fPtParaFunc)(const Double_t*, const Double_t*); //! Pointer to Pt parametrisation function |
72 | Double_t (*fYParaFunc )(const Double_t*, const Double_t*); //! Pointer to Y parametrisation function |
65fb704d |
73 | Int_t (*fIpParaFunc )(TRandom*); //! Pointer to particle type parametrisation function |
6078e216 |
74 | Double_t (*fV2ParaFunc )(const Double_t*, const Double_t*);//! Pointer to V2 parametrisation function |
c5c3e4b0 |
75 | TF1* fPtPara; // Transverse momentum parameterisation |
76 | TF1* fYPara; // Rapidity parameterisation |
6078e216 |
77 | TF1* fV2Para; // v2 parametrization |
78 | TF1* fdNdPhi; // Phi distribution depending on v2 |
34f60c01 |
79 | Int_t fParam; // Parameterisation type |
4d9d5cbd |
80 | Float_t fdNdy0; // central multiplicity per event |
81 | Float_t fYWgt; // Y-weight |
82 | Float_t fPtWgt; // Pt-weight |
4d9d5cbd |
83 | Float_t fBias; // Biasing factor |
84 | Int_t fTrials; // Number of trials |
34825e2b |
85 | Float_t fDeltaPt; // pT sampling in steps of fDeltaPt |
18e09c20 |
86 | Bool_t fSelectAll; // Flag for transportation of Background while using SetForceDecay() |
6c24ee7c |
87 | AliDecayer *fDecayer; // ! Pointer to pythia object for decays |
4ae1c9f0 |
88 | Bool_t fForceConv; // |
309923cb |
89 | Bool_t fKeepParent; // Store parent even if it does not have childs within cuts |
90 | Bool_t fKeepIfOneChildSelected; //Accept parent and child even if other children are not within cut. |
93a2041b |
91 | |
6078e216 |
92 | private: |
93a2041b |
93 | AliGenParam(const AliGenParam &Param); |
94 | AliGenParam & operator=(const AliGenParam & rhs); |
95 | |
18e09c20 |
96 | ClassDef(AliGenParam, 2) // Generator using parameterised pt- and y-distribution |
fe4da5cc |
97 | }; |
98 | #endif |
99 | |
100 | |
101 | |
102 | |
103 | |
104 | |
105 | |
106 | |
107 | |
108 | |