Update master to aliroot
[u/mrichter/AliRoot.git] / EVGEN / AliGenParam.h
... / ...
CommitLineData
1#ifndef ALIGENPARAM_H
2#define ALIGENPARAM_H
3/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4 * See cxx source for full Copyright notice */
5
6/* $Id$ */
7
8//
9// Class to generate particles from using parametrized pT and y distributions.
10// Distributions are obtained from pointer to object of type AliGenLib.
11// (For example AliGenMUONlib)
12//
13// andreas.morsch@cern.ch
14//
15
16#include "AliGenMC.h"
17
18class AliPythia;
19class TParticle;
20class AliGenLib;
21class TF1;
22
23typedef enum { kAnalog, kNonAnalog} Weighting_t;
24//-------------------------------------------------------------
25class AliGenParam : public AliGenMC
26{
27public:
28 AliGenParam();
29 AliGenParam(Int_t npart, const AliGenLib * Library, Int_t param, const char* tname = 0);
30 AliGenParam(Int_t npart, Int_t param, const char* tname = 0, const char* name = 0);
31 AliGenParam(Int_t npart, Int_t param,
32 Double_t (*PtPara)(const Double_t*, const Double_t*),
33 Double_t (*YPara )(const Double_t*, const Double_t*),
34 Double_t (*V2Para)(const Double_t*, const Double_t*),
35 Int_t (*IpPara)(TRandom*) );
36
37 virtual ~AliGenParam();
38 virtual void GenerateN(Int_t ntimes);
39 virtual void Generate();
40 virtual void Init();
41 // select particle type
42 virtual void SetParam(Int_t param) {fParam = param;}
43 //Setting the flag for Background transportation while using SetForceDecay()
44 void SetSelectAll(Bool_t selectall) {fSelectAll = selectall;}
45 // force decay type
46 virtual void SetWeighting(Weighting_t flag = kAnalog) {fAnalog = flag;}
47 virtual void SetDeltaPt(Float_t delta=0.01) {fDeltaPt = delta;}
48 virtual void SetDecayer(AliDecayer* decayer) {fDecayer = decayer;}
49 virtual void SetForceGammaConversion(Bool_t force=kTRUE) {fForceConv = force;}
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
53 virtual void Draw(const char * opt);
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
58 static TVector3 OrthogonalVector(TVector3 &inVec);
59 static void RotateVector(Double_t *pin, Double_t *pout, Double_t costheta, Double_t sintheta,
60 Double_t cosphi, Double_t sinphi);
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);
67 Int_t ForceGammaConversion(TClonesArray *particles, Int_t nPart);
68 virtual void SetSeed(UInt_t /*seed*/) {;}
69
70protected:
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
73 Int_t (*fIpParaFunc )(TRandom*); //! Pointer to particle type parametrisation function
74 Double_t (*fV2ParaFunc )(const Double_t*, const Double_t*);//! Pointer to V2 parametrisation function
75 TF1* fPtPara; // Transverse momentum parameterisation
76 TF1* fYPara; // Rapidity parameterisation
77 TF1* fV2Para; // v2 parametrization
78 TF1* fdNdPhi; // Phi distribution depending on v2
79 Int_t fParam; // Parameterisation type
80 Float_t fdNdy0; // central multiplicity per event
81 Float_t fYWgt; // Y-weight
82 Float_t fPtWgt; // Pt-weight
83 Float_t fBias; // Biasing factor
84 Int_t fTrials; // Number of trials
85 Float_t fDeltaPt; // pT sampling in steps of fDeltaPt
86 Bool_t fSelectAll; // Flag for transportation of Background while using SetForceDecay()
87 AliDecayer *fDecayer; // ! Pointer to pythia object for decays
88 Bool_t fForceConv; //
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.
91
92private:
93 AliGenParam(const AliGenParam &Param);
94 AliGenParam & operator=(const AliGenParam & rhs);
95
96 ClassDef(AliGenParam, 2) // Generator using parameterised pt- and y-distribution
97};
98#endif
99
100
101
102
103
104
105
106
107
108