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 //
   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
  18 class AliPythia;
  19 class TParticle;
  20 class AliGenLib;
  21 class TF1;
  22
  23 typedef enum { kAnalog, kNonAnalog} Weighting_t;
  24 //-------------------------------------------------------------
  25 class AliGenParam : public AliGenMC
  26 {
  27 public:
  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
  70 protected:
  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
  92 private:
  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