+#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$ */
-#ifndef AliGenParam_H
-#define AliGenParam_H
#include "AliGenerator.h"
-#include "AliPythia.h"
-#include "TNamed.h"
-#include "TF1.h"
-#include "TArrayF.h"
-#include "TArrayI.h"
-#include "TTree.h"
-#include "TParticle.h"
+#include "AliDecayer.h"
+#include "GenTypeDefs.h"
+#include <TArrayI.h>
+
+class AliPythia;
+class TParticle;
+class AliGenLib;
+class TF1;
//-------------------------------------------------------------
class AliGenParam : public AliGenerator
{
-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;
+ public:
+ AliGenParam();
+ AliGenParam(Int_t npart, AliGenLib * Library, Param_t param, char* tname=0);
+ AliGenParam(Int_t npart, Param_t param, char* tname=0);
+ AliGenParam(Int_t npart, Param_t param,
+ Double_t (*PtPara)(Double_t*, Double_t*),
+ Double_t (*YPara )(Double_t*, Double_t*),
+ Int_t (*IpPara)() );
+ AliGenParam(const AliGenParam &Param);
+
+ 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 SetForceDecay(Decay_t decay=dimuon) {fForceDecay=decay;}
+ virtual void SetWeighting(Weighting_t flag=analog) {fAnalog=flag;}
+ virtual void SetCutOnChild(Int_t flag=0) {fCutOnChild=flag;}
+ virtual void SetChildMomentumRange(Float_t pmin=0, Float_t pmax=1.e10)
+ {fChildPMin = pmin; fChildPMax = pmax;}
+ virtual void SetChildPtRange(Float_t ptmin=0, Float_t ptmax=20.)
+ {fChildPtMin = ptmin; fChildPtMax = ptmax;}
+ virtual void SetChildPhiRange(Float_t phimin=-180., Float_t phimax=180)
+ {fChildPhiMin = TMath::Pi()*phimin/180;
+ fChildPhiMax = TMath::Pi()*phimax/180;}
+ virtual void SetChildThetaRange(Float_t thetamin=0, Float_t thetamax=180)
+ {fChildThetaMin = TMath::Pi()*thetamin/180;
+ fChildThetaMax = TMath::Pi()*thetamax/180;}
+ virtual void SetDeltaPt(Float_t delta=0.01) {fDeltaPt=delta;}
+
+ AliGenParam & operator=(const AliGenParam & rhs);
+ 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; // Transverse momentum parameterisation
+ TF1* fYPara; // Rapidity parameterisation
+ Param_t fParam; // Parameterisation type
+ Float_t fdNdy0; // central multiplicity per event
+ Float_t fYWgt; // Y-weight
+ Float_t fPtWgt; // Pt-weight
+ Weighting_t fAnalog; // Flag for anolog or pt-weighted generation
+ Float_t fBias; // Biasing factor
+ Int_t fTrials; // Number of trials
+ Decay_t fForceDecay; // Decay channel forced
+ Int_t fCutOnChild; // Cuts on decay products (children) are enabled/disabled
+ Float_t fChildPtMin; // Children minimum pT
+ Float_t fChildPtMax; // Children maximum pT
+ Float_t fChildPMin; // Children minimum p
+ Float_t fChildPMax; // Children maximum p
+ Float_t fChildPhiMin; // Children minimum phi
+ Float_t fChildPhiMax; // Children maximum phi
+ Float_t fChildThetaMin;// Children minimum theta
+ Float_t fChildThetaMax;// Children maximum theta
+ Float_t fDeltaPt; // pT sampling in steps of fDeltaPt
+ TArrayI fChildSelect; // Children to be selected from decay products
+ AliDecayer *fDecayer; // Pointer to pythia object for decays
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
};
#endif