X-Git-Url: https://git.cweiske.de/enigma2.git/blobdiff_plain/c57ed609e8df226b0a912d249a0376c455a49704..056f4ed60ae323f41601b8acdc42794902eac0d1:/lib/dvb/pmt.cpp

diff --git a/lib/dvb/pmt.cpp b/lib/dvb/pmt.cpp
index 8df61304..74042ed0 100644
--- a/lib/dvb/pmt.cpp
+++ b/lib/dvb/pmt.cpp
@@ -5,6 +5,8 @@
 #include <lib/dvb/metaparser.h>
 #include <lib/dvb_ci/dvbci.h>
 #include <lib/dvb/epgcache.h>
+#include <lib/dvb/scan.h>
+#include <dvbsi++/ca_descriptor.h>
 #include <dvbsi++/ca_program_map_section.h>
 #include <dvbsi++/teletext_descriptor.h>
 #include <dvbsi++/descriptor_tag.h>
@@ -12,9 +14,10 @@
 #include <dvbsi++/stream_identifier_descriptor.h>
 
 eDVBServicePMTHandler::eDVBServicePMTHandler()
-	:m_ca_servicePtr(0), m_decode_demux_num(0xFF)
+	:m_ca_servicePtr(0), m_dvb_scan(0), m_decode_demux_num(0xFF)
 {
 	m_use_decode_demux = 0;
+	m_pmt_pid = -1;
 	eDVBResourceManager::getInstance(m_resourceManager);
 	CONNECT(m_PMT.tableReady, eDVBServicePMTHandler::PMTready);
 	CONNECT(m_PAT.tableReady, eDVBServicePMTHandler::PATready);
@@ -22,22 +25,7 @@ eDVBServicePMTHandler::eDVBServicePMTHandler()
 
 eDVBServicePMTHandler::~eDVBServicePMTHandler()
 {
-	if (m_ca_servicePtr)
-	{
-		eDebug("unregister caservice");
-		int demuxes[2] = {0,0};
-		uint8_t tmp;
-		m_demux->getCADemuxID(tmp);
-		demuxes[0]=tmp;
-		if (m_decode_demux_num != 0xFF)
-			demuxes[1]=m_decode_demux_num;
-		else
-			demuxes[1]=demuxes[0];
-		ePtr<eTable<ProgramMapSection> > ptr;
-		m_PMT.getCurrent(ptr);
-		eDVBCAService::unregister_service(m_reference, demuxes, ptr);
-		eDVBCIInterfaces::getInstance()->removePMTHandler(this);
-	}
+	free();
 }
 
 void eDVBServicePMTHandler::channelStateChanged(iDVBChannel *channel)
@@ -50,7 +38,7 @@ void eDVBServicePMTHandler::channelStateChanged(iDVBChannel *channel)
 	{
 		if (m_channel)
 			if (m_channel->getDemux(m_demux, (!m_use_decode_demux) ? 0 : iDVBChannel::capDecode))
-				eDebug("Allocating a demux for now tuned-in channel failed.");
+				eDebug("Allocating %s-decoding a demux for now tuned-in channel failed.", m_use_decode_demux ? "" : "non-");
 		
 		serviceEvent(eventTuned);
 		
@@ -58,7 +46,10 @@ void eDVBServicePMTHandler::channelStateChanged(iDVBChannel *channel)
 		{
 			eDebug("ok ... now we start!!");
 
-			m_PAT.begin(eApp, eDVBPATSpec(), m_demux);
+			if (m_pmt_pid == -1)
+				m_PAT.begin(eApp, eDVBPATSpec(), m_demux);
+			else
+				m_PMT.begin(eApp, eDVBPMTSpec(m_pmt_pid, m_reference.getServiceID().get()), m_demux);
 
 			if ( m_service && !m_service->cacheEmpty() )
 				serviceEvent(eventNewProgramInfo);
@@ -78,6 +69,9 @@ void eDVBServicePMTHandler::channelEvent(iDVBChannel *channel, int event)
 	case iDVBChannel::evtEOF:
 		serviceEvent(eventEOF);
 		break;
+	case iDVBChannel::evtSOF:
+		serviceEvent(eventSOF);
+		break;
 	default:
 		break;
 	}
@@ -142,6 +136,76 @@ void eDVBServicePMTHandler::PATready(int)
 		serviceEvent(eventNoPAT);
 }
 
+PyObject *eDVBServicePMTHandler::getCaIds()
+{
+	PyObject *ret=0;
+
+	ePtr<eTable<ProgramMapSection> > ptr;
+
+	if ( ((m_service && m_service->usePMT()) || !m_service) && !m_PMT.getCurrent(ptr))
+	{
+		uint16_t caids[255];
+		memset(caids, 0, sizeof(caids));
+		std::vector<ProgramMapSection*>::const_iterator i = ptr->getSections().begin();
+		for (; i != ptr->getSections().end(); ++i)
+		{
+			const ProgramMapSection &pmt = **i;
+			ElementaryStreamInfoConstIterator es = pmt.getEsInfo()->begin();
+			for (; es != pmt.getEsInfo()->end(); ++es)
+			{
+				for (DescriptorConstIterator desc = (*es)->getDescriptors()->begin();
+						desc != (*es)->getDescriptors()->end(); ++desc)
+				{
+					switch ((*desc)->getTag())
+					{
+						case CA_DESCRIPTOR:
+						{
+							const CaDescriptor *cadescr = (const CaDescriptor*)*desc;
+							uint16_t caid = cadescr->getCaSystemId();
+							int idx=0;
+							while (caids[idx] && caids[idx] != caid)
+								++idx;
+							caids[idx]=caid;
+							break;
+						}
+					}
+				}
+			}
+			for (DescriptorConstIterator desc = pmt.getDescriptors()->begin();
+				desc != pmt.getDescriptors()->end(); ++desc)
+			{
+				switch ((*desc)->getTag())
+				{
+					case CA_DESCRIPTOR:
+					{
+						const CaDescriptor *cadescr = (const CaDescriptor*)*desc;
+						uint16_t caid = cadescr->getCaSystemId();
+						int idx=0;
+						while (caids[idx] && caids[idx] != caid)
+							++idx;
+						caids[idx]=caid;
+						break;
+					}
+				}
+			}
+		}
+		int cnt=0;
+		while (caids[cnt])
+			++cnt;
+		if (cnt)
+		{
+			ret=PyList_New(cnt);
+			while(cnt--)
+				PyList_SET_ITEM(ret, cnt, PyInt_FromLong(caids[cnt]));
+		}
+	}
+
+	if (!ret)
+		ret=PyList_New(0);
+
+	return ret;
+}
+
 int eDVBServicePMTHandler::getProgramInfo(struct program &program)
 {
 	ePtr<eTable<ProgramMapSection> > ptr;
@@ -153,7 +217,7 @@ int eDVBServicePMTHandler::getProgramInfo(struct program &program)
 	program.pmtPid = -1;
 	program.textPid = -1;
 
-	if (!m_PMT.getCurrent(ptr))
+	if ( ((m_service && m_service->usePMT()) || !m_service) && !m_PMT.getCurrent(ptr))
 	{
 		int cached_apid_ac3 = -1;
 		int cached_apid_mpeg = -1;
@@ -358,6 +422,8 @@ int eDVBServicePMTHandler::getDecodeDemux(ePtr<iDVBDemux> &demux)
 		demux = m_demux;
 		return ret;
 	}
+	
+	ASSERT(m_channel); /* calling without a previous ::tune is certainly bad. */
 
 	ret = m_channel->getDemux(demux, iDVBChannel::capDecode);
 	if (!ret)
@@ -375,7 +441,40 @@ int eDVBServicePMTHandler::getPVRChannel(ePtr<iDVBPVRChannel> &pvr_channel)
 		return -1;
 }
 
-int eDVBServicePMTHandler::tune(eServiceReferenceDVB &ref, int use_decode_demux)
+void eDVBServicePMTHandler::SDTScanEvent(int event)
+{
+	eDebug("scan event %d!", event);
+
+	switch (event)
+	{
+		case eDVBScan::evtFinish:
+		{
+			ePtr<iDVBChannelList> db;
+			int err;
+			if ((err = m_resourceManager->getChannelList(db)) != 0)
+				eDebug("no channel list");
+			else
+			{
+				m_dvb_scan->insertInto(db);
+				eDebug("scan done!");
+			}
+			break;
+		}
+
+		case eDVBScan::evtNewService:
+			eDebug("scan new service");
+			break;
+
+		case eDVBScan::evtFail:
+			eDebug("scan failed.");
+			break;
+
+		default:
+			break;
+	}
+}
+
+int eDVBServicePMTHandler::tune(eServiceReferenceDVB &ref, int use_decode_demux, eCueSheet *cue)
 {
 	RESULT res;
 	m_reference = ref;
@@ -394,8 +493,22 @@ int eDVBServicePMTHandler::tune(eServiceReferenceDVB &ref, int use_decode_demux)
 		eDVBMetaParser parser;
 		
 		if (parser.parseFile(ref.path))
-			eWarning("no .meta file found, trying original service ref.");
-		else
+		{
+			eWarning("no .meta file found, trying to find PMT pid");
+			eDVBTSTools tstools;
+			if (tstools.openFile(ref.path.c_str()))
+				eWarning("failed to open file");
+			else
+			{
+				int service_id, pmt_pid;
+				if (!tstools.findPMT(pmt_pid, service_id))
+				{
+					eDebug("PMT pid found on pid %04x, service id %d", pmt_pid, service_id);
+					m_reference.setServiceID(service_id);
+					m_pmt_pid = pmt_pid;
+				}
+			}
+		} else
 			m_reference = parser.m_ref;
 		
 		eDebug("alloc PVR");
@@ -405,7 +518,11 @@ int eDVBServicePMTHandler::tune(eServiceReferenceDVB &ref, int use_decode_demux)
 			eDebug("allocatePVRChannel failed!\n");
 		m_channel = m_pvr_channel;
 	}
-	
+
+	ePtr<iDVBChannelList> db;
+	if (!m_resourceManager->getChannelList(db))
+		db->getService((eServiceReferenceDVB&)m_reference, m_service);
+
 	if (m_channel)
 	{
 		m_channel->connectStateChange(
@@ -417,6 +534,13 @@ int eDVBServicePMTHandler::tune(eServiceReferenceDVB &ref, int use_decode_demux)
 		m_channel->connectEvent(
 			slot(*this, &eDVBServicePMTHandler::channelEvent), 
 			m_channelEvent_connection);
+
+		if (ref.path.empty())
+		{
+			delete m_dvb_scan;
+			m_dvb_scan = new eDVBScan(m_channel);
+			m_dvb_scan->connectEvent(slot(*this, &eDVBServicePMTHandler::SDTScanEvent), m_scan_event_connection);
+		}
 	} else
 	{
 		serviceEvent(eventTuneFailed);
@@ -424,17 +548,42 @@ int eDVBServicePMTHandler::tune(eServiceReferenceDVB &ref, int use_decode_demux)
 	}
 
 	if (m_pvr_channel)
+	{
+		m_pvr_channel->setCueSheet(cue);
 		m_pvr_channel->playFile(ref.path.c_str());
-
-	ePtr<iDVBChannelList> db;
-	if (!m_resourceManager->getChannelList(db))
-		db->getService((eServiceReferenceDVB&)m_reference, m_service);
+	}
 
 	return res;
 }
 
 void eDVBServicePMTHandler::free()
 {
+	eDVBScan *tmp = m_dvb_scan;
+	m_dvb_scan = 0;
+	delete m_dvb_scan;
+
+	if (m_ca_servicePtr)
+	{
+		int demuxes[2] = {0,0};
+		uint8_t tmp;
+		m_demux->getCADemuxID(tmp);
+		demuxes[0]=tmp;
+		if (m_decode_demux_num != 0xFF)
+			demuxes[1]=m_decode_demux_num;
+		else
+			demuxes[1]=demuxes[0];
+		ePtr<eTable<ProgramMapSection> > ptr;
+		m_PMT.getCurrent(ptr);
+		eDVBCAService::unregister_service(m_reference, demuxes, ptr);
+		eDVBCIInterfaces::getInstance()->removePMTHandler(this);
+		m_ca_servicePtr = 0;
+	}
+
+	if (m_pvr_channel)
+	{
+		m_pvr_channel->stopFile();
+		m_pvr_channel->setCueSheet(0);
+	}
 	m_PMT.stop();
 	m_PAT.stop();
 	m_service = 0;
@@ -483,7 +632,7 @@ RESULT eDVBCAService::register_service( const eServiceReferenceDVB &ref, int dem
 		if ( iter < max_demux_slots )
 		{
 			caservice->m_used_demux[iter] = demux_nums[i] & 0xFF;
-			eDebug("[eDVBCAService] add demuxe %d to slot %d service %s", caservice->m_used_demux[iter], iter, ref.toString().c_str());
+			eDebug("[eDVBCAService] add demux %d to slot %d service %s", caservice->m_used_demux[iter], iter, ref.toString().c_str());
 		}
 		else
 		{