enable playback of multifile (split) movies

[enigma2.git] / lib / dvb / tstools.cpp

diff --git a/lib/dvb/tstools.cpp b/lib/dvb/tstools.cpp
index bec17a56922329026a7e3be8193364dfd0227086..5c99ea60c50027d03244d0d6c91b9516bc3d77a3 100644 (file)
--- a/lib/dvb/tstools.cpp
+++ b/lib/dvb/tstools.cpp
@@ -7,12 +7,13 @@
 
 eDVBTSTools::eDVBTSTools()
 {
-       m_fd = -1;
        m_pid = -1;
        m_maxrange = 256*1024;
        
        m_begin_valid = 0;
        m_end_valid = 0;
+       
+       m_use_streaminfo = 0;
 }
 
 eDVBTSTools::~eDVBTSTools()
@@ -23,16 +24,25 @@ eDVBTSTools::~eDVBTSTools()
 int eDVBTSTools::openFile(const char *filename)
 {
        closeFile();
-       m_fd = ::open(filename, O_RDONLY);
-       if (m_fd < 0)
+       
+       m_streaminfo.load((std::string(filename) + ".ap").c_str());
+       
+       if (!m_streaminfo.empty())
+               m_use_streaminfo = 1;
+       else
+       {
+               eDebug("no recorded stream information available");
+               m_use_streaminfo = 0;
+       }
+
+       if (m_file.open(filename) < 0)
                return -1;
        return 0;
 }
 
 void eDVBTSTools::closeFile()
 {
-       if (m_fd >= 0)
-               ::close(m_fd);
+       m_file.close();
 }
 
 void eDVBTSTools::setSyncPID(int pid)
@@ -46,15 +56,17 @@ void eDVBTSTools::setSearchRange(int maxrange)
 }
 
        /* getPTS extracts a pts value from any PID at a given offset. */
-int eDVBTSTools::getPTS(off_t &offset, pts_t &pts)
+int eDVBTSTools::getPTS(off_t &offset, pts_t &pts, int fixed)
 {
-       if (m_fd < 0)
+       if (m_use_streaminfo)
+               return m_streaminfo.getPTS(offset, pts);
+       
+       if (!m_file.valid() < 0)
                return -1;
 
        offset -= offset % 188;
        
-               // TODO: multiple files!        
-       if (lseek(m_fd, offset, SEEK_SET) < 0)
+       if (m_file.lseek(offset, SEEK_SET) < 0)
                return -1;
 
        int left = m_maxrange;
@@ -62,8 +74,11 @@ int eDVBTSTools::getPTS(off_t &offset, pts_t &pts)
        while (left >= 188)
        {
                unsigned char block[188];
-               if (read(m_fd, block, 188) != 188)
+               if (m_file.read(block, 188) != 188)
+               {
+                       eDebug("read error");
                        break;
+               }
                left -= 188;
                offset += 188;
                
@@ -76,7 +91,7 @@ int eDVBTSTools::getPTS(off_t &offset, pts_t &pts)
                                        break;
                                ++i;
                        }
-                       offset = lseek(m_fd, i - 188, SEEK_CUR);
+                       offset = m_file.lseek(i - 188, SEEK_CUR);
                        continue;
                }
                
@@ -112,6 +127,10 @@ int eDVBTSTools::getPTS(off_t &offset, pts_t &pts)
                        pts |= ((unsigned long long)(pes[12]&0xFF)) << 7;
                        pts |= ((unsigned long long)(pes[13]&0xFE)) >> 1;
                        offset -= 188;
+                       
+                               /* convert to zero-based */
+                       if (fixed)
+                               fixupPTS(offset, pts);
                        return 0;
                }
        }
@@ -119,9 +138,69 @@ int eDVBTSTools::getPTS(off_t &offset, pts_t &pts)
        return -1;
 }
 
+int eDVBTSTools::fixupPTS(const off_t &offset, pts_t &now)
+{
+       if (m_use_streaminfo)
+       {
+               return m_streaminfo.fixupPTS(offset, now);
+       } else
+       {
+                       /* for the simple case, we assume one epoch, with up to one wrap around in the middle. */
+               calcBegin();
+               if (!m_begin_valid)
+               {       
+                       eDebug("begin not valid, can't fixup");
+                       return -1;
+               }
+               
+               pts_t pos = m_pts_begin;
+               if ((now < pos) && ((pos - now) < 90000 * 10))
+               {       
+                       pos = 0;
+                       return 0;
+               }
+               
+               if (now < pos) /* wrap around */
+                       now = now + 0x200000000LL - pos;
+               else
+                       now -= pos;
+               return 0;
+       }
+}
+
+int eDVBTSTools::getOffset(off_t &offset, pts_t &pts)
+{
+       if (m_use_streaminfo)
+       {
+               offset = m_streaminfo.getAccessPoint(pts);
+               return 0;
+       } else
+       {
+               int bitrate = calcBitrate(); /* in bits/s */
+               if (bitrate <= 0)
+                       return -1;
+               
+               offset = (pts * (pts_t)bitrate) / 8ULL / 90000ULL;
+               offset -= offset % 188;
+
+               return 0;
+       }
+}
+
+int eDVBTSTools::getNextAccessPoint(pts_t &ts, const pts_t &start, int direction)
+{
+       if (m_use_streaminfo)
+               return m_streaminfo.getNextAccessPoint(ts, start, direction);
+       else
+       {
+               eDebug("can't get next access point without streaminfo");
+               return -1;
+       }
+}
+
 void eDVBTSTools::calcBegin()
 {
-       if (m_fd < 0)   
+       if (!m_file.valid())
                return;
 
        if (!m_begin_valid)
@@ -134,10 +213,10 @@ void eDVBTSTools::calcBegin()
 
 void eDVBTSTools::calcEnd()
 {
-       if (m_fd < 0)   
+       if (!m_file.valid())
                return;
        
-       off_t end = lseek(m_fd, 0, SEEK_END);
+       off_t end = m_file.lseek(0, SEEK_END);
        
        if (abs(end - m_offset_end) > 1*1024*1024)
        {