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"
20 static void work_func();
21 static void do_job( hb_job_t *, int cpu_count );
22 static void work_loop( void * );
25 * Allocates work object and launches work thread with work_func.
26 * @param jobs Handle to hb_list_t.
27 * @param cpu_count Humber of CPUs found in system.
28 * @param die Handle to user inititated exit indicator.
29 * @param error Handle to error indicator.
31 hb_thread_t * hb_work_init( hb_list_t * jobs, int cpu_count,
32 volatile int * die, int * error )
34 hb_work_t * work = calloc( sizeof( hb_work_t ), 1 );
37 work->cpu_count = cpu_count;
41 return hb_thread_init( "work", work_func, work, HB_LOW_PRIORITY );
45 * Iterates through job list and calls do_job for each job.
46 * @param _work Handle work object.
48 static void work_func( void * _work )
50 hb_work_t * work = _work;
53 hb_log( "%d job(s) to process", hb_list_count( work->jobs ) );
55 while( !*work->die && ( job = hb_list_item( work->jobs, 0 ) ) )
57 hb_list_rem( work->jobs, job );
59 do_job( job, work->cpu_count );
62 *(work->error) = HB_ERROR_NONE;
67 static hb_work_object_t * getWork( int id )
70 for( w = hb_objects; w; w = w->next )
81 * Job initialization rountine.
83 * Creates work objects for synchronizer, video decoder, video renderer, video decoder, audio decoder, audio encoder, reader, muxer.
84 * Launches thread for each work object with work_loop.
85 * Loops while monitoring status of work threads and fifos.
86 * Exits loop when conversion is done and fifos are empty.
87 * Closes threads and frees fifos.
88 * @param job Handle work hb_job_t.
89 * @param cpu_count number of CPUs found in system.
91 static void do_job( hb_job_t * job, int cpu_count )
97 /* FIXME: This feels really hackish, anything better? */
98 hb_work_object_t * audio_w = NULL;
101 hb_subtitle_t * subtitle;
106 job->list_work = hb_list_init();
108 hb_log( "starting job" );
109 hb_log( " + device %s", title->dvd );
110 hb_log( " + title %d, chapter(s) %d to %d", title->index,
111 job->chapter_start, job->chapter_end );
112 if ( job->pixel_ratio == 1 )
114 /* Correct the geometry of the output movie when using PixelRatio */
115 job->height=title->height-job->crop[0]-job->crop[1];
116 job->width=title->width-job->crop[2]-job->crop[3];
119 /* Keep width and height within these boundaries */
120 if (job->maxHeight && (job->height > job->maxHeight) )
122 job->height = job->maxHeight;
123 hb_fix_aspect( job, HB_KEEP_HEIGHT );
124 hb_log("Height out of bounds, scaling down to %i", job->maxHeight);
125 hb_log("New dimensions %i * %i", job->width, job->height);
127 if (job->maxWidth && (job->width > job->maxWidth) )
129 job->width = job->maxWidth;
130 hb_fix_aspect( job, HB_KEEP_WIDTH );
131 hb_log("Width out of bounds, scaling down to %i", job->maxWidth);
132 hb_log("New dimensions %i * %i", job->width, job->height);
135 hb_log( " + %dx%d -> %dx%d, crop %d/%d/%d/%d",
136 title->width, title->height, job->width, job->height,
137 job->crop[0], job->crop[1], job->crop[2], job->crop[3] );
138 hb_log( " + deinterlace %s", job->deinterlace ? "on" : "off" );
139 hb_log( " + grayscale %s", job->grayscale ? "on" : "off" );
140 if( job->vquality >= 0.0 && job->vquality <= 1.0 )
142 hb_log( " + %.3f fps, video quality %.2f", (float) job->vrate /
143 (float) job->vrate_base, job->vquality );
147 hb_log( " + %.3f fps, video bitrate %d kbps, pass %d",
148 (float) job->vrate / (float) job->vrate_base,
149 job->vbitrate, job->pass );
151 hb_log (" + PixelRatio: %d, width:%d, height: %d",job->pixel_ratio,job->width, job->height);
152 job->fifo_mpeg2 = hb_fifo_init( 2048 );
153 job->fifo_raw = hb_fifo_init( 8 );
154 job->fifo_sync = hb_fifo_init( 8 );
155 job->fifo_render = hb_fifo_init( 8 );
156 job->fifo_mpeg4 = hb_fifo_init( 8 );
158 /* Synchronization */
159 hb_list_add( job->list_work, ( w = getWork( WORK_SYNC ) ) );
164 hb_list_add( job->list_work, ( w = getWork( WORK_DECMPEG2 ) ) );
165 w->fifo_in = job->fifo_mpeg2;
166 w->fifo_out = job->fifo_raw;
169 hb_list_add( job->list_work, ( w = getWork( WORK_RENDER ) ) );
170 w->fifo_in = job->fifo_sync;
171 w->fifo_out = job->fifo_render;
174 switch( job->vcodec )
176 case HB_VCODEC_FFMPEG:
177 hb_log( " + encoder FFmpeg" );
178 w = getWork( WORK_ENCAVCODEC );
181 hb_log( " + encoder XviD" );
182 w = getWork( WORK_ENCXVID );
185 hb_log( " + encoder x264" );
186 w = getWork( WORK_ENCX264 );
189 w->fifo_in = job->fifo_render;
190 w->fifo_out = job->fifo_mpeg4;
191 w->config = &job->config;
192 hb_list_add( job->list_work, w );
194 subtitle = hb_list_item( title->list_subtitle, 0 );
197 hb_log( " + subtitle %x, %s", subtitle->id, subtitle->lang );
199 subtitle->fifo_in = hb_fifo_init( 8 );
200 subtitle->fifo_raw = hb_fifo_init( 8 );
202 hb_list_add( job->list_work, ( w = getWork( WORK_DECSUB ) ) );
203 w->fifo_in = subtitle->fifo_in;
204 w->fifo_out = subtitle->fifo_raw;
207 if( job->acodec & HB_ACODEC_AC3 )
209 hb_log( " + audio AC3 passthrough" );
213 hb_log( " + audio %d kbps, %d Hz", job->abitrate, job->arate );
214 hb_log( " + encoder %s", ( job->acodec & HB_ACODEC_FAAC ) ?
215 "faac" : ( ( job->acodec & HB_ACODEC_LAME ) ? "lame" :
219 for( i = 0; i < hb_list_count( title->list_audio ); i++ )
221 audio = hb_list_item( title->list_audio, i );
222 hb_log( " + %x, %s", audio->id, audio->lang );
224 /* sense-check the current mixdown options */
226 /* log the requested mixdown */
227 for (j = 0; j < hb_audio_mixdowns_count; j++) {
228 if (hb_audio_mixdowns[j].amixdown == job->audio_mixdowns[i]) {
229 hb_log( " + Requested mixdown: %s (%s)", hb_audio_mixdowns[j].human_readable_name, hb_audio_mixdowns[j].internal_name );
234 if (audio->codec == HB_ACODEC_AC3)
237 /* sense-check the AC3 mixdown */
239 /* audioCodecSupportsMono and audioCodecSupports6Ch are the same for now,
240 but this may change in the future, so they are separated for flexibility */
241 int audioCodecSupportsMono = (job->acodec == HB_ACODEC_FAAC);
242 int audioCodecSupports6Ch = (job->acodec == HB_ACODEC_FAAC);
244 /* find out what the format of our source AC3 audio is */
245 switch (audio->config.a52.ac3flags & A52_CHANNEL_MASK) {
251 /* regardless of what stereo mixdown we've requested, a mono source always get mixed down
252 to mono if we can, and mixed up to stereo if we can't */
253 if (job->audio_mixdowns[i] == HB_AMIXDOWN_MONO && audioCodecSupportsMono == 1) {
254 job->audio_mixdowns[i] = HB_AMIXDOWN_MONO;
256 job->audio_mixdowns[i] = HB_AMIXDOWN_STEREO;
263 /* if we've requested a mono mixdown, and it is supported, then do the mix */
264 /* use stereo if not supported */
265 if (job->audio_mixdowns[i] == HB_AMIXDOWN_MONO && audioCodecSupportsMono == 0) {
266 job->audio_mixdowns[i] = HB_AMIXDOWN_STEREO;
267 /* otherwise, preserve stereo regardless of if we requested something higher */
268 } else if (job->audio_mixdowns[i] > HB_AMIXDOWN_STEREO) {
269 job->audio_mixdowns[i] = HB_AMIXDOWN_STEREO;
273 /* dolby (DPL1 aka Dolby Surround = 4.0 matrix-encoded) input */
274 /* the A52 flags don't allow for a way to distinguish between DPL1 and DPL2 on a DVD,
275 so we always assume a DPL1 source for A52_DOLBY */
277 /* if we've requested a mono mixdown, and it is supported, then do the mix */
278 /* preserve dolby if not supported */
279 if (job->audio_mixdowns[i] == HB_AMIXDOWN_MONO && audioCodecSupportsMono == 0) {
280 job->audio_mixdowns[i] = HB_AMIXDOWN_DOLBY;
281 /* otherwise, preserve dolby even if we requested something higher */
282 /* a stereo mixdown will still be honoured here */
283 } else if (job->audio_mixdowns[i] > HB_AMIXDOWN_DOLBY) {
284 job->audio_mixdowns[i] = HB_AMIXDOWN_DOLBY;
290 /* if we've requested a mono mixdown, and it is supported, then do the mix */
291 /* use dpl2 if not supported */
292 if (job->audio_mixdowns[i] == HB_AMIXDOWN_MONO && audioCodecSupportsMono == 0) {
293 job->audio_mixdowns[i] = HB_AMIXDOWN_DOLBYPLII;
295 /* check if we have 3F2R input and also have an LFE - i.e. we have a 5.1 source) */
296 if (audio->config.a52.ac3flags & A52_LFE) {
297 /* we have a 5.1 source */
298 /* if we requested 6ch, but our audio format doesn't support it, then mix to DPLII instead */
299 if (job->audio_mixdowns[i] == HB_AMIXDOWN_6CH && audioCodecSupports6Ch == 0) {
300 job->audio_mixdowns[i] = HB_AMIXDOWN_DOLBYPLII;
303 /* we have a 5.0 source, so we can't do 6ch conversion
304 default to DPL II instead */
305 if (job->audio_mixdowns[i] > HB_AMIXDOWN_DOLBYPLII) {
306 job->audio_mixdowns[i] = HB_AMIXDOWN_DOLBYPLII;
310 /* all other mixdowns will have been preserved here */
315 /* if we've requested a mono mixdown, and it is supported, then do the mix */
316 /* use dpl1 if not supported */
317 if (job->audio_mixdowns[i] == HB_AMIXDOWN_MONO && audioCodecSupportsMono == 0) {
318 job->audio_mixdowns[i] = HB_AMIXDOWN_DOLBY;
320 /* we have a 4.0 or 4.1 source, so we can't do DPLII or 6ch conversion
321 default to DPL I instead */
322 if (job->audio_mixdowns[i] > HB_AMIXDOWN_DOLBY) {
323 job->audio_mixdowns[i] = HB_AMIXDOWN_DOLBY;
326 /* all other mixdowns will have been preserved here */
330 /* if we've requested a mono mixdown, and it is supported, then do the mix */
331 if (job->audio_mixdowns[i] == HB_AMIXDOWN_MONO && audioCodecSupportsMono == 1) {
332 job->audio_mixdowns[i] = HB_AMIXDOWN_MONO;
333 /* mix everything else down to stereo */
335 job->audio_mixdowns[i] = HB_AMIXDOWN_STEREO;
341 else if (audio->codec == HB_ACODEC_DCA)
344 /* sense-check the DCA mixdown */
346 /* audioCodecSupportsMono and audioCodecSupports6Ch are the same for now,
347 but this may change in the future, so they are separated for flexibility */
348 int audioCodecSupportsMono = (job->acodec == HB_ACODEC_FAAC);
349 int audioCodecSupports6Ch = (job->acodec == HB_ACODEC_FAAC);
351 /* find out what the format of our source DCA audio is */
352 switch (audio->config.dca.dcaflags & DCA_CHANNEL_MASK) {
356 /* regardless of what stereo mixdown we've requested, a mono source always get mixed down
357 to mono if we can, and mixed up to stereo if we can't */
358 if (job->audio_mixdowns[i] == HB_AMIXDOWN_MONO && audioCodecSupportsMono == 1) {
359 job->audio_mixdowns[i] = HB_AMIXDOWN_MONO;
361 job->audio_mixdowns[i] = HB_AMIXDOWN_STEREO;
368 case DCA_STEREO_SUMDIFF:
369 case DCA_STEREO_TOTAL:
370 /* if we've requested a mono mixdown, and it is supported, then do the mix */
371 /* use stereo if not supported */
372 if (job->audio_mixdowns[i] == HB_AMIXDOWN_MONO && audioCodecSupportsMono == 0) {
373 job->audio_mixdowns[i] = HB_AMIXDOWN_STEREO;
374 /* otherwise, preserve stereo regardless of if we requested something higher */
375 } else if (job->audio_mixdowns[i] > HB_AMIXDOWN_STEREO) {
376 job->audio_mixdowns[i] = HB_AMIXDOWN_STEREO;
380 /* dolby (DPL1 aka Dolby Surround = 4.0 matrix-encoded) input */
381 /* the A52 flags don't allow for a way to distinguish between DPL1 and DPL2 on a DVD,
382 so we always assume a DPL1 source for A52_DOLBY */
384 /* if we've requested a mono mixdown, and it is supported, then do the mix */
385 /* preserve dolby if not supported */
386 if (job->audio_mixdowns[i] == HB_AMIXDOWN_MONO && audioCodecSupportsMono == 0) {
387 job->audio_mixdowns[i] = HB_AMIXDOWN_DOLBY;
388 /* otherwise, preserve dolby even if we requested something higher */
389 /* a stereo mixdown will still be honoured here */
390 } else if (job->audio_mixdowns[i] > HB_AMIXDOWN_DOLBY) {
391 job->audio_mixdowns[i] = HB_AMIXDOWN_DOLBY;
397 /* if we've requested a mono mixdown, and it is supported, then do the mix */
398 /* use dpl2 if not supported */
399 if (job->audio_mixdowns[i] == HB_AMIXDOWN_MONO && audioCodecSupportsMono == 0) {
400 job->audio_mixdowns[i] = HB_AMIXDOWN_DOLBYPLII;
402 /* check if we have 3F2R input and also have an LFE - i.e. we have a 5.1 source) */
403 if (audio->config.dca.dcaflags & DCA_LFE) {
404 /* we have a 5.1 source */
405 /* if we requested 6ch, but our audio format doesn't support it, then mix to DPLII instead */
406 if (job->audio_mixdowns[i] == HB_AMIXDOWN_6CH && audioCodecSupports6Ch == 0) {
407 job->audio_mixdowns[i] = HB_AMIXDOWN_DOLBYPLII;
410 /* we have a 5.0 source, so we can't do 6ch conversion
411 default to DPL II instead */
412 if (job->audio_mixdowns[i] > HB_AMIXDOWN_DOLBYPLII) {
413 job->audio_mixdowns[i] = HB_AMIXDOWN_DOLBYPLII;
417 /* all other mixdowns will have been preserved here */
422 /* if we've requested a mono mixdown, and it is supported, then do the mix */
423 /* use dpl1 if not supported */
424 if (job->audio_mixdowns[i] == HB_AMIXDOWN_MONO && audioCodecSupportsMono == 0) {
425 job->audio_mixdowns[i] = HB_AMIXDOWN_DOLBY;
427 /* we have a 4.0 or 4.1 source, so we can't do DPLII or 6ch conversion
428 default to DPL I instead */
429 if (job->audio_mixdowns[i] > HB_AMIXDOWN_DOLBY) {
430 job->audio_mixdowns[i] = HB_AMIXDOWN_DOLBY;
433 /* all other mixdowns will have been preserved here */
437 /* if we've requested a mono mixdown, and it is supported, then do the mix */
438 if (job->audio_mixdowns[i] == HB_AMIXDOWN_MONO && audioCodecSupportsMono == 1) {
439 job->audio_mixdowns[i] = HB_AMIXDOWN_MONO;
440 /* mix everything else down to stereo */
442 job->audio_mixdowns[i] = HB_AMIXDOWN_STEREO;
450 /* assume a stereo input and output for non-AC3/DCA audio input (LPCM, MP2),
451 regardless of the mixdown passed to us */
452 job->audio_mixdowns[i] = HB_AMIXDOWN_STEREO;
456 /* log the output mixdown */
457 for (j = 0; j < hb_audio_mixdowns_count; j++) {
458 if (hb_audio_mixdowns[j].amixdown == job->audio_mixdowns[i]) {
459 hb_log( " + Actual mixdown: %s (%s)", hb_audio_mixdowns[j].human_readable_name, hb_audio_mixdowns[j].internal_name );
464 /* we now know we have a valid mixdown for the input source and the audio output format */
465 /* remember the mixdown for this track */
466 audio->amixdown = job->audio_mixdowns[i];
468 audio->config.vorbis.language = audio->lang_simple;
470 /* set up the audio work structures */
471 audio->fifo_in = hb_fifo_init( 2048 );
472 audio->fifo_raw = hb_fifo_init( 8 );
473 audio->fifo_sync = hb_fifo_init( 8 );
474 audio->fifo_out = hb_fifo_init( 8 );
476 switch( audio->codec )
479 w = getWork( WORK_DECA52 );
482 w = getWork( WORK_DECDCA );
485 w = getWork( WORK_DECAVCODEC );
488 w = getWork( WORK_DECLPCM );
491 w->fifo_in = audio->fifo_in;
492 w->fifo_out = audio->fifo_raw;
493 w->config = &audio->config;
494 w->amixdown = audio->amixdown;
496 /* FIXME: This feels really hackish, anything better? */
497 audio_w = calloc( sizeof( hb_work_object_t ), 1 );
498 audio_w = memcpy( audio_w, w, sizeof( hb_work_object_t ));
500 hb_list_add( job->list_work, audio_w );
502 switch( job->acodec )
505 w = getWork( WORK_ENCFAAC );
508 w = getWork( WORK_ENCLAME );
510 case HB_ACODEC_VORBIS:
511 w = getWork( WORK_ENCVORBIS );
515 if( job->acodec != HB_ACODEC_AC3 )
517 w->fifo_in = audio->fifo_sync;
518 w->fifo_out = audio->fifo_out;
519 w->config = &audio->config;
520 w->amixdown = audio->amixdown;
522 /* FIXME: This feels really hackish, anything better? */
523 audio_w = calloc( sizeof( hb_work_object_t ), 1 );
524 audio_w = memcpy( audio_w, w, sizeof( hb_work_object_t ));
526 hb_list_add( job->list_work, audio_w );
532 /* Init read & write threads */
533 job->reader = hb_reader_init( job );
535 hb_log( " + output: %s", job->file );
536 job->muxer = hb_muxer_init( job );
540 /* Launch processing threads */
541 for( i = 1; i < hb_list_count( job->list_work ); i++ )
543 w = hb_list_item( job->list_work, i );
544 w->done = &job->done;
545 w->thread_sleep_interval = 10;
547 w->thread = hb_thread_init( w->name, work_loop, w,
552 w = hb_list_item( job->list_work, 0 );
553 w->thread_sleep_interval = 50;
557 if( w->work( w, NULL, NULL ) == HB_WORK_DONE )
562 !hb_fifo_size( job->fifo_sync ) &&
563 !hb_fifo_size( job->fifo_render ) &&
564 !hb_fifo_size( job->fifo_mpeg4 ) )
568 hb_snooze( w->thread_sleep_interval );
570 hb_list_rem( job->list_work, w );
574 /* Close work objects */
575 while( ( w = hb_list_item( job->list_work, 0 ) ) )
577 hb_list_rem( job->list_work, w );
578 hb_thread_close( &w->thread );
581 /* FIXME: This feels really hackish, anything better? */
582 if ( w->id == WORK_DECA52 ||
583 w->id == WORK_DECDCA ||
584 w->id == WORK_DECLPCM ||
585 w->id == WORK_ENCFAAC ||
586 w->id == WORK_ENCLAME ||
587 w->id == WORK_ENCVORBIS )
594 hb_list_close( &job->list_work );
596 /* Stop read & write threads */
597 hb_thread_close( &job->reader );
598 hb_thread_close( &job->muxer );
601 hb_fifo_close( &job->fifo_mpeg2 );
602 hb_fifo_close( &job->fifo_raw );
603 hb_fifo_close( &job->fifo_sync );
604 hb_fifo_close( &job->fifo_render );
605 hb_fifo_close( &job->fifo_mpeg4 );
608 hb_fifo_close( &subtitle->fifo_in );
609 hb_fifo_close( &subtitle->fifo_raw );
611 for( i = 0; i < hb_list_count( title->list_audio ); i++ )
613 audio = hb_list_item( title->list_audio, i );
614 hb_fifo_close( &audio->fifo_in );
615 hb_fifo_close( &audio->fifo_raw );
616 hb_fifo_close( &audio->fifo_sync );
617 hb_fifo_close( &audio->fifo_out );
620 hb_title_close( &job->title );
625 * Performs the work objects specific work function.
626 * Loops calling work function for associated work object. Sleeps when fifo is full.
627 * Monitors work done indicator.
628 * Exits loop when work indiactor is set.
629 * @param _w Handle to work object.
631 static void work_loop( void * _w )
633 hb_work_object_t * w = _w;
634 hb_buffer_t * buf_in, * buf_out;
639 hb_lock( job->pause );
640 hb_unlock( job->pause );
642 if( hb_fifo_is_full( w->fifo_out ) ||
643 // if( (hb_fifo_percent_full( w->fifo_out ) > 0.8) ||
644 !( buf_in = hb_fifo_get( w->fifo_in ) ) )
646 hb_snooze( w->thread_sleep_interval );
647 // w->thread_sleep_interval += 1;
650 // w->thread_sleep_interval = MAX(1, (w->thread_sleep_interval - 1));
652 w->work( w, &buf_in, &buf_out );
654 // Propogate any chapter breaks for the worker
655 if( buf_in && buf_out && buf_in->new_chap )
657 printf("WORK: Copying Chapter Break\n");
658 buf_out->new_chap = 1;
663 hb_buffer_close( &buf_in );
667 hb_fifo_push( w->fifo_out, buf_out );