android_external_ffmpeg/tests/checkasm/rpi_sand.c
2025-06-24 22:46:48 -05:00

122 lines
4.2 KiB
C

/*
* Copyright (c) 2023 John Cox
*
* 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 "libavutil/common.h"
#include "libavutil/rpi_sand_fns.h"
#if ARCH_ARM
#include "libavutil/arm/cpu.h"
#include "libavutil/arm/rpi_sand_neon.h"
#elif ARCH_AARCH64
#include "libavutil/aarch64/cpu.h"
#include "libavutil/aarch64/rpi_sand_neon.h"
#else
#define have_neon(flags) 0
#define ff_rpi_sand30_lines_to_planar_y16 NULL
#define ff_rpi_sand30_lines_to_planar_c16 NULL
#endif
static inline uint32_t pack30(unsigned int a, unsigned int b, unsigned int c)
{
return (a & 0x3ff) | ((b & 0x3ff) << 10) | ((c & 0x3ff) << 20);
}
void checkasm_check_rpi_sand(void)
{
const unsigned int w = 1280;
const unsigned int h = 66;
const unsigned int stride1 = 128;
const unsigned int stride2 = h*3/2;
const unsigned int ssize = ((w+95)/96)*128*h*3/2;
const unsigned int ysize = ((w + 32) * (h + 32) * 2);
uint8_t * sbuf0 = malloc(ssize);
uint8_t * sbuf1 = malloc(ssize);
uint8_t * ybuf0 = malloc(ysize);
uint8_t * ybuf1 = malloc(ysize);
uint8_t * vbuf0 = malloc(ysize);
uint8_t * vbuf1 = malloc(ysize);
uint8_t * yframe0 = (w + 32) * 16 + ybuf0;
uint8_t * yframe1 = (w + 32) * 16 + ybuf1;
uint8_t * vframe0 = (w + 32) * 16 + vbuf0;
uint8_t * vframe1 = (w + 32) * 16 + vbuf1;
unsigned int i;
for (i = 0; i != ssize; i += 4)
*(uint32_t*)(sbuf0 + i) = rnd();
memcpy(sbuf1, sbuf0, ssize);
if (check_func(have_neon(av_get_cpu_flags()) ? ff_rpi_sand30_lines_to_planar_y16 : av_rpi_sand30_to_planar_y16, "rpi_sand30_to_planar_y16")) {
declare_func(void, uint8_t * dst, const unsigned int dst_stride,
const uint8_t * src,
unsigned int stride1, unsigned int stride2,
unsigned int _x, unsigned int y,
unsigned int _w, unsigned int h);
memset(ybuf0, 0xbb, ysize);
memset(ybuf1, 0xbb, ysize);
call_ref(yframe0, (w + 32) * 2, sbuf0, stride1, stride2, 0, 0, w, h);
call_new(yframe1, (w + 32) * 2, sbuf1, stride1, stride2, 0, 0, w, h);
if (memcmp(sbuf0, sbuf1, ssize)
|| memcmp(ybuf0, ybuf1, ysize))
fail();
bench_new(ybuf1, (w + 32) * 2, sbuf1, stride1, stride2, 0, 0, w, h);
}
if (check_func(have_neon(av_get_cpu_flags()) ? ff_rpi_sand30_lines_to_planar_c16 : av_rpi_sand30_to_planar_c16, "rpi_sand30_to_planar_c16")) {
declare_func(void, uint8_t * u_dst, const unsigned int u_stride,
uint8_t * v_dst, const unsigned int v_stride,
const uint8_t * src,
unsigned int stride1, unsigned int stride2,
unsigned int _x, unsigned int y,
unsigned int _w, unsigned int h);
memset(ybuf0, 0xbb, ysize);
memset(ybuf1, 0xbb, ysize);
memset(vbuf0, 0xbb, ysize);
memset(vbuf1, 0xbb, ysize);
call_ref(yframe0, (w + 32), vframe0, (w + 32), sbuf0, stride1, stride2, 0, 0, w/2, h/2);
call_new(yframe1, (w + 32), vframe1, (w + 32), sbuf1, stride1, stride2, 0, 0, w/2, h/2);
if (memcmp(sbuf0, sbuf1, ssize)
|| memcmp(ybuf0, ybuf1, ysize)
|| memcmp(vbuf0, vbuf1, ysize))
fail();
bench_new(yframe1, (w + 32), vframe1, (w + 32), sbuf1, stride1, stride2, 0, 0, w/2, h/2);
}
report("sand30");
free(sbuf0);
free(sbuf1);
free(ybuf0);
free(ybuf1);
free(vbuf0);
free(vbuf1);
}