#pragma once
#include "audio_math.h"
#include "synth.h"
#include <cmath>
#include <random>

namespace trnr {

//////////////
// SINE OSC //
//////////////

struct tx_sineosc {
	bool phase_reset;
	double samplerate;
	float phase_resolution;
	float phase;
	float history;
};

inline void tx_randomize_phase(float& phase)
{
	std::random_device random;
	std::mt19937 gen(random());
	std::uniform_real_distribution<> dis(0.0, 1.0);
	phase = dis(gen);
}

inline void tx_sineosc_init(tx_sineosc& s, double samplerate)
{
	s.phase_reset = false;
	s.samplerate = samplerate;
	s.phase_resolution = 16.f;
	s.phase = 0.f;
	s.history = 0.f;

	tx_randomize_phase(s.phase);
}

inline float tx_wrap(float& phase)
{
	while (phase < 0.) phase += 1.;
	while (phase >= 1.) phase -= 1.;
	return phase;
}

inline float tx_sineosc_process_sample(tx_sineosc& s, bool trigger, float frequency, float phase_modulation = 0.f)
{
	if (trigger) {
		if (s.phase_reset) s.phase = 0.f;
		else tx_randomize_phase(s.phase);
	}

	float lookup_phase = s.phase + phase_modulation;
	tx_wrap(lookup_phase);
	s.phase += frequency / s.samplerate;
	tx_wrap(s.phase);

	// redux
	s.phase = static_cast<int>(s.phase * s.phase_resolution) / s.phase_resolution;

	// x is scaled 0<=x<4096
	float output;
	float x = lookup_phase;
	const float a = -0.40319426317E-08;
	const float b = 0.21683205691E+03;
	const float c = 0.28463350538E-04;
	const float d = -0.30774648337E-02;
	float y;

	bool negate = false;
	if (x > 2048) {
		negate = true;
		x -= 2048;
	}
	if (x > 1024) x = 2048 - x;
	y = (a + x) / (b + c * x * x) + d * x;
	if (negate) output = (-y);
	else output = y;

	// filter
	output = 0.5 * (output + s.history);
	s.history = output;

	return output;
}

//////////////
// ENVELOPE //
//////////////

enum tx_env_state {
	idle = 0,
	attack1,
	attack2,
	hold,
	decay1,
	decay2,
	sustain,
	release1,
	release2
};

struct tx_envelope {
	tx_env_state state = idle;
	float attack1_rate = 0;
	float attack1_level = 0;
	float attack2_rate = 0;
	float hold_rate = 0;
	float decay1_rate = 0;
	float decay1_level = 0;
	float decay2_rate = 0;
	float sustain_level = 0;
	float release1_rate = 0;
	float release1_level = 0;
	float release2_rate = 0;
	bool skip_sustain = false;

	double samplerate = 44100.;
	size_t phase = 0;
	float level = 0.f;
	float start_level = 0.f;
	float h1 = 0.f;
	float h2 = 0.f;
	float h3 = 0.f;
	bool retrigger;
};

inline void tx_envelope_init(tx_envelope& e, double samplerate, bool retrigger = false)
{
	e.samplerate = samplerate;
	e.retrigger = retrigger;
}

inline size_t tx_mtos(float ms, double samplerate) { return static_cast<size_t>(ms * samplerate / 1000.f); }

inline float tx_lerp(float x1, float y1, float x2, float y2, float x)
{
	return y1 + (((x - x1) * (y2 - y1)) / (x2 - x1));
}

inline float tx_envelope_process_sample(tx_envelope& e, bool gate, bool trigger, float _attack_mod = 0,
										float _decay_mod = 0)
{
	size_t attack_mid_x1 = tx_mtos(e.attack1_rate + (float)_attack_mod, e.samplerate);
	size_t attack_mid_x2 = tx_mtos(e.attack2_rate + (float)_attack_mod, e.samplerate);
	size_t hold_samp = tx_mtos(e.hold_rate, e.samplerate);
	size_t decay_mid_x1 = tx_mtos(e.decay1_rate + (float)_decay_mod, e.samplerate);
	size_t decay_mid_x2 = tx_mtos(e.decay2_rate + (float)_decay_mod, e.samplerate);
	size_t release_mid_x1 = tx_mtos(e.release1_rate + (float)_decay_mod, e.samplerate);
	size_t release_mid_x2 = tx_mtos(e.release2_rate + (float)_decay_mod, e.samplerate);

	// if note on is triggered, transition to attack phase
	if (trigger) {
		if (e.retrigger) e.start_level = 0.f;
		else e.start_level = e.level;
		e.phase = 0;
		e.state = attack1;
	}
	// attack 1st half
	if (e.state == attack1) {
		// while in attack phase
		if (e.phase < attack_mid_x1) {
			e.level = tx_lerp(0, e.start_level, (float)attack_mid_x1, e.attack1_level, (float)e.phase);
			e.phase += 1;
		}
		// reset phase if parameter was changed
		if (e.phase > attack_mid_x1) { e.phase = attack_mid_x1; }
		// if attack phase is done, transition to decay phase
		if (e.phase == attack_mid_x1) {
			e.state = attack2;
			e.phase = 0;
		}
	}
	// attack 2nd half
	if (e.state == attack2) {
		// while in attack phase
		if (e.phase < attack_mid_x2) {
			e.level = tx_lerp(0, e.attack1_level, (float)attack_mid_x2, 1, (float)e.phase);
			e.phase += 1;
		}
		// reset phase if parameter was changed
		if (e.phase > attack_mid_x2) { e.phase = attack_mid_x2; }
		// if attack phase is done, transition to decay phase
		if (e.phase == attack_mid_x2) {
			e.state = hold;
			e.phase = 0;
		}
	}
	// hold
	if (e.state == hold) {
		if (e.phase < hold_samp) {
			e.level = 1.0;
			e.phase += 1;
		}
		if (e.phase > hold_samp) { e.phase = hold_samp; }
		if (e.phase == hold_samp) {
			e.state = decay1;
			e.phase = 0;
		}
	}
	// decay 1st half
	if (e.state == decay1) {
		// while in decay phase
		if (e.phase < decay_mid_x1) {
			e.level = tx_lerp(0, 1, (float)decay_mid_x1, e.decay1_level, (float)e.phase);
			e.phase += 1;
		}
		// reset phase if parameter was changed
		if (e.phase > decay_mid_x1) { e.phase = decay_mid_x1; }
		// if decay phase is done, transition to sustain phase
		if (e.phase == decay_mid_x1) {
			e.state = decay2;
			e.phase = 0;
		}
	}
	// decay 2nd half
	if (e.state == decay2) {
		// while in decay phase
		if (e.phase < decay_mid_x2) {
			e.level = tx_lerp(0, e.decay1_level, (float)decay_mid_x2, e.sustain_level, (float)e.phase);
			e.phase += 1;
		}
		// reset phase if parameter was changed
		if (e.phase > decay_mid_x2) { e.phase = decay_mid_x2; }
		// if decay phase is done, transition to sustain phase
		if (e.phase == decay_mid_x2) {
			e.state = sustain;
			e.phase = 0;
			e.level = e.sustain_level;
		}
	}
	// while sustain phase: if note off is triggered, transition to release phase
	if (e.state == sustain && (!gate || e.skip_sustain)) {
		e.state = release1;
		e.level = e.sustain_level;
	}
	// release 1st half
	if (e.state == release1) {
		// while in release phase
		if (e.phase < release_mid_x1) {
			e.level = tx_lerp(0, e.sustain_level, (float)release_mid_x1, e.release1_level, (float)e.phase);
			e.phase += 1;
		}
		// reset phase if parameter was changed
		if (e.phase > release_mid_x1) { e.phase = release_mid_x1; }
		// transition to 2nd release half
		if (e.phase == release_mid_x1) {
			e.phase = 0;
			e.state = release2;
		}
	}
	// release 2nd half
	if (e.state == release2) {
		// while in release phase
		if (e.phase < release_mid_x2) {
			e.level = tx_lerp(0, e.release1_level, (float)release_mid_x2, 0, (float)e.phase);
			e.phase += 1;
		}
		// reset phase if parameter was changed
		if (e.phase > release_mid_x2) { e.phase = release_mid_x2; }
		// reset
		if (e.phase == release_mid_x2) {
			e.phase = 0;
			e.state = idle;
			e.level = 0;
		}
	}

	// smooth output
	e.h3 = e.h2;
	e.h2 = e.h1;
	e.h1 = e.level;

	return (e.h1 + e.h2 + e.h3) / 3.f;
}

//////////////
// OPERATOR //
//////////////

struct tx_operator {
	tx_envelope envelope;
	tx_sineosc oscillator;
	float ratio = 1.f;
	float amplitude = 1.f;
};

inline void tx_operator_init(tx_operator& op, double samplerate)
{
	tx_envelope_init(op.envelope, samplerate);
	tx_sineosc_init(op.oscillator, samplerate);
}

inline float tx_operator_process_sample(tx_operator& op, bool gate, bool trigger, float frequency, float velocity,
										float pm = 0.f)
{
	float env = tx_envelope_process_sample(op.envelope, gate, trigger);

	// drifts and sounds better!
	if (op.envelope.state != idle) {
		float osc = tx_sineosc_process_sample(op.oscillator, trigger, frequency, pm);
		return osc * env * velocity;
	} else {
		return 0.f;
	}
}

////////////
// VOICE  //
////////////

constexpr float MOD_INDEX_COEFF = 4.f;

struct tx_voice {
	bool gate = false;
	bool trigger = false;
	int midi_note = 0;
	float velocity = 1.f;
	float additional_pitch_mod = 0.f; // modulates pitch in frequency

	int algorithm = 0;
	float pitch_env_amt = 0.f;
	float feedback_amt = 0.f;
	float bit_resolution = 12.f;
	tx_sineosc feedback_osc;
	tx_envelope pitch_env;
	tx_operator op1;
	tx_operator op2;
	tx_operator op3;

	float pitch_mod = 0.f;
};

inline float calc_algo1(tx_voice& v, const float frequency)
{
	float fb_freq = frequency * v.op3.ratio;
	float fb_mod_index = (v.feedback_amt * MOD_INDEX_COEFF);
	float fb_signal = tx_sineosc_process_sample(v.feedback_osc, v.trigger, fb_freq) * fb_mod_index;

	float op3_Freq = frequency * v.op3.ratio;
	float op3_mod_index = (v.op3.amplitude * MOD_INDEX_COEFF);
	float op3_signal =
		tx_operator_process_sample(v.op3, v.gate, v.trigger, op3_Freq, v.velocity, fb_signal) * op3_mod_index;

	float op2_freq = frequency * v.op2.ratio;
	float op2_mod_index = (v.op2.amplitude * MOD_INDEX_COEFF);
	float op2_signal =
		tx_operator_process_sample(v.op2, v.gate, v.trigger, op2_freq, v.velocity, op3_signal) * op2_mod_index;

	float op1_freq = frequency * v.op1.ratio;
	return tx_operator_process_sample(v.op1, v.gate, v.trigger, op1_freq, v.velocity, op2_signal) * v.op1.amplitude;
}

inline float calc_algo2(tx_voice& v, const float frequency)
{
	float fb_freq = frequency * v.op3.ratio;
	float fb_mod_index = (v.feedback_amt * MOD_INDEX_COEFF);
	float fb_signal = tx_sineosc_process_sample(v.feedback_osc, v.trigger, fb_freq) * fb_mod_index;

	float op3_freq = frequency * v.op3.ratio;
	float op3_signal =
		tx_operator_process_sample(v.op3, v.gate, v.trigger, op3_freq, v.velocity, fb_signal) * v.op3.amplitude;

	float op2_freq = frequency * v.op2.ratio;
	float op2_mod_index = (v.op2.amplitude * MOD_INDEX_COEFF);
	float op2_signal = tx_operator_process_sample(v.op2, v.gate, v.trigger, op2_freq, v.velocity) * op2_mod_index;

	float op1_freq = frequency * v.op1.ratio;
	float op1_signal =
		tx_operator_process_sample(v.op1, v.gate, v.trigger, op1_freq, v.velocity, op2_signal) * v.op1.amplitude;

	return op1_signal + op3_signal;
}

inline float calc_algo3(tx_voice& v, const float frequency)
{
	float fb_freq = frequency * v.op3.ratio;
	float fb_mod_index = (v.feedback_amt * MOD_INDEX_COEFF);
	float fb_signal = tx_sineosc_process_sample(v.feedback_osc, v.trigger, fb_freq) * fb_mod_index;

	float op3_freq = frequency * v.op3.ratio;
	float op3_signal =
		tx_operator_process_sample(v.op3, v.gate, v.trigger, op3_freq, v.velocity, fb_signal) * v.op3.amplitude;

	float op2_freq = frequency * v.op2.ratio;
	float op2_signal = tx_operator_process_sample(v.op2, v.gate, v.trigger, op2_freq, v.velocity) * v.op2.amplitude;

	float op1_freq = frequency * v.op1.ratio;
	float op1_signal = tx_operator_process_sample(v.op1, v.gate, v.trigger, op1_freq, v.velocity) * v.op1.amplitude;

	return op1_signal + op2_signal + op3_signal;
}

inline float calc_algo4(tx_voice& v, const float frequency)
{
	float fb_freq = frequency * v.op3.ratio;
	float fb_mod_index = (v.feedback_amt * MOD_INDEX_COEFF);
	float fb_signal = tx_sineosc_process_sample(v.feedback_osc, v.trigger, fb_freq) * fb_mod_index;

	float op3_freq = frequency * v.op3.ratio;
	float op3_mod_index = (v.op3.amplitude * MOD_INDEX_COEFF);
	float op3_signal =
		tx_operator_process_sample(v.op3, v.gate, v.trigger, op3_freq, v.velocity, fb_signal) * op3_mod_index;

	float op2_freq = frequency * v.op2.ratio;
	float op2_mod_index = (v.op2.amplitude * MOD_INDEX_COEFF);
	float op2_signal = tx_operator_process_sample(v.op2, v.gate, v.trigger, op2_freq, v.velocity) * op2_mod_index;

	float op1_freq = frequency * v.op1.ratio;
	return tx_operator_process_sample(v.op1, v.gate, v.trigger, op1_freq, v.velocity, op2_signal + op3_signal) *
		   v.op1.amplitude;
}

template <>
inline void voice_process_block<tx_voice, float>(tx_voice& v, float** frames, size_t num_frames, midi_event* events,
												 size_t num_events, const vector<audio_buffer<float>>& mods)
{
	float frequency = midi_to_frequency(v.midi_note + v.pitch_mod + v.additional_pitch_mod);

	for (int s = 0; s < num_frames; s++) {

		float pitch_env_signal = tx_envelope_process_sample(v.pitch_env, v.gate, v.trigger) * v.pitch_env_amt;
		float pitched_freq = frequency + pitch_env_signal;

		float output = 0.f;

		// mix operator signals according to selected algorithm
		switch (v.algorithm) {
		case 0:
			output = calc_algo1(v, pitched_freq);
			break;
		case 1:
			output = calc_algo2(v, pitched_freq);
			break;
		case 2:
			output = calc_algo3(v, pitched_freq);
			break;
		case 3:
			output = calc_algo4(v, pitched_freq);
			break;
		default:
			output = calc_algo1(v, pitched_freq);
			break;
		}

		// reset trigger
		v.trigger = false;

		float res = powf(2, v.bit_resolution);
		output = roundf(output * res) / res;

		frames[0][s] += output / 3.;
		frames[1][s] = frames[0][s];
	}
}

} // namespace trnr