[FFmpeg-devel] [PATCH] checkasm: add unit tests for v210enc
Henrik Gramner
henrik at gramner.com
Sun Sep 6 01:13:39 CEST 2015
---
libavcodec/v210enc.c | 15 +++++---
libavcodec/v210enc.h | 2 +
tests/checkasm/Makefile | 1 +
tests/checkasm/checkasm.c | 3 ++
tests/checkasm/checkasm.h | 1 +
tests/checkasm/v210enc.c | 94 +++++++++++++++++++++++++++++++++++++++++++++++
6 files changed, 111 insertions(+), 5 deletions(-)
create mode 100644 tests/checkasm/v210enc.c
diff --git a/libavcodec/v210enc.c b/libavcodec/v210enc.c
index 2c8b9cb..0612248 100644
--- a/libavcodec/v210enc.c
+++ b/libavcodec/v210enc.c
@@ -82,6 +82,15 @@ static void v210_planar_pack_10_c(const uint16_t *y, const uint16_t *u,
}
}
+av_cold void ff_v210enc_init(V210EncContext *s)
+{
+ s->pack_line_8 = v210_planar_pack_8_c;
+ s->pack_line_10 = v210_planar_pack_10_c;
+
+ if (ARCH_X86)
+ ff_v210enc_init_x86(s);
+}
+
static av_cold int encode_init(AVCodecContext *avctx)
{
V210EncContext *s = avctx->priv_data;
@@ -97,11 +106,7 @@ FF_DISABLE_DEPRECATION_WARNINGS
FF_ENABLE_DEPRECATION_WARNINGS
#endif
- s->pack_line_8 = v210_planar_pack_8_c;
- s->pack_line_10 = v210_planar_pack_10_c;
-
- if (ARCH_X86)
- ff_v210enc_init_x86(s);
+ ff_v210enc_init(s);
return 0;
}
diff --git a/libavcodec/v210enc.h b/libavcodec/v210enc.h
index 9179d73..a205427 100644
--- a/libavcodec/v210enc.h
+++ b/libavcodec/v210enc.h
@@ -30,6 +30,8 @@ typedef struct V210EncContext {
const uint16_t *v, uint8_t *dst, ptrdiff_t width);
} V210EncContext;
+void ff_v210enc_init(V210EncContext *s);
+
void ff_v210enc_init_x86(V210EncContext *s);
#endif /* AVCODEC_V210ENC_H */
diff --git a/tests/checkasm/Makefile b/tests/checkasm/Makefile
index 8be6eec..6aabcd7 100644
--- a/tests/checkasm/Makefile
+++ b/tests/checkasm/Makefile
@@ -2,6 +2,7 @@
AVCODECOBJS-$(CONFIG_BSWAPDSP) += bswapdsp.o
AVCODECOBJS-$(CONFIG_H264PRED) += h264pred.o
AVCODECOBJS-$(CONFIG_H264QPEL) += h264qpel.o
+AVCODECOBJS-$(CONFIG_V210_ENCODER) += v210enc.o
CHECKASMOBJS-$(CONFIG_AVCODEC) += $(AVCODECOBJS-yes)
diff --git a/tests/checkasm/checkasm.c b/tests/checkasm/checkasm.c
index 7ec8d67..5fdf0a3 100644
--- a/tests/checkasm/checkasm.c
+++ b/tests/checkasm/checkasm.c
@@ -66,6 +66,9 @@ static const struct {
#if CONFIG_H264QPEL
{ "h264qpel", checkasm_check_h264qpel },
#endif
+#if CONFIG_V210_ENCODER
+ { "v210enc", checkasm_check_v210enc },
+#endif
{ NULL }
};
diff --git a/tests/checkasm/checkasm.h b/tests/checkasm/checkasm.h
index 8028224..bf24d38 100644
--- a/tests/checkasm/checkasm.h
+++ b/tests/checkasm/checkasm.h
@@ -32,6 +32,7 @@
void checkasm_check_bswapdsp(void);
void checkasm_check_h264pred(void);
void checkasm_check_h264qpel(void);
+void checkasm_check_v210enc(void);
void *checkasm_check_func(void *func, const char *name, ...) av_printf_format(2, 3);
int checkasm_bench_func(void);
diff --git a/tests/checkasm/v210enc.c b/tests/checkasm/v210enc.c
new file mode 100644
index 0000000..23c73b9
--- /dev/null
+++ b/tests/checkasm/v210enc.c
@@ -0,0 +1,94 @@
+/*
+ * Copyright (c) 2015 Henrik Gramner
+ *
+ * This file is part of FFmpeg.
+ *
+ * FFmpeg is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 2 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 General Public License for more details.
+ *
+ * You should have received a copy of the GNU 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 <string.h>
+#include "checkasm.h"
+#include "libavcodec/v210enc.h"
+#include "libavutil/common.h"
+#include "libavutil/internal.h"
+#include "libavutil/intreadwrite.h"
+
+#define BUF_SIZE 512
+
+#define randomize_buffers(mask) \
+ do { \
+ int i, size = sizeof(*y0); \
+ for (i = 0; i < BUF_SIZE; i += 4/size) { \
+ uint32_t r = rnd() & mask; \
+ AV_WN32A(y0+i, r); \
+ AV_WN32A(y1+i, r); \
+ } \
+ for (i = 0; i < BUF_SIZE/2; i += 4/size) { \
+ uint32_t r = rnd() & mask; \
+ AV_WN32A(u0+i, r); \
+ AV_WN32A(u1+i, r); \
+ r = rnd() & mask; \
+ AV_WN32A(v0+i, r); \
+ AV_WN32A(v1+i, r); \
+ } \
+ for (i = 0; i < BUF_SIZE*8/3; i += 4) { \
+ uint32_t r = rnd(); \
+ AV_WN32A(dst0+i, r); \
+ AV_WN32A(dst1+i, r); \
+ } \
+ } while (0)
+
+#define check_pack_line(type, mask) \
+ do { \
+ LOCAL_ALIGNED_16(type, y0, [BUF_SIZE]); \
+ LOCAL_ALIGNED_16(type, y1, [BUF_SIZE]); \
+ LOCAL_ALIGNED_16(type, u0, [BUF_SIZE/2]); \
+ LOCAL_ALIGNED_16(type, u1, [BUF_SIZE/2]); \
+ LOCAL_ALIGNED_16(type, v0, [BUF_SIZE/2]); \
+ LOCAL_ALIGNED_16(type, v1, [BUF_SIZE/2]); \
+ LOCAL_ALIGNED_16(uint8_t, dst0, [BUF_SIZE*8/3]); \
+ LOCAL_ALIGNED_16(uint8_t, dst1, [BUF_SIZE*8/3]); \
+ \
+ declare_func(void, const type *y, const type *u, const type *v, \
+ uint8_t *dst, ptrdiff_t width); \
+ ptrdiff_t width, step = 12 / sizeof(type); \
+ \
+ for (width = step; width < BUF_SIZE - 15; width += step) { \
+ int y_offset = rnd() & 15; \
+ int uv_offset = y_offset / 2; \
+ randomize_buffers(mask); \
+ call_ref(y0+y_offset, u0+uv_offset, v0+uv_offset, dst0, width); \
+ call_new(y1+y_offset, u1+uv_offset, v1+uv_offset, dst1, width); \
+ if (memcmp(y0, y1, BUF_SIZE) || memcmp(u0, u1, BUF_SIZE/2) || \
+ memcmp(v0, v1, BUF_SIZE/2) || memcmp(dst0, dst1, width*8/3)) \
+ fail(); \
+ bench_new(y1+y_offset, u1+uv_offset, v1+uv_offset, dst1, width); \
+ } \
+ } while (0)
+
+void checkasm_check_v210enc(void)
+{
+ V210EncContext h;
+
+ ff_v210enc_init(&h);
+
+ if (check_func(h.pack_line_8, "v210_planar_pack_8"))
+ check_pack_line(uint8_t, 0xffffffff);
+
+ if (check_func(h.pack_line_10, "v210_planar_pack_10"))
+ check_pack_line(uint16_t, 0x03ff03ff);
+
+ report("planar_pack");
+}
--
1.8.3.2
More information about the ffmpeg-devel
mailing list