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 "eq.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 /* Generates impulse response */ 28 static void test_ir() 29 { 30 int N = 32768; 31 float *data; 32 struct eq *eq; 33 double NQ = 44100 / 2; /* nyquist frequency */ 34 struct timespec tp1, tp2; 35 int i; 36 FILE *ir; 37 38 data = calloc(1, sizeof(float) * N); 39 data[0] = 1; 40 41 eq = eq_new(); 42 eq_append_biquad(eq, BQ_PEAKING, 380/NQ, 3, -10); 43 eq_append_biquad(eq, BQ_PEAKING, 720/NQ, 3, -12); 44 eq_append_biquad(eq, BQ_PEAKING, 1705/NQ, 3, -8); 45 eq_append_biquad(eq, BQ_HIGHPASS, 218/NQ, 0.7, -10.2); 46 eq_append_biquad(eq, BQ_PEAKING, 580/NQ, 6, -8); 47 eq_append_biquad(eq, BQ_HIGHSHELF, 8000/NQ, 3, 2); 48 49 clock_gettime(CLOCK_THREAD_CPUTIME_ID, &tp1); 50 eq_process(eq, data, N); 51 clock_gettime(CLOCK_THREAD_CPUTIME_ID, &tp2); 52 printf("processing takes %g seconds\n", tp_diff(&tp2, &tp1)); 53 eq_free(eq); 54 55 ir = fopen("ir.dat", "w"); 56 for (i = 0; i < N; i++) 57 fprintf(ir, "%g\n", data[i]); 58 fclose(ir); 59 free(data); 60 } 61 62 /* Processes a buffer of data chunk by chunk using eq */ 63 static void process(struct eq *eq, float *data, int count) 64 { 65 int start; 66 for (start = 0; start < count; start += 2048) 67 eq_process(eq, data + start, min(2048, count - start)); 68 } 69 70 /* Runs the filters on an input file */ 71 static void test_file(const char *input_filename, const char *output_filename) 72 { 73 size_t frames; 74 int i; 75 double NQ = 44100 / 2; /* nyquist frequency */ 76 struct timespec tp1, tp2; 77 struct eq *eq; 78 79 float *data = read_raw(input_filename, &frames); 80 81 /* Set some data to 0 to test for denormals. */ 82 for (i = frames / 10; i < frames; i++) 83 data[i] = 0.0; 84 85 /* Left eq chain */ 86 eq = eq_new(); 87 eq_append_biquad(eq, BQ_PEAKING, 380/NQ, 3, -10); 88 eq_append_biquad(eq, BQ_PEAKING, 720/NQ, 3, -12); 89 eq_append_biquad(eq, BQ_PEAKING, 1705/NQ, 3, -8); 90 eq_append_biquad(eq, BQ_HIGHPASS, 218/NQ, 0.7, -10.2); 91 eq_append_biquad(eq, BQ_PEAKING, 580/NQ, 6, -8); 92 eq_append_biquad(eq, BQ_HIGHSHELF, 8000/NQ, 3, 2); 93 clock_gettime(CLOCK_THREAD_CPUTIME_ID, &tp1); 94 process(eq, data, frames); 95 clock_gettime(CLOCK_THREAD_CPUTIME_ID, &tp2); 96 printf("processing takes %g seconds for %zu samples\n", 97 tp_diff(&tp2, &tp1), frames); 98 eq_free(eq); 99 100 /* Right eq chain */ 101 eq = eq_new(); 102 eq_append_biquad(eq, BQ_PEAKING, 450/NQ, 3, -12); 103 eq_append_biquad(eq, BQ_PEAKING, 721/NQ, 3, -12); 104 eq_append_biquad(eq, BQ_PEAKING, 1800/NQ, 8, -10.2); 105 eq_append_biquad(eq, BQ_PEAKING, 580/NQ, 6, -8); 106 eq_append_biquad(eq, BQ_HIGHPASS, 250/NQ, 0.6578, 0); 107 eq_append_biquad(eq, BQ_HIGHSHELF, 8000/NQ, 0, 2); 108 clock_gettime(CLOCK_THREAD_CPUTIME_ID, &tp1); 109 process(eq, data + frames, frames); 110 clock_gettime(CLOCK_THREAD_CPUTIME_ID, &tp2); 111 printf("processing takes %g seconds for %zu samples\n", 112 tp_diff(&tp2, &tp1), frames); 113 eq_free(eq); 114 115 write_raw(output_filename, data, frames); 116 free(data); 117 } 118 119 int main(int argc, char **argv) 120 { 121 dsp_enable_flush_denormal_to_zero(); 122 if (dsp_util_has_denormal()) 123 printf("denormal still supported?\n"); 124 else 125 printf("denormal disabled\n"); 126 dsp_util_clear_fp_exceptions(); 127 128 if (argc == 1) 129 test_ir(); 130 else if (argc == 3) 131 test_file(argv[1], argv[2]); 132 else 133 printf("Usage: eq_test [input.raw output.raw]\n"); 134 135 dsp_util_print_fp_exceptions(); 136 return 0; 137 } 138
