ref: 190da6aedad3a43a9fedc88714c124d30d516ab1
parent: f2ee7a2a430992816e20b9d3a364492abf8fbdb8
author: Henrik Gramner <[email protected]>
date: Fri Sep 28 07:23:46 EDT 2018
checkasm: Add MC unit test
--- a/tests/checkasm/checkasm.c
+++ b/tests/checkasm/checkasm.c
@@ -52,6 +52,8 @@
const char *name;
void (*func)(void);
} tests[] = {+ { "mc_8bpc", checkasm_check_mc_8bpc },+ { "mc_10bpc", checkasm_check_mc_10bpc }, { NULL }};
--- a/tests/checkasm/checkasm.h
+++ b/tests/checkasm/checkasm.h
@@ -36,6 +36,9 @@
#include "include/common/attributes.h"
#include "include/common/intops.h"
+void checkasm_check_mc_8bpc(void);
+void checkasm_check_mc_10bpc(void);
+
void *checkasm_check_func(void *func, const char *name, ...);
int checkasm_bench_func(void);
void checkasm_fail_func(const char *msg, ...);
--- /dev/null
+++ b/tests/checkasm/mc.c
@@ -1,0 +1,240 @@
+/*
+ * Copyright © 2018, VideoLAN and dav1d authors
+ * Copyright © 2018, Two Orioles, LLC
+ * All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions are met:
+ *
+ * 1. Redistributions of source code must retain the above copyright notice, this
+ * list of conditions and the following disclaimer.
+ *
+ * 2. Redistributions in binary form must reproduce the above copyright notice,
+ * this list of conditions and the following disclaimer in the documentation
+ * and/or other materials provided with the distribution.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
+ * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
+ * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+ * DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR
+ * ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
+ * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
+ * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
+ * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+ * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
+ * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ */
+
+#include "src/levels.h"
+#include "src/mc.h"
+
+#include "tests/checkasm/checkasm.h"
+
+static const char *const filter_names[] = {+ "8tap_regular", "8tap_regular_smooth", "8tap_regular_sharp",
+ "8tap_sharp_regular", "8tap_sharp_smooth", "8tap_sharp",
+ "8tap_smooth_regular", "8tap_smooth", "8tap_smooth_sharp",
+ "bilinear"
+};
+
+static const char *const mxy_names[] = { "0", "h", "v", "hv" };+
+static void check_mc(Dav1dMCDSPContext *const c) {+ ALIGN_STK_32(pixel, src_buf, 135 * 135,);
+ ALIGN_STK_32(pixel, c_dst, 128 * 128,);
+ ALIGN_STK_32(pixel, a_dst, 128 * 128,);
+ const pixel *src = src_buf + 135 * 3 + 3;
+
+ for (int i = 0; i < 135 * 135; i++)
+ src_buf[i] = rand();
+
+ declare_func(void, pixel *dst, ptrdiff_t dst_stride, const pixel *src,
+ ptrdiff_t src_stride, int w, int h, int mx, int my);
+
+ for (int filter = 0; filter < N_2D_FILTERS; filter++)
+ for (int w = 2; w <= 128; w <<= 1)
+ for (int mxy = 0; mxy < 4; mxy++)
+ if (check_func(c->mc[filter], "mc_%s_w%d_%s_%dbpc",
+ filter_names[filter], w, mxy_names[mxy], BITDEPTH))
+ for (int h = imax(w / 4, 2); h <= imin(w * 4, 128); h <<= 1)
+ {+ const int mx = (mxy & 1) ? rand() % 15 + 1 : 0;
+ const int my = (mxy & 2) ? rand() % 15 + 1 : 0;
+
+ call_ref(c_dst, w, src, w, w, h, mx, my);
+ call_new(a_dst, w, src, w, w, h, mx, my);
+ if (memcmp(c_dst, a_dst, w * h * sizeof(*c_dst)))
+ fail();
+
+ bench_new(a_dst, w, src, w, w, h, mx, my);
+ }
+ report("mc");+}
+
+static void check_mct(Dav1dMCDSPContext *const c) {+ ALIGN_STK_32(pixel, src_buf, 135 * 135,);
+ ALIGN_STK_32(coef, c_tmp, 128 * 128,);
+ ALIGN_STK_32(coef, a_tmp, 128 * 128,);
+ const pixel *src = src_buf + 135 * 3 + 3;
+
+ for (int i = 0; i < 135 * 135; i++)
+ src_buf[i] = rand();
+
+ declare_func(void, coef *tmp, const pixel *src, ptrdiff_t src_stride,
+ int w, int h, int mx, int my);
+
+ for (int filter = 0; filter < N_2D_FILTERS; filter++)
+ for (int w = 4; w <= 128; w <<= 1)
+ for (int mxy = 0; mxy < 4; mxy++)
+ if (check_func(c->mct[filter], "mct_%s_w%d_%s_%dbpc",
+ filter_names[filter], w, mxy_names[mxy], BITDEPTH))
+ for (int h = imax(w / 4, 4); h <= imin(w * 4, 128); h <<= 1)
+ {+ const int mx = (mxy & 1) ? rand() % 15 + 1 : 0;
+ const int my = (mxy & 2) ? rand() % 15 + 1 : 0;
+
+ call_ref(c_tmp, src, w, w, h, mx, my);
+ call_new(a_tmp, src, w, w, h, mx, my);
+ if (memcmp(c_tmp, a_tmp, w * h * sizeof(*c_tmp)))
+ fail();
+
+ bench_new(a_tmp, src, w, w, h, mx, my);
+ }
+ report("mct");+}
+
+static void init_tmp(Dav1dMCDSPContext *const c, pixel *const buf,
+ coef (*const tmp)[128 * 128])
+{+ for (int i = 0; i < 2; i++) {+ for (int j = 0; j < 135 * 135; j++)
+ buf[j] = rand();
+ c->mct[rand() % N_2D_FILTERS](tmp[i], buf + 135 * 3 + 3,
+ 128 * sizeof(pixel), 128, 128,
+ rand() & 15, rand() & 15);
+ }
+}
+
+static void check_avg(Dav1dMCDSPContext *const c) {+ ALIGN_STK_32(coef, tmp, 2, [128 * 128]);
+ ALIGN_STK_32(pixel, c_dst, 135 * 135,);
+ ALIGN_STK_32(pixel, a_dst, 128 * 128,);
+
+ init_tmp(c, c_dst, tmp);
+
+ declare_func(void, pixel *dst, ptrdiff_t dst_stride, const coef *tmp1,
+ const coef *tmp2, int w, int h);
+
+ for (int w = 4; w <= 128; w <<= 1)
+ if (check_func(c->avg, "avg_w%d_%dbpc", w, BITDEPTH))
+ for (int h = imax(w / 4, 4); h <= imin(w * 4, 128); h <<= 1)
+ {+ call_ref(c_dst, w, tmp[0], tmp[1], w, h);
+ call_new(a_dst, w, tmp[0], tmp[1], w, h);
+ if (memcmp(c_dst, a_dst, w * h * sizeof(*c_dst)))
+ fail();
+
+ bench_new(a_dst, w, tmp[0], tmp[1], w, h);
+ }
+ report("avg");+}
+
+static void check_w_avg(Dav1dMCDSPContext *const c) {+ ALIGN_STK_32(coef, tmp, 2, [128 * 128]);
+ ALIGN_STK_32(pixel, c_dst, 135 * 135,);
+ ALIGN_STK_32(pixel, a_dst, 128 * 128,);
+
+ init_tmp(c, c_dst, tmp);
+
+ declare_func(void, pixel *dst, ptrdiff_t dst_stride, const coef *tmp1,
+ const coef *tmp2, int w, int h, int weight);
+
+ for (int w = 4; w <= 128; w <<= 1)
+ if (check_func(c->w_avg, "w_avg_w%d_%dbpc", w, BITDEPTH))
+ for (int h = imax(w / 4, 4); h <= imin(w * 4, 128); h <<= 1)
+ {+ int weight = rand() % 15 + 1;
+
+ call_ref(c_dst, w, tmp[0], tmp[1], w, h, weight);
+ call_new(a_dst, w, tmp[0], tmp[1], w, h, weight);
+ if (memcmp(c_dst, a_dst, w * h * sizeof(*c_dst)))
+ fail();
+
+ bench_new(a_dst, w, tmp[0], tmp[1], w, h, weight);
+ }
+ report("w_avg");+}
+
+static void check_mask(Dav1dMCDSPContext *const c) {+ ALIGN_STK_32(coef, tmp, 2, [128 * 128]);
+ ALIGN_STK_32(pixel, c_dst, 135 * 135,);
+ ALIGN_STK_32(pixel, a_dst, 128 * 128,);
+ ALIGN_STK_32(uint8_t, mask, 128 * 128,);
+
+ init_tmp(c, c_dst, tmp);
+ for (int i = 0; i < 128 * 128; i++)
+ mask[i] = rand() % 65;
+
+ declare_func(void, pixel *dst, ptrdiff_t dst_stride, const coef *tmp1,
+ const coef *tmp2, int w, int h, const uint8_t *mask);
+
+ for (int w = 4; w <= 128; w <<= 1)
+ if (check_func(c->mask, "mask_w%d_%dbpc", w, BITDEPTH))
+ for (int h = imax(w / 4, 4); h <= imin(w * 4, 128); h <<= 1)
+ {+ call_ref(c_dst, w, tmp[0], tmp[1], w, h, mask);
+ call_new(a_dst, w, tmp[0], tmp[1], w, h, mask);
+ if (memcmp(c_dst, a_dst, w * h * sizeof(*c_dst)))
+ fail();
+
+ bench_new(a_dst, w, tmp[0], tmp[1], w, h, mask);
+ }
+ report("mask");+}
+
+static void check_w_mask(Dav1dMCDSPContext *const c) {+ ALIGN_STK_32(coef, tmp, 2, [128 * 128]);
+ ALIGN_STK_32(pixel, c_dst, 135 * 135,);
+ ALIGN_STK_32(pixel, a_dst, 128 * 128,);
+ ALIGN_STK_32(uint8_t, c_mask, 128 * 128,);
+ ALIGN_STK_32(uint8_t, a_mask, 128 * 128,);
+
+ init_tmp(c, c_dst, tmp);
+
+ declare_func(void, pixel *dst, ptrdiff_t dst_stride, const coef *tmp1,
+ const coef *tmp2, int w, int h, uint8_t *mask, int sign);
+
+ static const uint16_t ss[] = { 444, 422, 420 };+
+ for (int i = 0; i < 3; i++)
+ for (int w = 4; w <= 128; w <<= 1)
+ if (check_func(c->w_mask[i], "w_mask_%d_w%d_%dbpc", ss[i], w,
+ BITDEPTH))
+ for (int h = imax(w / 4, 4); h <= imin(w * 4, 128); h <<= 1)
+ {+ int sign = rand() & 1;
+
+ call_ref(c_dst, w, tmp[0], tmp[1], w, h, c_mask, sign);
+ call_new(a_dst, w, tmp[0], tmp[1], w, h, a_mask, sign);
+ if (memcmp(c_dst, a_dst, w * h * sizeof(*c_dst)) ||
+ memcmp(c_mask, a_mask, (w * h * sizeof(*c_mask)) >> i))
+ {+ fail();
+ }
+
+ bench_new(a_dst, w, tmp[0], tmp[1], w, h, a_mask, sign);
+ }
+ report("w_mask");+}
+
+void bitfn(checkasm_check_mc)(void) {+ Dav1dMCDSPContext c;
+ bitfn(dav1d_mc_dsp_init)(&c);
+
+ check_mc(&c);
+ check_mct(&c);
+ check_avg(&c);
+ check_w_avg(&c);
+ check_mask(&c);
+ check_w_mask(&c);
+}