1 /* $Id: work.c,v 1.43 2005/03/17 16:38:49 titer Exp $
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. */
8 #include "a52dec/a52.h"
10 #include "libavformat/avformat.h"
15 hb_job_t ** current_job;
22 static void work_func();
23 static void do_job( hb_job_t *, int cpu_count );
24 static void work_loop( void * );
27 #define FIFO_LARGE_WAKE 16
29 #define FIFO_SMALL_WAKE 15
32 * Allocates work object and launches work thread with work_func.
33 * @param jobs Handle to hb_list_t.
34 * @param cpu_count Humber of CPUs found in system.
35 * @param die Handle to user inititated exit indicator.
36 * @param error Handle to error indicator.
38 hb_thread_t * hb_work_init( hb_list_t * jobs, int cpu_count,
39 volatile int * die, int * error, hb_job_t ** job )
41 hb_work_t * work = calloc( sizeof( hb_work_t ), 1 );
44 work->current_job = job;
45 work->cpu_count = cpu_count;
49 return hb_thread_init( "work", work_func, work, HB_LOW_PRIORITY );
52 static void InitWorkState( hb_handle_t * h )
56 state.state = HB_STATE_WORKING;
57 #define p state.param.working
66 hb_set_state( h, &state );
71 * Iterates through job list and calls do_job for each job.
72 * @param _work Handle work object.
74 static void work_func( void * _work )
76 hb_work_t * work = _work;
79 hb_log( "%d job(s) to process", hb_list_count( work->jobs ) );
81 while( !*work->die && ( job = hb_list_item( work->jobs, 0 ) ) )
83 hb_list_rem( work->jobs, job );
85 *(work->current_job) = job;
86 InitWorkState( job->h );
87 do_job( job, work->cpu_count );
88 *(work->current_job) = NULL;
91 *(work->error) = HB_ERROR_NONE;
96 hb_work_object_t * hb_get_work( int id )
99 for( w = hb_objects; w; w = w->next )
103 hb_work_object_t *wc = malloc( sizeof(*w) );
111 hb_work_object_t * hb_codec_decoder( int codec )
115 case HB_ACODEC_AC3: return hb_get_work( WORK_DECA52 );
116 case HB_ACODEC_DCA: return hb_get_work( WORK_DECDCA );
117 case HB_ACODEC_MPGA: return hb_get_work( WORK_DECAVCODEC );
118 case HB_ACODEC_LPCM: return hb_get_work( WORK_DECLPCM );
119 case HB_ACODEC_FFMPEG: return hb_get_work( WORK_DECAVCODECAI );
124 hb_work_object_t * hb_codec_encoder( int codec )
128 case HB_ACODEC_FAAC: return hb_get_work( WORK_ENCFAAC );
129 case HB_ACODEC_LAME: return hb_get_work( WORK_ENCLAME );
130 case HB_ACODEC_VORBIS: return hb_get_work( WORK_ENCVORBIS );
131 case HB_ACODEC_CA_AAC: return hb_get_work( WORK_ENC_CA_AAC );
137 * Displays job parameters in the debug log.
138 * @param job Handle work hb_job_t.
140 void hb_display_job_info( hb_job_t * job )
142 hb_title_t * title = job->title;
144 hb_subtitle_t * subtitle;
147 hb_log("job configuration:");
148 hb_log( " * source");
150 hb_log( " + %s", title->path );
152 hb_log( " + title %d, chapter(s) %d to %d", title->index,
153 job->chapter_start, job->chapter_end );
155 if( title->container_name != NULL )
156 hb_log( " + container: %s", title->container_name);
158 if( title->data_rate )
160 hb_log( " + data rate: %d kbps", title->data_rate / 1000 );
163 hb_log( " * destination");
165 hb_log( " + %s", job->file );
170 hb_log(" + container: MPEG-4 (.mp4 and .m4v)");
173 hb_log( " + compatibility atom for iPod 5G");
175 if( job->largeFileSize )
176 hb_log( " + 64-bit formatting");
178 if( job->mp4_optimize )
179 hb_log( " + optimized for progressive web downloads");
181 if( job->color_matrix )
182 hb_log( " + custom color matrix: %s", job->color_matrix == 1 ? "ITU Bt.601 (SD)" : "ITU Bt.709 (HD)");
186 hb_log(" + container: AVI");
190 hb_log(" + container: Matroska (.mkv)");
194 hb_log(" + conttainer: Ogg Media (.ogm)");
198 if( job->chapter_markers )
200 hb_log( " + chapter markers" );
203 hb_log(" * video track");
205 hb_log(" + decoder: %s", title->video_codec_name );
207 if( title->video_bitrate )
209 hb_log( " + bitrate %d kbps", title->video_bitrate / 1000 );
214 hb_log( " + frame rate: same as source (around %.3f fps)",
215 (float) title->rate / (float) title->rate_base );
219 static const char *frtypes[] = {
220 "", "constant", "peak rate limited to"
222 hb_log( " + frame rate: %.3f fps -> %s %.3f fps",
223 (float) title->rate / (float) title->rate_base, frtypes[job->cfr],
224 (float) job->vrate / (float) job->vrate_base );
227 if( job->anamorphic.mode )
229 hb_log( " + %s anamorphic", job->anamorphic.mode == 1 ? "strict" : job->anamorphic.mode == 2? "loose" : "custom" );
230 if( job->anamorphic.mode == 3 && job->anamorphic.keep_display_aspect )
232 hb_log( " + keeping source display aspect ratio");
234 hb_log( " + storage dimensions: %d * %d -> %d * %d, crop %d/%d/%d/%d, mod %i",
235 title->width, title->height, job->width, job->height,
236 job->crop[0], job->crop[1], job->crop[2], job->crop[3], job->modulus );
237 if( job->anamorphic.itu_par )
239 hb_log( " + using ITU pixel aspect ratio values");
241 hb_log( " + pixel aspect ratio: %i / %i", job->anamorphic.par_width, job->anamorphic.par_height );
242 hb_log( " + display dimensions: %.0f * %i",
243 (float)( job->width * job->anamorphic.par_width / job->anamorphic.par_height ), job->height );
247 hb_log( " + dimensions: %d * %d -> %d * %d, crop %d/%d/%d/%d",
248 title->width, title->height, job->width, job->height,
249 job->crop[0], job->crop[1], job->crop[2], job->crop[3] );
252 if ( job->grayscale )
253 hb_log( " + grayscale mode" );
255 if( hb_list_count( job->filters ) )
257 hb_log(" + %s", hb_list_count( job->filters) > 1 ? "filters" : "filter" );
258 for( i = 0; i < hb_list_count( job->filters ); i++ )
260 hb_filter_object_t * filter = hb_list_item( job->filters, i );
261 if (filter->settings)
262 hb_log(" + %s (%s)", filter->name, filter->settings);
264 hb_log(" + %s (default settings)", filter->name);
268 if( !job->indepth_scan )
271 switch( job->vcodec )
273 case HB_VCODEC_FFMPEG:
274 hb_log( " + encoder: FFmpeg" );
278 hb_log( " + encoder: x264" );
279 if( job->x264opts != NULL && *job->x264opts != '\0' )
280 hb_log( " + options: %s", job->x264opts);
283 case HB_VCODEC_THEORA:
284 hb_log( " + encoder: Theora" );
288 if( job->vquality >= 0.0 && job->vquality <= 1.0 )
290 hb_log( " + quality: %.2f", job->vquality );
292 else if( job->vquality > 1 )
294 hb_log( " + quality: %.2f %s", job->vquality, job->vcodec == HB_VCODEC_X264 ? "(RF)" : "(QP)" );
298 hb_log( " + bitrate: %d kbps, pass: %d", job->vbitrate, job->pass );
302 for( i=0; i < hb_list_count( title->list_subtitle ); i++ )
304 subtitle = hb_list_item( title->list_subtitle, i );
308 if( subtitle->source == SRTSUB )
310 /* For SRT, print offset and charset too */
311 hb_log( " * subtitle track %i, %s (id %x) %s [%s] -> %s%s, offset: %"PRId64", charset: %s",
312 subtitle->track, subtitle->lang, subtitle->id, "Text", "SRT", "Pass-Through",
313 subtitle->config.default_track ? ", Default" : "",
314 subtitle->config.offset, subtitle->config.src_codeset );
318 hb_log( " * subtitle track %i, %s (id %x) %s [%s] -> %s%s%s", subtitle->track, subtitle->lang, subtitle->id,
319 subtitle->format == PICTURESUB ? "Picture" : "Text",
320 subtitle->source == VOBSUB ? "VOBSUB" :
321 subtitle->source == CC608SUB || subtitle->source == CC708SUB ? "CC" :
322 subtitle->source == UTF8SUB ? "UTF-8" :
323 subtitle->source == TX3GSUB ? "TX3G" :
324 subtitle->source == SSASUB ? "SSA" : "Unknown",
325 job->indepth_scan ? "Foreign Audio Search" :
326 subtitle->config.dest == RENDERSUB ? "Render/Burn in" : "Pass-Through",
327 subtitle->config.force ? ", Forced Only" : "",
328 subtitle->config.default_track ? ", Default" : "" );
333 if( !job->indepth_scan )
335 for( i = 0; i < hb_list_count( title->list_audio ); i++ )
337 audio = hb_list_item( title->list_audio, i );
339 hb_log( " * audio track %d", audio->config.out.track );
341 if( audio->config.out.name )
342 hb_log( " + name: %s", audio->config.out.name );
344 hb_log( " + decoder: %s (track %d, id %x)", audio->config.lang.description, audio->config.in.track + 1, audio->id );
346 if( ( audio->config.in.codec == HB_ACODEC_AC3 ) || ( audio->config.in.codec == HB_ACODEC_DCA) )
348 hb_log( " + bitrate: %d kbps, samplerate: %d Hz", audio->config.in.bitrate / 1000, audio->config.in.samplerate );
351 if( (audio->config.out.codec != HB_ACODEC_AC3) && (audio->config.out.codec != HB_ACODEC_DCA) )
353 for (j = 0; j < hb_audio_mixdowns_count; j++)
355 if (hb_audio_mixdowns[j].amixdown == audio->config.out.mixdown) {
356 hb_log( " + mixdown: %s", hb_audio_mixdowns[j].human_readable_name );
362 if ( audio->config.out.dynamic_range_compression && (audio->config.out.codec != HB_ACODEC_AC3) && (audio->config.out.codec != HB_ACODEC_DCA))
364 hb_log(" + dynamic range compression: %f", audio->config.out.dynamic_range_compression);
367 if( (audio->config.out.codec == HB_ACODEC_AC3) || (audio->config.out.codec == HB_ACODEC_DCA) )
369 hb_log( " + %s passthrough", (audio->config.out.codec == HB_ACODEC_AC3) ?
374 hb_log( " + encoder: %s", ( audio->config.out.codec == HB_ACODEC_FAAC ) ?
375 "faac" : ( ( audio->config.out.codec == HB_ACODEC_LAME ) ?
376 "lame" : ( ( audio->config.out.codec == HB_ACODEC_CA_AAC ) ?
377 "ca_aac" : "vorbis" ) ) );
378 hb_log( " + bitrate: %d kbps, samplerate: %d Hz", audio->config.out.bitrate, audio->config.out.samplerate );
384 /* Corrects framerates when actual duration and frame count numbers are known. */
385 void correct_framerate( hb_job_t * job )
389 hb_interjob_t * interjob = hb_interjob_get( job->h );
391 if( ( job->sequence_id & 0xFFFFFF ) != ( interjob->last_job & 0xFFFFFF) )
392 return; // Interjob information is for a different encode.
394 /* Cache the original framerate before altering it. */
395 interjob->vrate = job->vrate;
396 interjob->vrate_base = job->vrate_base;
398 real_frames = interjob->frame_count - interjob->render_dropped;
400 job->vrate = job->vrate_base * ( (double)real_frames * 90000 / interjob->total_time );
404 * Job initialization rountine.
406 * Creates work objects for synchronizer, video decoder, video renderer, video decoder, audio decoder, audio encoder, reader, muxer.
407 * Launches thread for each work object with work_loop.
408 * Loops while monitoring status of work threads and fifos.
409 * Exits loop when conversion is done and fifos are empty.
410 * Closes threads and frees fifos.
411 * @param job Handle work hb_job_t.
412 * @param cpu_count number of CPUs found in system.
414 static void do_job( hb_job_t * job, int cpu_count )
418 hb_work_object_t * w;
419 hb_work_object_t * sync;
420 hb_work_object_t * muxer;
421 hb_interjob_t * interjob;
424 hb_subtitle_t * subtitle;
425 unsigned int subtitle_highest = 0;
426 unsigned int subtitle_highest_id = 0;
427 unsigned int subtitle_lowest = -1;
428 unsigned int subtitle_lowest_id = 0;
429 unsigned int subtitle_forced_id = 0;
430 unsigned int subtitle_hit = 0;
433 interjob = hb_interjob_get( job->h );
435 if( job->pass == 2 && !job->cfr )
437 correct_framerate( job );
440 job->list_work = hb_list_init();
442 hb_log( "starting job" );
444 if( job->anamorphic.mode )
446 hb_set_anamorphic_size(job, &job->width, &job->height, &job->anamorphic.par_width, &job->anamorphic.par_height);
448 if( job->vcodec == HB_VCODEC_FFMPEG )
450 /* Just to make working with ffmpeg even more fun,
451 lavc's MPEG-4 encoder can't handle PAR values >= 255,
452 even though AVRational does. Adjusting downwards
453 distorts the display aspect slightly, but such is life. */
454 while ((job->anamorphic.par_width & ~0xFF) ||
455 (job->anamorphic.par_height & ~0xFF))
457 job->anamorphic.par_width >>= 1;
458 job->anamorphic.par_height >>= 1;
463 /* Keep width and height within these boundaries,
464 but ignore for anamorphic. For "loose" anamorphic encodes,
465 this stuff is covered in the pixel_ratio section above. */
466 if ( job->maxHeight && ( job->height > job->maxHeight ) && ( !job->anamorphic.mode ) )
468 job->height = job->maxHeight;
469 hb_fix_aspect( job, HB_KEEP_HEIGHT );
470 hb_log( "Height out of bounds, scaling down to %i", job->maxHeight );
471 hb_log( "New dimensions %i * %i", job->width, job->height );
473 if ( job->maxWidth && ( job->width > job->maxWidth ) && ( !job->anamorphic.mode ) )
475 job->width = job->maxWidth;
476 hb_fix_aspect( job, HB_KEEP_WIDTH );
477 hb_log( "Width out of bounds, scaling down to %i", job->maxWidth );
478 hb_log( "New dimensions %i * %i", job->width, job->height );
481 if( job->mux & HB_MUX_AVI )
483 // The concept of variable frame rate video was a bit too advanced
484 // for Microsoft so AVI doesn't support it. Since almost all dvd
485 // video is VFR we have to convert it to constant frame rate to
486 // put it in an AVI container. So duplicate, drop and
487 // otherwise trash video frames to appease the gods of Redmond.
493 /* Ensure we're using "Same as source" FPS */
494 job->vrate_base = title->rate_base;
497 job->fifo_mpeg2 = hb_fifo_init( FIFO_LARGE, FIFO_LARGE_WAKE );
498 job->fifo_raw = hb_fifo_init( FIFO_SMALL, FIFO_SMALL_WAKE );
499 job->fifo_sync = hb_fifo_init( FIFO_SMALL, FIFO_SMALL_WAKE );
500 job->fifo_render = hb_fifo_init( FIFO_SMALL, FIFO_SMALL_WAKE );
501 job->fifo_mpeg4 = hb_fifo_init( FIFO_LARGE, FIFO_LARGE_WAKE );
504 * Audio fifos must be initialized before sync
506 if( !job->indepth_scan )
508 // if we are doing passthru, and the input codec is not the same as the output
509 // codec, then remove this audio from the job. If we're not doing passthru and
510 // the input codec is the 'internal' ffmpeg codec, make sure that only one
511 // audio references that audio stream since the codec context is specific to
512 // the audio id & multiple copies of the same stream will garble the audio
514 uint8_t aud_id_uses[MAX_STREAMS];
515 memset( aud_id_uses, 0, sizeof(aud_id_uses) );
516 for( i = 0; i < hb_list_count( title->list_audio ); )
518 audio = hb_list_item( title->list_audio, i );
519 if( ( ( audio->config.out.codec == HB_ACODEC_AC3 ) && ( audio->config.in.codec != HB_ACODEC_AC3 ) ) ||
520 ( ( audio->config.out.codec == HB_ACODEC_DCA ) && ( audio->config.in.codec != HB_ACODEC_DCA ) ) )
522 hb_log( "Passthru requested and input codec is not the same as output codec for track %d",
523 audio->config.out.track );
524 hb_list_rem( title->list_audio, audio );
528 if( audio->config.out.codec != HB_ACODEC_AC3 &&
529 audio->config.out.codec != HB_ACODEC_DCA &&
530 audio->config.out.samplerate > 48000 )
532 hb_log( "Sample rate %d not supported. Down-sampling to 48kHz.",
533 audio->config.out.samplerate );
534 audio->config.out.samplerate = 48000;
536 if ( audio->config.in.codec == HB_ACODEC_FFMPEG )
538 if ( aud_id_uses[audio->id] )
540 hb_log( "Multiple decodes of audio id %d, removing track %d",
541 audio->id, audio->config.out.track );
542 hb_list_rem( title->list_audio, audio );
546 ++aud_id_uses[audio->id];
548 /* Adjust output track number, in case we removed one.
549 * Output tracks sadly still need to be in sequential order.
551 audio->config.out.track = i++;
554 int requested_mixdown = 0;
555 int requested_mixdown_index = 0;
557 for( i = 0; i < hb_list_count( title->list_audio ); i++ )
559 audio = hb_list_item( title->list_audio, i );
561 if( audio->config.out.codec != audio->config.in.codec )
563 /* sense-check the current mixdown options */
565 /* log the requested mixdown */
566 for (j = 0; j < hb_audio_mixdowns_count; j++) {
567 if (hb_audio_mixdowns[j].amixdown == audio->config.out.mixdown) {
568 requested_mixdown = audio->config.out.mixdown;
569 requested_mixdown_index = j;
573 /* sense-check the requested mixdown */
575 if( audio->config.out.mixdown == 0 &&
576 audio->config.out.codec != HB_ACODEC_AC3 &&
577 audio->config.out.codec != HB_ACODEC_DCA )
580 * Mixdown wasn't specified and this is not pass-through,
581 * set a default mixdown of stereo.
583 audio->config.out.mixdown = HB_AMIXDOWN_STEREO;
586 // Here we try to sanitize the audio input to output mapping.
588 // 1. only the AC3 & DCA decoder libraries currently support mixdown
589 // 2. the lame encoder library only supports stereo.
590 // So if the encoder is lame we need the output to be stereo (or multichannel
591 // matrixed into stereo like dpl). If the decoder is not AC3 or DCA the
592 // encoder has to handle the input format since we can't do a mixdown.
593 switch (audio->config.in.channel_layout & HB_INPUT_CH_LAYOUT_DISCRETE_NO_LFE_MASK)
595 // stereo input or something not handled below
597 case HB_INPUT_CH_LAYOUT_STEREO:
598 // mono gets mixed up to stereo & more than stereo gets mixed down
599 if ( audio->config.out.mixdown > HB_AMIXDOWN_STEREO )
601 audio->config.out.mixdown = HB_AMIXDOWN_STEREO;
606 case HB_INPUT_CH_LAYOUT_MONO:
607 // everything else passes through
608 audio->config.out.mixdown = HB_AMIXDOWN_MONO;
611 // dolby (DPL1 aka Dolby Surround = 4.0 matrix-encoded) input
612 // the A52 flags don't allow for a way to distinguish between DPL1 and
613 // DPL2 on a DVD so we always assume a DPL1 source for A52_DOLBY.
614 case HB_INPUT_CH_LAYOUT_DOLBY:
615 if ( audio->config.out.mixdown > HB_AMIXDOWN_DOLBY )
617 audio->config.out.mixdown = HB_AMIXDOWN_DOLBY;
621 // 4 channel discrete
622 case HB_INPUT_CH_LAYOUT_2F2R:
623 case HB_INPUT_CH_LAYOUT_3F1R:
624 if ( audio->config.out.mixdown > HB_AMIXDOWN_DOLBY )
626 audio->config.out.mixdown = HB_AMIXDOWN_DOLBY;
630 // 5 or 6 channel discrete
631 case HB_INPUT_CH_LAYOUT_3F2R:
632 if ( ! ( audio->config.in.channel_layout &
633 HB_INPUT_CH_LAYOUT_HAS_LFE ) )
635 // we don't do 5 channel discrete so mixdown to DPLII
636 audio->config.out.mixdown = HB_AMIXDOWN_DOLBYPLII;
641 /* log the output mixdown */
642 for (j = 0; j < hb_audio_mixdowns_count; j++) {
643 if (hb_audio_mixdowns[j].amixdown == audio->config.out.mixdown) {
644 if ( audio->config.out.mixdown != requested_mixdown )
646 hb_log("work: sanitizing track %i mixdown %s to %s", i, hb_audio_mixdowns[requested_mixdown_index].human_readable_name, hb_audio_mixdowns[j].human_readable_name);
653 if (audio->config.out.codec == HB_ACODEC_VORBIS)
654 audio->priv.config.vorbis.language = audio->config.lang.simple;
656 /* set up the audio work structures */
657 audio->priv.fifo_in = hb_fifo_init( FIFO_LARGE, FIFO_LARGE_WAKE );
658 audio->priv.fifo_raw = hb_fifo_init( FIFO_SMALL, FIFO_SMALL_WAKE );
659 audio->priv.fifo_sync = hb_fifo_init( FIFO_SMALL, FIFO_SMALL_WAKE );
660 audio->priv.fifo_out = hb_fifo_init( FIFO_LARGE, FIFO_LARGE_WAKE );
664 /* Synchronization */
665 sync = hb_sync_init( job );
668 int vcodec = title->video_codec? title->video_codec : WORK_DECMPEG2;
669 hb_list_add( job->list_work, ( w = hb_get_work( vcodec ) ) );
670 w->codec_param = title->video_codec_param;
671 w->fifo_in = job->fifo_mpeg2;
672 w->fifo_out = job->fifo_raw;
675 hb_list_add( job->list_work, ( w = hb_get_work( WORK_RENDER ) ) );
676 w->fifo_in = job->fifo_sync;
677 if( !job->indepth_scan )
678 w->fifo_out = job->fifo_render;
682 if( !job->indepth_scan )
686 switch( job->vcodec )
688 case HB_VCODEC_FFMPEG:
689 w = hb_get_work( WORK_ENCAVCODEC );
692 w = hb_get_work( WORK_ENCX264 );
694 case HB_VCODEC_THEORA:
695 w = hb_get_work( WORK_ENCTHEORA );
698 w->fifo_in = job->fifo_render;
699 w->fifo_out = job->fifo_mpeg4;
700 w->config = &job->config;
702 hb_list_add( job->list_work, w );
706 * Look for the scanned subtitle in the existing subtitle list
707 * select_subtitle implies that we did a scan.
709 if ( !job->indepth_scan && interjob->select_subtitle &&
710 ( job->pass == 0 || job->pass == 2 ) )
713 * Disable forced subtitles if we didn't find any in the scan
714 * so that we display normal subtitles instead.
716 if( interjob->select_subtitle->config.force &&
717 interjob->select_subtitle->forced_hits == 0 )
719 interjob->select_subtitle->config.force = 0;
721 for( i=0; i < hb_list_count(title->list_subtitle); i++ )
723 subtitle = hb_list_item( title->list_subtitle, i );
728 * Remove the scanned subtitle from the subtitle list if
729 * it would result in an identical duplicate subtitle track
730 * or an emty track (forced and no forced hits).
732 if( ( interjob->select_subtitle->id == subtitle->id ) &&
733 ( ( interjob->select_subtitle->forced_hits == 0 &&
734 subtitle->config.force ) ||
735 ( subtitle->config.force == interjob->select_subtitle->config.force ) ) )
737 *subtitle = *(interjob->select_subtitle);
738 free( interjob->select_subtitle );
739 interjob->select_subtitle = NULL;
745 if( interjob->select_subtitle )
748 * Its not in the existing list
750 * Must be second pass of a two pass with subtitle scan enabled, so
751 * add the subtitle that we found on the first pass for use in this
754 hb_list_add( title->list_subtitle, interjob->select_subtitle );
755 interjob->select_subtitle = NULL;
760 for( i=0; i < hb_list_count(title->list_subtitle); i++ )
762 subtitle = hb_list_item( title->list_subtitle, i );
766 subtitle->fifo_in = hb_fifo_init( FIFO_SMALL, FIFO_SMALL_WAKE );
767 subtitle->fifo_raw = hb_fifo_init( FIFO_SMALL, FIFO_SMALL_WAKE );
768 subtitle->fifo_sync = hb_fifo_init( FIFO_SMALL, FIFO_SMALL_WAKE );
769 subtitle->fifo_out = hb_fifo_init( FIFO_SMALL, FIFO_SMALL_WAKE );
771 if( (!job->indepth_scan || job->select_subtitle_config.force) &&
772 subtitle->source == VOBSUB ) {
774 * Don't add threads for subtitles when we are scanning, unless
775 * looking for forced subtitles.
777 w = hb_get_work( WORK_DECVOBSUB );
778 w->fifo_in = subtitle->fifo_in;
779 w->fifo_out = subtitle->fifo_raw;
780 w->subtitle = subtitle;
781 hb_list_add( job->list_work, w );
784 if( !job->indepth_scan && subtitle->source == CC608SUB )
786 w = hb_get_work( WORK_DECCC608 );
787 w->fifo_in = subtitle->fifo_in;
788 w->fifo_out = subtitle->fifo_raw;
789 hb_list_add( job->list_work, w );
792 if( !job->indepth_scan && subtitle->source == SRTSUB )
794 w = hb_get_work( WORK_DECSRTSUB );
795 w->fifo_in = subtitle->fifo_in;
796 w->fifo_out = subtitle->fifo_raw;
797 w->subtitle = subtitle;
798 hb_list_add( job->list_work, w );
801 if( !job->indepth_scan && subtitle->source == UTF8SUB )
803 w = hb_get_work( WORK_DECUTF8SUB );
804 w->fifo_in = subtitle->fifo_in;
805 w->fifo_out = subtitle->fifo_raw;
806 hb_list_add( job->list_work, w );
809 if( !job->indepth_scan && subtitle->source == TX3GSUB )
811 w = hb_get_work( WORK_DECTX3GSUB );
812 w->fifo_in = subtitle->fifo_in;
813 w->fifo_out = subtitle->fifo_raw;
814 hb_list_add( job->list_work, w );
817 if( !job->indepth_scan && subtitle->source == SSASUB )
819 w = hb_get_work( WORK_DECSSASUB );
820 w->fifo_in = subtitle->fifo_in;
821 w->fifo_out = subtitle->fifo_raw;
822 hb_list_add( job->list_work, w );
825 if( !job->indepth_scan &&
826 subtitle->format == PICTURESUB
827 && subtitle->config.dest == PASSTHRUSUB )
830 * Passing through a subtitle picture, this will have to
831 * be rle encoded before muxing.
833 w = hb_get_work( WORK_ENCVOBSUB );
834 w->fifo_in = subtitle->fifo_sync;
835 w->fifo_out = subtitle->fifo_out;
836 w->subtitle = subtitle;
837 hb_list_add( job->list_work, w );
842 if( !job->indepth_scan )
844 for( i = 0; i < hb_list_count( title->list_audio ); i++ )
846 audio = hb_list_item( title->list_audio, i );
849 * Audio Decoder Thread
851 if ( ( w = hb_codec_decoder( audio->config.in.codec ) ) == NULL )
853 hb_error("Invalid input codec: %d", audio->config.in.codec);
857 w->fifo_in = audio->priv.fifo_in;
858 w->fifo_out = audio->priv.fifo_raw;
859 w->config = &audio->priv.config;
861 w->codec_param = audio->config.in.codec_param;
863 hb_list_add( job->list_work, w );
866 * Audio Encoder Thread
868 if( audio->config.out.codec != HB_ACODEC_AC3 &&
869 audio->config.out.codec != HB_ACODEC_DCA )
872 * Add the encoder thread if not doing AC-3 pass through
874 if ( ( w = hb_codec_encoder( audio->config.out.codec ) ) == NULL )
876 hb_error("Invalid audio codec: %#x", audio->config.out.codec);
881 w->fifo_in = audio->priv.fifo_sync;
882 w->fifo_out = audio->priv.fifo_out;
883 w->config = &audio->priv.config;
886 hb_list_add( job->list_work, w );
891 if( job->chapter_markers && job->chapter_start == job->chapter_end )
893 job->chapter_markers = 0;
894 hb_log("work: only 1 chapter, disabling chapter markers");
897 /* Display settings */
898 hb_display_job_info( job );
900 /* Init read & write threads */
901 job->reader = hb_reader_init( job );
906 /* Launch processing threads */
907 for( i = 0; i < hb_list_count( job->list_work ); i++ )
909 w = hb_list_item( job->list_work, i );
910 w->done = &job->done;
911 w->thread_sleep_interval = 10;
912 if( w->init( w, job ) )
914 hb_error( "Failure to initialise thread '%s'", w->name );
918 w->thread = hb_thread_init( w->name, work_loop, w,
922 if ( job->indepth_scan )
926 sync->done = &job->done;
930 sync->done = &job->done;
931 sync->thread_sleep_interval = 10;
932 if( sync->init( w, job ) )
934 hb_error( "Failure to initialise thread '%s'", w->name );
938 sync->thread = hb_thread_init( sync->name, work_loop, sync,
941 // The muxer requires track information that's set up by the encoder
942 // init routines so we have to init the muxer last.
943 muxer = hb_muxer_init( job );
947 while ( !*job->die && !*w->done && w->status != HB_WORK_DONE )
949 hb_buffer_t * buf_in, * buf_out;
951 buf_in = hb_fifo_get_wait( w->fifo_in );
952 if ( buf_in == NULL )
958 hb_buffer_close( &buf_in );
964 w->status = w->work( w, &buf_in, &buf_out );
968 hb_buffer_close( &buf_in );
970 if ( buf_out && w->fifo_out == NULL )
972 hb_buffer_close( &buf_out );
978 if ( hb_fifo_full_wait( w->fifo_out ) )
980 hb_fifo_push( w->fifo_out, buf_out );
987 hb_handle_t * h = job->h;
989 hb_get_state( h, &state );
991 hb_log("work: average encoding speed for job is %f fps", state.param.working.rate_avg);
996 muxer->close( muxer );
999 if( sync->thread != NULL )
1001 hb_thread_close( &sync->thread );
1002 sync->close( sync );
1008 /* Stop the write thread (thread_close will block until the muxer finishes) */
1011 /* Close work objects */
1012 while( ( w = hb_list_item( job->list_work, 0 ) ) )
1014 hb_list_rem( job->list_work, w );
1015 if( w->thread != NULL )
1017 hb_thread_close( &w->thread );
1023 hb_list_close( &job->list_work );
1025 /* Stop the read thread */
1026 if( job->reader != NULL )
1027 hb_thread_close( &job->reader );
1030 hb_fifo_close( &job->fifo_mpeg2 );
1031 hb_fifo_close( &job->fifo_raw );
1032 hb_fifo_close( &job->fifo_sync );
1033 hb_fifo_close( &job->fifo_render );
1034 hb_fifo_close( &job->fifo_mpeg4 );
1036 for (i=0; i < hb_list_count(title->list_subtitle); i++) {
1037 subtitle = hb_list_item( title->list_subtitle, i);
1040 hb_fifo_close( &subtitle->fifo_in );
1041 hb_fifo_close( &subtitle->fifo_raw );
1042 hb_fifo_close( &subtitle->fifo_sync );
1043 hb_fifo_close( &subtitle->fifo_out );
1046 for( i = 0; i < hb_list_count( title->list_audio ); i++ )
1048 audio = hb_list_item( title->list_audio, i );
1049 if( audio->priv.fifo_in != NULL )
1050 hb_fifo_close( &audio->priv.fifo_in );
1051 if( audio->priv.fifo_raw != NULL )
1052 hb_fifo_close( &audio->priv.fifo_raw );
1053 if( audio->priv.fifo_sync != NULL )
1054 hb_fifo_close( &audio->priv.fifo_sync );
1055 if( audio->priv.fifo_out != NULL )
1056 hb_fifo_close( &audio->priv.fifo_out );
1059 if( job->indepth_scan )
1062 * Before closing the title print out our subtitle stats if we need to
1063 * Find the highest and lowest.
1065 for( i=0; i < hb_list_count( title->list_subtitle ); i++ )
1067 subtitle = hb_list_item( title->list_subtitle, i );
1069 hb_log( "Subtitle stream 0x%x '%s': %d hits (%d forced)",
1070 subtitle->id, subtitle->lang, subtitle->hits,
1071 subtitle->forced_hits );
1073 if( subtitle->hits == 0 )
1076 if( subtitle->hits > subtitle_highest )
1078 subtitle_highest = subtitle->hits;
1079 subtitle_highest_id = subtitle->id;
1082 if( subtitle->hits < subtitle_lowest )
1084 subtitle_lowest = subtitle->hits;
1085 subtitle_lowest_id = subtitle->id;
1088 if( subtitle->forced_hits > 0 )
1090 if( subtitle_forced_id == 0 )
1092 subtitle_forced_id = subtitle->id;
1098 if( subtitle_forced_id )
1101 * If there are any subtitle streams with forced subtitles
1102 * then select it in preference to the lowest.
1104 subtitle_hit = subtitle_forced_id;
1105 hb_log("Found a subtitle candidate id 0x%x (contains forced subs)",
1107 } else if( subtitle_lowest < subtitle_highest )
1110 * OK we have more than one, and the lowest is lower,
1111 * but how much lower to qualify for turning it on by
1114 * Let's say 10% as a default.
1116 if( subtitle_lowest < ( subtitle_highest * 0.1 ) )
1118 subtitle_hit = subtitle_lowest_id;
1119 hb_log( "Found a subtitle candidate id 0x%x",
1122 hb_log( "No candidate subtitle detected during subtitle-scan");
1127 if( job->indepth_scan )
1129 for( i=0; i < hb_list_count( title->list_subtitle ); i++ )
1131 subtitle = hb_list_item( title->list_subtitle, i );
1132 if( subtitle->id == subtitle_hit )
1134 subtitle->config = job->select_subtitle_config;
1135 hb_list_rem( title->list_subtitle, subtitle );
1136 interjob->select_subtitle = subtitle;
1144 for( i = 0; i < hb_list_count( job->filters ); i++ )
1146 hb_filter_object_t * filter = hb_list_item( job->filters, i );
1147 hb_filter_close( &filter );
1149 hb_list_close( &job->filters );
1152 hb_buffer_pool_free();
1154 hb_title_close( &job->title );
1159 * Performs the work object's specific work function.
1160 * Loops calling work function for associated work object. Sleeps when fifo is full.
1161 * Monitors work done indicator.
1162 * Exits loop when work indiactor is set.
1163 * @param _w Handle to work object.
1165 static void work_loop( void * _w )
1167 hb_work_object_t * w = _w;
1168 hb_buffer_t * buf_in, * buf_out;
1170 while( !*w->done && w->status != HB_WORK_DONE )
1172 buf_in = hb_fifo_get_wait( w->fifo_in );
1173 if ( buf_in == NULL )
1179 hb_buffer_close( &buf_in );
1184 // Invalidate buf_out so that if there is no output
1185 // we don't try to pass along junk.
1187 w->status = w->work( w, &buf_in, &buf_out );
1189 // Propagate any chapter breaks for the worker if and only if the
1190 // output frame has the same time stamp as the input frame (any
1191 // worker that delays frames has to propagate the chapter marks itself
1192 // and workers that move chapter marks to a different time should set
1193 // 'buf_in' to NULL so that this code won't generate spurious duplicates.)
1194 if( buf_in && buf_out && buf_in->new_chap && buf_in->start == buf_out->start)
1196 // restore log below to debug chapter mark propagation problems
1197 //hb_log("work %s: Copying Chapter Break @ %"PRId64, w->name, buf_in->start);
1198 buf_out->new_chap = buf_in->new_chap;
1203 hb_buffer_close( &buf_in );
1205 if ( buf_out && w->fifo_out == NULL )
1207 hb_buffer_close( &buf_out );
1213 if ( hb_fifo_full_wait( w->fifo_out ) )
1215 hb_fifo_push( w->fifo_out, buf_out );
1221 // Consume data in incoming fifo till job complete so that
1222 // residual data does not stall the pipeline
1225 buf_in = hb_fifo_get_wait( w->fifo_in );
1226 if ( buf_in != NULL )
1227 hb_buffer_close( &buf_in );