OSDN Git Service

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