D and B distributions updated according to PPR.

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

#ifndef ALIGENHIJINGPARA_H
#define ALIGENHIJINGPARA_H
/* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
 * See cxx source for full Copyright notice                               */

/* $Id$ */

// Parameterisation of pi and K, eta and pt distributions
// used for the ALICE TDRs.
// Author: andreas.morsch@cern.ch

class TF1;

#include "AliGenerator.h"
#include "AliDecayer.h"

class AliGenHIJINGpara : public AliGenerator
{
 public:

  AliGenHIJINGpara();
  AliGenHIJINGpara(Int_t npart);
  AliGenHIJINGpara(const AliGenHIJINGpara &HIJINGpara);
  AliGenHIJINGpara& operator = (const AliGenHIJINGpara &para) 
  {para.Copy(*this); return (*this);}
  virtual ~AliGenHIJINGpara();
  virtual void SetCutVertexZ(Float_t cut=999999.) {fCutVertexZ = cut;}
  virtual void Generate();
  virtual void Init();
  virtual void SetPtRange(Float_t ptmin = 0., Float_t ptmax=15.);
  virtual void SetPi0Decays(Bool_t flag = kFALSE) {fPi0Decays = flag;}
  virtual void SetWeighting(Int_t flag = 0) {fAnalog = flag;}
  virtual void Draw( const char *);
  
 protected:
  Int_t   fNt;          // CurrentTrack;
  Bool_t  fPi0Decays;   // Flag for pi0 decays
  Float_t fPtWgtPi;     // pt weight for pions
  Float_t fPtWgtKa;     // pt weight for kaons
  TF1* fPtpi;           // Parametrised pt distribution for pi
  TF1* fPtka;           // Parametrised pt distribution for ka
  TF1* fETApic;         // Parametrised eta distribution for pi
  TF1* fETAkac;         // Parametrised eta distribution fro ka
  AliDecayer* fDecayer; // ! Pointer to pythia object for decays

  void DecayPi0(Float_t* orig, Float_t * p);
  void Copy(TObject &para) const;
  ClassDef(AliGenHIJINGpara,3) // Hijing parametrisation generator
};
#endif