OLD | NEW |
1 /* | 1 /* |
2 * Copyright (c) 2012 The WebRTC project authors. All Rights Reserved. | 2 * Copyright (c) 2012 The WebRTC project authors. All Rights Reserved. |
3 * | 3 * |
4 * Use of this source code is governed by a BSD-style license | 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 | 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 | 6 * tree. An additional intellectual property rights grant can be found |
7 * in the file PATENTS. All contributing project authors may | 7 * in the file PATENTS. All contributing project authors may |
8 * be found in the AUTHORS file in the root of the source tree. | 8 * be found in the AUTHORS file in the root of the source tree. |
9 */ | 9 */ |
10 | 10 |
11 #ifndef WEBRTC_MODULES_AUDIO_PROCESSING_GAIN_CONTROL_IMPL_H_ | 11 #ifndef WEBRTC_MODULES_AUDIO_PROCESSING_GAIN_CONTROL_IMPL_H_ |
12 #define WEBRTC_MODULES_AUDIO_PROCESSING_GAIN_CONTROL_IMPL_H_ | 12 #define WEBRTC_MODULES_AUDIO_PROCESSING_GAIN_CONTROL_IMPL_H_ |
13 | 13 |
14 #include <memory> | 14 #include <memory> |
15 #include <vector> | 15 #include <vector> |
16 | 16 |
| 17 #include "webrtc/base/constructormagic.h" |
17 #include "webrtc/base/criticalsection.h" | 18 #include "webrtc/base/criticalsection.h" |
18 #include "webrtc/base/thread_annotations.h" | 19 #include "webrtc/base/thread_annotations.h" |
19 #include "webrtc/common_audio/swap_queue.h" | 20 #include "webrtc/common_audio/swap_queue.h" |
20 #include "webrtc/modules/audio_processing/include/audio_processing.h" | 21 #include "webrtc/modules/audio_processing/include/audio_processing.h" |
21 #include "webrtc/modules/audio_processing/processing_component.h" | 22 #include "webrtc/modules/audio_processing/processing_component.h" |
22 | 23 |
23 namespace webrtc { | 24 namespace webrtc { |
24 | 25 |
25 class AudioBuffer; | 26 class AudioBuffer; |
26 | 27 |
27 class GainControlImpl : public GainControl, | 28 class GainControlImpl : public GainControl { |
28 public ProcessingComponent { | |
29 public: | 29 public: |
30 GainControlImpl(const AudioProcessing* apm, | 30 GainControlImpl(const AudioProcessing* apm, |
31 rtc::CriticalSection* crit_render, | 31 rtc::CriticalSection* crit_render, |
32 rtc::CriticalSection* crit_capture); | 32 rtc::CriticalSection* crit_capture); |
33 virtual ~GainControlImpl(); | 33 ~GainControlImpl() override; |
34 | 34 |
35 int ProcessRenderAudio(AudioBuffer* audio); | 35 int ProcessRenderAudio(AudioBuffer* audio); |
36 int AnalyzeCaptureAudio(AudioBuffer* audio); | 36 int AnalyzeCaptureAudio(AudioBuffer* audio); |
37 int ProcessCaptureAudio(AudioBuffer* audio); | 37 int ProcessCaptureAudio(AudioBuffer* audio); |
38 | 38 |
39 // ProcessingComponent implementation. | 39 void Initialize(); |
40 int Initialize() override; | |
41 | 40 |
42 // GainControl implementation. | 41 // GainControl implementation. |
43 bool is_enabled() const override; | 42 bool is_enabled() const override; |
44 int stream_analog_level() override; | 43 int stream_analog_level() override; |
45 bool is_limiter_enabled() const override; | 44 bool is_limiter_enabled() const override; |
46 Mode mode() const override; | 45 Mode mode() const override; |
47 | 46 |
48 // Reads render side data that has been queued on the render call. | 47 // Reads render side data that has been queued on the render call. |
49 void ReadQueuedRenderData(); | 48 void ReadQueuedRenderData(); |
50 | 49 |
51 private: | 50 private: |
| 51 class GainController; |
| 52 |
52 // GainControl implementation. | 53 // GainControl implementation. |
53 int Enable(bool enable) override; | 54 int Enable(bool enable) override; |
54 int set_stream_analog_level(int level) override; | 55 int set_stream_analog_level(int level) override; |
55 int set_mode(Mode mode) override; | 56 int set_mode(Mode mode) override; |
56 int set_target_level_dbfs(int level) override; | 57 int set_target_level_dbfs(int level) override; |
57 int target_level_dbfs() const override; | 58 int target_level_dbfs() const override; |
58 int set_compression_gain_db(int gain) override; | 59 int set_compression_gain_db(int gain) override; |
59 int compression_gain_db() const override; | 60 int compression_gain_db() const override; |
60 int enable_limiter(bool enable) override; | 61 int enable_limiter(bool enable) override; |
61 int set_analog_level_limits(int minimum, int maximum) override; | 62 int set_analog_level_limits(int minimum, int maximum) override; |
62 int analog_level_minimum() const override; | 63 int analog_level_minimum() const override; |
63 int analog_level_maximum() const override; | 64 int analog_level_maximum() const override; |
64 bool stream_is_saturated() const override; | 65 bool stream_is_saturated() const override; |
65 | 66 |
66 // ProcessingComponent implementation. | 67 size_t num_handles_required() const; |
67 void* CreateHandle() const override; | |
68 int InitializeHandle(void* handle) const override; | |
69 int ConfigureHandle(void* handle) const override; | |
70 void DestroyHandle(void* handle) const override; | |
71 size_t num_handles_required() const override; | |
72 int GetHandleError(void* handle) const override; | |
73 | 68 |
74 void AllocateRenderQueue(); | 69 void AllocateRenderQueue(); |
| 70 int Configure(); |
75 | 71 |
76 // Not guarded as its public API is thread safe. | 72 // Not guarded as its public API is thread safe. |
77 const AudioProcessing* apm_; | 73 const AudioProcessing* apm_; |
78 | 74 |
79 rtc::CriticalSection* const crit_render_ ACQUIRED_BEFORE(crit_capture_); | 75 rtc::CriticalSection* const crit_render_ ACQUIRED_BEFORE(crit_capture_); |
80 rtc::CriticalSection* const crit_capture_; | 76 rtc::CriticalSection* const crit_capture_; |
81 | 77 |
| 78 bool enabled_ = false; |
| 79 |
82 Mode mode_ GUARDED_BY(crit_capture_); | 80 Mode mode_ GUARDED_BY(crit_capture_); |
83 int minimum_capture_level_ GUARDED_BY(crit_capture_); | 81 int minimum_capture_level_ GUARDED_BY(crit_capture_); |
84 int maximum_capture_level_ GUARDED_BY(crit_capture_); | 82 int maximum_capture_level_ GUARDED_BY(crit_capture_); |
85 bool limiter_enabled_ GUARDED_BY(crit_capture_); | 83 bool limiter_enabled_ GUARDED_BY(crit_capture_); |
86 int target_level_dbfs_ GUARDED_BY(crit_capture_); | 84 int target_level_dbfs_ GUARDED_BY(crit_capture_); |
87 int compression_gain_db_ GUARDED_BY(crit_capture_); | 85 int compression_gain_db_ GUARDED_BY(crit_capture_); |
88 std::vector<int> capture_levels_ GUARDED_BY(crit_capture_); | |
89 int analog_capture_level_ GUARDED_BY(crit_capture_); | 86 int analog_capture_level_ GUARDED_BY(crit_capture_); |
90 bool was_analog_level_set_ GUARDED_BY(crit_capture_); | 87 bool was_analog_level_set_ GUARDED_BY(crit_capture_); |
91 bool stream_is_saturated_ GUARDED_BY(crit_capture_); | 88 bool stream_is_saturated_ GUARDED_BY(crit_capture_); |
92 | 89 |
93 size_t render_queue_element_max_size_ GUARDED_BY(crit_render_) | 90 size_t render_queue_element_max_size_ GUARDED_BY(crit_render_) |
94 GUARDED_BY(crit_capture_); | 91 GUARDED_BY(crit_capture_); |
95 std::vector<int16_t> render_queue_buffer_ GUARDED_BY(crit_render_); | 92 std::vector<int16_t> render_queue_buffer_ GUARDED_BY(crit_render_); |
96 std::vector<int16_t> capture_queue_buffer_ GUARDED_BY(crit_capture_); | 93 std::vector<int16_t> capture_queue_buffer_ GUARDED_BY(crit_capture_); |
97 | 94 |
98 // Lock protection not needed. | 95 // Lock protection not needed. |
99 std::unique_ptr< | 96 std::unique_ptr< |
100 SwapQueue<std::vector<int16_t>, RenderQueueItemVerifier<int16_t>>> | 97 SwapQueue<std::vector<int16_t>, RenderQueueItemVerifier<int16_t>>> |
101 render_signal_queue_; | 98 render_signal_queue_; |
| 99 |
| 100 std::vector<std::unique_ptr<GainController>> gain_controllers_; |
| 101 |
| 102 RTC_DISALLOW_IMPLICIT_CONSTRUCTORS(GainControlImpl); |
102 }; | 103 }; |
103 } // namespace webrtc | 104 } // namespace webrtc |
104 | 105 |
105 #endif // WEBRTC_MODULES_AUDIO_PROCESSING_GAIN_CONTROL_IMPL_H_ | 106 #endif // WEBRTC_MODULES_AUDIO_PROCESSING_GAIN_CONTROL_IMPL_H_ |
OLD | NEW |