annotate src/libsndfile-1.0.27/tests/dft_cmp.c @ 84:08ae793730bd

Add null config files
author Chris Cannam
date Mon, 02 Mar 2020 14:03:47 +0000
parents 1df64224f5ac
children
rev   line source
Chris@40 1 /*
Chris@40 2 ** Copyright (C) 2002-2015 Erik de Castro Lopo <erikd@mega-nerd.com>
Chris@40 3 **
Chris@40 4 ** This program is free software; you can redistribute it and/or modify
Chris@40 5 ** it under the terms of the GNU General Public License as published by
Chris@40 6 ** the Free Software Foundation; either version 2 of the License, or
Chris@40 7 ** (at your option) any later version.
Chris@40 8 **
Chris@40 9 ** This program is distributed in the hope that it will be useful,
Chris@40 10 ** but WITHOUT ANY WARRANTY; without even the implied warranty of
Chris@40 11 ** MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
Chris@40 12 ** GNU General Public License for more details.
Chris@40 13 **
Chris@40 14 ** You should have received a copy of the GNU General Public License
Chris@40 15 ** along with this program; if not, write to the Free Software
Chris@40 16 ** Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
Chris@40 17 */
Chris@40 18
Chris@40 19 #include <stdio.h>
Chris@40 20 #include <stdlib.h>
Chris@40 21 #include <math.h>
Chris@40 22
Chris@40 23 #include "dft_cmp.h"
Chris@40 24 #include "utils.h"
Chris@40 25
Chris@40 26 #ifndef M_PI
Chris@40 27 #define M_PI 3.14159265358979323846264338
Chris@40 28 #endif
Chris@40 29
Chris@40 30 #define DFT_SPEC_LENGTH (DFT_DATA_LENGTH / 2)
Chris@40 31
Chris@40 32 static void dft_magnitude (const double *data, double *spectrum) ;
Chris@40 33 static double calc_max_spectral_difference (const double *spec1, const double *spec2) ;
Chris@40 34
Chris@40 35 /*--------------------------------------------------------------------------------
Chris@40 36 ** Public functions.
Chris@40 37 */
Chris@40 38
Chris@40 39 double
Chris@40 40 dft_cmp_float (int linenum, const float *in_data, const float *test_data, int len, double target_snr, int allow_exit)
Chris@40 41 { static double orig [DFT_DATA_LENGTH] ;
Chris@40 42 static double test [DFT_DATA_LENGTH] ;
Chris@40 43 unsigned k ;
Chris@40 44
Chris@40 45 if (len != DFT_DATA_LENGTH)
Chris@40 46 { printf ("Error (line %d) : dft_cmp_float : Bad input array length.\n", linenum) ;
Chris@40 47 return 1 ;
Chris@40 48 } ;
Chris@40 49
Chris@40 50 for (k = 0 ; k < ARRAY_LEN (orig) ; k++)
Chris@40 51 { test [k] = test_data [k] ;
Chris@40 52 orig [k] = in_data [k] ;
Chris@40 53 } ;
Chris@40 54
Chris@40 55 return dft_cmp_double (linenum, orig, test, len, target_snr, allow_exit) ;
Chris@40 56 } /* dft_cmp_float */
Chris@40 57
Chris@40 58 double
Chris@40 59 dft_cmp_double (int linenum, const double *orig, const double *test, int len, double target_snr, int allow_exit)
Chris@40 60 { static double orig_spec [DFT_SPEC_LENGTH] ;
Chris@40 61 static double test_spec [DFT_SPEC_LENGTH] ;
Chris@40 62 double snr ;
Chris@40 63
Chris@40 64 if (! orig || ! test)
Chris@40 65 { printf ("Error (line %d) : dft_cmp_double : Bad input arrays.\n", linenum) ;
Chris@40 66 return 1 ;
Chris@40 67 } ;
Chris@40 68
Chris@40 69 if (len != DFT_DATA_LENGTH)
Chris@40 70 { printf ("Error (line %d) : dft_cmp_double : Bad input array length.\n", linenum) ;
Chris@40 71 return 1 ;
Chris@40 72 } ;
Chris@40 73
Chris@40 74 dft_magnitude (orig, orig_spec) ;
Chris@40 75 dft_magnitude (test, test_spec) ;
Chris@40 76
Chris@40 77 snr = calc_max_spectral_difference (orig_spec, test_spec) ;
Chris@40 78
Chris@40 79 if (snr > target_snr)
Chris@40 80 { printf ("\n\nLine %d: Actual SNR (% 4.1f) > target SNR (% 4.1f).\n\n", linenum, snr, target_snr) ;
Chris@40 81 oct_save_double (orig, test, len) ;
Chris@40 82 if (allow_exit)
Chris@40 83 exit (1) ;
Chris@40 84 } ;
Chris@40 85
Chris@40 86 if (snr < -500.0)
Chris@40 87 snr = -500.0 ;
Chris@40 88
Chris@40 89 return snr ;
Chris@40 90 } /* dft_cmp_double */
Chris@40 91
Chris@40 92 /*--------------------------------------------------------------------------------
Chris@40 93 ** Quick dirty calculation of magnitude spectrum for real valued data using
Chris@40 94 ** Discrete Fourier Transform. Since the data is real, the DFT is only
Chris@40 95 ** calculated for positive frequencies.
Chris@40 96 */
Chris@40 97
Chris@40 98 static void
Chris@40 99 dft_magnitude (const double *data, double *spectrum)
Chris@40 100 { static double cos_angle [DFT_DATA_LENGTH] = { 0.0 } ;
Chris@40 101 static double sin_angle [DFT_DATA_LENGTH] ;
Chris@40 102
Chris@40 103 double real_part, imag_part ;
Chris@40 104 int k, n ;
Chris@40 105
Chris@40 106 /* If sine and cosine tables haven't been initialised, do so. */
Chris@40 107 if (cos_angle [0] == 0.0)
Chris@40 108 for (n = 0 ; n < DFT_DATA_LENGTH ; n++)
Chris@40 109 { cos_angle [n] = cos (2.0 * M_PI * n / DFT_DATA_LENGTH) ;
Chris@40 110 sin_angle [n] = -1.0 * sin (2.0 * M_PI * n / DFT_DATA_LENGTH) ;
Chris@40 111 } ;
Chris@40 112
Chris@40 113 /* DFT proper. Since the data is real, only generate a half spectrum. */
Chris@40 114 for (k = 1 ; k < DFT_SPEC_LENGTH ; k++)
Chris@40 115 { real_part = 0.0 ;
Chris@40 116 imag_part = 0.0 ;
Chris@40 117
Chris@40 118 for (n = 0 ; n < DFT_DATA_LENGTH ; n++)
Chris@40 119 { real_part += data [n] * cos_angle [(k * n) % DFT_DATA_LENGTH] ;
Chris@40 120 imag_part += data [n] * sin_angle [(k * n) % DFT_DATA_LENGTH] ;
Chris@40 121 } ;
Chris@40 122
Chris@40 123 spectrum [k] = sqrt (real_part * real_part + imag_part * imag_part) ;
Chris@40 124 } ;
Chris@40 125
Chris@40 126 spectrum [DFT_SPEC_LENGTH - 1] = 0.0 ;
Chris@40 127
Chris@40 128 spectrum [0] = spectrum [1] = spectrum [2] = 0.0 ;
Chris@40 129
Chris@40 130 return ;
Chris@40 131 } /* dft_magnitude */
Chris@40 132
Chris@40 133 static double
Chris@40 134 calc_max_spectral_difference (const double *orig, const double *test)
Chris@40 135 { double orig_max = 0.0, max_diff = 0.0 ;
Chris@40 136 int k ;
Chris@40 137
Chris@40 138 for (k = 0 ; k < DFT_SPEC_LENGTH ; k++)
Chris@40 139 { if (orig_max < orig [k])
Chris@40 140 orig_max = orig [k] ;
Chris@40 141 if (max_diff < fabs (orig [k] - test [k]))
Chris@40 142 max_diff = fabs (orig [k] - test [k]) ;
Chris@40 143 } ;
Chris@40 144
Chris@40 145 if (max_diff < 1e-25)
Chris@40 146 return -500.0 ;
Chris@40 147
Chris@40 148 return 20.0 * log10 (max_diff / orig_max) ;
Chris@40 149 } /* calc_max_spectral_difference */