add LCD support
[enigma2.git] / lib / service / servicedvbrecord.cpp
index f05a35a398c2900a2006a187010c318b9936a21f..b34dad10e0ba397a3c7f0446c28b1a8f5021a6fe 100644 (file)
@@ -1,15 +1,19 @@
 #include <lib/service/servicedvbrecord.h>
 #include <lib/base/eerror.h>
-
+#include <lib/dvb/epgcache.h>
 #include <fcntl.h>
 
 DEFINE_REF(eDVBServiceRecord);
 
-eDVBServiceRecord::eDVBServiceRecord(const eServiceReferenceDVB &ref): m_ref(ref), m_service_handler(1)
+eDVBServiceRecord::eDVBServiceRecord(const eServiceReferenceDVB &ref): m_ref(ref)
 {
        CONNECT(m_service_handler.serviceEvent, eDVBServiceRecord::serviceEvent);
        m_state = stateIdle;
        m_want_record = 0;
+       m_tuned = 0;
+       m_target_fd = -1;
+       m_error = 0;
+       m_streaming = 0;
 }
 
 void eDVBServiceRecord::serviceEvent(int event)
@@ -20,33 +24,16 @@ void eDVBServiceRecord::serviceEvent(int event)
        case eDVBServicePMTHandler::eventTuned:
        {
                eDebug("tuned..");
-               if (!m_record)
-               {
-                       eDebug("Recording to %s...", m_filename.c_str());
-                       ::remove(m_filename.c_str());
-                       int fd = ::open(m_filename.c_str(), O_WRONLY|O_CREAT, 0644);
-                       if (fd == -1)
-                       {
-                               eDebug("eDVBServiceRecord - can't open hardcoded recording file!");
-                               return;
-//                             return -1;
-                       }
-                       ePtr<iDVBDemux> demux;
-                       if (m_service_handler.getDemux(demux))
-                       {
-                               eDebug("eDVBServiceRecord - NO DEMUX available!");
-                               return;
-//                             return -2;
-                       }
-                       demux->createTSRecorder(m_record);
-                       if (!m_record)
-                       {
-                               eDebug("eDVBServiceRecord - no ts recorder available.");
-                               return;
-//                             return -3;
-                       }
-                       m_record->setTargetFD(fd);
-               }
+               m_tuned = 1;
+               if (m_state == stateRecording && m_want_record)
+                       doRecord();
+               m_event((iRecordableService*)this, evTunedIn);
+               break;
+       }
+       case eDVBServicePMTHandler::eventTuneFailed:
+       {
+               eDebug("record failed to tune");
+               m_event((iRecordableService*)this, evTuneFailed);
                break;
        }
        case eDVBServicePMTHandler::eventNewProgramInfo:
@@ -55,43 +42,108 @@ void eDVBServiceRecord::serviceEvent(int event)
                        doPrepare();
                else if (m_want_record) /* doRecord can be called from Prepared and Recording state */
                        doRecord();
+               m_event((iRecordableService*)this, evNewProgramInfo);
                break;
        }
        }
 }
 
-
-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;
+       m_streaming = 0;
+       
        if (m_state == stateIdle)
-               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;
 }
 
+RESULT eDVBServiceRecord::prepareStreaming()
+{
+       m_filename = "";
+       m_streaming = 1;
+       if (m_state == stateIdle)
+               return doPrepare();
+       return -1;
+}
+
 RESULT eDVBServiceRecord::start()
 {
        m_want_record = 1;
                /* when tune wasn't yet successfully, doRecord stays in "prepared"-state which is fine. */
+       m_event((iRecordableService*)this, evStart);
        return doRecord();
 }
 
 
 RESULT eDVBServiceRecord::stop()
 {
-       eDebug("stop recording!!");
+       eDebug("stop recording!");
        if (m_state == stateRecording)
        {
-               m_record->stop();
+               if (m_record)
+                       m_record->stop();
+               if (m_target_fd >= 0)
+               {
+                       ::close(m_target_fd);
+                       m_target_fd = -1;
+               }
                m_state = statePrepared;
-       }
-       
+       } else
+               eDebug("(was not recording)");
        if (m_state == statePrepared)
        {
                m_record = 0;
                m_state = stateIdle;
        }
+       m_event((iRecordableService*)this, evRecordStopped);
        return 0;
 }
 
@@ -103,7 +155,7 @@ int eDVBServiceRecord::doPrepare()
        {
                m_pids_active.clear();
                m_state = statePrepared;
-               m_service_handler.tune(m_ref);
+               return m_service_handler.tune(m_ref, 0);
        }
        return 0;
 }
@@ -112,90 +164,228 @@ int eDVBServiceRecord::doRecord()
 {
        int err = doPrepare();
        if (err)
-               return err;
-       
-       if (!m_record)
        {
-               eDebug("demux not available (tune failed?). cannot record.");
-               return -1;
+               m_error = errTuneFailed;
+               m_event((iRecordableService*)this, evRecordFailed);
+               return err;
        }
-       eDebug("starting recording..");
        
-       eDVBServicePMTHandler::program program;
-       if (m_service_handler.getProgramInfo(program))
-               eDebug("getting program info failed.");
-       else
+       if (!m_tuned)
+               return 0; /* try it again when we are tuned in */
+       
+       if (!m_record && m_tuned && !m_streaming)
        {
-               std::set<int> pids_to_record;
-               
-               pids_to_record.insert(0); // PAT
-               
-               if (program.pmtPid != -1)
-                       pids_to_record.insert(program.pmtPid); // PMT
-               
-               eDebugNoNewLine("RECORD: have %d video stream(s)", program.videoStreams.size());
-               if (!program.videoStreams.empty())
+               eDebug("Recording to %s...", m_filename.c_str());
+               ::remove(m_filename.c_str());
+               int fd = ::open(m_filename.c_str(), O_WRONLY|O_CREAT|O_LARGEFILE, 0644);
+               if (fd == -1)
                {
-                       eDebugNoNewLine(" (");
-                       for (std::vector<eDVBServicePMTHandler::videoStream>::const_iterator
-                               i(program.videoStreams.begin()); 
-                               i != program.videoStreams.end(); ++i)
-                       {
-                               pids_to_record.insert(i->pid);
-                               if (i != program.videoStreams.begin())
-                                       eDebugNoNewLine(", ");
-                               eDebugNoNewLine("%04x", i->pid);
-                       }
-                       eDebugNoNewLine(")");
+                       eDebug("eDVBServiceRecord - can't open recording file!");
+                       m_error = errOpenRecordFile;
+                       m_event((iRecordableService*)this, evRecordFailed);
+                       return errOpenRecordFile;
                }
-               eDebugNoNewLine(", and %d audio stream(s)", program.audioStreams.size());
-               if (!program.audioStreams.empty())
-               {
-                       eDebugNoNewLine(" (");
-                       for (std::vector<eDVBServicePMTHandler::audioStream>::const_iterator
-                               i(program.audioStreams.begin()); 
-                               i != program.audioStreams.end(); ++i)
-                       {
-                               pids_to_record.insert(i->pid);
-                               if (i != program.audioStreams.begin())
-                                       eDebugNoNewLine(", ");
-                               eDebugNoNewLine("%04x", i->pid);
-                       }
-                       eDebugNoNewLine(")");
-               }
-               eDebug(", and the pcr pid is %04x", program.pcrPid);
-               if (program.pcrPid != 0x1fff)
-                       pids_to_record.insert(program.pcrPid);
-               
-                       /* find out which pids are NEW and which pids are obsolete.. */
-               std::set<int> new_pids, obsolete_pids;
-               
-               std::set_difference(pids_to_record.begin(), pids_to_record.end(), 
-                               m_pids_active.begin(), m_pids_active.end(),
-                               std::inserter(new_pids, new_pids.begin()));
-               
-               std::set_difference(
-                               m_pids_active.begin(), m_pids_active.end(),
-                               pids_to_record.begin(), pids_to_record.end(), 
-                               std::inserter(new_pids, new_pids.begin())
-                               );
-               
-               for (std::set<int>::iterator i(new_pids.begin()); i != new_pids.end(); ++i)
+
+                       /* turn off kernel caching strategies */
+               posix_fadvise(fd, 0, 0, POSIX_FADV_RANDOM);
+
+               ePtr<iDVBDemux> demux;
+               if (m_service_handler.getDataDemux(demux))
                {
-                       eDebug("ADD PID: %04x", *i);
-                       m_record->addPID(*i);
+                       eDebug("eDVBServiceRecord - NO DEMUX available!");
+                       m_error = errNoDemuxAvailable;
+                       m_event((iRecordableService*)this, evRecordFailed);
+                       return errNoDemuxAvailable;
                }
-               for (std::set<int>::iterator i(obsolete_pids.begin()); i != obsolete_pids.end(); ++i)
+               demux->createTSRecorder(m_record);
+               if (!m_record)
                {
-                       eDebug("REMOVED PID: %04x", *i);
-                       m_record->removePID(*i);
+                       eDebug("eDVBServiceRecord - no ts recorder available.");
+                       m_error = errNoTsRecorderAvailable;
+                       m_event((iRecordableService*)this, evRecordFailed);
+                       return errNoTsRecorderAvailable;
                }
-               
-               if (m_state != stateRecording)
+               m_record->setTargetFD(fd);
+               m_record->setTargetFilename(m_filename.c_str());
+               m_record->connectEvent(slot(*this, &eDVBServiceRecord::recordEvent), m_con_record_event);
+
+               m_target_fd = fd;
+       }
+       
+       if (m_streaming)
+       {
+               m_state = stateRecording;
+               eDebug("start streaming...");
+       } else
+       {
+               eDebug("start recording...");
+
+               eDVBServicePMTHandler::program program;
+               if (m_service_handler.getProgramInfo(program))
+                       eDebug("getting program info failed.");
+               else
                {
-                       m_record->start();
-                       m_state = stateRecording;
+                       std::set<int> pids_to_record;
+
+                       pids_to_record.insert(0); // PAT
+
+                       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())
+                       {
+                               eDebugNoNewLine(" (");
+                               for (std::vector<eDVBServicePMTHandler::videoStream>::const_iterator
+                                       i(program.videoStreams.begin()); 
+                                       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);
+                               }
+                               eDebugNoNewLine(")");
+                       }
+                       eDebugNoNewLine(", and %d audio stream(s)", program.audioStreams.size());
+                       if (!program.audioStreams.empty())
+                       {
+                               eDebugNoNewLine(" (");
+                               for (std::vector<eDVBServicePMTHandler::audioStream>::const_iterator
+                                       i(program.audioStreams.begin()); 
+                                       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);
+                               }
+                               eDebugNoNewLine(")");
+                       }
+                       if (!program.subtitleStreams.empty())
+                       {
+                               eDebugNoNewLine(" (");
+                               for (std::vector<eDVBServicePMTHandler::subtitleStream>::const_iterator
+                                       i(program.subtitleStreams.begin());
+                                       i != program.subtitleStreams.end(); ++i)
+                               {
+                                       pids_to_record.insert(i->pid);
+       
+                                       if (i != program.subtitleStreams.begin())
+                                               eDebugNoNewLine(", ");
+                                       eDebugNoNewLine("%04x", i->pid);
+                               }
+                               eDebugNoNewLine(")");
+                       }
+                       eDebugNoNewLine(", and the pcr pid is %04x", program.pcrPid);
+                       if (program.pcrPid != 0x1fff)
+                               pids_to_record.insert(program.pcrPid);
+                       eDebug(", and the text pid is %04x", program.textPid);
+                       if (program.textPid != -1)
+                               pids_to_record.insert(program.textPid); // Videotext
+
+                               /* find out which pids are NEW and which pids are obsolete.. */
+                       std::set<int> new_pids, obsolete_pids;
+
+                       std::set_difference(pids_to_record.begin(), pids_to_record.end(), 
+                                       m_pids_active.begin(), m_pids_active.end(),
+                                       std::inserter(new_pids, new_pids.begin()));
+
+                       std::set_difference(
+                                       m_pids_active.begin(), m_pids_active.end(),
+                                       pids_to_record.begin(), pids_to_record.end(), 
+                                       std::inserter(new_pids, new_pids.begin())
+                                       );
+                       
+                       for (std::set<int>::iterator i(new_pids.begin()); i != new_pids.end(); ++i)
+                       {
+                               eDebug("ADD PID: %04x", *i);
+                               m_record->addPID(*i);
+                       }
+
+                       for (std::set<int>::iterator i(obsolete_pids.begin()); i != obsolete_pids.end(); ++i)
+                       {
+                               eDebug("REMOVED PID: %04x", *i);
+                               m_record->removePID(*i);
+                       }
+
+                       if (timing_pid != -1)
+                               m_record->setTimingPID(timing_pid);
+
+                       m_pids_active = pids_to_record;
+
+                       if (m_state != stateRecording)
+                       {
+                               m_record->start();
+                               m_state = stateRecording;
+                       }
                }
        }
+       m_error = 0;
+       m_event((iRecordableService*)this, evRecordRunning);
        return 0;
 }
+
+RESULT eDVBServiceRecord::frontendInfo(ePtr<iFrontendInformation> &ptr)
+{
+       ptr = this;
+       return 0;
+}
+
+RESULT eDVBServiceRecord::connectEvent(const Slot2<void,iRecordableService*,int> &event, ePtr<eConnection> &connection)
+{
+       connection = new eConnection((iRecordableService*)this, m_event.connect(event));
+       return 0;
+}
+
+RESULT eDVBServiceRecord::stream(ePtr<iStreamableService> &ptr)
+{
+       ptr = this;
+       return 0;
+}
+
+PyObject *eDVBServiceRecord::getStreamingData()
+{
+       eDVBServicePMTHandler::program program;
+       if (!m_tuned || m_service_handler.getProgramInfo(program))
+       {
+               Py_INCREF(Py_None);
+               return Py_None;
+       }
+
+       PyObject *r = program.createPythonObject();
+       ePtr<iDVBDemux> demux;
+       if (!m_service_handler.getDataDemux(demux))
+       {
+               uint8_t demux_id;
+               demux->getCADemuxID(demux_id);
+               
+               PyDict_SetItemString(r, "demux", PyInt_FromLong(demux_id));
+       }
+
+       return r;
+}
+
+void eDVBServiceRecord::recordEvent(int event)
+{
+       switch (event)
+       {
+       case iDVBTSRecorder::eventWriteError:
+               eWarning("[eDVBServiceRecord] record write error");
+               stop();
+               m_event((iRecordableService*)this, evRecordWriteError);
+               return;
+       default:
+               eDebug("unhandled record event %d", event);
+       }
+}