#include "filters.h"

void init_rc(ResistorCapacitor *rc, float tau, float sample_rate) {
    rc->prev_sample = 0.0f;
    rc->alpha = exp(-1 / (tau * sample_rate));
}

float apply_pre_emphasis(ResistorCapacitor *rc, float sample) {
    float audio = sample-rc->alpha*rc->prev_sample;
    rc->prev_sample = audio;
    return audio*2;
}

void init_low_pass_filter(ResistorCapacitor *rc, float cutoff_frequency, float sample_rate) {
    init_rc(&rc, (-1 / (sample_rate * log(sample_rate/(sample_rate+(1/(M_2PI*cutoff_frequency)))))), sample_rate);
}

float apply_low_pass_filter(ResistorCapacitor *rc, float sample) {
    float output = rc->alpha*sample+(1-rc->alpha)*rc->prev_sample;
    rc->prev_sample = output;
    return output;
}

void init_delay_line(DelayLine *delay_line, int max_delay) {
    delay_line->buffer = (float *)calloc(max_delay, sizeof(float));
    delay_line->size = max_delay;
    delay_line->write_idx = 0;
    delay_line->read_idx = 0;
    delay_line->delay = 0;
}

void set_delay_line(DelayLine *delay_line, int new_delay) {
    if (new_delay >= delay_line->size) {
        new_delay = delay_line->size - 1;
    }
    if (new_delay < 0) {
        new_delay = 0;
    }
    
    delay_line->delay = new_delay;
    delay_line->read_idx = (delay_line->write_idx - new_delay + delay_line->size) % delay_line->size;
}

float delay_line(DelayLine *delay_line, float in) {
    float out;
    
    // Read the delayed sample
    out = delay_line->buffer[delay_line->read_idx];
    
    // Write the new sample
    delay_line->buffer[delay_line->write_idx] = in;
    
    // Update indices
    delay_line->write_idx = (delay_line->write_idx + 1) % delay_line->size;
    delay_line->read_idx = (delay_line->read_idx + 1) % delay_line->size;
    
    return out;
}

void exit_delay_line(DelayLine *delay_line) {
    free(delay_line->buffer);
}