[FFmpeg-cvslog] r12156 - in trunk: libavcodec/avcodec.h libavcodec/mpegvideo_parser.c libavformat/avformat.h libavformat/utils.c

[michael]

Author: michael
Date: Wed Feb 20 04:20:06 2008
New Revision: 12156

Log:
Make timestamp interpolation work with mpeg2 field pictures.
Cleaner/simpler solutions are welcome.


Modified:
   trunk/libavcodec/avcodec.h
   trunk/libavcodec/mpegvideo_parser.c
   trunk/libavformat/avformat.h
   trunk/libavformat/utils.c

Modified: trunk/libavcodec/avcodec.h
==============================================================================
--- trunk/libavcodec/avcodec.h	(original)
+++ trunk/libavcodec/avcodec.h	Wed Feb 20 04:20:06 2008
@@ -2788,6 +2788,7 @@ typedef struct AVCodecParserContext {
     /* video info */
     int pict_type; /* XXX: Put it back in AVCodecContext. */
     int repeat_pict; /* XXX: Put it back in AVCodecContext. */
+    int parity;
     int64_t pts;     /* pts of the current frame */
     int64_t dts;     /* dts of the current frame */
 

Modified: trunk/libavcodec/mpegvideo_parser.c
==============================================================================
--- trunk/libavcodec/mpegvideo_parser.c	(original)
+++ trunk/libavcodec/mpegvideo_parser.c	Wed Feb 20 04:20:06 2008
@@ -36,6 +36,7 @@ static void mpegvideo_extract_headers(AV
     int horiz_size_ext, vert_size_ext, bit_rate_ext;
 //FIXME replace the crap with get_bits()
     s->repeat_pict = 0;
+    s->parity = 0;
     buf_end = buf + buf_size;
     while (buf < buf_end) {
         start_code= -1;
@@ -105,8 +106,11 @@ static void mpegvideo_extract_headers(AV
 
                         /* the packet only represents half a frame
                            XXX,FIXME maybe find a different solution */
-                        if(picture_structure != 3)
+                        if(picture_structure != 3){
                             s->repeat_pict = -1;
+                            s->parity = picture_structure-1;
+                        }else
+                            s->parity = !top_field_first;
                     }
                     break;
                 }

Modified: trunk/libavformat/avformat.h
==============================================================================
--- trunk/libavformat/avformat.h	(original)
+++ trunk/libavformat/avformat.h	Wed Feb 20 04:20:06 2008
@@ -337,8 +337,8 @@ typedef struct AVStream {
     struct AVCodecParserContext *parser;
 
     int64_t cur_dts;
-    int last_IP_duration;
-    int64_t last_IP_pts;
+    int last_IP_duration[2];
+    int64_t last_IP_pts[2];
     /* av_seek_frame() support */
     AVIndexEntry *index_entries; /**< only used if the format does not
                                     support seeking natively */

Modified: trunk/libavformat/utils.c
==============================================================================
--- trunk/libavformat/utils.c	(original)
+++ trunk/libavformat/utils.c	Wed Feb 20 04:20:06 2008
@@ -702,21 +702,30 @@ static void compute_pkt_fields(AVFormatC
     /* interpolate PTS and DTS if they are not present */
     if(delay <=1){
         if (presentation_delayed) {
+            int fields= 2 + (pc ? pc->repeat_pict : 0);
+            int field_duration= pkt->duration / fields;
+            int parity= pc ? pc->parity : 0;
             /* DTS = decompression timestamp */
             /* PTS = presentation timestamp */
             if (pkt->dts == AV_NOPTS_VALUE)
-                pkt->dts = st->last_IP_pts;
+                pkt->dts = st->last_IP_pts[parity];
             update_initial_timestamps(s, pkt->stream_index, pkt->dts, pkt->pts);
             if (pkt->dts == AV_NOPTS_VALUE)
                 pkt->dts = st->cur_dts;
 
             /* this is tricky: the dts must be incremented by the duration
             of the frame we are displaying, i.e. the last I- or P-frame */
-            if (st->last_IP_duration == 0)
-                st->last_IP_duration = pkt->duration;
-            st->cur_dts = pkt->dts + st->last_IP_duration;
-            st->last_IP_duration  = pkt->duration;
-            st->last_IP_pts= pkt->pts;
+            st->cur_dts = pkt->dts;
+            for(i=0; i<fields; i++){
+                int p= (parity + i)&1;
+                if(!st->last_IP_duration[p])
+                    st->last_IP_duration[p]= field_duration;
+                st->cur_dts += st->last_IP_duration[p];
+                st->last_IP_pts[p]= pkt->pts;
+                if(pkt->pts != AV_NOPTS_VALUE)
+                    st->last_IP_pts[p] += i*field_duration;
+                st->last_IP_duration[p]= field_duration;
+            }
             /* cannot compute PTS if not present (we can compute it only
             by knowing the future */
         } else if(pkt->pts != AV_NOPTS_VALUE || pkt->dts != AV_NOPTS_VALUE || pkt->duration){
@@ -1014,7 +1023,8 @@ static void av_read_frame_flush(AVFormat
             av_parser_close(st->parser);
             st->parser = NULL;
         }
-        st->last_IP_pts = AV_NOPTS_VALUE;
+        st->last_IP_pts[0] =
+        st->last_IP_pts[1] = AV_NOPTS_VALUE;
         st->cur_dts = AV_NOPTS_VALUE; /* we set the current DTS to an unspecified origin */
     }
 }
@@ -1622,7 +1632,8 @@ static void av_estimate_timings_from_pts
     for(i=0; i<ic->nb_streams; i++){
         st= ic->streams[i];
         st->cur_dts= st->first_dts;
-        st->last_IP_pts = AV_NOPTS_VALUE;
+        st->last_IP_pts[0] =
+        st->last_IP_pts[1] = AV_NOPTS_VALUE;
     }
 }
 
@@ -2181,7 +2192,8 @@ AVStream *av_new_stream(AVFormatContext 
 
     /* default pts setting is MPEG-like */
     av_set_pts_info(st, 33, 1, 90000);
-    st->last_IP_pts = AV_NOPTS_VALUE;
+    st->last_IP_pts[0] =
+    st->last_IP_pts[1] = AV_NOPTS_VALUE;
     for(i=0; i<MAX_REORDER_DELAY+1; i++)
         st->pts_buffer[i]= AV_NOPTS_VALUE;