#ifndef __PythiaGenerator__
#define __PythiaGenerator__

///////////////////////////////////////////////////////////////////////
//
//  JSF Pythia generator
//
//$Id: PythiaGenerator.h,v 1.12 2002/09/20 01:00:05 miyamoto Exp $
///////////////////////////////////////////////////////////////////////

#include <TObject.h>
#include <TTree.h>
#include <TClonesArray.h>

#include "JSFConfig.h"
#include "JSFGenerator.h"

#if __PYTHIA_VERSION__ >= 6
#include <TPythia6.h>
#include <TMCParticle6.h>
#else
#include <TPythia.h>
#include <TMCParticle.h>
#endif

#include "JSFBeamGeneration.h"

class PythiaGenerator : public JSFGenerator {
private:
  Double_t fEcm      ; // Center of mass energy (GeV)
#if __PYTHIA_VERSION__ >= 6 
  TPythia6  *fPythia  ; //! Pointer to the pythia class
  Int_t     fMRPY[6];   // Random seed at the begining of event
  Double_t  fRRPY[100]; // 			     
#else
  TPythia  *fPythia  ; //! Pointer to the pythia class
  Int_t    fMRLU[6];   // Random seed at the begining of event
  Float_t  fRRLU[100]; // 			     
#endif
  Char_t   fFrame[8]; // Name of beam particle
  Char_t   fBeamParticle[8]; // Name of beam particle
  Char_t   fTargetParticle[8]; // Target particle.
  Int_t    fPrintStat; //! call PYSTAT at the end run if not equal 0.				    
  
// Following data are filled only at the end of run, or data is read.
  Int_t    fNUMSUB   ; // Number of defined subprocesses + 1 (for total)
  Int_t    *fISUB    ; //[fNUMSUB] ISUB of subprocesses whose MSUB=1. (0=total, 1 to fNUMSUB )
  Int_t    *fNGEN    ; //[fNUMSUB] Number of generated subprocesses. (0=total, 1 to fNUMSUB )
  Double_t *fXSEC    ; //[fNUMSUB] Integrated cross sections (0=total, 1 to fNUMSUB)
  Float_t   fEventWeight ; //! Weight of current event.    

  Int_t     fBeamStrahlung ; //  > 0 to include beam strahlung.

  // For beamstrahlung
  TFile             *fBSFile; //! Beamstrahlung file
  JSFBeamGenerationCain  *fBS;     //  Beamstrahlung spectrum
  Double_t           fEcmMax; //! Abolutely max ECM energy
  TString            fBSname; //  Name of BS spectrum 
  Double_t           fBSwidth; //  half-width of BS spectrum

  Double_t           fEMinus; //!
  Double_t           fEPlus;  //!
  Int_t              fNBSGen; //
  Int_t              fNBSGood; //
  Double_t           fBSThreshold; // 

public:
  PythiaGenerator(const Char_t *name="PythiaGenerator", 
		  const Char_t *title="Pythia Event generator",
		  const Option_t *opt="");
  virtual ~PythiaGenerator();

  inline Double_t GetEMinus(){ return fEMinus; }
  inline Double_t GetEPlus(){ return fEPlus; }

  virtual Bool_t Initialize();
  virtual Bool_t BeginRun(Int_t nrun);
  virtual Bool_t Process(Int_t ev=1);
  virtual Bool_t EndRun();
  virtual Bool_t Terminate();
  virtual Bool_t GetLastRunInfo();

#if __PYTHIA_VERSION__ >= 6
  TPythia6  *GetPythia(){ return fPythia ; }
  inline Int_t const GetVersionNumber(){ return __PYTHIA_VERSION__  ; }
#else
  TPythia  *GetPythia(){ return fPythia ; }
  inline Int_t const GetVersionNumber(){ return 5 ; }
#endif

  inline const JSFBeamGenerationCain *GetBS(){ return fBS; }
  inline const TString &GetBSName(){ return fBSname; }

  void SetEcm(Double_t ecm){fEcm=ecm; }
  Double_t GetEcm(){ return fEcm;}

  Int_t GetNUMSUB(){ return fNUMSUB; }
  Int_t GetISUB(Int_t i){ return fISUB[i]; }
  Int_t GetNGEN(Int_t i){ return fNGEN[i]; }
  Double_t GetXSEC(Int_t i){ return fXSEC[i]; }

  inline Int_t GetBeamStrahlung(){  return fBeamStrahlung; }
  inline void SetBeamStrahlung(Int_t ibs){ fBeamStrahlung=ibs; }

  void SetEventWeight(Float_t weight){ fEventWeight=weight; }
  Float_t GetEventWeight(){ return fEventWeight; }

#ifdef __LCLIBRAN_USE_RANMAR__
  void WriteRandomSeed(Char_t *fw="");
  void ReadRandomSeed(Char_t *fr="");
  void PrintRandomSeed(Int_t num=8);  // First num's data are printed
#endif

  ClassDef(PythiaGenerator, 5)  // Pythia Generator

};

#endif