Mercurial > hg > btrack
diff src/OnsetDetectionFunction.cpp @ 57:296af6af6c3d
Replaced switch statements in OnsetDetectionFunction with enums. Renamed lots of functions so that they have better names, in camel case. Added some unit tests for initialisation of BTrack.
author | Adam Stark <adamstark@users.noreply.github.com> |
---|---|
date | Thu, 23 Jan 2014 15:31:11 +0000 |
parents | 338f5eb29e41 |
children | a8e3e95d14e4 |
line wrap: on
line diff
--- a/src/OnsetDetectionFunction.cpp Thu Jan 23 12:17:06 2014 +0000 +++ b/src/OnsetDetectionFunction.cpp Thu Jan 23 15:31:11 2014 +0000 @@ -120,19 +120,19 @@ // set the window to the specified type switch (arg_win_type){ - case 0: + case RectangularWindow: set_win_rectangular(); // Rectangular window break; - case 1: + case HanningWindow: set_win_hanning(); // Hanning Window break; - case 2: + case HammingWindow: set_win_hamming(); // Hamming Window break; - case 3: + case BlackmanWindow: set_win_blackman(); // Blackman Window break; - case 4: + case TukeyWindow: set_win_tukey(); // Tukey Window break; default: @@ -187,38 +187,70 @@ } switch (df_type){ - case 0: - df_sample = energy_envelope(); // calculate energy envelope detection function sample - break; - case 1: - df_sample = energy_difference(); // calculate half-wave rectified energy difference detection function sample + case EnergyEnvelope: + { + // calculate energy envelope detection function sample + df_sample = energy_envelope(); break; - case 2: - df_sample = spectral_difference(); // calculate spectral difference detection function sample + } + case EnergyDifference: + { + // calculate half-wave rectified energy difference detection function sample + df_sample = energy_difference(); break; - case 3: - df_sample = spectral_difference_hwr(); // calculate spectral difference detection function sample (half wave rectified) + } + case SpectralDifference: + { + // calculate spectral difference detection function sample + df_sample = spectral_difference(); break; - case 4: - df_sample = phase_deviation(); // calculate phase deviation detection function sample (half wave rectified) + } + case SpectralDifferenceHWR: + { + // calculate spectral difference detection function sample (half wave rectified) + df_sample = spectral_difference_hwr(); break; - case 5: - df_sample = complex_spectral_difference(); // calcualte complex spectral difference detection function sample + } + case PhaseDeviation: + { + // calculate phase deviation detection function sample (half wave rectified) + df_sample = phase_deviation(); break; - case 6: - df_sample = complex_spectral_difference_hwr(); // calcualte complex spectral difference detection function sample (half-wave rectified) + } + case ComplexSpectralDifference: + { + // calcualte complex spectral difference detection function sample + df_sample = complex_spectral_difference(); break; - case 7: - df_sample = high_frequency_content(); // calculate high frequency content detection function sample + } + case ComplexSpectralDifferenceHWR: + { + // calcualte complex spectral difference detection function sample (half-wave rectified) + df_sample = complex_spectral_difference_hwr(); break; - case 8: - df_sample = high_frequency_spectral_difference(); // calculate high frequency spectral difference detection function sample + } + case HighFrequencyContent: + { + // calculate high frequency content detection function sample + df_sample = high_frequency_content(); break; - case 9: - df_sample = high_frequency_spectral_difference_hwr(); // calculate high frequency spectral difference detection function (half-wave rectified) + } + case HighFrequencySpectralDifference: + { + // calculate high frequency spectral difference detection function sample + df_sample = high_frequency_spectral_difference(); break; + } + case HighFrequencySpectralDifferenceHWR: + { + // calculate high frequency spectral difference detection function (half-wave rectified) + df_sample = high_frequency_spectral_difference_hwr(); + break; + } default: + { df_sample = 1.0; + } } return df_sample;