feat: complete dynamic filter

ZL-Audio · Dec 24, 2023 · af61845 · af61845
1 parent 7d747b3
commit af61845
Show file tree

Hide file tree

Showing 2 changed files with 77 additions and 9 deletions.
diff --git a/source/dsp/iir_filter/dynamic_filter.cpp b/source/dsp/iir_filter/dynamic_filter.cpp
@@ -7,8 +7,32 @@
 //
 // You should have received a copy of the GNU General Public License along with ZLEqualizer. If not, see <https://www.gnu.org/licenses/>.
 
-//
-// Created by Zishu Liu on 12/19/23.
-//
-
 #include "dynamic_filter.h"
+
+namespace zlIIR {
+    template<typename FloatType>
+    void DynamicFilter<FloatType>::prepare(const juce::dsp::ProcessSpec &spec) {
+        mFilter.prepare(spec);
+        tFilter.prepare(spec);
+        mixer.prepare(spec);
+        tBuffer.setSize(static_cast<int>(spec.numChannels),
+                        static_cast<int>(spec.maximumBlockSize));
+    }
+
+    template<typename FloatType>
+    void DynamicFilter<FloatType>::process(juce::AudioBuffer<FloatType> &buffer) {
+        mFilter.process(buffer);
+        if (dynamicON.load()) {
+            tBuffer.makeCopyOf(buffer, true);
+            tFilter.process(tBuffer);
+            mixer.pushDrySamples(juce::dsp::AudioBlock<FloatType>(buffer));
+            mixer.mixWetSamples(juce::dsp::AudioBlock<FloatType>(tBuffer));
+        }
+    }
+
+    template
+    class DynamicFilter<float>;
+
+    template
+    class DynamicFilter<double>;
+}
diff --git a/source/dsp/iir_filter/dynamic_filter.h b/source/dsp/iir_filter/dynamic_filter.h
@@ -7,10 +7,6 @@
 //
 // You should have received a copy of the GNU General Public License along with ZLEqualizer. If not, see <https://www.gnu.org/licenses/>.
 
-//
-// Created by Zishu Liu on 12/19/23.
-//
-
 #ifndef ZLEQUALIZER_IIR_FILTER_H
 #define ZLEQUALIZER_IIR_FILTER_H
 
@@ -19,8 +15,56 @@
 namespace zlIIR {
     template<typename FloatType>
     class DynamicFilter {
+    public:
+        DynamicFilter() = default;
+
+        void prepare(const juce::dsp::ProcessSpec &spec);
+
+        void process(juce::AudioBuffer<FloatType> &buffer);
+
+        inline void setFreq(FloatType x) {
+            mFilter.setFreq(x);
+            tFilter.setFreq(x);
+        }
+
+        inline void setGainMain(FloatType x) {
+            mFilter.setGain(x);
+        }
+
+        inline void setGainTarget(FloatType x) {
+            tFilter.setGain(x);
+        }
+
+        inline void setQMain(FloatType x) {
+            mFilter.setQ(x);
+        }
+
+        inline void setQTarget(FloatType x) {
+            tFilter.setQ(x);
+        }
+
+        inline void setFilterType(FilterType x) {
+            mFilter.setFilterType(x);
+            tFilter.setFilterType(x);
+        }
+
+        inline void setOrder(size_t x) {
+            mFilter.setOrder(x);
+            tFilter.setOrder(x);
+        }
+
+        inline void setTargetProportion(FloatType x) {
+            mixer.setWetMixProportion(x);
+        }
+
+        inline void setDynamicON(bool x) {
+            dynamicON.store(x);
+        }
+
     private:
-        SingleFilter<FloatType> filter0, filter1;
+        SingleFilter<FloatType> mFilter, tFilter;
+        juce::dsp::DryWetMixer<FloatType> mixer;
+        juce::AudioBuffer<FloatType> tBuffer;
         std::atomic<bool> dynamicON = false;
     };
 }