[FFmpeg-devel] [PATCH] checkasm/hevc_mc : add hevc_mc for checkasm

Yingming Fan yingmingfan at gmail.com
Thu Apr 12 15:04:54 EEST 2018


Hello,

Any review?

Yingming Fan

> On Apr 9, 2018, at 10:12 AM, Yingming Fan <yingmingfan at gmail.com> wrote:
> 
> From: Yingming Fan <yingmingfan at gmail.com>
> 
> ---
> Hi, there.
> I plane to submit our arm32 neon codes for qpel and epel.
> While before this i will submit hevc_mc checkasm codes.
> This hevc_mc checkasm codes check every qpel and epel function, including 8 10 and 12 bit.
> Passed test by using 'checkasm --test=hevc_mc' under Linux x86_64 MacOS x86_64 and Linux arm64 platform.
> Also passed FATE test. 
> 
> tests/checkasm/Makefile   |   2 +-
> tests/checkasm/checkasm.c |   1 +
> tests/checkasm/checkasm.h |   1 +
> tests/checkasm/hevc_mc.c  | 547 ++++++++++++++++++++++++++++++++++++++++++++++
> tests/fate/checkasm.mak   |   1 +
> 5 files changed, 551 insertions(+), 1 deletion(-)
> create mode 100644 tests/checkasm/hevc_mc.c
> 
> diff --git a/tests/checkasm/Makefile b/tests/checkasm/Makefile
> index 0233d2f989..e6c94cd676 100644
> --- a/tests/checkasm/Makefile
> +++ b/tests/checkasm/Makefile
> @@ -23,7 +23,7 @@ AVCODECOBJS-$(CONFIG_EXR_DECODER)       += exrdsp.o
> AVCODECOBJS-$(CONFIG_HUFFYUV_DECODER)   += huffyuvdsp.o
> AVCODECOBJS-$(CONFIG_JPEG2000_DECODER)  += jpeg2000dsp.o
> AVCODECOBJS-$(CONFIG_PIXBLOCKDSP)       += pixblockdsp.o
> -AVCODECOBJS-$(CONFIG_HEVC_DECODER)      += hevc_add_res.o hevc_idct.o hevc_sao.o
> +AVCODECOBJS-$(CONFIG_HEVC_DECODER)      += hevc_add_res.o hevc_idct.o hevc_sao.o hevc_mc.o
> AVCODECOBJS-$(CONFIG_UTVIDEO_DECODER)   += utvideodsp.o
> AVCODECOBJS-$(CONFIG_V210_ENCODER)      += v210enc.o
> AVCODECOBJS-$(CONFIG_VP9_DECODER)       += vp9dsp.o
> diff --git a/tests/checkasm/checkasm.c b/tests/checkasm/checkasm.c
> index 20ce56932f..b95efc674d 100644
> --- a/tests/checkasm/checkasm.c
> +++ b/tests/checkasm/checkasm.c
> @@ -117,6 +117,7 @@ static const struct {
>         { "hevc_add_res", checkasm_check_hevc_add_res },
>         { "hevc_idct", checkasm_check_hevc_idct },
>         { "hevc_sao", checkasm_check_hevc_sao },
> +        { "hevc_mc", checkasm_check_hevc_mc },
>     #endif
>     #if CONFIG_HUFFYUV_DECODER
>         { "huffyuvdsp", checkasm_check_huffyuvdsp },
> diff --git a/tests/checkasm/checkasm.h b/tests/checkasm/checkasm.h
> index dcab74de06..5a4a612da7 100644
> --- a/tests/checkasm/checkasm.h
> +++ b/tests/checkasm/checkasm.h
> @@ -58,6 +58,7 @@ void checkasm_check_h264qpel(void);
> void checkasm_check_hevc_add_res(void);
> void checkasm_check_hevc_idct(void);
> void checkasm_check_hevc_sao(void);
> +void checkasm_check_hevc_mc(void);
> void checkasm_check_huffyuvdsp(void);
> void checkasm_check_jpeg2000dsp(void);
> void checkasm_check_llviddsp(void);
> diff --git a/tests/checkasm/hevc_mc.c b/tests/checkasm/hevc_mc.c
> new file mode 100644
> index 0000000000..018f322c11
> --- /dev/null
> +++ b/tests/checkasm/hevc_mc.c
> @@ -0,0 +1,547 @@
> +/*
> + * Copyright (c) 2018 Yingming Fan <yingmingfan at gmail.com>
> + *
> + * 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 "libavutil/intreadwrite.h"
> +
> +#include "libavcodec/avcodec.h"
> +
> +#include "libavcodec/hevcdsp.h"
> +
> +#include "checkasm.h"
> +
> +static const uint32_t pixel_mask[3] = { 0xffffffff, 0x03ff03ff, 0x0fff0fff };
> +static const uint32_t idx_width_map[8][2] = {{1, 4}, {3, 8}, {4, 12}, {5, 16}, {6, 24}, {7, 32}, {8, 48}, {9, 64}};
> +#define SIZEOF_PIXEL ((bit_depth + 7) / 8)
> +#define PIXEL_STRIDE (AV_INPUT_BUFFER_PADDING_SIZE + MAX_PB_SIZE + AV_INPUT_BUFFER_PADDING_SIZE)
> +#define BUF_SIZE ((MAX_PB_SIZE+4+4) * PIXEL_STRIDE * 2)
> +
> +#define randomize_buffers(buf0, buf1, size)                 \
> +    do {                                                    \
> +        uint32_t mask = pixel_mask[(bit_depth - 8) >> 1];   \
> +        int k;                                              \
> +        for (k = 0; k < size; k += 4) {                     \
> +            uint32_t r = rnd() & mask;                      \
> +            AV_WN32A(buf0 + k, r);                          \
> +            AV_WN32A(buf1 + k, r);                          \
> +        }                                                   \
> +    } while (0)
> +
> +#define randomize_buffers2(buf0, buf1, size)                \
> +    do {                                                    \
> +        uint32_t mask = 0x1fff1fff;                         \
> +        int k;                                              \
> +        for (k = 0; k < size; k += 4) {                     \
> +            uint32_t r = rnd() & mask;                      \
> +            AV_WN32A(buf0 + k, r);                          \
> +            AV_WN32A(buf1 + k, r);                          \
> +        }                                                   \
> +    } while (0)
> +
> +static int get_random(min, max, mod)
> +{
> +    int value = rnd()%mod;
> +    int sign = rnd()&0x1 == 0 ? 1 : -1;
> +    return av_clip(sign*value, min, max);
> +}
> +
> +static void check_put_hevc_qpel(HEVCDSPContext h, int bit_depth)
> +{
> +    int i, x, y;
> +    LOCAL_ALIGNED_32(uint8_t, dst0, [BUF_SIZE]);
> +    LOCAL_ALIGNED_32(uint8_t, dst1, [BUF_SIZE]);
> +    LOCAL_ALIGNED_32(uint8_t, src0, [BUF_SIZE]);
> +    LOCAL_ALIGNED_32(uint8_t, src1, [BUF_SIZE]);
> +
> +    for (i = 0; i < FF_ARRAY_ELEMS(idx_width_map); i++) {
> +        for (y = 0; y < 4; y++) {
> +            for (x = 0; x < 4; x++) {
> +                int idx = idx_width_map[i][0];
> +                int width = idx_width_map[i][1];
> +                int height= width;
> +                ptrdiff_t stride = PIXEL_STRIDE*SIZEOF_PIXEL;
> +                int offset = (AV_INPUT_BUFFER_PADDING_SIZE+4*PIXEL_STRIDE)*SIZEOF_PIXEL;
> +                declare_func_emms(AV_CPU_FLAG_MMX, void, int16_t *dst, uint8_t *src, ptrdiff_t srcstride,
> +                                  int height, intptr_t mx, intptr_t my, int width);
> +
> +                randomize_buffers(src0, src1, BUF_SIZE);
> +                memset(dst0, 0, BUF_SIZE);
> +                memset(dst1, 0, BUF_SIZE);
> +
> +                if (check_func(h.put_hevc_qpel[idx][!!y][!!x], "put_hevc_qpel_%dx%d_%d_%d_%d", width, height,
> +                    x, y, bit_depth)) {
> +                    call_ref((int16_t *)dst0, src0 + offset, stride, height, x, y, width);
> +                    call_new((int16_t *)dst1, src1 + offset, stride, height, x, y, width);
> +                    if (memcmp(dst0, dst1, BUF_SIZE))
> +                        fail();
> +                    bench_new((int16_t *)dst1, src1 + offset, stride, height, x, y, width);
> +                }
> +            }
> +        }
> +    }
> +}
> +
> +static void check_put_hevc_epel(HEVCDSPContext h, int bit_depth)
> +{
> +    int i, x, y;
> +    LOCAL_ALIGNED_32(uint8_t, dst0, [BUF_SIZE]);
> +    LOCAL_ALIGNED_32(uint8_t, dst1, [BUF_SIZE]);
> +    LOCAL_ALIGNED_32(uint8_t, src0, [BUF_SIZE]);
> +    LOCAL_ALIGNED_32(uint8_t, src1, [BUF_SIZE]);
> +
> +    for (i = 0; i < FF_ARRAY_ELEMS(idx_width_map); i++) {
> +        for (y = 0; y < 8; y++) {
> +            for (x = 0; x < 8; x++) {
> +                int idx = idx_width_map[i][0];
> +                int width = idx_width_map[i][1];
> +                int height= width;
> +                ptrdiff_t stride = PIXEL_STRIDE*SIZEOF_PIXEL;
> +                int offset = (AV_INPUT_BUFFER_PADDING_SIZE+4*PIXEL_STRIDE)*SIZEOF_PIXEL;
> +                declare_func_emms(AV_CPU_FLAG_MMX, void, int16_t *dst, uint8_t *src, ptrdiff_t srcstride,
> +                                  int height, intptr_t mx, intptr_t my, int width);
> +
> +                randomize_buffers(src0, src1, BUF_SIZE);
> +                memset(dst0, 0, BUF_SIZE);
> +                memset(dst1, 0, BUF_SIZE);
> +
> +                if (check_func(h.put_hevc_epel[idx][!!y][!!x], "put_hevc_epel_%dx%d_%d_%d_%d", width, height,
> +                    x, y, bit_depth)) {
> +                    call_ref((int16_t *)dst0, src0 + offset, stride, height, x, y, width);
> +                    call_new((int16_t *)dst1, src1 + offset, stride, height, x, y, width);
> +                    if (memcmp(dst0, dst1, BUF_SIZE))
> +                        fail();
> +                    bench_new((int16_t *)dst1, src1 + offset, stride, height, x, y, width);
> +                }
> +            }
> +        }
> +    }
> +}
> +
> +static void check_put_hevc_qpel_uni(HEVCDSPContext h, int bit_depth)
> +{
> +    int i, x, y;
> +    LOCAL_ALIGNED_32(uint8_t, dst0, [BUF_SIZE]);
> +    LOCAL_ALIGNED_32(uint8_t, dst1, [BUF_SIZE]);
> +    LOCAL_ALIGNED_32(uint8_t, src0, [BUF_SIZE]);
> +    LOCAL_ALIGNED_32(uint8_t, src1, [BUF_SIZE]);
> +
> +    for (i = 0; i < FF_ARRAY_ELEMS(idx_width_map); i++) {
> +        for (y = 0; y < 4; y++) {
> +            for (x = 0; x < 4; x++) {
> +                int idx = idx_width_map[i][0];
> +                int width = idx_width_map[i][1];
> +                int height= width;
> +                ptrdiff_t stride = PIXEL_STRIDE*SIZEOF_PIXEL;
> +                int offset = (AV_INPUT_BUFFER_PADDING_SIZE+4*PIXEL_STRIDE)*SIZEOF_PIXEL;
> +                declare_func_emms(AV_CPU_FLAG_MMX, void, uint8_t *dst, ptrdiff_t dststride, uint8_t *src,
> +                                  ptrdiff_t srcstride, int height, intptr_t mx, intptr_t my, int width);
> +
> +                randomize_buffers(src0, src1, BUF_SIZE);
> +                memset(dst0, 0, BUF_SIZE);
> +                memset(dst1, 0, BUF_SIZE);
> +
> +                if (check_func(h.put_hevc_qpel_uni[idx][!!y][!!x], "put_hevc_qpel_uni_%dx%d_%d_%d_%d", width, height,
> +                    x, y, bit_depth)) {
> +                    call_ref(dst0, stride, src0 + offset, stride, height, x, y, width);
> +                    call_new(dst1, stride, src1 + offset, stride, height, x, y, width);
> +                    if (memcmp(dst0, dst1, BUF_SIZE))
> +                        fail();
> +                    bench_new(dst1, stride, src1 + offset, stride, height, x, y, width);
> +                }
> +            }
> +        }
> +    }
> +}
> +
> +static void check_put_hevc_epel_uni(HEVCDSPContext h, int bit_depth)
> +{
> +    int i, x, y;
> +    LOCAL_ALIGNED_32(uint8_t, dst0, [BUF_SIZE]);
> +    LOCAL_ALIGNED_32(uint8_t, dst1, [BUF_SIZE]);
> +    LOCAL_ALIGNED_32(uint8_t, src0, [BUF_SIZE]);
> +    LOCAL_ALIGNED_32(uint8_t, src1, [BUF_SIZE]);
> +
> +    for (i = 0; i < FF_ARRAY_ELEMS(idx_width_map); i++) {
> +        for (y = 0; y < 8; y++) {
> +            for (x = 0; x < 8; x++) {
> +                int idx = idx_width_map[i][0];
> +                int width = idx_width_map[i][1];
> +                int height= width;
> +                ptrdiff_t stride = PIXEL_STRIDE*SIZEOF_PIXEL;
> +                int offset = (AV_INPUT_BUFFER_PADDING_SIZE+4*PIXEL_STRIDE)*SIZEOF_PIXEL;
> +                declare_func_emms(AV_CPU_FLAG_MMX, void, uint8_t *dst, ptrdiff_t dststride, uint8_t *src,
> +                                  ptrdiff_t srcstride, int height, intptr_t mx, intptr_t my, int width);
> +
> +                randomize_buffers(src0, src1, BUF_SIZE);
> +                memset(dst0, 0, BUF_SIZE);
> +                memset(dst1, 0, BUF_SIZE);
> +
> +                if (check_func(h.put_hevc_epel_uni[idx][!!y][!!x], "put_hevc_epel_uni_%dx%d_%d_%d_%d", width, height,
> +                    x, y, bit_depth)) {
> +                    call_ref(dst0, stride, src0 + offset, stride, height, x, y, width);
> +                    call_new(dst1, stride, src1 + offset, stride, height, x, y, width);
> +                    if (memcmp(dst0, dst1, BUF_SIZE))
> +                        fail();
> +                    bench_new(dst1, stride, src1 + offset, stride, height, x, y, width);
> +                }
> +            }
> +        }
> +    }
> +}
> +
> +static void check_put_hevc_qpel_bi(HEVCDSPContext h, int bit_depth)
> +{
> +    int i, x, y;
> +    LOCAL_ALIGNED_32(uint8_t, dst0, [BUF_SIZE]);
> +    LOCAL_ALIGNED_32(uint8_t, dst1, [BUF_SIZE]);
> +    LOCAL_ALIGNED_32(uint8_t, src0, [BUF_SIZE]);
> +    LOCAL_ALIGNED_32(uint8_t, src1, [BUF_SIZE]);
> +    LOCAL_ALIGNED_32(uint8_t, src00, [BUF_SIZE]);
> +    LOCAL_ALIGNED_32(uint8_t, src11, [BUF_SIZE]);
> +
> +    for (i = 0; i < FF_ARRAY_ELEMS(idx_width_map); i++) {
> +        for (y = 0; y < 4; y++) {
> +            for (x = 0; x < 4; x++) {
> +                int idx = idx_width_map[i][0];
> +                int width = idx_width_map[i][1];
> +                int height= width;
> +                ptrdiff_t stride = PIXEL_STRIDE*SIZEOF_PIXEL;
> +                int offset = (AV_INPUT_BUFFER_PADDING_SIZE+4*PIXEL_STRIDE)*SIZEOF_PIXEL;
> +                declare_func_emms(AV_CPU_FLAG_MMX, void, uint8_t *dst, ptrdiff_t dststride, uint8_t *_src,
> +                                  ptrdiff_t _srcstride, int16_t *src2, int height, intptr_t mx, intptr_t my,
> +                                  int width);
> +
> +                randomize_buffers(src0, src1, BUF_SIZE);
> +                randomize_buffers2(src00, src11, BUF_SIZE);
> +                memset(dst0, 0, BUF_SIZE);
> +                memset(dst1, 0, BUF_SIZE);
> +
> +                if (check_func(h.put_hevc_qpel_bi[idx][!!y][!!x], "put_hevc_qpel_bi_%dx%d_%d_%d_%d", width, height,
> +                    x, y, bit_depth)) {
> +                    call_ref(dst0, stride, src0 + offset, stride, (int16_t *)src00, height, x, y, width);
> +                    call_new(dst1, stride, src1 + offset, stride, (int16_t *)src11, height, x, y, width);
> +                    if (memcmp(dst0, dst1, BUF_SIZE))
> +                        fail();
> +                    bench_new(dst1, stride, src1 + offset, stride, (int16_t *)src11, height, x, y, width);
> +                }
> +            }
> +        }
> +    }
> +}
> +
> +static void check_put_hevc_epel_bi(HEVCDSPContext h, int bit_depth)
> +{
> +    int i, x, y;
> +    LOCAL_ALIGNED_32(uint8_t, dst0, [BUF_SIZE]);
> +    LOCAL_ALIGNED_32(uint8_t, dst1, [BUF_SIZE]);
> +    LOCAL_ALIGNED_32(uint8_t, src0, [BUF_SIZE]);
> +    LOCAL_ALIGNED_32(uint8_t, src1, [BUF_SIZE]);
> +    LOCAL_ALIGNED_32(uint8_t, src00, [BUF_SIZE]);
> +    LOCAL_ALIGNED_32(uint8_t, src11, [BUF_SIZE]);
> +
> +    for (i = 0; i < FF_ARRAY_ELEMS(idx_width_map); i++) {
> +        for (y = 0; y < 8; y++) {
> +            for (x = 0; x < 8; x++) {
> +                int idx = idx_width_map[i][0];
> +                int width = idx_width_map[i][1];
> +                int height= width;
> +                ptrdiff_t stride = PIXEL_STRIDE*SIZEOF_PIXEL;
> +                int offset = (AV_INPUT_BUFFER_PADDING_SIZE+4*PIXEL_STRIDE)*SIZEOF_PIXEL;
> +                declare_func_emms(AV_CPU_FLAG_MMX, void, uint8_t *dst, ptrdiff_t dststride, uint8_t *_src,
> +                                  ptrdiff_t _srcstride, int16_t *src2, int height, intptr_t mx, intptr_t my,
> +                                  int width);
> +
> +                randomize_buffers(src0, src1, BUF_SIZE);
> +                randomize_buffers2(src00, src11, BUF_SIZE);
> +                memset(dst0, 0, BUF_SIZE);
> +                memset(dst1, 0, BUF_SIZE);
> +
> +                if (check_func(h.put_hevc_epel_bi[idx][!!y][!!x], "put_hevc_epel_bi_%dx%d_%d_%d_%d", width, height,
> +                    x, y, bit_depth)) {
> +                    call_ref(dst0, stride, src0 + offset, stride, (int16_t *)src00, height, x, y, width);
> +                    call_new(dst1, stride, src1 + offset, stride, (int16_t *)src11, height, x, y, width);
> +                    if (memcmp(dst0, dst1, BUF_SIZE))
> +                        fail();
> +                    bench_new(dst1, stride, src1 + offset, stride, (int16_t *)src11, height, x, y, width);
> +                }
> +            }
> +        }
> +    }
> +}
> +
> +static void check_put_hevc_qpel_uni_w(HEVCDSPContext h, int bit_depth)
> +{
> +    int i, x, y;
> +    LOCAL_ALIGNED_32(uint8_t, dst0, [BUF_SIZE]);
> +    LOCAL_ALIGNED_32(uint8_t, dst1, [BUF_SIZE]);
> +    LOCAL_ALIGNED_32(uint8_t, src0, [BUF_SIZE]);
> +    LOCAL_ALIGNED_32(uint8_t, src1, [BUF_SIZE]);
> +    LOCAL_ALIGNED_32(uint8_t, src00, [BUF_SIZE]);
> +    LOCAL_ALIGNED_32(uint8_t, src11, [BUF_SIZE]);
> +
> +    for (i = 0; i < FF_ARRAY_ELEMS(idx_width_map); i++) {
> +        for (y = 0; y < 4; y++) {
> +            for (x = 0; x < 4; x++) {
> +                int idx = idx_width_map[i][0];
> +                int width = idx_width_map[i][1];
> +                int height= width;
> +                int denom = rnd()%8;
> +                int wx = (1<<denom) + get_random(-127, 128, 129);
> +                int ox = get_random(-127, 128, 129);
> +                ptrdiff_t stride = PIXEL_STRIDE*SIZEOF_PIXEL;
> +                int offset = (AV_INPUT_BUFFER_PADDING_SIZE+4*PIXEL_STRIDE)*SIZEOF_PIXEL;
> +                declare_func_emms(AV_CPU_FLAG_MMX, void, uint8_t *_dst, ptrdiff_t _dststride, uint8_t *_src, ptrdiff_t _srcstride,
> +                                  int height, int denom, int wx, int ox, intptr_t mx, intptr_t my, int width);
> +
> +                randomize_buffers(src0, src1, BUF_SIZE);
> +                randomize_buffers(src00, src11, BUF_SIZE);
> +                memset(dst0, 0, BUF_SIZE);
> +                memset(dst1, 0, BUF_SIZE);
> +
> +                if (check_func(h.put_hevc_qpel_uni_w[idx][!!y][!!x], "put_hevc_qpel_uni_w_%dx%d_%d_%d_%d", width, height,
> +                    x, y, bit_depth)) {
> +                    call_ref(dst0, stride, src0 + offset, stride, height, denom, wx, ox, x, y, width);
> +                    call_new(dst1, stride, src1 + offset, stride, height, denom, wx, ox, x, y, width);
> +                    if (memcmp(dst0, dst1, BUF_SIZE))
> +                        fail();
> +                    bench_new(dst1, stride, src1 + offset, stride, height, denom, wx, ox, x, y, width);
> +                }
> +            }
> +        }
> +    }
> +}
> +
> +static void check_put_hevc_epel_uni_w(HEVCDSPContext h, int bit_depth)
> +{
> +    int i, x, y;
> +    LOCAL_ALIGNED_32(uint8_t, dst0, [BUF_SIZE]);
> +    LOCAL_ALIGNED_32(uint8_t, dst1, [BUF_SIZE]);
> +    LOCAL_ALIGNED_32(uint8_t, src0, [BUF_SIZE]);
> +    LOCAL_ALIGNED_32(uint8_t, src1, [BUF_SIZE]);
> +    LOCAL_ALIGNED_32(uint8_t, src00, [BUF_SIZE]);
> +    LOCAL_ALIGNED_32(uint8_t, src11, [BUF_SIZE]);
> +
> +    for (i = 0; i < FF_ARRAY_ELEMS(idx_width_map); i++) {
> +        for (y = 0; y < 8; y++) {
> +            for (x = 0; x < 8; x++) {
> +                int idx = idx_width_map[i][0];
> +                int width = idx_width_map[i][1];
> +                int height= width;
> +                int denom = rnd()%8;
> +                int wx = (1<<denom) + get_random(-127, 128, 129);
> +                int ox = get_random(-127, 128, 129);
> +                ptrdiff_t stride = PIXEL_STRIDE*SIZEOF_PIXEL;
> +                int offset = (AV_INPUT_BUFFER_PADDING_SIZE+4*PIXEL_STRIDE)*SIZEOF_PIXEL;
> +                declare_func_emms(AV_CPU_FLAG_MMX, void, uint8_t *_dst, ptrdiff_t _dststride, uint8_t *_src, ptrdiff_t _srcstride,
> +                                  int height, int denom, int wx, int ox, intptr_t mx, intptr_t my, int width);
> +
> +                randomize_buffers(src0, src1, BUF_SIZE);
> +                randomize_buffers(src00, src11, BUF_SIZE);
> +                memset(dst0, 0, BUF_SIZE);
> +                memset(dst1, 0, BUF_SIZE);
> +
> +                if (check_func(h.put_hevc_epel_uni_w[idx][!!y][!!x], "put_hevc_epel_uni_w_%dx%d_%d_%d_%d", width, height,
> +                    x, y, bit_depth)) {
> +                    call_ref(dst0, stride, src0 + offset, stride, height, denom, wx, ox, x, y, width);
> +                    call_new(dst1, stride, src1 + offset, stride, height, denom, wx, ox, x, y, width);
> +                    if (memcmp(dst0, dst1, BUF_SIZE))
> +                        fail();
> +                    bench_new(dst1, stride, src1 + offset, stride, height, denom, wx, ox, x, y, width);
> +                }
> +            }
> +        }
> +    }
> +}
> +
> +static void check_put_hevc_qpel_bi_w(HEVCDSPContext h, int bit_depth)
> +{
> +    int i, x, y;
> +    LOCAL_ALIGNED_32(uint8_t, dst0, [BUF_SIZE]);
> +    LOCAL_ALIGNED_32(uint8_t, dst1, [BUF_SIZE]);
> +    LOCAL_ALIGNED_32(uint8_t, src0, [BUF_SIZE]);
> +    LOCAL_ALIGNED_32(uint8_t, src1, [BUF_SIZE]);
> +    LOCAL_ALIGNED_32(uint8_t, src00, [BUF_SIZE]);
> +    LOCAL_ALIGNED_32(uint8_t, src11, [BUF_SIZE]);
> +
> +    for (i = 0; i < FF_ARRAY_ELEMS(idx_width_map); i++) {
> +        for (y = 0; y < 4; y++) {
> +            for (x = 0; x < 4; x++) {
> +                int idx = idx_width_map[i][0];
> +                int width = idx_width_map[i][1];
> +                int height= width;
> +                int denom = rnd()%8;
> +                int wx = (1<<denom) + get_random(-127, 128, 129);
> +                int ox = get_random(-127, 128, 129);
> +                int wx1 = (1<<denom) + get_random(-127, 128, 129);
> +                int ox1 = get_random(-127, 128, 129);
> +                ptrdiff_t stride = PIXEL_STRIDE*SIZEOF_PIXEL;
> +                int offset = (AV_INPUT_BUFFER_PADDING_SIZE+4*PIXEL_STRIDE)*SIZEOF_PIXEL;
> +                declare_func_emms(AV_CPU_FLAG_MMX, void, uint8_t *dst, ptrdiff_t dststride, uint8_t *_src, ptrdiff_t _srcstride,
> +                                  int16_t *src2, int height, int denom, int wx0, int wx1,
> +                                  int ox0, int ox1, intptr_t mx, intptr_t my, int width);
> +
> +                randomize_buffers(src0, src1, BUF_SIZE);
> +                randomize_buffers2(src00, src11, BUF_SIZE);
> +                memset(dst0, 0, BUF_SIZE);
> +                memset(dst1, 0, BUF_SIZE);
> +
> +                if (check_func(h.put_hevc_qpel_bi_w[idx][!!y][!!x], "put_hevc_qpel_bi_w_%dx%d_%d_%d_%d", width, height,
> +                    x, y, bit_depth)) {
> +                    call_ref(dst0, stride, src0 + offset, stride, (int16_t *)src00, height, denom, wx, wx1, ox, ox1, x, y, width);
> +                    call_new(dst1, stride, src1 + offset, stride, (int16_t *)src11, height, denom, wx, wx1, ox, ox1, x, y, width);
> +                    if (memcmp(dst0, dst1, BUF_SIZE))
> +                        fail();
> +                    bench_new(dst1, stride, src1 + offset, stride, (int16_t *)src11, height, denom, wx, wx1, ox, ox1, x, y, width);
> +                }
> +            }
> +        }
> +    }
> +}
> +
> +static void check_put_hevc_epel_bi_w(HEVCDSPContext h, int bit_depth)
> +{
> +    int i, x, y;
> +    LOCAL_ALIGNED_32(uint8_t, dst0, [BUF_SIZE]);
> +    LOCAL_ALIGNED_32(uint8_t, dst1, [BUF_SIZE]);
> +    LOCAL_ALIGNED_32(uint8_t, src0, [BUF_SIZE]);
> +    LOCAL_ALIGNED_32(uint8_t, src1, [BUF_SIZE]);
> +    LOCAL_ALIGNED_32(uint8_t, src00, [BUF_SIZE]);
> +    LOCAL_ALIGNED_32(uint8_t, src11, [BUF_SIZE]);
> +
> +    for (i = 0; i < FF_ARRAY_ELEMS(idx_width_map); i++) {
> +        for (y = 0; y < 8; y++) {
> +            for (x = 0; x < 8; x++) {
> +                int idx = idx_width_map[i][0];
> +                int width = idx_width_map[i][1];
> +                int height= width;
> +                int denom = rnd()%8;
> +                int wx = (1<<denom) + get_random(-127, 128, 129);
> +                int ox = get_random(-127, 128, 129);
> +                int wx1 = (1<<denom) + get_random(-127, 128, 129);
> +                int ox1 = get_random(-127, 128, 129);
> +                ptrdiff_t stride = PIXEL_STRIDE*SIZEOF_PIXEL;
> +                int offset = (AV_INPUT_BUFFER_PADDING_SIZE+4*PIXEL_STRIDE)*SIZEOF_PIXEL;
> +                declare_func_emms(AV_CPU_FLAG_MMX, void, uint8_t *dst, ptrdiff_t dststride, uint8_t *_src, ptrdiff_t _srcstride,
> +                                  int16_t *src2, int height, int denom, int wx0, int wx1,
> +                                  int ox0, int ox1, intptr_t mx, intptr_t my, int width);
> +
> +                randomize_buffers(src0, src1, BUF_SIZE);
> +                randomize_buffers2(src00, src11, BUF_SIZE);
> +                memset(dst0, 0, BUF_SIZE);
> +                memset(dst1, 0, BUF_SIZE);
> +
> +                if (check_func(h.put_hevc_epel_bi_w[idx][!!y][!!x], "put_hevc_epel_bi_w_%dx%d_%d_%d_%d", width, height,
> +                    x, y, bit_depth)) {
> +                    call_ref(dst0, stride, src0 + offset, stride, (int16_t *)src00, height, denom, wx, wx1, ox, ox1, x, y, width);
> +                    call_new(dst1, stride, src1 + offset, stride, (int16_t *)src11, height, denom, wx, wx1, ox, ox1, x, y, width);
> +                    if (memcmp(dst0, dst1, BUF_SIZE))
> +                        fail();
> +                    bench_new(dst1, stride, src1 + offset, stride, (int16_t *)src11, height, denom, wx, wx1, ox, ox1, x, y, width);
> +                }
> +            }
> +        }
> +    }
> +}
> +
> +void checkasm_check_hevc_mc(void)
> +{
> +    int bit_depth;
> +
> +    for (bit_depth = 8; bit_depth <= 12; bit_depth += 2) {
> +        HEVCDSPContext h;
> +
> +        ff_hevc_dsp_init(&h, bit_depth);
> +        check_put_hevc_qpel(h, bit_depth);
> +    }
> +    report("put_hevc_qpel");
> +
> +    for (bit_depth = 8; bit_depth <= 12; bit_depth += 2) {
> +        HEVCDSPContext h;
> +
> +        ff_hevc_dsp_init(&h, bit_depth);
> +        check_put_hevc_qpel_uni(h, bit_depth);
> +    }
> +    report("put_hevc_qpel_uni");
> +
> +    for (bit_depth = 8; bit_depth <= 12; bit_depth += 2) {
> +        HEVCDSPContext h;
> +
> +        ff_hevc_dsp_init(&h, bit_depth);
> +        check_put_hevc_qpel_bi(h, bit_depth);
> +    }
> +    report("put_hevc_qpel_bi");
> +
> +    for (bit_depth = 8; bit_depth <= 12; bit_depth += 2) {
> +        HEVCDSPContext h;
> +
> +        ff_hevc_dsp_init(&h, bit_depth);
> +        check_put_hevc_qpel_uni_w(h, bit_depth);
> +    }
> +    report("put_hevc_qpel_uni_w");
> +
> +    for (bit_depth = 8; bit_depth <= 12; bit_depth += 2) {
> +        HEVCDSPContext h;
> +
> +        ff_hevc_dsp_init(&h, bit_depth);
> +        check_put_hevc_qpel_bi_w(h, bit_depth);
> +    }
> +    report("put_hevc_qpel_bi_w");
> +
> +    for (bit_depth = 8; bit_depth <= 12; bit_depth += 2) {
> +        HEVCDSPContext h;
> +
> +        ff_hevc_dsp_init(&h, bit_depth);
> +        check_put_hevc_epel(h, bit_depth);
> +    }
> +    report("put_hevc_epel");
> +
> +    for (bit_depth = 8; bit_depth <= 12; bit_depth += 2) {
> +        HEVCDSPContext h;
> +
> +        ff_hevc_dsp_init(&h, bit_depth);
> +        check_put_hevc_epel_uni(h, bit_depth);
> +    }
> +    report("put_hevc_epel_uni");
> +
> +    for (bit_depth = 8; bit_depth <= 12; bit_depth += 2) {
> +        HEVCDSPContext h;
> +
> +        ff_hevc_dsp_init(&h, bit_depth);
> +        check_put_hevc_epel_bi(h, bit_depth);
> +    }
> +    report("put_hevc_epel_bi");
> +
> +    for (bit_depth = 8; bit_depth <= 12; bit_depth += 2) {
> +        HEVCDSPContext h;
> +
> +        ff_hevc_dsp_init(&h, bit_depth);
> +        check_put_hevc_epel_uni_w(h, bit_depth);
> +    }
> +    report("put_hevc_epel_uni_w");
> +
> +    for (bit_depth = 8; bit_depth <= 12; bit_depth += 2) {
> +        HEVCDSPContext h;
> +
> +        ff_hevc_dsp_init(&h, bit_depth);
> +        check_put_hevc_epel_bi_w(h, bit_depth);
> +    }
> +    report("put_hevc_epel_bi_w");
> +}
> diff --git a/tests/fate/checkasm.mak b/tests/fate/checkasm.mak
> index a722b4a917..aeb47091a1 100644
> --- a/tests/fate/checkasm.mak
> +++ b/tests/fate/checkasm.mak
> @@ -15,6 +15,7 @@ FATE_CHECKASM = fate-checkasm-aacpsdsp                                  \
>                 fate-checkasm-hevc_add_res                              \
>                 fate-checkasm-hevc_idct                                 \
>                 fate-checkasm-hevc_sao                                  \
> +                fate-checkasm-hevc_mc                                   \
>                 fate-checkasm-jpeg2000dsp                               \
>                 fate-checkasm-llviddsp                                  \
>                 fate-checkasm-llviddspenc                               \
> -- 
> 2.15.1 (Apple Git-101)
> 



More information about the ffmpeg-devel mailing list