juce_Chorus.h

/*
  ==============================================================================
 
   This file is part of the JUCE library.
   Copyright (c) 2022 - Raw Material Software Limited
 
   JUCE is an open source library subject to commercial or open-source
   licensing.
 
   By using JUCE, you agree to the terms of both the JUCE 7 End-User License
   Agreement and JUCE Privacy Policy.
 
   End User License Agreement: www.juce.com/juce-7-licence
   Privacy Policy: www.juce.com/juce-privacy-policy
 
   Or: You may also use this code under the terms of the GPL v3 (see
   www.gnu.org/licenses).
 
   JUCE IS PROVIDED "AS IS" WITHOUT ANY WARRANTY, AND ALL WARRANTIES, WHETHER
   EXPRESSED OR IMPLIED, INCLUDING MERCHANTABILITY AND FITNESS FOR PURPOSE, ARE
   DISCLAIMED.
 
  ==============================================================================
*/
 
namespace juce::dsp
{
 
template <typename SampleType>
class Chorus
{
public:
    //==============================================================================
    Chorus();
 
    //==============================================================================
    void setRate (SampleType newRateHz);
 
    void setDepth (SampleType newDepth);
 
    void setCentreDelay (SampleType newDelayMs);
 
    void setFeedback (SampleType newFeedback);
 
    void setMix (SampleType newMix);
 
    //==============================================================================
    void prepare (const ProcessSpec& spec);
 
    void reset();
 
    //==============================================================================
    template <typename ProcessContext>
    void process (const ProcessContext& context) noexcept
    {
        const auto& inputBlock = context.getInputBlock();
        auto& outputBlock      = context.getOutputBlock();
        const auto numChannels = outputBlock.getNumChannels();
        const auto numSamples  = outputBlock.getNumSamples();
 
        jassert (inputBlock.getNumChannels() == numChannels);
        jassert (inputBlock.getNumChannels() == lastOutput.size());
        jassert (inputBlock.getNumSamples()  == numSamples);
 
        if (context.isBypassed)
        {
            outputBlock.copyFrom (inputBlock);
            return;
        }
 
        auto delayValuesBlock = AudioBlock<SampleType> (bufferDelayTimes).getSubBlock (0, numSamples);
        auto contextDelay = ProcessContextReplacing<SampleType> (delayValuesBlock);
        delayValuesBlock.clear();
 
        osc.process (contextDelay);
        delayValuesBlock.multiplyBy (oscVolume);
 
        auto* delaySamples = bufferDelayTimes.getWritePointer (0);
 
        for (size_t i = 0; i < numSamples; ++i)
        {
            auto lfo = jmax (static_cast<SampleType> (1.0), maximumDelayModulation * delaySamples[i] + centreDelay);
            delaySamples[i] = static_cast<SampleType> (lfo * sampleRate / 1000.0);
        }
 
        dryWet.pushDrySamples (inputBlock);
 
        for (size_t channel = 0; channel < numChannels; ++channel)
        {
            auto* inputSamples  = inputBlock .getChannelPointer (channel);
            auto* outputSamples = outputBlock.getChannelPointer (channel);
 
            for (size_t i = 0; i < numSamples; ++i)
            {
                auto input = inputSamples[i];
                auto output = input - lastOutput[channel];
 
                delay.pushSample ((int) channel, output);
                delay.setDelay (delaySamples[i]);
                output = delay.popSample ((int) channel);
 
                outputSamples[i] = output;
                lastOutput[channel] = output * feedbackVolume[channel].getNextValue();
            }
        }
 
        dryWet.mixWetSamples (outputBlock);
    }
 
private:
    //==============================================================================
    void update();
 
    //==============================================================================
    Oscillator<SampleType> osc;
    DelayLine<SampleType, DelayLineInterpolationTypes::Linear> delay;
    SmoothedValue<SampleType, ValueSmoothingTypes::Linear> oscVolume;
    std::vector<SmoothedValue<SampleType, ValueSmoothingTypes::Linear>> feedbackVolume { 2 };
    DryWetMixer<SampleType> dryWet;
    std::vector<SampleType> lastOutput { 2 };
    AudioBuffer<SampleType> bufferDelayTimes;
 
    double sampleRate = 44100.0;
    SampleType rate = 1.0, depth = 0.25, feedback = 0.0, mix = 0.5,
               centreDelay = 7.0;
 
    static constexpr SampleType maxDepth               = 1.0,
                                maxCentreDelayMs       = 100.0,
                                oscVolumeMultiplier    = 0.5,
                                maximumDelayModulation = 20.0;
};
 
} // namespace juce::dsp