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