Browse code

asfenc: mux chapters in ASF files using an ASF "marker" section

ASF markers only have a start time, so we lose the chapter end times,
but that is ASF for you

Signed-off-by: Vladimir Pantelic <vladoman@gmail.com>
Signed-off-by: Anton Khirnov <anton@khirnov.net>

Vladimir Pantelic authored on 2013/09/20 00:50:38
Showing 3 changed files
... ...
@@ -36,6 +36,7 @@ version 10:
36 36
 - WebP decoder
37 37
 - Error Resilient AAC syntax (ER AAC LC) decoding
38 38
 - Low Delay AAC (ER AAC LD) decoding
39
+- mux chapters in ASF files
39 40
 
40 41
 
41 42
 version 9:
... ...
@@ -20,6 +20,7 @@
20 20
  */
21 21
 
22 22
 #include "libavutil/dict.h"
23
+#include "libavutil/mathematics.h"
23 24
 #include "avformat.h"
24 25
 #include "avio_internal.h"
25 26
 #include "internal.h"
... ...
@@ -287,6 +288,64 @@ static int64_t unix_to_file_time(int ti)
287 287
     return t;
288 288
 }
289 289
 
290
+static int32_t get_send_time(ASFContext *asf, int64_t pres_time, uint64_t *offset)
291
+{
292
+    int i;
293
+    int32_t send_time = 0;
294
+    *offset = asf->data_offset + DATA_HEADER_SIZE;
295
+    for (i = 0; i < asf->nb_index_count; i++) {
296
+        if (pres_time <= asf->index_ptr[i].send_time)
297
+            break;
298
+        send_time = asf->index_ptr[i].send_time;
299
+        *offset   = asf->index_ptr[i].offset;
300
+    }
301
+
302
+    return send_time / 10000;
303
+}
304
+
305
+static int asf_write_markers(AVFormatContext *s)
306
+{
307
+    ASFContext *asf = s->priv_data;
308
+    AVIOContext *pb = s->pb;
309
+    int i;
310
+    AVRational scale = {1, 10000000};
311
+    int64_t hpos = put_header(pb, &ff_asf_marker_header);
312
+
313
+    put_guid(pb, &ff_asf_reserved_4);  // ASF spec mandates this reserved value
314
+    avio_wl32(pb, s->nb_chapters);     // markers count
315
+    avio_wl16(pb, 0);                  // ASF spec mandates 0 for this
316
+    avio_wl16(pb, 0);                  // name length 0, no name given
317
+
318
+    for (i = 0; i < s->nb_chapters; i++) {
319
+        AVChapter *c = s->chapters[i];
320
+        AVDictionaryEntry *t = av_dict_get(c->metadata, "title", NULL, 0);
321
+        int64_t pres_time = av_rescale_q(c->start, c->time_base, scale);
322
+        uint64_t offset;
323
+        int32_t send_time = get_send_time(asf, pres_time, &offset);
324
+        int len = 0;
325
+        uint8_t *buf;
326
+        AVIOContext *dyn_buf;
327
+        if (t) {
328
+            if (avio_open_dyn_buf(&dyn_buf) < 0)
329
+                return AVERROR(ENOMEM);
330
+            avio_put_str16le(dyn_buf, t->value);
331
+            len = avio_close_dyn_buf(dyn_buf, &buf);
332
+        }
333
+        avio_wl64(pb, offset);            // offset of the packet with send_time
334
+        avio_wl64(pb, pres_time + PREROLL_TIME * 10000); // presentation time
335
+        avio_wl16(pb, 12 + len);          // entry length
336
+        avio_wl32(pb, send_time);         // send time
337
+        avio_wl32(pb, 0);                 // flags, should be 0
338
+        avio_wl32(pb, len / 2);           // marker desc length in WCHARS!
339
+        if (t) {
340
+            avio_write(pb, buf, len);     // marker desc
341
+            av_freep(&buf);
342
+        }
343
+    }
344
+    end_header(pb, hpos);
345
+    return 0;
346
+}
347
+
290 348
 /* write the header (used two times if non streamed) */
291 349
 static int asf_write_header1(AVFormatContext *s, int64_t file_size,
292 350
                              int64_t data_chunk_size)
... ...
@@ -388,7 +447,12 @@ static int asf_write_header1(AVFormatContext *s, int64_t file_size,
388 388
         }
389 389
         end_header(pb, hpos);
390 390
     }
391
-
391
+    /* chapters using ASF markers */
392
+    if (!asf->is_streamed && s->nb_chapters) {
393
+        int ret;
394
+        if (ret = asf_write_markers(s))
395
+            return ret;
396
+    }
392 397
     /* stream headers */
393 398
     for (n = 0; n < s->nb_streams; n++) {
394 399
         int64_t es_pos;
... ...
@@ -31,7 +31,7 @@
31 31
 
32 32
 #define LIBAVFORMAT_VERSION_MAJOR 55
33 33
 #define LIBAVFORMAT_VERSION_MINOR  5
34
-#define LIBAVFORMAT_VERSION_MICRO  0
34
+#define LIBAVFORMAT_VERSION_MICRO  1
35 35
 
36 36
 #define LIBAVFORMAT_VERSION_INT AV_VERSION_INT(LIBAVFORMAT_VERSION_MAJOR, \
37 37
                                                LIBAVFORMAT_VERSION_MINOR, \