ffmpeg/libavutil/tests/pixfmt_best.c

241 lines
9.0 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/pixdesc.c"
static const enum AVPixelFormat pixfmt_list[] = {
AV_PIX_FMT_MONOWHITE,
AV_PIX_FMT_GRAY8,
AV_PIX_FMT_GRAY10,
AV_PIX_FMT_GRAY16,
AV_PIX_FMT_YUV420P,
AV_PIX_FMT_YUV420P10,
AV_PIX_FMT_YUV420P16,
AV_PIX_FMT_YUV422P,
AV_PIX_FMT_YUV422P10,
AV_PIX_FMT_YUV422P16,
AV_PIX_FMT_YUV444P,
AV_PIX_FMT_YUV444P10,
AV_PIX_FMT_YUV444P16,
AV_PIX_FMT_RGB565,
AV_PIX_FMT_RGB24,
AV_PIX_FMT_RGB48,
AV_PIX_FMT_VDPAU,
AV_PIX_FMT_VAAPI,
};
static const enum AVPixelFormat semiplanar_list[] = {
AV_PIX_FMT_P016,
AV_PIX_FMT_P012,
AV_PIX_FMT_P010,
AV_PIX_FMT_NV12,
};
static const enum AVPixelFormat packed_list[] = {
AV_PIX_FMT_XV48,
AV_PIX_FMT_XV36,
AV_PIX_FMT_XV30,
AV_PIX_FMT_VUYX,
AV_PIX_FMT_Y216,
AV_PIX_FMT_Y212,
AV_PIX_FMT_Y210,
AV_PIX_FMT_YUYV422,
};
static const enum AVPixelFormat subsampled_list[] = {
AV_PIX_FMT_YUV411P,
AV_PIX_FMT_YUV420P,
AV_PIX_FMT_YUV422P,
AV_PIX_FMT_YUV444P,
};
static const enum AVPixelFormat depthchroma_list[] = {
AV_PIX_FMT_YUV420P14,
AV_PIX_FMT_YUV422P14,
AV_PIX_FMT_YUV444P16,
};
typedef enum AVPixelFormat (*find_best_t)(enum AVPixelFormat pixfmt);
#define find_best_wrapper(name, list) \
static enum AVPixelFormat find_best_ ## name (enum AVPixelFormat pixfmt) \
{ \
enum AVPixelFormat best = AV_PIX_FMT_NONE; \
int i; \
for (i = 0; i < FF_ARRAY_ELEMS(list); i++) \
best = av_find_best_pix_fmt_of_2(best, list[i], \
pixfmt, 0, NULL); \
return best; \
}
find_best_wrapper(base, pixfmt_list)
find_best_wrapper(seminplanar, semiplanar_list)
find_best_wrapper(packed, packed_list)
find_best_wrapper(subsampled, subsampled_list)
find_best_wrapper(depthchroma, depthchroma_list)
static void test(enum AVPixelFormat input, enum AVPixelFormat expected,
int *pass, int *fail, find_best_t find_best_fn)
{
enum AVPixelFormat output = find_best_fn(input);
if (output != expected) {
printf("Matching %s: got %s, expected %s\n",
av_get_pix_fmt_name(input),
av_get_pix_fmt_name(output),
av_get_pix_fmt_name(expected));
++(*fail);
} else
++(*pass);
}
int main(void)
{
int i, pass = 0, fail = 0;
#define TEST(input, expected) \
test(input, expected, &pass, &fail, find_best_base)
// Same formats.
for (i = 0; i < FF_ARRAY_ELEMS(pixfmt_list); i++)
TEST(pixfmt_list[i], pixfmt_list[i]);
// Formats containing the same data in different layouts.
TEST(AV_PIX_FMT_MONOBLACK, AV_PIX_FMT_MONOWHITE);
TEST(AV_PIX_FMT_NV12, AV_PIX_FMT_YUV420P);
TEST(AV_PIX_FMT_P010, AV_PIX_FMT_YUV420P10);
TEST(AV_PIX_FMT_P016, AV_PIX_FMT_YUV420P16);
TEST(AV_PIX_FMT_NV16, AV_PIX_FMT_YUV422P);
TEST(AV_PIX_FMT_NV24, AV_PIX_FMT_YUV444P);
TEST(AV_PIX_FMT_YUYV422, AV_PIX_FMT_YUV422P);
TEST(AV_PIX_FMT_UYVY422, AV_PIX_FMT_YUV422P);
TEST(AV_PIX_FMT_VYU444, AV_PIX_FMT_YUV444P);
TEST(AV_PIX_FMT_BGR565, AV_PIX_FMT_RGB565);
TEST(AV_PIX_FMT_BGR24, AV_PIX_FMT_RGB24);
TEST(AV_PIX_FMT_GBRP, AV_PIX_FMT_RGB24);
TEST(AV_PIX_FMT_0RGB, AV_PIX_FMT_RGB24);
TEST(AV_PIX_FMT_GBRP16, AV_PIX_FMT_RGB48);
TEST(AV_PIX_FMT_VUYX, AV_PIX_FMT_YUV444P);
// Formats additionally containing alpha (here ignored).
TEST(AV_PIX_FMT_YA8, AV_PIX_FMT_GRAY8);
TEST(AV_PIX_FMT_YA16, AV_PIX_FMT_GRAY16);
TEST(AV_PIX_FMT_YUVA420P, AV_PIX_FMT_YUV420P);
TEST(AV_PIX_FMT_YUVA422P, AV_PIX_FMT_YUV422P);
TEST(AV_PIX_FMT_YUVA444P, AV_PIX_FMT_YUV444P);
TEST(AV_PIX_FMT_VUYA, AV_PIX_FMT_YUV444P);
TEST(AV_PIX_FMT_AYUV, AV_PIX_FMT_YUV444P);
TEST(AV_PIX_FMT_UYVA, AV_PIX_FMT_YUV444P);
TEST(AV_PIX_FMT_AYUV64, AV_PIX_FMT_YUV444P16);
TEST(AV_PIX_FMT_RGBA, AV_PIX_FMT_RGB24);
TEST(AV_PIX_FMT_ABGR, AV_PIX_FMT_RGB24);
TEST(AV_PIX_FMT_GBRAP, AV_PIX_FMT_RGB24);
TEST(AV_PIX_FMT_RGBA64, AV_PIX_FMT_RGB48);
TEST(AV_PIX_FMT_BGRA64, AV_PIX_FMT_RGB48);
TEST(AV_PIX_FMT_GBRAP16, AV_PIX_FMT_RGB48);
// Formats requiring upsampling to represent exactly.
TEST(AV_PIX_FMT_GRAY12, AV_PIX_FMT_GRAY16);
TEST(AV_PIX_FMT_YUV410P, AV_PIX_FMT_YUV420P);
TEST(AV_PIX_FMT_YUV411P, AV_PIX_FMT_YUV422P);
TEST(AV_PIX_FMT_UYYVYY411, AV_PIX_FMT_YUV422P);
TEST(AV_PIX_FMT_YUV440P, AV_PIX_FMT_YUV444P);
TEST(AV_PIX_FMT_YUV440P10, AV_PIX_FMT_YUV444P10);
TEST(AV_PIX_FMT_YUV440P12, AV_PIX_FMT_YUV444P16);
TEST(AV_PIX_FMT_YUV420P9, AV_PIX_FMT_YUV420P10);
TEST(AV_PIX_FMT_YUV420P12, AV_PIX_FMT_YUV420P16);
TEST(AV_PIX_FMT_YUV444P9, AV_PIX_FMT_YUV444P10);
TEST(AV_PIX_FMT_YUV444P12, AV_PIX_FMT_YUV444P16);
TEST(AV_PIX_FMT_BGR4, AV_PIX_FMT_RGB565);
TEST(AV_PIX_FMT_RGB444, AV_PIX_FMT_RGB565);
TEST(AV_PIX_FMT_RGB555, AV_PIX_FMT_RGB565);
TEST(AV_PIX_FMT_GBRP10, AV_PIX_FMT_RGB48);
TEST(AV_PIX_FMT_GBRAP10, AV_PIX_FMT_RGB48);
TEST(AV_PIX_FMT_GBRAP12, AV_PIX_FMT_RGB48);
// Formats containing the same data in different endianness.
TEST(AV_PIX_FMT_GRAY10BE, AV_PIX_FMT_GRAY10);
TEST(AV_PIX_FMT_GRAY10LE, AV_PIX_FMT_GRAY10);
TEST(AV_PIX_FMT_GRAY16BE, AV_PIX_FMT_GRAY16);
TEST(AV_PIX_FMT_GRAY16LE, AV_PIX_FMT_GRAY16);
TEST(AV_PIX_FMT_YUV422P10BE, AV_PIX_FMT_YUV422P10);
TEST(AV_PIX_FMT_YUV422P10LE, AV_PIX_FMT_YUV422P10);
TEST(AV_PIX_FMT_YUV444P16BE, AV_PIX_FMT_YUV444P16);
TEST(AV_PIX_FMT_YUV444P16LE, AV_PIX_FMT_YUV444P16);
TEST(AV_PIX_FMT_RGB565BE, AV_PIX_FMT_RGB565);
TEST(AV_PIX_FMT_RGB565LE, AV_PIX_FMT_RGB565);
TEST(AV_PIX_FMT_RGB48BE, AV_PIX_FMT_RGB48);
TEST(AV_PIX_FMT_RGB48LE, AV_PIX_FMT_RGB48);
// Opaque formats are least unlike each other.
TEST(AV_PIX_FMT_DXVA2_VLD, AV_PIX_FMT_VDPAU);
#define TEST_SEMIPLANAR(input, expected) \
test(input, expected, &pass, &fail, find_best_seminplanar)
// Same formats.
for (i = 0; i < FF_ARRAY_ELEMS(semiplanar_list); i++)
TEST_SEMIPLANAR(semiplanar_list[i], semiplanar_list[i]);
// Formats containing the same data in different layouts.
TEST_SEMIPLANAR(AV_PIX_FMT_YUV420P, AV_PIX_FMT_NV12);
TEST_SEMIPLANAR(AV_PIX_FMT_YUV420P10, AV_PIX_FMT_P010);
TEST_SEMIPLANAR(AV_PIX_FMT_YUV420P12, AV_PIX_FMT_P012);
TEST_SEMIPLANAR(AV_PIX_FMT_YUV420P16, AV_PIX_FMT_P016);
TEST_SEMIPLANAR(AV_PIX_FMT_YUV420P9, AV_PIX_FMT_P010);
#define TEST_PACKED(input, expected) \
test(input, expected, &pass, &fail, find_best_packed)
// Same formats.
for (i = 0; i < FF_ARRAY_ELEMS(packed_list); i++)
TEST_PACKED(packed_list[i], packed_list[i]);
// Formats containing the same data in different layouts.
TEST_PACKED(AV_PIX_FMT_YUV444P, AV_PIX_FMT_VUYX);
TEST_PACKED(AV_PIX_FMT_YUV444P10, AV_PIX_FMT_XV30);
TEST_PACKED(AV_PIX_FMT_YUV444P12, AV_PIX_FMT_XV36);
TEST_PACKED(AV_PIX_FMT_YUV444P16, AV_PIX_FMT_XV48);
TEST_PACKED(AV_PIX_FMT_YUV422P, AV_PIX_FMT_YUYV422);
TEST_PACKED(AV_PIX_FMT_YUV422P10, AV_PIX_FMT_Y210);
TEST_PACKED(AV_PIX_FMT_YUV422P12, AV_PIX_FMT_Y212);
TEST_PACKED(AV_PIX_FMT_YUV422P16, AV_PIX_FMT_Y216);
#define TEST_SUBSAMPLED(input, expected) \
test(input, expected, &pass, &fail, find_best_subsampled)
// Same formats.
for (i = 0; i < FF_ARRAY_ELEMS(subsampled_list); i++)
TEST_SUBSAMPLED(subsampled_list[i], subsampled_list[i]);
TEST_SUBSAMPLED(AV_PIX_FMT_YUV410P, AV_PIX_FMT_YUV420P);
#define TEST_DEPTH_CHROMA(input, expected) \
test(input, expected, &pass, &fail, find_best_depthchroma)
// Same formats.
for (i = 0; i < FF_ARRAY_ELEMS(depthchroma_list); i++)
TEST_DEPTH_CHROMA(depthchroma_list[i], depthchroma_list[i]);
TEST_DEPTH_CHROMA(AV_PIX_FMT_YUV420P16, AV_PIX_FMT_YUV444P16);
TEST_DEPTH_CHROMA(AV_PIX_FMT_YUV422P16, AV_PIX_FMT_YUV444P16);
printf("%d tests passed, %d tests failed.\n", pass, fail);
return !!fail;
}