tests/checkasm/sbrdsp.c - manifest_repos/ffmpeg - Git at Google

 /*
  * 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/sbrdsp.h"
 #include <float.h>

 #include "checkasm.h"

 #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.0001

 static void test_sum64x5(void)
 {
     LOCAL_ALIGNED_16(INTFLOAT, dst0, [64 + 256]);
     LOCAL_ALIGNED_16(INTFLOAT, dst1, [64 + 256]);

     declare_func(void, INTFLOAT *z);

     randomize((INTFLOAT *)dst0, 64 + 256);
     memcpy(dst1, dst0, (64 + 256) * sizeof(INTFLOAT));
     call_ref(dst0);
     call_new(dst1);
     if (!float_near_abs_eps_array(dst0, dst1, EPS, 64 + 256))
         fail();
     bench_new(dst1);
 }

 static void test_sum_square(void)
 {
     INTFLOAT res0;
     INTFLOAT res1;
     LOCAL_ALIGNED_16(INTFLOAT, src, [256], [2]);
     double t = 4 * 256;

     declare_func_float(INTFLOAT, INTFLOAT (*x)[2], int n);

     randomize((INTFLOAT *)src, 256 * 2);
     res0 = call_ref(src, 256);
     res1 = call_new(src, 256);
     if (!float_near_abs_eps(res0, res1, t * 2 * FLT_EPSILON))
         fail();
     bench_new(src, 256);
 }

 static void test_neg_odd_64(void)
 {
     LOCAL_ALIGNED_16(INTFLOAT, dst0, [64]);
     LOCAL_ALIGNED_16(INTFLOAT, dst1, [64]);

     declare_func(void, INTFLOAT *x);

     randomize((INTFLOAT *)dst0, 64);
     memcpy(dst1, dst0, (64) * sizeof(INTFLOAT));
     call_ref(dst0);
     call_new(dst1);
     if (!float_near_abs_eps_array(dst0, dst1, EPS, 64))
         fail();
     bench_new(dst1);
 }

 static void test_qmf_pre_shuffle(void)
 {
     LOCAL_ALIGNED_16(INTFLOAT, dst0, [128]);
     LOCAL_ALIGNED_16(INTFLOAT, dst1, [128]);

     declare_func(void, INTFLOAT *z);

     randomize((INTFLOAT *)dst0, 128);
     memcpy(dst1, dst0, (128) * sizeof(INTFLOAT));
     call_ref(dst0);
     call_new(dst1);
     if (!float_near_abs_eps_array(dst0, dst1, EPS, 128))
         fail();
     bench_new(dst1);
 }

 static void test_qmf_post_shuffle(void)
 {
     LOCAL_ALIGNED_16(INTFLOAT, src, [64]);
     LOCAL_ALIGNED_16(INTFLOAT, dst0, [32], [2]);
     LOCAL_ALIGNED_16(INTFLOAT, dst1, [32], [2]);

     declare_func(void, INTFLOAT W[32][2], const INTFLOAT *z);

     randomize((INTFLOAT *)src, 64);
     call_ref(dst0, src);
     call_new(dst1, src);
     if (!float_near_abs_eps_array((INTFLOAT *)dst0, (INTFLOAT *)dst1, EPS, 64))
         fail();
     bench_new(dst1, src);
 }

 static void test_qmf_deint_neg(void)
 {
     LOCAL_ALIGNED_16(INTFLOAT, src, [64]);
     LOCAL_ALIGNED_16(INTFLOAT, dst0, [64]);
     LOCAL_ALIGNED_16(INTFLOAT, dst1, [64]);

     declare_func(void, INTFLOAT *v, const INTFLOAT *src);

     randomize((INTFLOAT *)src, 64);
     call_ref(dst0, src);
     call_new(dst1, src);
     if (!float_near_abs_eps_array(dst0, dst1, EPS, 64))
         fail();
     bench_new(dst1, src);
 }

 static void test_qmf_deint_bfly(void)
 {
     LOCAL_ALIGNED_16(INTFLOAT, src0, [64]);
     LOCAL_ALIGNED_16(INTFLOAT, src1, [64]);
     LOCAL_ALIGNED_16(INTFLOAT, dst0, [128]);
     LOCAL_ALIGNED_16(INTFLOAT, dst1, [128]);

     declare_func(void, INTFLOAT *v, const INTFLOAT *src0, const INTFLOAT *src1);

     memset(dst0, 0, 128 * sizeof(INTFLOAT));
     memset(dst1, 0, 128 * sizeof(INTFLOAT));

     randomize((INTFLOAT *)src0, 64);
     randomize((INTFLOAT *)src1, 64);
     call_ref(dst0, src0, src1);
     call_new(dst1, src0, src1);
     if (!float_near_abs_eps_array(dst0, dst1, EPS, 128))
         fail();
     bench_new(dst1, src0, src1);
 }

 static void test_autocorrelate(void)
 {
     LOCAL_ALIGNED_16(INTFLOAT, src, [40], [2]);
     LOCAL_ALIGNED_16(INTFLOAT, dst0, [3], [2][2]);
     LOCAL_ALIGNED_16(INTFLOAT, dst1, [3], [2][2]);

     declare_func(void, const INTFLOAT x[40][2], INTFLOAT phi[3][2][2]);

     memset(dst0, 0, 3 * 2 * 2 * sizeof(INTFLOAT));
     memset(dst1, 0, 3 * 2 * 2 * sizeof(INTFLOAT));

     randomize((INTFLOAT *)src, 80);
     call_ref(src, dst0);
     call_new(src, dst1);
     if (!float_near_abs_eps_array((INTFLOAT *)dst0, (INTFLOAT *)dst1, EPS, 3 * 2 * 2))
         fail();
     bench_new(src, dst1);
 }

 static void test_hf_gen(void)
 {
     LOCAL_ALIGNED_16(INTFLOAT, low, [128], [2]);
     LOCAL_ALIGNED_16(INTFLOAT, alpha0, [2]);
     LOCAL_ALIGNED_16(INTFLOAT, alpha1, [2]);
     LOCAL_ALIGNED_16(INTFLOAT, dst0, [128], [2]);
     LOCAL_ALIGNED_16(INTFLOAT, dst1, [128], [2]);
     INTFLOAT bw = (INTFLOAT)rnd() / UINT_MAX;
     int i;

     declare_func(void, INTFLOAT (*X_high)[2], const INTFLOAT (*X_low)[2],
                        const INTFLOAT alpha0[2], const INTFLOAT alpha1[2],
                        INTFLOAT bw, int start, int end);

     randomize((INTFLOAT *)low, 128 * 2);
     randomize((INTFLOAT *)alpha0, 2);
     randomize((INTFLOAT *)alpha1, 2);
     for (i = 2; i < 64; i += 2) {
         memset(dst0, 0, 128 * 2 * sizeof(INTFLOAT));
         memset(dst1, 0, 128 * 2 * sizeof(INTFLOAT));
         call_ref(dst0, low, alpha0, alpha1, 0.0, i, 128);
         call_new(dst1, low, alpha0, alpha1, 0.0, i, 128);
         if (!float_near_abs_eps_array((INTFLOAT *)dst0, (INTFLOAT *)dst1, EPS, 128 * 2))
             fail();
         bench_new(dst1, low, alpha0, alpha1, bw, i, 128);
     }
 }

 static void test_hf_g_filt(void)
 {
     LOCAL_ALIGNED_16(INTFLOAT, high, [128], [40][2]);
     LOCAL_ALIGNED_16(INTFLOAT, g_filt, [128]);
     LOCAL_ALIGNED_16(INTFLOAT, dst0, [128], [2]);
     LOCAL_ALIGNED_16(INTFLOAT, dst1, [128], [2]);

     declare_func(void, INTFLOAT (*Y)[2], const INTFLOAT (*X_high)[40][2],
                        const INTFLOAT *g_filt, int m_max, intptr_t ixh);

     randomize((INTFLOAT *)high, 128 * 40 * 2);
     randomize((INTFLOAT *)g_filt, 128);

     call_ref(dst0, high, g_filt, 128, 20);
     call_new(dst1, high, g_filt, 128, 20);
     if (!float_near_abs_eps_array((INTFLOAT *)dst0, (INTFLOAT *)dst1, EPS, 128 * 2))
         fail();
     bench_new(dst1, high, g_filt, 128, 20);
 }

 static void test_hf_apply_noise(const SBRDSPContext *sbrdsp)
 {
     LOCAL_ALIGNED_16(AAC_FLOAT, s_m, [128]);
     LOCAL_ALIGNED_16(AAC_FLOAT, q_filt, [128]);
     LOCAL_ALIGNED_16(INTFLOAT, ref, [128], [2]);
     LOCAL_ALIGNED_16(INTFLOAT, dst0, [128], [2]);
     LOCAL_ALIGNED_16(INTFLOAT, dst1, [128], [2]);
     int noise = 0x2a;
     int i, j;

     declare_func(void, INTFLOAT (*Y)[2], const AAC_FLOAT *s_m,
                        const AAC_FLOAT *q_filt, int noise,
                        int kx, int m_max);

     randomize((INTFLOAT *)ref, 128 * 2);
     randomize((INTFLOAT *)s_m, 128);
     randomize((INTFLOAT *)q_filt, 128);

     for (i = 0; i < 4; i++) {
         if (check_func(sbrdsp->hf_apply_noise[i], "hf_apply_noise_%d", i)) {
             for (j = 0; j < 2; j++) {
                 memcpy(dst0, ref, 128 * 2 * sizeof(INTFLOAT));
                 memcpy(dst1, ref, 128 * 2 * sizeof(INTFLOAT));
                 call_ref(dst0, s_m, q_filt, noise, j, 128);
                 call_new(dst1, s_m, q_filt, noise, j, 128);
                 if (!float_near_abs_eps_array((INTFLOAT *)dst0, (INTFLOAT *)dst1, EPS, 128 * 2))
                     fail();
                 bench_new(dst1, s_m, q_filt, noise, j, 128);
             }
         }
     }
 }

 void checkasm_check_sbrdsp(void)
 {
     SBRDSPContext sbrdsp;

     ff_sbrdsp_init(&sbrdsp);

     if (check_func(sbrdsp.sum64x5, "sum64x5"))
         test_sum64x5();
     report("sum64x5");

     if (check_func(sbrdsp.sum_square, "sum_square"))
         test_sum_square();
     report("sum_square");

     if (check_func(sbrdsp.neg_odd_64, "neg_odd_64"))
         test_neg_odd_64();
     report("neg_odd_64");

     if (check_func(sbrdsp.qmf_pre_shuffle, "qmf_pre_shuffle"))
         test_qmf_pre_shuffle();
     report("qmf_pre_shuffle");

     if (check_func(sbrdsp.qmf_post_shuffle, "qmf_post_shuffle"))
         test_qmf_post_shuffle();
     report("qmf_post_shuffle");

     if (check_func(sbrdsp.qmf_deint_neg, "qmf_deint_neg"))
         test_qmf_deint_neg();
     report("qmf_deint_neg");

     if (check_func(sbrdsp.qmf_deint_bfly, "qmf_deint_bfly"))
         test_qmf_deint_bfly();
     report("qmf_deint_bfly");

     if (check_func(sbrdsp.autocorrelate, "autocorrelate"))
         test_autocorrelate();
     report("autocorrelate");

     if (check_func(sbrdsp.hf_gen, "hf_gen"))
         test_hf_gen();
     report("hf_gen");

     if (check_func(sbrdsp.hf_g_filt, "hf_g_filt"))
         test_hf_g_filt();
     report("hf_g_filt");

     test_hf_apply_noise(&sbrdsp);
     report("hf_apply_noise");
 }
	/*
	* 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/sbrdsp.h"
	#include <float.h>

	#include "checkasm.h"

	#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.0001

	static void test_sum64x5(void)
	{
	LOCAL_ALIGNED_16(INTFLOAT, dst0, [64 + 256]);
	LOCAL_ALIGNED_16(INTFLOAT, dst1, [64 + 256]);

	declare_func(void, INTFLOAT *z);

	randomize((INTFLOAT *)dst0, 64 + 256);
	memcpy(dst1, dst0, (64 + 256) * sizeof(INTFLOAT));
	call_ref(dst0);
	call_new(dst1);
	if (!float_near_abs_eps_array(dst0, dst1, EPS, 64 + 256))
	fail();
	bench_new(dst1);
	}

	static void test_sum_square(void)
	{
	INTFLOAT res0;
	INTFLOAT res1;
	LOCAL_ALIGNED_16(INTFLOAT, src, [256], [2]);
	double t = 4 * 256;

	declare_func_float(INTFLOAT, INTFLOAT (*x)[2], int n);

	randomize((INTFLOAT )src, 256 2);
	res0 = call_ref(src, 256);
	res1 = call_new(src, 256);
	if (!float_near_abs_eps(res0, res1, t * 2 * FLT_EPSILON))
	fail();
	bench_new(src, 256);
	}

	static void test_neg_odd_64(void)
	{
	LOCAL_ALIGNED_16(INTFLOAT, dst0, [64]);
	LOCAL_ALIGNED_16(INTFLOAT, dst1, [64]);

	declare_func(void, INTFLOAT *x);

	randomize((INTFLOAT *)dst0, 64);
	memcpy(dst1, dst0, (64) * sizeof(INTFLOAT));
	call_ref(dst0);
	call_new(dst1);
	if (!float_near_abs_eps_array(dst0, dst1, EPS, 64))
	fail();
	bench_new(dst1);
	}

	static void test_qmf_pre_shuffle(void)
	{
	LOCAL_ALIGNED_16(INTFLOAT, dst0, [128]);
	LOCAL_ALIGNED_16(INTFLOAT, dst1, [128]);

	declare_func(void, INTFLOAT *z);

	randomize((INTFLOAT *)dst0, 128);
	memcpy(dst1, dst0, (128) * sizeof(INTFLOAT));
	call_ref(dst0);
	call_new(dst1);
	if (!float_near_abs_eps_array(dst0, dst1, EPS, 128))
	fail();
	bench_new(dst1);
	}

	static void test_qmf_post_shuffle(void)
	{
	LOCAL_ALIGNED_16(INTFLOAT, src, [64]);
	LOCAL_ALIGNED_16(INTFLOAT, dst0, [32], [2]);
	LOCAL_ALIGNED_16(INTFLOAT, dst1, [32], [2]);

	declare_func(void, INTFLOAT W[32][2], const INTFLOAT *z);

	randomize((INTFLOAT *)src, 64);
	call_ref(dst0, src);
	call_new(dst1, src);
	if (!float_near_abs_eps_array((INTFLOAT )dst0, (INTFLOAT )dst1, EPS, 64))
	fail();
	bench_new(dst1, src);
	}

	static void test_qmf_deint_neg(void)
	{
	LOCAL_ALIGNED_16(INTFLOAT, src, [64]);
	LOCAL_ALIGNED_16(INTFLOAT, dst0, [64]);
	LOCAL_ALIGNED_16(INTFLOAT, dst1, [64]);

	declare_func(void, INTFLOAT v, const INTFLOAT src);

	randomize((INTFLOAT *)src, 64);
	call_ref(dst0, src);
	call_new(dst1, src);
	if (!float_near_abs_eps_array(dst0, dst1, EPS, 64))
	fail();
	bench_new(dst1, src);
	}

	static void test_qmf_deint_bfly(void)
	{
	LOCAL_ALIGNED_16(INTFLOAT, src0, [64]);
	LOCAL_ALIGNED_16(INTFLOAT, src1, [64]);
	LOCAL_ALIGNED_16(INTFLOAT, dst0, [128]);
	LOCAL_ALIGNED_16(INTFLOAT, dst1, [128]);

	declare_func(void, INTFLOAT v, const INTFLOAT src0, const INTFLOAT *src1);

	memset(dst0, 0, 128 * sizeof(INTFLOAT));
	memset(dst1, 0, 128 * sizeof(INTFLOAT));

	randomize((INTFLOAT *)src0, 64);
	randomize((INTFLOAT *)src1, 64);
	call_ref(dst0, src0, src1);
	call_new(dst1, src0, src1);
	if (!float_near_abs_eps_array(dst0, dst1, EPS, 128))
	fail();
	bench_new(dst1, src0, src1);
	}

	static void test_autocorrelate(void)
	{
	LOCAL_ALIGNED_16(INTFLOAT, src, [40], [2]);
	LOCAL_ALIGNED_16(INTFLOAT, dst0, [3], [2][2]);
	LOCAL_ALIGNED_16(INTFLOAT, dst1, [3], [2][2]);

	declare_func(void, const INTFLOAT x[40][2], INTFLOAT phi[3][2][2]);

	memset(dst0, 0, 3 * 2 * 2 * sizeof(INTFLOAT));
	memset(dst1, 0, 3 * 2 * 2 * sizeof(INTFLOAT));

	randomize((INTFLOAT *)src, 80);
	call_ref(src, dst0);
	call_new(src, dst1);
	if (!float_near_abs_eps_array((INTFLOAT )dst0, (INTFLOAT )dst1, EPS, 3 * 2 * 2))
	fail();
	bench_new(src, dst1);
	}

	static void test_hf_gen(void)
	{
	LOCAL_ALIGNED_16(INTFLOAT, low, [128], [2]);
	LOCAL_ALIGNED_16(INTFLOAT, alpha0, [2]);
	LOCAL_ALIGNED_16(INTFLOAT, alpha1, [2]);
	LOCAL_ALIGNED_16(INTFLOAT, dst0, [128], [2]);
	LOCAL_ALIGNED_16(INTFLOAT, dst1, [128], [2]);
	INTFLOAT bw = (INTFLOAT)rnd() / UINT_MAX;
	int i;

	declare_func(void, INTFLOAT (X_high)[2], const INTFLOAT (X_low)[2],
	const INTFLOAT alpha0[2], const INTFLOAT alpha1[2],
	INTFLOAT bw, int start, int end);

	randomize((INTFLOAT )low, 128 2);
	randomize((INTFLOAT *)alpha0, 2);
	randomize((INTFLOAT *)alpha1, 2);
	for (i = 2; i < 64; i += 2) {
	memset(dst0, 0, 128 * 2 * sizeof(INTFLOAT));
	memset(dst1, 0, 128 * 2 * sizeof(INTFLOAT));
	call_ref(dst0, low, alpha0, alpha1, 0.0, i, 128);
	call_new(dst1, low, alpha0, alpha1, 0.0, i, 128);
	if (!float_near_abs_eps_array((INTFLOAT )dst0, (INTFLOAT )dst1, EPS, 128 * 2))
	fail();
	bench_new(dst1, low, alpha0, alpha1, bw, i, 128);
	}
	}

	static void test_hf_g_filt(void)
	{
	LOCAL_ALIGNED_16(INTFLOAT, high, [128], [40][2]);
	LOCAL_ALIGNED_16(INTFLOAT, g_filt, [128]);
	LOCAL_ALIGNED_16(INTFLOAT, dst0, [128], [2]);
	LOCAL_ALIGNED_16(INTFLOAT, dst1, [128], [2]);

	declare_func(void, INTFLOAT (Y)[2], const INTFLOAT (X_high)[40][2],
	const INTFLOAT *g_filt, int m_max, intptr_t ixh);

	randomize((INTFLOAT )high, 128 40 * 2);
	randomize((INTFLOAT *)g_filt, 128);

	call_ref(dst0, high, g_filt, 128, 20);
	call_new(dst1, high, g_filt, 128, 20);
	if (!float_near_abs_eps_array((INTFLOAT )dst0, (INTFLOAT )dst1, EPS, 128 * 2))
	fail();
	bench_new(dst1, high, g_filt, 128, 20);
	}

	static void test_hf_apply_noise(const SBRDSPContext *sbrdsp)
	{
	LOCAL_ALIGNED_16(AAC_FLOAT, s_m, [128]);
	LOCAL_ALIGNED_16(AAC_FLOAT, q_filt, [128]);
	LOCAL_ALIGNED_16(INTFLOAT, ref, [128], [2]);
	LOCAL_ALIGNED_16(INTFLOAT, dst0, [128], [2]);
	LOCAL_ALIGNED_16(INTFLOAT, dst1, [128], [2]);
	int noise = 0x2a;
	int i, j;

	declare_func(void, INTFLOAT (Y)[2], const AAC_FLOAT s_m,
	const AAC_FLOAT *q_filt, int noise,
	int kx, int m_max);

	randomize((INTFLOAT )ref, 128 2);
	randomize((INTFLOAT *)s_m, 128);
	randomize((INTFLOAT *)q_filt, 128);

	for (i = 0; i < 4; i++) {
	if (check_func(sbrdsp->hf_apply_noise[i], "hf_apply_noise_%d", i)) {
	for (j = 0; j < 2; j++) {
	memcpy(dst0, ref, 128 * 2 * sizeof(INTFLOAT));
	memcpy(dst1, ref, 128 * 2 * sizeof(INTFLOAT));
	call_ref(dst0, s_m, q_filt, noise, j, 128);
	call_new(dst1, s_m, q_filt, noise, j, 128);
	if (!float_near_abs_eps_array((INTFLOAT )dst0, (INTFLOAT )dst1, EPS, 128 * 2))
	fail();
	bench_new(dst1, s_m, q_filt, noise, j, 128);
	}
	}
	}
	}

	void checkasm_check_sbrdsp(void)
	{
	SBRDSPContext sbrdsp;

	ff_sbrdsp_init(&sbrdsp);

	if (check_func(sbrdsp.sum64x5, "sum64x5"))
	test_sum64x5();
	report("sum64x5");

	if (check_func(sbrdsp.sum_square, "sum_square"))
	test_sum_square();
	report("sum_square");

	if (check_func(sbrdsp.neg_odd_64, "neg_odd_64"))
	test_neg_odd_64();
	report("neg_odd_64");

	if (check_func(sbrdsp.qmf_pre_shuffle, "qmf_pre_shuffle"))
	test_qmf_pre_shuffle();
	report("qmf_pre_shuffle");

	if (check_func(sbrdsp.qmf_post_shuffle, "qmf_post_shuffle"))
	test_qmf_post_shuffle();
	report("qmf_post_shuffle");

	if (check_func(sbrdsp.qmf_deint_neg, "qmf_deint_neg"))
	test_qmf_deint_neg();
	report("qmf_deint_neg");

	if (check_func(sbrdsp.qmf_deint_bfly, "qmf_deint_bfly"))
	test_qmf_deint_bfly();
	report("qmf_deint_bfly");

	if (check_func(sbrdsp.autocorrelate, "autocorrelate"))
	test_autocorrelate();
	report("autocorrelate");

	if (check_func(sbrdsp.hf_gen, "hf_gen"))
	test_hf_gen();
	report("hf_gen");

	if (check_func(sbrdsp.hf_g_filt, "hf_g_filt"))
	test_hf_g_filt();
	report("hf_g_filt");

	test_hf_apply_noise(&sbrdsp);
	report("hf_apply_noise");
	}