Update and FFT implementation

1 files changed, 33 insertions, 11 deletions

diff --git a/src/dft.h b/src/dft.h
index 982cc6f..30b911c 100644
--- a/src/dft.h
+++ b/src/dft.h
@@ -2,6 +2,7 @@
 
 #include "common.h"
 #include "smrt_arena.h"
+#include "vec2.h"
 #include "wav.h"
 
 typedef struct {
@@ -17,6 +18,14 @@ dft_data_t discrete_fourier_transform(smrt_arena_t *      arena ,
                                                u64 sample_count ,
                                                u64  sample_rate);
 
+/// Run FFT algorithm on samples
+dft_data_t fast_fourier_transform( smrt_arena_t *         arena ,
+                                            f64 *       samples ,
+                                              u64  sample_count ,
+                                              u64   sample_rate ,
+                                  smrt_arena_t **     conflicts ,
+                                              u64 num_conflicts);
+
 /// Convert DFT frequency data to unsigned 8-bit PCM WAV amplitude data
 wav_data_t dft_data_to_wav(smrt_arena_t *     arena ,
                              dft_data_t         dft ,
@@ -35,14 +44,27 @@ typedef struct {
                u64   sample_rate;
 } stft_data_t;
 
-/// Run STFT algorithm on samples.
-stft_data_t short_time_fourier_transform(smrt_arena_t *             arena ,
-                                                  u64 samples_per_segment ,
-                                                  f64 *           samples ,
-                                                  u64        sample_count ,
-                                                  u64         sample_rate);
-
-/// Convert STFT frequency data to unsigned 8-bit PCM WAV amplitude data
-wav_data_t stft_data_to_wav(smrt_arena_t *      arena ,
-                             stft_data_t         stft ,
-                                     u64 sample_count);
+/// Run STFT algorithm on samples, sliding a window_size window by hop_size each step
+stft_data_t short_time_fourier_transform(smrt_arena_t *        arena ,
+                                                    u64  window_size ,
+                                                    u64     hop_size ,
+                                                  f64 *      samples ,
+                                                    u64 sample_count ,
+                                                    u64  sample_rate);
+
+/// Run in-place inverse FFT on a complex spectrum
+void inverse_fast_fourier_transform(vec2d_soa_t vs);
+
+/// Reconstruct samples from STFT data via overlap-add inverse FFT synthesis
+f64 *inverse_short_time_fourier_transform( smrt_arena_t *         arena ,
+                                              stft_data_t          stft ,
+                                                      u64   window_size ,
+                                                      u64      hop_size ,
+                                          smrt_arena_t **     conflicts ,
+                                                      u64 num_conflicts);
+
+/// Rebuild a window's full mirrored complex spectrum from DFT amplitude/phase data
+void reconstruct_spectrum(dft_data_t *         data ,
+                                   u64 sample_count ,
+                           vec2d_soa_t         vs_o);
+