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 "AliDecayer.h"
  10 #include "GenTypeDefs.h"
  11 #include <TArrayI.h>
  12
  13 class AliPythia;
  14 class TParticle;
  15 class AliGenLib;
  16 class TF1;
  17
  18 //-------------------------------------------------------------
  19 class AliGenParam : public AliGenerator
  20 {
  21  public:
  22     AliGenParam();
  23     AliGenParam(Int_t npart, AliGenLib * Library, Param_t param, char* tname=0);
  24     AliGenParam(Int_t npart, Param_t param, char* tname=0);
  25     AliGenParam(Int_t npart, Param_t param,
  26                 Double_t (*PtPara)(Double_t*, Double_t*),
  27                 Double_t (*YPara )(Double_t*, Double_t*),
  28                 Int_t    (*IpPara)()                      );
  29     AliGenParam(const AliGenParam &Param);
  30
  31     virtual ~AliGenParam();
  32     virtual void Generate();
  33     virtual void Init();
  34     // select particle type
  35     virtual void SetParam(Param_t param=jpsi_p) {fParam=param;}
  36     // force decay type
  37     virtual void SetForceDecay(Decay_t decay=dimuon) {fForceDecay=decay;}
  38     virtual void SetWeighting(Weighting_t flag=analog) {fAnalog=flag;}
  39     virtual void SetCutOnChild(Int_t flag=0) {fCutOnChild=flag;}
  40     virtual void SetChildMomentumRange(Float_t pmin=0, Float_t pmax=1.e10)
  41         {fChildPMin = pmin; fChildPMax = pmax;}
  42     virtual void SetChildPtRange(Float_t ptmin=0, Float_t ptmax=20.)
  43         {fChildPtMin = ptmin; fChildPtMax = ptmax;}
  44     virtual void SetChildPhiRange(Float_t phimin=-180., Float_t phimax=180)
  45         {fChildPhiMin = TMath::Pi()*phimin/180;
  46         fChildPhiMax = TMath::Pi()*phimax/180;}
  47     virtual void SetChildThetaRange(Float_t thetamin=0, Float_t thetamax=180)
  48         {fChildThetaMin = TMath::Pi()*thetamin/180;
  49         fChildThetaMax = TMath::Pi()*thetamax/180;}
  50     virtual void SetDeltaPt(Float_t delta=0.01) {fDeltaPt=delta;}
  51
  52     AliGenParam & operator=(const AliGenParam & rhs);
  53  protected:
  54     Double_t (*fPtParaFunc)(Double_t*, Double_t*); //! Pointer to Pt parametrisation function
  55     Double_t (*fYParaFunc )(Double_t*, Double_t*); //! Pointer to Y parametrisation function
  56     Int_t    (*fIpParaFunc )();    //! Pointer to particle type parametrisation function
  57     TF1* fPtPara;              // Transverse momentum parameterisation
  58     TF1* fYPara;               // Rapidity parameterisation
  59     Param_t     fParam;        // Parameterisation type
  60     Float_t     fdNdy0;        // central multiplicity per event
  61     Float_t     fYWgt;         // Y-weight
  62     Float_t     fPtWgt;        // Pt-weight
  63     Weighting_t fAnalog;       // Flag for anolog or pt-weighted generation
  64     Float_t     fBias;         // Biasing factor
  65     Int_t       fTrials;       // Number of trials
  66     Decay_t     fForceDecay;   // Decay channel forced
  67     Int_t       fCutOnChild;   // Cuts on decay products (children)  are enabled/disabled
  68     Float_t     fChildPtMin;   // Children minimum pT
  69     Float_t     fChildPtMax;   // Children maximum pT
  70     Float_t     fChildPMin;    // Children minimum p
  71     Float_t     fChildPMax;    // Children maximum p
  72     Float_t     fChildPhiMin;  // Children minimum phi
  73     Float_t     fChildPhiMax;  // Children maximum phi
  74     Float_t     fChildThetaMin;// Children minimum theta
  75     Float_t     fChildThetaMax;// Children maximum theta
  76     Float_t     fDeltaPt;      // pT sampling in steps of fDeltaPt
  77     TArrayI     fChildSelect;  // Children to be selected from decay products
  78     AliDecayer  *fDecayer;      // Pointer to pythia object for decays
  79  private:
  80     // check if particle is selected as child
  81     Bool_t ChildSelected(Int_t ip);
  82     // all kinematic selection goes here
  83     Bool_t KinematicSelection(TParticle *particle);
  84
  85   ClassDef(AliGenParam,1) // Generator using parameterised pt- and y-distribution
  86 };
  87 #endif
  88
  89
  90
  91
  92
  93
  94
  95
  96
  97