fix recording (thx to tmb)

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

diff --git a/lib/service/servicedvbrecord.cpp b/lib/service/servicedvbrecord.cpp
index 773aeb39ed9f6d0602539809502272a431725023..33f0497f177d260acf9e49a7b669fe432ffac09b 100644 (file)
--- a/lib/service/servicedvbrecord.cpp
+++ b/lib/service/servicedvbrecord.cpp
@@ -5,7 +5,7 @@
 
 DEFINE_REF(eDVBServiceRecord);
 
-eDVBServiceRecord::eDVBServiceRecord(const eServiceReferenceDVB &ref): m_ref(ref)
+eDVBServiceRecord::eDVBServiceRecord(const eServiceReferenceDVB &ref): m_ref(ref), m_service_handler(1)
 {
        CONNECT(m_service_handler.serviceEvent, eDVBServiceRecord::serviceEvent);
        m_state = stateIdle;
@@ -20,6 +20,33 @@ 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);
+               }
                break;
        }
        case eDVBServicePMTHandler::eventNewProgramInfo:
@@ -34,10 +61,14 @@ void eDVBServiceRecord::serviceEvent(int event)
 }
 
 
-RESULT eDVBServiceRecord::prepare()
+RESULT eDVBServiceRecord::prepare(const char *filename)
 {
+       m_filename = filename;
        if (m_state == stateIdle)
-               return m_service_handler.tune(m_ref);
+       {
+               doPrepare();
+               return 0;
+       }
        else
                return -1;
 }
@@ -73,32 +104,9 @@ int eDVBServiceRecord::doPrepare()
                /* allocate a ts recorder if we don't already have one. */
        if (m_state == stateIdle)
        {
-               ::remove("recordings.ts");
-               int fd = ::open("recording.ts", O_WRONLY|O_CREAT, 0644);
-               if (fd == -1)
-               {
-                       eDebug("eDVBServiceRecord - can't open hardcoded recording file!");
-                       return -1;
-               }
-               ePtr<iDVBDemux> demux;
-               if (m_service_handler.getDemux(demux))
-               {
-                       eDebug("eDVBServiceRecord - NO DEMUX available!");
-                       return -2;
-               }
-               demux->createTSRecorder(m_record);
-               if (!m_record)
-               {
-                       eDebug("eDVBServiceRecord - no ts recorder available.");
-                       return -3;
-               }
-               m_record->setTargetFD(fd);
                m_pids_active.clear();
                m_state = statePrepared;
-       } else if ((m_state == statePrepared) || (m_state == stateRecording))
-       {
-                       /* when we're already recording, we already have a recorder allocated. */
-               assert(m_record);
+               m_service_handler.tune(m_ref);
        }
        return 0;
 }
@@ -109,6 +117,11 @@ int eDVBServiceRecord::doRecord()
        if (err)
                return err;
        
+       if (!m_record)
+       {
+               eDebug("demux not available (tune failed?). cannot record.");
+               return -1;
+       }
        eDebug("starting recording..");
        
        eDVBServicePMTHandler::program program;
@@ -118,6 +131,11 @@ int eDVBServiceRecord::doRecord()
        {
                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())
                {