All coding rule violations except RS3 corrected
[u/mrichter/AliRoot.git] / EVGEN / AliGenParam.h
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 #include "AliGenerator.h"
9 #include "TArrayI.h"
10 #include "GenTypeDefs.h"
11
12 class AliPythia;
13 class TParticle;
14 class TF1;
15
16 //-------------------------------------------------------------
17 class AliGenParam : public AliGenerator
18 {
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*),
25                 Int_t    (*IpPara)()                      );
26     AliGenParam(const AliGenParam &Param);
27      
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     
49     AliGenParam & operator=(const AliGenParam & rhs);
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
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
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
76  private:
77     // check if particle is selected as child
78     Bool_t ChildSelected(Int_t ip);
79     // all kinematic selection goes here
80     Bool_t KinematicSelection(TParticle *particle);
81
82   ClassDef(AliGenParam,1) // Generator using parameterised pt- and y-distribution
83 };
84 #endif
85
86
87
88
89
90
91
92
93
94