libhb/work.c

   1 /* $Id: work.c,v 1.43 2005/03/17 16:38:49 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_list_t * jobs;
  12     int         cpu_count;
  13     int       * error;
  14     volatile int * die;
  15
  16 } hb_work_t;
  17
  18 static void work_func();
  19 static void do_job( hb_job_t *, int cpu_count );
  20 static void work_loop( void * );
  21
  22 hb_thread_t * hb_work_init( hb_list_t * jobs, int cpu_count,
  23                             volatile int * die, int * error )
  24 {
  25     hb_work_t * work = calloc( sizeof( hb_work_t ), 1 );
  26
  27     work->jobs      = jobs;
  28     work->cpu_count = cpu_count;
  29     work->die       = die;
  30     work->error     = error;
  31
  32     return hb_thread_init( "work", work_func, work, HB_LOW_PRIORITY );
  33 }
  34
  35 static void work_func( void * _work )
  36 {
  37     hb_work_t  * work = _work;
  38     hb_job_t   * job;
  39
  40     hb_log( "%d job(s) to process", hb_list_count( work->jobs ) );
  41
  42     while( !*work->die && ( job = hb_list_item( work->jobs, 0 ) ) )
  43     {
  44         hb_list_rem( work->jobs, job );
  45         job->die = work->die;
  46         do_job( job, work->cpu_count );
  47     }
  48
  49     *(work->error) = HB_ERROR_NONE;
  50
  51     free( work );
  52 }
  53
  54 static hb_work_object_t * getWork( int id )
  55 {
  56     hb_work_object_t * w;
  57     for( w = hb_objects; w; w = w->next )
  58     {
  59         if( w->id == id )
  60         {
  61             return w;
  62         }
  63     }
  64     return NULL;
  65 }
  66
  67 static void do_job( hb_job_t * job, int cpu_count )
  68 {
  69     hb_title_t    * title;
  70     int             i;
  71     hb_work_object_t * w;
  72     hb_audio_t   * audio;
  73     hb_subtitle_t * subtitle;
  74
  75     title = job->title;
  76
  77     job->list_work = hb_list_init();
  78
  79     hb_log( "starting job" );
  80     hb_log( " + device %s", title->dvd );
  81     hb_log( " + title %d, chapter(s) %d to %d", title->index,
  82             job->chapter_start, job->chapter_end );
  83     hb_log( " + %dx%d -> %dx%d, crop %d/%d/%d/%d",
  84             title->width, title->height, job->width, job->height,
  85             job->crop[0], job->crop[1], job->crop[2], job->crop[3] );
  86     hb_log( " + deinterlace %s", job->deinterlace ? "on" : "off" );
  87     hb_log( " + grayscale %s", job->grayscale ? "on" : "off" );
  88     if( job->vquality >= 0.0 && job->vquality <= 1.0 )
  89     {
  90         hb_log( " + %.3f fps, video quality %.2f", (float) job->vrate /
  91                 (float) job->vrate_base, job->vquality );
  92     }
  93     else
  94     {
  95         hb_log( " + %.3f fps, video bitrate %d kbps, pass %d",
  96                 (float) job->vrate / (float) job->vrate_base,
  97                 job->vbitrate, job->pass );
  98     }
  99
 100     job->fifo_mpeg2  = hb_fifo_init( 2048 );
 101     job->fifo_raw    = hb_fifo_init( 8 );
 102     job->fifo_sync   = hb_fifo_init( 8 );
 103     job->fifo_render = hb_fifo_init( 8 );
 104     job->fifo_mpeg4  = hb_fifo_init( 8 );
 105
 106     /* Synchronization */
 107     hb_list_add( job->list_work, ( w = getWork( WORK_SYNC ) ) );
 108     w->fifo_in  = NULL;
 109     w->fifo_out = NULL;
 110
 111     /* Video decoder */
 112     hb_list_add( job->list_work, ( w = getWork( WORK_DECMPEG2 ) ) );
 113     w->fifo_in  = job->fifo_mpeg2;
 114     w->fifo_out = job->fifo_raw;
 115
 116     /* Video renderer */
 117     hb_list_add( job->list_work, ( w = getWork( WORK_RENDER ) ) );
 118     w->fifo_in  = job->fifo_sync;
 119     w->fifo_out = job->fifo_render;
 120
 121     /* Video encoder */
 122     switch( job->vcodec )
 123     {
 124         case HB_VCODEC_FFMPEG:
 125             hb_log( " + encoder FFmpeg" );
 126             w = getWork( WORK_ENCAVCODEC );
 127             break;
 128         case HB_VCODEC_XVID:
 129             hb_log( " + encoder XviD" );
 130             w = getWork( WORK_ENCXVID );
 131             break;
 132         case HB_VCODEC_X264:
 133             hb_log( " + encoder x264" );
 134             w = getWork( WORK_ENCX264 );
 135             break;
 136     }
 137     w->fifo_in  = job->fifo_render;
 138     w->fifo_out = job->fifo_mpeg4;
 139     w->config   = &job->config;
 140     hb_list_add( job->list_work, w );
 141
 142     subtitle = hb_list_item( title->list_subtitle, 0 );
 143     if( subtitle )
 144     {
 145         hb_log( " + subtitle %x, %s", subtitle->id, subtitle->lang );
 146
 147         subtitle->fifo_in  = hb_fifo_init( 8 );
 148         subtitle->fifo_raw = hb_fifo_init( 8 );
 149
 150         hb_list_add( job->list_work, ( w = getWork( WORK_DECSUB ) ) );
 151         w->fifo_in  = subtitle->fifo_in;
 152         w->fifo_out = subtitle->fifo_raw;
 153     }
 154
 155     if( job->acodec & HB_ACODEC_AC3 )
 156     {
 157         hb_log( " + audio AC3 passthrough" );
 158     }
 159     else
 160     {
 161         hb_log( " + audio %d kbps, %d Hz", job->abitrate, job->arate );
 162         hb_log( " + encoder %s", ( job->acodec & HB_ACODEC_FAAC ) ?
 163                 "faac" : ( ( job->acodec & HB_ACODEC_LAME ) ? "lame" :
 164                 "vorbis" ) );
 165     }
 166
 167     for( i = 0; i < hb_list_count( title->list_audio ); i++ )
 168     {
 169         audio = hb_list_item( title->list_audio, i );
 170         hb_log( "   + %x, %s", audio->id, audio->lang );
 171
 172         audio->fifo_in   = hb_fifo_init( 2048 );
 173         audio->fifo_raw  = hb_fifo_init( 8 );
 174         audio->fifo_sync = hb_fifo_init( 8 );
 175         audio->fifo_out  = hb_fifo_init( 8 );
 176
 177         switch( audio->codec )
 178         {
 179             case HB_ACODEC_AC3:
 180                 w = getWork( WORK_DECA52 );
 181                 break;
 182             case HB_ACODEC_MPGA:
 183                 w = getWork( WORK_DECAVCODEC );
 184                 break;
 185             case HB_ACODEC_LPCM:
 186                 w = getWork( WORK_DECLPCM );
 187                 break;
 188         }
 189         w->fifo_in  = audio->fifo_in;
 190         w->fifo_out = audio->fifo_raw;
 191         hb_list_add( job->list_work, w );
 192
 193         switch( job->acodec )
 194         {
 195             case HB_ACODEC_FAAC:
 196                 w = getWork( WORK_ENCFAAC );
 197                 break;
 198             case HB_ACODEC_LAME:
 199                 w = getWork( WORK_ENCLAME );
 200                 break;
 201             case HB_ACODEC_VORBIS:
 202                 w = getWork( WORK_ENCVORBIS );
 203                 break;
 204         }
 205         if( job->acodec != HB_ACODEC_AC3 )
 206         {
 207             w->fifo_in  = audio->fifo_sync;
 208             w->fifo_out = audio->fifo_out;
 209             w->config   = &audio->config;
 210             hb_list_add( job->list_work, w );
 211         }
 212     }
 213
 214     /* Init read & write threads */
 215     job->reader = hb_reader_init( job );
 216
 217     hb_log( " + output: %s", job->file );
 218     job->muxer = hb_muxer_init( job );
 219
 220     job->done = 0;
 221
 222     /* Launch processing threads */
 223     for( i = 1; i < hb_list_count( job->list_work ); i++ )
 224     {
 225         w = hb_list_item( job->list_work, i );
 226         w->done = &job->done;
 227         w->init( w, job );
 228         w->thread = hb_thread_init( w->name, work_loop, w,
 229                                     HB_LOW_PRIORITY );
 230     }
 231
 232     int done = 0;
 233     w = hb_list_item( job->list_work, 0 );
 234     w->init( w, job );
 235     while( !*job->die )
 236     {
 237         if( w->work( w, NULL, NULL ) == HB_WORK_DONE )
 238         {
 239             done = 1;
 240         }
 241         if( done &&
 242             !hb_fifo_size( job->fifo_sync ) &&
 243             !hb_fifo_size( job->fifo_render ) &&
 244             hb_fifo_size( job->fifo_mpeg4 ) < 2 )
 245         {
 246             break;
 247         }
 248         hb_snooze( 50 );
 249     }
 250     hb_list_rem( job->list_work, w );
 251     w->close( w );
 252     job->done = 1;
 253
 254     /* Close work objects */
 255     while( ( w = hb_list_item( job->list_work, 0 ) ) )
 256     {
 257         hb_list_rem( job->list_work, w );
 258         hb_thread_close( &w->thread );
 259         w->close( w );
 260     }
 261
 262     /* Stop read & write threads */
 263     hb_thread_close( &job->reader );
 264     hb_thread_close( &job->muxer );
 265
 266     /* Close fifos */
 267     hb_fifo_close( &job->fifo_mpeg2 );
 268     hb_fifo_close( &job->fifo_raw );
 269     hb_fifo_close( &job->fifo_sync );
 270     hb_fifo_close( &job->fifo_render );
 271     hb_fifo_close( &job->fifo_mpeg4 );
 272     if( subtitle )
 273     {
 274         hb_fifo_close( &subtitle->fifo_in );
 275         hb_fifo_close( &subtitle->fifo_raw );
 276     }
 277     for( i = 0; i < hb_list_count( title->list_audio ); i++ )
 278     {
 279         audio = hb_list_item( title->list_audio, i );
 280         hb_fifo_close( &audio->fifo_in );
 281         hb_fifo_close( &audio->fifo_raw );
 282         hb_fifo_close( &audio->fifo_sync );
 283         hb_fifo_close( &audio->fifo_out );
 284     }
 285 }
 286
 287 static void work_loop( void * _w )
 288 {
 289     hb_work_object_t * w = _w;
 290     hb_buffer_t      * buf_in, * buf_out;
 291
 292     while( !*w->done )
 293     {
 294 #if 0
 295         hb_lock( job->pause );
 296         hb_unlock( job->pause );
 297 #endif
 298         if( hb_fifo_is_full( w->fifo_out ) ||
 299             !( buf_in = hb_fifo_get( w->fifo_in ) ) )
 300         {
 301             hb_snooze( 50 );
 302             continue;
 303         }
 304
 305         w->work( w, &buf_in, &buf_out );
 306         if( buf_in )
 307         {
 308             hb_buffer_close( &buf_in );
 309         }
 310         if( buf_out )
 311         {
 312             hb_fifo_push( w->fifo_out, buf_out );
 313         }
 314     }
 315 }