123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262 |
- /*
- * 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 "libavcodec/aacpsdsp.h"
- #include "libavutil/intfloat.h"
- #include "libavutil/mem_internal.h"
- #include "checkasm.h"
- #define N 32
- #define STRIDE 128
- #define BUF_SIZE (N * STRIDE)
- #define randomize(buf, len) do { \
- int i; \
- for (i = 0; i < len; i++) { \
- const INTFLOAT f = (INTFLOAT)rnd() / UINT_MAX; \
- (buf)[i] = f; \
- } \
- } while (0)
- #define EPS 0.005
- static void clear_less_significant_bits(INTFLOAT *buf, int len, int bits)
- {
- int i;
- for (i = 0; i < len; i++) {
- union av_intfloat32 u = { .f = buf[i] };
- u.i &= (0xffffffff << bits);
- buf[i] = u.f;
- }
- }
- static void test_add_squares(void)
- {
- LOCAL_ALIGNED_16(INTFLOAT, dst0, [BUF_SIZE]);
- LOCAL_ALIGNED_16(INTFLOAT, dst1, [BUF_SIZE]);
- LOCAL_ALIGNED_16(INTFLOAT, src, [BUF_SIZE], [2]);
- declare_func(void, INTFLOAT *dst,
- const INTFLOAT (*src)[2], int n);
- randomize((INTFLOAT *)src, BUF_SIZE * 2);
- randomize(dst0, BUF_SIZE);
- memcpy(dst1, dst0, BUF_SIZE * sizeof(INTFLOAT));
- call_ref(dst0, src, BUF_SIZE);
- call_new(dst1, src, BUF_SIZE);
- if (!float_near_abs_eps_array(dst0, dst1, EPS, BUF_SIZE))
- fail();
- bench_new(dst1, src, BUF_SIZE);
- }
- static void test_mul_pair_single(void)
- {
- LOCAL_ALIGNED_16(INTFLOAT, dst0, [BUF_SIZE], [2]);
- LOCAL_ALIGNED_16(INTFLOAT, dst1, [BUF_SIZE], [2]);
- LOCAL_ALIGNED_16(INTFLOAT, src0, [BUF_SIZE], [2]);
- LOCAL_ALIGNED_16(INTFLOAT, src1, [BUF_SIZE]);
- declare_func(void, INTFLOAT (*dst)[2],
- INTFLOAT (*src0)[2], INTFLOAT *src1, int n);
- randomize((INTFLOAT *)src0, BUF_SIZE * 2);
- randomize(src1, BUF_SIZE);
- call_ref(dst0, src0, src1, BUF_SIZE);
- call_new(dst1, src0, src1, BUF_SIZE);
- if (!float_near_abs_eps_array((float *)dst0, (float *)dst1, EPS, BUF_SIZE * 2))
- fail();
- bench_new(dst1, src0, src1, BUF_SIZE);
- }
- static void test_hybrid_analysis(void)
- {
- LOCAL_ALIGNED_16(INTFLOAT, dst0, [BUF_SIZE], [2]);
- LOCAL_ALIGNED_16(INTFLOAT, dst1, [BUF_SIZE], [2]);
- LOCAL_ALIGNED_16(INTFLOAT, in, [13], [2]);
- LOCAL_ALIGNED_16(INTFLOAT, filter, [N], [8][2]);
- declare_func(void, INTFLOAT (*out)[2], INTFLOAT (*in)[2],
- const INTFLOAT (*filter)[8][2],
- ptrdiff_t stride, int n);
- randomize((INTFLOAT *)in, 13 * 2);
- randomize((INTFLOAT *)filter, N * 8 * 2);
- randomize((INTFLOAT *)dst0, BUF_SIZE * 2);
- memcpy(dst1, dst0, BUF_SIZE * 2 * sizeof(INTFLOAT));
- call_ref(dst0, in, filter, STRIDE, N);
- call_new(dst1, in, filter, STRIDE, N);
- if (!float_near_abs_eps_array((float *)dst0, (float *)dst1, EPS, BUF_SIZE * 2))
- fail();
- bench_new(dst1, in, filter, STRIDE, N);
- }
- static void test_hybrid_analysis_ileave(void)
- {
- LOCAL_ALIGNED_16(INTFLOAT, in, [2], [38][64]);
- LOCAL_ALIGNED_16(INTFLOAT, out0, [91], [32][2]);
- LOCAL_ALIGNED_16(INTFLOAT, out1, [91], [32][2]);
- declare_func(void, INTFLOAT (*out)[32][2], INTFLOAT L[2][38][64],
- int i, int len);
- randomize((INTFLOAT *)out0, 91 * 32 * 2);
- randomize((INTFLOAT *)in, 2 * 38 * 64);
- memcpy(out1, out0, 91 * 32 * 2 * sizeof(INTFLOAT));
- /* len is hardcoded to 32 as that's the only value used in
- libavcodec. asm functions are likely to be optimized
- hardcoding this value in their loops and could fail with
- anything else.
- i is hardcoded to the two values currently used by the
- aac decoder because the arm neon implementation is
- micro-optimized for them and will fail for almost every
- other value. */
- call_ref(out0, in, 3, 32);
- call_new(out1, in, 3, 32);
- /* the function just moves data around, so memcmp is enough */
- if (memcmp(out0, out1, 91 * 32 * 2 * sizeof(INTFLOAT)))
- fail();
- call_ref(out0, in, 5, 32);
- call_new(out1, in, 5, 32);
- if (memcmp(out0, out1, 91 * 32 * 2 * sizeof(INTFLOAT)))
- fail();
- bench_new(out1, in, 3, 32);
- }
- static void test_hybrid_synthesis_deint(void)
- {
- LOCAL_ALIGNED_16(INTFLOAT, out0, [2], [38][64]);
- LOCAL_ALIGNED_16(INTFLOAT, out1, [2], [38][64]);
- LOCAL_ALIGNED_16(INTFLOAT, in, [91], [32][2]);
- declare_func(void, INTFLOAT out[2][38][64], INTFLOAT (*in)[32][2],
- int i, int len);
- randomize((INTFLOAT *)in, 91 * 32 * 2);
- randomize((INTFLOAT *)out0, 2 * 38 * 64);
- memcpy(out1, out0, 2 * 38 * 64 * sizeof(INTFLOAT));
- /* len is hardcoded to 32 as that's the only value used in
- libavcodec. asm functions are likely to be optimized
- hardcoding this value in their loops and could fail with
- anything else.
- i is hardcoded to the two values currently used by the
- aac decoder because the arm neon implementation is
- micro-optimized for them and will fail for almost every
- other value. */
- call_ref(out0, in, 3, 32);
- call_new(out1, in, 3, 32);
- /* the function just moves data around, so memcmp is enough */
- if (memcmp(out0, out1, 2 * 38 * 64 * sizeof(INTFLOAT)))
- fail();
- call_ref(out0, in, 5, 32);
- call_new(out1, in, 5, 32);
- if (memcmp(out0, out1, 2 * 38 * 64 * sizeof(INTFLOAT)))
- fail();
- bench_new(out1, in, 3, 32);
- }
- static void test_stereo_interpolate(PSDSPContext *psdsp)
- {
- int i;
- LOCAL_ALIGNED_16(INTFLOAT, l, [BUF_SIZE], [2]);
- LOCAL_ALIGNED_16(INTFLOAT, r, [BUF_SIZE], [2]);
- LOCAL_ALIGNED_16(INTFLOAT, l0, [BUF_SIZE], [2]);
- LOCAL_ALIGNED_16(INTFLOAT, r0, [BUF_SIZE], [2]);
- LOCAL_ALIGNED_16(INTFLOAT, l1, [BUF_SIZE], [2]);
- LOCAL_ALIGNED_16(INTFLOAT, r1, [BUF_SIZE], [2]);
- LOCAL_ALIGNED_16(INTFLOAT, h, [2], [4]);
- LOCAL_ALIGNED_16(INTFLOAT, h_step, [2], [4]);
- declare_func(void, INTFLOAT (*l)[2], INTFLOAT (*r)[2],
- INTFLOAT h[2][4], INTFLOAT h_step[2][4], int len);
- randomize((INTFLOAT *)l, BUF_SIZE * 2);
- randomize((INTFLOAT *)r, BUF_SIZE * 2);
- for (i = 0; i < 2; i++) {
- if (check_func(psdsp->stereo_interpolate[i], "ps_stereo_interpolate%s", i ? "_ipdopd" : "")) {
- memcpy(l0, l, BUF_SIZE * 2 * sizeof(INTFLOAT));
- memcpy(l1, l, BUF_SIZE * 2 * sizeof(INTFLOAT));
- memcpy(r0, r, BUF_SIZE * 2 * sizeof(INTFLOAT));
- memcpy(r1, r, BUF_SIZE * 2 * sizeof(INTFLOAT));
- randomize((INTFLOAT *)h, 2 * 4);
- randomize((INTFLOAT *)h_step, 2 * 4);
- // Clear the least significant 14 bits of h_step, to avoid
- // divergence when accumulating h_step BUF_SIZE times into
- // a float variable which may or may not have extra intermediate
- // precision. Therefore clear roughly log2(BUF_SIZE) less
- // significant bits, to get the same result regardless of any
- // extra precision in the accumulator.
- clear_less_significant_bits((INTFLOAT *)h_step, 2 * 4, 14);
- call_ref(l0, r0, h, h_step, BUF_SIZE);
- call_new(l1, r1, h, h_step, BUF_SIZE);
- if (!float_near_abs_eps_array((float *)l0, (float *)l1, EPS, BUF_SIZE * 2) ||
- !float_near_abs_eps_array((float *)r0, (float *)r1, EPS, BUF_SIZE * 2))
- fail();
- memcpy(l1, l, BUF_SIZE * 2 * sizeof(INTFLOAT));
- memcpy(r1, r, BUF_SIZE * 2 * sizeof(INTFLOAT));
- bench_new(l1, r1, h, h_step, BUF_SIZE);
- }
- }
- }
- void checkasm_check_aacpsdsp(void)
- {
- PSDSPContext psdsp;
- ff_psdsp_init(&psdsp);
- if (check_func(psdsp.add_squares, "ps_add_squares"))
- test_add_squares();
- report("add_squares");
- if (check_func(psdsp.mul_pair_single, "ps_mul_pair_single"))
- test_mul_pair_single();
- report("mul_pair_single");
- if (check_func(psdsp.hybrid_analysis, "ps_hybrid_analysis"))
- test_hybrid_analysis();
- report("hybrid_analysis");
- if (check_func(psdsp.hybrid_analysis_ileave, "ps_hybrid_analysis_ileave"))
- test_hybrid_analysis_ileave();
- report("hybrid_analysis_ileave");
- if (check_func(psdsp.hybrid_synthesis_deint, "ps_hybrid_synthesis_deint"))
- test_hybrid_synthesis_deint();
- report("hybrid_synthesis_deint");
- test_stereo_interpolate(&psdsp);
- report("stereo_interpolate");
- }
|