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