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 static hb_work_object_t * getWork( 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             return w;
  81         }
  82     }
  83     return NULL;
  84 }
  85
  86 /**
  87  * Job initialization rountine.
  88  * Initializes fifos.
  89  * Creates work objects for synchronizer, video decoder, video renderer, video decoder, audio decoder, audio encoder, reader, muxer.
  90  * Launches thread for each work object with work_loop.
  91  * Loops while monitoring status of work threads and fifos.
  92  * Exits loop when conversion is done and fifos are empty.
  93  * Closes threads and frees fifos.
  94  * @param job Handle work hb_job_t.
  95  * @param cpu_count number of CPUs found in system.
  96  */
  97 static void do_job( hb_job_t * job, int cpu_count )
  98 {
  99     hb_title_t    * title;
 100     int             i, j;
 101     hb_work_object_t * w;
 102
 103     /* FIXME: This feels really hackish, anything better? */
 104     hb_work_object_t * audio_w = NULL;
 105     hb_work_object_t * sub_w = NULL;
 106     hb_work_object_t * final_w = NULL;
 107
 108     hb_audio_t   * audio;
 109     hb_subtitle_t * subtitle;
 110     int done;
 111     unsigned int subtitle_highest = 0;
 112     unsigned int subtitle_highest_id = 0;
 113     unsigned int subtitle_lowest = -1;
 114     unsigned int subtitle_lowest_id = 0;
 115     unsigned int subtitle_forced_id = 0;
 116     unsigned int subtitle_hit = 0;
 117
 118     title = job->title;
 119
 120     job->list_work = hb_list_init();
 121
 122     hb_log( "starting job" );
 123     hb_log( " + device %s", title->dvd );
 124     hb_log( " + title %d, chapter(s) %d to %d", title->index,
 125             job->chapter_start, job->chapter_end );
 126     if ( job->pixel_ratio == 1 )
 127     {
 128         /* Correct the geometry of the output movie when using PixelRatio */
 129         job->height=title->height-job->crop[0]-job->crop[1];
 130         job->width=title->width-job->crop[2]-job->crop[3];
 131     }
 132     else if ( job->pixel_ratio == 2 )
 133     {
 134
 135         /* While keeping the DVD storage aspect, resize the job width and height
 136            so they fit into the user's specified dimensions. */
 137         hb_set_anamorphic_size(job, &job->width, &job->height, &job->pixel_aspect_width, &job->pixel_aspect_height);
 138     }
 139
 140     if( job->pixel_ratio && job->vcodec == HB_VCODEC_FFMPEG)
 141     {
 142         /* Just to make working with ffmpeg even more fun,
 143            lavc's MPEG-4 encoder can't handle PAR values >= 255,
 144            even though AVRational does. Adjusting downwards
 145            distorts the display aspect slightly, but such is life. */
 146         while ((job->pixel_aspect_width & ~0xFF) ||
 147                (job->pixel_aspect_height & ~0xFF))
 148         {
 149             job->pixel_aspect_width >>= 1;
 150             job->pixel_aspect_height >>= 1;
 151         }
 152     }
 153
 154         /* Keep width and height within these boundaries,
 155            but ignore for "loose" anamorphic encodes, for
 156            which this stuff is covered in the pixel_ratio
 157            section right above.*/
 158         if (job->maxHeight && (job->height > job->maxHeight) && (job->pixel_ratio != 2))
 159         {
 160                 job->height = job->maxHeight;
 161                 hb_fix_aspect( job, HB_KEEP_HEIGHT );
 162                 hb_log("Height out of bounds, scaling down to %i", job->maxHeight);
 163                 hb_log("New dimensions %i * %i", job->width, job->height);
 164         }
 165         if (job->maxWidth && (job->width > job->maxWidth) && (job->pixel_ratio != 2))
 166         {
 167                 job->width = job->maxWidth;
 168                 hb_fix_aspect( job, HB_KEEP_WIDTH );
 169                 hb_log("Width out of bounds, scaling down to %i", job->maxWidth);
 170                 hb_log("New dimensions %i * %i", job->width, job->height);
 171         }
 172
 173     hb_log( " + %dx%d -> %dx%d, crop %d/%d/%d/%d",
 174             title->width, title->height, job->width, job->height,
 175             job->crop[0], job->crop[1], job->crop[2], job->crop[3] );
 176
 177     if ( job->grayscale )
 178         hb_log( " + grayscale mode" );
 179
 180     if ( job->vfr )
 181     {
 182         job->vrate_base = title->rate_base;
 183
 184         int detelecine_present = 0;
 185         if ( job->filters )
 186         {
 187             for( i = 0; i < hb_list_count( job->filters ); i++ )
 188             {
 189                 hb_filter_object_t * filter = hb_list_item( job->filters, i );
 190                 if (filter->id == FILTER_DETELECINE)
 191                     detelecine_present = 1;
 192             }
 193         }
 194
 195         if (!detelecine_present)
 196         {
 197             /* Allocate the filter. */
 198             hb_filter_object_t * filter =  malloc( sizeof( hb_filter_object_t ) );
 199
 200             /* Copy in the contents of the detelecine struct. */
 201             memcpy( filter, &hb_filter_detelecine, sizeof( hb_filter_object_t ) );
 202
 203             /* Set the name to a copy of the template name so render.c has something to free. */
 204             filter->name = strdup(hb_filter_detelecine.name);
 205
 206             /* Add it to the list. */
 207             hb_list_add( job->filters, filter );
 208
 209             hb_log("work: VFR mode -- adding detelecine filter");
 210         }
 211     }
 212
 213     if( hb_list_count( job->filters ) )
 214     {
 215         hb_log(" + filters");
 216         for( i = 0; i < hb_list_count( job->filters ); i++ )
 217         {
 218             hb_filter_object_t * filter = hb_list_item( job->filters, i );
 219             if (filter->settings)
 220                 hb_log("   + %s (%s)", filter->name, filter->settings);
 221             else
 222                 hb_log("   + %s (default settings)", filter->name);
 223         }
 224     }
 225
 226     if( job->vfr)
 227     {
 228         hb_log( " + video frame rate: %.3f fps -> variable fps", (float) job->vrate /
 229             (float) job->vrate_base );
 230     }
 231     else
 232     {
 233         hb_log( " + video frame rate: %.3f fps", (float) job->vrate / (float) job->vrate_base);
 234     }
 235
 236     if( job->vquality >= 0.0 && job->vquality <= 1.0 )
 237     {
 238         hb_log( " + video quality %.2f", job->vquality );
 239     }
 240     else
 241     {
 242         hb_log( " + video bitrate %d kbps, pass %d", job->vbitrate, job->pass );
 243     }
 244
 245         hb_log (" + PixelRatio: %d, width:%d, height: %d",job->pixel_ratio,job->width, job->height);
 246     job->fifo_mpeg2  = hb_fifo_init( 2048 );
 247     job->fifo_raw    = hb_fifo_init( FIFO_CPU_MULT * cpu_count );
 248     job->fifo_sync   = hb_fifo_init( FIFO_CPU_MULT * cpu_count );
 249     job->fifo_render = hb_fifo_init( FIFO_CPU_MULT * cpu_count );
 250     job->fifo_mpeg4  = hb_fifo_init( FIFO_CPU_MULT * cpu_count );
 251
 252     /* Synchronization */
 253     hb_list_add( job->list_work, ( w = getWork( WORK_SYNC ) ) );
 254     w->fifo_in  = NULL;
 255     w->fifo_out = NULL;
 256
 257     /* Video decoder */
 258     hb_list_add( job->list_work, ( w = getWork( WORK_DECMPEG2 ) ) );
 259     w->fifo_in  = job->fifo_mpeg2;
 260     w->fifo_out = job->fifo_raw;
 261
 262     /* Video renderer */
 263     hb_list_add( job->list_work, ( w = getWork( WORK_RENDER ) ) );
 264     w->fifo_in  = job->fifo_sync;
 265     w->fifo_out = job->fifo_render;
 266     if ( job->indepth_scan )
 267     {
 268         /*
 269          * if we're doing a subtitle scan the last thread in the
 270          * processing pipeline is render - remember it so we can
 271          * wait for its completion below.
 272          */
 273         final_w = w;
 274     }
 275
 276     /* Video encoder */
 277     switch( job->vcodec )
 278     {
 279         case HB_VCODEC_FFMPEG:
 280             hb_log( " + encoder FFmpeg" );
 281             w = getWork( WORK_ENCAVCODEC );
 282             break;
 283         case HB_VCODEC_XVID:
 284             hb_log( " + encoder XviD" );
 285             w = getWork( WORK_ENCXVID );
 286             break;
 287         case HB_VCODEC_X264:
 288             hb_log( " + encoder x264" );
 289             if( job->x264opts != NULL && *job->x264opts != '\0' )
 290                 hb_log( "   + x264 options: %s", job->x264opts);
 291             w = getWork( WORK_ENCX264 );
 292             break;
 293         case HB_VCODEC_THEORA:
 294             hb_log( " + encoder Theora" );
 295             w = getWork( WORK_ENCTHEORA );
 296             break;
 297     }
 298     w->fifo_in  = job->fifo_render;
 299     w->fifo_out = job->fifo_mpeg4;
 300     w->config   = &job->config;
 301
 302     hb_list_add( job->list_work, w );
 303     if ( !job->indepth_scan )
 304     {
 305         /*
 306          * if we're not doing a subtitle scan the last thread in the
 307          * processing pipeline is the encoder - remember it so we can
 308          * wait for its completion below.
 309          */
 310         final_w = w;
 311     }
 312
 313     if( job->select_subtitle && !job->indepth_scan )
 314     {
 315         /*
 316          * Must be second pass of a two pass with subtitle scan enabled, so
 317          * add the subtitle that we found on the first pass for use in this
 318          * pass.
 319          */
 320         if (*(job->select_subtitle))
 321         {
 322             hb_list_add( title->list_subtitle, *( job->select_subtitle ) );
 323         }
 324     }
 325
 326     for( i=0; i < hb_list_count(title->list_subtitle); i++ )
 327     {
 328         subtitle =  hb_list_item( title->list_subtitle, i );
 329
 330         if( subtitle )
 331         {
 332             hb_log( " + subtitle %x, %s", subtitle->id, subtitle->lang );
 333
 334             subtitle->fifo_in  = hb_fifo_init( FIFO_CPU_MULT * cpu_count );
 335             subtitle->fifo_raw = hb_fifo_init( FIFO_CPU_MULT * cpu_count );
 336
 337             /*
 338              * Disable forced subtitles if we didn't find any in the scan
 339              * so that we display normal subtitles instead.
 340              *
 341              * select_subtitle implies that we did a scan.
 342              */
 343             if( !job->indepth_scan && job->subtitle_force &&
 344                 job->select_subtitle )
 345             {
 346                 if( subtitle->forced_hits == 0 )
 347                 {
 348                     job->subtitle_force = 0;
 349                 }
 350             }
 351
 352             if (!job->indepth_scan || job->subtitle_force) {
 353                 /*
 354                  * Don't add threads for subtitles when we are scanning, unless
 355                  * looking for forced subtitles.
 356                  */
 357                 if( sub_w != NULL )
 358                 {
 359                     /*
 360                      * Need to copy the prior subtitle structure so that we
 361                      * don't overwrite the fifos.
 362                      */
 363                     sub_w = calloc( sizeof( hb_work_object_t ), 1 );
 364                     sub_w = memcpy( sub_w, w, sizeof( hb_work_object_t ));
 365                 } else {
 366                     w = sub_w = getWork( WORK_DECSUB );
 367                 }
 368                 hb_list_add( job->list_work, sub_w );
 369                 sub_w->fifo_in  = subtitle->fifo_in;
 370                 sub_w->fifo_out = subtitle->fifo_raw;
 371             }
 372         }
 373     }
 374
 375     /* if we are doing passthru, and the input codec is not the same as the output
 376      * codec, then remove this audio from the job */
 377     /* otherwise, Bad Things will happen */
 378     for( i = 0; i < hb_list_count( title->list_audio ); )
 379     {
 380         audio = hb_list_item( title->list_audio, i );
 381         if( ( ( audio->config.out.codec == HB_ACODEC_AC3 ) && ( audio->config.in.codec != HB_ACODEC_AC3 ) ) ||
 382             ( ( audio->config.out.codec == HB_ACODEC_DCA ) && ( audio->config.in.codec != HB_ACODEC_DCA ) ) )
 383         {
 384             hb_log( "Passthru requested and input codec is not the same as output codec for track %d",
 385                     audio->config.out.track );
 386             hb_list_rem( title->list_audio, audio );
 387             free( audio );
 388             continue;
 389         }
 390         /* Adjust output track number, in case we removed one.
 391          * Output tracks sadly still need to be in sequential order.
 392          */
 393         audio->config.out.track = i++;
 394     }
 395
 396     for( i = 0; i < hb_list_count( title->list_audio ); i++ )
 397     {
 398         audio = hb_list_item( title->list_audio, i );
 399         hb_log( " + audio track %d", audio->config.out.track );
 400         hb_log( "   + input track %d", audio->config.in.track );
 401         if( (audio->config.out.codec == HB_ACODEC_AC3) || (audio->config.out.codec == HB_ACODEC_DCA) )
 402         {
 403             hb_log( "   + %s passthrough", (audio->config.out.codec == HB_ACODEC_AC3) ?
 404                 "AC3" : "DCA" );
 405         }
 406         else
 407         {
 408             hb_log( "   + audio %d kbps, %d Hz", audio->config.out.bitrate, audio->config.out.samplerate );
 409             hb_log( "   + encoder %s", ( audio->config.out.codec == HB_ACODEC_FAAC ) ?
 410             "faac" : ( ( audio->config.out.codec == HB_ACODEC_LAME ) ? "lame" :
 411             "vorbis" ) );
 412             if ( audio->config.out.dynamic_range_compression > 1 )
 413                 hb_log("   + dynamic range compression: %f", audio->config.out.dynamic_range_compression);
 414         }
 415
 416         hb_log( "     + %x, %s", audio->id, audio->config.lang.description );
 417
 418                 /* sense-check the current mixdown options */
 419
 420                 /* log the requested mixdown */
 421                 for (j = 0; j < hb_audio_mixdowns_count; j++) {
 422                         if (hb_audio_mixdowns[j].amixdown == audio->config.out.mixdown) {
 423                                 hb_log( "       + Requested mixdown: %s (%s)", hb_audio_mixdowns[j].human_readable_name, hb_audio_mixdowns[j].internal_name );
 424                                 break;
 425                         }
 426                 }
 427
 428         /* sense-check the requested mixdown */
 429
 430                 if( audio->config.out.mixdown == 0 && audio->config.out.codec != HB_ACODEC_AC3 )
 431                 {
 432                     /*
 433                      * Mixdown wasn't specified and this is not pass-through, set a default mixdown
 434                      * of stereo.
 435                      */
 436                     audio->config.out.mixdown = HB_AMIXDOWN_STEREO;
 437                 }
 438
 439         /* audioCodecsSupportMono and audioCodecsSupport6Ch are the same for now,
 440            but this may change in the future, so they are separated for flexibility */
 441         int audioCodecsSupportMono = ( (audio->config.in.codec == HB_ACODEC_AC3 || audio->config.in.codec == HB_ACODEC_DCA) &&
 442             (audio->config.out.codec == HB_ACODEC_FAAC || audio->config.out.codec == HB_ACODEC_VORBIS) );
 443         int audioCodecsSupport6Ch =  ( (audio->config.in.codec == HB_ACODEC_AC3 || audio->config.in.codec == HB_ACODEC_DCA) &&
 444             (audio->config.out.codec == HB_ACODEC_FAAC || audio->config.out.codec == HB_ACODEC_VORBIS));
 445
 446         /* find out what the format of our source audio is */
 447         switch (audio->config.in.channel_layout & HB_INPUT_CH_LAYOUT_DISCRETE_NO_LFE_MASK) {
 448
 449             /* mono sources */
 450             case HB_INPUT_CH_LAYOUT_MONO:
 451                 /* regardless of what stereo mixdown we've requested, a mono source always get mixed down
 452                 to mono if we can, and mixed up to stereo if we can't */
 453                 if (audio->config.out.mixdown == HB_AMIXDOWN_MONO && audioCodecsSupportMono == 1) {
 454                     audio->config.out.mixdown = HB_AMIXDOWN_MONO;
 455                 } else {
 456                     audio->config.out.mixdown = HB_AMIXDOWN_STEREO;
 457                 }
 458                 break;
 459
 460             /* stereo input */
 461             case HB_INPUT_CH_LAYOUT_STEREO:
 462                 /* if we've requested a mono mixdown, and it is supported, then do the mix */
 463                 /* use stereo if not supported */
 464                 if (audio->config.out.mixdown == HB_AMIXDOWN_MONO && audioCodecsSupportMono == 0) {
 465                     audio->config.out.mixdown = HB_AMIXDOWN_STEREO;
 466                 /* otherwise, preserve stereo regardless of if we requested something higher */
 467                 } else if (audio->config.out.mixdown > HB_AMIXDOWN_STEREO) {
 468                     audio->config.out.mixdown = HB_AMIXDOWN_STEREO;
 469                 }
 470                 break;
 471
 472             /* dolby (DPL1 aka Dolby Surround = 4.0 matrix-encoded) input */
 473             /* the A52 flags don't allow for a way to distinguish between DPL1 and DPL2 on a DVD,
 474                so we always assume a DPL1 source for A52_DOLBY */
 475             case HB_INPUT_CH_LAYOUT_DOLBY:
 476                 /* if we've requested a mono mixdown, and it is supported, then do the mix */
 477                 /* preserve dolby if not supported */
 478                 if (audio->config.out.mixdown == HB_AMIXDOWN_MONO && audioCodecsSupportMono == 0) {
 479                     audio->config.out.mixdown = HB_AMIXDOWN_DOLBY;
 480                 /* otherwise, preserve dolby even if we requested something higher */
 481                 /* a stereo mixdown will still be honoured here */
 482                 } else if (audio->config.out.mixdown > HB_AMIXDOWN_DOLBY) {
 483                     audio->config.out.mixdown = HB_AMIXDOWN_DOLBY;
 484                 }
 485                 break;
 486
 487             /* 3F/2R input */
 488             case HB_INPUT_CH_LAYOUT_3F2R:
 489                 /* if we've requested a mono mixdown, and it is supported, then do the mix */
 490                 /* use dpl2 if not supported */
 491                 if (audio->config.out.mixdown == HB_AMIXDOWN_MONO && audioCodecsSupportMono == 0) {
 492                     audio->config.out.mixdown = HB_AMIXDOWN_DOLBYPLII;
 493                 } else {
 494                     /* check if we have 3F2R input and also have an LFE - i.e. we have a 5.1 source) */
 495                     if (audio->config.in.channel_layout & HB_INPUT_CH_LAYOUT_HAS_LFE) {
 496                         /* we have a 5.1 source */
 497                         /* if we requested 6ch, but our audio format doesn't support it, then mix to DPLII instead */
 498                         if (audio->config.out.mixdown == HB_AMIXDOWN_6CH && audioCodecsSupport6Ch == 0) {
 499                             audio->config.out.mixdown = HB_AMIXDOWN_DOLBYPLII;
 500                         }
 501                     } else {
 502                         /* we have a 5.0 source, so we can't do 6ch conversion
 503                         default to DPL II instead */
 504                         if (audio->config.out.mixdown > HB_AMIXDOWN_DOLBYPLII) {
 505                             audio->config.out.mixdown = HB_AMIXDOWN_DOLBYPLII;
 506                         }
 507                     }
 508                 }
 509                 /* all other mixdowns will have been preserved here */
 510                 break;
 511
 512             /* 3F/1R input */
 513             case HB_INPUT_CH_LAYOUT_3F1R:
 514                 /* if we've requested a mono mixdown, and it is supported, then do the mix */
 515                 /* use dpl1 if not supported */
 516                 if (audio->config.out.mixdown == HB_AMIXDOWN_MONO && audioCodecsSupportMono == 0) {
 517                     audio->config.out.mixdown = HB_AMIXDOWN_DOLBY;
 518                 } else {
 519                     /* we have a 4.0 or 4.1 source, so we can't do DPLII or 6ch conversion
 520                     default to DPL I instead */
 521                     if (audio->config.out.mixdown > HB_AMIXDOWN_DOLBY) {
 522                         audio->config.out.mixdown = HB_AMIXDOWN_DOLBY;
 523                     }
 524                 }
 525                 /* all other mixdowns will have been preserved here */
 526                 break;
 527
 528             default:
 529                 /* if we've requested a mono mixdown, and it is supported, then do the mix */
 530                 if (audio->config.out.mixdown == HB_AMIXDOWN_MONO && audioCodecsSupportMono == 1) {
 531                     audio->config.out.mixdown = HB_AMIXDOWN_MONO;
 532                 /* mix everything else down to stereo */
 533                 } else {
 534                     audio->config.out.mixdown = HB_AMIXDOWN_STEREO;
 535                 }
 536
 537         }
 538
 539                 /* log the output mixdown */
 540                 for (j = 0; j < hb_audio_mixdowns_count; j++) {
 541                         if (hb_audio_mixdowns[j].amixdown == audio->config.out.mixdown) {
 542                                 hb_log( "       + Actual mixdown: %s (%s)", hb_audio_mixdowns[j].human_readable_name, hb_audio_mixdowns[j].internal_name );
 543                                 break;
 544                         }
 545                 }
 546
 547         if (audio->config.out.codec == HB_ACODEC_VORBIS)
 548             audio->priv.config.vorbis.language = audio->config.lang.simple;
 549
 550         /* set up the audio work structures */
 551         audio->priv.fifo_in   = hb_fifo_init( 256 );
 552         audio->priv.fifo_raw  = hb_fifo_init( FIFO_CPU_MULT * cpu_count );
 553         audio->priv.fifo_sync = hb_fifo_init( 256 );
 554         audio->priv.fifo_out  = hb_fifo_init( FIFO_CPU_MULT * cpu_count );
 555
 556
 557         /*
 558          * Audio Decoder Thread
 559          */
 560         switch( audio->config.in.codec )
 561         {
 562             case HB_ACODEC_AC3:
 563                 w = getWork( WORK_DECA52 );
 564                 break;
 565             case HB_ACODEC_DCA:
 566                 w = getWork( WORK_DECDCA );
 567                 break;
 568             case HB_ACODEC_MPGA:
 569                 w = getWork( WORK_DECAVCODEC );
 570                 break;
 571             case HB_ACODEC_LPCM:
 572                 w = getWork( WORK_DECLPCM );
 573                 break;
 574             default:
 575                 /* Invalid input codec */
 576                 hb_error("Invalid input codec: %d", audio->config.in.codec);
 577                 *job->die = 1;
 578                 goto cleanup;
 579         }
 580         w->fifo_in       = audio->priv.fifo_in;
 581         w->fifo_out      = audio->priv.fifo_raw;
 582         w->config        = &audio->priv.config;
 583         w->audio         = audio;
 584
 585         /* FIXME: This feels really hackish, anything better? */
 586         audio_w = calloc( sizeof( hb_work_object_t ), 1 );
 587         audio_w = memcpy( audio_w, w, sizeof( hb_work_object_t ));
 588
 589         hb_list_add( job->list_work, audio_w );
 590
 591         /*
 592          * Audio Encoder Thread
 593          */
 594         switch( audio->config.out.codec )
 595         {
 596             case HB_ACODEC_FAAC:
 597                 w = getWork( WORK_ENCFAAC );
 598                 break;
 599             case HB_ACODEC_LAME:
 600                 w = getWork( WORK_ENCLAME );
 601                 break;
 602             case HB_ACODEC_VORBIS:
 603                 w = getWork( WORK_ENCVORBIS );
 604                 break;
 605             case HB_ACODEC_AC3:
 606                 break;
 607             case HB_ACODEC_DCA:    /* These are all invalid output codecs. */
 608             default:
 609                 hb_error("Invalid audio codec: %#x", audio->config.out.codec);
 610                 w = NULL;
 611                 *job->die = 1;
 612                 goto cleanup;
 613         }
 614
 615         if( audio->config.out.codec != HB_ACODEC_AC3 )
 616         {
 617             /*
 618              * Add the encoder thread if not doing AC-3 pass through
 619              */
 620             w->fifo_in       = audio->priv.fifo_sync;
 621             w->fifo_out      = audio->priv.fifo_out;
 622             w->config        = &audio->priv.config;
 623             w->audio         = audio;
 624
 625             /* FIXME: This feels really hackish, anything better? */
 626             audio_w = calloc( sizeof( hb_work_object_t ), 1 );
 627             audio_w = memcpy( audio_w, w, sizeof( hb_work_object_t ));
 628
 629             hb_list_add( job->list_work, audio_w );
 630         }
 631
 632
 633     }
 634
 635     /* Init read & write threads */
 636     job->reader = hb_reader_init( job );
 637
 638     hb_log( " + output: %s", job->file );
 639     job->muxer = hb_muxer_init( job );
 640
 641     job->done = 0;
 642
 643     /* Launch processing threads */
 644     for( i = 1; i < hb_list_count( job->list_work ); i++ )
 645     {
 646         w = hb_list_item( job->list_work, i );
 647         w->done = &job->done;
 648         w->thread_sleep_interval = 10;
 649         w->init( w, job );
 650         w->thread = hb_thread_init( w->name, work_loop, w,
 651                                     HB_LOW_PRIORITY );
 652     }
 653
 654     done = 0;
 655     w = hb_list_item( job->list_work, 0 );
 656     w->thread_sleep_interval = 50;
 657     w->init( w, job );
 658     while( !*job->die )
 659     {
 660         if( ( w->status = w->work( w, NULL, NULL ) ) == HB_WORK_DONE )
 661         {
 662             done = 1;
 663         }
 664         if( done &&
 665             final_w->status == HB_WORK_DONE &&
 666             !hb_fifo_size( job->fifo_mpeg4 ) )
 667         {
 668             break;
 669         }
 670         hb_snooze( w->thread_sleep_interval );
 671     }
 672     hb_list_rem( job->list_work, w );
 673     w->close( w );
 674     job->done = 1;
 675
 676 cleanup:
 677     /* Close work objects */
 678     while( ( w = hb_list_item( job->list_work, 0 ) ) )
 679     {
 680         hb_list_rem( job->list_work, w );
 681         if( w != NULL && w->thread != NULL )
 682         {
 683             hb_thread_close( &w->thread );
 684             w->close( w );
 685         }
 686
 687         /* FIXME: This feels really hackish, anything better? */
 688         if ( w->id == WORK_DECA52 ||
 689              w->id == WORK_DECDCA ||
 690              w->id == WORK_DECLPCM ||
 691              w->id == WORK_ENCFAAC ||
 692              w->id == WORK_ENCLAME ||
 693              w->id == WORK_ENCVORBIS )
 694         {
 695             free( w );
 696             w = NULL;
 697         }
 698     }
 699
 700     hb_list_close( &job->list_work );
 701
 702     /* Stop read & write threads */
 703     if( job->reader != NULL )
 704         hb_thread_close( &job->reader );
 705     if( job->muxer != NULL )
 706         hb_thread_close( &job->muxer );
 707
 708     /* Close fifos */
 709     hb_fifo_close( &job->fifo_mpeg2 );
 710     hb_fifo_close( &job->fifo_raw );
 711     hb_fifo_close( &job->fifo_sync );
 712     hb_fifo_close( &job->fifo_render );
 713     hb_fifo_close( &job->fifo_mpeg4 );
 714
 715     for (i=0; i < hb_list_count(title->list_subtitle); i++) {
 716         subtitle =  hb_list_item( title->list_subtitle, i);
 717         if( subtitle )
 718         {
 719             hb_fifo_close( &subtitle->fifo_in );
 720             hb_fifo_close( &subtitle->fifo_raw );
 721         }
 722     }
 723     for( i = 0; i < hb_list_count( title->list_audio ); i++ )
 724     {
 725         audio = hb_list_item( title->list_audio, i );
 726         if( audio->priv.fifo_in != NULL )
 727             hb_fifo_close( &audio->priv.fifo_in );
 728         if( audio->priv.fifo_raw != NULL )
 729             hb_fifo_close( &audio->priv.fifo_raw );
 730         if( audio->priv.fifo_sync != NULL )
 731             hb_fifo_close( &audio->priv.fifo_sync );
 732         if( audio->priv.fifo_out != NULL )
 733             hb_fifo_close( &audio->priv.fifo_out );
 734     }
 735
 736     if( job->indepth_scan )
 737     {
 738         /*
 739          * Before closing the title print out our subtitle stats if we need to
 740          * Find the highest and lowest.
 741          */
 742         for( i=0; i < hb_list_count( title->list_subtitle ); i++ )
 743         {
 744             subtitle =  hb_list_item( title->list_subtitle, i );
 745             hb_log( "Subtitle stream 0x%x '%s': %d hits (%d forced)",
 746                     subtitle->id, subtitle->lang, subtitle->hits,
 747                     subtitle->forced_hits );
 748             if( subtitle->hits > subtitle_highest )
 749             {
 750                 subtitle_highest = subtitle->hits;
 751                 subtitle_highest_id = subtitle->id;
 752             }
 753
 754             if( subtitle->hits < subtitle_lowest )
 755             {
 756                 subtitle_lowest = subtitle->hits;
 757                 subtitle_lowest_id = subtitle->id;
 758             }
 759
 760             if( subtitle->forced_hits > 0 )
 761             {
 762                 if( subtitle_forced_id == 0 )
 763                 {
 764                     subtitle_forced_id = subtitle->id;
 765                 }
 766             }
 767         }
 768
 769         if( job->native_language ) {
 770             /*
 771              * We still have a native_language, so the audio and subtitles are
 772              * different, so in this case it is a foreign film and we want to
 773              * select the subtitle with the highest hits in our language.
 774              */
 775             subtitle_hit = subtitle_highest_id;
 776             hb_log( "Found a native-language subtitle id 0x%x", subtitle_hit);
 777         } else {
 778             if( subtitle_forced_id )
 779             {
 780                 /*
 781                  * If there are any subtitle streams with forced subtitles
 782                  * then select it in preference to the lowest.
 783                  */
 784                 subtitle_hit = subtitle_forced_id;
 785                 hb_log("Found a subtitle candidate id 0x%x (contains forced subs)",
 786                        subtitle_hit);
 787             } else if( subtitle_lowest < subtitle_highest )
 788             {
 789                 /*
 790                  * OK we have more than one, and the lowest is lower,
 791                  * but how much lower to qualify for turning it on by
 792                  * default?
 793                  *
 794                  * Let's say 10% as a default.
 795                  */
 796                 if( subtitle_lowest < ( subtitle_highest * 0.1 ) )
 797                 {
 798                     subtitle_hit = subtitle_lowest_id;
 799                     hb_log( "Found a subtitle candidate id 0x%x",
 800                             subtitle_hit );
 801                 } else {
 802                     hb_log( "No candidate subtitle detected during subtitle-scan");
 803                 }
 804             }
 805         }
 806     }
 807
 808     if( job->select_subtitle )
 809     {
 810         if( job->indepth_scan )
 811         {
 812             for( i=0; i < hb_list_count( title->list_subtitle ); i++ )
 813             {
 814                 subtitle =  hb_list_item( title->list_subtitle, i );
 815                 if( subtitle->id == subtitle_hit )
 816                 {
 817                     hb_list_rem( title->list_subtitle, subtitle );
 818                     *( job->select_subtitle ) = subtitle;
 819                 }
 820             }
 821         } else {
 822             /*
 823              * Must be the end of pass 0 or 2 - we don't need this anymore.
 824              *
 825              * Have to put the subtitle list back together in the title though
 826              * or the GUI will have a hissy fit.
 827              */
 828             free( job->select_subtitle );
 829             job->select_subtitle = NULL;
 830         }
 831     }
 832
 833     if( job->filters )
 834     {
 835         for( i = 0; i < hb_list_count( job->filters ); i++ )
 836         {
 837             hb_filter_object_t * filter = hb_list_item( job->filters, i );
 838             hb_filter_close( &filter );
 839         }
 840         hb_list_close( &job->filters );
 841     }
 842
 843     hb_buffer_pool_free();
 844
 845     hb_title_close( &job->title );
 846     free( job );
 847 }
 848
 849 /**
 850  * Performs the work objects specific work function.
 851  * Loops calling work function for associated work object. Sleeps when fifo is full.
 852  * Monitors work done indicator.
 853  * Exits loop when work indiactor is set.
 854  * @param _w Handle to work object.
 855  */
 856 static void work_loop( void * _w )
 857 {
 858     hb_work_object_t * w = _w;
 859     hb_buffer_t      * buf_in, * buf_out;
 860
 861     while( !*w->done )
 862     {
 863 #if 0
 864         hb_lock( job->pause );
 865         hb_unlock( job->pause );
 866 #endif
 867         if( hb_fifo_is_full( w->fifo_out ) ||
 868 //        if( (hb_fifo_percent_full( w->fifo_out ) > 0.8) ||
 869             !( buf_in = hb_fifo_get( w->fifo_in ) ) )
 870         {
 871             hb_snooze( w->thread_sleep_interval );
 872 //                      w->thread_sleep_interval += 1;
 873             continue;
 874         }
 875 //              w->thread_sleep_interval = MAX(1, (w->thread_sleep_interval - 1));
 876
 877         w->status = w->work( w, &buf_in, &buf_out );
 878
 879         // Propagate any chapter breaks for the worker if and only if the
 880         // output frame has the same time stamp as the input frame (any
 881         // worker that delays frames has to propagate the chapter marks itself
 882         // and workers that move chapter marks to a different time should set
 883         // 'buf_in' to NULL so that this code won't generate spurious duplicates.)
 884         if( buf_in && buf_out && buf_in->new_chap && buf_in->start == buf_out->start)
 885         {
 886             // restore log below to debug chapter mark propagation problems
 887             //hb_log("work %s: Copying Chapter Break @ %lld", w->name, buf_in->start);
 888             buf_out->new_chap = buf_in->new_chap;
 889         }
 890
 891         if( buf_in )
 892         {
 893             hb_buffer_close( &buf_in );
 894         }
 895         if( buf_out )
 896         {
 897             hb_fifo_push( w->fifo_out, buf_out );
 898         }
 899     }
 900 }