#ifndef _AliPythia_H
#define _AliPythia_H

#include <TPythia.h>
#include "GenTypeDefs.h"

class AliPythia:public TPythia
{
 protected:
    Process_t     fProcess;
    Decay_t       fDecay;
    Float_t       fEcms;
    StrucFunc_t   fStrucFunc;
    Int_t         fGPCode[501][2];
    Float_t       fBraPart[501];
    
 public:
    static Int_t fgInit;
    AliPythia();
    virtual ~AliPythia(){;}
    // convert to compressed code and print result (for debugging only)
    virtual Int_t CheckedLuComp(Int_t kf)
	{
	    Int_t kc=LuComp(kf);
	    printf("\n Lucomp kf,kc %d %d",kf,kc);
	    return kc;
	}
    // entry to the corresponding lujet function
    virtual void Lu1Ent(int flag, int idpart, 
			float mom, float theta,float phi);
    // Decay a Particle
    virtual void DecayParticle
	(Int_t idpart, Float_t mom, Float_t theta,Float_t phi);
    //
    // Pythia initialisation for selected processes
    virtual void ProcInit
	(Process_t process, Float_t energy, StrucFunc_t strucfunc);
    //
    // Count decay products
    virtual Int_t CountProducts(Int_t channel, Int_t particle);
    
    //
    // Force decay modes
    //
    // select type and multiplicity of the decay product
    virtual void ForceParticleDecay(Int_t particle,Int_t product,Int_t mult);
    //
    // force a particular decy-type
    virtual void ForceDecay(Decay_t  decay);
    // don't force any decay
    virtual void AllowAllDecays();
    
    //
    // Define heavy mesons to GEANT and make correspondance
    // GEANT - Pythia particle code
    virtual void DefineParticles();
    //
    // Convert from Pythia to Geant particle code
    virtual Int_t  GetGeantCode(Int_t kf);
    //
    // Get sum of branching ratios for forced decays
    virtual Float_t GetBraPart(Int_t kf);
    ClassDef(AliPythia,1) //ALICE UI to PYTHIA
};

#endif