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.fr/>.
   5    It may be used under the terms of the GNU General Public License. */
   6
   7 #include "hb.h"
   8 #include "a52dec/a52.h"
   9 #include "dca.h"
  10 #include "libavformat/avformat.h"
  11
  12 typedef struct
  13 {
  14     hb_list_t * jobs;
  15     hb_job_t  ** current_job;
  16     int         cpu_count;
  17     int       * error;
  18     volatile int * die;
  19
  20 } hb_work_t;
  21
  22 static void work_func();
  23 static void do_job( hb_job_t *, int cpu_count );
  24 static void work_loop( void * );
  25
  26 #define FIFO_UNBOUNDED 65536
  27 #define FIFO_UNBOUNDED_WAKE 65535
  28 #define FIFO_LARGE 32
  29 #define FIFO_LARGE_WAKE 16
  30 #define FIFO_SMALL 16
  31 #define FIFO_SMALL_WAKE 15
  32
  33 /**
  34  * Allocates work object and launches work thread with work_func.
  35  * @param jobs Handle to hb_list_t.
  36  * @param cpu_count Humber of CPUs found in system.
  37  * @param die Handle to user inititated exit indicator.
  38  * @param error Handle to error indicator.
  39  */
  40 hb_thread_t * hb_work_init( hb_list_t * jobs, int cpu_count,
  41                             volatile int * die, int * error, hb_job_t ** job )
  42 {
  43     hb_work_t * work = calloc( sizeof( hb_work_t ), 1 );
  44
  45     work->jobs      = jobs;
  46     work->current_job = job;
  47     work->cpu_count = cpu_count;
  48     work->die       = die;
  49     work->error     = error;
  50
  51     return hb_thread_init( "work", work_func, work, HB_LOW_PRIORITY );
  52 }
  53
  54 static void InitWorkState( hb_handle_t * h )
  55 {
  56     hb_state_t state;
  57
  58     state.state = HB_STATE_WORKING;
  59 #define p state.param.working
  60     p.progress  = 0.0;
  61     p.rate_cur  = 0.0;
  62     p.rate_avg  = 0.0;
  63     p.hours     = -1;
  64     p.minutes   = -1;
  65     p.seconds   = -1;
  66 #undef p
  67
  68     hb_set_state( h, &state );
  69
  70 }
  71
  72 /**
  73  * Iterates through job list and calls do_job for each job.
  74  * @param _work Handle work object.
  75  */
  76 static void work_func( void * _work )
  77 {
  78     hb_work_t  * work = _work;
  79     hb_job_t   * job;
  80
  81     hb_log( "%d job(s) to process", hb_list_count( work->jobs ) );
  82
  83     while( !*work->die && ( job = hb_list_item( work->jobs, 0 ) ) )
  84     {
  85         hb_list_rem( work->jobs, job );
  86         job->die = work->die;
  87         *(work->current_job) = job;
  88         InitWorkState( job->h );
  89         do_job( job, work->cpu_count );
  90         *(work->current_job) = NULL;
  91     }
  92
  93     *(work->error) = HB_ERROR_NONE;
  94
  95     free( work );
  96 }
  97
  98 hb_work_object_t * hb_get_work( int id )
  99 {
 100     hb_work_object_t * w;
 101     for( w = hb_objects; w; w = w->next )
 102     {
 103         if( w->id == id )
 104         {
 105             hb_work_object_t *wc = malloc( sizeof(*w) );
 106             *wc = *w;
 107             return wc;
 108         }
 109     }
 110     return NULL;
 111 }
 112
 113 hb_work_object_t * hb_codec_decoder( int codec )
 114 {
 115     switch( codec )
 116     {
 117         case HB_ACODEC_AC3:  return hb_get_work( WORK_DECA52 );
 118         case HB_ACODEC_DCA:  return hb_get_work( WORK_DECDCA );
 119         case HB_ACODEC_MPGA: return hb_get_work( WORK_DECAVCODEC );
 120         case HB_ACODEC_LPCM: return hb_get_work( WORK_DECLPCM );
 121         case HB_ACODEC_FFMPEG: return hb_get_work( WORK_DECAVCODECAI );
 122     }
 123     return NULL;
 124 }
 125
 126 hb_work_object_t * hb_codec_encoder( int codec )
 127 {
 128     switch( codec )
 129     {
 130         case HB_ACODEC_FAAC:   return hb_get_work( WORK_ENCFAAC );
 131         case HB_ACODEC_LAME:   return hb_get_work( WORK_ENCLAME );
 132         case HB_ACODEC_VORBIS: return hb_get_work( WORK_ENCVORBIS );
 133         case HB_ACODEC_CA_AAC: return hb_get_work( WORK_ENC_CA_AAC );
 134         case HB_ACODEC_AC3:    return hb_get_work( WORK_ENCAC3 );
 135     }
 136     return NULL;
 137 }
 138
 139 /**
 140  * Displays job parameters in the debug log.
 141  * @param job Handle work hb_job_t.
 142  */
 143 void hb_display_job_info( hb_job_t * job )
 144 {
 145     hb_title_t * title = job->title;
 146     hb_audio_t   * audio;
 147     hb_subtitle_t * subtitle;
 148     int             i, j;
 149
 150     hb_log("job configuration:");
 151     hb_log( " * source");
 152
 153     hb_log( "   + %s", title->path );
 154
 155     hb_log( "   + title %d, chapter(s) %d to %d", title->index,
 156             job->chapter_start, job->chapter_end );
 157
 158     if( title->container_name != NULL )
 159         hb_log( "   + container: %s", title->container_name);
 160
 161     if( title->data_rate )
 162     {
 163         hb_log( "   + data rate: %d kbps", title->data_rate / 1000 );
 164     }
 165
 166     hb_log( " * destination");
 167
 168     hb_log( "   + %s", job->file );
 169
 170     switch( job->mux )
 171     {
 172         case HB_MUX_MP4:
 173             hb_log("   + container: MPEG-4 (.mp4 and .m4v)");
 174
 175             if( job->ipod_atom )
 176                 hb_log( "     + compatibility atom for iPod 5G");
 177
 178             if( job->largeFileSize )
 179                 hb_log( "     + 64-bit formatting");
 180
 181             if( job->mp4_optimize )
 182                 hb_log( "     + optimized for progressive web downloads");
 183
 184             if( job->color_matrix )
 185                 hb_log( "     + custom color matrix: %s", job->color_matrix == 1 ? "ITU Bt.601 (SD)" : "ITU Bt.709 (HD)");
 186             break;
 187
 188         case HB_MUX_MKV:
 189             hb_log("   + container: Matroska (.mkv)");
 190             break;
 191     }
 192
 193     if( job->chapter_markers )
 194     {
 195         hb_log( "     + chapter markers" );
 196     }
 197
 198     hb_log(" * video track");
 199
 200     hb_log("   + decoder: %s", title->video_codec_name );
 201
 202     if( title->video_bitrate )
 203     {
 204         hb_log( "     + bitrate %d kbps", title->video_bitrate / 1000 );
 205     }
 206
 207     if( !job->cfr )
 208     {
 209         hb_log( "   + frame rate: same as source (around %.3f fps)",
 210             (float) title->rate / (float) title->rate_base );
 211     }
 212     else
 213     {
 214         static const char *frtypes[] = {
 215             "", "constant", "peak rate limited to"
 216         };
 217         hb_log( "   + frame rate: %.3f fps -> %s %.3f fps",
 218             (float) title->rate / (float) title->rate_base, frtypes[job->cfr],
 219             (float) job->vrate / (float) job->vrate_base );
 220     }
 221
 222     if( job->anamorphic.mode )
 223     {
 224         hb_log( "   + %s anamorphic", job->anamorphic.mode == 1 ? "strict" : job->anamorphic.mode == 2? "loose" : "custom" );
 225         if( job->anamorphic.mode == 3 && job->anamorphic.keep_display_aspect )
 226         {
 227             hb_log( "     + keeping source display aspect ratio");
 228         }
 229         hb_log( "     + storage dimensions: %d * %d -> %d * %d, crop %d/%d/%d/%d, mod %i",
 230                     title->width, title->height, job->width, job->height,
 231                     job->crop[0], job->crop[1], job->crop[2], job->crop[3], job->modulus );
 232         if( job->anamorphic.itu_par )
 233         {
 234             hb_log( "     + using ITU pixel aspect ratio values");
 235         }
 236         hb_log( "     + pixel aspect ratio: %i / %i", job->anamorphic.par_width, job->anamorphic.par_height );
 237         hb_log( "     + display dimensions: %.0f * %i",
 238             (float)( job->width * job->anamorphic.par_width / job->anamorphic.par_height ), job->height );
 239     }
 240     else
 241     {
 242         hb_log( "   + dimensions: %d * %d -> %d * %d, crop %d/%d/%d/%d, mod %i",
 243                 title->width, title->height, job->width, job->height,
 244                 job->crop[0], job->crop[1], job->crop[2], job->crop[3], job->modulus );
 245     }
 246
 247     if ( job->grayscale )
 248         hb_log( "   + grayscale mode" );
 249
 250     if( hb_list_count( job->filters ) )
 251     {
 252         hb_log("   + %s", hb_list_count( job->filters) > 1 ? "filters" : "filter" );
 253         for( i = 0; i < hb_list_count( job->filters ); i++ )
 254         {
 255             hb_filter_object_t * filter = hb_list_item( job->filters, i );
 256             if (filter->settings)
 257                 hb_log("     + %s (%s)", filter->name, filter->settings);
 258             else
 259                 hb_log("     + %s (default settings)", filter->name);
 260         }
 261     }
 262
 263     if( !job->indepth_scan )
 264     {
 265         /* Video encoder */
 266         switch( job->vcodec )
 267         {
 268             case HB_VCODEC_FFMPEG:
 269                 hb_log( "   + encoder: FFmpeg" );
 270                 break;
 271
 272             case HB_VCODEC_X264:
 273                 hb_log( "   + encoder: x264" );
 274                 if( job->x264opts != NULL && *job->x264opts != '\0' )
 275                     hb_log( "     + options: %s", job->x264opts);
 276                 break;
 277
 278             case HB_VCODEC_THEORA:
 279                 hb_log( "   + encoder: Theora" );
 280                 break;
 281         }
 282
 283         if( job->vquality >= 0.0 && job->vquality <= 1.0 )
 284         {
 285             hb_log( "     + quality: %.2f", job->vquality );
 286         }
 287         else if( job->vquality > 1 )
 288         {
 289             hb_log( "     + quality: %.2f %s", job->vquality, job->vcodec == HB_VCODEC_X264 ? "(RF)" : "(QP)" );
 290         }
 291         else
 292         {
 293             hb_log( "     + bitrate: %d kbps, pass: %d", job->vbitrate, job->pass );
 294         }
 295     }
 296
 297     for( i=0; i < hb_list_count( title->list_subtitle ); i++ )
 298     {
 299         subtitle =  hb_list_item( title->list_subtitle, i );
 300
 301         if( subtitle )
 302         {
 303             if( subtitle->source == SRTSUB )
 304             {
 305                 /* For SRT, print offset and charset too */
 306                 hb_log( " * subtitle track %i, %s (id %x) %s [%s] -> %s%s, offset: %"PRId64", charset: %s",
 307                         subtitle->track, subtitle->lang, subtitle->id, "Text", "SRT", "Pass-Through",
 308                         subtitle->config.default_track ? ", Default" : "",
 309                         subtitle->config.offset, subtitle->config.src_codeset );
 310             }
 311             else
 312             {
 313                 hb_log( " * subtitle track %i, %s (id %x) %s [%s] -> %s%s%s", subtitle->track, subtitle->lang, subtitle->id,
 314                         subtitle->format == PICTURESUB ? "Picture" : "Text",
 315                         hb_subsource_name( subtitle->source ),
 316                         job->indepth_scan ? "Foreign Audio Search" :
 317                         subtitle->config.dest == RENDERSUB ? "Render/Burn in" : "Pass-Through",
 318                         subtitle->config.force ? ", Forced Only" : "",
 319                         subtitle->config.default_track ? ", Default" : "" );
 320             }
 321         }
 322     }
 323
 324     if( !job->indepth_scan )
 325     {
 326         for( i = 0; i < hb_list_count( title->list_audio ); i++ )
 327         {
 328             audio = hb_list_item( title->list_audio, i );
 329
 330             hb_log( " * audio track %d", audio->config.out.track );
 331
 332             if( audio->config.out.name )
 333                 hb_log( "   + name: %s", audio->config.out.name );
 334
 335             hb_log( "   + decoder: %s (track %d, id %x)", audio->config.lang.description, audio->config.in.track + 1, audio->id );
 336
 337             if( ( audio->config.in.codec == HB_ACODEC_AC3 ) || ( audio->config.in.codec == HB_ACODEC_DCA) )
 338             {
 339                 hb_log( "     + bitrate: %d kbps, samplerate: %d Hz", audio->config.in.bitrate / 1000, audio->config.in.samplerate );
 340             }
 341
 342             if( (audio->config.out.codec != HB_ACODEC_AC3_PASS) && (audio->config.out.codec != HB_ACODEC_DCA_PASS) )
 343             {
 344                 for (j = 0; j < hb_audio_mixdowns_count; j++)
 345                 {
 346                     if (hb_audio_mixdowns[j].amixdown == audio->config.out.mixdown) {
 347                         hb_log( "   + mixdown: %s", hb_audio_mixdowns[j].human_readable_name );
 348                         break;
 349                     }
 350                 }
 351             }
 352
 353             if ( audio->config.out.dynamic_range_compression && (audio->config.out.codec != HB_ACODEC_AC3_PASS) && (audio->config.out.codec != HB_ACODEC_DCA_PASS))
 354             {
 355                 hb_log("   + dynamic range compression: %f", audio->config.out.dynamic_range_compression);
 356             }
 357
 358             if( (audio->config.out.codec == HB_ACODEC_AC3_PASS) || (audio->config.out.codec == HB_ACODEC_DCA_PASS) )
 359             {
 360                 hb_log( "   + %s passthrough", (audio->config.out.codec == HB_ACODEC_AC3_PASS) ?
 361                     "AC3" : "DCA" );
 362             }
 363             else
 364             {
 365                 hb_log( "   + encoder: %s",
 366                     ( audio->config.out.codec == HB_ACODEC_FAAC ) ?  "faac" :
 367                     ( ( audio->config.out.codec == HB_ACODEC_LAME ) ?  "lame" :
 368                     ( ( audio->config.out.codec == HB_ACODEC_CA_AAC ) ?  "ca_aac" :
 369                     ( ( audio->config.out.codec == HB_ACODEC_AC3 ) ?  "ffac3" :
 370                     "vorbis"  ) ) ) );
 371                 hb_log( "     + bitrate: %d kbps, samplerate: %d Hz", audio->config.out.bitrate, audio->config.out.samplerate );
 372             }
 373         }
 374     }
 375 }
 376
 377 /* Corrects framerates when actual duration and frame count numbers are known. */
 378 void correct_framerate( hb_job_t * job )
 379 {
 380     int real_frames;
 381
 382     hb_interjob_t * interjob = hb_interjob_get( job->h );
 383
 384     if( ( job->sequence_id & 0xFFFFFF ) != ( interjob->last_job & 0xFFFFFF) )
 385         return; // Interjob information is for a different encode.
 386
 387     /* Cache the original framerate before altering it. */
 388     interjob->vrate = job->vrate;
 389     interjob->vrate_base = job->vrate_base;
 390
 391     real_frames = interjob->frame_count - interjob->render_dropped;
 392
 393     job->vrate = job->vrate_base * ( (double)real_frames * 90000 / interjob->total_time );
 394 }
 395
 396 /**
 397  * Job initialization rountine.
 398  * Initializes fifos.
 399  * Creates work objects for synchronizer, video decoder, video renderer, video decoder, audio decoder, audio encoder, reader, muxer.
 400  * Launches thread for each work object with work_loop.
 401  * Loops while monitoring status of work threads and fifos.
 402  * Exits loop when conversion is done and fifos are empty.
 403  * Closes threads and frees fifos.
 404  * @param job Handle work hb_job_t.
 405  * @param cpu_count number of CPUs found in system.
 406  */
 407 static void do_job( hb_job_t * job, int cpu_count )
 408 {
 409     hb_title_t    * title;
 410     int             i, j;
 411     hb_work_object_t * w;
 412     hb_work_object_t * sync;
 413     hb_work_object_t * muxer;
 414     hb_interjob_t * interjob;
 415
 416     hb_audio_t   * audio;
 417     hb_subtitle_t * subtitle;
 418     hb_attachment_t * attachment;
 419     unsigned int subtitle_highest = 0;
 420     unsigned int subtitle_highest_id = 0;
 421     unsigned int subtitle_lowest = -1;
 422     unsigned int subtitle_lowest_id = 0;
 423     unsigned int subtitle_forced_id = 0;
 424     unsigned int subtitle_hit = 0;
 425
 426     title = job->title;
 427     interjob = hb_interjob_get( job->h );
 428
 429     if( job->pass == 2 && !job->cfr )
 430     {
 431         correct_framerate( job );
 432     }
 433
 434     job->list_work = hb_list_init();
 435
 436     hb_log( "starting job" );
 437
 438     if( job->anamorphic.mode )
 439     {
 440         hb_set_anamorphic_size(job, &job->width, &job->height, &job->anamorphic.par_width, &job->anamorphic.par_height);
 441
 442         if( job->vcodec == HB_VCODEC_FFMPEG )
 443         {
 444             /* Just to make working with ffmpeg even more fun,
 445                lavc's MPEG-4 encoder can't handle PAR values >= 255,
 446                even though AVRational does. Adjusting downwards
 447                distorts the display aspect slightly, but such is life. */
 448             while ((job->anamorphic.par_width & ~0xFF) ||
 449                    (job->anamorphic.par_height & ~0xFF))
 450             {
 451                 job->anamorphic.par_width >>= 1;
 452                 job->anamorphic.par_height >>= 1;
 453             }
 454             hb_reduce( &job->anamorphic.par_width, &job->anamorphic.par_height, job->anamorphic.par_width, job->anamorphic.par_height );
 455         }
 456     }
 457
 458     /* Keep width and height within these boundaries,
 459        but ignore for anamorphic. For "loose" anamorphic encodes,
 460        this stuff is covered in the pixel_ratio section above.    */
 461     if ( job->maxHeight && ( job->height > job->maxHeight ) && ( !job->anamorphic.mode ) )
 462     {
 463         job->height = job->maxHeight;
 464         hb_fix_aspect( job, HB_KEEP_HEIGHT );
 465         hb_log( "Height out of bounds, scaling down to %i", job->maxHeight );
 466         hb_log( "New dimensions %i * %i", job->width, job->height );
 467     }
 468     if ( job->maxWidth && ( job->width > job->maxWidth ) && ( !job->anamorphic.mode ) )
 469     {
 470         job->width = job->maxWidth;
 471         hb_fix_aspect( job, HB_KEEP_WIDTH );
 472         hb_log( "Width out of bounds, scaling down to %i", job->maxWidth );
 473         hb_log( "New dimensions %i * %i", job->width, job->height );
 474     }
 475
 476     if ( job->cfr == 0 )
 477     {
 478         /* Ensure we're using "Same as source" FPS */
 479         job->vrate_base = title->rate_base;
 480     }
 481
 482     job->fifo_mpeg2  = hb_fifo_init( FIFO_LARGE, FIFO_LARGE_WAKE );
 483     job->fifo_raw    = hb_fifo_init( FIFO_SMALL, FIFO_SMALL_WAKE );
 484     job->fifo_sync   = hb_fifo_init( FIFO_SMALL, FIFO_SMALL_WAKE );
 485     job->fifo_render = hb_fifo_init( FIFO_SMALL, FIFO_SMALL_WAKE );
 486     job->fifo_mpeg4  = hb_fifo_init( FIFO_LARGE, FIFO_LARGE_WAKE );
 487
 488     /*
 489      * Audio fifos must be initialized before sync
 490      */
 491     if( !job->indepth_scan )
 492     {
 493     // if we are doing passthru, and the input codec is not the same as the output
 494     // codec, then remove this audio from the job. If we're not doing passthru and
 495     // the input codec is the 'internal' ffmpeg codec, make sure that only one
 496     // audio references that audio stream since the codec context is specific to
 497     // the audio id & multiple copies of the same stream will garble the audio
 498     // or cause aborts.
 499     uint8_t aud_id_uses[MAX_STREAMS];
 500     memset( aud_id_uses, 0, sizeof(aud_id_uses) );
 501     for( i = 0; i < hb_list_count( title->list_audio ); )
 502     {
 503         audio = hb_list_item( title->list_audio, i );
 504         if( ( ( audio->config.out.codec == HB_ACODEC_AC3_PASS ) && ( audio->config.in.codec != HB_ACODEC_AC3 ) ) ||
 505             ( ( audio->config.out.codec == HB_ACODEC_DCA_PASS ) && ( audio->config.in.codec != HB_ACODEC_DCA ) ) )
 506         {
 507             hb_log( "Passthru requested and input codec is not the same as output codec for track %d",
 508                     audio->config.out.track );
 509             hb_list_rem( title->list_audio, audio );
 510             free( audio );
 511             continue;
 512         }
 513         if( audio->config.out.codec != HB_ACODEC_AC3_PASS &&
 514             audio->config.out.codec != HB_ACODEC_DCA_PASS &&
 515             audio->config.out.samplerate > 48000 )
 516         {
 517             hb_log( "Sample rate %d not supported.  Down-sampling to 48kHz.",
 518                     audio->config.out.samplerate );
 519             audio->config.out.samplerate = 48000;
 520         }
 521         if( audio->config.out.codec == HB_ACODEC_AC3 &&
 522             audio->config.out.bitrate > 640 )
 523         {
 524             hb_log( "Bitrate %d not supported.  Reducing to 640Kbps.",
 525                     audio->config.out.bitrate );
 526             audio->config.out.bitrate = 640;
 527         }
 528         if ( audio->config.in.codec == HB_ACODEC_FFMPEG )
 529         {
 530             if ( aud_id_uses[audio->id] )
 531             {
 532                 hb_log( "Multiple decodes of audio id %d, removing track %d",
 533                         audio->id, audio->config.out.track );
 534                 hb_list_rem( title->list_audio, audio );
 535                 free( audio );
 536                 continue;
 537             }
 538             ++aud_id_uses[audio->id];
 539         }
 540         /* Adjust output track number, in case we removed one.
 541          * Output tracks sadly still need to be in sequential order.
 542          */
 543         audio->config.out.track = i++;
 544     }
 545
 546     int requested_mixdown = 0;
 547     int requested_mixdown_index = 0;
 548     int best_mixdown = 0;
 549     int requested_bitrate = 0;
 550     int best_bitrate = 0;
 551
 552     for( i = 0; i < hb_list_count( title->list_audio ); i++ )
 553     {
 554         audio = hb_list_item( title->list_audio, i );
 555
 556         best_mixdown = hb_get_best_mixdown( audio->config.out.codec,
 557                                             audio->config.in.channel_layout, 0 );
 558
 559         /* sense-check the current mixdown options */
 560
 561         /* sense-check the requested mixdown */
 562         if( audio->config.out.mixdown == 0 &&
 563             !( audio->config.out.codec & HB_ACODEC_PASS_FLAG ) )
 564         {
 565             /*
 566              * Mixdown wasn't specified and this is not pass-through,
 567              * set a default mixdown
 568              */
 569             audio->config.out.mixdown = hb_get_default_mixdown( audio->config.out.codec,
 570                                                                 audio->config.in.channel_layout );
 571             for (j = 0; j < hb_audio_mixdowns_count; j++)
 572             {
 573                 if (hb_audio_mixdowns[j].amixdown == audio->config.out.mixdown)
 574                 {
 575                     hb_log("work: mixdown not specified, track %i setting mixdown %s", i, hb_audio_mixdowns[j].human_readable_name);
 576                     break;
 577                 }
 578             }
 579         }
 580
 581         /* log the requested mixdown */
 582         for (j = 0; j < hb_audio_mixdowns_count; j++) {
 583             if (hb_audio_mixdowns[j].amixdown == audio->config.out.mixdown) {
 584                 requested_mixdown = audio->config.out.mixdown;
 585                 requested_mixdown_index = j;
 586                 break;
 587             }
 588         }
 589
 590         if ( !( audio->config.out.codec & HB_ACODEC_PASS_FLAG ) )
 591         {
 592             if ( audio->config.out.mixdown > best_mixdown )
 593             {
 594                 audio->config.out.mixdown = best_mixdown;
 595             }
 596         }
 597
 598         if ( audio->config.out.mixdown != requested_mixdown )
 599         {
 600             /* log the output mixdown */
 601             for (j = 0; j < hb_audio_mixdowns_count; j++)
 602             {
 603                 if (hb_audio_mixdowns[j].amixdown == audio->config.out.mixdown)
 604                 {
 605                     hb_log("work: sanitizing track %i mixdown %s to %s", i, hb_audio_mixdowns[requested_mixdown_index].human_readable_name, hb_audio_mixdowns[j].human_readable_name);
 606                     break;
 607                 }
 608             }
 609         }
 610
 611         /* sense-check the current bitrates */
 612
 613         /* sense-check the requested bitrate */
 614         if( audio->config.out.bitrate == 0 &&
 615             !( audio->config.out.codec & HB_ACODEC_PASS_FLAG ) )
 616         {
 617             /*
 618              * Bitrate wasn't specified and this is not pass-through,
 619              * set a default bitrate
 620              */
 621             audio->config.out.bitrate = hb_get_default_audio_bitrate( audio->config.out.codec,
 622                                                                       audio->config.out.samplerate,
 623                                                                       audio->config.out.mixdown );
 624
 625             hb_log( "work: bitrate not specified, track %d setting bitrate %d",
 626                     i, audio->config.out.bitrate );
 627         }
 628
 629         /* log the requested bitrate */
 630         requested_bitrate = audio->config.out.bitrate;
 631         best_bitrate = hb_get_best_audio_bitrate( audio->config.out.codec,
 632                                                   audio->config.out.bitrate,
 633                                                   audio->config.out.samplerate,
 634                                                   audio->config.out.mixdown );
 635
 636         if ( !( audio->config.out.codec & HB_ACODEC_PASS_FLAG ) )
 637         {
 638             if ( audio->config.out.bitrate != best_bitrate )
 639             {
 640                 audio->config.out.bitrate = best_bitrate;
 641             }
 642         }
 643
 644         if ( audio->config.out.bitrate != requested_bitrate )
 645         {
 646             /* log the output bitrate */
 647             hb_log( "work: sanitizing track %d audio bitrate %d to %d",
 648                     i, requested_bitrate, audio->config.out.bitrate);
 649         }
 650
 651         if (audio->config.out.codec == HB_ACODEC_VORBIS)
 652             audio->priv.config.vorbis.language = audio->config.lang.simple;
 653
 654         /* set up the audio work structures */
 655         audio->priv.fifo_in   = hb_fifo_init( FIFO_LARGE, FIFO_LARGE_WAKE );
 656         audio->priv.fifo_raw  = hb_fifo_init( FIFO_SMALL, FIFO_SMALL_WAKE );
 657         audio->priv.fifo_sync = hb_fifo_init( FIFO_SMALL, FIFO_SMALL_WAKE );
 658         audio->priv.fifo_out  = hb_fifo_init( FIFO_LARGE, FIFO_LARGE_WAKE );
 659     }
 660
 661     }
 662     /* Synchronization */
 663     sync = hb_sync_init( job );
 664
 665     /* Video decoder */
 666     int vcodec = title->video_codec? title->video_codec : WORK_DECMPEG2;
 667 #if defined(USE_FF_MPEG2)
 668     if (vcodec == WORK_DECMPEG2)
 669     {
 670         vcodec = WORK_DECAVCODECV;
 671         title->video_codec_param = CODEC_ID_MPEG2VIDEO;
 672     }
 673 #endif
 674     hb_list_add( job->list_work, ( w = hb_get_work( vcodec ) ) );
 675     w->codec_param = title->video_codec_param;
 676     w->fifo_in  = job->fifo_mpeg2;
 677     w->fifo_out = job->fifo_raw;
 678
 679     /* Video renderer */
 680     hb_list_add( job->list_work, ( w = hb_get_work( WORK_RENDER ) ) );
 681     w->fifo_in  = job->fifo_sync;
 682     if( !job->indepth_scan )
 683         w->fifo_out = job->fifo_render;
 684     else
 685         w->fifo_out = NULL;
 686
 687     if( !job->indepth_scan )
 688     {
 689
 690         /* Video encoder */
 691         switch( job->vcodec )
 692         {
 693         case HB_VCODEC_FFMPEG:
 694             w = hb_get_work( WORK_ENCAVCODEC );
 695             break;
 696         case HB_VCODEC_X264:
 697             w = hb_get_work( WORK_ENCX264 );
 698             break;
 699         case HB_VCODEC_THEORA:
 700             w = hb_get_work( WORK_ENCTHEORA );
 701             break;
 702         }
 703         w->fifo_in  = job->fifo_render;
 704         w->fifo_out = job->fifo_mpeg4;
 705         w->config   = &job->config;
 706
 707         hb_list_add( job->list_work, w );
 708     }
 709
 710     /*
 711      * Look for the scanned subtitle in the existing subtitle list
 712      * select_subtitle implies that we did a scan.
 713      */
 714     if ( !job->indepth_scan && interjob->select_subtitle &&
 715          ( job->pass == 0 || job->pass == 2 ) )
 716     {
 717         /*
 718          * Disable forced subtitles if we didn't find any in the scan
 719          * so that we display normal subtitles instead.
 720          */
 721         if( interjob->select_subtitle->config.force &&
 722             interjob->select_subtitle->forced_hits == 0 )
 723         {
 724             interjob->select_subtitle->config.force = 0;
 725         }
 726         for( i=0; i < hb_list_count(title->list_subtitle); i++ )
 727         {
 728             subtitle =  hb_list_item( title->list_subtitle, i );
 729
 730             if( subtitle )
 731             {
 732                 /*
 733                 * Remove the scanned subtitle from the subtitle list if
 734                 * it would result in an identical duplicate subtitle track
 735                 * or an emty track (forced and no forced hits).
 736                 */
 737                 if( ( interjob->select_subtitle->id == subtitle->id ) &&
 738                     ( ( interjob->select_subtitle->forced_hits == 0 &&
 739                         subtitle->config.force ) ||
 740                     ( subtitle->config.force == interjob->select_subtitle->config.force ) ) )
 741                 {
 742                     *subtitle = *(interjob->select_subtitle);
 743                     free( interjob->select_subtitle );
 744                     interjob->select_subtitle = NULL;
 745                     break;
 746                 }
 747             }
 748         }
 749
 750         if( interjob->select_subtitle )
 751         {
 752             /*
 753              * Its not in the existing list
 754              *
 755              * Must be second pass of a two pass with subtitle scan enabled, so
 756              * add the subtitle that we found on the first pass for use in this
 757              * pass.
 758              */
 759             hb_list_add( title->list_subtitle, interjob->select_subtitle );
 760             interjob->select_subtitle = NULL;
 761         }
 762     }
 763
 764
 765     for( i=0; i < hb_list_count(title->list_subtitle); i++ )
 766     {
 767         subtitle =  hb_list_item( title->list_subtitle, i );
 768
 769         if( subtitle )
 770         {
 771             subtitle->fifo_in   = hb_fifo_init( FIFO_SMALL, FIFO_SMALL_WAKE );
 772             // Must set capacity of the raw-FIFO to be set >= the maximum number of subtitle
 773             // lines that could be decoded prior to a video frame in order to prevent the following
 774             // deadlock condition:
 775             //   1. Subtitle decoder blocks trying to generate more subtitle lines than will fit in the FIFO.
 776             //   2. Blocks the processing of further subtitle packets read from the input stream.
 777             //   3. And that blocks the processing of any further video packets read from the input stream.
 778             //   4. And that blocks the sync work-object from running, which is needed to consume the subtitle lines in the raw-FIFO.
 779             // Since that number is unbounded, the FIFO must be made (effectively) unbounded in capacity.
 780             subtitle->fifo_raw  = hb_fifo_init( FIFO_UNBOUNDED, FIFO_UNBOUNDED_WAKE );
 781             subtitle->fifo_sync = hb_fifo_init( FIFO_SMALL, FIFO_SMALL_WAKE );
 782             subtitle->fifo_out  = hb_fifo_init( FIFO_SMALL, FIFO_SMALL_WAKE );
 783
 784             if( (!job->indepth_scan || job->select_subtitle_config.force) &&
 785                 subtitle->source == VOBSUB ) {
 786                 /*
 787                  * Don't add threads for subtitles when we are scanning, unless
 788                  * looking for forced subtitles.
 789                  */
 790                 w = hb_get_work( WORK_DECVOBSUB );
 791                 w->fifo_in  = subtitle->fifo_in;
 792                 w->fifo_out = subtitle->fifo_raw;
 793                 w->subtitle = subtitle;
 794                 hb_list_add( job->list_work, w );
 795             }
 796
 797             if( !job->indepth_scan && subtitle->source == CC608SUB )
 798             {
 799                 w = hb_get_work( WORK_DECCC608 );
 800                 w->fifo_in  = subtitle->fifo_in;
 801                 w->fifo_out = subtitle->fifo_raw;
 802                 hb_list_add( job->list_work, w );
 803             }
 804
 805             if( !job->indepth_scan && subtitle->source == SRTSUB )
 806             {
 807                 w = hb_get_work( WORK_DECSRTSUB );
 808                 w->fifo_in  = subtitle->fifo_in;
 809                 w->fifo_out = subtitle->fifo_raw;
 810                 w->subtitle = subtitle;
 811                 hb_list_add( job->list_work, w );
 812             }
 813
 814             if( !job->indepth_scan && subtitle->source == UTF8SUB )
 815             {
 816                 w = hb_get_work( WORK_DECUTF8SUB );
 817                 w->fifo_in  = subtitle->fifo_in;
 818                 w->fifo_out = subtitle->fifo_raw;
 819                 hb_list_add( job->list_work, w );
 820             }
 821
 822             if( !job->indepth_scan && subtitle->source == TX3GSUB )
 823             {
 824                 w = hb_get_work( WORK_DECTX3GSUB );
 825                 w->fifo_in  = subtitle->fifo_in;
 826                 w->fifo_out = subtitle->fifo_raw;
 827                 hb_list_add( job->list_work, w );
 828             }
 829
 830             if( !job->indepth_scan && subtitle->source == SSASUB )
 831             {
 832                 w = hb_get_work( WORK_DECSSASUB );
 833                 w->fifo_in  = subtitle->fifo_in;
 834                 w->fifo_out = subtitle->fifo_raw;
 835                 w->subtitle = subtitle;
 836                 hb_list_add( job->list_work, w );
 837             }
 838
 839             if( !job->indepth_scan &&
 840                 subtitle->format == PICTURESUB
 841                 && subtitle->config.dest == PASSTHRUSUB )
 842             {
 843                 /*
 844                  * Passing through a subtitle picture, this will have to
 845                  * be rle encoded before muxing.
 846                  */
 847                 w = hb_get_work( WORK_ENCVOBSUB );
 848                 w->fifo_in  = subtitle->fifo_sync;
 849                 w->fifo_out = subtitle->fifo_out;
 850                 w->subtitle = subtitle;
 851                 hb_list_add( job->list_work, w );
 852             }
 853         }
 854     }
 855
 856     if( !job->indepth_scan )
 857     {
 858         for( i = 0; i < hb_list_count( title->list_audio ); i++ )
 859         {
 860             audio = hb_list_item( title->list_audio, i );
 861
 862             /*
 863             * Audio Decoder Thread
 864             */
 865             if ( ( w = hb_codec_decoder( audio->config.in.codec ) ) == NULL )
 866             {
 867                 hb_error("Invalid input codec: %d", audio->config.in.codec);
 868                 *job->die = 1;
 869                 goto cleanup;
 870             }
 871             w->fifo_in  = audio->priv.fifo_in;
 872             w->fifo_out = audio->priv.fifo_raw;
 873             w->config   = &audio->priv.config;
 874             w->audio    = audio;
 875             w->codec_param = audio->config.in.codec_param;
 876
 877             hb_list_add( job->list_work, w );
 878
 879             /*
 880             * Audio Encoder Thread
 881             */
 882             if( audio->config.out.codec != HB_ACODEC_AC3_PASS &&
 883                 audio->config.out.codec != HB_ACODEC_DCA_PASS )
 884             {
 885                 /*
 886                 * Add the encoder thread if not doing AC-3 pass through
 887                 */
 888                 if ( ( w = hb_codec_encoder( audio->config.out.codec ) ) == NULL )
 889                 {
 890                     hb_error("Invalid audio codec: %#x", audio->config.out.codec);
 891                     w = NULL;
 892                     *job->die = 1;
 893                     goto cleanup;
 894                 }
 895                 w->fifo_in  = audio->priv.fifo_sync;
 896                 w->fifo_out = audio->priv.fifo_out;
 897                 w->config   = &audio->priv.config;
 898                 w->audio    = audio;
 899
 900                 hb_list_add( job->list_work, w );
 901             }
 902         }
 903     }
 904
 905     if( job->chapter_markers && job->chapter_start == job->chapter_end )
 906     {
 907         job->chapter_markers = 0;
 908         hb_log("work: only 1 chapter, disabling chapter markers");
 909     }
 910
 911     /* Display settings */
 912     hb_display_job_info( job );
 913
 914     /* Init read & write threads */
 915     job->reader = hb_reader_init( job );
 916
 917
 918     job->done = 0;
 919
 920     /* Launch processing threads */
 921     for( i = 0; i < hb_list_count( job->list_work ); i++ )
 922     {
 923         w = hb_list_item( job->list_work, i );
 924         w->done = &job->done;
 925         w->thread_sleep_interval = 10;
 926         if( w->init( w, job ) )
 927         {
 928             hb_error( "Failure to initialise thread '%s'", w->name );
 929             *job->die = 1;
 930             goto cleanup;
 931         }
 932         w->thread = hb_thread_init( w->name, work_loop, w,
 933                                     HB_LOW_PRIORITY );
 934     }
 935
 936     if ( job->indepth_scan )
 937     {
 938         muxer = NULL;
 939         w = sync;
 940         sync->done = &job->done;
 941     }
 942     else
 943     {
 944         sync->done = &job->done;
 945         sync->thread_sleep_interval = 10;
 946         if( sync->init( w, job ) )
 947         {
 948             hb_error( "Failure to initialise thread '%s'", w->name );
 949             *job->die = 1;
 950             goto cleanup;
 951         }
 952         sync->thread = hb_thread_init( sync->name, work_loop, sync,
 953                                     HB_LOW_PRIORITY );
 954
 955         // The muxer requires track information that's set up by the encoder
 956         // init routines so we have to init the muxer last.
 957         muxer = hb_muxer_init( job );
 958         w = muxer;
 959     }
 960
 961     while ( !*job->die && !*w->done && w->status != HB_WORK_DONE )
 962     {
 963         hb_buffer_t      * buf_in, * buf_out;
 964
 965         buf_in = hb_fifo_get_wait( w->fifo_in );
 966         if ( buf_in == NULL )
 967             continue;
 968         if ( *job->die )
 969         {
 970             if( buf_in )
 971             {
 972                 hb_buffer_close( &buf_in );
 973             }
 974             break;
 975         }
 976
 977         buf_out = NULL;
 978         w->status = w->work( w, &buf_in, &buf_out );
 979
 980         if( buf_in )
 981         {
 982             hb_buffer_close( &buf_in );
 983         }
 984         if ( buf_out && w->fifo_out == NULL )
 985         {
 986             hb_buffer_close( &buf_out );
 987         }
 988         if( buf_out )
 989         {
 990             while ( !*job->die )
 991             {
 992                 if ( hb_fifo_full_wait( w->fifo_out ) )
 993                 {
 994                     hb_fifo_push( w->fifo_out, buf_out );
 995                     break;
 996                 }
 997             }
 998         }
 999     }
1000
1001     hb_handle_t * h = job->h;
1002     hb_state_t state;
1003     hb_get_state( h, &state );
1004
1005     hb_log("work: average encoding speed for job is %f fps", state.param.working.rate_avg);
1006
1007     job->done = 1;
1008     if( muxer != NULL )
1009     {
1010         muxer->close( muxer );
1011         free( muxer );
1012
1013         if( sync->thread != NULL )
1014         {
1015             hb_thread_close( &sync->thread );
1016             sync->close( sync );
1017         }
1018         free( sync );
1019     }
1020
1021 cleanup:
1022     /* Stop the write thread (thread_close will block until the muxer finishes) */
1023     job->done = 1;
1024
1025     /* Close work objects */
1026     while( ( w = hb_list_item( job->list_work, 0 ) ) )
1027     {
1028         hb_list_rem( job->list_work, w );
1029         if( w->thread != NULL )
1030         {
1031             hb_thread_close( &w->thread );
1032             w->close( w );
1033         }
1034         free( w );
1035     }
1036
1037     hb_list_close( &job->list_work );
1038
1039     /* Stop the read thread */
1040     if( job->reader != NULL )
1041         hb_thread_close( &job->reader );
1042
1043     /* Close fifos */
1044     hb_fifo_close( &job->fifo_mpeg2 );
1045     hb_fifo_close( &job->fifo_raw );
1046     hb_fifo_close( &job->fifo_sync );
1047     hb_fifo_close( &job->fifo_render );
1048     hb_fifo_close( &job->fifo_mpeg4 );
1049
1050     for (i=0; i < hb_list_count(title->list_subtitle); i++) {
1051         subtitle =  hb_list_item( title->list_subtitle, i);
1052         if( subtitle )
1053         {
1054             hb_fifo_close( &subtitle->fifo_in );
1055             hb_fifo_close( &subtitle->fifo_raw );
1056             hb_fifo_close( &subtitle->fifo_sync );
1057             hb_fifo_close( &subtitle->fifo_out );
1058         }
1059     }
1060     for( i = 0; i < hb_list_count( title->list_audio ); i++ )
1061     {
1062         audio = hb_list_item( title->list_audio, i );
1063         if( audio->priv.fifo_in != NULL )
1064             hb_fifo_close( &audio->priv.fifo_in );
1065         if( audio->priv.fifo_raw != NULL )
1066             hb_fifo_close( &audio->priv.fifo_raw );
1067         if( audio->priv.fifo_sync != NULL )
1068             hb_fifo_close( &audio->priv.fifo_sync );
1069         if( audio->priv.fifo_out != NULL )
1070             hb_fifo_close( &audio->priv.fifo_out );
1071     }
1072
1073     if( job->indepth_scan )
1074     {
1075         /*
1076          * Before closing the title print out our subtitle stats if we need to
1077          * Find the highest and lowest.
1078          */
1079         for( i=0; i < hb_list_count( title->list_subtitle ); i++ )
1080         {
1081             subtitle =  hb_list_item( title->list_subtitle, i );
1082
1083             hb_log( "Subtitle stream 0x%x '%s': %d hits (%d forced)",
1084                     subtitle->id, subtitle->lang, subtitle->hits,
1085                     subtitle->forced_hits );
1086
1087             if( subtitle->hits == 0 )
1088                 continue;
1089
1090             if( subtitle->hits > subtitle_highest )
1091             {
1092                 subtitle_highest = subtitle->hits;
1093                 subtitle_highest_id = subtitle->id;
1094             }
1095
1096             if( subtitle->hits < subtitle_lowest )
1097             {
1098                 subtitle_lowest = subtitle->hits;
1099                 subtitle_lowest_id = subtitle->id;
1100             }
1101
1102             if( subtitle->forced_hits > 0 )
1103             {
1104                 if( subtitle_forced_id == 0 )
1105                 {
1106                     subtitle_forced_id = subtitle->id;
1107                 }
1108             }
1109         }
1110
1111
1112         if( subtitle_forced_id )
1113         {
1114             /*
1115              * If there are any subtitle streams with forced subtitles
1116              * then select it in preference to the lowest.
1117              */
1118             subtitle_hit = subtitle_forced_id;
1119             hb_log("Found a subtitle candidate id 0x%x (contains forced subs)",
1120                    subtitle_hit);
1121         } else if( subtitle_lowest < subtitle_highest )
1122         {
1123             /*
1124              * OK we have more than one, and the lowest is lower,
1125              * but how much lower to qualify for turning it on by
1126              * default?
1127              *
1128              * Let's say 10% as a default.
1129              */
1130             if( subtitle_lowest < ( subtitle_highest * 0.1 ) )
1131             {
1132                 subtitle_hit = subtitle_lowest_id;
1133                 hb_log( "Found a subtitle candidate id 0x%x",
1134                         subtitle_hit );
1135             } else {
1136                 hb_log( "No candidate subtitle detected during subtitle-scan");
1137             }
1138         }
1139     }
1140
1141     if( job->indepth_scan )
1142     {
1143         for( i=0; i < hb_list_count( title->list_subtitle ); i++ )
1144         {
1145             subtitle =  hb_list_item( title->list_subtitle, i );
1146             if( subtitle->id == subtitle_hit )
1147             {
1148                 subtitle->config = job->select_subtitle_config;
1149                 hb_list_rem( title->list_subtitle, subtitle );
1150                 interjob->select_subtitle = subtitle;
1151                 break;
1152             }
1153         }
1154     }
1155
1156     if( job->filters )
1157     {
1158         for( i = 0; i < hb_list_count( job->filters ); i++ )
1159         {
1160             hb_filter_object_t * filter = hb_list_item( job->filters, i );
1161             hb_filter_close( &filter );
1162         }
1163         hb_list_close( &job->filters );
1164     }
1165
1166     hb_buffer_pool_free();
1167
1168     hb_title_close( &job->title );
1169     free( job );
1170 }
1171
1172 /**
1173  * Performs the work object's specific work function.
1174  * Loops calling work function for associated work object. Sleeps when fifo is full.
1175  * Monitors work done indicator.
1176  * Exits loop when work indiactor is set.
1177  * @param _w Handle to work object.
1178  */
1179 static void work_loop( void * _w )
1180 {
1181     hb_work_object_t * w = _w;
1182     hb_buffer_t      * buf_in, * buf_out;
1183
1184     while( !*w->done && w->status != HB_WORK_DONE )
1185     {
1186         buf_in = hb_fifo_get_wait( w->fifo_in );
1187         if ( buf_in == NULL )
1188             continue;
1189         if ( *w->done )
1190         {
1191             if( buf_in )
1192             {
1193                 hb_buffer_close( &buf_in );
1194             }
1195             break;
1196         }
1197
1198         // Invalidate buf_out so that if there is no output
1199         // we don't try to pass along junk.
1200         buf_out = NULL;
1201         w->status = w->work( w, &buf_in, &buf_out );
1202
1203         // Propagate any chapter breaks for the worker if and only if the
1204         // output frame has the same time stamp as the input frame (any
1205         // worker that delays frames has to propagate the chapter marks itself
1206         // and workers that move chapter marks to a different time should set
1207         // 'buf_in' to NULL so that this code won't generate spurious duplicates.)
1208         if( buf_in && buf_out && buf_in->new_chap && buf_in->start == buf_out->start)
1209         {
1210             // restore log below to debug chapter mark propagation problems
1211             //hb_log("work %s: Copying Chapter Break @ %"PRId64, w->name, buf_in->start);
1212             buf_out->new_chap = buf_in->new_chap;
1213         }
1214
1215         if( buf_in )
1216         {
1217             hb_buffer_close( &buf_in );
1218         }
1219         if ( buf_out && w->fifo_out == NULL )
1220         {
1221             hb_buffer_close( &buf_out );
1222         }
1223         if( buf_out )
1224         {
1225             while ( !*w->done )
1226             {
1227                 if ( hb_fifo_full_wait( w->fifo_out ) )
1228                 {
1229                     hb_fifo_push( w->fifo_out, buf_out );
1230                     break;
1231                 }
1232             }
1233         }
1234     }
1235     // Consume data in incoming fifo till job complete so that
1236     // residual data does not stall the pipeline
1237     while( !*w->done )
1238     {
1239         buf_in = hb_fifo_get_wait( w->fifo_in );
1240         if ( buf_in != NULL )
1241             hb_buffer_close( &buf_in );
1242     }
1243 }