From 24d9fc797297c902033d27343efa29268decf317 Mon Sep 17 00:00:00 2001
From: Lucas Schumacher <technolucas@protonmail.com>
Date: Sun, 12 May 2024 15:17:59 -0400
Subject: [PATCH] Fix audio_fft outputing phase instead of magnitude

---
 src/app/audio_fft.rs | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/src/app/audio_fft.rs b/src/app/audio_fft.rs
index 7df784f..26df136 100644
--- a/src/app/audio_fft.rs
+++ b/src/app/audio_fft.rs
@@ -54,7 +54,12 @@ impl AudioFFT {
                         .send(("FFT Output", PlotData::Bode32(fft_out.clone())))
                         .unwrap();
                     fft_in.clear();
-                    let output: Vec<u8> = fft_out.iter().map(|c| (c.arg() * 255.0) as u8).collect();
+                    let output: Vec<u8> = fft_out
+                        .iter()
+                        .map(|c| {
+                            (((c.re * c.re) + (c.im * c.im)).sqrt() / size as f32 * 255.0) as u8
+                        })
+                        .collect();
                     assert_eq!(output_len, output.len());
                     plot_tx
                         .send(("FFT Processed Output", PlotData::U8(output.clone())))