]> git.sesse.net Git - nageru/blobdiff - audio_mixer.h
Implement the EQ in AudioMixer. (No UI yet.)
[nageru] / audio_mixer.h
index 75b468a120f7d1f39aa293009e9b76153ae4e9a3..c332671e0bebd6f385d66b8ddc40fd426fd765ad 100644 (file)
@@ -52,6 +52,13 @@ struct DeviceInfo {
        unsigned num_channels;
 };
 
+enum EQBand {
+       EQ_BAND_BASS = 0,
+       EQ_BAND_MID,
+       EQ_BAND_TREBLE,
+       NUM_EQ_BANDS
+};
+
 static inline uint64_t DeviceSpec_to_key(const DeviceSpec &device_spec)
 {
        return (uint64_t(device_spec.type) << 32) | device_spec.index;
@@ -101,14 +108,20 @@ public:
                locut_cutoff_hz = cutoff_hz;
        }
 
-       void set_locut_enabled(bool enabled)
+       void set_locut_enabled(unsigned bus, bool enabled)
+       {
+               locut_enabled[bus] = enabled;
+       }
+
+       bool get_locut_enabled(unsigned bus)
        {
-               locut_enabled = enabled;
+               return locut_enabled[bus];
        }
 
-       bool get_locut_enabled() const
+       void set_eq(unsigned bus_index, EQBand band, float db_gain)
        {
-               return locut_enabled;
+               assert(band >= 0 && band < NUM_EQ_BANDS);
+               eq_level_db[bus_index][band] = db_gain;
        }
 
        float get_limiter_threshold_dbfs() const
@@ -116,9 +129,9 @@ public:
                return limiter_threshold_dbfs;
        }
 
-       float get_compressor_threshold_dbfs() const
+       float get_compressor_threshold_dbfs(unsigned bus_index) const
        {
-               return compressor_threshold_dbfs;
+               return compressor_threshold_dbfs[bus_index];
        }
 
        void set_limiter_threshold_dbfs(float threshold_dbfs)
@@ -126,9 +139,9 @@ public:
                limiter_threshold_dbfs = threshold_dbfs;
        }
 
-       void set_compressor_threshold_dbfs(float threshold_dbfs)
+       void set_compressor_threshold_dbfs(unsigned bus_index, float threshold_dbfs)
        {
-               compressor_threshold_dbfs = threshold_dbfs;
+               compressor_threshold_dbfs[bus_index] = threshold_dbfs;
        }
 
        void set_limiter_enabled(bool enabled)
@@ -141,39 +154,39 @@ public:
                return limiter_enabled;
        }
 
-       void set_compressor_enabled(bool enabled)
+       void set_compressor_enabled(unsigned bus_index, bool enabled)
        {
-               compressor_enabled = enabled;
+               compressor_enabled[bus_index] = enabled;
        }
 
-       bool get_compressor_enabled() const
+       bool get_compressor_enabled(unsigned bus_index) const
        {
-               return compressor_enabled;
+               return compressor_enabled[bus_index];
        }
 
-       void set_gain_staging_db(float gain_db)
+       void set_gain_staging_db(unsigned bus_index, float gain_db)
        {
                std::unique_lock<std::mutex> lock(compressor_mutex);
-               level_compressor_enabled = false;
-               gain_staging_db = gain_db;
+               level_compressor_enabled[bus_index] = false;
+               gain_staging_db[bus_index] = gain_db;
        }
 
-       float get_gain_staging_db() const
+       float get_gain_staging_db(unsigned bus_index) const
        {
                std::unique_lock<std::mutex> lock(compressor_mutex);
-               return gain_staging_db;
+               return gain_staging_db[bus_index];
        }
 
-       void set_gain_staging_auto(bool enabled)
+       void set_gain_staging_auto(unsigned bus_index, bool enabled)
        {
                std::unique_lock<std::mutex> lock(compressor_mutex);
-               level_compressor_enabled = enabled;
+               level_compressor_enabled[bus_index] = enabled;
        }
 
-       bool get_gain_staging_auto() const
+       bool get_gain_staging_auto(unsigned bus_index) const
        {
                std::unique_lock<std::mutex> lock(compressor_mutex);
-               return level_compressor_enabled;
+               return level_compressor_enabled[bus_index];
        }
 
        void set_final_makeup_gain_db(float gain_db)
@@ -201,9 +214,20 @@ public:
                return final_makeup_gain_auto;
        }
 
+       void reset_peak(unsigned bus_index);
+
+       struct BusLevel {
+               float current_level_dbfs[2];  // Digital peak of last frame, left and right.
+               float peak_level_dbfs[2];  // Digital peak with hold, left and right.
+               float historic_peak_dbfs;
+               float gain_staging_db;
+               float compressor_attenuation_db;  // A positive number; 0.0 for no attenuation.
+       };
+
        typedef std::function<void(float level_lufs, float peak_db,
+                                  std::vector<BusLevel> bus_levels,
                                   float global_level_lufs, float range_low_lufs, float range_high_lufs,
-                                  float gain_staging_db, float final_makeup_gain_db,
+                                  float final_makeup_gain_db,
                                   float correlation)> audio_level_callback_t;
        void set_audio_level_callback(audio_level_callback_t callback)
        {
@@ -228,7 +252,10 @@ private:
        void reset_resampler_mutex_held(DeviceSpec device_spec);
        void reset_alsa_mutex_held(DeviceSpec device_spec);
        std::map<DeviceSpec, DeviceInfo> get_devices_mutex_held() const;
+       void apply_eq(unsigned bus_index, std::vector<float> *samples_bus);
        void update_meters(const std::vector<float> &samples);
+       void add_bus_to_master(unsigned bus_index, const std::vector<float> &samples_bus, std::vector<float> *samples_out);
+       void measure_bus_levels(unsigned bus_index, const std::vector<float> &left, const std::vector<float> &right);
        void send_audio_level_callback();
 
        unsigned num_cards;
@@ -241,15 +268,16 @@ private:
        AudioDevice alsa_inputs[MAX_ALSA_CARDS];  // Under audio_mutex.
        std::vector<ALSAInput::Device> available_alsa_cards;
 
-       StereoFilter locut;  // Default cutoff 120 Hz, 24 dB/oct.
        std::atomic<float> locut_cutoff_hz;
-       std::atomic<bool> locut_enabled{true};
+       StereoFilter locut[MAX_BUSES];  // Default cutoff 120 Hz, 24 dB/oct.
+       std::atomic<bool> locut_enabled[MAX_BUSES];
+       StereoFilter eq[MAX_BUSES][NUM_EQ_BANDS];  // The one for EQBand::MID isn't actually used (see comments in apply_eq()).
 
        // First compressor; takes us up to about -12 dBFS.
        mutable std::mutex compressor_mutex;
-       StereoCompressor level_compressor;  // Under compressor_mutex. Used to set/override gain_staging_db if <level_compressor_enabled>.
-       float gain_staging_db = 0.0f;  // Under compressor_mutex.
-       bool level_compressor_enabled = true;  // Under compressor_mutex.
+       std::unique_ptr<StereoCompressor> level_compressor[MAX_BUSES];  // Under compressor_mutex. Used to set/override gain_staging_db if <level_compressor_enabled>.
+       float gain_staging_db[MAX_BUSES];  // Under compressor_mutex.
+       bool level_compressor_enabled[MAX_BUSES];  // Under compressor_mutex.
 
        static constexpr float ref_level_dbfs = -14.0f;  // Chosen so that we end up around 0 LU in practice.
        static constexpr float ref_level_lufs = -23.0f;  // 0 LU, more or less by definition.
@@ -257,15 +285,27 @@ private:
        StereoCompressor limiter;
        std::atomic<float> limiter_threshold_dbfs{ref_level_dbfs + 4.0f};   // 4 dB.
        std::atomic<bool> limiter_enabled{true};
-       StereoCompressor compressor;
-       std::atomic<float> compressor_threshold_dbfs{ref_level_dbfs - 12.0f};  // -12 dB.
-       std::atomic<bool> compressor_enabled{true};
+       std::unique_ptr<StereoCompressor> compressor[MAX_BUSES];
+       std::atomic<float> compressor_threshold_dbfs[MAX_BUSES];
+       std::atomic<bool> compressor_enabled[MAX_BUSES];
+
+       // Note: The values here are not in dB.
+       struct PeakHistory {
+               float current_level = 0.0f;  // Peak of the last frame.
+               float historic_peak = 0.0f;  // Highest peak since last reset; no falloff.
+               float current_peak = 0.0f;  // Current peak of the peak meter.
+               float last_peak = 0.0f;
+               float age_seconds = 0.0f;   // Time since "last_peak" was set.
+       };
+       PeakHistory peak_history[MAX_BUSES][2];  // Separate for each channel. Under audio_mutex.
 
        double final_makeup_gain = 1.0;  // Under compressor_mutex. Read/write by the user. Note: Not in dB, we want the numeric precision so that we can change it slowly.
        bool final_makeup_gain_auto = true;  // Under compressor_mutex.
 
        InputMapping input_mapping;  // Under audio_mutex.
        std::atomic<float> fader_volume_db[MAX_BUSES] {{ 0.0f }};
+       float last_fader_volume_db[MAX_BUSES] { 0.0f };  // Under audio_mutex.
+       std::atomic<float> eq_level_db[MAX_BUSES][NUM_EQ_BANDS] {{{ 0.0f }}};
 
        audio_level_callback_t audio_level_callback = nullptr;
        mutable std::mutex audio_measure_mutex;