HexoDSP/src/dsp/node_quant.rs

// 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::nodes::{NodeAudioContext, NodeExecContext};
use crate::dsp::{NodeId, SAtom, ProcBuf, DspNode, LedPhaseVals, NodeContext};
use crate::dsp::helpers::{Trigger};

#[macro_export]
macro_rules! fa_quant { ($formatter: expr, $v: expr, $denorm_v: expr) => { {
    write!($formatter, "?")
} } }

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

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

/// A 9 channel signal multiplexer
#[derive(Debug, Clone)]
pub struct Quant {
}

impl Quant {
    pub fn new(_nid: &NodeId) -> Self {
        Self {
        }
    }
    pub const inp : &'static str =
        "Quant inp\n\nRange: (0..1)";
    pub const oct : &'static str =
        "Quant oct\n\nRange: (-1..1)";
    pub const omin : &'static str =
        "Quant omin\n\nRange: (-1..1)";
    pub const omax : &'static str =
        "Quant omax\n\nRange: (-1..1)";
    pub const sig : &'static str =
        "Quant sig\n\nRange: (-1..1)";
    pub const keys : &'static str =
        "Quant keys\n";
    pub const DESC : &'static str =
r#"Pitch/Note Quantizer

"#;
    pub const HELP : &'static str =
r#"Quant - A pitch quantizer

"#;
}

impl DspNode for Quant {
    fn outputs() -> usize { 1 }

    fn set_sample_rate(&mut self, _srate: f32) { }
    fn reset(&mut self) { }

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

        let inp = inp::Quant::inp(inputs);
        let oct = inp::Quant::oct(inputs);
        let out = out::Quant::sig(outputs);
        let keys = at::Quant::keys(atoms);
        let omin = at::Quant::omin(atoms);
        let omax = at::Quant::omax(atoms);

        let mut key_count = 0;
        let mut used_keys = [0.0; 12];

        let mask = keys.i();
        let tune_to_a4 = (9.0 / 12.0) * 0.1;
        for i in 0..9 {
            if mask & (0x1 << i) > 0 {
                used_keys[key_count] = ((i as f32 / 12.0) * 0.1) - tune_to_a4;
                key_count += 1;
            }
        }

        for i in 9..12 {
            let key_pitch_idx = (i + 9 + 12) % 12;
            if mask & (0x1 << i) > 0 {
                used_keys[key_count] = (i as f32 / 12.0) * 0.1 - tune_to_a4;
                key_count += 1;
            }
        }

        let mut all_keys = [0.0; 12 * 10]; // -4 and +4 octaves + 1 center
        let mut max_all  = 0;

        let omin = omin.i() as usize;
        for o in 0..omin {
            let o = omin - o;

            for i in 0..key_count {
                all_keys[max_all] = used_keys[i] - (o as f32) * 0.1;
                max_all += 1;
            }
        }

        for i in 0..key_count {
            all_keys[max_all] = used_keys[i];
            max_all += 1;
        }

        let omax = omax.i() as usize;
        for o in 1..=omax {
            for i in 0..key_count {
                all_keys[max_all] = used_keys[i] + (o as f32) * 0.1;
                max_all += 1;
            }
        }

        if max_all == 0 {
            for frame in 0..ctx.nframes() {
                out.write(
                    frame,
                    denorm::Quant::inp(inp, frame)
                    + denorm::Quant::oct(oct, frame));
            }

            ctx_vals[1].set(100.0); // some unreachable value for Keys widget
            ctx_vals[0].set(out.read(ctx.nframes() - 1));

        } else {
            let mut last_key = 0;

            for frame in 0..ctx.nframes() {
                let key =
                    (denorm::Quant::inp(inp, frame) * (max_all as f32))
                    .floor();
                let key = key as usize % max_all;
                let pitch = all_keys[key];
                last_key = key;
                out.write(frame, pitch + denorm::Quant::oct(oct, frame));
            }

            let last_pitch = used_keys[last_key as usize % key_count];
            ctx_vals[1].set((last_pitch + tune_to_a4) * 10.0 + 0.0001);
            ctx_vals[0].set(((last_pitch + tune_to_a4) * 10.0 - 0.5) * 2.0);
        }
    }
}