From 2e0106d686bb69faa1182dabbceb58969a93956f Mon Sep 17 00:00:00 2001
From: awk <awk@awkward.org>
Date: Wed, 4 Jul 2007 02:14:42 +0000
Subject: Implement transport and program stream support. With these changes
 it's now possible to open a .ts or .mpg files and transcode to standard
 Handbrake Output files. This fixes Ticket #21.

git-svn-id: svn://svn.handbrake.fr/HandBrake/trunk@648 b64f7644-9d1e-0410-96f1-a4d463321fa5
---
 libhb/reader.c | 61 ++++++++++++++++++++++++++++++++++++++++++++++------------
 1 file changed, 49 insertions(+), 12 deletions(-)

(limited to 'libhb/reader.c')

diff --git a/libhb/reader.c b/libhb/reader.c
index 4767f87a7..335dfb80d 100644
--- a/libhb/reader.c
+++ b/libhb/reader.c
@@ -14,6 +14,7 @@ typedef struct
 
     hb_dvd_t     * dvd;
     hb_buffer_t  * ps;
+    hb_stream_t  * stream;
 
 } hb_reader_t;
 
@@ -53,25 +54,43 @@ static void ReaderFunc( void * _r )
     hb_fifo_t    * fifo;
     hb_buffer_t  * buf;
     hb_list_t    * list;
-    int            chapter;
+    int            chapter = -1;
 
     if( !( r->dvd = hb_dvd_init( r->title->dvd ) ) )
     {
-        return;
+        if ( !(r->stream = hb_stream_open(r->title->dvd) ) )
+        {
+          return;
+        }
     }
 
-    if( !hb_dvd_start( r->dvd, r->title->index, r->job->chapter_start ) )
+    if (r->dvd)
     {
-        hb_dvd_close( &r->dvd );
-        return;
+      if( !hb_dvd_start( r->dvd, r->title->index, r->job->chapter_start ) )
+      {
+          hb_dvd_close( &r->dvd );
+          return;
+      }
     }
-
+    
+	if (r->stream)
+	{
+		// At this point r->audios[0] gives us the index of the selected audio track for output track 0
+		// we cannot effectively demux multiple PID's into the seperate output tracks unfortunately
+		// so we'll just specifiy things here for a single track.
+		hb_stream_set_selected_audio_pid_index(r->stream, r->job->audios[0]);
+	}
+	
     list  = hb_list_init();
-    r->ps = hb_buffer_init( 2048 );
+    r->ps = hb_buffer_init( HB_DVD_READ_BUFFER_SIZE );
 
     while( !*r->die && !r->job->done )
     {
-        chapter = hb_dvd_chapter( r->dvd );
+        if (r->dvd)
+          chapter = hb_dvd_chapter( r->dvd );
+        else if (r->stream)
+          chapter = 1;
+          
         if( chapter < 0 )
         {
             hb_log( "reader: end of the title reached" );
@@ -84,11 +103,21 @@ static void ReaderFunc( void * _r )
             break;
         }
 
-        if( !hb_dvd_read( r->dvd, r->ps ) )
+        if (r->dvd)
+        {
+          if( !hb_dvd_read( r->dvd, r->ps ) )
+          {
+              break;
+          }
+        }
+        else if (r->stream)
         {
+          if ( !hb_stream_read( r->stream, r->ps ) )
+          {
             break;
+          }
         }
-
+        
         hb_demux_ps( r->ps, list );
 
         while( ( buf = hb_list_item( list, 0 ) ) )
@@ -113,8 +142,16 @@ static void ReaderFunc( void * _r )
 
     hb_list_empty( &list );
     hb_buffer_close( &r->ps );
-    hb_dvd_stop( r->dvd );
-    hb_dvd_close( &r->dvd );
+    if (r->dvd)
+    {
+      hb_dvd_stop( r->dvd );
+      hb_dvd_close( &r->dvd );
+    }
+    else if (r->stream)
+    {
+      hb_stream_close(&r->stream);
+    }
+    
     free( r );
     _r = NULL;
 
-- 
cgit v1.2.3