OSDN Git Service

More code to deal with the flakey streams from NZ TV. They like to change the PCR...
[handbrake-jp/handbrake-jp-git.git] / 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.fr/>.
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 #include "libavformat/avformat.h"
11
12 typedef struct
13 {
14     hb_list_t * jobs;
15     hb_job_t  ** current_job;
16     int         cpu_count;
17     int       * error;
18     volatile int * die;
19
20 } hb_work_t;
21
22 static void work_func();
23 static void do_job( hb_job_t *, int cpu_count );
24 static void work_loop( void * );
25
26 #define FIFO_CPU_MULT 8
27
28 /**
29  * Allocates work object and launches work thread with work_func.
30  * @param jobs Handle to hb_list_t.
31  * @param cpu_count Humber of CPUs found in system.
32  * @param die Handle to user inititated exit indicator.
33  * @param error Handle to error indicator.
34  */
35 hb_thread_t * hb_work_init( hb_list_t * jobs, int cpu_count,
36                             volatile int * die, int * error, hb_job_t ** job )
37 {
38     hb_work_t * work = calloc( sizeof( hb_work_t ), 1 );
39
40     work->jobs      = jobs;
41     work->current_job = job;
42     work->cpu_count = cpu_count;
43     work->die       = die;
44     work->error     = error;
45
46     return hb_thread_init( "work", work_func, work, HB_LOW_PRIORITY );
47 }
48
49 /**
50  * Iterates through job list and calls do_job for each job.
51  * @param _work Handle work object.
52  */
53 static void work_func( void * _work )
54 {
55     hb_work_t  * work = _work;
56     hb_job_t   * job;
57
58     hb_log( "%d job(s) to process", hb_list_count( work->jobs ) );
59
60     while( !*work->die && ( job = hb_list_item( work->jobs, 0 ) ) )
61     {
62         hb_list_rem( work->jobs, job );
63         job->die = work->die;
64         *(work->current_job) = job;
65         do_job( job, work->cpu_count );
66         *(work->current_job) = NULL;
67     }
68
69     *(work->error) = HB_ERROR_NONE;
70
71     free( work );
72 }
73
74 hb_work_object_t * hb_get_work( int id )
75 {
76     hb_work_object_t * w;
77     for( w = hb_objects; w; w = w->next )
78     {
79         if( w->id == id )
80         {
81             hb_work_object_t *wc = malloc( sizeof(*w) );
82             *wc = *w;
83             return wc;
84         }
85     }
86     return NULL;
87 }
88
89 hb_work_object_t * hb_codec_decoder( int codec )
90 {
91     switch( codec )
92     {
93         case HB_ACODEC_AC3:  return hb_get_work( WORK_DECA52 );
94         case HB_ACODEC_DCA:  return hb_get_work( WORK_DECDCA );
95         case HB_ACODEC_MPGA: return hb_get_work( WORK_DECAVCODEC );
96         case HB_ACODEC_LPCM: return hb_get_work( WORK_DECLPCM );
97         case HB_ACODEC_FFMPEG: return hb_get_work( WORK_DECAVCODECAI );
98     }
99     return NULL;
100 }
101
102 hb_work_object_t * hb_codec_encoder( int codec )
103 {
104     switch( codec )
105     {
106         case HB_ACODEC_FAAC:   return hb_get_work( WORK_ENCFAAC );
107         case HB_ACODEC_LAME:   return hb_get_work( WORK_ENCLAME );
108         case HB_ACODEC_VORBIS: return hb_get_work( WORK_ENCVORBIS );
109     }
110     return NULL;
111 }
112
113 /**
114  * Displays job parameters in the debug log.
115  * @param job Handle work hb_job_t.
116  */
117 void hb_display_job_info( hb_job_t * job )
118 {
119     hb_title_t * title = job->title;
120     hb_audio_t   * audio;
121     hb_subtitle_t * subtitle;
122     int             i, j;
123     
124     hb_log("job configuration:");
125     hb_log( " * source");
126     
127     hb_log( "   + %s", title->dvd );
128
129     hb_log( "   + title %d, chapter(s) %d to %d", title->index,
130             job->chapter_start, job->chapter_end );
131
132     if( title->container_name != NULL )
133         hb_log( "   + container: %s", title->container_name);
134
135     if( title->data_rate )
136     {
137         hb_log( "   + data rate: %d kbps", title->data_rate / 1000 );
138     }
139     
140     hb_log( " * destination");
141
142     hb_log( "   + %s", job->file );
143
144     switch( job->mux )
145     {
146         case HB_MUX_MP4:
147             hb_log("   + container: MPEG-4 (.mp4 and .m4v)");
148             
149             if( job->ipod_atom )
150                 hb_log( "     + compatibility atom for iPod 5G");
151
152             if( job->largeFileSize )
153                 hb_log( "     + 64-bit formatting");
154
155             if( job->mp4_optimize )
156                 hb_log( "     + optimized for progressive web downloads");
157             
158             if( job->color_matrix )
159                 hb_log( "     + custom color matrix: %s", job->color_matrix == 1 ? "ITU Bt.601 (SD)" : "ITU Bt.709 (HD)");
160             break;
161
162         case HB_MUX_AVI:
163             hb_log("   + container: AVI");
164             break;
165
166         case HB_MUX_MKV:
167             hb_log("   + container: Matroska (.mkv)");
168             break;
169
170         case HB_MUX_OGM:
171             hb_log("   + conttainer: Ogg Media (.ogm)");
172             break;
173     }
174
175     if( job->chapter_markers )
176     {
177         hb_log( "     + chapter markers" );
178     }
179     
180     hb_log(" * video track");
181     
182     hb_log("   + decoder: %s", title->video_codec_name );
183     
184     if( title->video_bitrate )
185     {
186         hb_log( "     + bitrate %d kbps", title->video_bitrate / 1000 );
187     }
188     
189     if( job->vfr)
190     {
191         hb_log( "   + frame rate: %.3f fps -> variable fps",
192             (float) title->rate / (float) title->rate_base );
193     }
194     else if( !job->cfr )
195     {
196         hb_log( "   + frame rate: same as source (around %.3f fps)",
197             (float) title->rate / (float) title->rate_base );
198     }
199     else
200     {
201         hb_log( "   + frame rate: %.3f fps -> constant %.3f fps",
202             (float) title->rate / (float) title->rate_base, (float) job->vrate / (float) job->vrate_base );
203     }
204
205     if( job->pixel_ratio )
206     {
207         hb_log( "   + %s anamorphic", job->pixel_ratio == 1 ? "strict" : "loose" );
208         hb_log( "     + storage dimensions: %d * %d -> %d * %d, crop %d/%d/%d/%d",
209                     title->width, title->height, job->width, job->height,
210                     job->crop[0], job->crop[1], job->crop[2], job->crop[3] );
211         hb_log( "     + pixel aspect ratio: %i / %i", job->pixel_aspect_width, job->pixel_aspect_height );
212         hb_log( "     + display dimensions: %.0f * %i",
213             (float)( job->width * job->pixel_aspect_width / job->pixel_aspect_height ), job->height );
214     }
215     else
216     {
217         hb_log( "   + dimensions: %d * %d -> %d * %d, crop %d/%d/%d/%d",
218                 title->width, title->height, job->width, job->height,
219                 job->crop[0], job->crop[1], job->crop[2], job->crop[3] );
220     }
221
222     if ( job->grayscale )
223         hb_log( "   + grayscale mode" );
224
225     if( hb_list_count( job->filters ) )
226     {
227         hb_log("   + %s", hb_list_count( job->filters) > 1 ? "filters" : "filter" );
228         for( i = 0; i < hb_list_count( job->filters ); i++ )
229         {
230             hb_filter_object_t * filter = hb_list_item( job->filters, i );
231             if (filter->settings)
232                 hb_log("     + %s (%s)", filter->name, filter->settings);
233             else
234                 hb_log("     + %s (default settings)", filter->name);
235         }
236     }
237     
238     if( !job->indepth_scan )
239     {
240         /* Video encoder */
241         switch( job->vcodec )
242         {
243             case HB_VCODEC_FFMPEG:
244                 hb_log( "   + encoder: FFmpeg" );
245                 break;
246
247             case HB_VCODEC_XVID:
248                 hb_log( "   + encoder: XviD" );
249                 break;
250
251             case HB_VCODEC_X264:
252                 hb_log( "   + encoder: x264" );
253                 if( job->x264opts != NULL && *job->x264opts != '\0' )
254                     hb_log( "     + options: %s", job->x264opts);
255                 break;
256
257             case HB_VCODEC_THEORA:
258                 hb_log( "   + encoder: Theora" );
259                 break;
260         }
261
262         if( job->vquality >= 0.0 && job->vquality <= 1.0 )
263         {
264             hb_log( "     + quality: %.2f", job->vquality );
265         }
266         else if( job->vquality > 1 )
267         {
268             hb_log( "     + quality: %.0f %s", job->vquality, job->crf && job->vcodec == HB_VCODEC_X264 ? "(RF)" : "(QP)" ); 
269         }
270         else
271         {
272             hb_log( "     + bitrate: %d kbps, pass: %d", job->vbitrate, job->pass );
273         }
274     }
275
276     for( i=0; i < hb_list_count(title->list_subtitle); i++ )
277     {
278         subtitle =  hb_list_item( title->list_subtitle, i );
279
280         if( subtitle )
281         {
282             hb_log( " * subtitle track %i, %s (id %x)", job->subtitle+1, subtitle->lang, subtitle->id);
283         }
284     }
285
286     if( !job->indepth_scan )
287     {
288         for( i = 0; i < hb_list_count( title->list_audio ); i++ )
289         {
290             audio = hb_list_item( title->list_audio, i );
291
292             hb_log( " * audio track %d", audio->config.out.track );
293             
294             if( audio->config.out.name )
295                 hb_log( "   + name: %s", audio->config.out.name );
296
297             hb_log( "   + decoder: %s (track %d, id %x)", audio->config.lang.description, audio->config.in.track + 1, audio->id );
298
299             if( ( audio->config.in.codec == HB_ACODEC_AC3 ) || ( audio->config.in.codec == HB_ACODEC_DCA) )
300             {
301                 hb_log( "     + bitrate: %d kbps, samplerate: %d Hz", audio->config.in.bitrate / 1000, audio->config.in.samplerate );
302             }
303
304             if( (audio->config.out.codec != HB_ACODEC_AC3) && (audio->config.out.codec != HB_ACODEC_DCA) )
305             {
306                 for (j = 0; j < hb_audio_mixdowns_count; j++)
307                 {
308                     if (hb_audio_mixdowns[j].amixdown == audio->config.out.mixdown) {
309                         hb_log( "   + mixdown: %s", hb_audio_mixdowns[j].human_readable_name );
310                         break;
311                     }
312                 }
313             }
314
315             if ( audio->config.out.dynamic_range_compression > 1 && (audio->config.out.codec != HB_ACODEC_AC3) && (audio->config.out.codec != HB_ACODEC_DCA))
316             {
317                 hb_log("   + dynamic range compression: %f", audio->config.out.dynamic_range_compression);
318             }
319             
320             if( (audio->config.out.codec == HB_ACODEC_AC3) || (audio->config.out.codec == HB_ACODEC_DCA) )
321             {
322                 hb_log( "   + %s passthrough", (audio->config.out.codec == HB_ACODEC_AC3) ?
323                     "AC3" : "DCA" );
324             }
325             else
326             {
327                 hb_log( "   + encoder: %s", ( audio->config.out.codec == HB_ACODEC_FAAC ) ?
328                     "faac" : ( ( audio->config.out.codec == HB_ACODEC_LAME ) ? "lame" :
329                     "vorbis" ) );
330                 hb_log( "     + bitrate: %d kbps, samplerate: %d Hz", audio->config.out.bitrate, audio->config.out.samplerate );            
331             }
332         }
333     }
334 }
335
336 /**
337  * Job initialization rountine.
338  * Initializes fifos.
339  * Creates work objects for synchronizer, video decoder, video renderer, video decoder, audio decoder, audio encoder, reader, muxer.
340  * Launches thread for each work object with work_loop.
341  * Loops while monitoring status of work threads and fifos.
342  * Exits loop when conversion is done and fifos are empty.
343  * Closes threads and frees fifos.
344  * @param job Handle work hb_job_t.
345  * @param cpu_count number of CPUs found in system.
346  */
347 static void do_job( hb_job_t * job, int cpu_count )
348 {
349     hb_title_t    * title;
350     int             i, j;
351     hb_work_object_t * w;
352
353     hb_audio_t   * audio;
354     hb_subtitle_t * subtitle;
355     unsigned int subtitle_highest = 0;
356     unsigned int subtitle_highest_id = 0;
357     unsigned int subtitle_lowest = -1;
358     unsigned int subtitle_lowest_id = 0;
359     unsigned int subtitle_forced_id = 0;
360     unsigned int subtitle_hit = 0;
361
362     title = job->title;
363
364     job->list_work = hb_list_init();
365
366     hb_log( "starting job" );
367
368     if ( job->pixel_ratio == 1 )
369     {
370         /* Correct the geometry of the output movie when using PixelRatio */
371         job->height=title->height-job->crop[0]-job->crop[1];
372         job->width=title->width-job->crop[2]-job->crop[3];
373     }
374     else if ( job->pixel_ratio == 2 )
375     {
376
377         /* While keeping the DVD storage aspect, resize the job width and height
378            so they fit into the user's specified dimensions. */
379         hb_set_anamorphic_size(job, &job->width, &job->height, &job->pixel_aspect_width, &job->pixel_aspect_height);
380     }
381
382     if( job->pixel_ratio && job->vcodec == HB_VCODEC_FFMPEG)
383     {
384         /* Just to make working with ffmpeg even more fun,
385            lavc's MPEG-4 encoder can't handle PAR values >= 255,
386            even though AVRational does. Adjusting downwards
387            distorts the display aspect slightly, but such is life. */
388         while ((job->pixel_aspect_width & ~0xFF) ||
389                (job->pixel_aspect_height & ~0xFF))
390         {
391             job->pixel_aspect_width >>= 1;
392             job->pixel_aspect_height >>= 1;
393         }
394     }
395
396     /* Keep width and height within these boundaries,
397        but ignore for anamorphic. For "loose" anamorphic encodes,
398        this stuff is covered in the pixel_ratio section above.    */
399     if ( job->maxHeight && ( job->height > job->maxHeight ) && ( !job->pixel_ratio ) )
400     {
401         job->height = job->maxHeight;
402         hb_fix_aspect( job, HB_KEEP_HEIGHT );
403         hb_log( "Height out of bounds, scaling down to %i", job->maxHeight );
404         hb_log( "New dimensions %i * %i", job->width, job->height );
405     }
406     if ( job->maxWidth && ( job->width > job->maxWidth ) && ( !job->pixel_ratio ) )
407     {
408         job->width = job->maxWidth;
409         hb_fix_aspect( job, HB_KEEP_WIDTH );
410         hb_log( "Width out of bounds, scaling down to %i", job->maxWidth );
411         hb_log( "New dimensions %i * %i", job->width, job->height );
412     }
413
414     if( ( job->mux & HB_MUX_AVI ) || job->cfr )
415     {
416         /* VFR detelecine is not compatible with AVI or constant frame rates. */
417         job->vfr = 0;
418     }
419
420     if ( job->vfr )
421     {
422         /* Ensure we're using "Same as source" FPS,
423            aka VFR, if we're doing VFR detelecine. */
424         job->vrate_base = title->rate_base;
425     }
426
427     job->fifo_mpeg2  = hb_fifo_init( 256 );
428     job->fifo_raw    = hb_fifo_init( FIFO_CPU_MULT * cpu_count );
429     job->fifo_sync   = hb_fifo_init( FIFO_CPU_MULT * cpu_count );
430     job->fifo_render = hb_fifo_init( FIFO_CPU_MULT * cpu_count );
431     job->fifo_mpeg4  = hb_fifo_init( FIFO_CPU_MULT * cpu_count );
432
433     /* Synchronization */
434     hb_list_add( job->list_work, ( w = hb_get_work( WORK_SYNC ) ) );
435     w->fifo_in  = NULL;
436     w->fifo_out = NULL;
437
438     /* Video decoder */
439     int vcodec = title->video_codec? title->video_codec : WORK_DECMPEG2;
440     hb_list_add( job->list_work, ( w = hb_get_work( vcodec ) ) );
441     w->codec_param = title->video_codec_param;
442     w->fifo_in  = job->fifo_mpeg2;
443     w->fifo_out = job->fifo_raw;
444
445     /* Video renderer */
446     hb_list_add( job->list_work, ( w = hb_get_work( WORK_RENDER ) ) );
447     w->fifo_in  = job->fifo_sync;
448     w->fifo_out = job->fifo_render;
449
450     if( !job->indepth_scan )
451     {
452
453         /* Video encoder */
454         switch( job->vcodec )
455         {
456         case HB_VCODEC_FFMPEG:
457             w = hb_get_work( WORK_ENCAVCODEC );
458             break;
459         case HB_VCODEC_XVID:
460             w = hb_get_work( WORK_ENCXVID );
461             break;
462         case HB_VCODEC_X264:
463             w = hb_get_work( WORK_ENCX264 );
464             break;
465         case HB_VCODEC_THEORA:
466             w = hb_get_work( WORK_ENCTHEORA );
467             break;
468         }
469         w->fifo_in  = job->fifo_render;
470         w->fifo_out = job->fifo_mpeg4;
471         w->config   = &job->config;
472
473         hb_list_add( job->list_work, w );
474     }
475
476     if( job->select_subtitle && !job->indepth_scan )
477     {
478         /*
479          * Must be second pass of a two pass with subtitle scan enabled, so
480          * add the subtitle that we found on the first pass for use in this
481          * pass.
482          */
483         if (*(job->select_subtitle))
484         {
485             hb_list_add( title->list_subtitle, *( job->select_subtitle ) );
486         }
487     }
488
489     for( i=0; i < hb_list_count(title->list_subtitle); i++ )
490     {
491         subtitle =  hb_list_item( title->list_subtitle, i );
492
493         if( subtitle )
494         {
495             subtitle->fifo_in  = hb_fifo_init( FIFO_CPU_MULT * cpu_count );
496             subtitle->fifo_raw = hb_fifo_init( FIFO_CPU_MULT * cpu_count );
497
498             /*
499              * Disable forced subtitles if we didn't find any in the scan
500              * so that we display normal subtitles instead.
501              *
502              * select_subtitle implies that we did a scan.
503              */
504             if( !job->indepth_scan && job->subtitle_force &&
505                 job->select_subtitle )
506             {
507                 if( subtitle->forced_hits == 0 )
508                 {
509                     job->subtitle_force = 0;
510                 }
511             }
512
513             if (!job->indepth_scan || job->subtitle_force) {
514                 /*
515                  * Don't add threads for subtitles when we are scanning, unless
516                  * looking for forced subtitles.
517                  */
518                 w = hb_get_work( WORK_DECSUB );
519                 w->fifo_in  = subtitle->fifo_in;
520                 w->fifo_out = subtitle->fifo_raw;
521                 hb_list_add( job->list_work, w );
522             }
523         }
524     }
525
526     if( !job->indepth_scan )
527     {
528     // if we are doing passthru, and the input codec is not the same as the output
529     // codec, then remove this audio from the job. If we're not doing passthru and
530     // the input codec is the 'internal' ffmpeg codec, make sure that only one
531     // audio references that audio stream since the codec context is specific to
532     // the audio id & multiple copies of the same stream will garble the audio
533     // or cause aborts.
534     uint8_t aud_id_uses[MAX_STREAMS];
535     memset( aud_id_uses, 0, sizeof(aud_id_uses) );
536     for( i = 0; i < hb_list_count( title->list_audio ); )
537     {
538         audio = hb_list_item( title->list_audio, i );
539         if( ( ( audio->config.out.codec == HB_ACODEC_AC3 ) && ( audio->config.in.codec != HB_ACODEC_AC3 ) ) ||
540             ( ( audio->config.out.codec == HB_ACODEC_DCA ) && ( audio->config.in.codec != HB_ACODEC_DCA ) ) )
541         {
542             hb_log( "Passthru requested and input codec is not the same as output codec for track %d",
543                     audio->config.out.track );
544             hb_list_rem( title->list_audio, audio );
545             free( audio );
546             continue;
547         }
548         if ( audio->config.in.codec == HB_ACODEC_FFMPEG )
549         {
550             if ( aud_id_uses[audio->id] )
551             {
552                 hb_log( "Multiple decodes of audio id %d, removing track %d",
553                         audio->id, audio->config.out.track );
554                 hb_list_rem( title->list_audio, audio );
555                 free( audio );
556                 continue;
557             }
558             ++aud_id_uses[audio->id];
559         }
560         /* Adjust output track number, in case we removed one.
561          * Output tracks sadly still need to be in sequential order.
562          */
563         audio->config.out.track = i++;
564     }
565
566     int requested_mixdown = 0;
567     int requested_mixdown_index = 0;
568
569     for( i = 0; i < hb_list_count( title->list_audio ); i++ )
570     {
571         audio = hb_list_item( title->list_audio, i );
572
573         if( audio->config.out.codec != audio->config.in.codec )
574         {
575             /* sense-check the current mixdown options */
576
577             /* log the requested mixdown */
578             for (j = 0; j < hb_audio_mixdowns_count; j++) {
579                 if (hb_audio_mixdowns[j].amixdown == audio->config.out.mixdown) {
580                     requested_mixdown = audio->config.out.mixdown;
581                     requested_mixdown_index = j;
582                 }
583             }
584
585             /* sense-check the requested mixdown */
586
587             if( audio->config.out.mixdown == 0 &&
588                 audio->config.out.codec != HB_ACODEC_AC3 )
589             {
590                 /*
591                  * Mixdown wasn't specified and this is not pass-through,
592                  * set a default mixdown of stereo.
593                  */
594                 audio->config.out.mixdown = HB_AMIXDOWN_STEREO;
595             }
596
597             // Here we try to sanitize the audio input to output mapping.
598             // Constraints are:
599             //   1. only the AC3 & DCA decoder libraries currently support mixdown
600             //   2. the lame encoder library only supports stereo.
601             // So if the encoder is lame we need the output to be stereo (or multichannel
602             // matrixed into stereo like dpl). If the decoder is not AC3 or DCA the
603             // encoder has to handle the input format since we can't do a mixdown.
604 #define CAN_MIXDOWN(a) ( a->config.in.codec & (HB_ACODEC_AC3|HB_ACODEC_DCA) )
605 #define STEREO_ONLY(a) ( a->config.out.codec & HB_ACODEC_LAME )
606
607             switch (audio->config.in.channel_layout & HB_INPUT_CH_LAYOUT_DISCRETE_NO_LFE_MASK)
608             {
609                 // stereo input or something not handled below
610                 default:
611                 case HB_INPUT_CH_LAYOUT_STEREO:
612                     // mono gets mixed up to stereo & more than stereo gets mixed down
613                     if ( STEREO_ONLY( audio ) ||
614                          audio->config.out.mixdown > HB_AMIXDOWN_STEREO)
615                     {
616                         audio->config.out.mixdown = HB_AMIXDOWN_STEREO;
617                     }
618                     break;
619
620                 // mono input
621                 case HB_INPUT_CH_LAYOUT_MONO:
622                     if ( STEREO_ONLY( audio ) )
623                     {
624                         if ( !CAN_MIXDOWN( audio ) )
625                         {
626                             // XXX we're hosed - we can't mix up & lame can't handle
627                             // the input format. The user shouldn't be able to make
628                             // this choice. It's too late to do anything about it now
629                             // so complain in the log & let things abort in lame.
630                             hb_log( "ERROR - can't use lame mp3 audio output with "
631                                     "mono audio stream %x - output will be messed up",
632                                     audio->id );
633                         }
634                         audio->config.out.mixdown = HB_AMIXDOWN_STEREO;
635                     }
636                     else
637                     {
638                         // everything else passes through
639                         audio->config.out.mixdown = HB_AMIXDOWN_MONO;
640                     }
641                     break;
642
643                 // dolby (DPL1 aka Dolby Surround = 4.0 matrix-encoded) input
644                 // the A52 flags don't allow for a way to distinguish between DPL1 and
645                 // DPL2 on a DVD so we always assume a DPL1 source for A52_DOLBY.
646                 case HB_INPUT_CH_LAYOUT_DOLBY:
647                     if ( STEREO_ONLY( audio ) || !CAN_MIXDOWN( audio ) ||
648                          audio->config.out.mixdown > HB_AMIXDOWN_DOLBY )
649                     {
650                         audio->config.out.mixdown = HB_AMIXDOWN_DOLBY;
651                     }
652                     break;
653
654                 // 4 channel discrete
655                 case HB_INPUT_CH_LAYOUT_2F2R:
656                 case HB_INPUT_CH_LAYOUT_3F1R:
657                     if ( CAN_MIXDOWN( audio ) )
658                     {
659                         if ( STEREO_ONLY( audio ) ||
660                              audio->config.out.mixdown > HB_AMIXDOWN_DOLBY )
661                         {
662                             audio->config.out.mixdown = HB_AMIXDOWN_DOLBY;
663                         }
664                     }
665                     else
666                     {
667                         // XXX we can't mixdown & don't have any way to specify
668                         // 4 channel discrete output so we're hosed.
669                         hb_log( "ERROR - can't handle 4 channel discrete audio stream "
670                                 "%x - output will be messed up", audio->id );
671                     }
672                     break;
673
674                 // 5 or 6 channel discrete
675                 case HB_INPUT_CH_LAYOUT_3F2R:
676                     if ( CAN_MIXDOWN( audio ) )
677                     {
678                         if ( STEREO_ONLY( audio ) )
679                         {
680                             if ( audio->config.out.mixdown < HB_AMIXDOWN_STEREO )
681                             {
682                                 audio->config.out.mixdown = HB_AMIXDOWN_STEREO;
683                             }
684                             else if ( audio->config.out.mixdown > HB_AMIXDOWN_DOLBYPLII )
685                             {
686                                 audio->config.out.mixdown = HB_AMIXDOWN_DOLBYPLII;
687                             }
688                         }
689                         else if ( ! ( audio->config.in.channel_layout &
690                                         HB_INPUT_CH_LAYOUT_HAS_LFE ) )
691                         {
692                             // we don't do 5 channel discrete so mixdown to DPLII
693                             audio->config.out.mixdown = HB_AMIXDOWN_DOLBYPLII;
694                         }
695                     }
696                     else if ( ! ( audio->config.in.channel_layout &
697                                         HB_INPUT_CH_LAYOUT_HAS_LFE ) )
698                     {
699                         // XXX we can't mixdown & don't have any way to specify
700                         // 5 channel discrete output so we're hosed.
701                         hb_log( "ERROR - can't handle 5 channel discrete audio stream "
702                                 "%x - output will be messed up", audio->id );
703                     }
704                     else
705                     {
706                         // we can't mixdown so force 6 channel discrete
707                         audio->config.out.mixdown = HB_AMIXDOWN_6CH;
708                     }
709                     break;
710             }
711
712             /* log the output mixdown */
713             for (j = 0; j < hb_audio_mixdowns_count; j++) {
714                 if (hb_audio_mixdowns[j].amixdown == audio->config.out.mixdown) {
715                     if ( audio->config.out.mixdown != requested_mixdown )
716                     {
717                         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);
718                     }
719                     break;
720                 }
721             }
722         }
723
724         if (audio->config.out.codec == HB_ACODEC_VORBIS)
725             audio->priv.config.vorbis.language = audio->config.lang.simple;
726
727         /* set up the audio work structures */
728         audio->priv.fifo_in   = hb_fifo_init( 32 );
729         audio->priv.fifo_raw  = hb_fifo_init( FIFO_CPU_MULT * cpu_count );
730         audio->priv.fifo_sync = hb_fifo_init( 32 );
731         audio->priv.fifo_out  = hb_fifo_init( FIFO_CPU_MULT * cpu_count );
732
733
734         /*
735          * Audio Decoder Thread
736          */
737         if ( ( w = hb_codec_decoder( audio->config.in.codec ) ) == NULL )
738         {
739             hb_error("Invalid input codec: %d", audio->config.in.codec);
740             *job->die = 1;
741             goto cleanup;
742         }
743         w->fifo_in  = audio->priv.fifo_in;
744         w->fifo_out = audio->priv.fifo_raw;
745         w->config   = &audio->priv.config;
746         w->audio    = audio;
747         w->codec_param = audio->config.in.codec_param;
748
749         hb_list_add( job->list_work, w );
750
751         /*
752          * Audio Encoder Thread
753          */
754         if( audio->config.out.codec != HB_ACODEC_AC3 )
755         {
756             /*
757              * Add the encoder thread if not doing AC-3 pass through
758              */
759             if ( ( w = hb_codec_encoder( audio->config.out.codec ) ) == NULL )
760             {
761                 hb_error("Invalid audio codec: %#x", audio->config.out.codec);
762                 w = NULL;
763                 *job->die = 1;
764                 goto cleanup;
765             }
766             w->fifo_in  = audio->priv.fifo_sync;
767             w->fifo_out = audio->priv.fifo_out;
768             w->config   = &audio->priv.config;
769             w->audio    = audio;
770
771             hb_list_add( job->list_work, w );
772         }
773     }
774
775     }
776
777     /* Display settings */
778     hb_display_job_info( job );
779
780     /* Init read & write threads */
781     job->reader = hb_reader_init( job );
782
783
784     job->done = 0;
785
786     /* Launch processing threads */
787     for( i = 1; i < hb_list_count( job->list_work ); i++ )
788     {
789         w = hb_list_item( job->list_work, i );
790         w->done = &job->done;
791         w->thread_sleep_interval = 10;
792         if( w->init( w, job ) )
793         {
794             hb_error( "Failure to initialise thread '%s'", w->name );
795             *job->die = 1;
796             goto cleanup;
797         }
798         w->thread = hb_thread_init( w->name, work_loop, w,
799                                     HB_LOW_PRIORITY );
800     }
801
802     // The muxer requires track information that's set up by the encoder
803     // init routines so we have to init the muxer last.
804     job->muxer = job->indepth_scan? NULL : hb_muxer_init( job );
805
806     w = hb_list_item( job->list_work, 0 );
807     w->thread_sleep_interval = 10;
808     w->init( w, job );
809     while( !*job->die )
810     {
811         if ( ( w->status = w->work( w, NULL, NULL ) ) == HB_WORK_DONE )
812         {
813             break;
814         }
815         hb_snooze( w->thread_sleep_interval );
816     }
817     hb_list_rem( job->list_work, w );
818     w->close( w );
819     free( w );
820
821     hb_handle_t * h = job->h;
822     hb_state_t state;
823     hb_get_state( h, &state );
824     
825     hb_log("work: average encoding speed for job is %f fps", state.param.working.rate_avg);
826
827 cleanup:
828     /* Stop the write thread (thread_close will block until the muxer finishes) */
829     if( job->muxer != NULL )
830         hb_thread_close( &job->muxer );
831
832     job->done = 1;
833
834     /* Close work objects */
835     while( ( w = hb_list_item( job->list_work, 0 ) ) )
836     {
837         hb_list_rem( job->list_work, w );
838         if( w->thread != NULL )
839         {
840             hb_thread_close( &w->thread );
841             w->close( w );
842         }
843         free( w );
844     }
845
846     hb_list_close( &job->list_work );
847
848     /* Stop the read thread */
849     if( job->reader != NULL )
850         hb_thread_close( &job->reader );
851
852     /* Close fifos */
853     hb_fifo_close( &job->fifo_mpeg2 );
854     hb_fifo_close( &job->fifo_raw );
855     hb_fifo_close( &job->fifo_sync );
856     hb_fifo_close( &job->fifo_render );
857     hb_fifo_close( &job->fifo_mpeg4 );
858
859     for (i=0; i < hb_list_count(title->list_subtitle); i++) {
860         subtitle =  hb_list_item( title->list_subtitle, i);
861         if( subtitle )
862         {
863             hb_fifo_close( &subtitle->fifo_in );
864             hb_fifo_close( &subtitle->fifo_raw );
865         }
866     }
867     for( i = 0; i < hb_list_count( title->list_audio ); i++ )
868     {
869         audio = hb_list_item( title->list_audio, i );
870         if( audio->priv.fifo_in != NULL )
871             hb_fifo_close( &audio->priv.fifo_in );
872         if( audio->priv.fifo_raw != NULL )
873             hb_fifo_close( &audio->priv.fifo_raw );
874         if( audio->priv.fifo_sync != NULL )
875             hb_fifo_close( &audio->priv.fifo_sync );
876         if( audio->priv.fifo_out != NULL )
877             hb_fifo_close( &audio->priv.fifo_out );
878     }
879
880     if( job->indepth_scan )
881     {
882         /*
883          * Before closing the title print out our subtitle stats if we need to
884          * Find the highest and lowest.
885          */
886         for( i=0; i < hb_list_count( title->list_subtitle ); i++ )
887         {
888             subtitle =  hb_list_item( title->list_subtitle, i );
889             hb_log( "Subtitle stream 0x%x '%s': %d hits (%d forced)",
890                     subtitle->id, subtitle->lang, subtitle->hits,
891                     subtitle->forced_hits );
892             if( subtitle->hits > subtitle_highest )
893             {
894                 subtitle_highest = subtitle->hits;
895                 subtitle_highest_id = subtitle->id;
896             }
897
898             if( subtitle->hits < subtitle_lowest )
899             {
900                 subtitle_lowest = subtitle->hits;
901                 subtitle_lowest_id = subtitle->id;
902             }
903
904             if( subtitle->forced_hits > 0 )
905             {
906                 if( subtitle_forced_id == 0 )
907                 {
908                     subtitle_forced_id = subtitle->id;
909                 }
910             }
911         }
912
913         if( job->native_language ) {
914             /*
915              * We still have a native_language, so the audio and subtitles are
916              * different, so in this case it is a foreign film and we want to
917              * select the subtitle with the highest hits in our language.
918              */
919             subtitle_hit = subtitle_highest_id;
920             hb_log( "Found a native-language subtitle id 0x%x", subtitle_hit);
921         } else {
922             if( subtitle_forced_id )
923             {
924                 /*
925                  * If there are any subtitle streams with forced subtitles
926                  * then select it in preference to the lowest.
927                  */
928                 subtitle_hit = subtitle_forced_id;
929                 hb_log("Found a subtitle candidate id 0x%x (contains forced subs)",
930                        subtitle_hit);
931             } else if( subtitle_lowest < subtitle_highest )
932             {
933                 /*
934                  * OK we have more than one, and the lowest is lower,
935                  * but how much lower to qualify for turning it on by
936                  * default?
937                  *
938                  * Let's say 10% as a default.
939                  */
940                 if( subtitle_lowest < ( subtitle_highest * 0.1 ) )
941                 {
942                     subtitle_hit = subtitle_lowest_id;
943                     hb_log( "Found a subtitle candidate id 0x%x",
944                             subtitle_hit );
945                 } else {
946                     hb_log( "No candidate subtitle detected during subtitle-scan");
947                 }
948             }
949         }
950     }
951
952     if( job->select_subtitle )
953     {
954         if( job->indepth_scan )
955         {
956             for( i=0; i < hb_list_count( title->list_subtitle ); i++ )
957             {
958                 subtitle =  hb_list_item( title->list_subtitle, i );
959                 if( subtitle->id == subtitle_hit )
960                 {
961                     hb_list_rem( title->list_subtitle, subtitle );
962                     *( job->select_subtitle ) = subtitle;
963                 }
964             }
965         } else {
966             /*
967              * Must be the end of pass 0 or 2 - we don't need this anymore.
968              *
969              * Have to put the subtitle list back together in the title though
970              * or the GUI will have a hissy fit.
971              */
972             free( job->select_subtitle );
973             job->select_subtitle = NULL;
974         }
975     }
976
977     if( job->filters )
978     {
979         for( i = 0; i < hb_list_count( job->filters ); i++ )
980         {
981             hb_filter_object_t * filter = hb_list_item( job->filters, i );
982             hb_filter_close( &filter );
983         }
984         hb_list_close( &job->filters );
985     }
986
987     hb_buffer_pool_free();
988
989     hb_title_close( &job->title );
990     free( job );
991 }
992
993 /**
994  * Performs the work object's specific work function.
995  * Loops calling work function for associated work object. Sleeps when fifo is full.
996  * Monitors work done indicator.
997  * Exits loop when work indiactor is set.
998  * @param _w Handle to work object.
999  */
1000 static void work_loop( void * _w )
1001 {
1002     hb_work_object_t * w = _w;
1003     hb_buffer_t      * buf_in, * buf_out;
1004
1005     while( !*w->done )
1006     {
1007 #if 0
1008         hb_lock( job->pause );
1009         hb_unlock( job->pause );
1010 #endif
1011         if( hb_fifo_is_full( w->fifo_out ) ||
1012 //        if( (hb_fifo_percent_full( w->fifo_out ) > 0.8) ||
1013             !( buf_in = hb_fifo_get( w->fifo_in ) ) )
1014         {
1015             hb_snooze( w->thread_sleep_interval );
1016 //                      w->thread_sleep_interval += 1;
1017             continue;
1018         }
1019 //              w->thread_sleep_interval = MAX(1, (w->thread_sleep_interval - 1));
1020
1021         w->status = w->work( w, &buf_in, &buf_out );
1022
1023         // Propagate any chapter breaks for the worker if and only if the
1024         // output frame has the same time stamp as the input frame (any
1025         // worker that delays frames has to propagate the chapter marks itself
1026         // and workers that move chapter marks to a different time should set
1027         // 'buf_in' to NULL so that this code won't generate spurious duplicates.)
1028         if( buf_in && buf_out && buf_in->new_chap && buf_in->start == buf_out->start)
1029         {
1030             // restore log below to debug chapter mark propagation problems
1031             //hb_log("work %s: Copying Chapter Break @ %lld", w->name, buf_in->start);
1032             buf_out->new_chap = buf_in->new_chap;
1033         }
1034
1035         if( buf_in )
1036         {
1037             hb_buffer_close( &buf_in );
1038         }
1039         if( buf_out )
1040         {
1041             hb_fifo_push( w->fifo_out, buf_out );
1042         }
1043     }
1044 }