libhb/reader.c

   1 /* $Id: reader.c,v 1.21 2005/11/25 15:05:25 titer Exp $
   2
   3    This file is part of the HandBrake source code.
   4    Homepage: <http://handbrake.m0k.org/>.
   5    It may be used under the terms of the GNU General Public License. */
   6
   7 #include "hb.h"
   8
   9 typedef struct
  10 {
  11     hb_job_t     * job;
  12     hb_title_t   * title;
  13     volatile int * die;
  14
  15     hb_dvd_t     * dvd;
  16     hb_buffer_t  * ps;
  17
  18 } hb_reader_t;
  19
  20 /***********************************************************************
  21  * Local prototypes
  22  **********************************************************************/
  23 static void        ReaderFunc( void * );
  24 static hb_fifo_t * GetFifoForId( hb_job_t * job, int id );
  25
  26 /***********************************************************************
  27  * hb_reader_init
  28  ***********************************************************************
  29  *
  30  **********************************************************************/
  31 hb_thread_t * hb_reader_init( hb_job_t * job )
  32 {
  33     hb_reader_t * r;
  34
  35     r = calloc( sizeof( hb_reader_t ), 1 );
  36
  37     r->job   = job;
  38     r->title = job->title;
  39     r->die   = job->die;
  40
  41     return hb_thread_init( "reader", ReaderFunc, r,
  42                            HB_NORMAL_PRIORITY );
  43 }
  44
  45 /***********************************************************************
  46  * ReaderFunc
  47  ***********************************************************************
  48  *
  49  **********************************************************************/
  50 static void ReaderFunc( void * _r )
  51 {
  52     hb_reader_t  * r = _r;
  53     hb_fifo_t    * fifo;
  54     hb_buffer_t  * buf;
  55     hb_list_t    * list;
  56     int            chapter;
  57
  58     if( !( r->dvd = hb_dvd_init( r->title->dvd ) ) )
  59     {
  60         return;
  61     }
  62
  63     if( !hb_dvd_start( r->dvd, r->title->index, r->job->chapter_start ) )
  64     {
  65         hb_dvd_close( &r->dvd );
  66         return;
  67     }
  68
  69     list  = hb_list_init();
  70     r->ps = hb_buffer_init( 2048 );
  71
  72     while( !*r->die && !r->job->done )
  73     {
  74         chapter = hb_dvd_chapter( r->dvd );
  75         if( chapter < 0 )
  76         {
  77             hb_log( "reader: end of the title reached" );
  78             break;
  79         }
  80         if( chapter > r->job->chapter_end )
  81         {
  82             hb_log( "reader: end of chapter %d reached (%d)",
  83                     r->job->chapter_end, chapter );
  84             break;
  85         }
  86
  87         if( !hb_dvd_read( r->dvd, r->ps ) )
  88         {
  89             break;
  90         }
  91
  92         hb_demux_ps( r->ps, list );
  93
  94         while( ( buf = hb_list_item( list, 0 ) ) )
  95         {
  96             hb_list_rem( list, buf );
  97             fifo = GetFifoForId( r->job, buf->id );
  98             if( fifo )
  99             {
 100                 while( !*r->die && !r->job->done &&
 101                        hb_fifo_is_full( fifo ) )
 102                 {
 103                     hb_snooze( 50 );
 104                 }
 105                 hb_fifo_push( fifo, buf );
 106             }
 107             else
 108             {
 109                 hb_buffer_close( &buf );
 110             }
 111         }
 112     }
 113
 114     hb_list_empty( &list );
 115     hb_buffer_close( &r->ps );
 116     hb_dvd_stop( r->dvd );
 117     hb_dvd_close( &r->dvd );
 118
 119     hb_log( "reader: done" );
 120 }
 121
 122 /***********************************************************************
 123  * GetFifoForId
 124  ***********************************************************************
 125  *
 126  **********************************************************************/
 127 static hb_fifo_t * GetFifoForId( hb_job_t * job, int id )
 128 {
 129     hb_title_t    * title = job->title;
 130     hb_audio_t    * audio;
 131     hb_subtitle_t * subtitle;
 132     int             i;
 133
 134     if( id == 0xE0 )
 135     {
 136         return job->fifo_mpeg2;
 137     }
 138
 139     if( ( subtitle = hb_list_item( title->list_subtitle, 0 ) ) &&
 140         id == subtitle->id )
 141     {
 142         return subtitle->fifo_in;
 143     }
 144
 145     for( i = 0; i < hb_list_count( title->list_audio ); i++ )
 146     {
 147         audio = hb_list_item( title->list_audio, i );
 148         if( id == audio->id )
 149         {
 150             return audio->fifo_in;
 151         }
 152     }
 153
 154     return NULL;
 155 }
 156