1/* 2 * Copyright (c) 2022 Ben Avison 3 * 4 * This file is part of FFmpeg. 5 * 6 * FFmpeg is free software; you can redistribute it and/or modify 7 * it under the terms of the GNU General Public License as published by 8 * the Free Software Foundation; either version 2 of the License, or 9 * (at your option) any later version. 10 * 11 * FFmpeg is distributed in the hope that it will be useful, 12 * but WITHOUT ANY WARRANTY; without even the implied warranty of 13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 14 * GNU General Public License for more details. 15 * 16 * You should have received a copy of the GNU General Public License along 17 * with FFmpeg; if not, write to the Free Software Foundation, Inc., 18 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA. 19 */ 20 21#include <string.h> 22 23#include "checkasm.h" 24 25#include "libavcodec/idctdsp.h" 26 27#include "libavutil/common.h" 28#include "libavutil/internal.h" 29#include "libavutil/intreadwrite.h" 30#include "libavutil/mem_internal.h" 31 32#define IDCTDSP_TEST(func) { #func, offsetof(IDCTDSPContext, func) }, 33 34typedef struct { 35 const char *name; 36 size_t offset; 37} test; 38 39#define RANDOMIZE_BUFFER16(name, size) \ 40 do { \ 41 int i; \ 42 for (i = 0; i < size; ++i) { \ 43 uint16_t r = rnd() % 0x201 - 0x100; \ 44 AV_WN16A(name##0 + i, r); \ 45 AV_WN16A(name##1 + i, r); \ 46 } \ 47 } while (0) 48 49#define RANDOMIZE_BUFFER8(name, size) \ 50 do { \ 51 int i; \ 52 for (i = 0; i < size; ++i) { \ 53 uint8_t r = rnd(); \ 54 name##0[i] = r; \ 55 name##1[i] = r; \ 56 } \ 57 } while (0) 58 59static void check_add_put_clamped(void) 60{ 61 /* Source buffers are only as big as needed, since any over-read won't affect results */ 62 LOCAL_ALIGNED_16(int16_t, src0, [64]); 63 LOCAL_ALIGNED_16(int16_t, src1, [64]); 64 /* Destination buffers have borders of one row above/below and 8 columns left/right to catch overflows */ 65 LOCAL_ALIGNED_8(uint8_t, dst0, [10 * 24]); 66 LOCAL_ALIGNED_8(uint8_t, dst1, [10 * 24]); 67 68 AVCodecContext avctx = { 0 }; 69 IDCTDSPContext h; 70 71 const test tests[] = { 72 IDCTDSP_TEST(add_pixels_clamped) 73 IDCTDSP_TEST(put_pixels_clamped) 74 IDCTDSP_TEST(put_signed_pixels_clamped) 75 }; 76 77 ff_idctdsp_init(&h, &avctx); 78 79 for (size_t t = 0; t < FF_ARRAY_ELEMS(tests); ++t) { 80 void (*func)(const int16_t *, uint8_t * ptrdiff_t) = *(void **)((intptr_t) &h + tests[t].offset); 81 if (check_func(func, "idctdsp.%s", tests[t].name)) { 82 declare_func_emms(AV_CPU_FLAG_MMX, void, const int16_t *, uint8_t *, ptrdiff_t); 83 RANDOMIZE_BUFFER16(src, 64); 84 RANDOMIZE_BUFFER8(dst, 10 * 24); 85 call_ref(src0, dst0 + 24 + 8, 24); 86 call_new(src1, dst1 + 24 + 8, 24); 87 if (memcmp(dst0, dst1, 10 * 24)) 88 fail(); 89 bench_new(src1, dst1 + 24 + 8, 24); 90 } 91 } 92} 93 94void checkasm_check_idctdsp(void) 95{ 96 check_add_put_clamped(); 97 report("idctdsp"); 98} 99