[FFmpeg-devel] [PATCH v2 1/3] checkasm/diracdsp: test add_dirac_obmc
Kyosuke Kawakami
kawakami150708 at gmail.com
Thu Nov 14 00:30:08 EET 2024
Signed-off-by: Kyosuke Kawakami <kawakami150708 at gmail.com>
---
tests/checkasm/Makefile | 1 +
tests/checkasm/checkasm.c | 3 ++
tests/checkasm/checkasm.h | 1 +
tests/checkasm/diracdsp.c | 81 +++++++++++++++++++++++++++++++++++++++
tests/fate/checkasm.mak | 1 +
5 files changed, 87 insertions(+)
create mode 100644 tests/checkasm/diracdsp.c
diff --git a/tests/checkasm/Makefile b/tests/checkasm/Makefile
index ae324ced3f..c7268d836e 100644
--- a/tests/checkasm/Makefile
+++ b/tests/checkasm/Makefile
@@ -29,6 +29,7 @@ AVCODECOBJS-$(CONFIG_AAC_DECODER) += aacpsdsp.o \
AVCODECOBJS-$(CONFIG_AAC_ENCODER) += aacencdsp.o
AVCODECOBJS-$(CONFIG_ALAC_DECODER) += alacdsp.o
AVCODECOBJS-$(CONFIG_DCA_DECODER) += synth_filter.o
+AVCODECOBJS-$(CONFIG_DIRAC_DECODER) += diracdsp.o
AVCODECOBJS-$(CONFIG_EXR_DECODER) += exrdsp.o
AVCODECOBJS-$(CONFIG_FLAC_DECODER) += flacdsp.o
AVCODECOBJS-$(CONFIG_HUFFYUV_DECODER) += huffyuvdsp.o
diff --git a/tests/checkasm/checkasm.c b/tests/checkasm/checkasm.c
index c9d2b5faf1..fb307af0ae 100644
--- a/tests/checkasm/checkasm.c
+++ b/tests/checkasm/checkasm.c
@@ -138,6 +138,9 @@ static const struct {
#if CONFIG_DCA_DECODER
{ "synth_filter", checkasm_check_synth_filter },
#endif
+ #if CONFIG_DIRAC_DECODER
+ { "diracdsp", checkasm_check_diracdsp },
+ #endif
#if CONFIG_EXR_DECODER
{ "exrdsp", checkasm_check_exrdsp },
#endif
diff --git a/tests/checkasm/checkasm.h b/tests/checkasm/checkasm.h
index 866eef01e9..0ba5c3040d 100644
--- a/tests/checkasm/checkasm.h
+++ b/tests/checkasm/checkasm.h
@@ -84,6 +84,7 @@ void checkasm_check_blend(void);
void checkasm_check_blockdsp(void);
void checkasm_check_bswapdsp(void);
void checkasm_check_colorspace(void);
+void checkasm_check_diracdsp(void);
void checkasm_check_exrdsp(void);
void checkasm_check_fdctdsp(void);
void checkasm_check_fixed_dsp(void);
diff --git a/tests/checkasm/diracdsp.c b/tests/checkasm/diracdsp.c
new file mode 100644
index 0000000000..70cd788320
--- /dev/null
+++ b/tests/checkasm/diracdsp.c
@@ -0,0 +1,81 @@
+/*
+ * Copyright (c) 2024 Kyosuke Kawakami
+ *
+ * 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 "checkasm.h"
+
+#include "libavcodec/diracdsp.h"
+
+#include "libavutil/intreadwrite.h"
+#include "libavutil/mem_internal.h"
+
+#define RANDOMIZE_DESTS(name, size) \
+ do { \
+ int i; \
+ for (i = 0; i < size; ++i) { \
+ uint16_t r = rnd(); \
+ AV_WN16A(name##0 + i, r); \
+ AV_WN16A(name##1 + i, r); \
+ } \
+ } while (0)
+
+#define RANDOMIZE_BUFFER8(name, size) \
+ do { \
+ int i; \
+ for (i = 0; i < size; ++i) { \
+ uint8_t r = rnd(); \
+ name[i] = r; \
+ } \
+ } while (0)
+
+#define OBMC_STRIDE 32
+
+#define CHECK_ADD_OBMC(func_index, yblen, xblen) \
+ static void check_add_obmc ## xblen(void) \
+ { \
+ LOCAL_ALIGNED_8(uint8_t, src, [yblen * xblen]); \
+ LOCAL_ALIGNED_16(uint16_t, dst0, [yblen * xblen]); \
+ LOCAL_ALIGNED_16(uint16_t, dst1, [yblen * xblen]); \
+ LOCAL_ALIGNED_8(uint8_t, obmc_weight, [yblen * OBMC_STRIDE]); \
+ DiracDSPContext h; \
+ ff_diracdsp_init(&h); \
+ if (check_func(h.add_dirac_obmc[func_index], "diracdsp.add_dirac_obmc_%d", xblen)) { \
+ declare_func(void, uint16_t*, const uint8_t*, int, const uint8_t *, int); \
+ RANDOMIZE_BUFFER8(src, yblen * xblen); \
+ RANDOMIZE_DESTS(dst, yblen * xblen); \
+ RANDOMIZE_BUFFER8(obmc_weight, yblen * OBMC_STRIDE); \
+ call_ref(dst0, src, xblen, obmc_weight, yblen); \
+ call_new(dst1, src, xblen, obmc_weight, yblen); \
+ if (memcmp(dst0, dst1, yblen * xblen)) \
+ fail(); \
+ bench_new(dst1, src, xblen, obmc_weight, yblen); \
+ } \
+ } \
+
+CHECK_ADD_OBMC(0, 64, 8)
+CHECK_ADD_OBMC(1, 64, 16)
+CHECK_ADD_OBMC(2, 64, 32)
+
+void checkasm_check_diracdsp(void)
+{
+ check_add_obmc8();
+ check_add_obmc16();
+ check_add_obmc32();
+ report("diracdsp");
+}
diff --git a/tests/fate/checkasm.mak b/tests/fate/checkasm.mak
index d1396cb641..8a2c04e1cd 100644
--- a/tests/fate/checkasm.mak
+++ b/tests/fate/checkasm.mak
@@ -7,6 +7,7 @@ FATE_CHECKASM = fate-checkasm-aacencdsp \
fate-checkasm-av_tx \
fate-checkasm-blockdsp \
fate-checkasm-bswapdsp \
+ fate-checkasm-diracdsp \
fate-checkasm-exrdsp \
fate-checkasm-fdctdsp \
fate-checkasm-fixed_dsp \
--
2.47.0
More information about the ffmpeg-devel
mailing list