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