- Some coded moved to AliGenMC

[u/mrichter/AliRoot.git] / EVGEN / AliGenParam.h

diff --git a/EVGEN/AliGenParam.h b/EVGEN/AliGenParam.h
index a769e8a8c9aa6297f529e2755cb6b9823fcd0426..dfd4dc8ef686a48dc8c0bdd7e597726152e9c63e 100644 (file)
--- a/EVGEN/AliGenParam.h
+++ b/EVGEN/AliGenParam.h
@@ -1,52 +1,56 @@
+#ifndef ALIGENPARAM_H
+#define ALIGENPARAM_H
+/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
+ * See cxx source for full Copyright notice                               */
 
-#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"
+/* $Id$ */
 
+#include "AliGenMC.h"
+#include <TArrayI.h>
+
+class AliPythia;
+class TParticle;
+class AliGenLib;
+class TF1;
+
+typedef enum { kAnalog, kNonAnalog} Weighting_t;
 //-------------------------------------------------------------
-class AliGenParam : public AliGenerator
+class AliGenParam : public AliGenMC
 {
-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;
-    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;}   
+    AliGenParam();
+    AliGenParam(Int_t npart, AliGenLib * Library, Int_t param, char* tname = 0);
+    AliGenParam(Int_t npart, Int_t param, char* tname = 0);
+    AliGenParam(Int_t npart, Int_t param,
+               Double_t (*PtPara)(Double_t*, Double_t*),
+               Double_t (*YPara )(Double_t*, Double_t*),
+               Int_t    (*IpPara)(TRandom*)           );
+    AliGenParam(const AliGenParam &Param);
+     
+    virtual ~AliGenParam();
+    virtual void Generate();
+    virtual void Init();
+    // select particle type
+    virtual void SetParam(Int_t param) {fParam = param;}
+    // force decay type
+    virtual void SetWeighting(Weighting_t flag = kAnalog) {fAnalog = flag;}    
+    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 )(TRandom*);    //! Pointer to particle type parametrisation function
+    TF1* fPtPara;              // Transverse momentum parameterisation
+    TF1* fYPara;               // Rapidity parameterisation
+    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
+    AliDecayer  *fDecayer;     // ! Pointer to pythia object for decays
   ClassDef(AliGenParam,1) // Generator using parameterised pt- and y-distribution
 };
 #endif