OSDN Git Service

Set an minimum subtitle display time of three seconds *or* until the next subtitle...
[handbrake-jp/handbrake-jp-git.git] / libhb / sync.c
index 9d6679d..04bdf16 100644 (file)
@@ -20,7 +20,7 @@ typedef struct
 {
     hb_audio_t * audio;
     int64_t      count_frames;
-    
+
     /* Raw */
     SRC_STATE  * state;
     SRC_DATA     data;
@@ -40,6 +40,7 @@ struct hb_work_private_s
     hb_subtitle_t * subtitle;
     int64_t pts_offset;
     int64_t pts_offset_old;
+    int64_t next_start;
     int64_t count_frames;
     int64_t count_frames_max;
     int64_t video_sequence;
@@ -105,7 +106,7 @@ int syncInit( hb_work_object_t * w, hb_job_t * job )
     {
         chapter   = hb_list_item( title->list_chapter, i - 1 );
         duration += chapter->duration;
-    }                                                                           
+    }
     duration += 90000;
         /* 1 second safety so we're sure we won't miss anything */
     pv->count_frames_max = duration * job->vrate / job->vrate_base / 90000;
@@ -136,14 +137,15 @@ void syncClose( hb_work_object_t * w )
     hb_work_private_t * pv = w->private_data;
     hb_job_t          * job   = pv->job;
     hb_title_t        * title = job->title;
-    
+
     int i;
 
     if( pv->cur ) hb_buffer_close( &pv->cur );
 
     for( i = 0; i < hb_list_count( title->list_audio ); i++ )
     {
-        if( job->acodec & HB_ACODEC_AC3 )
+        if( job->acodec & HB_ACODEC_AC3 ||
+            job->audio_mixdowns[i] == HB_AMIXDOWN_AC3 )
         {
             free( pv->sync_audio[i].ac3_buf );
         }
@@ -152,7 +154,7 @@ void syncClose( hb_work_object_t * w )
             src_delete( pv->sync_audio[i].state );
         }
     }
-    
+
     free( pv );
     w->private_data = NULL;
 }
@@ -205,7 +207,8 @@ static void InitAudio( hb_work_object_t * w, int i )
     sync        = &pv->sync_audio[i];
     sync->audio = hb_list_item( title->list_audio, i );
 
-    if( job->acodec & HB_ACODEC_AC3 )
+    if( job->acodec & HB_ACODEC_AC3 ||
+        job->audio_mixdowns[i] == HB_AMIXDOWN_AC3 )
     {
         /* Have a silent AC-3 frame ready in case we have to fill a
            gap */
@@ -237,7 +240,7 @@ static void InitAudio( hb_work_object_t * w, int i )
         {
             hb_log( "sync: avcodec_encode_audio failed" );
         }
-        
+
         free( zeros );
         avcodec_close( c );
         av_free( c );
@@ -258,7 +261,7 @@ static void InitAudio( hb_work_object_t * w, int i )
 /***********************************************************************
  * SyncVideo
  ***********************************************************************
- * 
+ *
  **********************************************************************/
 static int SyncVideo( hb_work_object_t * w )
 {
@@ -282,14 +285,14 @@ static int SyncVideo( hb_work_object_t * w )
         hb_log( "sync: got %lld frames, %lld expected",
                 pv->count_frames, pv->count_frames_max );
         pv->done = 1;
-        
+
         hb_buffer_t * buf_tmp;
 
        // Drop an empty buffer into our output to ensure that things
        // get flushed all the way out.
         buf_tmp = hb_buffer_init(0); // Empty end buffer
         hb_fifo_push( job->fifo_sync, buf_tmp );
-        
+
         return HB_WORK_DONE;
     }
 
@@ -328,7 +331,7 @@ static int SyncVideo( hb_work_object_t * w )
         {
            hb_log( "Sync: Video PTS discontinuity %s (current buffer start=%lld, next buffer start=%lld)",
                     pv->discontinuity ? "second" : "first", cur->start, next->start );
-            
+
             /*
              * Do we need to trash the subtitle, is it from the next->start period
              * or is it from our old position. If the latter then trash it.
@@ -363,8 +366,13 @@ static int SyncVideo( hb_work_object_t * w )
 
             /* Calculate new offset */
             pv->pts_offset_old = pv->pts_offset;
-            pv->pts_offset     = cur->start -
-                pv->count_frames * pv->job->vrate_base / 300;
+            if ( job->vfr )
+            {
+                pv->pts_offset = cur->start - pv->next_start;
+            } else {
+                pv->pts_offset = cur->start -
+                    pv->count_frames * pv->job->vrate_base / 300;
+            }
 
             if( !pv->discontinuity )
             {
@@ -387,7 +395,7 @@ static int SyncVideo( hb_work_object_t * w )
                 if( sub2 && sub->stop > sub2->start )
                     sub->stop = sub2->start;
 
-                // hb_log("0x%x: video seq: %lld  subtitle sequence: %lld", 
+                // hb_log("0x%x: video seq: %lld  subtitle sequence: %lld",
                 //       sub, cur->sequence, sub->sequence);
 
                 if( sub->sequence > cur->sequence )
@@ -409,8 +417,8 @@ static int SyncVideo( hb_work_object_t * w )
                      * code.
                      */
                     break;
-                } 
-                else 
+                }
+                else
                 {
                     /*
                      * The stop time is in the past. But is it due to
@@ -428,8 +436,8 @@ static int SyncVideo( hb_work_object_t * w )
                     }
                 }
 
-                /* 
-                 * The subtitle is older than this picture, trash it 
+                /*
+                 * The subtitle is older than this picture, trash it
                  */
                 sub = hb_fifo_get( pv->subtitle->fifo_raw );
                 hb_buffer_close( &sub );
@@ -443,7 +451,7 @@ static int SyncVideo( hb_work_object_t * w )
                 if( sub->stop > sub->start)
                 {
                     /*
-                     * Normal subtitle which ends after it starts, check to 
+                     * Normal subtitle which ends after it starts, check to
                      * see that the current video is between the start and end.
                      */
                     if( cur->start > sub->start &&
@@ -455,7 +463,27 @@ static int SyncVideo( hb_work_object_t * w )
                          *
                          * fall through to display
                          */
-                    } 
+                        if( ( sub->stop - sub->start ) < ( 3 * 90000 ) )
+                        {
+                            /*
+                             * Subtitle is on for less than three seconds, extend
+                             * the time that it is displayed to make it easier
+                             * to read. Make it 3 seconds or until the next
+                             * subtitle is displayed. 
+                             *
+                             * This is in response to Indochine which only 
+                             * displays subs for 1 second - too fast to read.
+                             */
+                            sub->stop = sub->start + ( 3 * 90000 );
+
+                            sub2 = hb_fifo_see2( pv->subtitle->fifo_raw );
+
+                            if( sub2 && sub->stop > sub2->start )
+                            {
+                                sub->stop = sub2->start;
+                            }
+                        }
+                    }
                     else
                     {
                         /*
@@ -485,8 +513,8 @@ static int SyncVideo( hb_work_object_t * w )
                              *
                              * fall through to display.
                              */
-                        } 
-                        else 
+                        }
+                        else
                         {
                             /*
                              * Defer until the play point is within the subtitle
@@ -505,48 +533,72 @@ static int SyncVideo( hb_work_object_t * w )
             }
         }
 
-        /* The PTS of the frame we are expecting now */
-        pts_expected = pv->pts_offset +
-            pv->count_frames * pv->job->vrate_base / 300;
-
-        //hb_log("Video expecting PTS %lld, current frame: %lld, next frame: %lld, cf: %lld", 
-        //       pts_expected, cur->start, next->start, pv->count_frames * pv->job->vrate_base / 300 );
-
-        if( cur->start < pts_expected - pv->job->vrate_base / 300 / 2 &&
-            next->start < pts_expected + pv->job->vrate_base / 300 / 2 )
+        if ( job->vfr )
         {
-            /* The current frame is too old but the next one matches,
-               let's trash */
-            /* Also, make sure we don't trash a chapter break */
-            chap_break = cur->new_chap;
-            hb_buffer_close( &cur );
+            /*
+             * adjust the pts of the current frame so that it's contiguous
+             * with the previous frame. pts_offset tracks the time difference
+             * between the pts values in the input content (which start at some
+             * random time) and our timestamps (which start at zero). We don't
+             * make any adjustments to the source timestamps other than removing
+             * the clock offsets (which also removes pts discontinuities).
+             * This means we automatically encode at the source's frame rate.
+             * MP2 uses an implicit duration (frames end when the next frame
+             * starts) but more advanced containers like MP4 use an explicit
+             * duration. Since we're looking ahead one frame we set the
+             * explicit stop time from the start time of the next frame.
+             */
+            buf_tmp = cur;
             pv->cur = cur = hb_fifo_get( job->fifo_raw );
-            cur->new_chap |= chap_break; // Make sure we don't stomp the existing one.
-            
-            continue;
-        }
-
-        if( next->start > pts_expected + 3 * pv->job->vrate_base / 300 / 2 )
-        {
-            /* We'll need the current frame more than one time. Make a
-               copy of it and keep it */
-            buf_tmp = hb_buffer_init( cur->size );
-            memcpy( buf_tmp->data, cur->data, cur->size ); 
-            buf_tmp->sequence = cur->sequence;
+            buf_tmp->start = pv->next_start;
+            pv->next_start = next->start - pv->pts_offset;
+            buf_tmp->stop = pv->next_start;
         }
         else
         {
-            /* The frame has the expected date and won't have to be
-               duplicated, just put it through */
-            buf_tmp = cur;
-            pv->cur = cur = hb_fifo_get( job->fifo_raw );
+            /* The PTS of the frame we are expecting now */
+            pts_expected = pv->pts_offset +
+                pv->count_frames * pv->job->vrate_base / 300;
+
+            //hb_log("Video expecting PTS %lld, current frame: %lld, next frame: %lld, cf: %lld",
+            //       pts_expected, cur->start, next->start, pv->count_frames * pv->job->vrate_base / 300 );
+
+            if( cur->start < pts_expected - pv->job->vrate_base / 300 / 2 &&
+                next->start < pts_expected + pv->job->vrate_base / 300 / 2 )
+            {
+                /* The current frame is too old but the next one matches,
+                   let's trash */
+                /* Also, make sure we don't trash a chapter break */
+                chap_break = cur->new_chap;
+                hb_buffer_close( &cur );
+                pv->cur = cur = hb_fifo_get( job->fifo_raw );
+                cur->new_chap |= chap_break; // Make sure we don't stomp the existing one.
+
+                continue;
+            }
+
+            if( next->start > pts_expected + 3 * pv->job->vrate_base / 300 / 2 )
+            {
+                /* We'll need the current frame more than one time. Make a
+                   copy of it and keep it */
+                buf_tmp = hb_buffer_init( cur->size );
+                memcpy( buf_tmp->data, cur->data, cur->size );
+                buf_tmp->sequence = cur->sequence;
+            }
+            else
+            {
+                /* The frame has the expected date and won't have to be
+                   duplicated, just put it through */
+                buf_tmp = cur;
+                pv->cur = cur = hb_fifo_get( job->fifo_raw );
+            }
+
+            /* Replace those MPEG-2 dates with our dates */
+            buf_tmp->start = (uint64_t) pv->count_frames *
+                pv->job->vrate_base / 300;
+            buf_tmp->stop  = (uint64_t) ( pv->count_frames + 1 ) *
+                pv->job->vrate_base / 300;
         }
-        
-        /* Replace those MPEG-2 dates with our dates */
-        buf_tmp->start = (uint64_t) pv->count_frames *
-            pv->job->vrate_base / 300;
-        buf_tmp->stop  = (uint64_t) ( pv->count_frames + 1 ) *
-            pv->job->vrate_base / 300;
 
         /* If we have a subtitle for this picture, copy it */
         /* FIXME: we should avoid this memcpy */
@@ -571,12 +623,12 @@ static int SyncVideo( hb_work_object_t * w )
         {
             hb_log( "sync: got too many frames (%lld), exiting early", pv->count_frames );
             pv->done = 1;
-            
+
            // Drop an empty buffer into our output to ensure that things
            // get flushed all the way out.
            buf_tmp = hb_buffer_init(0); // Empty end buffer
            hb_fifo_push( job->fifo_sync, buf_tmp );
-            
+
             break;
         }
     }
@@ -587,7 +639,7 @@ static int SyncVideo( hb_work_object_t * w )
 /***********************************************************************
  * SyncAudio
  ***********************************************************************
- * 
+ *
  **********************************************************************/
 static void SyncAudio( hb_work_object_t * w, int i )
 {
@@ -607,7 +659,8 @@ static void SyncAudio( hb_work_object_t * w, int i )
     sync   = &pv->sync_audio[i];
     audio  = sync->audio;
 
-    if( job->acodec & HB_ACODEC_AC3 )
+    if( job->acodec & HB_ACODEC_AC3 ||
+        job->audio_mixdowns[i] == HB_AMIXDOWN_AC3 )
     {
         fifo = audio->fifo_out;
         rate = audio->rate;
@@ -640,7 +693,7 @@ static void SyncAudio( hb_work_object_t * w, int i )
             if( pv->discontinuity )
             {
                 /*
-                 * There is an outstanding discontinuity, so use the offset from 
+                 * There is an outstanding discontinuity, so use the offset from
                  * that discontinuity.
                  */
                 pts_expected = pv->pts_offset_old + sync->count_frames *
@@ -688,7 +741,7 @@ static void SyncAudio( hb_work_object_t * w, int i )
                  * Try and reconverge regardless. so continue on to
                  * our convergence code below which will kick in as
                  * it will be more than 100ms out.
-                 * 
+                 *
                  * Note that trashing the Audio could make things
                  * worse if the Audio is in front because we will end
                  * up diverging even more. We need to hold on to the
@@ -699,7 +752,7 @@ static void SyncAudio( hb_work_object_t * w, int i )
                     hb_log("Sync: Audio is way out of sync, attempt to reconverge from current video PTS");
                     pv->way_out_of_sync = 1;
                 }
-               
+
                 /*
                  * It wasn't from the old place, so we must be from
                  * the new, but just too far out. So attempt to
@@ -752,8 +805,8 @@ static void SyncAudio( hb_work_object_t * w, int i )
             }
             InsertSilence( w, i );
             continue;
-        } 
-        else 
+        }
+        else
         {
             if( pv->trashing_audio || pv->inserting_silence )
             {
@@ -769,7 +822,8 @@ static void SyncAudio( hb_work_object_t * w, int i )
             }
         }
 
-        if( job->acodec & HB_ACODEC_AC3 )
+        if( job->acodec & HB_ACODEC_AC3 ||
+            job->audio_mixdowns[i] == HB_AMIXDOWN_AC3 )
         {
             buf        = hb_fifo_get( audio->fifo_raw );
             buf->start = start;
@@ -822,7 +876,7 @@ static void SyncAudio( hb_work_object_t * w, int i )
     }
 
     if( hb_fifo_is_full( fifo ) &&
-        pv->way_out_of_sync ) 
+        pv->way_out_of_sync )
     {
         /*
          * Trash the top audio packet to avoid dead lock as we reconverge.
@@ -885,7 +939,8 @@ static void InsertSilence( hb_work_object_t * w, int i )
     job    = pv->job;
     sync   = &pv->sync_audio[i];
 
-    if( job->acodec & HB_ACODEC_AC3 )
+    if( job->acodec & HB_ACODEC_AC3 ||
+        job->audio_mixdowns[i] == HB_AMIXDOWN_AC3 )
     {
         buf        = hb_buffer_init( sync->ac3_size );
         buf->start = sync->count_frames * 90000 / sync->audio->rate;
@@ -933,14 +988,14 @@ static void UpdateState( hb_work_object_t * w )
                  3 * sizeof( uint64_t ) );
         pv->st_dates[3]  = hb_get_date();
         pv->st_counts[3] = pv->count_frames;
-    } 
+    }
 
 #define p state.param.working
     state.state = HB_STATE_WORKING;
     p.progress  = (float) pv->count_frames / (float) pv->count_frames_max;
     if( p.progress > 1.0 )
     {
-        p.progress = 1.0; 
+        p.progress = 1.0;
     }
     p.rate_cur   = 1000.0 *
         (float) ( pv->st_counts[3] - pv->st_counts[0] ) /