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