1 files changed, 13 insertions, 17 deletions
diff --git a/src/audio/fft.cc b/src/audio/fft.cc
index 7523b42..982f35f 100644
--- a/src/audio/fft.cc
+++ b/src/audio/fft.cc
@@ -7,6 +7,10 @@
 #include <cmath>
 #include <cstring>
 
+// Max supported FFT size for stack-allocated temp buffers.
+// All callers use N <= 512 (DCT_SIZE). imdct needs 2*N = 1024.
+static const size_t kMaxFFTSize = 1024;
+
 // 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) {
@@ -102,9 +106,9 @@ void fft_inverse(float* real, float* imag, size_t N) {
 void dct_fft(const float* input, float* output, size_t N) {
   const float PI = 3.14159265358979323846f;
 
-  // Allocate temporary arrays for N-point FFT
-  float* real = new float[N];
-  float* imag = new float[N];
+  // Stack-allocated temp arrays (N <= kMaxFFTSize)
+  float real[kMaxFFTSize];
+  float imag[kMaxFFTSize];
 
   // Reorder input: even indices first, then odd indices reversed
   // [x[0], x[2], x[4], ...] followed by [x[N-1], x[N-3], x[N-5], ...]
@@ -135,9 +139,6 @@ void dct_fft(const float* input, float* output, size_t N) {
       output[k] = dct_value * sqrtf(2.0f / N);
     }
   }
-
-  delete[] real;
-  delete[] imag;
 }
 
 // IMDCT via FFT
@@ -149,8 +150,9 @@ void imdct_fft(const float* input, float* output, size_t N) {
   const float PI = 3.14159265358979323846f;
   const size_t M = 2 * N; // output length
 
-  float* real = new float[M];
-  float* imag = new float[M];
+  // Stack-allocated temp arrays (M = 2*N <= kMaxFFTSize)
+  float real[kMaxFFTSize];
+  float imag[kMaxFFTSize];
 
   // Pre-multiply X[k] by exp(-j*pi*(2k+1)/(4N)), build 2N complex FFT input
   // via standard IMDCT-via-FFT algorithm (N-point complex FFT)
@@ -180,9 +182,6 @@ void imdct_fft(const float* input, float* output, size_t N) {
     const float angle = -PI * (2.0f * n + 1.0f) / (4.0f * N);
     output[n] = gain * (real[n] * cosf(angle) - imag[n] * sinf(angle));
   }
-
-  delete[] real;
-  delete[] imag;
 }
 
 // IDCT (DCT-III) via FFT - inverse of the DCT-II reordering method
@@ -190,9 +189,9 @@ void imdct_fft(const float* input, float* output, size_t N) {
 void idct_fft(const float* input, float* output, size_t N) {
   const float PI = 3.14159265358979323846f;
 
-  // Allocate temporary arrays for N-point FFT
-  float* real = new float[N];
-  float* imag = new float[N];
+  // Stack-allocated temp arrays (N <= kMaxFFTSize)
+  float real[kMaxFFTSize];
+  float imag[kMaxFFTSize];
 
   // Prepare FFT input with inverse phase correction
   // FFT[k] = DCT[k] * exp(+j*pi*k/(2*N)) / normalization
@@ -226,7 +225,4 @@ void idct_fft(const float* input, float* output, size_t N) {
     output[2 * i] = real[i];             // Even positions
     output[2 * i + 1] = real[N - 1 - i]; // Odd positions (reversed)
   }
-
-  delete[] real;
-  delete[] imag;
 }