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