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.m0k.org/>.
5 It may be used under the terms of the GNU General Public License. */
8 #include "a52dec/a52.h"
14 hb_job_t ** current_job;
21 static void work_func();
22 static void do_job( hb_job_t *, int cpu_count );
23 static void work_loop( void * );
25 #define FIFO_CPU_MULT 8
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.
34 hb_thread_t * hb_work_init( hb_list_t * jobs, int cpu_count,
35 volatile int * die, int * error, hb_job_t ** job )
37 hb_work_t * work = calloc( sizeof( hb_work_t ), 1 );
40 work->current_job = job;
41 work->cpu_count = cpu_count;
45 return hb_thread_init( "work", work_func, work, HB_LOW_PRIORITY );
49 * Iterates through job list and calls do_job for each job.
50 * @param _work Handle work object.
52 static void work_func( void * _work )
54 hb_work_t * work = _work;
57 hb_log( "%d job(s) to process", hb_list_count( work->jobs ) );
59 while( !*work->die && ( job = hb_list_item( work->jobs, 0 ) ) )
61 hb_list_rem( work->jobs, job );
63 *(work->current_job) = job;
64 do_job( job, work->cpu_count );
65 *(work->current_job) = NULL;
68 *(work->error) = HB_ERROR_NONE;
73 static hb_work_object_t * getWork( int id )
76 for( w = hb_objects; w; w = w->next )
87 * Job initialization rountine.
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.
97 static void do_job( hb_job_t * job, int cpu_count )
101 hb_work_object_t * w;
103 /* FIXME: This feels really hackish, anything better? */
104 hb_work_object_t * audio_w = NULL;
105 hb_work_object_t * sub_w = NULL;
108 hb_subtitle_t * subtitle;
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;
119 job->list_work = hb_list_init();
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 )
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];
131 else if ( job->pixel_ratio == 2 )
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);
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))
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);
150 if (job->maxWidth && (job->width > job->maxWidth) && (job->pixel_ratio != 2))
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);
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 hb_log( " + grayscale %s", job->grayscale ? "on" : "off" );
165 job->vrate_base = 900900;
167 int detelecine_present = 0;
170 for( i = 0; i < hb_list_count( job->filters ); i++ )
172 hb_filter_object_t * filter = hb_list_item( job->filters, i );
173 if (filter->id == FILTER_DETELECINE)
174 detelecine_present = 1;
178 if (!detelecine_present)
179 hb_list_add( job->filters, &hb_filter_detelecine );
181 hb_log("work: VFR mode -- Switching FPS to 29.97 and detelecining.");
186 hb_log(" + filters");
187 for( i = 0; i < hb_list_count( job->filters ); i++ )
189 hb_filter_object_t * filter = hb_list_item( job->filters, i );
190 if (filter->settings)
191 hb_log(" + %s (%s)", filter->name, filter->settings);
193 hb_log(" + %s (default settings)", filter->name);
197 if( job->vquality >= 0.0 && job->vquality <= 1.0 )
199 hb_log( " + %.3f fps, video quality %.2f", (float) job->vrate /
200 (float) job->vrate_base, job->vquality );
204 hb_log( " + %.3f fps, video bitrate %d kbps, pass %d",
205 (float) job->vrate / (float) job->vrate_base,
206 job->vbitrate, job->pass );
208 hb_log (" + PixelRatio: %d, width:%d, height: %d",job->pixel_ratio,job->width, job->height);
209 job->fifo_mpeg2 = hb_fifo_init( 2048 );
210 job->fifo_raw = hb_fifo_init( FIFO_CPU_MULT * cpu_count );
211 job->fifo_sync = hb_fifo_init( FIFO_CPU_MULT * cpu_count );
212 job->fifo_render = hb_fifo_init( FIFO_CPU_MULT * cpu_count );
213 job->fifo_mpeg4 = hb_fifo_init( FIFO_CPU_MULT * cpu_count );
215 /* Synchronization */
216 hb_list_add( job->list_work, ( w = getWork( WORK_SYNC ) ) );
221 hb_list_add( job->list_work, ( w = getWork( WORK_DECMPEG2 ) ) );
222 w->fifo_in = job->fifo_mpeg2;
223 w->fifo_out = job->fifo_raw;
226 hb_list_add( job->list_work, ( w = getWork( WORK_RENDER ) ) );
227 w->fifo_in = job->fifo_sync;
228 w->fifo_out = job->fifo_render;
231 switch( job->vcodec )
233 case HB_VCODEC_FFMPEG:
234 hb_log( " + encoder FFmpeg" );
235 w = getWork( WORK_ENCAVCODEC );
238 hb_log( " + encoder XviD" );
239 w = getWork( WORK_ENCXVID );
242 hb_log( " + encoder x264" );
243 if( job->x264opts != NULL && *job->x264opts != '\0' )
244 hb_log( " + x264 options: %s", job->x264opts);
245 w = getWork( WORK_ENCX264 );
248 w->fifo_in = job->fifo_render;
249 w->fifo_out = job->fifo_mpeg4;
250 w->config = &job->config;
252 hb_list_add( job->list_work, w );
254 if( job->select_subtitle && !job->indepth_scan )
257 * Must be second pass of a two pass with subtitle scan enabled, so
258 * add the subtitle that we found on the first pass for use in this
261 if (*(job->select_subtitle))
263 hb_list_add( title->list_subtitle, *( job->select_subtitle ) );
267 for( i=0; i < hb_list_count(title->list_subtitle); i++ )
269 subtitle = hb_list_item( title->list_subtitle, i );
273 hb_log( " + subtitle %x, %s", subtitle->id, subtitle->lang );
275 subtitle->fifo_in = hb_fifo_init( FIFO_CPU_MULT * cpu_count );
276 subtitle->fifo_raw = hb_fifo_init( FIFO_CPU_MULT * cpu_count );
279 * Disable forced subtitles if we didn't find any in the scan
280 * so that we display normal subtitles instead.
282 * select_subtitle implies that we did a scan.
284 if( !job->indepth_scan && job->subtitle_force &&
285 job->select_subtitle )
287 if( subtitle->forced_hits == 0 )
289 job->subtitle_force = 0;
293 if (!job->indepth_scan || job->subtitle_force) {
295 * Don't add threads for subtitles when we are scanning, unless
296 * looking for forced subtitles.
301 * Need to copy the prior subtitle structure so that we
302 * don't overwrite the fifos.
304 sub_w = calloc( sizeof( hb_work_object_t ), 1 );
305 sub_w = memcpy( sub_w, w, sizeof( hb_work_object_t ));
307 w = sub_w = getWork( WORK_DECSUB );
309 hb_list_add( job->list_work, sub_w );
310 sub_w->fifo_in = subtitle->fifo_in;
311 sub_w->fifo_out = subtitle->fifo_raw;
316 if( job->acodec & HB_ACODEC_AC3 )
318 hb_log( " + audio AC3 passthrough" );
322 hb_log( " + audio %d kbps, %d Hz", job->abitrate, job->arate );
323 hb_log( " + encoder %s", ( job->acodec & HB_ACODEC_FAAC ) ?
324 "faac" : ( ( job->acodec & HB_ACODEC_LAME ) ? "lame" :
328 /* if we are doing AC3 passthru, then remove any non-AC3 audios from the job */
329 /* otherwise, Bad Things will happen */
330 for( i = 0; i < hb_list_count( title->list_audio ); )
332 audio = hb_list_item( title->list_audio, i );
333 if( ( job->acodec & HB_ACODEC_AC3 ) && ( audio->codec != HB_ACODEC_AC3 ) )
335 hb_list_rem( title->list_audio, audio );
342 for( i = 0; i < hb_list_count( title->list_audio ); i++ )
344 audio = hb_list_item( title->list_audio, i );
345 hb_log( " + %x, %s", audio->id, audio->lang );
347 /* sense-check the current mixdown options */
349 /* log the requested mixdown */
350 for (j = 0; j < hb_audio_mixdowns_count; j++) {
351 if (hb_audio_mixdowns[j].amixdown == job->audio_mixdowns[i]) {
352 hb_log( " + Requested mixdown: %s (%s)", hb_audio_mixdowns[j].human_readable_name, hb_audio_mixdowns[j].internal_name );
357 /* sense-check the requested mixdown */
359 /* audioCodecsSupportMono and audioCodecsSupport6Ch are the same for now,
360 but this may change in the future, so they are separated for flexibility */
361 int audioCodecsSupportMono = ((audio->codec == HB_ACODEC_AC3 ||
362 audio->codec == HB_ACODEC_DCA) && (job->acodec == HB_ACODEC_FAAC || job->acodec == HB_ACODEC_VORBIS));
363 int audioCodecsSupport6Ch = ((audio->codec == HB_ACODEC_AC3 ||
364 audio->codec == HB_ACODEC_DCA) && (job->acodec == HB_ACODEC_FAAC || job->acodec == HB_ACODEC_VORBIS));
366 /* find out what the format of our source audio is */
367 switch (audio->input_channel_layout & HB_INPUT_CH_LAYOUT_DISCRETE_NO_LFE_MASK) {
370 case HB_INPUT_CH_LAYOUT_MONO:
371 /* regardless of what stereo mixdown we've requested, a mono source always get mixed down
372 to mono if we can, and mixed up to stereo if we can't */
373 if (job->audio_mixdowns[i] == HB_AMIXDOWN_MONO && audioCodecsSupportMono == 1) {
374 job->audio_mixdowns[i] = HB_AMIXDOWN_MONO;
376 job->audio_mixdowns[i] = HB_AMIXDOWN_STEREO;
381 case HB_INPUT_CH_LAYOUT_STEREO:
382 /* if we've requested a mono mixdown, and it is supported, then do the mix */
383 /* use stereo if not supported */
384 if (job->audio_mixdowns[i] == HB_AMIXDOWN_MONO && audioCodecsSupportMono == 0) {
385 job->audio_mixdowns[i] = HB_AMIXDOWN_STEREO;
386 /* otherwise, preserve stereo regardless of if we requested something higher */
387 } else if (job->audio_mixdowns[i] > HB_AMIXDOWN_STEREO) {
388 job->audio_mixdowns[i] = HB_AMIXDOWN_STEREO;
392 /* dolby (DPL1 aka Dolby Surround = 4.0 matrix-encoded) input */
393 /* the A52 flags don't allow for a way to distinguish between DPL1 and DPL2 on a DVD,
394 so we always assume a DPL1 source for A52_DOLBY */
395 case HB_INPUT_CH_LAYOUT_DOLBY:
396 /* if we've requested a mono mixdown, and it is supported, then do the mix */
397 /* preserve dolby if not supported */
398 if (job->audio_mixdowns[i] == HB_AMIXDOWN_MONO && audioCodecsSupportMono == 0) {
399 job->audio_mixdowns[i] = HB_AMIXDOWN_DOLBY;
400 /* otherwise, preserve dolby even if we requested something higher */
401 /* a stereo mixdown will still be honoured here */
402 } else if (job->audio_mixdowns[i] > HB_AMIXDOWN_DOLBY) {
403 job->audio_mixdowns[i] = HB_AMIXDOWN_DOLBY;
408 case HB_INPUT_CH_LAYOUT_3F2R:
409 /* if we've requested a mono mixdown, and it is supported, then do the mix */
410 /* use dpl2 if not supported */
411 if (job->audio_mixdowns[i] == HB_AMIXDOWN_MONO && audioCodecsSupportMono == 0) {
412 job->audio_mixdowns[i] = HB_AMIXDOWN_DOLBYPLII;
414 /* check if we have 3F2R input and also have an LFE - i.e. we have a 5.1 source) */
415 if (audio->input_channel_layout & HB_INPUT_CH_LAYOUT_HAS_LFE) {
416 /* we have a 5.1 source */
417 /* if we requested 6ch, but our audio format doesn't support it, then mix to DPLII instead */
418 if (job->audio_mixdowns[i] == HB_AMIXDOWN_6CH && audioCodecsSupport6Ch == 0) {
419 job->audio_mixdowns[i] = HB_AMIXDOWN_DOLBYPLII;
422 /* we have a 5.0 source, so we can't do 6ch conversion
423 default to DPL II instead */
424 if (job->audio_mixdowns[i] > HB_AMIXDOWN_DOLBYPLII) {
425 job->audio_mixdowns[i] = HB_AMIXDOWN_DOLBYPLII;
429 /* all other mixdowns will have been preserved here */
433 case HB_INPUT_CH_LAYOUT_3F1R:
434 /* if we've requested a mono mixdown, and it is supported, then do the mix */
435 /* use dpl1 if not supported */
436 if (job->audio_mixdowns[i] == HB_AMIXDOWN_MONO && audioCodecsSupportMono == 0) {
437 job->audio_mixdowns[i] = HB_AMIXDOWN_DOLBY;
439 /* we have a 4.0 or 4.1 source, so we can't do DPLII or 6ch conversion
440 default to DPL I instead */
441 if (job->audio_mixdowns[i] > HB_AMIXDOWN_DOLBY) {
442 job->audio_mixdowns[i] = HB_AMIXDOWN_DOLBY;
445 /* all other mixdowns will have been preserved here */
449 /* if we've requested a mono mixdown, and it is supported, then do the mix */
450 if (job->audio_mixdowns[i] == HB_AMIXDOWN_MONO && audioCodecsSupportMono == 1) {
451 job->audio_mixdowns[i] = HB_AMIXDOWN_MONO;
452 /* mix everything else down to stereo */
454 job->audio_mixdowns[i] = HB_AMIXDOWN_STEREO;
459 /* log the output mixdown */
460 for (j = 0; j < hb_audio_mixdowns_count; j++) {
461 if (hb_audio_mixdowns[j].amixdown == job->audio_mixdowns[i]) {
462 hb_log( " + Actual mixdown: %s (%s)", hb_audio_mixdowns[j].human_readable_name, hb_audio_mixdowns[j].internal_name );
467 /* we now know we have a valid mixdown for the input source and the audio output format */
468 /* remember the mixdown for this track */
469 audio->amixdown = job->audio_mixdowns[i];
471 audio->config.vorbis.language = audio->lang_simple;
473 /* set up the audio work structures */
474 audio->fifo_in = hb_fifo_init( 2048 );
475 audio->fifo_raw = hb_fifo_init( FIFO_CPU_MULT * cpu_count );
476 audio->fifo_sync = hb_fifo_init( FIFO_CPU_MULT * cpu_count );
477 audio->fifo_out = hb_fifo_init( FIFO_CPU_MULT * cpu_count );
479 switch( audio->codec )
482 w = getWork( WORK_DECA52 );
485 w = getWork( WORK_DECDCA );
488 w = getWork( WORK_DECAVCODEC );
491 w = getWork( WORK_DECLPCM );
494 w->fifo_in = audio->fifo_in;
495 w->fifo_out = audio->fifo_raw;
496 w->config = &audio->config;
497 w->amixdown = audio->amixdown;
498 w->source_acodec = audio->codec;
500 /* FIXME: This feels really hackish, anything better? */
501 audio_w = calloc( sizeof( hb_work_object_t ), 1 );
502 audio_w = memcpy( audio_w, w, sizeof( hb_work_object_t ));
504 hb_list_add( job->list_work, audio_w );
506 switch( job->acodec )
509 w = getWork( WORK_ENCFAAC );
512 w = getWork( WORK_ENCLAME );
514 case HB_ACODEC_VORBIS:
515 w = getWork( WORK_ENCVORBIS );
519 if( job->acodec != HB_ACODEC_AC3 )
521 w->fifo_in = audio->fifo_sync;
522 w->fifo_out = audio->fifo_out;
523 w->config = &audio->config;
524 w->amixdown = audio->amixdown;
525 w->source_acodec = audio->codec;
527 /* FIXME: This feels really hackish, anything better? */
528 audio_w = calloc( sizeof( hb_work_object_t ), 1 );
529 audio_w = memcpy( audio_w, w, sizeof( hb_work_object_t ));
531 hb_list_add( job->list_work, audio_w );
537 /* Init read & write threads */
538 job->reader = hb_reader_init( job );
540 hb_log( " + output: %s", job->file );
541 job->muxer = hb_muxer_init( job );
545 /* Launch processing threads */
546 for( i = 1; i < hb_list_count( job->list_work ); i++ )
548 w = hb_list_item( job->list_work, i );
549 w->done = &job->done;
550 w->thread_sleep_interval = 10;
552 w->thread = hb_thread_init( w->name, work_loop, w,
557 w = hb_list_item( job->list_work, 0 );
558 w->thread_sleep_interval = 50;
562 if( w->work( w, NULL, NULL ) == HB_WORK_DONE )
567 !hb_fifo_size( job->fifo_sync ) &&
568 !hb_fifo_size( job->fifo_render ) &&
569 !hb_fifo_size( job->fifo_mpeg4 ) )
573 hb_snooze( w->thread_sleep_interval );
575 hb_list_rem( job->list_work, w );
579 /* Close work objects */
580 while( ( w = hb_list_item( job->list_work, 0 ) ) )
582 hb_list_rem( job->list_work, w );
583 hb_thread_close( &w->thread );
586 /* FIXME: This feels really hackish, anything better? */
587 if ( w->id == WORK_DECA52 ||
588 w->id == WORK_DECDCA ||
589 w->id == WORK_DECLPCM ||
590 w->id == WORK_ENCFAAC ||
591 w->id == WORK_ENCLAME ||
592 w->id == WORK_ENCVORBIS )
599 hb_list_close( &job->list_work );
601 /* Stop read & write threads */
602 hb_thread_close( &job->reader );
603 hb_thread_close( &job->muxer );
606 hb_fifo_close( &job->fifo_mpeg2 );
607 hb_fifo_close( &job->fifo_raw );
608 hb_fifo_close( &job->fifo_sync );
609 hb_fifo_close( &job->fifo_render );
610 hb_fifo_close( &job->fifo_mpeg4 );
612 for (i=0; i < hb_list_count(title->list_subtitle); i++) {
613 subtitle = hb_list_item( title->list_subtitle, i);
616 hb_fifo_close( &subtitle->fifo_in );
617 hb_fifo_close( &subtitle->fifo_raw );
620 for( i = 0; i < hb_list_count( title->list_audio ); i++ )
622 audio = hb_list_item( title->list_audio, i );
623 hb_fifo_close( &audio->fifo_in );
624 hb_fifo_close( &audio->fifo_raw );
625 hb_fifo_close( &audio->fifo_sync );
626 hb_fifo_close( &audio->fifo_out );
629 if( job->indepth_scan )
632 * Before closing the title print out our subtitle stats if we need to
633 * Find the highest and lowest.
635 for( i=0; i < hb_list_count( title->list_subtitle ); i++ )
637 subtitle = hb_list_item( title->list_subtitle, i );
638 hb_log( "Subtitle stream 0x%x '%s': %d hits (%d forced)",
639 subtitle->id, subtitle->lang, subtitle->hits,
640 subtitle->forced_hits );
641 if( subtitle->hits > subtitle_highest )
643 subtitle_highest = subtitle->hits;
644 subtitle_highest_id = subtitle->id;
647 if( subtitle->hits < subtitle_lowest )
649 subtitle_lowest = subtitle->hits;
650 subtitle_lowest_id = subtitle->id;
653 if( subtitle->forced_hits > 0 )
655 if( subtitle_forced_id == 0 )
657 subtitle_forced_id = subtitle->id;
662 if( job->native_language ) {
664 * We still have a native_language, so the audio and subtitles are
665 * different, so in this case it is a foreign film and we want to
666 * select the subtitle with the highest hits in our language.
668 subtitle_hit = subtitle_highest_id;
669 hb_log( "Found a native-language subtitle id 0x%x", subtitle_hit);
671 if( subtitle_forced_id )
674 * If there are any subtitle streams with forced subtitles
675 * then select it in preference to the lowest.
677 subtitle_hit = subtitle_forced_id;
678 hb_log("Found a subtitle candidate id 0x%x (contains forced subs)",
680 } else if( subtitle_lowest < subtitle_highest )
683 * OK we have more than one, and the lowest is lower,
684 * but how much lower to qualify for turning it on by
687 * Let's say 10% as a default.
689 if( subtitle_lowest < ( subtitle_highest * 0.1 ) )
691 subtitle_hit = subtitle_lowest_id;
692 hb_log( "Found a subtitle candidate id 0x%x",
695 hb_log( "No candidate subtitle detected during subtitle-scan");
701 if( job->select_subtitle )
703 if( job->indepth_scan )
705 for( i=0; i < hb_list_count( title->list_subtitle ); i++ )
707 subtitle = hb_list_item( title->list_subtitle, i );
708 if( subtitle->id == subtitle_hit )
710 hb_list_rem( title->list_subtitle, subtitle );
711 *( job->select_subtitle ) = subtitle;
716 * Must be the end of pass 0 or 2 - we don't need this anymore.
718 * Have to put the subtitle list back together in the title though
719 * or the GUI will have a hissy fit.
721 free( job->select_subtitle );
722 job->select_subtitle = NULL;
726 hb_buffer_pool_free();
728 hb_title_close( &job->title );
733 * Performs the work objects specific work function.
734 * Loops calling work function for associated work object. Sleeps when fifo is full.
735 * Monitors work done indicator.
736 * Exits loop when work indiactor is set.
737 * @param _w Handle to work object.
739 static void work_loop( void * _w )
741 hb_work_object_t * w = _w;
742 hb_buffer_t * buf_in, * buf_out;
747 hb_lock( job->pause );
748 hb_unlock( job->pause );
750 if( hb_fifo_is_full( w->fifo_out ) ||
751 // if( (hb_fifo_percent_full( w->fifo_out ) > 0.8) ||
752 !( buf_in = hb_fifo_get( w->fifo_in ) ) )
754 hb_snooze( w->thread_sleep_interval );
755 // w->thread_sleep_interval += 1;
758 // w->thread_sleep_interval = MAX(1, (w->thread_sleep_interval - 1));
760 w->work( w, &buf_in, &buf_out );
762 // Propogate any chapter breaks for the worker
763 if( buf_in && buf_out && buf_in->new_chap )
765 hb_log("WORK: Copying Chapter Break");
766 buf_out->new_chap = 1;
771 hb_buffer_close( &buf_in );
775 hb_fifo_push( w->fifo_out, buf_out );