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