Home | History | Annotate | Download | only in tests
      1 /* Copyright (c) 2013 The Chromium OS Authors. All rights reserved.
      2  * Use of this source code is governed by a BSD-style license that can be
      3  * found in the LICENSE file.
      4  */
      5 
      6 #include <stdio.h>
      7 #include <stdlib.h>
      8 #include <time.h>
      9 
     10 #include "dsp_test_util.h"
     11 #include "dsp_util.h"
     12 #include "eq2.h"
     13 #include "raw.h"
     14 
     15 #ifndef min
     16 #define min(a, b) ({ __typeof__(a) _a = (a);	\
     17 			__typeof__(b) _b = (b);	\
     18 			_a < _b ? _a : _b; })
     19 #endif
     20 
     21 static double tp_diff(struct timespec *tp2, struct timespec *tp1)
     22 {
     23 	return (tp2->tv_sec - tp1->tv_sec)
     24 		+ (tp2->tv_nsec - tp1->tv_nsec) * 1e-9;
     25 }
     26 
     27 /* Processes a buffer of data chunk by chunk using eq2 */
     28 static void process(struct eq2 *eq2, float *data0, float *data1, int count)
     29 {
     30 	int start;
     31 	for (start = 0; start < count; start += 2048)
     32 		eq2_process(eq2, data0 + start, data1 + start,
     33 			    min(2048, count - start));
     34 }
     35 
     36 /* Runs the filters on an input file */
     37 static void test_file(const char *input_filename, const char *output_filename)
     38 {
     39 	size_t frames;
     40 	int i;
     41 	double NQ = 44100 / 2; /* nyquist frequency */
     42 	struct timespec tp1, tp2;
     43 	struct eq2 *eq2;
     44 
     45 	float *data = read_raw(input_filename, &frames);
     46 
     47 	/* Set some data to 0 to test for denormals. */
     48 	for (i = frames / 10; i < frames; i++)
     49 		data[i] = 0.0;
     50 
     51 	/* eq chain */
     52 	eq2 = eq2_new();
     53 	eq2_append_biquad(eq2, 0, BQ_PEAKING, 380/NQ, 3, -10);
     54 	eq2_append_biquad(eq2, 0, BQ_PEAKING, 720/NQ, 3, -12);
     55 	eq2_append_biquad(eq2, 0, BQ_PEAKING, 1705/NQ, 3, -8);
     56 	eq2_append_biquad(eq2, 0, BQ_HIGHPASS, 218/NQ, 0.7, -10.2);
     57 	eq2_append_biquad(eq2, 0, BQ_PEAKING, 580/NQ, 6, -8);
     58 	eq2_append_biquad(eq2, 0, BQ_HIGHSHELF, 8000/NQ, 3, 2);
     59 	eq2_append_biquad(eq2, 1, BQ_PEAKING, 450/NQ, 3, -12);
     60 	eq2_append_biquad(eq2, 1, BQ_PEAKING, 721/NQ, 3, -12);
     61 	eq2_append_biquad(eq2, 1, BQ_PEAKING, 1800/NQ, 8, -10.2);
     62 	eq2_append_biquad(eq2, 1, BQ_PEAKING, 580/NQ, 6, -8);
     63 	eq2_append_biquad(eq2, 1, BQ_HIGHPASS, 250/NQ, 0.6578, 0);
     64 	eq2_append_biquad(eq2, 1, BQ_HIGHSHELF, 8000/NQ, 0, 2);
     65 	clock_gettime(CLOCK_THREAD_CPUTIME_ID, &tp1);
     66 	process(eq2, data, data + frames, frames);
     67 	clock_gettime(CLOCK_THREAD_CPUTIME_ID, &tp2);
     68 	printf("processing takes %g seconds for %zu samples\n",
     69 	       tp_diff(&tp2, &tp1), frames * 2);
     70 	eq2_free(eq2);
     71 
     72 	write_raw(output_filename, data, frames);
     73 	free(data);
     74 }
     75 
     76 int main(int argc, char **argv)
     77 {
     78 	dsp_enable_flush_denormal_to_zero();
     79 	if (dsp_util_has_denormal())
     80 		printf("denormal still supported?\n");
     81 	else
     82 		printf("denormal disabled\n");
     83 	dsp_util_clear_fp_exceptions();
     84 
     85 	if (argc == 3)
     86 		test_file(argv[1], argv[2]);
     87 	else
     88 		printf("Usage: eq2_test [input.raw output.raw]\n");
     89 
     90 	dsp_util_print_fp_exceptions();
     91 	return 0;
     92 }
     93