mirror of https://github.com/FFmpeg/FFmpeg.git
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
752 lines
24 KiB
752 lines
24 KiB
/* |
|
* Copyright (c) 2015 Martin Storsjo |
|
* |
|
* This file is part of FFmpeg. |
|
* |
|
* FFmpeg is free software; you can redistribute it and/or |
|
* modify it under the terms of the GNU Lesser General Public |
|
* License as published by the Free Software Foundation; either |
|
* version 2.1 of the License, or (at your option) any later version. |
|
* |
|
* FFmpeg is distributed in the hope that it will be useful, |
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of |
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU |
|
* Lesser General Public License for more details. |
|
* |
|
* You should have received a copy of the GNU Lesser General Public |
|
* License along with FFmpeg; if not, write to the Free Software |
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA |
|
*/ |
|
|
|
#include "config.h" |
|
|
|
#include "libavutil/intreadwrite.h" |
|
#include "libavutil/mathematics.h" |
|
#include "libavutil/md5.h" |
|
|
|
#include "libavformat/avformat.h" |
|
|
|
#if HAVE_UNISTD_H |
|
#include <unistd.h> |
|
#endif |
|
|
|
#if !HAVE_GETOPT |
|
#include "compat/getopt.c" |
|
#endif |
|
|
|
#define HASH_SIZE 16 |
|
|
|
static const uint8_t h264_extradata[] = { |
|
0x01, 0x4d, 0x40, 0x1e, 0xff, 0xe1, 0x00, 0x02, 0x67, 0x4d, 0x01, 0x00, 0x02, 0x68, 0xef |
|
}; |
|
static const uint8_t aac_extradata[] = { |
|
0x12, 0x10 |
|
}; |
|
|
|
|
|
static const char *format = "mp4"; |
|
AVFormatContext *ctx; |
|
uint8_t iobuf[32768]; |
|
AVDictionary *opts; |
|
|
|
int write_file; |
|
const char *cur_name; |
|
FILE* out; |
|
int out_size; |
|
struct AVMD5* md5; |
|
uint8_t hash[HASH_SIZE]; |
|
|
|
AVStream *video_st, *audio_st; |
|
int64_t audio_dts, video_dts; |
|
|
|
int bframes; |
|
int64_t duration; |
|
int64_t audio_duration; |
|
int frames; |
|
int gop_size; |
|
int64_t next_p_pts; |
|
enum AVPictureType last_picture; |
|
int skip_write; |
|
int skip_write_audio; |
|
int clear_duration; |
|
int force_iobuf_size; |
|
int do_interleave; |
|
int fake_pkt_duration; |
|
|
|
int num_warnings; |
|
|
|
int check_faults; |
|
|
|
|
|
static void count_warnings(void *avcl, int level, const char *fmt, va_list vl) |
|
{ |
|
if (level == AV_LOG_WARNING) |
|
num_warnings++; |
|
} |
|
|
|
static void init_count_warnings(void) |
|
{ |
|
av_log_set_callback(count_warnings); |
|
num_warnings = 0; |
|
} |
|
|
|
static void reset_count_warnings(void) |
|
{ |
|
av_log_set_callback(av_log_default_callback); |
|
} |
|
|
|
static int io_write(void *opaque, uint8_t *buf, int size) |
|
{ |
|
out_size += size; |
|
av_md5_update(md5, buf, size); |
|
if (out) |
|
fwrite(buf, 1, size, out); |
|
return size; |
|
} |
|
|
|
static int io_write_data_type(void *opaque, uint8_t *buf, int size, |
|
enum AVIODataMarkerType type, int64_t time) |
|
{ |
|
char timebuf[30], content[5] = { 0 }; |
|
const char *str; |
|
switch (type) { |
|
case AVIO_DATA_MARKER_HEADER: str = "header"; break; |
|
case AVIO_DATA_MARKER_SYNC_POINT: str = "sync"; break; |
|
case AVIO_DATA_MARKER_BOUNDARY_POINT: str = "boundary"; break; |
|
case AVIO_DATA_MARKER_UNKNOWN: str = "unknown"; break; |
|
case AVIO_DATA_MARKER_TRAILER: str = "trailer"; break; |
|
} |
|
if (time == AV_NOPTS_VALUE) |
|
snprintf(timebuf, sizeof(timebuf), "nopts"); |
|
else |
|
snprintf(timebuf, sizeof(timebuf), "%"PRId64, time); |
|
// There can be multiple header/trailer callbacks, only log the box type |
|
// for header at out_size == 0 |
|
if (type != AVIO_DATA_MARKER_UNKNOWN && |
|
type != AVIO_DATA_MARKER_TRAILER && |
|
(type != AVIO_DATA_MARKER_HEADER || out_size == 0) && |
|
size >= 8) |
|
memcpy(content, &buf[4], 4); |
|
else |
|
snprintf(content, sizeof(content), "-"); |
|
printf("write_data len %d, time %s, type %s atom %s\n", size, timebuf, str, content); |
|
return io_write(opaque, buf, size); |
|
} |
|
|
|
static void init_out(const char *name) |
|
{ |
|
char buf[100]; |
|
cur_name = name; |
|
snprintf(buf, sizeof(buf), "%s.%s", cur_name, format); |
|
|
|
av_md5_init(md5); |
|
if (write_file) { |
|
out = fopen(buf, "wb"); |
|
if (!out) |
|
perror(buf); |
|
} |
|
out_size = 0; |
|
} |
|
|
|
static void close_out(void) |
|
{ |
|
int i; |
|
av_md5_final(md5, hash); |
|
for (i = 0; i < HASH_SIZE; i++) |
|
printf("%02x", hash[i]); |
|
printf(" %d %s\n", out_size, cur_name); |
|
if (out) |
|
fclose(out); |
|
out = NULL; |
|
} |
|
|
|
static void check_func(int value, int line, const char *msg, ...) |
|
{ |
|
if (!value) { |
|
va_list ap; |
|
va_start(ap, msg); |
|
printf("%d: ", line); |
|
vprintf(msg, ap); |
|
printf("\n"); |
|
check_faults++; |
|
va_end(ap); |
|
} |
|
} |
|
#define check(value, ...) check_func(value, __LINE__, __VA_ARGS__) |
|
|
|
static void init_fps(int bf, int audio_preroll, int fps) |
|
{ |
|
AVStream *st; |
|
int iobuf_size = force_iobuf_size ? force_iobuf_size : sizeof(iobuf); |
|
ctx = avformat_alloc_context(); |
|
if (!ctx) |
|
exit(1); |
|
ctx->oformat = av_guess_format(format, NULL, NULL); |
|
if (!ctx->oformat) |
|
exit(1); |
|
ctx->pb = avio_alloc_context(iobuf, iobuf_size, AVIO_FLAG_WRITE, NULL, NULL, io_write, NULL); |
|
if (!ctx->pb) |
|
exit(1); |
|
ctx->pb->write_data_type = io_write_data_type; |
|
ctx->flags |= AVFMT_FLAG_BITEXACT; |
|
|
|
st = avformat_new_stream(ctx, NULL); |
|
if (!st) |
|
exit(1); |
|
st->codecpar->codec_type = AVMEDIA_TYPE_VIDEO; |
|
st->codecpar->codec_id = AV_CODEC_ID_H264; |
|
st->codecpar->width = 640; |
|
st->codecpar->height = 480; |
|
st->time_base.num = 1; |
|
st->time_base.den = 30; |
|
st->codecpar->extradata_size = sizeof(h264_extradata); |
|
st->codecpar->extradata = av_mallocz(st->codecpar->extradata_size + AV_INPUT_BUFFER_PADDING_SIZE); |
|
if (!st->codecpar->extradata) |
|
exit(1); |
|
memcpy(st->codecpar->extradata, h264_extradata, sizeof(h264_extradata)); |
|
video_st = st; |
|
|
|
st = avformat_new_stream(ctx, NULL); |
|
if (!st) |
|
exit(1); |
|
st->codecpar->codec_type = AVMEDIA_TYPE_AUDIO; |
|
st->codecpar->codec_id = AV_CODEC_ID_AAC; |
|
st->codecpar->sample_rate = 44100; |
|
st->codecpar->channels = 2; |
|
st->time_base.num = 1; |
|
st->time_base.den = 44100; |
|
st->codecpar->extradata_size = sizeof(aac_extradata); |
|
st->codecpar->extradata = av_mallocz(st->codecpar->extradata_size + AV_INPUT_BUFFER_PADDING_SIZE); |
|
if (!st->codecpar->extradata) |
|
exit(1); |
|
memcpy(st->codecpar->extradata, aac_extradata, sizeof(aac_extradata)); |
|
audio_st = st; |
|
|
|
if (avformat_write_header(ctx, &opts) < 0) |
|
exit(1); |
|
av_dict_free(&opts); |
|
|
|
frames = 0; |
|
gop_size = 30; |
|
duration = video_st->time_base.den / fps; |
|
audio_duration = 1024LL * audio_st->time_base.den / audio_st->codecpar->sample_rate; |
|
if (audio_preroll) |
|
audio_preroll = 2048LL * audio_st->time_base.den / audio_st->codecpar->sample_rate; |
|
|
|
bframes = bf; |
|
video_dts = bframes ? -duration : 0; |
|
audio_dts = -audio_preroll; |
|
} |
|
|
|
static void init(int bf, int audio_preroll) |
|
{ |
|
init_fps(bf, audio_preroll, 30); |
|
} |
|
|
|
static void mux_frames(int n) |
|
{ |
|
int end_frames = frames + n; |
|
while (1) { |
|
AVPacket pkt; |
|
uint8_t pktdata[8] = { 0 }; |
|
av_init_packet(&pkt); |
|
|
|
if (av_compare_ts(audio_dts, audio_st->time_base, video_dts, video_st->time_base) < 0) { |
|
pkt.dts = pkt.pts = audio_dts; |
|
pkt.stream_index = 1; |
|
pkt.duration = audio_duration; |
|
audio_dts += audio_duration; |
|
} else { |
|
if (frames == end_frames) |
|
break; |
|
pkt.dts = video_dts; |
|
pkt.stream_index = 0; |
|
pkt.duration = duration; |
|
if ((frames % gop_size) == 0) { |
|
pkt.flags |= AV_PKT_FLAG_KEY; |
|
last_picture = AV_PICTURE_TYPE_I; |
|
pkt.pts = pkt.dts + duration; |
|
video_dts = pkt.pts; |
|
} else { |
|
if (last_picture == AV_PICTURE_TYPE_P) { |
|
last_picture = AV_PICTURE_TYPE_B; |
|
pkt.pts = pkt.dts; |
|
video_dts = next_p_pts; |
|
} else { |
|
last_picture = AV_PICTURE_TYPE_P; |
|
if (((frames + 1) % gop_size) == 0) { |
|
pkt.pts = pkt.dts + duration; |
|
video_dts = pkt.pts; |
|
} else { |
|
next_p_pts = pkt.pts = pkt.dts + 2 * duration; |
|
video_dts += duration; |
|
} |
|
} |
|
} |
|
if (!bframes) |
|
pkt.pts = pkt.dts; |
|
if (fake_pkt_duration) |
|
pkt.duration = fake_pkt_duration; |
|
frames++; |
|
} |
|
|
|
if (clear_duration) |
|
pkt.duration = 0; |
|
AV_WB32(pktdata + 4, pkt.pts); |
|
pkt.data = pktdata; |
|
pkt.size = 8; |
|
if (skip_write) |
|
continue; |
|
if (skip_write_audio && pkt.stream_index == 1) |
|
continue; |
|
if (do_interleave) |
|
av_interleaved_write_frame(ctx, &pkt); |
|
else |
|
av_write_frame(ctx, &pkt); |
|
} |
|
} |
|
|
|
static void mux_gops(int n) |
|
{ |
|
mux_frames(gop_size * n); |
|
} |
|
|
|
static void skip_gops(int n) |
|
{ |
|
skip_write = 1; |
|
mux_gops(n); |
|
skip_write = 0; |
|
} |
|
|
|
static void signal_init_ts(void) |
|
{ |
|
AVPacket pkt; |
|
av_init_packet(&pkt); |
|
pkt.size = 0; |
|
pkt.data = NULL; |
|
|
|
pkt.stream_index = 0; |
|
pkt.dts = video_dts; |
|
pkt.pts = 0; |
|
av_write_frame(ctx, &pkt); |
|
|
|
pkt.stream_index = 1; |
|
pkt.dts = pkt.pts = audio_dts; |
|
av_write_frame(ctx, &pkt); |
|
} |
|
|
|
static void finish(void) |
|
{ |
|
av_write_trailer(ctx); |
|
av_free(ctx->pb); |
|
avformat_free_context(ctx); |
|
ctx = NULL; |
|
} |
|
|
|
static void help(void) |
|
{ |
|
printf("movenc-test [-w]\n" |
|
"-w write output into files\n"); |
|
} |
|
|
|
int main(int argc, char **argv) |
|
{ |
|
int c; |
|
uint8_t header[HASH_SIZE]; |
|
uint8_t content[HASH_SIZE]; |
|
int empty_moov_pos; |
|
int prev_pos; |
|
|
|
for (;;) { |
|
c = getopt(argc, argv, "wh"); |
|
if (c == -1) |
|
break; |
|
switch (c) { |
|
case 'w': |
|
write_file = 1; |
|
break; |
|
default: |
|
case 'h': |
|
help(); |
|
return 0; |
|
} |
|
} |
|
|
|
av_register_all(); |
|
|
|
md5 = av_md5_alloc(); |
|
if (!md5) |
|
return 1; |
|
|
|
// Write a fragmented file with an initial moov that actually contains some |
|
// samples. One moov+mdat with 1 second of data and one moof+mdat with 1 |
|
// second of data. |
|
init_out("non-empty-moov"); |
|
av_dict_set(&opts, "movflags", "frag_keyframe", 0); |
|
init(0, 0); |
|
mux_gops(2); |
|
finish(); |
|
close_out(); |
|
|
|
// Write a similar file, but with B-frames and audio preroll, handled |
|
// via an edit list. |
|
init_out("non-empty-moov-elst"); |
|
av_dict_set(&opts, "movflags", "frag_keyframe", 0); |
|
av_dict_set(&opts, "use_editlist", "1", 0); |
|
init(1, 1); |
|
mux_gops(2); |
|
finish(); |
|
close_out(); |
|
|
|
// Use B-frames but no audio-preroll, but without an edit list. |
|
// Due to avoid_negative_ts == AVFMT_AVOID_NEG_TS_MAKE_ZERO, the dts |
|
// of the first audio packet is > 0, but it is set to zero since edit |
|
// lists aren't used, increasing the duration of the first packet instead. |
|
init_out("non-empty-moov-no-elst"); |
|
av_dict_set(&opts, "movflags", "frag_keyframe", 0); |
|
av_dict_set(&opts, "use_editlist", "0", 0); |
|
init(1, 0); |
|
mux_gops(2); |
|
finish(); |
|
close_out(); |
|
|
|
format = "ismv"; |
|
// Write an ISMV, with B-frames and audio preroll. |
|
init_out("ismv"); |
|
av_dict_set(&opts, "movflags", "frag_keyframe", 0); |
|
init(1, 1); |
|
mux_gops(2); |
|
finish(); |
|
close_out(); |
|
format = "mp4"; |
|
|
|
// An initial moov that doesn't contain any samples, followed by two |
|
// moof+mdat pairs. |
|
init_out("empty-moov"); |
|
av_dict_set(&opts, "movflags", "frag_keyframe+empty_moov", 0); |
|
av_dict_set(&opts, "use_editlist", "0", 0); |
|
init(0, 0); |
|
mux_gops(2); |
|
finish(); |
|
close_out(); |
|
memcpy(content, hash, HASH_SIZE); |
|
|
|
// Similar to the previous one, but with input that doesn't start at |
|
// pts/dts 0. avoid_negative_ts behaves in the same way as |
|
// in non-empty-moov-no-elst above. |
|
init_out("empty-moov-no-elst"); |
|
av_dict_set(&opts, "movflags", "frag_keyframe+empty_moov", 0); |
|
init(1, 0); |
|
mux_gops(2); |
|
finish(); |
|
close_out(); |
|
|
|
// Same as the previous one, but disable avoid_negative_ts (which |
|
// would require using an edit list, but with empty_moov, one can't |
|
// write a sensible edit list, when the start timestamps aren't known). |
|
// This should trigger a warning - we check that the warning is produced. |
|
init_count_warnings(); |
|
init_out("empty-moov-no-elst-no-adjust"); |
|
av_dict_set(&opts, "movflags", "frag_keyframe+empty_moov", 0); |
|
av_dict_set(&opts, "avoid_negative_ts", "0", 0); |
|
init(1, 0); |
|
mux_gops(2); |
|
finish(); |
|
close_out(); |
|
|
|
reset_count_warnings(); |
|
check(num_warnings > 0, "No warnings printed for unhandled start offset"); |
|
|
|
// Verify that delay_moov produces the same as empty_moov for |
|
// simple input |
|
init_out("delay-moov"); |
|
av_dict_set(&opts, "movflags", "frag_keyframe+delay_moov", 0); |
|
av_dict_set(&opts, "use_editlist", "0", 0); |
|
init(0, 0); |
|
mux_gops(2); |
|
finish(); |
|
close_out(); |
|
check(!memcmp(hash, content, HASH_SIZE), "delay_moov differs from empty_moov"); |
|
|
|
// Test writing content that requires an edit list using delay_moov |
|
init_out("delay-moov-elst"); |
|
av_dict_set(&opts, "movflags", "frag_keyframe+delay_moov", 0); |
|
init(1, 1); |
|
mux_gops(2); |
|
finish(); |
|
close_out(); |
|
|
|
// Test writing a file with one track lacking packets, with delay_moov. |
|
skip_write_audio = 1; |
|
init_out("delay-moov-empty-track"); |
|
av_dict_set(&opts, "movflags", "frag_keyframe+delay_moov", 0); |
|
init(0, 0); |
|
mux_gops(2); |
|
// The automatic flushing shouldn't output anything, since we're still |
|
// waiting for data for some tracks |
|
check(out_size == 0, "delay_moov flushed prematurely"); |
|
// When closed (or manually flushed), all the written data should still |
|
// be output. |
|
finish(); |
|
close_out(); |
|
check(out_size > 0, "delay_moov didn't output anything"); |
|
|
|
// Check that manually flushing still outputs things as expected. This |
|
// produces two fragments, while the one above produces only one. |
|
init_out("delay-moov-empty-track-flush"); |
|
av_dict_set(&opts, "movflags", "frag_custom+delay_moov", 0); |
|
init(0, 0); |
|
mux_gops(1); |
|
av_write_frame(ctx, NULL); // Force writing the moov |
|
check(out_size > 0, "No moov written"); |
|
av_write_frame(ctx, NULL); |
|
mux_gops(1); |
|
av_write_frame(ctx, NULL); |
|
finish(); |
|
close_out(); |
|
|
|
skip_write_audio = 0; |
|
|
|
|
|
|
|
// Verify that the header written by delay_moov when manually flushed |
|
// is identical to the one by empty_moov. |
|
init_out("empty-moov-header"); |
|
av_dict_set(&opts, "movflags", "frag_keyframe+empty_moov", 0); |
|
av_dict_set(&opts, "use_editlist", "0", 0); |
|
init(0, 0); |
|
close_out(); |
|
memcpy(header, hash, HASH_SIZE); |
|
init_out("empty-moov-content"); |
|
mux_gops(2); |
|
// Written 2 seconds of content, with an automatic flush after 1 second. |
|
check(out_size > 0, "No automatic flush?"); |
|
empty_moov_pos = prev_pos = out_size; |
|
// Manually flush the second fragment |
|
av_write_frame(ctx, NULL); |
|
check(out_size > prev_pos, "No second fragment flushed?"); |
|
prev_pos = out_size; |
|
// Check that an extra flush doesn't output any more data |
|
av_write_frame(ctx, NULL); |
|
check(out_size == prev_pos, "More data written?"); |
|
close_out(); |
|
memcpy(content, hash, HASH_SIZE); |
|
// Ignore the trailer written here |
|
finish(); |
|
|
|
init_out("delay-moov-header"); |
|
av_dict_set(&opts, "movflags", "frag_custom+delay_moov", 0); |
|
av_dict_set(&opts, "use_editlist", "0", 0); |
|
init(0, 0); |
|
check(out_size == 0, "Output written during init with delay_moov"); |
|
mux_gops(1); // Write 1 second of content |
|
av_write_frame(ctx, NULL); // Force writing the moov |
|
close_out(); |
|
check(!memcmp(hash, header, HASH_SIZE), "delay_moov header differs from empty_moov"); |
|
init_out("delay-moov-content"); |
|
av_write_frame(ctx, NULL); // Flush the first fragment |
|
check(out_size == empty_moov_pos, "Manually flushed content differs from automatically flushed, %d vs %d", out_size, empty_moov_pos); |
|
mux_gops(1); // Write the rest of the content |
|
av_write_frame(ctx, NULL); // Flush the second fragment |
|
close_out(); |
|
check(!memcmp(hash, content, HASH_SIZE), "delay_moov content differs from empty_moov"); |
|
finish(); |
|
|
|
|
|
// Verify that we can produce an identical second fragment without |
|
// writing the first one. First write the reference fragments that |
|
// we want to reproduce. |
|
av_dict_set(&opts, "movflags", "frag_custom+empty_moov+dash", 0); |
|
init(0, 0); |
|
mux_gops(1); |
|
av_write_frame(ctx, NULL); // Output the first fragment |
|
init_out("empty-moov-second-frag"); |
|
mux_gops(1); |
|
av_write_frame(ctx, NULL); // Output the second fragment |
|
close_out(); |
|
memcpy(content, hash, HASH_SIZE); |
|
finish(); |
|
|
|
// Produce the same second fragment without actually writing the first |
|
// one before. |
|
av_dict_set(&opts, "movflags", "frag_custom+empty_moov+dash+frag_discont", 0); |
|
av_dict_set(&opts, "fragment_index", "2", 0); |
|
av_dict_set(&opts, "avoid_negative_ts", "0", 0); |
|
av_dict_set(&opts, "use_editlist", "0", 0); |
|
init(0, 0); |
|
skip_gops(1); |
|
init_out("empty-moov-second-frag-discont"); |
|
mux_gops(1); |
|
av_write_frame(ctx, NULL); // Output the second fragment |
|
close_out(); |
|
check(!memcmp(hash, content, HASH_SIZE), "discontinuously written fragment differs"); |
|
finish(); |
|
|
|
// Produce the same thing by using delay_moov, which requires a slightly |
|
// different call sequence. |
|
av_dict_set(&opts, "movflags", "frag_custom+delay_moov+dash+frag_discont", 0); |
|
av_dict_set(&opts, "fragment_index", "2", 0); |
|
init(0, 0); |
|
skip_gops(1); |
|
mux_gops(1); |
|
av_write_frame(ctx, NULL); // Output the moov |
|
init_out("delay-moov-second-frag-discont"); |
|
av_write_frame(ctx, NULL); // Output the second fragment |
|
close_out(); |
|
check(!memcmp(hash, content, HASH_SIZE), "discontinuously written fragment differs"); |
|
finish(); |
|
|
|
|
|
// Test discontinuously written fragments with B-frames (where the |
|
// assumption of starting at pts=0 works) but not with audio preroll |
|
// (which can't be guessed). |
|
av_dict_set(&opts, "movflags", "frag_custom+delay_moov+dash", 0); |
|
init(1, 0); |
|
mux_gops(1); |
|
init_out("delay-moov-elst-init"); |
|
av_write_frame(ctx, NULL); // Output the moov |
|
close_out(); |
|
memcpy(header, hash, HASH_SIZE); |
|
av_write_frame(ctx, NULL); // Output the first fragment |
|
init_out("delay-moov-elst-second-frag"); |
|
mux_gops(1); |
|
av_write_frame(ctx, NULL); // Output the second fragment |
|
close_out(); |
|
memcpy(content, hash, HASH_SIZE); |
|
finish(); |
|
|
|
av_dict_set(&opts, "movflags", "frag_custom+delay_moov+dash+frag_discont", 0); |
|
av_dict_set(&opts, "fragment_index", "2", 0); |
|
init(1, 0); |
|
skip_gops(1); |
|
mux_gops(1); // Write the second fragment |
|
init_out("delay-moov-elst-init-discont"); |
|
av_write_frame(ctx, NULL); // Output the moov |
|
close_out(); |
|
check(!memcmp(hash, header, HASH_SIZE), "discontinuously written header differs"); |
|
init_out("delay-moov-elst-second-frag-discont"); |
|
av_write_frame(ctx, NULL); // Output the second fragment |
|
close_out(); |
|
check(!memcmp(hash, content, HASH_SIZE), "discontinuously written fragment differs"); |
|
finish(); |
|
|
|
|
|
// Test discontinuously written fragments with B-frames and audio preroll, |
|
// properly signaled. |
|
av_dict_set(&opts, "movflags", "frag_custom+delay_moov+dash", 0); |
|
init(1, 1); |
|
mux_gops(1); |
|
init_out("delay-moov-elst-signal-init"); |
|
av_write_frame(ctx, NULL); // Output the moov |
|
close_out(); |
|
memcpy(header, hash, HASH_SIZE); |
|
av_write_frame(ctx, NULL); // Output the first fragment |
|
init_out("delay-moov-elst-signal-second-frag"); |
|
mux_gops(1); |
|
av_write_frame(ctx, NULL); // Output the second fragment |
|
close_out(); |
|
memcpy(content, hash, HASH_SIZE); |
|
finish(); |
|
|
|
av_dict_set(&opts, "movflags", "frag_custom+delay_moov+dash+frag_discont", 0); |
|
av_dict_set(&opts, "fragment_index", "2", 0); |
|
init(1, 1); |
|
signal_init_ts(); |
|
skip_gops(1); |
|
mux_gops(1); // Write the second fragment |
|
init_out("delay-moov-elst-signal-init-discont"); |
|
av_write_frame(ctx, NULL); // Output the moov |
|
close_out(); |
|
check(!memcmp(hash, header, HASH_SIZE), "discontinuously written header differs"); |
|
init_out("delay-moov-elst-signal-second-frag-discont"); |
|
av_write_frame(ctx, NULL); // Output the second fragment |
|
close_out(); |
|
check(!memcmp(hash, content, HASH_SIZE), "discontinuously written fragment differs"); |
|
finish(); |
|
|
|
|
|
// Test VFR content, with sidx atoms (which declare the pts duration |
|
// of a fragment, forcing overriding the start pts of the next one). |
|
// Here, the fragment duration in pts is significantly different from |
|
// the duration in dts. The video stream starts at dts=-10,pts=0, and |
|
// the second fragment starts at dts=155,pts=156. The trun duration sum |
|
// of the first fragment is 165, which also is written as |
|
// baseMediaDecodeTime in the tfdt in the second fragment. The sidx for |
|
// the first fragment says earliest_presentation_time = 0 and |
|
// subsegment_duration = 156, which also matches the sidx in the second |
|
// fragment. For the audio stream, the pts and dts durations also don't |
|
// match - the input stream starts at pts=-2048, but that part is excluded |
|
// by the edit list. |
|
init_out("vfr"); |
|
av_dict_set(&opts, "movflags", "frag_keyframe+delay_moov+dash", 0); |
|
init_fps(1, 1, 3); |
|
mux_frames(gop_size/2); |
|
duration /= 10; |
|
mux_frames(gop_size/2); |
|
mux_gops(1); |
|
finish(); |
|
close_out(); |
|
|
|
// Test VFR content, with cleared duration fields. In these cases, |
|
// the muxer must guess the duration of the last packet of each |
|
// fragment. As long as the framerate doesn't vary (too much) at the |
|
// fragment edge, it works just fine. Additionally, when automatically |
|
// cutting fragments, the muxer already know the timestamps of the next |
|
// packet for one stream (in most cases the video stream), avoiding |
|
// having to use guesses for that one. |
|
init_count_warnings(); |
|
clear_duration = 1; |
|
init_out("vfr-noduration"); |
|
av_dict_set(&opts, "movflags", "frag_keyframe+delay_moov+dash", 0); |
|
init_fps(1, 1, 3); |
|
mux_frames(gop_size/2); |
|
duration /= 10; |
|
mux_frames(gop_size/2); |
|
mux_gops(1); |
|
finish(); |
|
close_out(); |
|
clear_duration = 0; |
|
reset_count_warnings(); |
|
check(num_warnings > 0, "No warnings printed for filled in durations"); |
|
|
|
// Test with an IO buffer size that is too small to hold a full fragment; |
|
// this will cause write_data_type to be called with the type unknown. |
|
force_iobuf_size = 1500; |
|
init_out("large_frag"); |
|
av_dict_set(&opts, "movflags", "frag_keyframe+delay_moov", 0); |
|
init_fps(1, 1, 3); |
|
mux_gops(2); |
|
finish(); |
|
close_out(); |
|
force_iobuf_size = 0; |
|
|
|
// Test VFR content with bframes with interleaving. |
|
// Here, using av_interleaved_write_frame allows the muxer to get the |
|
// fragment end durations right. We always set the packet duration to |
|
// the expected, but we simulate dropped frames at one point. |
|
do_interleave = 1; |
|
init_out("vfr-noduration-interleave"); |
|
av_dict_set(&opts, "movflags", "frag_keyframe+delay_moov", 0); |
|
av_dict_set(&opts, "frag_duration", "650000", 0); |
|
init_fps(1, 1, 30); |
|
mux_frames(gop_size/2); |
|
// Pretend that the packet duration is the normal, even if |
|
// we actually skip a bunch of frames. (I.e., simulate that |
|
// we don't know of the framedrop in advance.) |
|
fake_pkt_duration = duration; |
|
duration *= 10; |
|
mux_frames(1); |
|
fake_pkt_duration = 0; |
|
duration /= 10; |
|
mux_frames(gop_size/2 - 1); |
|
mux_gops(1); |
|
finish(); |
|
close_out(); |
|
clear_duration = 0; |
|
do_interleave = 0; |
|
|
|
|
|
av_free(md5); |
|
|
|
return check_faults > 0 ? 1 : 0; |
|
}
|
|
|