mirror of
https://github.com/portapack-mayhem/mayhem-firmware.git
synced 2024-12-15 04:28:10 +00:00
31e8019642
* WIP Refactoring * WordExtractor building * Fix buffer sizes and squelch execute * Move impls to cpp file * Baud indicator * WIP new bit extractor * New approach for bit extraction. * Code fit and finish * Fix case on button * Cleanup * Adjust rate miss threshold * Fix count bits error calculation.
126 lines
3.3 KiB
C++
126 lines
3.3 KiB
C++
/*
|
|
* Copyright (C) 2023 Bernd Herzog
|
|
*
|
|
* This file is part of PortaPack.
|
|
*
|
|
* This program is free software; you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation; either version 2, or (at your option)
|
|
* any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program; see the file COPYING. If not, write to
|
|
* the Free Software Foundation, Inc., 51 Franklin Street,
|
|
* Boston, MA 02110-1301, USA.
|
|
*/
|
|
|
|
#include "dsp_fft.hpp"
|
|
#include "doctest.h"
|
|
|
|
TEST_CASE("ifft successfully calculates dc on zero frequency") {
|
|
uint32_t fft_width = 8;
|
|
complex16_t* v = new complex16_t[fft_width];
|
|
complex16_t* tmp = new complex16_t[fft_width];
|
|
|
|
v[0] = {1024, 0}; // DC bin
|
|
v[1] = {0, 0};
|
|
v[2] = {0, 0};
|
|
v[3] = {0, 0};
|
|
v[4] = {0, 0};
|
|
v[5] = {0, 0};
|
|
v[6] = {0, 0};
|
|
v[7] = {0, 0};
|
|
|
|
ifft<complex16_t>(v, fft_width, tmp);
|
|
|
|
CHECK(v[0].real() == 1024);
|
|
CHECK(v[1].real() == 1024);
|
|
CHECK(v[2].real() == 1024);
|
|
CHECK(v[3].real() == 1024);
|
|
CHECK(v[4].real() == 1024);
|
|
CHECK(v[5].real() == 1024);
|
|
CHECK(v[6].real() == 1024);
|
|
CHECK(v[7].real() == 1024);
|
|
|
|
for (uint32_t i = 0; i < fft_width; i++)
|
|
CHECK(v[i].imag() == 0);
|
|
|
|
delete[] v;
|
|
delete[] tmp;
|
|
}
|
|
|
|
TEST_CASE("ifft successfully calculates sine of quarter the sample rate") {
|
|
uint32_t fft_width = 8;
|
|
complex16_t* v = new complex16_t[fft_width];
|
|
complex16_t* tmp = new complex16_t[fft_width];
|
|
|
|
v[0] = {0, 0};
|
|
v[1] = {0, 0};
|
|
v[2] = {1024, 0}; // sample rate /4 bin
|
|
v[3] = {0, 0};
|
|
v[4] = {0, 0};
|
|
v[5] = {0, 0};
|
|
v[6] = {0, 0};
|
|
v[7] = {0, 0};
|
|
|
|
ifft<complex16_t>(v, fft_width, tmp);
|
|
|
|
CHECK(v[0].real() == 1024);
|
|
CHECK(v[1].real() == 0);
|
|
CHECK(v[2].real() == -1024);
|
|
CHECK(v[3].real() == 0);
|
|
CHECK(v[4].real() == 1024);
|
|
CHECK(v[5].real() == 0);
|
|
CHECK(v[6].real() == -1024);
|
|
CHECK(v[7].real() == 0);
|
|
|
|
CHECK(v[0].imag() == 0);
|
|
CHECK(v[1].imag() == 1024);
|
|
CHECK(v[2].imag() == 0);
|
|
CHECK(v[3].imag() == -1024);
|
|
CHECK(v[4].imag() == 0);
|
|
CHECK(v[5].imag() == 1024);
|
|
CHECK(v[6].imag() == 0);
|
|
CHECK(v[7].imag() == -1024);
|
|
|
|
delete[] v;
|
|
delete[] tmp;
|
|
}
|
|
|
|
TEST_CASE("ifft successfully calculates pure sine of half the sample rate") {
|
|
uint32_t fft_width = 8;
|
|
complex16_t* v = new complex16_t[fft_width];
|
|
complex16_t* tmp = new complex16_t[fft_width];
|
|
|
|
v[0] = {0, 0};
|
|
v[1] = {0, 0};
|
|
v[2] = {0, 0};
|
|
v[3] = {0, 0};
|
|
v[4] = {1024, 0}; // sample rate /2 bin
|
|
v[5] = {0, 0};
|
|
v[6] = {0, 0};
|
|
v[7] = {0, 0};
|
|
|
|
ifft<complex16_t>(v, fft_width, tmp);
|
|
|
|
CHECK(v[0].real() == 1024);
|
|
CHECK(v[1].real() == -1024);
|
|
CHECK(v[2].real() == 1024);
|
|
CHECK(v[3].real() == -1024);
|
|
CHECK(v[4].real() == 1024);
|
|
CHECK(v[5].real() == -1024);
|
|
CHECK(v[6].real() == 1024);
|
|
CHECK(v[7].real() == -1024);
|
|
|
|
for (uint32_t i = 0; i < fft_width; i++)
|
|
CHECK(v[i].imag() == 0);
|
|
|
|
delete[] v;
|
|
delete[] tmp;
|
|
}
|