pointillism 0.4.3

//! TODO: missing docs

use crate::prelude::*;

/// Plays a signal back with a delay. Can have some optional feedback. For a dry/wet effect, you can
/// add this to the original signal in different proportions.
///
/// ## Instanciation
///
/// The [`Delay`] type is quite generic. It allows for a buffer that might or might not be owned,
/// and it lets the feedback function be essentially anything. However in most cases, you'll want an
/// owned buffer, and you'll want one of the following three feedback behaviors, for which we make
/// type aliases.
///
/// - No feedback: [`Pure`].
/// - Exponential feedback: [`Exp`].
/// - Exponential feedback + ping-pong: [`Flip`].
#[derive(Clone, Debug)]
pub struct Delay<
    S: Signal<Sample = B::Item>,
    B: buf::BufferMut,
    F: Map<Input = B::Item, Output = B::Item>,
> {
    /// The played signal.
    pub sgn: S,

    /// The backing [`gen::LoopBuf`]. This contains the buffer, and an index controlling the current
    /// position.
    ///
    /// The length of the buffer deterimes the length of the delay. Since buffers can only hold an
    /// integral amount of samples, a delay might not be exact to the frame. This probably isn't a
    /// problem unless you have a really long delay though.
    pub loop_gen: gen::LoopBuf<B>,

    /// A function that determines how a sample from the buffer is fed back into itself. This is
    /// most often used to lower its volume, and flip the channels for a ping-pong effect.
    ///
    /// The zero signal should be mapped to zero, or you'll get all sorts of unexpected behavior.
    pub feedback: F,
}

impl<S: Signal<Sample = B::Item>, B: buf::BufferMut, F: Map<Input = B::Item, Output = B::Item>>
    Delay<S, B, F>
{
    /// Initializes a new delay.
    ///    
    /// To use an empty, owned buffer, see [`Self::new_owned`].
    pub const fn new(sgn: S, buffer: B, feedback: F) -> Self {
        Self {
            sgn,
            loop_gen: gen::LoopBuf::new(buffer),
            feedback,
        }
    }

    /// Returns a reference to the inner buffer.
    pub const fn buffer(&self) -> &B {
        &self.loop_gen.buffer
    }

    /// Returns a mutable reference to the inner buffer.
    pub fn buffer_mut(&mut self) -> &mut B {
        &mut self.loop_gen.buffer
    }

    /// Clears the inner buffer.
    pub fn clear(&mut self) {
        self.buffer_mut().clear();
    }

    /// Reads the current sample from the signal and advances the buffer position. The signal is not
    /// advanced.
    ///
    /// You can use this if you're advancing the signal manually.
    pub fn read_sgn(&mut self) {
        // Computes the new signal.
        let sgn = self.sgn.get();
        let buf = self.loop_gen._get();
        let new_sgn = sgn + self.feedback.eval(buf);

        // Updates the buffer, advances its position.
        let idx = self.loop_gen.index();
        self.buffer_mut()[idx] = new_sgn;
        self.loop_gen.advance();
    }
}

impl<S: Signal, F: Map<Input = S::Sample, Output = S::Sample>> Delay<S, buf::Dyn<S::Sample>, F>
where
    S::Sample: Audio,
{
    /// Initializes a new delay that owns its buffer. The size of the buffer is determined by the
    /// delay time.
    pub fn new_owned(sgn: S, delay: unt::Time, feedback: F) -> Self {
        Self::new(sgn, buf::Dyn::new_time(delay), feedback)
    }
}

impl<S: Signal<Sample = B::Item>, B: buf::BufferMut, F: Map<Input = B::Item, Output = B::Item>>
    Signal for Delay<S, B, F>
{
    type Sample = S::Sample;

    fn get(&self) -> Self::Sample {
        self.loop_gen._get()
    }
}

impl<
        S: SignalMut<Sample = B::Item>,
        B: buf::BufferMut,
        F: Map<Input = B::Item, Output = B::Item>,
    > SignalMut for Delay<S, B, F>
{
    fn advance(&mut self) {
        self.read_sgn();
        self.sgn.advance();
    }

    fn retrigger(&mut self) {
        // Retriggers the signals.
        self.sgn.retrigger();
        self.loop_gen.retrigger();

        // Clears the buffer.
        self.clear();
    }
}

impl<
        S: Frequency<Sample = B::Item>,
        B: buf::BufferMut,
        F: Map<Input = B::Item, Output = B::Item>,
    > Frequency for Delay<S, B, F>
{
    fn freq(&self) -> unt::Freq {
        self.sgn.freq()
    }

    fn freq_mut(&mut self) -> &mut unt::Freq {
        self.sgn.freq_mut()
    }
}

impl<S: Base<Sample = B::Item>, B: buf::BufferMut, F: Map<Input = B::Item, Output = B::Item>> Base
    for Delay<S, B, F>
{
    type Base = S::Base;

    fn base(&self) -> &S::Base {
        self.sgn.base()
    }

    fn base_mut(&mut self) -> &mut S::Base {
        self.sgn.base_mut()
    }
}

impl<S: Stop<Sample = B::Item>, B: buf::BufferMut, F: Map<Input = B::Item, Output = B::Item>> Stop
    for Delay<S, B, F>
{
    fn stop(&mut self) {
        self.sgn.stop();
    }
}

impl<S: Panic<Sample = B::Item>, B: buf::BufferMut, F: Map<Input = B::Item, Output = B::Item>> Panic
    for Delay<S, B, F>
{
    fn panic(&mut self) {
        self.sgn.panic();
        self.clear();
    }
}

/// A delay that only plays once.
pub type Pure<S, B> = Delay<S, B, map::Zero<<S as Signal>::Sample, <S as Signal>::Sample>>;

impl<S: Signal<Sample = B::Item>, B: buf::BufferMut> Pure<S, B> {
    /// Initializes a delay that only plays once.
    ///
    /// To use an empty, owned buffer, see [`Self::new_pure_owned`].
    pub const fn new_pure(sgn: S, buffer: B) -> Self {
        Self::new(sgn, buffer, map::Zero::new())
    }
}

impl<S: Signal> Pure<S, buf::Dyn<S::Sample>>
where
    S::Sample: Audio,
{
    /// Initializes a delay that only plays once and owns its buffer. The size of the buffer is
    /// determined by the delay time.
    pub fn new_pure_owned(sgn: S, delay: unt::Time) -> Self {
        Self::new_owned(sgn, delay, map::Zero::new())
    }
}

/// A delay that feeds back into itself with some gain factor.
///
/// This causes the signal to decay exponentially. You can set the volume to `1.0` for an infinite
/// delay, but other than that, you'll probably want a value exclusively between `0.0` and `1.0`.
pub type Exp<S, B> = Delay<S, B, map::Pw<<S as Signal>::Sample, unt::Vol>>;

impl<S: Signal<Sample = B::Item>, B: buf::BufferMut> Exp<S, B> {
    /// Initializes a new [`Exp`].
    ///
    /// To use an empty, owned buffer, see [`Self::new_exp_owned`].
    pub const fn new_exp(sgn: S, buffer: B, vol: unt::Vol) -> Self {
        Self::new(sgn, buffer, map::Pw::new(vol))
    }

    /// Returns the feedback volume.
    pub const fn vol(&self) -> unt::Vol {
        self.feedback.func
    }

    /// Returns a mutable reference to the feedback volume.
    pub fn vol_mut(&mut self) -> &mut unt::Vol {
        &mut self.feedback.func
    }
}

impl<S: Signal> Exp<S, buf::Dyn<S::Sample>>
where
    S::Sample: Audio,
{
    /// Initializes a delay with exponential decay that owns its buffer. The size of the buffer is
    /// determined by the delay time.
    pub fn new_exp_owned(sgn: S, delay: unt::Time, vol: unt::Vol) -> Self {
        Self::new_owned(sgn, delay, map::Pw::new(vol))
    }
}

/// An exponential delay with a ping-pong effect.
pub type Flip<S, B> = Delay<S, B, map::Comp<map::Pw<smp::Stereo, unt::Vol>, map::Flip>>;

/// Simple auxiliary function.
const fn comp_flip(vol: unt::Vol) -> map::Comp<map::Pw<smp::Stereo, unt::Vol>, map::Flip> {
    map::Comp::new(map::Pw::new(vol), map::Flip)
}

impl<S: Signal<Sample = smp::Stereo>, B: buf::BufferMut<Item = smp::Stereo>> Flip<S, B> {
    /// Initializes a new [`Flip`].
    ///
    /// You can set the volume to `1.0` for an infinite delay, but other than that, you'll probably
    /// want a value exclusively between `0.0` and `1.0`.
    pub const fn new_flip(sgn: S, buffer: B, vol: unt::Vol) -> Self {
        Self::new(sgn, buffer, comp_flip(vol))
    }

    /// Returns the feedback volume.
    pub const fn vol(&self) -> unt::Vol {
        self.feedback.inner.func
    }

    /// Returns a mutable reference to the feedback volume.
    pub fn vol_mut(&mut self) -> &mut unt::Vol {
        &mut self.feedback.inner.func
    }
}

impl<S: Signal<Sample = smp::Stereo>> Flip<S, buf::Dyn<S::Sample>> {
    /// Initializes a ping-pong delay that owns its buffer. The size of the buffer is determined by
    /// the delay time.
    pub fn new_flip_owned(sgn: S, delay: unt::Time, vol: unt::Vol) -> Self {
        Self::new_owned(sgn, delay, comp_flip(vol))
    }
}