ffmpeg/libavcodec/tests/h264_levels.c

236 lines
7.4 KiB
C

/*
* 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 "libavutil/common.h"
#include "libavcodec/h264_levels.h"
static const struct {
int width;
int height;
int level_idc;
} test_sizes[] = {
// First level usable at some standard sizes.
// (From H.264 table A-6.)
{ 176, 144, 10 }, // QCIF
{ 352, 288, 11 }, // CIF
{ 640, 480, 22 }, // VGA
{ 720, 480, 22 }, // NTSC
{ 720, 576, 22 }, // PAL
{ 800, 600, 31 }, // SVGA
{ 1280, 720, 31 }, // 720p
{ 1280, 1024, 32 }, // SXGA
{ 1920, 1080, 40 }, // 1080p
{ 2048, 1080, 42 }, // 2Kx1080
{ 2048, 1536, 50 }, // 4XGA
{ 3840, 2160, 51 }, // 4K
{ 7680, 4320, 60 }, // 8K
// Overly wide or tall sizes.
{ 1, 256, 10 },
{ 1, 512, 11 },
{ 1, 1024, 21 },
{ 1, 1808, 22 },
{ 1, 1824, 31 },
{ 256, 1, 10 },
{ 512, 1, 11 },
{ 1024, 1, 21 },
{ 1808, 1, 22 },
{ 1824, 1, 31 },
{ 512, 4096, 40 },
{ 256, 4112, 42 },
{ 8688, 1024, 51 },
{ 8704, 512, 60 },
{ 16880, 1, 60 },
{ 16896, 1, 0 },
};
static const struct {
int width;
int height;
int framerate;
int level_idc;
} test_framerate[] = {
// Some typical sizes and frame rates.
// (From H.264 table A-1 and table A-6)
{ 176, 144, 15, 10 },
{ 176, 144, 16, 11 },
{ 320, 240, 10, 11 },
{ 320, 240, 20, 12 },
{ 320, 240, 40, 21 },
{ 352, 288, 30, 13 },
{ 352, 288, 51, 22 },
{ 352, 576, 25, 21 },
{ 352, 576, 26, 30 },
{ 640, 480, 33, 30 },
{ 640, 480, 34, 31 },
{ 720, 480, 50, 31 },
{ 720, 576, 25, 30 },
{ 800, 600, 55, 31 },
{ 1024, 768, 35, 31 },
{ 1024, 768, 70, 32 },
{ 1280, 720, 30, 31 },
{ 1280, 720, 31, 32 },
{ 1280, 960, 45, 32 },
{ 1280, 960, 46, 40 },
{ 1280, 1024, 42, 32 },
{ 1600, 1200, 32, 40 },
{ 1600, 1200, 33, 42 },
{ 1920, 1088, 30, 40 },
{ 1920, 1088, 55, 42 },
{ 2048, 1024, 30, 40 },
{ 2048, 1024, 62, 42 },
{ 2048, 1088, 60, 42 },
{ 3680, 1536, 26, 50 },
{ 4096, 2048, 30, 51 },
{ 4096, 2048, 59, 52 },
{ 4096, 2160, 60, 52 },
};
static const struct {
int width;
int height;
int dpb_size;
int level_idc;
} test_dpb[] = {
// First level usable for some DPB sizes.
// (From H.264 table A-7.)
{ 176, 144, 4, 10 },
{ 176, 144, 8, 11 },
{ 176, 144, 16, 12 },
{ 1280, 720, 1, 31 },
{ 1280, 720, 5, 31 },
{ 1280, 720, 9, 40 },
{ 1280, 720, 10, 50 },
{ 1920, 1080, 1, 40 },
{ 1920, 1080, 5, 50 },
{ 1920, 1080, 13, 50 },
{ 1920, 1080, 14, 51 },
{ 3840, 2160, 5, 51 },
{ 3840, 2160, 6, 60 },
{ 3840, 2160, 16, 60 },
{ 7680, 4320, 5, 60 },
{ 7680, 4320, 6, 0 },
};
static const struct {
int64_t bitrate;
int profile_idc;
int level_idc;
} test_bitrate[] = {
// Values where profile affects level at a given bitrate.
{ 2500000, 77, 21 },
{ 2500000, 100, 20 },
{ 2500000, 244, 13 },
{ 100000000, 77, 50 },
{ 100000000, 100, 50 },
{ 100000000, 244, 41 },
{ 999999999, 77, 0 },
{ 999999999, 100, 62 },
// Check level 1b.
{ 32 * 1200, 66, 10 },
{ 32 * 1500, 100, 10 },
{ 96 * 1200, 66, 11 },
{ 96 * 1500, 100, 9 },
{ 144 * 1200, 66, 11 },
{ 144 * 1500, 100, 11 },
};
static const struct {
const char *name;
int profile_idc;
int64_t bitrate;
int width;
int height;
int dpb_frames;
int level_idc;
} test_all[] = {
{ "Bluray 1080p 40Mb/s", 100, 40000000, 1920, 1080, 4, 41 },
{ "Bluray 1080p 24Mb/s", 100, 24000000, 1920, 1080, 4, 40 },
{ "Bluray 720p 40Mb/s", 100, 40000000, 1280, 720, 6, 41 },
{ "Bluray 720p 24Mb/s", 100, 24000000, 1280, 720, 6, 40 },
{ "Bluray PAL 40Mb/s", 100, 40000000, 720, 576, 6, 41 },
{ "Bluray PAL 24Mb/s", 100, 24000000, 720, 576, 6, 32 },
{ "Bluray PAL 16Mb/s", 100, 16800000, 720, 576, 6, 31 },
{ "Bluray PAL 12Mb/s", 100, 12000000, 720, 576, 5, 30 },
{ "Bluray NTSC 40Mb/s", 100, 40000000, 720, 480, 6, 41 },
{ "Bluray NTSC 24Mb/s", 100, 24000000, 720, 480, 6, 32 },
{ "Bluray NTSC 16Mb/s", 100, 16800000, 720, 480, 6, 31 },
{ "Bluray NTSC 12Mb/s", 100, 12000000, 720, 480, 6, 30 },
};
int main(void)
{
const H264LevelDescriptor *level;
int i;
#define CHECK(expected, format, ...) do { \
if (expected ? (!level || level->level_idc != expected) \
: !!level) { \
av_log(NULL, AV_LOG_ERROR, "Incorrect level for " \
format ": expected %d, got %d.\n", __VA_ARGS__, \
expected, level ? level->level_idc : -1); \
return 1; \
} \
} while (0)
for (i = 0; i < FF_ARRAY_ELEMS(test_sizes); i++) {
level = ff_h264_guess_level(0, 0, 0, test_sizes[i].width,
test_sizes[i].height, 0);
CHECK(test_sizes[i].level_idc, "size %dx%d",
test_sizes[i].width, test_sizes[i].height);
}
for (i = 0; i < FF_ARRAY_ELEMS(test_framerate); i++) {
level = ff_h264_guess_level(0, 0, test_framerate[i].framerate,
test_framerate[i].width,
test_framerate[i].height, 0);
CHECK(test_framerate[i].level_idc, "framerate %d, size %dx%d",
test_framerate[i].framerate, test_framerate[i].width,
test_framerate[i].height);
}
for (i = 0; i < FF_ARRAY_ELEMS(test_dpb); i++) {
level = ff_h264_guess_level(0, 0, 0, test_dpb[i].width,
test_dpb[i].height,
test_dpb[i].dpb_size);
CHECK(test_dpb[i].level_idc, "size %dx%d dpb %d",
test_dpb[i].width, test_dpb[i].height,
test_dpb[i].dpb_size);
}
for (i = 0; i < FF_ARRAY_ELEMS(test_bitrate); i++) {
level = ff_h264_guess_level(test_bitrate[i].profile_idc,
test_bitrate[i].bitrate,
0, 0, 0, 0);
CHECK(test_bitrate[i].level_idc, "bitrate %"PRId64" profile %d",
test_bitrate[i].bitrate, test_bitrate[i].profile_idc);
}
for (i = 0; i < FF_ARRAY_ELEMS(test_all); i++) {
level = ff_h264_guess_level(test_all[i].profile_idc,
test_all[i].bitrate,
0,
test_all[i].width,
test_all[i].height,
test_all[i].dpb_frames);
CHECK(test_all[i].level_idc, "%s", test_all[i].name);
}
return 0;
}