f8110dd2ade511ea8d3c53147f3a6dbbec45701d
[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 "AliPythia.h"
10 #include "TNamed.h"
11 #include "TF1.h"
12 #include "TArrayF.h"
13 #include "TArrayI.h"
14 #include "TTree.h"
15 #include "TParticle.h"
16
17 //-------------------------------------------------------------
18 class AliGenParam : public AliGenerator
19 {
20 protected:
21   Double_t (*fPtParaFunc)(Double_t*, Double_t*); //! Pointer to Pt parametrisation function
22   Double_t (*fYParaFunc )(Double_t*, Double_t*); //! Pointer to Y parametrisation function
23   Int_t    (*fIpParaFunc )();    //! Pointer to particle type parametrisation function
24     TF1* fPtPara;
25     TF1* fYPara;
26     Param_t     fParam;
27     Float_t     fdNdy0;
28     Float_t     fYWgt;
29     Float_t     fPtWgt;
30     Weighting_t fAnalog;       //Flaf for anolog or pt-weighted generation
31     Float_t     fBias;
32     Int_t       fTrials;
33     Decay_t     fForceDecay;
34     Int_t       fCutOnChild;
35     TArrayI   fChildSelect;
36     AliPythia *fPythia;
37  private:
38     // check if particle is selected as child
39     Bool_t ChildSelected(Int_t ip);
40     // all kinematic selection goes here
41     Bool_t KinematicSelection(TParticle *particle);
42  public:
43   AliGenParam();
44   AliGenParam(Int_t npart, Param_t param);
45   AliGenParam(Int_t npart, Param_t param,
46                    Double_t (*PtPara)(Double_t*, Double_t*),
47                    Double_t (*YPara )(Double_t*, Double_t*),
48                    Int_t    (*IpPara)()                      ); 
49   virtual ~AliGenParam();
50   virtual void Generate();
51   virtual void Init();
52   // select particle type
53   virtual void SetParam(Param_t param=jpsi_p) {fParam=param;}
54   // force decay type
55   virtual void SetForceDecay(Decay_t decay=dimuon) {fForceDecay=decay;}
56   virtual void SetWeighting(Weighting_t flag=analog) {fAnalog=flag;}    
57   virtual void SetCutOnChild(Int_t flag=0) {fCutOnChild=flag;}
58   ClassDef(AliGenParam,1) // Generator using parameterised pt- and y-distribution
59 };
60 #endif
61
62
63
64
65
66
67
68
69
70