#include <lib/dvb/pmt.h>
#include <lib/dvb/specs.h>
#include <lib/dvb/dvb.h>

eDVBServicePMTHandler::eDVBServicePMTHandler()
{
	ePtr<eDVBResourceManager> mgr;
	eDVBResourceManager::getInstance(mgr);
	m_resourceManager = mgr;
	CONNECT(m_PMT.tableReady, eDVBServicePMTHandler::PMTready);
	CONNECT(m_PAT.tableReady, eDVBServicePMTHandler::PATready);
}

void eDVBServicePMTHandler::channelStateChanged(iDVBChannel *channel)
{
	int state;
	channel->getState(state);
	
	if ((m_last_channel_state != iDVBChannel::state_ok)
		&& (state == iDVBChannel::state_ok) && (!m_demux))
	{
		if (m_channel)
			if (m_channel->getDemux(m_demux))
				eDebug("shit it failed.. again.");
		
		if (m_demux)
		{	
			eDebug("ok ... now we start!!");
			m_PAT.begin(eApp, eDVBPATSpec(), m_demux);
		}
	}
}

void eDVBServicePMTHandler::PMTready(int error)
{
	if (error)
		serviceEvent(eventNoPMT);
	else
		serviceEvent(eventNewProgramInfo);
}

void eDVBServicePMTHandler::PATready(int)
{
	ePtr<eTable<ProgramAssociationTable> > ptr;
	if (!m_PAT.getCurrent(ptr))
	{
		int pmtpid = -1;
		ProgramAssociationTableConstIterator i;
		for (i = ptr->getSections().begin(); i != ptr->getSections().end(); ++i)
		{
			const ProgramAssociationTable &pat = **i;
			ProgramAssociationConstIterator program;
			for (program = pat.getPrograms()->begin(); program != pat.getPrograms()->end(); ++program)
				if (eServiceID((*program)->getProgramNumber()) == m_reference.getServiceID())
					pmtpid = (*program)->getProgramMapPid();
		}
		if (pmtpid == -1)
			serviceEvent(eventNoPATEntry);
		else
			m_PMT.begin(eApp, eDVBPMTSpec(pmtpid, m_reference.getServiceID().get()), m_demux);
	} else
		serviceEvent(eventNoPAT);
}

int eDVBServicePMTHandler::getProgramInfo(struct program &program)
{
	ePtr<eTable<ProgramMapTable> > ptr;
	
	program.videoStreams.clear();
	program.audioStreams.clear();
	program.pcrPid = -1;
	
	if (!m_PMT.getCurrent(ptr))
	{
		ProgramMapTableConstIterator i;
		for (i = ptr->getSections().begin(); i != ptr->getSections().end(); ++i)
		{
			const ProgramMapTable &pmt = **i;
			program.pcrPid = pmt.getPcrPid();
			
			ElementaryStreamInfoConstIterator es;
			for (es = pmt.getEsInfo()->begin(); es != pmt.getEsInfo()->end(); ++es)
			{
				int isaudio = 0, isvideo = 0;
				videoStream video;
				audioStream audio;
				
				video.pid = (*es)->getPid();
				audio.pid = (*es)->getPid();
				
				switch ((*es)->getType())
				{
				case 0x01: // MPEG 1 video
				case 0x02: // MPEG 2 video
					isvideo = 1;
					break;
				case 0x03: // MPEG 1 audio
				case 0x04: // MPEG 2 audio:
					isaudio = 1;
					audio.type = audioStream::atMPEG;
					break;
				}
				if (isaudio)
					program.audioStreams.push_back(audio);
				if (isvideo)
					program.videoStreams.push_back(video);
			}
		}
		return 0;
	} else
		return -1;
}

int eDVBServicePMTHandler::getDemux(ePtr<iDVBDemux> &demux)
{
	demux = m_demux;
	if (demux)
		return 0;
	else
		return -1;
}

int eDVBServicePMTHandler::tune(eServiceReferenceDVB &ref)
{
	RESULT res;
	m_channel = 0;
	m_channelStateChanged_connection = 0;
	m_reference = ref;
	eDVBChannelID chid;
	ref.getChannelID(chid);
	res = m_resourceManager->allocateChannel(chid, m_channel);
	eDebug("eDVBServicePMTHandler: tune %d", res);
	if (m_channel)
	{
		m_channel->connectStateChange(
			slot(*this, &eDVBServicePMTHandler::channelStateChanged), 
			m_channelStateChanged_connection);
		m_last_channel_state = -1;
		channelStateChanged(m_channel);
	}
	return res;
}