// Copyright (c) 2021 Weird Constructor <weirdconstructor@gmail.com>
// This file is a part of HexoDSP. Released under GPL-3.0-or-later.
// See README.md and COPYING for details.

use crate::dsp::{
    DspNode, GraphAtomData, GraphFun, LedPhaseVals, NodeContext, NodeGlobalRef, NodeId, ProcBuf,
    SAtom,
};
use crate::nodes::{NodeAudioContext, NodeExecContext};
use synfx_dsp::PolyBlepOscillator;

#[macro_export]
macro_rules! fa_bosc_wtype {
    ($formatter: expr, $v: expr, $denorm_v: expr) => {{
        let s = match ($v.round() as usize) {
            0 => "Sin",
            1 => "Tri",
            2 => "Saw",
            3 => "Pulse",
            4 => "Pulse-DC",
            _ => "?",
        };
        write!($formatter, "{}", s)
    }};
}

/// A simple amplifier
#[derive(Debug, Clone)]
pub struct BOsc {
    osc: PolyBlepOscillator,
    israte: f32,
}

impl BOsc {
    pub fn new(nid: &NodeId, _node_global: &NodeGlobalRef) -> Self {
        let init_phase = nid.init_phase();

        Self { osc: PolyBlepOscillator::new(init_phase), israte: 1.0 / 44100.0 }
    }

    pub const freq: &'static str = "Base frequency of the oscillator.\n";
    pub const det: &'static str = "Detune the oscillator in semitones and cents. \
         the input of this value is rounded to semitones on coarse input. \
         Fine input lets you detune in cents (rounded). \
         A signal sent to this port is not rounded.\n\
         Note: The signal input allows detune +-10 octaves.";
    pub const pw: &'static str = "";
    pub const wtype: &'static str = "Waveform type. Available waveforms:\n\
          - **Sin**   - Sine Waveform\n\
          - **Tri**   - Triangle Waveform\n\
          - **Saw**   - Sawtooth Waveform\n\
          - **Pulse** - Pulse Waveform\n\
          - **Pulse-DC** - Pulse Waveform (DC corrected)";
    pub const sig: &'static str = "Oscillator output";
    pub const DESC: &'static str = r#"Basic Oscillator

A very basic band limited oscillator with a sine, triangle, pulse and sawtooth waveform.
"#;
    pub const HELP: &'static str = r#"Basic Waveform Oscillator

A very basic band limited oscillator with a sine, triangle, pulse and sawtooth
waveform.  The pulse width ~~pw~~ parameter only has an effect for the
**Pulse** waveform.

There are two pulse waveforms: **Pulse** and **Pulse-DC**. Depending on the pulse width
setting of the oscillator the output of the pulse might introduce DC (direct current) into
the signal. The **Pulse-DC** variant compensates that DC component by shifting the signal,
just like a high pass filter would do.
"#;

    pub fn graph_fun() -> Option<GraphFun> {
        let mut osc = Box::new(PolyBlepOscillator::new(0.0));
        let israte = 1.0 / 128.0;

        Some(Box::new(move |gd: &dyn GraphAtomData, init: bool, _x: f32, _xn: f32| -> f32 {
            let wtype = NodeId::BOsc(0).inp_param("wtype").unwrap().inp();
            let pw = NodeId::BOsc(0).inp_param("pw").unwrap().inp();
            // let det   = NodeId::BOsc(0).inp_param("det").unwrap().inp();

            let wtype = gd.get(wtype as u32).map(|a| a.i()).unwrap_or(0);
            let pw = gd.get_denorm(pw as u32);
            // let det   = gd.get_norm(det as u32);

            // the detune scaling with lerp is wrong...
            // let pow = lerp((det + 0.2) * (1.0 / 0.4), 0.25, 4.0);
            // let freq = (2.0_f32).powf(pow);
            let freq = 2.0;

            if init {
                osc.reset();
                if wtype == 1 {
                    // we need to initialize the leaky integrator
                    // in the triangle wave form, or it would look
                    // a bit weird.
                    for _ in 0..256 {
                        osc.next_tri(freq, israte);
                    }
                }
            }

            let s = match wtype {
                0 => (osc.next_sin(freq, israte) + 1.0) * 0.5,
                1 => (osc.next_tri(freq, israte) + 1.0) * 0.5,
                2 => (osc.next_saw(freq, israte) + 1.0) * 0.5,
                3 => (osc.next_pulse_no_dc(freq, israte, pw) + 1.0) * 0.5,
                _ => (osc.next_pulse(freq, israte, pw) + 1.0) * 0.5,
            };

            s * 0.9 + 0.05
        }))
    }
}

impl DspNode for BOsc {
    fn set_sample_rate(&mut self, srate: f32) {
        self.israte = 1.0 / srate;
    }

    fn reset(&mut self) {
        self.osc.reset();
    }

    #[inline]
    fn process(
        &mut self,
        ctx: &mut dyn NodeAudioContext,
        _ectx: &mut NodeExecContext,
        _nctx: &NodeContext,
        atoms: &[SAtom],
        inputs: &[ProcBuf],
        outputs: &mut [ProcBuf],
        ctx_vals: LedPhaseVals,
    ) {
        use crate::dsp::{at, denorm, denorm_offs, inp, out};

        let freq = inp::BOsc::freq(inputs);
        let det = inp::BOsc::det(inputs);
        let pw = inp::BOsc::pw(inputs);
        let out = out::BOsc::sig(outputs);

        let wtype = at::BOsc::wtype(atoms);

        let israte = self.israte;

        match wtype.i() {
            0 => {
                // sin
                for frame in 0..ctx.nframes() {
                    let freq = denorm_offs::BOsc::freq(freq, det.read(frame), frame);
                    out.write(frame, self.osc.next_sin(freq, israte));
                }
            }
            1 => {
                // tri
                for frame in 0..ctx.nframes() {
                    let freq = denorm_offs::BOsc::freq(freq, det.read(frame), frame);
                    out.write(frame, self.osc.next_tri(freq, israte));
                }
            }
            2 => {
                // saw
                for frame in 0..ctx.nframes() {
                    let freq = denorm_offs::BOsc::freq(freq, det.read(frame), frame);
                    out.write(frame, self.osc.next_saw(freq, israte));
                }
            }
            3 => {
                // pulse
                for frame in 0..ctx.nframes() {
                    let freq = denorm_offs::BOsc::freq(freq, det.read(frame), frame);
                    let pw = denorm::BOsc::pw(pw, frame);
                    out.write(frame, self.osc.next_pulse_no_dc(freq, israte, pw));
                }
            }
            _ => {
                // pulse
                for frame in 0..ctx.nframes() {
                    let freq = denorm_offs::BOsc::freq(freq, det.read(frame), frame);
                    let pw = denorm::BOsc::pw(pw, frame);
                    out.write(frame, self.osc.next_pulse(freq, israte, pw));
                }
            }
        }

        ctx_vals[0].set(out.read(ctx.nframes() - 1));
    }
}