aleloi | 24899e5 | 2017-02-21 05:06:29 -0800 | [diff] [blame] | 1 | /* |
| 2 | * Copyright (c) 2017 The WebRTC project authors. All Rights Reserved. |
| 3 | * |
| 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 Bonadei | 92ea95e | 2017-09-15 06:47:31 +0200 | [diff] [blame] | 11 | #ifndef MODULES_AUDIO_MIXER_FRAME_COMBINER_H_ |
| 12 | #define MODULES_AUDIO_MIXER_FRAME_COMBINER_H_ |
aleloi | 24899e5 | 2017-02-21 05:06:29 -0800 | [diff] [blame] | 13 | |
| 14 | #include <memory> |
| 15 | #include <vector> |
| 16 | |
Alex Loiko | 8396e34 | 2018-06-21 12:04:05 +0200 | [diff] [blame] | 17 | #include "api/audio/audio_frame.h" |
Alessio Bazzica | 3e4c77f | 2018-11-01 21:31:38 +0100 | [diff] [blame] | 18 | #include "modules/audio_processing/agc2/limiter.h" |
aleloi | 24899e5 | 2017-02-21 05:06:29 -0800 | [diff] [blame] | 19 | |
| 20 | namespace webrtc { |
Alex Loiko | 507e8d1 | 2018-02-27 13:51:47 +0100 | [diff] [blame] | 21 | class ApmDataDumper; |
aleloi | 24899e5 | 2017-02-21 05:06:29 -0800 | [diff] [blame] | 22 | |
| 23 | class FrameCombiner { |
| 24 | public: |
Alex Loiko | 507e8d1 | 2018-02-27 13:51:47 +0100 | [diff] [blame] | 25 | enum class LimiterType { kNoLimiter, kApmAgcLimiter, kApmAgc2Limiter }; |
Alex Loiko | 507e8d1 | 2018-02-27 13:51:47 +0100 | [diff] [blame] | 26 | explicit FrameCombiner(bool use_limiter); |
aleloi | 24899e5 | 2017-02-21 05:06:29 -0800 | [diff] [blame] | 27 | ~FrameCombiner(); |
| 28 | |
| 29 | // Combine several frames into one. Assumes sample_rate, |
| 30 | // samples_per_channel of the input frames match the parameters. The |
aleloi | 2c9306e | 2017-03-29 04:25:16 -0700 | [diff] [blame] | 31 | // parameters 'number_of_channels' and 'sample_rate' are needed |
| 32 | // because 'mix_list' can be empty. The parameter |
| 33 | // 'number_of_streams' is used for determining whether to pass the |
| 34 | // data through a limiter. |
aleloi | 24899e5 | 2017-02-21 05:06:29 -0800 | [diff] [blame] | 35 | void Combine(const std::vector<AudioFrame*>& mix_list, |
| 36 | size_t number_of_channels, |
| 37 | int sample_rate, |
aleloi | 2c9306e | 2017-03-29 04:25:16 -0700 | [diff] [blame] | 38 | size_t number_of_streams, |
Alex Loiko | 507e8d1 | 2018-02-27 13:51:47 +0100 | [diff] [blame] | 39 | AudioFrame* audio_frame_for_mixing); |
aleloi | 24899e5 | 2017-02-21 05:06:29 -0800 | [diff] [blame] | 40 | |
Alex Loiko | b4977de | 2019-01-28 16:38:38 +0100 | [diff] [blame^] | 41 | // Stereo, 48 kHz, 10 ms. |
| 42 | static constexpr size_t kMaximumNumberOfChannels = 8; |
| 43 | static constexpr size_t kMaximumChannelSize = 48 * 10; |
| 44 | |
| 45 | using MixingBuffer = std::array<std::array<float, kMaximumChannelSize>, |
| 46 | kMaximumNumberOfChannels>; |
| 47 | |
aleloi | 24899e5 | 2017-02-21 05:06:29 -0800 | [diff] [blame] | 48 | private: |
Alex Loiko | 6f2fcb4 | 2018-03-14 12:27:05 +0100 | [diff] [blame] | 49 | void LogMixingStats(const std::vector<AudioFrame*>& mix_list, |
| 50 | int sample_rate, |
| 51 | size_t number_of_streams) const; |
| 52 | |
Alex Loiko | 507e8d1 | 2018-02-27 13:51:47 +0100 | [diff] [blame] | 53 | std::unique_ptr<ApmDataDumper> data_dumper_; |
Alex Loiko | b4977de | 2019-01-28 16:38:38 +0100 | [diff] [blame^] | 54 | std::unique_ptr<MixingBuffer> mixing_buffer_; |
Alessio Bazzica | 3e4c77f | 2018-11-01 21:31:38 +0100 | [diff] [blame] | 55 | Limiter limiter_; |
Alex Loiko | 8396e34 | 2018-06-21 12:04:05 +0200 | [diff] [blame] | 56 | const bool use_limiter_; |
Alex Loiko | 6f2fcb4 | 2018-03-14 12:27:05 +0100 | [diff] [blame] | 57 | mutable int uma_logging_counter_ = 0; |
aleloi | 24899e5 | 2017-02-21 05:06:29 -0800 | [diff] [blame] | 58 | }; |
| 59 | } // namespace webrtc |
| 60 | |
Mirko Bonadei | 92ea95e | 2017-09-15 06:47:31 +0200 | [diff] [blame] | 61 | #endif // MODULES_AUDIO_MIXER_FRAME_COMBINER_H_ |