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