#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;
}