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