portapack-mayhem/firmware/test/baseband/dsp_fft_test.cpp
Kyle Reed 31e8019642
POCSAG Processor Rewrite (#1437)
* 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.
2023-09-08 19:41:09 +02:00

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;
}