changeset 2437:36350b4c94fa libavformat

Start a new cluster every 5 MB or 5 seconds
author conrad
date Wed, 05 Sep 2007 00:23:00 +0000
parents fab3556370d0
children 5e587452dc48
files matroskaenc.c
diffstat 1 files changed, 12 insertions(+), 1 deletions(-) [+]
line wrap: on
line diff
--- a/matroskaenc.c	Wed Sep 05 00:22:58 2007 +0000
+++ b/matroskaenc.c	Wed Sep 05 00:23:00 2007 +0000
@@ -27,6 +27,7 @@
 typedef struct MatroskaMuxContext {
     offset_t    segment;
     offset_t    cluster;
+    uint64_t    cluster_pts;
 } MatroskaMuxContext;
 
 static void put_ebml_id(ByteIOContext *pb, unsigned int id)
@@ -249,18 +250,28 @@
 
     mkv->cluster = start_ebml_master(pb, MATROSKA_ID_CLUSTER);
     put_ebml_uint(pb, MATROSKA_ID_CLUSTERTIMECODE, 0);
+    mkv->cluster_pts = 0;
 
     return 0;
 }
 
 static int mkv_write_packet(AVFormatContext *s, AVPacket *pkt)
 {
+    MatroskaMuxContext *mkv = s->priv_data;
     ByteIOContext *pb = &s->pb;
     offset_t block;
 
+    // start a new cluster every 5 MB or 5 sec
+    if (url_ftell(pb) > mkv->cluster + 5*1024*1024 || pkt->pts > mkv->cluster_pts + 5000) {
+        end_ebml_master(pb, mkv->cluster);
+        mkv->cluster = start_ebml_master(pb, MATROSKA_ID_CLUSTER);
+        put_ebml_uint(pb, MATROSKA_ID_CLUSTERTIMECODE, pkt->pts);
+        mkv->cluster_pts = pkt->pts;
+    }
+
     block = start_ebml_master(pb, MATROSKA_ID_SIMPLEBLOCK);
     put_byte(pb, 0x80 | pkt->stream_index);     // this assumes stream_index is less than 127
-    put_be16(pb, pkt->pts);
+    put_be16(pb, pkt->pts - mkv->cluster_pts);
     put_byte(pb, !!(pkt->flags & PKT_FLAG_KEY));
     put_buffer(pb, pkt->data, pkt->size);
     end_ebml_master(pb, block);