summaryrefslogtreecommitdiff
path: root/src/audio/fft.cc
diff options
context:
space:
mode:
authorskal <pascal.massimino@gmail.com>2026-03-24 07:52:25 +0100
committerskal <pascal.massimino@gmail.com>2026-03-24 07:52:25 +0100
commit994c8e29bac4a5969cffb9eb913d2e74692bc71c (patch)
tree025ae4e62346c0f6266cffac986dff96d747e8f3 /src/audio/fft.cc
parent8a3da8213cd8ef58b04a2147f51d849b5a22e795 (diff)
fix(fft): replace iterative twiddle with direct cosf/sinf, add tests A-E
fft_radix2 now computes wr=cosf(angle*k)/wi=sinf(angle*k) directly per k, eliminating float drift over long iteration runs. Iterative approach documented in comment for reference. Tests A-E added (bit-reverse, small-N DFT, twiddle drift, DCT small/large N). arrays_match tolerance reverted to 5e-3. TODO.md updated. handoff(Gemini): fft twiddle fix complete, 38/38 tests passing.
Diffstat (limited to 'src/audio/fft.cc')
-rw-r--r--src/audio/fft.cc25
1 files changed, 13 insertions, 12 deletions
diff --git a/src/audio/fft.cc b/src/audio/fft.cc
index 6b8ba8e..6512fdc 100644
--- a/src/audio/fft.cc
+++ b/src/audio/fft.cc
@@ -9,7 +9,7 @@
// Bit-reversal permutation (in-place)
// Reorders array elements by reversing their binary indices
-static void bit_reverse_permute(float* real, float* imag, size_t N) {
+void bit_reverse_permute(float* real, float* imag, size_t N) {
size_t temp_bits = N;
size_t num_bits = 0;
while (temp_bits > 1) {
@@ -48,13 +48,19 @@ static void fft_radix2(float* real, float* imag, size_t N, int direction) {
const size_t half_stage = stage_size / 2;
const float angle = direction * 2.0f * PI / stage_size;
- // Precompute twiddle factors for this stage
- float wr = 1.0f;
- float wi = 0.0f;
- const float wr_delta = cosf(angle);
- const float wi_delta = sinf(angle);
-
for (size_t k = 0; k < half_stage; k++) {
+ // Direct twiddle factor: numerically stable, no drift.
+ // Faster iterative alternative (~4 mul+add vs 2 transcendentals) but
+ // accumulates float drift over many iterations (e.g. 256 steps for
+ // N=512 final stage). Shown here for reference:
+ // float wr = 1.0f, wi = 0.0f;
+ // const float wr_delta = cosf(angle), wi_delta = sinf(angle);
+ // // per k: const float wr_old = wr;
+ // // wr = wr_old * wr_delta - wi * wi_delta;
+ // // wi = wr_old * wi_delta + wi * wr_delta;
+ const float wr = cosf(angle * (float)k);
+ const float wi = sinf(angle * (float)k);
+
// Apply butterfly to all groups at this stage
for (size_t group_start = k; group_start < N; group_start += stage_size) {
const size_t i = group_start;
@@ -70,11 +76,6 @@ static void fft_radix2(float* real, float* imag, size_t N, int direction) {
real[i] = real[i] + temp_real;
imag[i] = imag[i] + temp_imag;
}
-
- // Update twiddle factor for next k (rotation)
- const float wr_old = wr;
- wr = wr_old * wr_delta - wi * wi_delta;
- wi = wr_old * wi_delta + wi * wr_delta;
}
}
}