Browse code

nutenc: mux chapters.

Signed-off-by: Luca Barbato <lu_zero@gentoo.org>

Anton Khirnov authored on 2011/03/16 14:42:43
Showing 2 changed files
... ...
@@ -82,6 +82,10 @@ typedef struct {
82 82
 } StreamContext;
83 83
 
84 84
 typedef struct {
85
+    AVRational *time_base;
86
+} ChapterContext;
87
+
88
+typedef struct {
85 89
     AVFormatContext *avf;
86 90
 //    int written_packet_size;
87 91
 //    int64_t packet_start;
... ...
@@ -90,6 +94,7 @@ typedef struct {
90 90
     const uint8_t *header[128];
91 91
     uint64_t next_startcode;     ///< stores the next startcode if it has already been parsed but the stream is not seekable
92 92
     StreamContext *stream;
93
+    ChapterContext *chapter;
93 94
     unsigned int max_distance;
94 95
     unsigned int time_base_count;
95 96
     int64_t last_syncpoint_pos;
... ...
@@ -241,9 +241,9 @@ static void build_frame_code(AVFormatContext *s){
241 241
     nut->frame_code['N'].flags= FLAG_INVALID;
242 242
 }
243 243
 
244
-static void put_tt(NUTContext *nut, StreamContext *nus, AVIOContext *bc, uint64_t val){
244
+static void put_tt(NUTContext *nut, AVRational *time_base, AVIOContext *bc, uint64_t val){
245 245
     val *= nut->time_base_count;
246
-    val += nus->time_base - nut->time_base;
246
+    val += time_base - nut->time_base;
247 247
     ff_put_v(bc, val);
248 248
 }
249 249
 
... ...
@@ -486,6 +486,34 @@ static int write_streaminfo(NUTContext *nut, AVIOContext *bc, int stream_id){
486 486
     return count;
487 487
 }
488 488
 
489
+static int write_chapter(NUTContext *nut, AVIOContext *bc, int id)
490
+{
491
+    AVIOContext *dyn_bc;
492
+    uint8_t *dyn_buf = NULL;
493
+    AVMetadataTag *t = NULL;
494
+    AVChapter *ch    = nut->avf->chapters[id];
495
+    int ret, dyn_size, count = 0;
496
+
497
+    ret = url_open_dyn_buf(&dyn_bc);
498
+    if (ret < 0)
499
+        return ret;
500
+
501
+    ff_put_v(bc, 0);                                        // stream_id_plus1
502
+    put_s(bc, id + 1);                                      // chapter_id
503
+    put_tt(nut, nut->chapter[id].time_base, bc, ch->start); // chapter_start
504
+    ff_put_v(bc, ch->end - ch->start);                      // chapter_len
505
+
506
+    while ((t = av_metadata_get(ch->metadata, "", t, AV_METADATA_IGNORE_SUFFIX)))
507
+        count += add_info(dyn_bc, t->key, t->value);
508
+
509
+    ff_put_v(bc, count);
510
+
511
+    dyn_size = url_close_dyn_buf(dyn_bc, &dyn_buf);
512
+    avio_write(bc, dyn_buf, dyn_size);
513
+    av_freep(&dyn_buf);
514
+    return 0;
515
+}
516
+
489 517
 static int write_headers(AVFormatContext *avctx, AVIOContext *bc){
490 518
     NUTContext *nut = avctx->priv_data;
491 519
     AVIOContext *dyn_bc;
... ...
@@ -530,6 +558,20 @@ static int write_headers(AVFormatContext *avctx, AVIOContext *bc){
530 530
         }
531 531
     }
532 532
 
533
+    for (i = 0; i < nut->avf->nb_chapters; i++) {
534
+        ret = url_open_dyn_buf(&dyn_bc);
535
+        if (ret < 0)
536
+            return ret;
537
+        ret = write_chapter(nut, dyn_bc, i);
538
+        if (ret < 0) {
539
+            uint8_t *buf;
540
+            url_close_dyn_buf(dyn_bc, &buf);
541
+            av_freep(&buf);
542
+            return ret;
543
+        }
544
+        put_packet(nut, bc, dyn_bc, 1, INFO_STARTCODE);
545
+    }
546
+
533 547
     nut->last_syncpoint_pos= INT_MIN;
534 548
     nut->header_count++;
535 549
     return 0;
... ...
@@ -543,7 +585,9 @@ static int write_header(AVFormatContext *s){
543 543
     nut->avf= s;
544 544
 
545 545
     nut->stream   = av_mallocz(sizeof(StreamContext)*s->nb_streams);
546
-    nut->time_base= av_mallocz(sizeof(AVRational   )*s->nb_streams);
546
+    nut->chapter  = av_mallocz(sizeof(ChapterContext)*s->nb_chapters);
547
+    nut->time_base= av_mallocz(sizeof(AVRational   )*(s->nb_streams +
548
+                                                      s->nb_chapters));
547 549
 
548 550
     for(i=0; i<s->nb_streams; i++){
549 551
         AVStream *st= s->streams[i];
... ...
@@ -570,6 +614,20 @@ static int write_header(AVFormatContext *s){
570 570
         nut->stream[i].max_pts_distance= FFMAX(time_base.den, time_base.num) / time_base.num;
571 571
     }
572 572
 
573
+    for (i = 0; i < s->nb_chapters; i++) {
574
+        AVChapter *ch = s->chapters[i];
575
+
576
+        for (j = 0; j < nut->time_base_count; j++) {
577
+            if (!memcmp(&ch->time_base, &nut->time_base[j], sizeof(AVRational)))
578
+                break;
579
+        }
580
+
581
+        nut->time_base[j] = ch->time_base;
582
+        nut->chapter[i].time_base = &nut->time_base[j];
583
+        if(j == nut->time_base_count)
584
+            nut->time_base_count++;
585
+    }
586
+
573 587
     nut->max_distance = MAX_DISTANCE;
574 588
     build_elision_headers(s);
575 589
     build_frame_code(s);
... ...
@@ -672,7 +730,7 @@ static int write_packet(AVFormatContext *s, AVPacket *pkt){
672 672
         ret = url_open_dyn_buf(&dyn_bc);
673 673
         if(ret < 0)
674 674
             return ret;
675
-        put_tt(nut, nus, dyn_bc, pkt->dts);
675
+        put_tt(nut, nus->time_base, dyn_bc, pkt->dts);
676 676
         ff_put_v(dyn_bc, sp ? (nut->last_syncpoint_pos - sp->pos)>>4 : 0);
677 677
         put_packet(nut, bc, dyn_bc, 1, SYNCPOINT_STARTCODE);
678 678