1 files changed, 53 insertions, 0 deletions
diff --git a/input/common.c b/input/common.c
new file mode 100644
index 0000000..29fdc89
--- /dev/null
+++ b/input/common.c
@@ -0,0 +1,53 @@
+#include "input/common.h"
+
+#include <string.h>
+
+void reset_output_buffers(struct audio_data *data) {
+    memset(data->in_bass_r, 0, sizeof(double) * 2 * (data->FFTbassbufferSize / 2 + 1));
+    memset(data->in_bass_l, 0, sizeof(double) * 2 * (data->FFTbassbufferSize / 2 + 1));
+    memset(data->in_mid_r, 0, sizeof(double) * 2 * (data->FFTmidbufferSize / 2 + 1));
+    memset(data->in_mid_l, 0, sizeof(double) * 2 * (data->FFTmidbufferSize / 2 + 1));
+    memset(data->in_treble_r, 0, sizeof(double) * 2 * (data->FFTtreblebufferSize / 2 + 1));
+    memset(data->in_treble_l, 0, sizeof(double) * 2 * (data->FFTtreblebufferSize / 2 + 1));
+}
+
+int write_to_fftw_input_buffers(int16_t buf[], int16_t frames, void *data) {
+    struct audio_data *audio = (struct audio_data *)data;
+
+    for (uint16_t i = 0; i < frames * 2; i += 2) {
+        if (audio->channels == 1) {
+            if (audio->average) {
+                audio->in_bass_l[audio->bass_index] = (buf[i] + buf[i + 1]) / 2;
+            }
+            if (audio->left) {
+                audio->in_bass_l[audio->bass_index] = buf[i];
+            }
+            if (audio->right) {
+                audio->in_bass_l[audio->bass_index] = buf[i + 1];
+            }
+        }
+        // stereo storing channels in buffer
+        if (audio->channels == 2) {
+            audio->in_bass_l[audio->bass_index] = buf[i];
+            audio->in_bass_r[audio->bass_index] = buf[i + 1];
+
+            audio->in_mid_r[audio->mid_index] = audio->in_bass_r[audio->bass_index];
+            audio->in_treble_r[audio->treble_index] = audio->in_bass_r[audio->bass_index];
+        }
+
+        audio->in_mid_l[audio->mid_index] = audio->in_bass_l[audio->bass_index];
+        audio->in_treble_l[audio->treble_index] = audio->in_bass_l[audio->bass_index];
+
+        audio->bass_index++;
+        audio->mid_index++;
+        audio->treble_index++;
+        if (audio->bass_index == audio->FFTbassbufferSize - 1)
+            audio->bass_index = 0;
+        if (audio->mid_index == audio->FFTmidbufferSize - 1)
+            audio->mid_index = 0;
+        if (audio->treble_index == audio->FFTtreblebufferSize - 1)
+            audio->treble_index = 0;
+    }
+
+    return 0;
+}