even more stuff

core/src/dsp/demod/am.h

@@ -0,0 +1,88 @@
+#pragma once
+#include "../processor.h"
+#include "../loop/agc.h"
+
+namespace dsp::demod {
+    class AM : public Processor<dsp::complex_t, float> {
+        using base_type = Processor<dsp::complex_t, float>;
+    public:
+        enum AGCMode {
+            CARRIER,
+            AUDIO,
+        };
+
+        AM() {}
+
+        AM(stream<complex_t>* in, AGCMode agcMode, double agcRate) { init(in, agcMode, agcRate); }
+
+        void init(stream<complex_t>* in, AGCMode agcMode, double agcRate) {
+            _agcMode = agcMode;
+
+            carrierAgc.init(NULL, 1.0, agcRate);
+            audioAgc.init(NULL, 1.0, agcRate);
+            
+            base_type::init(in);
+        }
+
+        void setAGCMode(AGCMode agcMode) {
+            assert(base_type::_block_init);
+            std::lock_guard<std::recursive_mutex> lck(base_type::ctrlMtx);
+            base_type::tempStop();
+            _agcMode = agcMode;
+            reset();
+            base_type::tempStart();
+        }
+
+        void setAGCRate(double agcRate) {
+            assert(base_type::_block_init);
+            std::lock_guard<std::recursive_mutex> lck(base_type::ctrlMtx);
+            carrierAgc.setRate(agcRate);
+            audioAgc.setRate(agcRate);
+        }
+
+        void reset() {
+            assert(base_type::_block_init);
+            std::lock_guard<std::recursive_mutex> lck(base_type::ctrlMtx);
+            base_type::tempStop();
+            carrierAgc.reset();
+            audioAgc.reset();
+            base_type::tempStart();
+        }
+
+        int process(int count, complex_t* in, float* out) {
+            // Apply carrier AGC if needed
+            if (_agcMode == AGCMode::CARRIER) {
+                carrierAgc.process(count, in, carrierAgc.out.writeBuf);
+                in = carrierAgc.out.writeBuf;
+            }
+
+            // Get magnitude of each sample (TODO: use block instead)
+            volk_32fc_magnitude_32f(out, (lv_32fc_t*)in, count);
+
+            // Apply audio AGC if needed
+            if (_agcMode == AGCMode::AUDIO) {
+                audioAgc.process(count, out, out);
+            }
+
+            return count;
+        }
+
+        int run() {
+            int count = base_type::_in->read();
+            if (count < 0) { return -1; }
+
+            process(count, base_type::_in->readBuf, base_type::out.writeBuf);
+
+            base_type::_in->flush();
+            if (!base_type::out.swap(count)) { return -1; }
+            return count;
+        }
+
+    protected:
+        AGCMode _agcMode;
+
+        loop::AGC<complex_t> carrierAgc;
+        loop::AGC<float> audioAgc;
+
+    };
+}