fix typo

[enigma2.git] / lib / service / servicedvbrecord.cpp

diff --git a/lib/service/servicedvbrecord.cpp b/lib/service/servicedvbrecord.cpp
index f4bf557726cbae7de71be225594cddd0f2f4c92a..4c41b5edef539d82e28d9fc70d0822fdeaf685ba 100644 (file)
--- 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;
        }
        
@@ -102,9 +155,13 @@ int eDVBServiceRecord::doRecord()
                int fd = ::open(m_filename.c_str(), O_WRONLY|O_CREAT|O_LARGEFILE, 0644);
                if (fd == -1)
                {
-                       eDebug("eDVBServiceRecord - can't open hardcoded recording file!");
+                       eDebug("eDVBServiceRecord - can't open recording file!");
                        return -1;
                }
+               
+                       /* turn off kernel caching strategies */
+               posix_fadvise(fd, 0, 0, POSIX_FADV_RANDOM);
+               
                ePtr<iDVBDemux> demux;
                if (m_service_handler.getDataDemux(demux))
                {
@@ -118,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..");
        
@@ -133,6 +192,8 @@ int eDVBServiceRecord::doRecord()
                if (program.pmtPid != -1)
                        pids_to_record.insert(program.pmtPid); // PMT
                
+               int timing_pid = -1;
+               
                eDebugNoNewLine("RECORD: have %d video stream(s)", program.videoStreams.size());
                if (!program.videoStreams.empty())
                {
@@ -142,6 +203,10 @@ int eDVBServiceRecord::doRecord()
                                i != program.videoStreams.end(); ++i)
                        {
                                pids_to_record.insert(i->pid);
+                               
+                               if (timing_pid == -1)
+                                       timing_pid = i->pid;
+                               
                                if (i != program.videoStreams.begin())
                                        eDebugNoNewLine(", ");
                                eDebugNoNewLine("%04x", i->pid);
@@ -157,6 +222,10 @@ int eDVBServiceRecord::doRecord()
                                i != program.audioStreams.end(); ++i)
                        {
                                pids_to_record.insert(i->pid);
+
+                               if (timing_pid == -1)
+                                       timing_pid = i->pid;
+                               
                                if (i != program.audioStreams.begin())
                                        eDebugNoNewLine(", ");
                                eDebugNoNewLine("%04x", i->pid);
@@ -194,6 +263,9 @@ int eDVBServiceRecord::doRecord()
                        m_record->removePID(*i);
                }
                
+               if (timing_pid != -1)
+                       m_record->setTimingPID(timing_pid);
+               
                m_pids_active = pids_to_record;
                
                if (m_state != stateRecording)
@@ -204,3 +276,9 @@ int eDVBServiceRecord::doRecord()
        }
        return 0;
 }
+
+RESULT eDVBServiceRecord::frontendInfo(ePtr<iFrontendInformation> &ptr)
+{
+       ptr = this;
+       return 0;
+}