blob: 797661387678b86fd1bbef65c37e5bb577ca7a2a [file] [log] [blame]
niklase@google.com470e71d2011-07-07 08:21:25 +00001/*
bjornv@webrtc.org0c6f9312012-01-30 09:39:08 +00002 * Copyright (c) 2012 The WebRTC project authors. All Rights Reserved.
niklase@google.com470e71d2011-07-07 08:21:25 +00003 *
4 * Use of this source code is governed by a BSD-style license
5 * that can be found in the LICENSE file in the root of the source
6 * tree. An additional intellectual property rights grant can be found
7 * in the file PATENTS. All contributing project authors may
8 * be found in the AUTHORS file in the root of the source tree.
9 */
10
Mirko Bonadei92ea95e2017-09-15 06:47:31 +020011#ifndef MODULES_AUDIO_PROCESSING_GAIN_CONTROL_IMPL_H_
12#define MODULES_AUDIO_PROCESSING_GAIN_CONTROL_IMPL_H_
niklase@google.com470e71d2011-07-07 08:21:25 +000013
Yves Gerey988cc082018-10-23 12:03:01 +020014#include <stddef.h>
15#include <stdint.h>
Jonas Olssona4d87372019-07-05 19:08:33 +020016
kwiberg88788ad2016-02-19 07:04:49 -080017#include <memory>
niklase@google.com470e71d2011-07-07 08:21:25 +000018#include <vector>
19
Yves Gerey988cc082018-10-23 12:03:01 +020020#include "absl/types/optional.h"
21#include "api/array_view.h"
Sam Zackrisson41478c72019-10-15 10:10:26 +020022#include "modules/audio_processing/agc/gain_control.h"
Steve Anton10542f22019-01-11 09:11:00 -080023#include "rtc_base/constructor_magic.h"
niklase@google.com470e71d2011-07-07 08:21:25 +000024
25namespace webrtc {
andrew@webrtc.org56e4a052014-02-27 22:23:17 +000026
peah135259a2016-10-28 03:12:11 -070027class ApmDataDumper;
niklase@google.com470e71d2011-07-07 08:21:25 +000028class AudioBuffer;
29
peahbfa97112016-03-10 21:09:04 -080030class GainControlImpl : public GainControl {
niklase@google.com470e71d2011-07-07 08:21:25 +000031 public:
Sam Zackrissonf0d1c032019-03-27 13:28:08 +010032 GainControlImpl();
33 GainControlImpl(const GainControlImpl&) = delete;
34 GainControlImpl& operator=(const GainControlImpl&) = delete;
35
peahbfa97112016-03-10 21:09:04 -080036 ~GainControlImpl() override;
niklase@google.com470e71d2011-07-07 08:21:25 +000037
peah701d6282016-10-25 05:42:20 -070038 void ProcessRenderAudio(rtc::ArrayView<const int16_t> packed_render_audio);
niklase@google.com470e71d2011-07-07 08:21:25 +000039 int AnalyzeCaptureAudio(AudioBuffer* audio);
peahb8fbb542016-03-15 02:28:08 -070040 int ProcessCaptureAudio(AudioBuffer* audio, bool stream_has_echo);
niklase@google.com470e71d2011-07-07 08:21:25 +000041
peahb8fbb542016-03-15 02:28:08 -070042 void Initialize(size_t num_proc_channels, int sample_rate_hz);
niklase@google.com470e71d2011-07-07 08:21:25 +000043
peah701d6282016-10-25 05:42:20 -070044 static void PackRenderAudioBuffer(AudioBuffer* audio,
45 std::vector<int16_t>* packed_buffer);
46
niklase@google.com470e71d2011-07-07 08:21:25 +000047 // GainControl implementation.
kjellander@webrtc.org14665ff2015-03-04 12:58:35 +000048 bool is_enabled() const override;
Sam Zackrissonf0d1c032019-03-27 13:28:08 +010049 int stream_analog_level() const override;
Minyue13b96ba2015-10-03 00:39:14 +020050 bool is_limiter_enabled() const override;
51 Mode mode() const override;
Per Åhgren0e3198e2019-11-18 08:52:22 +010052 int Enable(bool enable) override;
53 int set_mode(Mode mode) override;
aluebs11d4a422016-04-28 14:58:32 -070054 int compression_gain_db() const override;
Per Åhgren0e3198e2019-11-18 08:52:22 +010055 int set_analog_level_limits(int minimum, int maximum) override;
56 int set_compression_gain_db(int gain) override;
57 int set_target_level_dbfs(int level) override;
58 int enable_limiter(bool enable) override;
59 int set_stream_analog_level(int level) override;
aluebs11d4a422016-04-28 14:58:32 -070060
niklase@google.com470e71d2011-07-07 08:21:25 +000061 private:
peahbfa97112016-03-10 21:09:04 -080062 class GainController;
63
niklase@google.com470e71d2011-07-07 08:21:25 +000064 // GainControl implementation.
kjellander@webrtc.org14665ff2015-03-04 12:58:35 +000065 int target_level_dbfs() const override;
kjellander@webrtc.org14665ff2015-03-04 12:58:35 +000066 int analog_level_minimum() const override;
67 int analog_level_maximum() const override;
68 bool stream_is_saturated() const override;
niklase@google.com470e71d2011-07-07 08:21:25 +000069
Sam Zackrissonf0d1c032019-03-27 13:28:08 +010070 int Configure();
peahdf3efa82015-11-28 12:35:15 -080071
peah135259a2016-10-28 03:12:11 -070072 std::unique_ptr<ApmDataDumper> data_dumper_;
73
peahbfa97112016-03-10 21:09:04 -080074 bool enabled_ = false;
75
Sam Zackrissonf0d1c032019-03-27 13:28:08 +010076 Mode mode_;
77 int minimum_capture_level_;
78 int maximum_capture_level_;
79 bool limiter_enabled_;
80 int target_level_dbfs_;
81 int compression_gain_db_;
82 int analog_capture_level_;
83 bool was_analog_level_set_;
84 bool stream_is_saturated_;
peahdf3efa82015-11-28 12:35:15 -080085
peahbfa97112016-03-10 21:09:04 -080086 std::vector<std::unique_ptr<GainController>> gain_controllers_;
87
Sam Zackrissonf0d1c032019-03-27 13:28:08 +010088 absl::optional<size_t> num_proc_channels_;
89 absl::optional<int> sample_rate_hz_;
peahb8fbb542016-03-15 02:28:08 -070090
peah135259a2016-10-28 03:12:11 -070091 static int instance_counter_;
niklase@google.com470e71d2011-07-07 08:21:25 +000092};
93} // namespace webrtc
94
Mirko Bonadei92ea95e2017-09-15 06:47:31 +020095#endif // MODULES_AUDIO_PROCESSING_GAIN_CONTROL_IMPL_H_