Signed-off-by: Luca Barbato <lu_zero@gentoo.org>
| ... | ... |
@@ -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 |
|