X-Git-Url: https://git.cweiske.de/enigma2.git/blobdiff_plain/a39683740f78cc2463900d1ba73f7d7e2c69d807..0f70a75ff64c1583ccd620631462499f9d5fbc26:/lib/service/servicedvbrecord.cpp

diff --git a/lib/service/servicedvbrecord.cpp b/lib/service/servicedvbrecord.cpp
index 5b4f35ed..d0da074a 100644
--- a/lib/service/servicedvbrecord.cpp
+++ b/lib/service/servicedvbrecord.cpp
@@ -1,5 +1,6 @@
 #include <lib/service/servicedvbrecord.h>
 #include <lib/base/eerror.h>
+#include <lib/dvb/epgcache.h>
 
 #include <fcntl.h>
 
@@ -11,6 +12,7 @@ eDVBServiceRecord::eDVBServiceRecord(const eServiceReferenceDVB &ref): m_ref(ref
 	m_state = stateIdle;
 	m_want_record = 0;
 	m_tuned = 0;
+	m_target_fd = -1;
 }
 
 void eDVBServiceRecord::serviceEvent(int event)
@@ -37,11 +39,57 @@ void eDVBServiceRecord::serviceEvent(int event)
 	}
 }
 
-RESULT eDVBServiceRecord::prepare(const char *filename)
+RESULT eDVBServiceRecord::prepare(const char *filename, time_t begTime, time_t endTime, int eit_event_id)
 {
 	m_filename = filename;
 	if (m_state == stateIdle)
-		return doPrepare();
+	{
+		int ret = doPrepare();
+		if (!ret)
+		{
+			eEPGCache::getInstance()->Lock();
+			const eit_event_struct *event = 0;
+			eServiceReferenceDVB ref = m_ref.getParentServiceReference();
+			if (!ref.valid())
+				ref = m_ref;
+			if ( eit_event_id != -1 )
+			{
+				eDebug("query epg event id %d", eit_event_id);
+				eEPGCache::getInstance()->lookupEventId(ref, eit_event_id, event);
+			}
+			if ( !event && (begTime != -1 && endTime != -1) )
+			{
+				time_t queryTime = begTime + ((endTime-begTime)/2);
+				tm beg, end, query;
+				localtime_r(&begTime, &beg);
+				localtime_r(&endTime, &end);
+				localtime_r(&queryTime, &query);
+				eDebug("query stime %d:%d:%d, etime %d:%d:%d, qtime %d:%d:%d",
+					beg.tm_hour, beg.tm_min, beg.tm_sec,
+					end.tm_hour, end.tm_min, end.tm_sec,
+					query.tm_hour, query.tm_min, query.tm_sec);
+				eEPGCache::getInstance()->lookupEventTime(ref, queryTime, event);
+			}
+			if ( event )
+			{
+				eDebug("found event.. store to disc");
+				std::string fname = filename;
+				fname.erase(fname.length()-2, 2);
+				fname+="eit";
+				int fd = open(fname.c_str(), O_CREAT|O_WRONLY, 0777);
+				if (fd>-1)
+				{
+					int evLen=HILO(event->descriptors_loop_length)+12/*EIT_LOOP_SIZE*/;
+					int wr = ::write( fd, (unsigned char*)event, evLen );
+					if ( wr != evLen )
+						eDebug("eit write error (%m)");
+					::close(fd);
+				}
+			}
+			eEPGCache::getInstance()->Unlock();
+		}
+		return ret;
+	}
 	else
 		return -1;
 }
@@ -61,6 +109,11 @@ RESULT eDVBServiceRecord::stop()
 	{
 		if (m_record)
 			m_record->stop();
+		if (m_target_fd >= 0)
+		{
+			::close(m_target_fd);
+			m_target_fd = -1;
+		}
 		m_state = statePrepared;
 	}
 	
@@ -122,6 +175,8 @@ int eDVBServiceRecord::doRecord()
 			return -3;
 		}
 		m_record->setTargetFD(fd);
+		m_record->setTargetFilename(m_filename.c_str());
+		m_target_fd = fd;
 	}
 	eDebug("starting recording..");