X-Git-Url: http://git.uio.no/git/?p=u%2Fmrichter%2FAliRoot.git;a=blobdiff_plain;f=EVGEN%2FAliGenParam.h;h=bbd61cebc653437502af3a4f6ab8b0c4dc80c286;hp=86e5d51f8bf3bc6fb39d625659e6ad948854210b;hb=dd028c53090c8bd4eb79119d6a7aae3928ed4410;hpb=3da306186bdfe9b8dedf54c4b9cdcf2bff5b132c diff --git a/EVGEN/AliGenParam.h b/EVGEN/AliGenParam.h index 86e5d51f8bf..bbd61cebc65 100644 --- a/EVGEN/AliGenParam.h +++ b/EVGEN/AliGenParam.h @@ -1,59 +1,98 @@ -#ifndef AliGenParam_H -#define AliGenParam_H +#ifndef ALIGENPARAM_H +#define ALIGENPARAM_H /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. * * See cxx source for full Copyright notice */ /* $Id$ */ -#include "AliGenerator.h" -#include "AliPythia.h" -#include "TNamed.h" -#include "TF1.h" -#include "TArrayF.h" -#include "TArrayI.h" -#include "TTree.h" -#include "TParticle.h" +// +// Class to generate particles from using parametrized pT and y distributions. +// Distributions are obtained from pointer to object of type AliGenLib. +// (For example AliGenMUONlib) +// +// andreas.morsch@cern.ch +// +#include "AliGenMC.h" + +class AliPythia; +class TParticle; +class AliGenLib; +class TF1; + +typedef enum { kAnalog, kNonAnalog} Weighting_t; //------------------------------------------------------------- -class AliGenParam : public AliGenerator +class AliGenParam : public AliGenMC { +public: + AliGenParam(); + AliGenParam(Int_t npart, const AliGenLib * Library, Int_t param, const char* tname = 0); + AliGenParam(Int_t npart, Int_t param, const char* tname = 0, const char* name = 0); + AliGenParam(Int_t npart, Int_t param, + Double_t (*PtPara)(const Double_t*, const Double_t*), + Double_t (*YPara )(const Double_t*, const Double_t*), + Double_t (*V2Para)(const Double_t*, const Double_t*), + Int_t (*IpPara)(TRandom*) ); + + virtual ~AliGenParam(); + virtual void GenerateN(Int_t ntimes); + virtual void Generate(); + virtual void Init(); + // select particle type + virtual void SetParam(Int_t param) {fParam = param;} + //Setting the flag for Background transportation while using SetForceDecay() + void SetSelectAll(Bool_t selectall) {fSelectAll = selectall;} + // force decay type + virtual void SetWeighting(Weighting_t flag = kAnalog) {fAnalog = flag;} + virtual void SetDeltaPt(Float_t delta=0.01) {fDeltaPt = delta;} + virtual void SetDecayer(AliDecayer* decayer) {fDecayer = decayer;} + virtual void SetForceGammaConversion(Bool_t force=kTRUE) {fForceConv = force;} + virtual void SetKeepParent(Bool_t keep=kTRUE){fKeepParent= keep;} //Store parent even if it does not have childs within cuts + virtual void SetKeepIfOneChildSelected(Bool_t keep=kTRUE){fKeepIfOneChildSelected = keep;} //Accept parent and child even if other children are not within cut. + + virtual void Draw(const char * opt); + TF1 * GetPt() { return fPtPara;} + TF1 * GetY() {return fYPara;} + Float_t GetRelativeArea(Float_t ptMin, Float_t ptMax, Float_t yMin, Float_t yMax, Float_t phiMin, Float_t phiMax); + + static TVector3 OrthogonalVector(TVector3 &inVec); + static void RotateVector(Double_t *pin, Double_t *pout, Double_t costheta, Double_t sintheta, + Double_t cosphi, Double_t sinphi); + static double ScreenFunction1(double d); + static double ScreenFunction2(double d); + double RandomEnergyFraction(double Z, double E); + double RandomPolarAngle(); + double RandomMass(Double_t mh); + Int_t VirtualGammaPairProduction(TClonesArray *particles, Int_t nPart); + Int_t ForceGammaConversion(TClonesArray *particles, Int_t nPart); + protected: - Double_t (*fPtParaFunc)(Double_t*, Double_t*); //! Pointer to Pt parametrisation function - Double_t (*fYParaFunc )(Double_t*, Double_t*); //! Pointer to Y parametrisation function - Int_t (*fIpParaFunc )(); //! Pointer to particle type parametrisation function - TF1* fPtPara; - TF1* fYPara; - Param_t fParam; - Float_t fdNdy0; - Float_t fYWgt; - Float_t fPtWgt; - Weighting_t fAnalog; //Flaf for anolog or pt-weighted generation - Float_t fBias; - Int_t fTrials; - Decay_t fForceDecay; - Int_t fCutOnChild; - TArrayI fChildSelect; - AliPythia *fPythia; - private: - // check if particle is selected as child - Bool_t ChildSelected(Int_t ip); - // all kinematic selection goes here - Bool_t KinematicSelection(TParticle *particle); - public: - AliGenParam(); - AliGenParam(Int_t npart, Param_t param); -// Double_t (*PtPara)(Double_t*, Double_t*), -// Double_t (*YPara )(Double_t*, Double_t*)); - virtual ~AliGenParam(); - virtual void Generate(); - virtual void Init(); - // select particle type - virtual void SetParam(Param_t param=jpsi_p) {fParam=param;} - // force decay type - virtual void ForceDecay(Decay_t decay=dimuon) {fForceDecay=decay;} - virtual void SetWeighting(Weighting_t flag=analog) {fAnalog=flag;} - virtual void SetCutOnChild(Int_t flag=0) {fCutOnChild=flag;} - ClassDef(AliGenParam,1) // Generator using parameterised pt- and y-distribution + Double_t (*fPtParaFunc)(const Double_t*, const Double_t*); //! Pointer to Pt parametrisation function + Double_t (*fYParaFunc )(const Double_t*, const Double_t*); //! Pointer to Y parametrisation function + Int_t (*fIpParaFunc )(TRandom*); //! Pointer to particle type parametrisation function + Double_t (*fV2ParaFunc )(const Double_t*, const Double_t*);//! Pointer to V2 parametrisation function + TF1* fPtPara; // Transverse momentum parameterisation + TF1* fYPara; // Rapidity parameterisation + TF1* fV2Para; // v2 parametrization + TF1* fdNdPhi; // Phi distribution depending on v2 + Int_t fParam; // Parameterisation type + Float_t fdNdy0; // central multiplicity per event + Float_t fYWgt; // Y-weight + Float_t fPtWgt; // Pt-weight + Float_t fBias; // Biasing factor + Int_t fTrials; // Number of trials + Float_t fDeltaPt; // pT sampling in steps of fDeltaPt + Bool_t fSelectAll; // Flag for transportation of Background while using SetForceDecay() + AliDecayer *fDecayer; // ! Pointer to pythia object for decays + Bool_t fForceConv; // + Bool_t fKeepParent; // Store parent even if it does not have childs within cuts + Bool_t fKeepIfOneChildSelected; //Accept parent and child even if other children are not within cut. + +private: + AliGenParam(const AliGenParam &Param); + AliGenParam & operator=(const AliGenParam & rhs); + + ClassDef(AliGenParam, 2) // Generator using parameterised pt- and y-distribution }; #endif