1/* 2 * nut muxer 3 * Copyright (c) 2004-2007 Michael Niedermayer 4 * 5 * This file is part of FFmpeg. 6 * 7 * FFmpeg is free software; you can redistribute it and/or 8 * modify it under the terms of the GNU Lesser General Public 9 * License as published by the Free Software Foundation; either 10 * version 2.1 of the License, or (at your option) any later version. 11 * 12 * FFmpeg is distributed in the hope that it will be useful, 13 * but WITHOUT ANY WARRANTY; without even the implied warranty of 14 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU 15 * Lesser General Public License for more details. 16 * 17 * You should have received a copy of the GNU Lesser General Public 18 * License along with FFmpeg; if not, write to the Free Software 19 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA 20 */ 21 22#include <stdint.h> 23 24#include "libavutil/intreadwrite.h" 25#include "libavutil/mathematics.h" 26#include "libavutil/tree.h" 27#include "libavutil/dict.h" 28#include "libavutil/avassert.h" 29#include "libavutil/time.h" 30#include "libavutil/opt.h" 31#include "libavcodec/bytestream.h" 32#include "libavcodec/mpegaudiodata.h" 33#include "nut.h" 34#include "internal.h" 35#include "avio_internal.h" 36#include "riff.h" 37 38static int find_expected_header(AVCodecContext *c, int size, int key_frame, 39 uint8_t out[64]) 40{ 41 int sample_rate = c->sample_rate; 42 43 if (size > 4096) 44 return 0; 45 46 AV_WB24(out, 1); 47 48 if (c->codec_id == AV_CODEC_ID_MPEG4) { 49 if (key_frame) { 50 return 3; 51 } else { 52 out[3] = 0xB6; 53 return 4; 54 } 55 } else if (c->codec_id == AV_CODEC_ID_MPEG1VIDEO || 56 c->codec_id == AV_CODEC_ID_MPEG2VIDEO) { 57 return 3; 58 } else if (c->codec_id == AV_CODEC_ID_H264) { 59 return 3; 60 } else if (c->codec_id == AV_CODEC_ID_MP3 || 61 c->codec_id == AV_CODEC_ID_MP2) { 62 int lsf, mpeg25, sample_rate_index, bitrate_index, frame_size; 63 int layer = c->codec_id == AV_CODEC_ID_MP3 ? 3 : 2; 64 unsigned int header = 0xFFF00000; 65 66 lsf = sample_rate < (24000 + 32000) / 2; 67 mpeg25 = sample_rate < (12000 + 16000) / 2; 68 sample_rate <<= lsf + mpeg25; 69 if (sample_rate < (32000 + 44100) / 2) sample_rate_index = 2; 70 else if (sample_rate < (44100 + 48000) / 2) sample_rate_index = 0; 71 else sample_rate_index = 1; 72 73 sample_rate = avpriv_mpa_freq_tab[sample_rate_index] >> (lsf + mpeg25); 74 75 for (bitrate_index = 2; bitrate_index < 30; bitrate_index++) { 76 frame_size = 77 avpriv_mpa_bitrate_tab[lsf][layer - 1][bitrate_index >> 1]; 78 frame_size = (frame_size * 144000) / (sample_rate << lsf) + 79 (bitrate_index & 1); 80 81 if (frame_size == size) 82 break; 83 } 84 85 header |= (!lsf) << 19; 86 header |= (4 - layer) << 17; 87 header |= 1 << 16; //no crc 88 AV_WB32(out, header); 89 if (size <= 0) 90 return 2; //we guess there is no crc, if there is one the user clearly does not care about overhead 91 if (bitrate_index == 30) 92 return -1; //something is wrong ... 93 94 header |= (bitrate_index >> 1) << 12; 95 header |= sample_rate_index << 10; 96 header |= (bitrate_index & 1) << 9; 97 98 return 2; //FIXME actually put the needed ones in build_elision_headers() 99 //return 3; //we guess that the private bit is not set 100//FIXME the above assumptions should be checked, if these turn out false too often something should be done 101 } 102 return 0; 103} 104 105static int find_header_idx(AVFormatContext *s, AVCodecContext *c, int size, int frame_type) 106{ 107 NUTContext *nut = s->priv_data; 108 uint8_t out[64]; 109 int i; 110 int len = find_expected_header(c, size, frame_type, out); 111 112 for (i = 1; i < nut->header_count; i++) { 113 if (len == nut->header_len[i] && !memcmp(out, nut->header[i], len)) { 114 return i; 115 } 116 } 117 118 return 0; 119} 120 121static void build_elision_headers(AVFormatContext *s) 122{ 123 NUTContext *nut = s->priv_data; 124 int i; 125 //FIXME this is lame 126 //FIXME write a 2pass mode to find the maximal headers 127 static const uint8_t headers[][5] = { 128 { 3, 0x00, 0x00, 0x01 }, 129 { 4, 0x00, 0x00, 0x01, 0xB6}, 130 { 2, 0xFF, 0xFA }, //mp3+crc 131 { 2, 0xFF, 0xFB }, //mp3 132 { 2, 0xFF, 0xFC }, //mp2+crc 133 { 2, 0xFF, 0xFD }, //mp2 134 }; 135 136 nut->header_count = 7; 137 for (i = 1; i < nut->header_count; i++) { 138 nut->header_len[i] = headers[i - 1][0]; 139 nut->header[i] = &headers[i - 1][1]; 140 } 141} 142 143static void build_frame_code(AVFormatContext *s) 144{ 145 NUTContext *nut = s->priv_data; 146 int key_frame, index, pred, stream_id; 147 int start = 1; 148 int end = 254; 149 int keyframe_0_esc = s->nb_streams > 2; 150 int pred_table[10]; 151 FrameCode *ft; 152 153 ft = &nut->frame_code[start]; 154 ft->flags = FLAG_CODED; 155 ft->size_mul = 1; 156 ft->pts_delta = 1; 157 start++; 158 159 if (keyframe_0_esc) { 160 /* keyframe = 0 escape */ 161 FrameCode *ft = &nut->frame_code[start]; 162 ft->flags = FLAG_STREAM_ID | FLAG_SIZE_MSB | FLAG_CODED_PTS; 163 ft->size_mul = 1; 164 start++; 165 } 166 167 for (stream_id = 0; stream_id < s->nb_streams; stream_id++) { 168 int start2 = start + (end - start) * stream_id / s->nb_streams; 169 int end2 = start + (end - start) * (stream_id + 1) / s->nb_streams; 170 AVCodecContext *codec = s->streams[stream_id]->codec; 171 int is_audio = codec->codec_type == AVMEDIA_TYPE_AUDIO; 172 int intra_only = /*codec->intra_only || */ is_audio; 173 int pred_count; 174 int frame_size = 0; 175 176 if (codec->codec_type == AVMEDIA_TYPE_AUDIO) { 177 frame_size = av_get_audio_frame_duration(codec, 0); 178 if (codec->codec_id == AV_CODEC_ID_VORBIS && !frame_size) 179 frame_size = 64; 180 } else { 181 AVRational f = av_div_q(codec->time_base, *nut->stream[stream_id].time_base); 182 if (f.den == 1 && f.num>0) 183 frame_size = f.num; 184 } 185 if (!frame_size) 186 frame_size = 1; 187 188 for (key_frame = 0; key_frame < 2; key_frame++) { 189 if (!intra_only || !keyframe_0_esc || key_frame != 0) { 190 FrameCode *ft = &nut->frame_code[start2]; 191 ft->flags = FLAG_KEY * key_frame; 192 ft->flags |= FLAG_SIZE_MSB | FLAG_CODED_PTS; 193 ft->stream_id = stream_id; 194 ft->size_mul = 1; 195 if (is_audio) 196 ft->header_idx = find_header_idx(s, codec, -1, key_frame); 197 start2++; 198 } 199 } 200 201 key_frame = intra_only; 202#if 1 203 if (is_audio) { 204 int frame_bytes = codec->frame_size * (int64_t)codec->bit_rate / 205 (8 * codec->sample_rate); 206 int pts; 207 for (pts = 0; pts < 2; pts++) { 208 for (pred = 0; pred < 2; pred++) { 209 FrameCode *ft = &nut->frame_code[start2]; 210 ft->flags = FLAG_KEY * key_frame; 211 ft->stream_id = stream_id; 212 ft->size_mul = frame_bytes + 2; 213 ft->size_lsb = frame_bytes + pred; 214 ft->pts_delta = pts * frame_size; 215 ft->header_idx = find_header_idx(s, codec, frame_bytes + pred, key_frame); 216 start2++; 217 } 218 } 219 } else { 220 FrameCode *ft = &nut->frame_code[start2]; 221 ft->flags = FLAG_KEY | FLAG_SIZE_MSB; 222 ft->stream_id = stream_id; 223 ft->size_mul = 1; 224 ft->pts_delta = frame_size; 225 start2++; 226 } 227#endif 228 229 if (codec->has_b_frames) { 230 pred_count = 5; 231 pred_table[0] = -2; 232 pred_table[1] = -1; 233 pred_table[2] = 1; 234 pred_table[3] = 3; 235 pred_table[4] = 4; 236 } else if (codec->codec_id == AV_CODEC_ID_VORBIS) { 237 pred_count = 3; 238 pred_table[0] = 2; 239 pred_table[1] = 9; 240 pred_table[2] = 16; 241 } else { 242 pred_count = 1; 243 pred_table[0] = 1; 244 } 245 246 for (pred = 0; pred < pred_count; pred++) { 247 int start3 = start2 + (end2 - start2) * pred / pred_count; 248 int end3 = start2 + (end2 - start2) * (pred + 1) / pred_count; 249 250 pred_table[pred] *= frame_size; 251 252 for (index = start3; index < end3; index++) { 253 FrameCode *ft = &nut->frame_code[index]; 254 ft->flags = FLAG_KEY * key_frame; 255 ft->flags |= FLAG_SIZE_MSB; 256 ft->stream_id = stream_id; 257//FIXME use single byte size and pred from last 258 ft->size_mul = end3 - start3; 259 ft->size_lsb = index - start3; 260 ft->pts_delta = pred_table[pred]; 261 if (is_audio) 262 ft->header_idx = find_header_idx(s, codec, -1, key_frame); 263 } 264 } 265 } 266 memmove(&nut->frame_code['N' + 1], &nut->frame_code['N'], sizeof(FrameCode) * (255 - 'N')); 267 nut->frame_code[0].flags = 268 nut->frame_code[255].flags = 269 nut->frame_code['N'].flags = FLAG_INVALID; 270} 271 272static void put_tt(NUTContext *nut, AVRational *time_base, AVIOContext *bc, uint64_t val) 273{ 274 val *= nut->time_base_count; 275 val += time_base - nut->time_base; 276 ff_put_v(bc, val); 277} 278/** 279 * Store a string as vb. 280 */ 281static void put_str(AVIOContext *bc, const char *string) 282{ 283 int len = strlen(string); 284 285 ff_put_v(bc, len); 286 avio_write(bc, string, len); 287} 288 289static void put_s(AVIOContext *bc, int64_t val) 290{ 291 ff_put_v(bc, 2 * FFABS(val) - (val > 0)); 292} 293 294#ifdef TRACE 295static inline void ff_put_v_trace(AVIOContext *bc, uint64_t v, const char *file, 296 const char *func, int line) 297{ 298 av_log(NULL, AV_LOG_DEBUG, "ff_put_v %5"PRId64" / %"PRIX64" in %s %s:%d\n", v, v, file, func, line); 299 300 ff_put_v(bc, v); 301} 302 303static inline void put_s_trace(AVIOContext *bc, int64_t v, const char *file, const char *func, int line) 304{ 305 av_log(NULL, AV_LOG_DEBUG, "put_s %5"PRId64" / %"PRIX64" in %s %s:%d\n", v, v, file, func, line); 306 307 put_s(bc, v); 308} 309#define ff_put_v(bc, v) ff_put_v_trace(bc, v, __FILE__, __PRETTY_FUNCTION__, __LINE__) 310#define put_s(bc, v) put_s_trace(bc, v, __FILE__, __PRETTY_FUNCTION__, __LINE__) 311#endif 312 313//FIXME remove calculate_checksum 314static void put_packet(NUTContext *nut, AVIOContext *bc, AVIOContext *dyn_bc, 315 int calculate_checksum, uint64_t startcode) 316{ 317 uint8_t *dyn_buf = NULL; 318 int dyn_size = avio_close_dyn_buf(dyn_bc, &dyn_buf); 319 int forw_ptr = dyn_size + 4 * calculate_checksum; 320 321 if (forw_ptr > 4096) 322 ffio_init_checksum(bc, ff_crc04C11DB7_update, 0); 323 avio_wb64(bc, startcode); 324 ff_put_v(bc, forw_ptr); 325 if (forw_ptr > 4096) 326 avio_wl32(bc, ffio_get_checksum(bc)); 327 328 if (calculate_checksum) 329 ffio_init_checksum(bc, ff_crc04C11DB7_update, 0); 330 avio_write(bc, dyn_buf, dyn_size); 331 if (calculate_checksum) 332 avio_wl32(bc, ffio_get_checksum(bc)); 333 334 av_free(dyn_buf); 335} 336 337static void write_mainheader(NUTContext *nut, AVIOContext *bc) 338{ 339 int i, j, tmp_pts, tmp_flags, tmp_stream, tmp_mul, tmp_size, tmp_fields, 340 tmp_head_idx; 341 int64_t tmp_match; 342 343 ff_put_v(bc, nut->version); 344 if (nut->version > 3) 345 ff_put_v(bc, nut->minor_version = 1); 346 ff_put_v(bc, nut->avf->nb_streams); 347 ff_put_v(bc, nut->max_distance); 348 ff_put_v(bc, nut->time_base_count); 349 350 for (i = 0; i < nut->time_base_count; i++) { 351 ff_put_v(bc, nut->time_base[i].num); 352 ff_put_v(bc, nut->time_base[i].den); 353 } 354 355 tmp_pts = 0; 356 tmp_mul = 1; 357 tmp_stream = 0; 358 tmp_match = 1 - (1LL << 62); 359 tmp_head_idx = 0; 360 for (i = 0; i < 256; ) { 361 tmp_fields = 0; 362 tmp_size = 0; 363// tmp_res=0; 364 if (tmp_pts != nut->frame_code[i].pts_delta ) tmp_fields = 1; 365 if (tmp_mul != nut->frame_code[i].size_mul ) tmp_fields = 2; 366 if (tmp_stream != nut->frame_code[i].stream_id ) tmp_fields = 3; 367 if (tmp_size != nut->frame_code[i].size_lsb ) tmp_fields = 4; 368// if (tmp_res != nut->frame_code[i].res ) tmp_fields=5; 369 if (tmp_head_idx != nut->frame_code[i].header_idx) tmp_fields = 8; 370 371 tmp_pts = nut->frame_code[i].pts_delta; 372 tmp_flags = nut->frame_code[i].flags; 373 tmp_stream = nut->frame_code[i].stream_id; 374 tmp_mul = nut->frame_code[i].size_mul; 375 tmp_size = nut->frame_code[i].size_lsb; 376// tmp_res = nut->frame_code[i].res; 377 tmp_head_idx = nut->frame_code[i].header_idx; 378 379 for (j = 0; i < 256; j++, i++) { 380 if (i == 'N') { 381 j--; 382 continue; 383 } 384 if (nut->frame_code[i].pts_delta != tmp_pts || 385 nut->frame_code[i].flags != tmp_flags || 386 nut->frame_code[i].stream_id != tmp_stream || 387 nut->frame_code[i].size_mul != tmp_mul || 388 nut->frame_code[i].size_lsb != tmp_size + j || 389// nut->frame_code[i].res != tmp_res || 390 nut->frame_code[i].header_idx != tmp_head_idx) 391 break; 392 } 393 if (j != tmp_mul - tmp_size) 394 tmp_fields = 6; 395 396 ff_put_v(bc, tmp_flags); 397 ff_put_v(bc, tmp_fields); 398 if (tmp_fields > 0) put_s(bc, tmp_pts); 399 if (tmp_fields > 1) ff_put_v(bc, tmp_mul); 400 if (tmp_fields > 2) ff_put_v(bc, tmp_stream); 401 if (tmp_fields > 3) ff_put_v(bc, tmp_size); 402 if (tmp_fields > 4) ff_put_v(bc, 0 /*tmp_res*/); 403 if (tmp_fields > 5) ff_put_v(bc, j); 404 if (tmp_fields > 6) ff_put_v(bc, tmp_match); 405 if (tmp_fields > 7) ff_put_v(bc, tmp_head_idx); 406 } 407 ff_put_v(bc, nut->header_count - 1); 408 for (i = 1; i < nut->header_count; i++) { 409 ff_put_v(bc, nut->header_len[i]); 410 avio_write(bc, nut->header[i], nut->header_len[i]); 411 } 412 // flags had been effectively introduced in version 4 413 if (nut->version > 3) 414 ff_put_v(bc, nut->flags); 415} 416 417static int write_streamheader(AVFormatContext *avctx, AVIOContext *bc, 418 AVStream *st, int i) 419{ 420 NUTContext *nut = avctx->priv_data; 421 AVCodecContext *codec = st->codec; 422 423 ff_put_v(bc, i); 424 switch (codec->codec_type) { 425 case AVMEDIA_TYPE_VIDEO: ff_put_v(bc, 0); break; 426 case AVMEDIA_TYPE_AUDIO: ff_put_v(bc, 1); break; 427 case AVMEDIA_TYPE_SUBTITLE: ff_put_v(bc, 2); break; 428 default: ff_put_v(bc, 3); break; 429 } 430 ff_put_v(bc, 4); 431 if (codec->codec_tag) { 432 avio_wl32(bc, codec->codec_tag); 433 } else { 434 av_log(avctx, AV_LOG_ERROR, "No codec tag defined for stream %d\n", i); 435 return AVERROR(EINVAL); 436 } 437 438 ff_put_v(bc, nut->stream[i].time_base - nut->time_base); 439 ff_put_v(bc, nut->stream[i].msb_pts_shift); 440 ff_put_v(bc, nut->stream[i].max_pts_distance); 441 ff_put_v(bc, codec->has_b_frames); 442 avio_w8(bc, 0); /* flags: 0x1 - fixed_fps, 0x2 - index_present */ 443 444 ff_put_v(bc, codec->extradata_size); 445 avio_write(bc, codec->extradata, codec->extradata_size); 446 447 switch (codec->codec_type) { 448 case AVMEDIA_TYPE_AUDIO: 449 ff_put_v(bc, codec->sample_rate); 450 ff_put_v(bc, 1); 451 ff_put_v(bc, codec->channels); 452 break; 453 case AVMEDIA_TYPE_VIDEO: 454 ff_put_v(bc, codec->width); 455 ff_put_v(bc, codec->height); 456 457 if (st->sample_aspect_ratio.num <= 0 || 458 st->sample_aspect_ratio.den <= 0) { 459 ff_put_v(bc, 0); 460 ff_put_v(bc, 0); 461 } else { 462 ff_put_v(bc, st->sample_aspect_ratio.num); 463 ff_put_v(bc, st->sample_aspect_ratio.den); 464 } 465 ff_put_v(bc, 0); /* csp type -- unknown */ 466 break; 467 default: 468 break; 469 } 470 return 0; 471} 472 473static int add_info(AVIOContext *bc, const char *type, const char *value) 474{ 475 put_str(bc, type); 476 put_s(bc, -1); 477 put_str(bc, value); 478 return 1; 479} 480 481static int write_globalinfo(NUTContext *nut, AVIOContext *bc) 482{ 483 AVFormatContext *s = nut->avf; 484 AVDictionaryEntry *t = NULL; 485 AVIOContext *dyn_bc; 486 uint8_t *dyn_buf = NULL; 487 int count = 0, dyn_size; 488 int ret = avio_open_dyn_buf(&dyn_bc); 489 if (ret < 0) 490 return ret; 491 492 while ((t = av_dict_get(s->metadata, "", t, AV_DICT_IGNORE_SUFFIX))) 493 count += add_info(dyn_bc, t->key, t->value); 494 495 ff_put_v(bc, 0); //stream_if_plus1 496 ff_put_v(bc, 0); //chapter_id 497 ff_put_v(bc, 0); //timestamp_start 498 ff_put_v(bc, 0); //length 499 500 ff_put_v(bc, count); 501 502 dyn_size = avio_close_dyn_buf(dyn_bc, &dyn_buf); 503 avio_write(bc, dyn_buf, dyn_size); 504 av_free(dyn_buf); 505 return 0; 506} 507 508static int write_streaminfo(NUTContext *nut, AVIOContext *bc, int stream_id) { 509 AVFormatContext *s= nut->avf; 510 AVStream* st = s->streams[stream_id]; 511 AVDictionaryEntry *t = NULL; 512 AVIOContext *dyn_bc; 513 uint8_t *dyn_buf=NULL; 514 int count=0, dyn_size, i; 515 int ret = avio_open_dyn_buf(&dyn_bc); 516 if (ret < 0) 517 return ret; 518 519 while ((t = av_dict_get(st->metadata, "", t, AV_DICT_IGNORE_SUFFIX))) 520 count += add_info(dyn_bc, t->key, t->value); 521 for (i=0; ff_nut_dispositions[i].flag; ++i) { 522 if (st->disposition & ff_nut_dispositions[i].flag) 523 count += add_info(dyn_bc, "Disposition", ff_nut_dispositions[i].str); 524 } 525 if (st->codec->codec_type == AVMEDIA_TYPE_VIDEO) { 526 uint8_t buf[256]; 527 snprintf(buf, sizeof(buf), "%d/%d", st->codec->time_base.den, st->codec->time_base.num); 528 count += add_info(dyn_bc, "r_frame_rate", buf); 529 } 530 dyn_size = avio_close_dyn_buf(dyn_bc, &dyn_buf); 531 532 if (count) { 533 ff_put_v(bc, stream_id + 1); //stream_id_plus1 534 ff_put_v(bc, 0); //chapter_id 535 ff_put_v(bc, 0); //timestamp_start 536 ff_put_v(bc, 0); //length 537 538 ff_put_v(bc, count); 539 540 avio_write(bc, dyn_buf, dyn_size); 541 } 542 543 av_free(dyn_buf); 544 return count; 545} 546 547static int write_chapter(NUTContext *nut, AVIOContext *bc, int id) 548{ 549 AVIOContext *dyn_bc; 550 uint8_t *dyn_buf = NULL; 551 AVDictionaryEntry *t = NULL; 552 AVChapter *ch = nut->avf->chapters[id]; 553 int ret, dyn_size, count = 0; 554 555 ret = avio_open_dyn_buf(&dyn_bc); 556 if (ret < 0) 557 return ret; 558 559 ff_put_v(bc, 0); // stream_id_plus1 560 put_s(bc, id + 1); // chapter_id 561 put_tt(nut, nut->chapter[id].time_base, bc, ch->start); // chapter_start 562 ff_put_v(bc, ch->end - ch->start); // chapter_len 563 564 while ((t = av_dict_get(ch->metadata, "", t, AV_DICT_IGNORE_SUFFIX))) 565 count += add_info(dyn_bc, t->key, t->value); 566 567 ff_put_v(bc, count); 568 569 dyn_size = avio_close_dyn_buf(dyn_bc, &dyn_buf); 570 avio_write(bc, dyn_buf, dyn_size); 571 av_freep(&dyn_buf); 572 return 0; 573} 574 575static int write_index(NUTContext *nut, AVIOContext *bc) { 576 int i; 577 Syncpoint dummy= { .pos= 0 }; 578 Syncpoint *next_node[2] = { NULL }; 579 int64_t startpos = avio_tell(bc); 580 int64_t payload_size; 581 582 put_tt(nut, nut->max_pts_tb, bc, nut->max_pts); 583 584 ff_put_v(bc, nut->sp_count); 585 586 for (i=0; i<nut->sp_count; i++) { 587 av_tree_find(nut->syncpoints, &dummy, (void *) ff_nut_sp_pos_cmp, (void**)next_node); 588 ff_put_v(bc, (next_node[1]->pos >> 4) - (dummy.pos>>4)); 589 dummy.pos = next_node[1]->pos; 590 } 591 592 for (i=0; i<nut->avf->nb_streams; i++) { 593 StreamContext *nus= &nut->stream[i]; 594 int64_t last_pts= -1; 595 int j, k; 596 for (j=0; j<nut->sp_count; j++) { 597 int flag; 598 int n = 0; 599 600 if (j && nus->keyframe_pts[j] == nus->keyframe_pts[j-1]) { 601 av_log(nut->avf, AV_LOG_WARNING, "Multiple keyframes with same PTS\n"); 602 nus->keyframe_pts[j] = AV_NOPTS_VALUE; 603 } 604 605 flag = (nus->keyframe_pts[j] != AV_NOPTS_VALUE) ^ (j+1 == nut->sp_count); 606 for (; j<nut->sp_count && (nus->keyframe_pts[j] != AV_NOPTS_VALUE) == flag; j++) 607 n++; 608 609 ff_put_v(bc, 1 + 2*flag + 4*n); 610 for (k= j - n; k<=j && k<nut->sp_count; k++) { 611 if (nus->keyframe_pts[k] == AV_NOPTS_VALUE) 612 continue; 613 av_assert0(nus->keyframe_pts[k] > last_pts); 614 ff_put_v(bc, nus->keyframe_pts[k] - last_pts); 615 last_pts = nus->keyframe_pts[k]; 616 } 617 } 618 } 619 620 payload_size = avio_tell(bc) - startpos + 8 + 4; 621 622 avio_wb64(bc, 8 + payload_size + av_log2(payload_size) / 7 + 1 + 4*(payload_size > 4096)); 623 624 return 0; 625} 626 627static int write_headers(AVFormatContext *avctx, AVIOContext *bc) 628{ 629 NUTContext *nut = avctx->priv_data; 630 AVIOContext *dyn_bc; 631 int i, ret; 632 633 ff_metadata_conv_ctx(avctx, ff_nut_metadata_conv, NULL); 634 635 ret = avio_open_dyn_buf(&dyn_bc); 636 if (ret < 0) 637 return ret; 638 write_mainheader(nut, dyn_bc); 639 put_packet(nut, bc, dyn_bc, 1, MAIN_STARTCODE); 640 641 for (i = 0; i < nut->avf->nb_streams; i++) { 642 ret = avio_open_dyn_buf(&dyn_bc); 643 if (ret < 0) 644 return ret; 645 ret = write_streamheader(avctx, dyn_bc, nut->avf->streams[i], i); 646 if (ret < 0) 647 return ret; 648 put_packet(nut, bc, dyn_bc, 1, STREAM_STARTCODE); 649 } 650 651 ret = avio_open_dyn_buf(&dyn_bc); 652 if (ret < 0) 653 return ret; 654 write_globalinfo(nut, dyn_bc); 655 put_packet(nut, bc, dyn_bc, 1, INFO_STARTCODE); 656 657 for (i = 0; i < nut->avf->nb_streams; i++) { 658 ret = avio_open_dyn_buf(&dyn_bc); 659 if (ret < 0) 660 return ret; 661 ret = write_streaminfo(nut, dyn_bc, i); 662 if (ret < 0) 663 return ret; 664 if (ret > 0) 665 put_packet(nut, bc, dyn_bc, 1, INFO_STARTCODE); 666 else { 667 uint8_t *buf; 668 avio_close_dyn_buf(dyn_bc, &buf); 669 av_free(buf); 670 } 671 } 672 673 for (i = 0; i < nut->avf->nb_chapters; i++) { 674 ret = avio_open_dyn_buf(&dyn_bc); 675 if (ret < 0) 676 return ret; 677 ret = write_chapter(nut, dyn_bc, i); 678 if (ret < 0) { 679 uint8_t *buf; 680 avio_close_dyn_buf(dyn_bc, &buf); 681 av_freep(&buf); 682 return ret; 683 } 684 put_packet(nut, bc, dyn_bc, 1, INFO_STARTCODE); 685 } 686 687 nut->last_syncpoint_pos = INT_MIN; 688 nut->header_count++; 689 return 0; 690} 691 692static int nut_write_header(AVFormatContext *s) 693{ 694 NUTContext *nut = s->priv_data; 695 AVIOContext *bc = s->pb; 696 int i, j, ret; 697 698 nut->avf = s; 699 700 nut->version = FFMAX(NUT_STABLE_VERSION, 3 + !!nut->flags); 701 if (nut->version > 3 && s->strict_std_compliance > FF_COMPLIANCE_EXPERIMENTAL) { 702 av_log(s, AV_LOG_ERROR, 703 "The additional syncpoint modes require version %d, " 704 "that is currently not finalized, " 705 "please set -f_strict experimental in order to enable it.\n", 706 nut->version); 707 return AVERROR_EXPERIMENTAL; 708 } 709 710 nut->stream = av_calloc(s->nb_streams, sizeof(*nut->stream )); 711 nut->chapter = av_calloc(s->nb_chapters, sizeof(*nut->chapter)); 712 nut->time_base= av_calloc(s->nb_streams + 713 s->nb_chapters, sizeof(*nut->time_base)); 714 if (!nut->stream || !nut->chapter || !nut->time_base) { 715 av_freep(&nut->stream); 716 av_freep(&nut->chapter); 717 av_freep(&nut->time_base); 718 return AVERROR(ENOMEM); 719 } 720 721 for (i = 0; i < s->nb_streams; i++) { 722 AVStream *st = s->streams[i]; 723 int ssize; 724 AVRational time_base; 725 ff_parse_specific_params(st, &time_base.den, &ssize, &time_base.num); 726 727 if (st->codec->codec_type == AVMEDIA_TYPE_AUDIO && st->codec->sample_rate) { 728 time_base = (AVRational) {1, st->codec->sample_rate}; 729 } else { 730 time_base = ff_choose_timebase(s, st, 48000); 731 } 732 733 avpriv_set_pts_info(st, 64, time_base.num, time_base.den); 734 735 for (j = 0; j < nut->time_base_count; j++) 736 if (!memcmp(&time_base, &nut->time_base[j], sizeof(AVRational))) { 737 break; 738 } 739 nut->time_base[j] = time_base; 740 nut->stream[i].time_base = &nut->time_base[j]; 741 if (j == nut->time_base_count) 742 nut->time_base_count++; 743 744 if (INT64_C(1000) * time_base.num >= time_base.den) 745 nut->stream[i].msb_pts_shift = 7; 746 else 747 nut->stream[i].msb_pts_shift = 14; 748 nut->stream[i].max_pts_distance = 749 FFMAX(time_base.den, time_base.num) / time_base.num; 750 } 751 752 for (i = 0; i < s->nb_chapters; i++) { 753 AVChapter *ch = s->chapters[i]; 754 755 for (j = 0; j < nut->time_base_count; j++) 756 if (!memcmp(&ch->time_base, &nut->time_base[j], sizeof(AVRational))) 757 break; 758 759 nut->time_base[j] = ch->time_base; 760 nut->chapter[i].time_base = &nut->time_base[j]; 761 if (j == nut->time_base_count) 762 nut->time_base_count++; 763 } 764 765 nut->max_distance = MAX_DISTANCE; 766 build_elision_headers(s); 767 build_frame_code(s); 768 av_assert0(nut->frame_code['N'].flags == FLAG_INVALID); 769 770 avio_write(bc, ID_STRING, strlen(ID_STRING)); 771 avio_w8(bc, 0); 772 773 if ((ret = write_headers(s, bc)) < 0) 774 return ret; 775 776 if (s->avoid_negative_ts < 0) 777 s->avoid_negative_ts = 1; 778 779 avio_flush(bc); 780 781 return 0; 782} 783 784static int get_needed_flags(NUTContext *nut, StreamContext *nus, FrameCode *fc, 785 AVPacket *pkt) 786{ 787 int flags = 0; 788 789 if (pkt->flags & AV_PKT_FLAG_KEY) 790 flags |= FLAG_KEY; 791 if (pkt->stream_index != fc->stream_id) 792 flags |= FLAG_STREAM_ID; 793 if (pkt->size / fc->size_mul) 794 flags |= FLAG_SIZE_MSB; 795 if (pkt->pts - nus->last_pts != fc->pts_delta) 796 flags |= FLAG_CODED_PTS; 797 if (pkt->side_data_elems && nut->version > 3) 798 flags |= FLAG_SM_DATA; 799 if (pkt->size > 2 * nut->max_distance) 800 flags |= FLAG_CHECKSUM; 801 if (FFABS(pkt->pts - nus->last_pts) > nus->max_pts_distance) 802 flags |= FLAG_CHECKSUM; 803 if (pkt->size < nut->header_len[fc->header_idx] || 804 (pkt->size > 4096 && fc->header_idx) || 805 memcmp(pkt->data, nut->header[fc->header_idx], 806 nut->header_len[fc->header_idx])) 807 flags |= FLAG_HEADER_IDX; 808 809 return flags | (fc->flags & FLAG_CODED); 810} 811 812static int find_best_header_idx(NUTContext *nut, AVPacket *pkt) 813{ 814 int i; 815 int best_i = 0; 816 int best_len = 0; 817 818 if (pkt->size > 4096) 819 return 0; 820 821 for (i = 1; i < nut->header_count; i++) 822 if (pkt->size >= nut->header_len[i] 823 && nut->header_len[i] > best_len 824 && !memcmp(pkt->data, nut->header[i], nut->header_len[i])) { 825 best_i = i; 826 best_len = nut->header_len[i]; 827 } 828 return best_i; 829} 830 831static int write_sm_data(AVFormatContext *s, AVIOContext *bc, AVPacket *pkt, int is_meta) 832{ 833 int ret, i, dyn_size; 834 unsigned flags; 835 AVIOContext *dyn_bc; 836 int sm_data_count = 0; 837 uint8_t tmp[256]; 838 uint8_t *dyn_buf; 839 840 ret = avio_open_dyn_buf(&dyn_bc); 841 if (ret < 0) 842 return ret; 843 844 for (i = 0; i<pkt->side_data_elems; i++) { 845 const uint8_t *data = pkt->side_data[i].data; 846 int size = pkt->side_data[i].size; 847 const uint8_t *data_end = data + size; 848 849 if (is_meta) { 850 if ( pkt->side_data[i].type == AV_PKT_DATA_METADATA_UPDATE 851 || pkt->side_data[i].type == AV_PKT_DATA_STRINGS_METADATA) { 852 if (!size || data[size-1]) { 853 ret = AVERROR(EINVAL); 854 goto fail; 855 } 856 while (data < data_end) { 857 const uint8_t *key = data; 858 const uint8_t *val = data + strlen(key) + 1; 859 860 if(val >= data_end) { 861 ret = AVERROR(EINVAL); 862 goto fail; 863 } 864 put_str(dyn_bc, key); 865 put_s(dyn_bc, -1); 866 put_str(dyn_bc, val); 867 data = val + strlen(val) + 1; 868 sm_data_count++; 869 } 870 } 871 } else { 872 switch (pkt->side_data[i].type) { 873 case AV_PKT_DATA_PALETTE: 874 case AV_PKT_DATA_NEW_EXTRADATA: 875 case AV_PKT_DATA_MATROSKA_BLOCKADDITIONAL: 876 default: 877 if (pkt->side_data[i].type == AV_PKT_DATA_PALETTE) { 878 put_str(dyn_bc, "Palette"); 879 } else if(pkt->side_data[i].type == AV_PKT_DATA_NEW_EXTRADATA) { 880 put_str(dyn_bc, "Extradata"); 881 } else if(pkt->side_data[i].type == AV_PKT_DATA_MATROSKA_BLOCKADDITIONAL) { 882 snprintf(tmp, sizeof(tmp), "CodecSpecificSide%"PRId64"", AV_RB64(data)); 883 put_str(dyn_bc, tmp); 884 } else { 885 snprintf(tmp, sizeof(tmp), "UserData%s-SD-%d", 886 (s->flags & AVFMT_FLAG_BITEXACT) ? "Lavf" : LIBAVFORMAT_IDENT, 887 pkt->side_data[i].type); 888 put_str(dyn_bc, tmp); 889 } 890 put_s(dyn_bc, -2); 891 put_str(dyn_bc, "bin"); 892 ff_put_v(dyn_bc, pkt->side_data[i].size); 893 avio_write(dyn_bc, data, pkt->side_data[i].size); 894 sm_data_count++; 895 break; 896 case AV_PKT_DATA_PARAM_CHANGE: 897 flags = bytestream_get_le32(&data); 898 if (flags & AV_SIDE_DATA_PARAM_CHANGE_CHANNEL_COUNT) { 899 put_str(dyn_bc, "Channels"); 900 put_s(dyn_bc, bytestream_get_le32(&data)); 901 sm_data_count++; 902 } 903 if (flags & AV_SIDE_DATA_PARAM_CHANGE_CHANNEL_LAYOUT) { 904 put_str(dyn_bc, "ChannelLayout"); 905 put_s(dyn_bc, -2); 906 put_str(dyn_bc, "u64"); 907 ff_put_v(bc, 8); 908 avio_write(dyn_bc, data, 8); data+=8; 909 sm_data_count++; 910 } 911 if (flags & AV_SIDE_DATA_PARAM_CHANGE_SAMPLE_RATE) { 912 put_str(dyn_bc, "SampleRate"); 913 put_s(dyn_bc, bytestream_get_le32(&data)); 914 sm_data_count++; 915 } 916 if (flags & AV_SIDE_DATA_PARAM_CHANGE_DIMENSIONS) { 917 put_str(dyn_bc, "Width"); 918 put_s(dyn_bc, bytestream_get_le32(&data)); 919 put_str(dyn_bc, "Height"); 920 put_s(dyn_bc, bytestream_get_le32(&data)); 921 sm_data_count+=2; 922 } 923 break; 924 case AV_PKT_DATA_SKIP_SAMPLES: 925 if (AV_RL32(data)) { 926 put_str(dyn_bc, "SkipStart"); 927 put_s(dyn_bc, (unsigned)AV_RL32(data)); 928 sm_data_count++; 929 } 930 if (AV_RL32(data+4)) { 931 put_str(dyn_bc, "SkipEnd"); 932 put_s(dyn_bc, (unsigned)AV_RL32(data+4)); 933 sm_data_count++; 934 } 935 break; 936 case AV_PKT_DATA_METADATA_UPDATE: 937 case AV_PKT_DATA_STRINGS_METADATA: 938 // belongs into meta, not side data 939 break; 940 } 941 } 942 } 943 944fail: 945 ff_put_v(bc, sm_data_count); 946 dyn_size = avio_close_dyn_buf(dyn_bc, &dyn_buf); 947 avio_write(bc, dyn_buf, dyn_size); 948 av_freep(&dyn_buf); 949 950 return ret; 951} 952 953static int nut_write_packet(AVFormatContext *s, AVPacket *pkt) 954{ 955 NUTContext *nut = s->priv_data; 956 StreamContext *nus = &nut->stream[pkt->stream_index]; 957 AVIOContext *bc = s->pb, *dyn_bc, *sm_bc = NULL; 958 FrameCode *fc; 959 int64_t coded_pts; 960 int best_length, frame_code, flags, needed_flags, i, header_idx; 961 int best_header_idx; 962 int key_frame = !!(pkt->flags & AV_PKT_FLAG_KEY); 963 int store_sp = 0; 964 int ret = 0; 965 int sm_size = 0; 966 int data_size = pkt->size; 967 uint8_t *sm_buf = NULL; 968 969 if (pkt->pts < 0) { 970 av_log(s, AV_LOG_ERROR, 971 "Negative pts not supported stream %d, pts %"PRId64"\n", 972 pkt->stream_index, pkt->pts); 973 if (pkt->pts == AV_NOPTS_VALUE) 974 av_log(s, AV_LOG_ERROR, "Try to enable the genpts flag\n"); 975 return AVERROR(EINVAL); 976 } 977 978 if (pkt->side_data_elems && nut->version > 3) { 979 ret = avio_open_dyn_buf(&sm_bc); 980 if (ret < 0) 981 return ret; 982 ret = write_sm_data(s, sm_bc, pkt, 0); 983 if (ret >= 0) 984 ret = write_sm_data(s, sm_bc, pkt, 1); 985 sm_size = avio_close_dyn_buf(sm_bc, &sm_buf); 986 if (ret < 0) 987 goto fail; 988 data_size += sm_size; 989 } 990 991 if (1LL << (20 + 3 * nut->header_count) <= avio_tell(bc)) 992 write_headers(s, bc); 993 994 if (key_frame && !(nus->last_flags & FLAG_KEY)) 995 store_sp = 1; 996 997 if (data_size + 30 /*FIXME check*/ + avio_tell(bc) >= nut->last_syncpoint_pos + nut->max_distance) 998 store_sp = 1; 999 1000//FIXME: Ensure store_sp is 1 in the first place. 1001 1002 if (store_sp && 1003 (!(nut->flags & NUT_PIPE) || nut->last_syncpoint_pos == INT_MIN)) { 1004 int64_t sp_pos = INT64_MAX; 1005 1006 ff_nut_reset_ts(nut, *nus->time_base, pkt->dts); 1007 for (i = 0; i < s->nb_streams; i++) { 1008 AVStream *st = s->streams[i]; 1009 int64_t dts_tb = av_rescale_rnd(pkt->dts, 1010 nus->time_base->num * (int64_t)nut->stream[i].time_base->den, 1011 nus->time_base->den * (int64_t)nut->stream[i].time_base->num, 1012 AV_ROUND_DOWN); 1013 int index = av_index_search_timestamp(st, dts_tb, 1014 AVSEEK_FLAG_BACKWARD); 1015 if (index >= 0) { 1016 sp_pos = FFMIN(sp_pos, st->index_entries[index].pos); 1017 if (!nut->write_index && 2*index > st->nb_index_entries) { 1018 memmove(st->index_entries, 1019 st->index_entries + index, 1020 sizeof(*st->index_entries) * (st->nb_index_entries - index)); 1021 st->nb_index_entries -= index; 1022 } 1023 } 1024 } 1025 1026 nut->last_syncpoint_pos = avio_tell(bc); 1027 ret = avio_open_dyn_buf(&dyn_bc); 1028 if (ret < 0) 1029 goto fail; 1030 put_tt(nut, nus->time_base, dyn_bc, pkt->dts); 1031 ff_put_v(dyn_bc, sp_pos != INT64_MAX ? (nut->last_syncpoint_pos - sp_pos) >> 4 : 0); 1032 1033 if (nut->flags & NUT_BROADCAST) { 1034 put_tt(nut, nus->time_base, dyn_bc, 1035 av_rescale_q(av_gettime(), AV_TIME_BASE_Q, *nus->time_base)); 1036 } 1037 put_packet(nut, bc, dyn_bc, 1, SYNCPOINT_STARTCODE); 1038 1039 if (nut->write_index) { 1040 if ((ret = ff_nut_add_sp(nut, nut->last_syncpoint_pos, 0 /*unused*/, pkt->dts)) < 0) 1041 goto fail; 1042 1043 if ((1ll<<60) % nut->sp_count == 0) 1044 for (i=0; i<s->nb_streams; i++) { 1045 int j; 1046 StreamContext *nus = &nut->stream[i]; 1047 av_reallocp_array(&nus->keyframe_pts, 2*nut->sp_count, sizeof(*nus->keyframe_pts)); 1048 if (!nus->keyframe_pts) { 1049 ret = AVERROR(ENOMEM); 1050 goto fail; 1051 } 1052 for (j=nut->sp_count == 1 ? 0 : nut->sp_count; j<2*nut->sp_count; j++) 1053 nus->keyframe_pts[j] = AV_NOPTS_VALUE; 1054 } 1055 } 1056 } 1057 av_assert0(nus->last_pts != AV_NOPTS_VALUE); 1058 1059 coded_pts = pkt->pts & ((1 << nus->msb_pts_shift) - 1); 1060 if (ff_lsb2full(nus, coded_pts) != pkt->pts) 1061 coded_pts = pkt->pts + (1 << nus->msb_pts_shift); 1062 1063 best_header_idx = find_best_header_idx(nut, pkt); 1064 1065 best_length = INT_MAX; 1066 frame_code = -1; 1067 for (i = 0; i < 256; i++) { 1068 int length = 0; 1069 FrameCode *fc = &nut->frame_code[i]; 1070 int flags = fc->flags; 1071 1072 if (flags & FLAG_INVALID) 1073 continue; 1074 needed_flags = get_needed_flags(nut, nus, fc, pkt); 1075 1076 if (flags & FLAG_CODED) { 1077 length++; 1078 flags = needed_flags; 1079 } 1080 1081 if ((flags & needed_flags) != needed_flags) 1082 continue; 1083 1084 if ((flags ^ needed_flags) & FLAG_KEY) 1085 continue; 1086 1087 if (flags & FLAG_STREAM_ID) 1088 length += ff_get_v_length(pkt->stream_index); 1089 1090 if (data_size % fc->size_mul != fc->size_lsb) 1091 continue; 1092 if (flags & FLAG_SIZE_MSB) 1093 length += ff_get_v_length(data_size / fc->size_mul); 1094 1095 if (flags & FLAG_CHECKSUM) 1096 length += 4; 1097 1098 if (flags & FLAG_CODED_PTS) 1099 length += ff_get_v_length(coded_pts); 1100 1101 if ( (flags & FLAG_CODED) 1102 && nut->header_len[best_header_idx] > nut->header_len[fc->header_idx] + 1) { 1103 flags |= FLAG_HEADER_IDX; 1104 } 1105 1106 if (flags & FLAG_HEADER_IDX) { 1107 length += 1 - nut->header_len[best_header_idx]; 1108 } else { 1109 length -= nut->header_len[fc->header_idx]; 1110 } 1111 1112 length *= 4; 1113 length += !(flags & FLAG_CODED_PTS); 1114 length += !(flags & FLAG_CHECKSUM); 1115 1116 if (length < best_length) { 1117 best_length = length; 1118 frame_code = i; 1119 } 1120 } 1121 av_assert0(frame_code != -1); 1122 fc = &nut->frame_code[frame_code]; 1123 flags = fc->flags; 1124 needed_flags = get_needed_flags(nut, nus, fc, pkt); 1125 header_idx = fc->header_idx; 1126 1127 ffio_init_checksum(bc, ff_crc04C11DB7_update, 0); 1128 avio_w8(bc, frame_code); 1129 if (flags & FLAG_CODED) { 1130 ff_put_v(bc, (flags ^ needed_flags) & ~(FLAG_CODED)); 1131 flags = needed_flags; 1132 } 1133 if (flags & FLAG_STREAM_ID) ff_put_v(bc, pkt->stream_index); 1134 if (flags & FLAG_CODED_PTS) ff_put_v(bc, coded_pts); 1135 if (flags & FLAG_SIZE_MSB ) ff_put_v(bc, data_size / fc->size_mul); 1136 if (flags & FLAG_HEADER_IDX) ff_put_v(bc, header_idx = best_header_idx); 1137 1138 if (flags & FLAG_CHECKSUM) avio_wl32(bc, ffio_get_checksum(bc)); 1139 else ffio_get_checksum(bc); 1140 1141 if (flags & FLAG_SM_DATA) { 1142 avio_write(bc, sm_buf, sm_size); 1143 } 1144 avio_write(bc, pkt->data + nut->header_len[header_idx], pkt->size - nut->header_len[header_idx]); 1145 1146 nus->last_flags = flags; 1147 nus->last_pts = pkt->pts; 1148 1149 //FIXME just store one per syncpoint 1150 if (flags & FLAG_KEY && !(nut->flags & NUT_PIPE)) { 1151 av_add_index_entry( 1152 s->streams[pkt->stream_index], 1153 nut->last_syncpoint_pos, 1154 pkt->pts, 1155 0, 1156 0, 1157 AVINDEX_KEYFRAME); 1158 if (nus->keyframe_pts && nus->keyframe_pts[nut->sp_count] == AV_NOPTS_VALUE) 1159 nus->keyframe_pts[nut->sp_count] = pkt->pts; 1160 } 1161 1162 if (!nut->max_pts_tb || av_compare_ts(nut->max_pts, *nut->max_pts_tb, pkt->pts, *nus->time_base) < 0) { 1163 nut->max_pts = pkt->pts; 1164 nut->max_pts_tb = nus->time_base; 1165 } 1166 1167fail: 1168 av_freep(&sm_buf); 1169 1170 return ret; 1171} 1172 1173static int nut_write_trailer(AVFormatContext *s) 1174{ 1175 NUTContext *nut = s->priv_data; 1176 AVIOContext *bc = s->pb, *dyn_bc; 1177 int i, ret; 1178 1179 while (nut->header_count < 3) 1180 write_headers(s, bc); 1181 1182 ret = avio_open_dyn_buf(&dyn_bc); 1183 if (ret >= 0 && nut->sp_count) { 1184 av_assert1(nut->write_index); 1185 write_index(nut, dyn_bc); 1186 put_packet(nut, bc, dyn_bc, 1, INDEX_STARTCODE); 1187 } 1188 1189 ff_nut_free_sp(nut); 1190 for (i=0; i<s->nb_streams; i++) 1191 av_freep(&nut->stream[i].keyframe_pts); 1192 1193 av_freep(&nut->stream); 1194 av_freep(&nut->chapter); 1195 av_freep(&nut->time_base); 1196 1197 return 0; 1198} 1199 1200#define OFFSET(x) offsetof(NUTContext, x) 1201#define E AV_OPT_FLAG_ENCODING_PARAM 1202static const AVOption options[] = { 1203 { "syncpoints", "NUT syncpoint behaviour", OFFSET(flags), AV_OPT_TYPE_FLAGS, {.i64 = 0}, INT_MIN, INT_MAX, E, "syncpoints" }, 1204 { "default", "", 0, AV_OPT_TYPE_CONST, {.i64 = 0}, INT_MIN, INT_MAX, E, "syncpoints" }, 1205 { "none", "Disable syncpoints, low overhead and unseekable", 0, AV_OPT_TYPE_CONST, {.i64 = NUT_PIPE}, INT_MIN, INT_MAX, E, "syncpoints" }, 1206 { "timestamped", "Extend syncpoints with a wallclock timestamp", 0, AV_OPT_TYPE_CONST, {.i64 = NUT_BROADCAST}, INT_MIN, INT_MAX, E, "syncpoints" }, 1207 { "write_index", "Write index", OFFSET(write_index), AV_OPT_TYPE_INT, {.i64 = 1}, 0, 1, E, }, 1208 { NULL }, 1209}; 1210 1211static const AVClass class = { 1212 .class_name = "nutenc", 1213 .item_name = av_default_item_name, 1214 .option = options, 1215 .version = LIBAVUTIL_VERSION_INT, 1216}; 1217 1218AVOutputFormat ff_nut_muxer = { 1219 .name = "nut", 1220 .long_name = NULL_IF_CONFIG_SMALL("NUT"), 1221 .mime_type = "video/x-nut", 1222 .extensions = "nut", 1223 .priv_data_size = sizeof(NUTContext), 1224 .audio_codec = CONFIG_LIBVORBIS ? AV_CODEC_ID_VORBIS : 1225 CONFIG_LIBMP3LAME ? AV_CODEC_ID_MP3 : AV_CODEC_ID_MP2, 1226 .video_codec = AV_CODEC_ID_MPEG4, 1227 .write_header = nut_write_header, 1228 .write_packet = nut_write_packet, 1229 .write_trailer = nut_write_trailer, 1230 .flags = AVFMT_GLOBALHEADER | AVFMT_VARIABLE_FPS, 1231 .codec_tag = ff_nut_codec_tags, 1232 .priv_class = &class, 1233}; 1234