143 lines
4.9 KiB
C++
143 lines
4.9 KiB
C++
|
|
// g++ --std=c++11 test.cpp
|
|
|
|
#include "test.h"
|
|
|
|
#include "test.h"
|
|
|
|
#include "fl/fft.h"
|
|
#include "fl/fft_impl.h"
|
|
#include "fl/math.h"
|
|
|
|
// // Proof of concept FFTImpl using KISS FFTImpl. Right now this is fixed sized blocks
|
|
// of 512. But this is
|
|
// // intended to change with a C++ wrapper around ot/
|
|
// typedef int16_t fft_audio_buffer_t[512];
|
|
|
|
// void fft_init(); // Explicit initialization of FFTImpl, otherwise it will be
|
|
// initialized on first run. bool fft_is_initialized(); void fft_unit_test(const
|
|
// fft_audio_buffer_t &buffer);
|
|
|
|
using namespace fl;
|
|
|
|
|
|
TEST_CASE("fft tester 512") {
|
|
int16_t buffer[512] = {0};
|
|
const int n = 512;
|
|
// fill in with a sine wave
|
|
for (int i = 0; i < n; ++i) {
|
|
float rot = fl::map_range<float, float>(i, 0, n - 1, 0, 2 * PI * 10);
|
|
float sin_x = sin(rot);
|
|
buffer[i] = int16_t(32767 * sin_x);
|
|
}
|
|
FFTBins out(16);
|
|
// fft_unit_test(buffer, &out);
|
|
const int samples = n;
|
|
FFTImpl fft(samples);
|
|
fft.run(buffer, &out);
|
|
|
|
FASTLED_WARN("FFTImpl output raw bins: " << out.bins_raw);
|
|
FASTLED_WARN("FFTImpl output db bins: " << out.bins_db);
|
|
const float expected_output[16] = {
|
|
3, 2, 2, 6, 6.08, 15.03, 3078.22, 4346.29,
|
|
4033.16, 3109, 38.05, 4.47, 4, 2, 1.41, 1.41};
|
|
for (int i = 0; i < 16; ++i) {
|
|
// CHECK(out[i] == Approx(expected_output[i]).epsilon(0.1));
|
|
float a = out.bins_raw[i];
|
|
float b = expected_output[i];
|
|
bool almost_equal = ALMOST_EQUAL(a, b, 0.1);
|
|
if (!almost_equal) {
|
|
FASTLED_WARN("FFTImpl output mismatch at index " << i << ": " << a
|
|
<< " != " << b);
|
|
}
|
|
CHECK(almost_equal);
|
|
}
|
|
|
|
fl::string info = fft.info();
|
|
FASTLED_WARN("FFTImpl info: " << info);
|
|
FASTLED_WARN("Done");
|
|
}
|
|
|
|
TEST_CASE("fft tester 256") {
|
|
// fft_audio_buffer_t buffer = {0};
|
|
fl::vector<int16_t> buffer;
|
|
const int n = 256;
|
|
// fill in with a sine wave
|
|
for (int i = 0; i < n; ++i) {
|
|
float rot = fl::map_range<float, float>(i, 0, n - 1, 0, 2 * PI * 10);
|
|
float sin_x = sin(rot);
|
|
auto v = int16_t(32767 * sin_x);
|
|
buffer.push_back(v);
|
|
}
|
|
FFTBins out(16);
|
|
// fft_unit_test(buffer, &out);
|
|
const int samples = n;
|
|
FFTImpl fft(samples);
|
|
fft.run(buffer, &out);
|
|
|
|
FASTLED_WARN("FFTImpl output: " << out);
|
|
const float expected_output[16] = {
|
|
3, 2, 4, 5, 5.10, 9.06, 11.05, 27.66,
|
|
2779.93, 3811.66, 4176.58, 4185.02, 4174.50, 4017.63, 3638.46, 3327.60};
|
|
for (int i = 0; i < 16; ++i) {
|
|
// CHECK(out[i] == Approx(expected_output[i]).epsilon(0.1));
|
|
float a = out.bins_raw[i];
|
|
float b = expected_output[i];
|
|
bool almost_equal = ALMOST_EQUAL(a, b, 0.1);
|
|
if (!almost_equal) {
|
|
FASTLED_WARN("FFTImpl output mismatch at index " << i << ": " << a
|
|
<< " != " << b);
|
|
}
|
|
CHECK(almost_equal);
|
|
}
|
|
|
|
fl::string info = fft.info();
|
|
FASTLED_WARN("FFTImpl info: " << info);
|
|
FASTLED_WARN("Done");
|
|
}
|
|
|
|
TEST_CASE("fft tester 256 with 64 bands") {
|
|
// fft_audio_buffer_t buffer = {0};
|
|
fl::vector<int16_t> buffer;
|
|
const int n = 256;
|
|
// fill in with a sine wave
|
|
for (int i = 0; i < n; ++i) {
|
|
float rot = fl::map_range<float, float>(i, 0, n - 1, 0, 2 * PI * 10);
|
|
float sin_x = sin(rot);
|
|
auto v = int16_t(32767 * sin_x);
|
|
buffer.push_back(v);
|
|
}
|
|
FFTBins out(64);
|
|
// fft_unit_test(buffer, &out);
|
|
const int samples = n;
|
|
FFT_Args args(samples, 64);
|
|
FFTImpl fft(args);
|
|
fft.run(buffer, &out);
|
|
FASTLED_WARN("FFTImpl output: " << out);
|
|
const float expected_output[64] = {
|
|
3, 3, 1, 2, 2, 3, 3,
|
|
3, 3, 4, 3, 4, 4, 5,
|
|
5, 3.16, 4.12, 5.10, 5.10, 6.08, 7,
|
|
9.06, 9.06, 9.06, 10.20, 11.18, 15.13, 18.25,
|
|
20.22, 26.31, 30.59, 63.95, 71.85, 2601.78, 2896.46,
|
|
3281.87, 3473.71, 3678.96, 3876.88, 3960.81, 4023.50, 4203.33,
|
|
4176.58, 4286.66, 4199.48, 4273.51, 4274.82, 4155.52, 4170.46,
|
|
4121.19, 4131.86, 4044.44, 4072.49, 4120.38, 3966.60, 4017.84,
|
|
3815.20, 3815.66, 3964.51, 3628.27, 3599.13, 3863.29, 3823.06,
|
|
3327.600};
|
|
for (int i = 0; i < 64; ++i) {
|
|
// CHECK(out[i] == Approx(expected_output[i]).epsilon(0.1));
|
|
float a = out.bins_raw[i];
|
|
float b = expected_output[i];
|
|
bool almost_equal = ALMOST_EQUAL(a, b, 0.1);
|
|
if (!almost_equal) {
|
|
FASTLED_WARN("FFTImpl output mismatch at index " << i << ": " << a
|
|
<< " != " << b);
|
|
}
|
|
CHECK(almost_equal);
|
|
}
|
|
fl::string info = fft.info();
|
|
FASTLED_WARN("FFTImpl info: " << info);
|
|
FASTLED_WARN("Done");
|
|
}
|