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