Cuts on primary particles are separated from those on the decay products. Methods

[u/mrichter/AliRoot.git] / EVGEN / 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 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;
 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);
  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 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;}
  ClassDef(AliGenParam,1) // Generator using parameterised pt- and y-distribution
};
#endif