A bit more spike mitigation in Tone King and RONN

Chowdhury-DSP · Mar 14, 2022 · 64f794e · 64f794e
1 parent fb512f5
commit 64f794e
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 5 deletions.
diff --git a/src/processors/drive/RONN.cpp b/src/processors/drive/RONN.cpp
@@ -179,6 +179,8 @@ void RONN::reloadModel (int randomSeed)
 
     if (std::isnan (makeupGain) || std::isinf (makeupGain))
         reloadModel (randomSeed + 1);
+
+    makeupGain = jmax (makeupGain, 30.0f);
 }
 
 void RONN::prepare (double sampleRate, int samplesPerBlock)

diff --git a/src/processors/drive/king_of_tone/KingOfToneDrive.cpp b/src/processors/drive/king_of_tone/KingOfToneDrive.cpp
@@ -127,15 +127,17 @@ void KingOfToneDrive::prepare (double sampleRate, int samplesPerBlock)
     dcBlocker.prepare (sampleRate, samplesPerBlock);
 
     prevMode = (int) *modeParam;
+    prevNumChannels = 2;
 
     preBuffer.setSize (2, samplesPerBlock);
     doPreBuffering();
 }
 
 void KingOfToneDrive::doPreBuffering()
 {
+    preBuffer.setSize (prevNumChannels, preBuffer.getNumSamples(), false, false, true);
     const auto numSamples = preBuffer.getNumSamples();
-    for (int i = 0; i < int (fs * 0.5f); i += numSamples)
+    for (int i = 0; i < (int) fs; i += numSamples)
     {
         preBuffer.clear();
         processAudio (preBuffer);
@@ -144,16 +146,17 @@ void KingOfToneDrive::doPreBuffering()
 
 void KingOfToneDrive::processAudio (AudioBuffer<float>& buffer)
 {
+    const auto numSamples = (int) buffer.getNumSamples();
+    const auto numChannels = (int) buffer.getNumChannels();
+
     const auto currentMode = (int) *modeParam;
-    if (currentMode != prevMode)
+    if (currentMode != prevMode || numChannels != prevNumChannels)
     {
         prevMode = currentMode;
+        prevNumChannels = numChannels;
         doPreBuffering();
     }
 
-    const auto numSamples = (int) buffer.getNumSamples();
-    const auto numChannels = (int) buffer.getNumChannels();
-
     buffer.applyGain (0.2f); // voltage scaling
 
     for (int ch = 0; ch < numChannels; ++ch)
@@ -197,7 +200,9 @@ void KingOfToneDrive::processAudio (AudioBuffer<float>& buffer)
             FloatVectorOperations::multiply (x, Decibels::decibelsToGain (makeupGainDB), numSamples);
         }
         else
+        {
             FloatVectorOperations::multiply (x, Decibels::decibelsToGain (-12.0f), numSamples);
+        }
     }
 
     dcBlocker.processAudio (buffer);

diff --git a/src/processors/drive/king_of_tone/KingOfToneDrive.h b/src/processors/drive/king_of_tone/KingOfToneDrive.h
@@ -24,6 +24,7 @@ class KingOfToneDrive : public BaseProcessor
 
     SmoothedValue<float> driveParamSmooth[2];
     int prevMode = 0;
+    int prevNumChannels = 0;
 
     float fs = 48000.0f;
     chowdsp::FirstOrderHPF<float> inputFilter[2];