AudioEffectsSuite.h

#pragma once
 
#ifndef NO_EFFECTS_SUITE
#include <cmath>
#include <cstdint>
#include <iostream>
#include "AudioEffects/AudioEffect.h"
 
namespace audio_tools {
 
typedef float effectsuite_t;
 
static effectsuite_t **interpolationTable = nullptr;
 
class EffectSuiteBase  : public AudioEffect {
  virtual effectsuite_t processDouble(effectsuite_t inputSample) = 0;
 
  virtual effect_t process(effect_t inputSample) override {
    return active_flag ? 32767.0f * processDouble(static_cast<effectsuite_t>(inputSample)/32767.0f) : inputSample;
  }
 
};
 
 
class ModulationBaseClass {
public:
  ModulationBaseClass() { srand(static_cast<unsigned>(time(0))); }
 
  ModulationBaseClass(ModulationBaseClass &copy) = default; 
 
  ModulationBaseClass(effectsuite_t extSampRate) {
    this->sampleRate = extSampRate;
    timeStep = 1. / extSampRate;
    allocateMemory();
    // setInterpTable();
    srand(static_cast<unsigned>(time(0)));
  }
  ~ModulationBaseClass() = default;
 
  void setupModulationBaseClass(effectsuite_t extSampRate) {
    if (waveTable != nullptr) {
      delete[] waveTable;
    }
    sampleRate = extSampRate;
    timeStep = 1. / extSampRate;
    allocateMemory();
  }
 
  void setTriangle() {
    std::fill(waveTable, waveTable + sampleRate, 0);
    const effectsuite_t radPerSec = 2 * 3.1415926536f * timeStep;
    for (int i = 0; i < sampleRate; i++) {
      for (int j = 0; j < 35; j += 1)
        waveTable[i] += pow(-1., j) *
                        (sin((2. * effectsuite_t(j) + 1) * i * radPerSec)) /
                        (2. * effectsuite_t(j) + 1);
    }
  }
  void setSquare() {
    std::fill(waveTable, waveTable + sampleRate, 0);
    const effectsuite_t radPerSec = 2 * 3.1415926536f * timeStep;
    for (int i = 0; i < sampleRate; i++) {
      for (int j = 0; j < 35; j += 1)
        waveTable[i] += (sin((2 * j + 1) * i * radPerSec)) / (2 * j + 1);
    }
  }
  void setSawtooth() {
    std::fill(waveTable, waveTable + sampleRate, 0);
    const effectsuite_t radPerSec = 2 * 3.1415926536f * timeStep;
    for (int i = 0; i < sampleRate; i++) {
      for (int j = 1; j < 11; j += 1)
        waveTable[i] += pow(-1, j) * sin(j * radPerSec * i) / effectsuite_t(j);
    }
  }
  void setSine() {
    const effectsuite_t radPerSec = 2 * 3.1415926536f * timeStep;
    for (int i = 0; i < sampleRate; i++)
      waveTable[i] = sin(i * radPerSec);
  }
  void setOffSine() {
    const effectsuite_t radPerSec = 2 * 3.1415926536f * timeStep;
    for (int i = 0; i < sampleRate; i++)
      waveTable[i] = (sin(i * radPerSec) + 1) * .5f;
  }
 
  void setNoise() {
    is_noise = true;
  }
 
  bool isNoise() {
    return is_noise;
  }
 
  void setDC() {
    for (int i = 0; i < sampleRate; i++)
      waveTable[i] = 1.0;
  }
  void setRamp() {
    for (int i = 0; i < sampleRate; i++)
      waveTable[i] = i / effectsuite_t(sampleRate);
  }
  effectsuite_t readNoise() {
    const effectsuite_t lo = -1.;
    const effectsuite_t hi = 1.;
    return lo + static_cast<effectsuite_t>(rand()) /
                    (static_cast<effectsuite_t>(RAND_MAX / (hi - lo)));
  }
  void clipWave(effectsuite_t amp) {
    if (amp < .01) {
      amp = .01;
    }
 
    for (int i = 0; i < sampleRate; i++)
      waveTable[i] = tanh(amp * waveTable[i]) / tanh(amp);
  }
 
  effectsuite_t readTable(effectsuite_t freq) {
    if (freq > 0) {
      //    const effectsuite_t out = getInterpOut(tableIndex);
      const effectsuite_t out = getSplineOut(tableIndex, int(freq));
      tableIndex += freq;
      if (tableIndex - sampleRate > 0)
        tableIndex -= sampleRate;
 
      return out;
    } else {
      return 0.;
    }
  }
  void printInterpTable() {
    for (int j = 0; j < res; j++) {
      for (int i = 0; i < order; i++) {
        std::cout << interpTable[i][j] << '\t';
      }
      std::cout << '\n';
    }
  }
  bool setInterpTable() {
    effectsuite_t *polynomial_normaliser = new effectsuite_t[order];
    if (!polynomial_normaliser) {
      return false;
    }
    std::fill(polynomial_normaliser, polynomial_normaliser + order, 1);
    effectsuite_t *alphas = new effectsuite_t[res];
    if (!alphas) {
      return false;
    }
 
    for (int i = 0; i < res; i++) {
      alphas[i] = (i / float(res)) - 0.5;
    }
 
    effectsuite_t *anchors = new effectsuite_t[order];
 
    if ((order % 2) == 0) {
      for (int i = 0; i < order; i++) {
        anchors[i] = -(effectsuite_t(order) - 1) * 0.5 + effectsuite_t(i);
        std::fill(interpTable[i], interpTable[i] + res, 1);
      }
    } else {
      for (int i = 0; i < order; i++) {
        anchors[i] = (-(effectsuite_t(order)) * 0.5) + effectsuite_t(i);
      }
    }
 
    // loop for every value of alpha
    for (int q = 0; q < res; q++) {
      // loop for sub polynomial
      for (int j = 0; j < order; j++) {
        // loop for each point in subpoly
        for (int m = 0; m < order; m++) {
          if (m != j) {
            if (q == 0) {
              polynomial_normaliser[j] =
                  polynomial_normaliser[j] * (anchors[j] - anchors[m]);
            }
            interpTable[j][q] *= (alphas[q] - anchors[m]);
          }
        }
        interpTable[j][q] /= polynomial_normaliser[j];
      }
    }
 
    delete[] polynomial_normaliser;
    delete[] alphas;
    delete[] anchors;
    return true;
  }
 
protected:
  bool allocateMemory() {
    waveTable = new effectsuite_t[sampleRate];
    assert(waveTable!=nullptr);
    if (!waveTable) {
      return false;
    }
    std::fill(waveTable, waveTable + sampleRate, 0);
    return true;
  }
  effectsuite_t getInterpOut(effectsuite_t bufferIndex) {
    const int order = 4;
    const int orderHalf = order * .5;
    const int res = 100;
    effectsuite_t interpOut = 0;
    int intBufferIndex = floor(bufferIndex);
    int alphaIndex = int(floor((bufferIndex - intBufferIndex) * res));
 
    for (int i = 0; i < order; i++) {
      int interpIndex =
          (((i + 1 - orderHalf) + intBufferIndex) + sampleRate) % sampleRate;
      interpOut += (interpTable[i][alphaIndex]) * (waveTable[interpIndex]);
    }
    return interpOut;
  }
 
  effectsuite_t getSplineOut(effectsuite_t bufferIndex, int freq) {
    if (freq < 1) {
      freq = 1;
    }
    const int n0 = floor(bufferIndex);
    const int n1 = (n0 + freq) % sampleRate;
    const int n2 = (n0 + (2 * freq)) % sampleRate;
    const effectsuite_t alpha = bufferIndex - n0;
    const effectsuite_t a = waveTable[n1];
    const effectsuite_t c = ((3 * (waveTable[n2] - waveTable[n1])) -
                      (3 * (waveTable[n1] - waveTable[n0]))) *
                     .25;
    const effectsuite_t b = (waveTable[n2] - waveTable[n1]) - ((2 * c)) / 3;
    const effectsuite_t d = (-c) / 3;
    return a + (b * alpha) + (c * alpha * alpha) + (d * alpha * alpha * alpha);
  }
 
public:
  effectsuite_t tableIndex = 0;
  int sampleRate;
  effectsuite_t timeStep;
  effectsuite_t *waveTable;
 
protected:
  static const int order = 4;
  static const int res = 100;
  effectsuite_t interpTable[order][res];// = {1};
  bool is_noise = false;
};
 
 template <class T>
 class SoundGeneratorModulation : public SoundGenerator<T>{
    public:
      SoundGeneratorModulation(ModulationBaseClass &mod, int freq){
        p_mod = &mod;
        this->freq = freq;
      }
      bool begin(AudioInfo info) override{
          max_value = pow(2, info.bits_per_sample)/2-1;
          return SoundGenerator<T>::begin(info);
      }
      virtual  T readSample() override {
        return p_mod->isNoise() ? max_value * p_mod->readNoise() : max_value * p_mod->readTable(freq);
      }
      
  protected:
    ModulationBaseClass *p_mod=nullptr;
    int freq;
    float max_value=32767;
};
 
 
class DelayEffectBase  {
public:
  DelayEffectBase() = default;
 
  DelayEffectBase(DelayEffectBase &copy) = default;
 
  DelayEffectBase(int bufferSizeSamples) {
    error = setDelayBuffer(bufferSizeSamples);
    delayTimeSamples = bufferSizeSamples;
    if (interpolationTable == nullptr) {
      interpolationTable = DelayEffectBase::setInterpolationTable();
    }
  }
 
  ~DelayEffectBase() {
    if (delayBuffer != nullptr)
      delete[] delayBuffer;
  }
 
  void setupDelayEffectBase(const int bufferSizeSamples) {
    error = setDelayBuffer(bufferSizeSamples);
    delayTimeSamples = bufferSizeSamples;
  }
 
protected:
  static effectsuite_t **setInterpolationTable() {
    const int order = interpOrder;
    const int res = interpResolution;
    effectsuite_t **interpTable = new effectsuite_t *[order];
    if (!interpTable) {
      return NULL;
    }
 
    for (int i = 0; i < order; i++) {
      interpTable[i] = new effectsuite_t[res + 1];
      if (!interpTable[i]) {
        return NULL;
      }
      std::fill(interpTable[i], interpTable[i] + res, 1);
    }
 
    effectsuite_t *polynomial_normaliser = new effectsuite_t[order];
    if (!polynomial_normaliser) {
      return NULL;
    }
    std::fill(polynomial_normaliser, polynomial_normaliser + order, 1);
    effectsuite_t *alphas = new effectsuite_t[res];
    if (!alphas) {
      return NULL;
    }
 
    for (int i = 0; i < res; i++) {
      alphas[i] = (i / float(res)) - 0.5;
    }
 
    effectsuite_t *anchors = new effectsuite_t[order];
 
    if ((order % 2) == 0) {
      for (int i = 0; i < order; i++) {
        anchors[i] = -(effectsuite_t(order) - 1) * 0.5 + effectsuite_t(i);
      }
    } else {
      for (int i = 0; i < order; i++) {
        anchors[i] = (-(effectsuite_t(order)) * 0.5) + effectsuite_t(i);
      }
    }
 
    // loop for every value of alpha
    for (int q = 0; q < res; q++) {
      // loop for sub polynomial
      for (int j = 0; j < order; j++) {
        // loop for each point in subpoly
        for (int m = 0; m < order; m++) {
          if (m != j) {
            if (q == 0) {
              polynomial_normaliser[j] =
                  polynomial_normaliser[j] * (anchors[j] - anchors[m]);
            }
            interpTable[j][q] *= (alphas[q] - anchors[m]);
          }
        }
        interpTable[j][q] /= polynomial_normaliser[j];
      }
    }
    delete[] polynomial_normaliser;
    delete[] alphas;
    delete[] anchors;
    return interpTable;
  }
 
  // void printInterpTable() {
  //   for (int j = 0; j < interpResolution; j++) {
  //     for (int i = 0; i < interpOrder; i++) {
  //       printf("index %d: %.2f \t", i, interpolationTable[i][j]);
  //     }
  //     printf("\n");
  //   }
  // }
 
protected:
  bool setDelayBuffer(int bufferSizeSamples) {
    maxDelayBufferSize = bufferSizeSamples;
    delayBuffer = new effectsuite_t[maxDelayBufferSize];
    if (!delayBuffer) {
      return false;
    }
    std::fill(delayBuffer, delayBuffer + maxDelayBufferSize, 0);
    return true;
  }
 
  void storeSample(effectsuite_t inputSample) {
    delayBuffer[currentDelayWriteIndex] = inputSample;
  }
 
  void incDelayBuffWriteIndex() {
    currentDelayWriteIndex++;
    currentDelayWriteIndex %= delayTimeSamples;
  }
 
  void incDelayBuffReadIndex(effectsuite_t indexInc) {
    currentDelayReadIndex += indexInc;
    if (currentDelayReadIndex >= effectsuite_t(delayTimeSamples)) {
      currentDelayReadIndex = 0;
    }
    if (currentDelayReadIndex < 0) {
      currentDelayReadIndex = 0;
    }
  }
 
  void setDelayBuffReadIndex(effectsuite_t index) {
    currentDelayReadIndex = index;
    if (currentDelayReadIndex >= effectsuite_t(delayTimeSamples)) {
      currentDelayReadIndex = 0;
    }
    if (currentDelayReadIndex < 0) {
      currentDelayReadIndex = 0;
    }
  }
 
  void delaySample(effectsuite_t inputSample) {
    storeSample(inputSample);
    incDelayBuffWriteIndex();
  }
  effectsuite_t getInterpolatedOut(effectsuite_t bufferIndex) {
    const int order = interpOrder;
    const int orderHalf = order * .5;
    const int res = interpResolution;
    effectsuite_t interpOut = 0;
    int intBufferIndex = floor(bufferIndex);
    int alphaIndex = int(floor((bufferIndex - intBufferIndex) * res));
 
    for (int i = 0; i < order; i++) {
      int interpIndex = (i + 1 - orderHalf) + intBufferIndex;
      if (interpIndex < 0 || (interpIndex >= maxDelayBufferSize)) {
        if (interpIndex < 0) {
          interpIndex = maxDelayBufferSize + interpIndex;
        } else {
          interpIndex = interpIndex - maxDelayBufferSize;
        }
      }
 
      interpOut +=
          (interpolationTable[i][alphaIndex]) * (delayBuffer[interpIndex]);
    }
    return interpOut;
  }
 
protected: // member variables
  effectsuite_t *delayBuffer = 0;
  int maxDelayBufferSize = 441000;
  int delayTimeSamples = 44100;
  int currentDelayWriteIndex = 0;
  effectsuite_t currentDelayReadIndex = 0;
  static const int interpOrder = 4;
  static const int interpResolution = 1000;
 
  bool error;
};
 
class FilterEffectBase : public EffectSuiteBase  {
public:
  FilterEffectBase() { std::fill(rmsBuffer, rmsBuffer + rmsWindowSize, 0); }
 
  FilterEffectBase(FilterEffectBase &copy) = default;
 
  ~FilterEffectBase() = default;
 
  virtual effectsuite_t applyFilter(effectsuite_t sampVal) {
    effectsuite_t outSample = 0;
    firBuffer[bufferIndex] = sampVal;
 
    for (int j = 0; j < filterOrder; j++) {
      int i = ((bufferIndex - j) + filterOrder) % filterOrder;
      outSample +=
          firCoefficients[j] * firBuffer[i] + iirCoefficients[j] * iirBuffer[i];
    }
 
    iirBuffer[bufferIndex] = outSample;
    incBufferIndex();
 
    return outSample;
  }
 
  virtual effectsuite_t processDouble(effectsuite_t inputSample) override {
    return applyFilter(inputSample);
  }
 
  virtual effect_t process(effect_t inputSample) override {
    return active_flag ? 32767.0 * processDouble(static_cast<effectsuite_t>(inputSample)/32767.0) : inputSample;
  }
 
  effectsuite_t envelope(effectsuite_t sample) { return applyFilter(rms(sample)); }
 
  // void printBuffers() {
  //   printf("FIRb\t\tIIRb\n");
  //   for (int i = 0; i < filterOrder; i++) {
  //     printf("%.4e\t%.4e\n", firBuffer[i], iirBuffer[i]);
  //   }
  //   printf("\n");
  // }
 
  // void printCoefs() {
  //   printf("FIR\t\tIIR\n");
  //   for (int i = 0; i < filterOrder; i++) {
  //     printf("%.4e\t%.4e\n", firCoefficients[i], iirCoefficients[i]);
  //   }
  //   printf("\n");
  // }
 
  bool setChebyICoefficients(effectsuite_t cutFreq, bool shelfType, effectsuite_t ripple) {
    // NOTE: coefficient buffers must be cleared as are additive in the
    // following        code
    std::fill(firCoefficients, firCoefficients + 22, 0);
    std::fill(iirCoefficients, iirCoefficients + 22, 0);
 
    effectsuite_t poles = (effectsuite_t)filterOrder - 1;
    int order = (int)poles;
 
    firCoefficients[2] = 1;
    iirCoefficients[2] = 1;
 
    effectsuite_t Es, Vx, Kx;
    if (ripple != 0) {
      Es = sqrt(pow(1 / (1 - ripple), 2) - 1);
      Vx = (1 / poles) * log(1 / Es + sqrt(1 / (pow(Es, 2)) + 1));
      Kx = (1 / poles) * log(1 / Es + sqrt(1 / (pow(Es, 2)) - 1));
      Kx = cosh(Kx);
    } else {
      Vx = 1;
      Kx = 1;
    }
 
    const effectsuite_t T = 2 * tan(.5);
    const effectsuite_t W = 2 * pi * cutFreq;
 
    effectsuite_t K;
 
    if (shelfType == 0) 
    {
      K = sin(.5 - W / 2) / sin(.5 + W / 2);
    } else 
    {
 
      K = -cos(.5 + W / 2) / cos(W / 2 - .5);
    }
 
    for (int i = 0; i < (order / 2); i++) {
      const effectsuite_t alpha = pi / (2 * poles) + (i - 1) * (pi / poles);
 
      effectsuite_t Rp, Ip;
      if (ripple != 0) {
        Rp = -cos(alpha) * sinh(Vx) / Kx;
        Ip = sin(alpha) * cosh(Vx) / Kx;
      } else {
        Rp = -cos(alpha);
        Ip = sin(alpha);
      }
 
      const effectsuite_t M = pow(Rp, 2) + pow(Ip, 2);
      const effectsuite_t D = 4 - 4 * Rp * T + M * T;
 
      const effectsuite_t X0 = (pow(T, 2)) / D;
      const effectsuite_t X1 = (2 * pow(T, 2)) / D;
      const effectsuite_t X2 = X0;
 
      const effectsuite_t Y1 = (8 - (2 * M * pow(T, 2))) / D;
      const effectsuite_t Y2 = (-4 - 4 * Rp * T - M * T) / D;
 
      // renamed and inverted from original algorithm
      const effectsuite_t _D1 = 1 / (1 + Y1 * K - Y2 * pow(K, 2));
 
      const effectsuite_t _A0 = (X0 - X1 * K + X2 * pow(K, 2)) * _D1;
      effectsuite_t _A1 = (-2 * X0 * K + X1 + X1 * pow(K, 2) - 2 * X2 * K) * _D1;
      const effectsuite_t _A2 = (X0 * pow(K, 2) - X1 * K + X2) * _D1;
 
      effectsuite_t _B1 = (2 * K + Y1 + Y1 * pow(K, 2) - 2 * Y2 * K) * _D1;
      const effectsuite_t B2 = (-(pow(K, 2)) - Y1 * K + Y2) * _D1;
 
      if (shelfType == 1) {
        _A1 = -_A1;
        _B1 = -_B1;
      }
 
      for (int j = 0; j < 22; j++) {
        firTemp[j] = firCoefficients[j];
        iirTemp[j] = iirCoefficients[j];
      }
      for (int j = 2; j < 22; j++) {
        firCoefficients[j] =
            _A0 * firTemp[j] + _A1 * firTemp[j - 1] + _A2 * firTemp[j - 2];
        iirCoefficients[j] =
            iirTemp[j] - _B1 * iirTemp[j - 1] - B2 * iirTemp[j - 2];
      }
    } 
 
    iirCoefficients[2] = 0;
    for (int j = 0; j < filterOrder; j++) {
      firCoefficients[j] = firCoefficients[j + 2];
      iirCoefficients[j] = -iirCoefficients[j + 2];
    }
    effectsuite_t SA = 0;
    effectsuite_t SB = 0;
    if (shelfType == 0) {
      for (int j = 0; j < filterOrder; j++) {
        SA += firCoefficients[j];
        SB += iirCoefficients[j];
      }
    } else {
      for (int j = 0; j < order; j++) {
        SA += firCoefficients[j] * pow(-1, j);
        SB += iirCoefficients[j] * pow(-1, j);
      }
    }
 
    const effectsuite_t gain = SA / (1 - SB);
    for (int j = 0; j < filterOrder; j++) {
      firCoefficients[j] /= gain;
    }
 
    return true;
  }
 
protected:
  bool changeChebyICoefficients(effectsuite_t cutFreq, bool shelfType, effectsuite_t ripple,
                                int poles) {
    filterOrder = poles + 1;
    clearMemory();
    allocateBufferMemory();
    setChebyICoefficients(cutFreq, shelfType, ripple);
 
    return true;
  }
 
  bool setSimpleLpf(int order) {
    filterOrder = order;
    clearMemory();
    allocateBufferMemory();
    firCoefficients = new effectsuite_t[filterOrder];
    iirCoefficients = new effectsuite_t[filterOrder];
    std::fill(iirCoefficients, iirCoefficients + filterOrder, 0);
    int coef = 1;
    effectsuite_t gain = 0;
    for (int j = 0; j < filterOrder; j++) {
      if (j == 0) {
        coef = 1;
      } else {
        coef = coef * (filterOrder - j) / j;
      }
 
      firCoefficients[j] = (effectsuite_t)coef;
      gain += firCoefficients[j];
    }
 
    for (int j = 0; j <= filterOrder; j++) {
      firCoefficients[j] /= gain;
    }
 
    return true;
  }
 
protected:
  void incBufferIndex() {
    bufferIndex++;
    bufferIndex %= filterOrder;
  }
 
  void clearMemory() {
    if (firCoefficients) {
      delete[] firCoefficients;
    }
 
    if (iirCoefficients) {
      delete[] iirCoefficients;
    }
  }
 
  void allocateBufferMemory() {
    if (firBuffer) {
      delete[] firBuffer;
    }
 
    if (iirBuffer) {
      delete[] iirBuffer;
    }
    firBuffer = new effectsuite_t[filterOrder];
    iirBuffer = new effectsuite_t[filterOrder];
    std::fill(firBuffer, firBuffer + filterOrder, 0);
    std::fill(iirBuffer, iirBuffer + filterOrder, 0);
 
    if (firCoefficients) {
      delete[] firCoefficients;
    }
 
    if (iirCoefficients) {
      delete[] iirCoefficients;
    }
 
    if (firTemp) {
      delete[] firTemp;
    }
 
    if (iirTemp) {
      delete[] iirTemp;
    }
 
    firCoefficients = new effectsuite_t[22];
    iirCoefficients = new effectsuite_t[22];
    firTemp = new effectsuite_t[22];
    iirTemp = new effectsuite_t[22];
    std::fill(firCoefficients, firCoefficients + 22, 0);
    std::fill(iirCoefficients, iirCoefficients + 22, 0);
    std::fill(firTemp, firTemp + 22, 0);
    std::fill(iirTemp, iirTemp + 22, 0);
  }
 
  effectsuite_t rms(effectsuite_t sample) {
    rmsBuffer[rmsBufferIndex] = sample;
    effectsuite_t rmsValue = 0;
    for (int j = 0; j < rmsBufferIndex; j++) {
      int i = ((rmsBufferIndex - j) + rmsWindowSize) % rmsWindowSize;
      rmsValue += rmsBuffer[i] * rmsBuffer[i];
    }
 
    //  printf("samp: %e\tsum: %e\n", sample, rmsValue);
    rmsValue /= rmsWindowSize;
    rmsValue = sqrt(rmsValue);
 
    rmsBufferIndex++;
    rmsBufferIndex %= rmsWindowSize;
 
    return rmsValue;
  }
 
public:
protected:
  effectsuite_t *firCoefficients = 0;
  effectsuite_t *iirCoefficients = 0;
  effectsuite_t *firTemp = 0;
  effectsuite_t *iirTemp = 0;
  effectsuite_t *firBuffer = 0;
  effectsuite_t *iirBuffer = 0;
  int bufferIndex = 0;
  int filterOrder = 0;
  /***/
  int samplingRate = 0;
  const int rmsWindowSize = 128;
  int rmsBufferIndex = 0;
  effectsuite_t *rmsBuffer = new effectsuite_t[rmsWindowSize];
 
protected: // variables
  constexpr static const effectsuite_t pi = 3.141592653589793;
};
 
class SimpleLPF : public FilterEffectBase {
public:
  SimpleLPF(effectsuite_t cutoff, int order) {
    changeChebyICoefficients(cutoff, false, .1, order);
  };
 
  SimpleLPF(SimpleLPF &copy) = default;
 
  ~SimpleLPF() = default;
 
  SimpleLPF* clone(){
    return new SimpleLPF(*this);
  }
 
};
 
class SimpleChorus : public DelayEffectBase,
                     public ModulationBaseClass,
                     public SimpleLPF{
public:
//  SimpleChorus() : SimpleLPF(0.0001, 4) {}
  SimpleChorus(int extSampleRate=44100) :
        DelayEffectBase(static_cast<int>(0.031 * extSampleRate)), 
        ModulationBaseClass(extSampleRate),
        SimpleLPF(0.0001, 4) {
    swing = 0.005 * sampleRate;
    base = 0.015 * sampleRate;
    if (sampleRate != 0)
      setRandLfo();
  }
 
  SimpleChorus(SimpleChorus &copy)=default;
 
  ~SimpleChorus() = default;
 
  virtual effectsuite_t processDouble(effectsuite_t inputSample) override {
    delaySample(inputSample);
    const effectsuite_t waveDelay = getModSignal();
    const effectsuite_t delayAmount =
        ((int(currentDelayWriteIndex - waveDelay) + delayTimeSamples) %
         delayTimeSamples) +
        ((currentDelayWriteIndex - waveDelay) -
         trunc(currentDelayWriteIndex - waveDelay));
    const effectsuite_t out = .0 * inputSample + 1. * getInterpolatedOut(delayAmount);
    return out;
  }
 
  void setupChorus(effectsuite_t extSampleRate) {
    setupModulationBaseClass(extSampleRate);
    setupDelayEffectBase(effectsuite_t(extSampleRate) * .1);
    //    SimpleLPF(0.0004,4)
    setChebyICoefficients(0.00005, false, 0);
 
    swing = readSpeed * extSampleRate * 5;
    base = readSpeed * extSampleRate * 20;
    setRandLfo();
  }
 
  void setSwing(effectsuite_t swingAmount) { swing = swingAmount * sampleRate; }
 
  void setBase(effectsuite_t baseAmount) { base = baseAmount * sampleRate; }
 
  SimpleChorus* clone() override {
    return new SimpleChorus(*this);
  }
 
protected:
  effectsuite_t swing;
  effectsuite_t base;
  effectsuite_t modMin = .5;
  effectsuite_t modMax = .5;
  effectsuite_t modNorm = 1 / (modMax - modMin);
  const effectsuite_t readSpeed = ((readNoise() + 1) * .5) * .0005;
 
  effectsuite_t getModSignal() { return (readTable(readSpeed) * swing) + base; }
 
  void setRandLfo() {
    std::fill(iirBuffer, iirBuffer + filterOrder, .5);
    for (int i = 0; i < sampleRate; i++) {
      waveTable[i] = (readNoise() + 1) * .5;
      //        waveTable[i] = applyFilter((readNoise()+1)*.5);
      if (waveTable[i] < modMin)
        modMin = waveTable[i];
      if (waveTable[i] > modMax) {
        modMax = waveTable[i];
      }
    }
 
    modNorm = 1 / (modMax - modMin);
 
    // normalises the delay signal
    for (int i = 0; i < sampleRate; i++) {
      waveTable[i] -= modMin;
      waveTable[i] *= modNorm;
    }
 
    //  setOffSine();
 
    // this code fades out at the end and fades in at the start
    // to avoid any discontinuities int the signal.
    //    const int fadeSize = 10000;
    //    const effectsuite_t fadeSpeed = 2*M_PI/fadeSize;
    //    for (int i = 0; i < fadeSize; i++)
    //    {
    //        const int fadeIndex = ((sampleRate-fadeSize/2)+i)%sampleRate;
    //        waveTable[fadeIndex] *= (1+cos(fadeSpeed*i))*.5;
    //    }
  }
};
 
class FilteredDelay : public DelayEffectBase, public FilterEffectBase {
public:
  FilteredDelay(int delayInSamples, int sample_rate=44100) : DelayEffectBase(sample_rate) {
    delayTimeSamples = delayInSamples;
    changeChebyICoefficients(.05, true, .1, 4);
  };
 
  FilteredDelay(FilteredDelay&copy) = default;
 
  ~FilteredDelay() = default;
 
  void setDelayGain(effectsuite_t gain) {
    capGain(gain);
    delayGain = gain;
  }
 
  void setFeedbackGain(effectsuite_t gain) {
    capGain(gain);
    feedbackGain = gain;
  }
 
  effectsuite_t processDouble(effectsuite_t inputSample) override {
    delaySample(
        applyFilter((inputSample * delayGain) +
                    feedbackGain * getInterpolatedOut(currentDelayWriteIndex)));
    const effectsuite_t out = getInterpolatedOut(currentDelayWriteIndex) + inputSample;
    return out;
  }
 
  effect_t process(effect_t inputSample) override {
    return active_flag ? 32767.0 * processDouble(static_cast<effectsuite_t>(inputSample)/32767.0) : inputSample;
  }
 
  FilteredDelay *clone() override {
    return new FilteredDelay(*this);
  }
 
protected:
  void capGain(effectsuite_t &gain) {
    if (gain > 1.) {
      gain = 1.;
    } else if (gain < -1.) {
      gain = -1.;
    }
    return;
  }
 
protected:
  effectsuite_t delayGain = .707, feedbackGain = 0.0;
};
 
class SimpleDelay : public DelayEffectBase, public EffectSuiteBase {
public:
  SimpleDelay(int maxDelayInSamples=8810, int samplingRate=44100) {
    writeHeadIndex = 0;
    readHeadIndex = 1;
    currentDelaySamples = maxDelayInSamples;
    targetDelaySamples = maxDelayInSamples;
    setDelayTransitionTime(0.5);
  }
 
  SimpleDelay(SimpleDelay &copy) = default;
 
  ~SimpleDelay() = default;
 
  void setDelayGain(effectsuite_t gain) {
    capGain(gain);
    delayGain = gain;
  }
 
  void setFeedbackGain(effectsuite_t gain) {
    capGain(gain);
    feedbackGain = gain;
  }
 
  effectsuite_t processDouble(effectsuite_t inputSample) override {
    // write sample
    delayBuffer[writeHeadIndex] = inputSample;
    writeHeadIndex++;
    writeHeadIndex %= maxDelayBufferSize;
 
    // read sample
    effectsuite_t outSample = getSplineOut(readHeadIndex) + (inputSample * 1);
    if (delayTimeChanged) {
      count++;
      const effectsuite_t difference = (currentDelaySamples - targetDelaySamples);
      const effectsuite_t increment = delayIncrement * (difference / fabs(difference));
      currentDelaySamples -= increment;
      readHeadIndex += 1 + increment;
      readHeadIndex = std::fmod(readHeadIndex, maxDelayBufferSize);
      if (count > floor(delayTransitionTimeInSamples)) {
        currentDelaySamples = targetDelaySamples;
        readHeadIndex = floor(readHeadIndex);
        delayTimeChanged = false;
      }
    } else {
      readHeadIndex++;
      readHeadIndex = std::fmod(readHeadIndex, maxDelayBufferSize);
    }
    return outSample;
  }
 
  effect_t process(effect_t inputSample) override {
    return active_flag ? 32767.0 * processDouble(static_cast<effectsuite_t>(inputSample)/32767.0) : inputSample;
  }
 
  void setupSimpleDelay(int delayInSamples) {
    setupDelayEffectBase(delayInSamples);
  }
  void setDelayTime(effectsuite_t delayInSamples) {
    delayTimeChanged = true;
    targetDelaySamples = delayInSamples;
    const effectsuite_t delayTimeDifference = currentDelaySamples - targetDelaySamples;
    delayIncrement = delayTimeDifference / delayTransitionTimeInSamples;
    count = 0;
  }
  void setDelayTransitionTime(effectsuite_t seconds) {
    delayTransitionTime = seconds;
    delayTransitionTimeInSamples = seconds * sampleRate;
  }
 
  SimpleDelay* clone() override {
    return new SimpleDelay(*this);
  }
 
protected:
  void capGain(effectsuite_t &gain) {
    if (gain > 1.) {
      gain = 1.;
    } else if (gain < -1.) {
      gain = -1.;
    }
    return;
  }
  effectsuite_t getSplineOut(effectsuite_t bufferIndex) {
    const int n0 = floor(bufferIndex);
    const int n1 = (n0 + 1) % maxDelayBufferSize;
    const int n2 = (n0 + 2) % maxDelayBufferSize;
    const effectsuite_t alpha = bufferIndex - n0;
 
    const effectsuite_t a = delayBuffer[n1];
    const effectsuite_t c = ((3 * (delayBuffer[n2] - delayBuffer[n1])) -
                      (3 * (delayBuffer[n1] - delayBuffer[n0]))) *
                     0.25;
    const effectsuite_t b = (delayBuffer[n2] - delayBuffer[n1]) - (2 * c * 0.33333);
    const effectsuite_t d = (-c) * 0.33333;
    return a + (b * alpha) + (c * alpha * alpha) + (d * alpha * alpha * alpha);
  }
 
protected: // member vairables
  effectsuite_t delayGain = .707;
  effectsuite_t feedbackGain = 0.;
  effectsuite_t readHeadIndex;
  unsigned int writeHeadIndex;
  effectsuite_t currentDelaySamples;
  effectsuite_t targetDelaySamples;
  effectsuite_t delayIncrement;
  effectsuite_t invDelayIncrement;
  effectsuite_t delayTransitionTime;
  effectsuite_t delayTransitionTimeInSamples;
  int sampleRate;
  int count = 0;
  bool delayTimeChanged = false;
};
 
class SimpleFlanger : public DelayEffectBase, public EffectSuiteBase {
public:
  SimpleFlanger() = default;
  SimpleFlanger(SimpleFlanger&copy) = default;
  SimpleFlanger(effectsuite_t extSampleRate=44100)
      : DelayEffectBase(static_cast<int>(extSampleRate * 0.02)) {}
 
  ~SimpleFlanger() = default;
 
  void setEffectGain(effectsuite_t gain) { effectGain = capGain(gain); }
 
  void setDepth(const effectsuite_t depth) {
    if (depth > effectsuite_t(delayTimeSamples))
      modulationDepth = effectsuite_t(delayTimeSamples) - 1;
    else
      modulationDepth = depth;
  }
 
  void setRate(const effectsuite_t rate) {
    modulationRate = rate;
    setAngleDelta();
  }
 
  void setEffectParams(effectsuite_t gain, effectsuite_t depth, effectsuite_t rate) {
    setEffectGain(gain);
    setDepth(depth);
    setRate(rate);
  }
 
  effectsuite_t processDouble(effectsuite_t inputSample) override {
    delaySample(inputSample);
    const effectsuite_t out = ((1 - fabs(effectGain * .2)) * (inputSample) +
                        (effectGain * getInterpolatedOut(modulationIndex)));
    updateModulation();
    return out;
  }
 
  void setupSimpleFlanger(effectsuite_t extSampleRate) {
    setupDelayEffectBase(extSampleRate * .02);
    timeStep = 1. / extSampleRate;
    setEffectParams(.707, extSampleRate * .02, .1);
  }
 
  SimpleFlanger* clone() override {
    return new SimpleFlanger(*this);
  }
 
protected:
  effectsuite_t capGain(effectsuite_t gain) {
    if (gain > 1.) {
      gain = 1.;
    } else if (gain < -1.) {
      gain = -1.;
    }
    return gain;
  }
 
  void setAngleDelta() {
    const effectsuite_t cyclesPerSample = modulationRate * timeStep;
    angleDelta = cyclesPerSample * 2.0 * internal_Pi;
  }
 
  void updateModulation() {
    modulationAngle += angleDelta;
    modulationIndex = (currentDelayWriteIndex -
                       (modulationDepth * (1 + (sin(modulationAngle))))) -
                      12;
    modulationIndex =
        ((int(modulationIndex) + delayTimeSamples) % delayTimeSamples) +
        (modulationIndex - floor(modulationIndex));
  }
 
protected:
  constexpr static const effectsuite_t internal_Pi = 3.141592653589793;
 
  effectsuite_t modulationDepth = 1000, modulationRate = 0, effectGain = .01;
 
  effectsuite_t modulationIndex = 0;
 
  effectsuite_t timeStep = 1. / 44100.;
 
  effectsuite_t modulationConstant, modulationAngle = 0;
 
  // const effectsuite_t cyclesPerSample = modulationRate * timeStep;
  effectsuite_t angleDelta = 2 * internal_Pi * timeStep;
};
 
class EnvelopeFilter : public FilterEffectBase {
public:
  EnvelopeFilter() : envelopeFollower(.00006, 4) {
    // NOTE: Initialising chebyshev coeffcients allocates memory, perhaps alter
    // so that memory is already pre allocated
    changeChebyICoefficients(.01, false, .1, 4);
    envelopeFollower.setChebyICoefficients(.00006, false, 0);
  };
 
  ~EnvelopeFilter() = default;
  effectsuite_t processDouble(effectsuite_t sample) {
    setChebyICoefficients(0.001 + envelopeFollower.envelope(2 * sample), false,
                          .1); // Offset avoids zero cutoff value
    return applyFilter(sample);
  }
 
protected:
  SimpleLPF envelopeFollower;
};
 
} // namespace effectsuite_tools
 
#endif