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