OSDN Git Service

Matroska muxer!
[handbrake-jp/handbrake-jp-git.git] / libhb / work.c
index 6529e2f..92707cc 100644 (file)
@@ -5,6 +5,8 @@
    It may be used under the terms of the GNU General Public License. */
 
 #include "hb.h"
+#include "a52dec/a52.h"
+#include "dca.h"
 
 typedef struct
 {
@@ -89,11 +91,20 @@ static hb_work_object_t * getWork( int id )
 static void do_job( hb_job_t * job, int cpu_count )
 {
     hb_title_t    * title;
-    int             i;
+    int             i, j;
     hb_work_object_t * w;
+    
+    /* FIXME: This feels really hackish, anything better? */
+    hb_work_object_t * audio_w = NULL;
+
     hb_audio_t   * audio;
     hb_subtitle_t * subtitle;
     int done;
+    unsigned int subtitle_highest = 0;
+    unsigned int subtitle_highest_id = 0;
+    unsigned int subtitle_lowest = -1;
+    unsigned int subtitle_lowest_id = 0;
+    unsigned int subtitle_hit = 0;
 
     title = job->title;
 
@@ -103,6 +114,29 @@ static void do_job( hb_job_t * job, int cpu_count )
     hb_log( " + device %s", title->dvd );
     hb_log( " + title %d, chapter(s) %d to %d", title->index,
             job->chapter_start, job->chapter_end );
+    if ( job->pixel_ratio == 1 )
+    {
+       /* Correct the geometry of the output movie when using PixelRatio */
+       job->height=title->height-job->crop[0]-job->crop[1];
+       job->width=title->width-job->crop[2]-job->crop[3];
+    }
+
+       /* Keep width and height within these boundaries */
+       if (job->maxHeight && (job->height > job->maxHeight) )
+       {
+               job->height = job->maxHeight;
+               hb_fix_aspect( job, HB_KEEP_HEIGHT );
+               hb_log("Height out of bounds, scaling down to %i", job->maxHeight);
+               hb_log("New dimensions %i * %i", job->width, job->height);
+       }
+       if (job->maxWidth && (job->width > job->maxWidth) )
+       {
+               job->width = job->maxWidth;
+               hb_fix_aspect( job, HB_KEEP_WIDTH );   
+               hb_log("Width out of bounds, scaling down to %i", job->maxWidth);
+               hb_log("New dimensions %i * %i", job->width, job->height);
+       }
+
     hb_log( " + %dx%d -> %dx%d, crop %d/%d/%d/%d",
             title->width, title->height, job->width, job->height,
             job->crop[0], job->crop[1], job->crop[2], job->crop[3] );
@@ -119,7 +153,7 @@ static void do_job( hb_job_t * job, int cpu_count )
                 (float) job->vrate / (float) job->vrate_base,
                 job->vbitrate, job->pass );
     }
-
+       hb_log (" + PixelRatio: %d, width:%d, height: %d",job->pixel_ratio,job->width, job->height);
     job->fifo_mpeg2  = hb_fifo_init( 2048 );
     job->fifo_raw    = hb_fifo_init( 8 );
     job->fifo_sync   = hb_fifo_init( 8 );
@@ -160,19 +194,34 @@ static void do_job( hb_job_t * job, int cpu_count )
     w->fifo_in  = job->fifo_render;
     w->fifo_out = job->fifo_mpeg4;
     w->config   = &job->config;
+    
     hb_list_add( job->list_work, w );
 
-    subtitle = hb_list_item( title->list_subtitle, 0 );
-    if( subtitle )
+    if( job->select_subtitle && !job->subtitle_scan ) 
     {
-        hb_log( " + subtitle %x, %s", subtitle->id, subtitle->lang );
+        hb_list_add( title->list_subtitle, *( job->select_subtitle ) );
+    }
 
-        subtitle->fifo_in  = hb_fifo_init( 8 );
-        subtitle->fifo_raw = hb_fifo_init( 8 );
+    for( i=0; i < hb_list_count(title->list_subtitle); i++ ) 
+    {
+        subtitle =  hb_list_item( title->list_subtitle, i );
 
-        hb_list_add( job->list_work, ( w = getWork( WORK_DECSUB ) ) );
-        w->fifo_in  = subtitle->fifo_in;
-        w->fifo_out = subtitle->fifo_raw;
+        if( subtitle )
+        {
+            hb_log( " + subtitle %x, %s", subtitle->id, subtitle->lang );
+            
+            subtitle->fifo_in  = hb_fifo_init( 8 );
+            subtitle->fifo_raw = hb_fifo_init( 8 );
+            
+            if (!job->subtitle_scan) {
+                /*
+                 * Don't add threads for subtitles when we are scanning
+                 */
+                hb_list_add( job->list_work, ( w = getWork( WORK_DECSUB ) ) );
+                w->fifo_in  = subtitle->fifo_in;
+                w->fifo_out = subtitle->fifo_raw;
+            }
+        }
     }
 
     if( job->acodec & HB_ACODEC_AC3 )
@@ -187,11 +236,152 @@ static void do_job( hb_job_t * job, int cpu_count )
                 "vorbis" ) );
     }
 
+    /* if we are doing AC3 passthru, then remove any non-AC3 audios from the job */
+    /* otherwise, Bad Things will happen */
+    for( i = 0; i < hb_list_count( title->list_audio ); )
+    {
+        audio = hb_list_item( title->list_audio, i );
+        if( ( job->acodec & HB_ACODEC_AC3 ) && ( audio->codec != HB_ACODEC_AC3 ) )
+        {
+            hb_list_rem( title->list_audio, audio );
+            free( audio );
+            continue;
+        }
+        i++;
+    }
+
     for( i = 0; i < hb_list_count( title->list_audio ); i++ )
     {
         audio = hb_list_item( title->list_audio, i );
         hb_log( "   + %x, %s", audio->id, audio->lang );
+                       
+               /* sense-check the current mixdown options */
+
+               /* log the requested mixdown */
+               for (j = 0; j < hb_audio_mixdowns_count; j++) {
+                       if (hb_audio_mixdowns[j].amixdown == job->audio_mixdowns[i]) {
+                               hb_log( "     + Requested mixdown: %s (%s)", hb_audio_mixdowns[j].human_readable_name, hb_audio_mixdowns[j].internal_name );
+                               break;
+                       }
+               }
+
+        /* sense-check the requested mixdown */
+
+        /* audioCodecsSupportMono and audioCodecsSupport6Ch are the same for now,
+           but this may change in the future, so they are separated for flexibility */
+        int audioCodecsSupportMono = ((audio->codec == HB_ACODEC_AC3 ||
+            audio->codec == HB_ACODEC_DCA) && job->acodec == HB_ACODEC_FAAC);
+        int audioCodecsSupport6Ch =  ((audio->codec == HB_ACODEC_AC3 ||
+            audio->codec == HB_ACODEC_DCA) && job->acodec == HB_ACODEC_FAAC);
+
+        /* find out what the format of our source audio is */
+        switch (audio->input_channel_layout & HB_INPUT_CH_LAYOUT_DISCRETE_NO_LFE_MASK) {
+        
+            /* mono sources */
+            case HB_INPUT_CH_LAYOUT_MONO:
+                /* regardless of what stereo mixdown we've requested, a mono source always get mixed down
+                to mono if we can, and mixed up to stereo if we can't */
+                if (job->audio_mixdowns[i] == HB_AMIXDOWN_MONO && audioCodecsSupportMono == 1) {
+                    job->audio_mixdowns[i] = HB_AMIXDOWN_MONO;
+                } else {
+                    job->audio_mixdowns[i] = HB_AMIXDOWN_STEREO;
+                }
+                break;
+
+            /* stereo input */
+            case HB_INPUT_CH_LAYOUT_STEREO:
+                /* if we've requested a mono mixdown, and it is supported, then do the mix */
+                /* use stereo if not supported */
+                if (job->audio_mixdowns[i] == HB_AMIXDOWN_MONO && audioCodecsSupportMono == 0) {
+                    job->audio_mixdowns[i] = HB_AMIXDOWN_STEREO;
+                /* otherwise, preserve stereo regardless of if we requested something higher */
+                } else if (job->audio_mixdowns[i] > HB_AMIXDOWN_STEREO) {
+                    job->audio_mixdowns[i] = HB_AMIXDOWN_STEREO;
+                }
+                break;
+
+            /* dolby (DPL1 aka Dolby Surround = 4.0 matrix-encoded) input */
+            /* the A52 flags don't allow for a way to distinguish between DPL1 and DPL2 on a DVD,
+               so we always assume a DPL1 source for A52_DOLBY */
+            case HB_INPUT_CH_LAYOUT_DOLBY:
+                /* if we've requested a mono mixdown, and it is supported, then do the mix */
+                /* preserve dolby if not supported */
+                if (job->audio_mixdowns[i] == HB_AMIXDOWN_MONO && audioCodecsSupportMono == 0) {
+                    job->audio_mixdowns[i] = HB_AMIXDOWN_DOLBY;
+                /* otherwise, preserve dolby even if we requested something higher */
+                /* a stereo mixdown will still be honoured here */
+                } else if (job->audio_mixdowns[i] > HB_AMIXDOWN_DOLBY) {
+                    job->audio_mixdowns[i] = HB_AMIXDOWN_DOLBY;
+                }
+                break;
+
+            /* 3F/2R input */
+            case HB_INPUT_CH_LAYOUT_3F2R:
+                /* if we've requested a mono mixdown, and it is supported, then do the mix */
+                /* use dpl2 if not supported */
+                if (job->audio_mixdowns[i] == HB_AMIXDOWN_MONO && audioCodecsSupportMono == 0) {
+                    job->audio_mixdowns[i] = HB_AMIXDOWN_DOLBYPLII;
+                } else {
+                    /* check if we have 3F2R input and also have an LFE - i.e. we have a 5.1 source) */
+                    if (audio->input_channel_layout & HB_INPUT_CH_LAYOUT_HAS_LFE) {
+                        /* we have a 5.1 source */
+                        /* if we requested 6ch, but our audio format doesn't support it, then mix to DPLII instead */
+                        if (job->audio_mixdowns[i] == HB_AMIXDOWN_6CH && audioCodecsSupport6Ch == 0) {
+                            job->audio_mixdowns[i] = HB_AMIXDOWN_DOLBYPLII;
+                        }
+                    } else {
+                        /* we have a 5.0 source, so we can't do 6ch conversion
+                        default to DPL II instead */
+                        if (job->audio_mixdowns[i] > HB_AMIXDOWN_DOLBYPLII) {
+                            job->audio_mixdowns[i] = HB_AMIXDOWN_DOLBYPLII;
+                        }
+                    }
+                }
+                /* all other mixdowns will have been preserved here */
+                break;
+
+            /* 3F/1R input */
+            case HB_INPUT_CH_LAYOUT_3F1R:
+                /* if we've requested a mono mixdown, and it is supported, then do the mix */
+                /* use dpl1 if not supported */
+                if (job->audio_mixdowns[i] == HB_AMIXDOWN_MONO && audioCodecsSupportMono == 0) {
+                    job->audio_mixdowns[i] = HB_AMIXDOWN_DOLBY;
+                } else {
+                    /* we have a 4.0 or 4.1 source, so we can't do DPLII or 6ch conversion
+                    default to DPL I instead */
+                    if (job->audio_mixdowns[i] > HB_AMIXDOWN_DOLBY) {
+                        job->audio_mixdowns[i] = HB_AMIXDOWN_DOLBY;
+                    }
+                }
+                /* all other mixdowns will have been preserved here */
+                break;
+
+            default:
+                /* if we've requested a mono mixdown, and it is supported, then do the mix */
+                if (job->audio_mixdowns[i] == HB_AMIXDOWN_MONO && audioCodecsSupportMono == 1) {
+                    job->audio_mixdowns[i] = HB_AMIXDOWN_MONO;
+                /* mix everything else down to stereo */
+                } else {
+                    job->audio_mixdowns[i] = HB_AMIXDOWN_STEREO;
+                }
+
+        }
+
+               /* log the output mixdown */
+               for (j = 0; j < hb_audio_mixdowns_count; j++) {
+                       if (hb_audio_mixdowns[j].amixdown == job->audio_mixdowns[i]) {
+                               hb_log( "     + Actual mixdown: %s (%s)", hb_audio_mixdowns[j].human_readable_name, hb_audio_mixdowns[j].internal_name );
+                               break;
+                       }
+               }
 
+               /* we now know we have a valid mixdown for the input source and the audio output format */
+               /* remember the mixdown for this track */
+               audio->amixdown = job->audio_mixdowns[i];
+
+        audio->config.vorbis.language = audio->lang_simple;
+
+               /* set up the audio work structures */
         audio->fifo_in   = hb_fifo_init( 2048 );
         audio->fifo_raw  = hb_fifo_init( 8 );
         audio->fifo_sync = hb_fifo_init( 8 );
@@ -202,6 +392,9 @@ static void do_job( hb_job_t * job, int cpu_count )
             case HB_ACODEC_AC3:
                 w = getWork( WORK_DECA52 );
                 break;
+            case HB_ACODEC_DCA:
+                w = getWork( WORK_DECDCA );
+                break;
             case HB_ACODEC_MPGA:
                 w = getWork( WORK_DECAVCODEC );
                 break;
@@ -209,9 +402,17 @@ static void do_job( hb_job_t * job, int cpu_count )
                 w = getWork( WORK_DECLPCM );
                 break;
         }
-        w->fifo_in  = audio->fifo_in;
-        w->fifo_out = audio->fifo_raw;
-        hb_list_add( job->list_work, w );
+        w->fifo_in       = audio->fifo_in;
+        w->fifo_out      = audio->fifo_raw;
+        w->config        = &audio->config;
+        w->amixdown      = audio->amixdown;
+        w->source_acodec = audio->codec;
+        
+        /* FIXME: This feels really hackish, anything better? */
+        audio_w = calloc( sizeof( hb_work_object_t ), 1 );
+        audio_w = memcpy( audio_w, w, sizeof( hb_work_object_t ));
+        
+        hb_list_add( job->list_work, audio_w );
 
         switch( job->acodec )
         {
@@ -225,13 +426,23 @@ static void do_job( hb_job_t * job, int cpu_count )
                 w = getWork( WORK_ENCVORBIS );
                 break;
         }
+
         if( job->acodec != HB_ACODEC_AC3 )
         {
-            w->fifo_in  = audio->fifo_sync;
-            w->fifo_out = audio->fifo_out;
-            w->config   = &audio->config;
-            hb_list_add( job->list_work, w );
+            w->fifo_in       = audio->fifo_sync;
+            w->fifo_out      = audio->fifo_out;
+            w->config        = &audio->config;
+            w->amixdown      = audio->amixdown;
+            w->source_acodec = audio->codec;
+            
+            /* FIXME: This feels really hackish, anything better? */
+            audio_w = calloc( sizeof( hb_work_object_t ), 1 );
+            audio_w = memcpy( audio_w, w, sizeof( hb_work_object_t ));
+        
+            hb_list_add( job->list_work, audio_w );
         }
+
+
     }
 
     /* Init read & write threads */
@@ -266,7 +477,7 @@ static void do_job( hb_job_t * job, int cpu_count )
         if( done &&
             !hb_fifo_size( job->fifo_sync ) &&
             !hb_fifo_size( job->fifo_render ) &&
-            hb_fifo_size( job->fifo_mpeg4 ) < 2 )
+            !hb_fifo_size( job->fifo_mpeg4 ) )
         {
             break;
         }
@@ -282,7 +493,21 @@ static void do_job( hb_job_t * job, int cpu_count )
         hb_list_rem( job->list_work, w );
         hb_thread_close( &w->thread );
         w->close( w );
+        
+        /* FIXME: This feels really hackish, anything better? */
+        if ( w->id == WORK_DECA52 ||
+             w->id == WORK_DECDCA ||
+             w->id == WORK_DECLPCM ||
+             w->id == WORK_ENCFAAC ||
+             w->id == WORK_ENCLAME ||
+             w->id == WORK_ENCVORBIS )
+        {
+            free( w );
+            w = NULL;
+        }
     }
+    
+    hb_list_close( &job->list_work );
 
     /* Stop read & write threads */
     hb_thread_close( &job->reader );
@@ -294,10 +519,13 @@ static void do_job( hb_job_t * job, int cpu_count )
     hb_fifo_close( &job->fifo_sync );
     hb_fifo_close( &job->fifo_render );
     hb_fifo_close( &job->fifo_mpeg4 );
-    if( subtitle )
-    {
-        hb_fifo_close( &subtitle->fifo_in );
-        hb_fifo_close( &subtitle->fifo_raw );
+    for (i=0; i < hb_list_count(title->list_subtitle); i++) {
+        subtitle =  hb_list_item( title->list_subtitle, i);
+        if( subtitle )
+        {
+            hb_fifo_close( &subtitle->fifo_in );
+            hb_fifo_close( &subtitle->fifo_raw );
+        }
     }
     for( i = 0; i < hb_list_count( title->list_audio ); i++ )
     {
@@ -307,6 +535,82 @@ static void do_job( hb_job_t * job, int cpu_count )
         hb_fifo_close( &audio->fifo_sync );
         hb_fifo_close( &audio->fifo_out );
     }
+    
+    /*
+     * Before closing the title print out our subtitle stats if we need to
+     * Find the highest and lowest.
+     */
+    for( i=0; i < hb_list_count( title->list_subtitle ); i++ ) 
+    {
+        subtitle =  hb_list_item( title->list_subtitle, i );
+        hb_log( "Subtitle stream 0x%x '%s': %d hits",
+               subtitle->id, subtitle->lang, subtitle->hits );
+        if( subtitle->hits > subtitle_highest ) 
+        {
+            subtitle_highest = subtitle->hits;
+            subtitle_highest_id = subtitle->id;
+        } 
+
+        if( subtitle->hits < subtitle_lowest ) 
+        {
+            subtitle_lowest = subtitle->hits;
+            subtitle_lowest_id = subtitle->id;
+        }
+    }
+
+    if( job->native_language ) {
+        /*
+         * We still have a native_language, so the audio and subtitles are
+         * different, so in this case it is a foreign film and we want to
+         * select the first subtitle in our language.
+         */
+        subtitle =  hb_list_item( title->list_subtitle, 0 );
+        subtitle_hit = subtitle->id;
+        hb_log( "Found a native-language subtitle id 0x%x", subtitle_hit);
+    } else {
+        if( subtitle_lowest < subtitle_highest ) 
+        {
+            /*
+             * OK we have more than one, and the lowest is lower, but how much
+             * lower to qualify for turning it on by default?
+             *
+             * Let's say 10% as a default.
+             */
+            if( subtitle_lowest < ( subtitle_highest * 0.1 ) ) 
+            {
+                subtitle_hit = subtitle_lowest_id;
+                hb_log( "Found a subtitle candidate id 0x%x",
+                        subtitle_hit );
+            } else {
+                hb_log( "No candidate subtitle detected during subtitle-scan");
+            }
+        }
+    }
+
+    if( job->select_subtitle ) 
+    {
+        if( job->subtitle_scan ) 
+        {
+            for( i=0; i < hb_list_count( title->list_subtitle ); i++ ) 
+            {
+                subtitle =  hb_list_item( title->list_subtitle, i );
+                if( subtitle->id = subtitle_hit ) 
+                {
+                    hb_list_rem( title->list_subtitle, subtitle );
+                    *( job->select_subtitle ) = subtitle;
+                }
+            }
+        } else {
+            /*
+             * Must be the second pass - we don't need this anymore.
+             */
+            free( job->select_subtitle );
+            job->select_subtitle = NULL;
+        }
+    }
+
+    hb_title_close( &job->title );
+    free( job );
 }
 
 /**
@@ -338,6 +642,14 @@ static void work_loop( void * _w )
 //             w->thread_sleep_interval = MAX(1, (w->thread_sleep_interval - 1));
 
         w->work( w, &buf_in, &buf_out );
+
+        // Propogate any chapter breaks for the worker
+        if( buf_in && buf_out && buf_in->new_chap )
+        {
+            printf("WORK: Copying Chapter Break\n");
+            buf_out->new_chap = 1;
+        }
+        
         if( buf_in )
         {
             hb_buffer_close( &buf_in );