SetChildMomentumRange, SetChildPtRange, SetChildPhiRange, SetChildThetaRange added.
[u/mrichter/AliRoot.git] / EVGEN / AliGenParam.h
index 5f2fe0a23ee0e22b77b63dae2f4727911fa0bf66..67906bc55d18b89e399192d695168336dcdc5b4f 100644 (file)
@@ -1,6 +1,10 @@
-
 #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 "TArrayF.h"
 #include "TArrayI.h"
 #include "TTree.h"
-#include "TMCParticle.h"
+#include "TParticle.h"
 
 //-------------------------------------------------------------
-// Generators specific to MUON Arm
-
-// Generate heavy mesons - J/Psi, Upsilon, Phi
-
 class AliGenParam : public AliGenerator
 {
 protected:
-    Double_t (*fPtParaFunc)(Double_t*, Double_t*);
-    Double_t (*fYParaFunc )(Double_t*, Double_t*);
-    Int_t    (*fIpParaFunc )();    
+  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;
-    Int_t fIpart;
-    Float_t fdNdy0;
-    Float_t fYWgt;
-    Float_t fPtWgt;
-    Float_t fBias;
-    Int_t   fTrials;
-    Decay_t fForceDecay;
+    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
     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(TMCParticle *particle);
+    Bool_t KinematicSelection(TParticle *particle);
  public:
   AliGenParam();
-  AliGenParam(Int_t npart, Int_t ipart);
-//                Double_t (*PtPara)(Double_t*, Double_t*),
-//                Double_t (*YPara )(Double_t*, Double_t*));
+  AliGenParam(Int_t npart, Param_t param);
+  AliGenParam(Int_t npart, Param_t param,
+                  Double_t (*PtPara)(Double_t*, Double_t*),
+                  Double_t (*YPara )(Double_t*, Double_t*),
+                  Int_t    (*IpPara)()                      ); 
   virtual ~AliGenParam();
   virtual void Generate();
   virtual void Init();
   // select particle type
-  virtual void SetPart(Int_t part=443) {fIpart=part;}
+  virtual void SetParam(Param_t param=jpsi_p) {fParam=param;}
   // force decay type
-  virtual void ForceDecay(Decay_t decay=dimuon) {fForceDecay=decay;}
-  ClassDef(AliGenParam,1)
+  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;}
+  ClassDef(AliGenParam,1) // Generator using parameterised pt- and y-distribution
 };
 #endif