cosmetics: reindent

[ffmpeg] / libavformat / nutenc.c

diff --git a/libavformat/nutenc.c b/libavformat/nutenc.c
index 9248ef5bd9c66a863838993f162d4a48001be9da..b28566100cc2779b2f31cf83688d3b2911161f0f 100644 (file)
--- a/libavformat/nutenc.c
+++ b/libavformat/nutenc.c
@@ -22,8 +22,6 @@
 #include "nut.h"
 #include "tree.h"
 
-#define TRACE
-
 static void build_frame_code(AVFormatContext *s){
     NUTContext *nut = s->priv_data;
     int key_frame, index, pred, stream_id;
@@ -162,7 +160,7 @@ static void put_t(NUTContext *nut, StreamContext *nus, ByteIOContext *bc, uint64
 }
 
 /**
- * stores a string as vb.
+ * Stores a string as vb.
  */
 static void put_str(ByteIOContext *bc, const char *string){
     int len= strlen(string);
@@ -216,7 +214,7 @@ static void put_packet(NUTContext *nut, ByteIOContext *bc, ByteIOContext *dyn_bc
 static void write_mainheader(NUTContext *nut, ByteIOContext *bc){
     int i, j, tmp_pts, tmp_flags, tmp_stream, tmp_mul, tmp_size, tmp_fields;
 
-    put_v(bc, 2); /* version */
+    put_v(bc, 3); /* version */
     put_v(bc, nut->avf->nb_streams);
     put_v(bc, MAX_DISTANCE);
     put_v(bc, nut->time_base_count);
@@ -303,8 +301,14 @@ static int write_streamheader(NUTContext *nut, ByteIOContext *bc, AVCodecContext
     case CODEC_TYPE_VIDEO:
         put_v(bc, codec->width);
         put_v(bc, codec->height);
-        put_v(bc, codec->sample_aspect_ratio.num);
-        put_v(bc, codec->sample_aspect_ratio.den);
+
+        if(codec->sample_aspect_ratio.num<=0 || codec->sample_aspect_ratio.den<=0){
+            put_v(bc, 0);
+            put_v(bc, 0);
+        }else{
+            put_v(bc, codec->sample_aspect_ratio.num);
+            put_v(bc, codec->sample_aspect_ratio.den);
+        }
         put_v(bc, 0); /* csp type -- unknown */
         break;
     default:
@@ -313,6 +317,39 @@ static int write_streamheader(NUTContext *nut, ByteIOContext *bc, AVCodecContext
     return 0;
 }
 
+static int add_info(ByteIOContext *bc, char *type, char *value){
+    put_str(bc, type);
+    put_s(bc, -1);
+    put_str(bc, value);
+    return 1;
+}
+
+static void write_globalinfo(NUTContext *nut, ByteIOContext *bc){
+    AVFormatContext *s= nut->avf;
+    ByteIOContext dyn_bc;
+    uint8_t *dyn_buf=NULL;
+    int count=0, dyn_size;
+
+    url_open_dyn_buf(&dyn_bc);
+
+    if(s->title    [0]) count+= add_info(&dyn_bc, "Title"    , s->title);
+    if(s->author   [0]) count+= add_info(&dyn_bc, "Author"   , s->author);
+    if(s->copyright[0]) count+= add_info(&dyn_bc, "Copyright", s->copyright);
+    if(!(s->streams[0]->codec->flags & CODEC_FLAG_BITEXACT))
+                        count+= add_info(&dyn_bc, "Encoder"  , LIBAVFORMAT_IDENT);
+
+    put_v(bc, 0); //stream_if_plus1
+    put_v(bc, 0); //chapter_id
+    put_v(bc, 0); //timestamp_start
+    put_v(bc, 0); //length
+
+    put_v(bc, count);
+
+    dyn_size= url_close_dyn_buf(&dyn_bc, &dyn_buf);
+    put_buffer(bc, dyn_buf, dyn_size);
+    av_free(dyn_buf);
+}
+
 static void write_headers(NUTContext *nut, ByteIOContext *bc){
     ByteIOContext dyn_bc;
     int i;
@@ -328,6 +365,13 @@ static void write_headers(NUTContext *nut, ByteIOContext *bc){
         write_streamheader(nut, &dyn_bc, codec, i);
         put_packet(nut, bc, &dyn_bc, 1, STREAM_STARTCODE);
     }
+
+    url_open_dyn_buf(&dyn_bc);
+    write_globalinfo(nut, &dyn_bc);
+    put_packet(nut, bc, &dyn_bc, 1, INFO_STARTCODE);
+
+    nut->last_syncpoint_pos= INT_MIN;
+    nut->header_count++;
 }
 
 static int write_header(AVFormatContext *s){
@@ -335,10 +379,6 @@ static int write_header(AVFormatContext *s){
     ByteIOContext *bc = &s->pb;
     int i, j;
 
-    if(s->streams[0]->codec->strict_std_compliance > FF_COMPLIANCE_EXPERIMENTAL){
-        return -1;
-    }
-
     nut->avf= s;
 
     nut->stream   = av_mallocz(sizeof(StreamContext)*s->nb_streams);
@@ -379,7 +419,7 @@ static int write_header(AVFormatContext *s){
 
     put_flush_packet(bc);
 
-    //FIXME info header, header repeation, index
+    //FIXME index
 
     return 0;
 }
@@ -395,7 +435,7 @@ static int get_needed_flags(NUTContext *nut, StreamContext *nus, FrameCode *fc,
     if(FFABS(pkt->pts - nus->last_pts)
                          > nus->max_pts_distance) flags |= FLAG_CHECKSUM;
 
-    return flags;
+    return flags | (fc->flags & FLAG_CODED);
 }
 
 static int write_packet(AVFormatContext *s, AVPacket *pkt){
@@ -408,13 +448,16 @@ static int write_packet(AVFormatContext *s, AVPacket *pkt){
     int key_frame = !!(pkt->flags & PKT_FLAG_KEY);
     int store_sp=0;
 
+    if(1LL<<(20+3*nut->header_count) <= url_ftell(bc))
+        write_headers(nut, bc);
+
     if(key_frame && !!(nus->last_flags & FLAG_KEY))
         store_sp= 1;
 
     if(pkt->size + 30/*FIXME check*/ + url_ftell(bc) >= nut->last_syncpoint_pos + nut->max_distance)
         store_sp= 1;
 
-//FIXME ensure store_sp is 1 for the first thing
+//FIXME: Ensure store_sp is 1 in the first place.
 
     if(store_sp){
         syncpoint_t *sp, dummy= {.pos= INT64_MAX};
@@ -455,8 +498,7 @@ static int write_packet(AVFormatContext *s, AVPacket *pkt){
 
         if(flags & FLAG_CODED){
             length++;
-            flags &= ~needed_flags;
-            flags |=  needed_flags;
+            flags = needed_flags;
         }
 
         if((flags & needed_flags) != needed_flags)
@@ -526,7 +568,8 @@ static int write_trailer(AVFormatContext *s){
     NUTContext *nut= s->priv_data;
     ByteIOContext *bc= &s->pb;
 
-    write_headers(nut, bc);
+    while(nut->header_count<3)
+        write_headers(nut, bc);
     put_flush_packet(bc);
 
     return 0;