OSDN Git Service

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