xref: /aosp_15_r20/external/webrtc/modules/audio_mixer/frame_combiner.h (revision d9f758449e529ab9291ac668be2861e7a55c2422)
1*d9f75844SAndroid Build Coastguard Worker /*
2*d9f75844SAndroid Build Coastguard Worker  *  Copyright (c) 2017 The WebRTC project authors. All Rights Reserved.
3*d9f75844SAndroid Build Coastguard Worker  *
4*d9f75844SAndroid Build Coastguard Worker  *  Use of this source code is governed by a BSD-style license
5*d9f75844SAndroid Build Coastguard Worker  *  that can be found in the LICENSE file in the root of the source
6*d9f75844SAndroid Build Coastguard Worker  *  tree. An additional intellectual property rights grant can be found
7*d9f75844SAndroid Build Coastguard Worker  *  in the file PATENTS.  All contributing project authors may
8*d9f75844SAndroid Build Coastguard Worker  *  be found in the AUTHORS file in the root of the source tree.
9*d9f75844SAndroid Build Coastguard Worker  */
10*d9f75844SAndroid Build Coastguard Worker 
11*d9f75844SAndroid Build Coastguard Worker #ifndef MODULES_AUDIO_MIXER_FRAME_COMBINER_H_
12*d9f75844SAndroid Build Coastguard Worker #define MODULES_AUDIO_MIXER_FRAME_COMBINER_H_
13*d9f75844SAndroid Build Coastguard Worker 
14*d9f75844SAndroid Build Coastguard Worker #include <memory>
15*d9f75844SAndroid Build Coastguard Worker #include <vector>
16*d9f75844SAndroid Build Coastguard Worker 
17*d9f75844SAndroid Build Coastguard Worker #include "api/array_view.h"
18*d9f75844SAndroid Build Coastguard Worker #include "api/audio/audio_frame.h"
19*d9f75844SAndroid Build Coastguard Worker #include "modules/audio_processing/agc2/limiter.h"
20*d9f75844SAndroid Build Coastguard Worker 
21*d9f75844SAndroid Build Coastguard Worker namespace webrtc {
22*d9f75844SAndroid Build Coastguard Worker class ApmDataDumper;
23*d9f75844SAndroid Build Coastguard Worker 
24*d9f75844SAndroid Build Coastguard Worker class FrameCombiner {
25*d9f75844SAndroid Build Coastguard Worker  public:
26*d9f75844SAndroid Build Coastguard Worker   enum class LimiterType { kNoLimiter, kApmAgcLimiter, kApmAgc2Limiter };
27*d9f75844SAndroid Build Coastguard Worker   explicit FrameCombiner(bool use_limiter);
28*d9f75844SAndroid Build Coastguard Worker   ~FrameCombiner();
29*d9f75844SAndroid Build Coastguard Worker 
30*d9f75844SAndroid Build Coastguard Worker   // Combine several frames into one. Assumes sample_rate,
31*d9f75844SAndroid Build Coastguard Worker   // samples_per_channel of the input frames match the parameters. The
32*d9f75844SAndroid Build Coastguard Worker   // parameters 'number_of_channels' and 'sample_rate' are needed
33*d9f75844SAndroid Build Coastguard Worker   // because 'mix_list' can be empty. The parameter
34*d9f75844SAndroid Build Coastguard Worker   // 'number_of_streams' is used for determining whether to pass the
35*d9f75844SAndroid Build Coastguard Worker   // data through a limiter.
36*d9f75844SAndroid Build Coastguard Worker   void Combine(rtc::ArrayView<AudioFrame* const> mix_list,
37*d9f75844SAndroid Build Coastguard Worker                size_t number_of_channels,
38*d9f75844SAndroid Build Coastguard Worker                int sample_rate,
39*d9f75844SAndroid Build Coastguard Worker                size_t number_of_streams,
40*d9f75844SAndroid Build Coastguard Worker                AudioFrame* audio_frame_for_mixing);
41*d9f75844SAndroid Build Coastguard Worker 
42*d9f75844SAndroid Build Coastguard Worker   // Stereo, 48 kHz, 10 ms.
43*d9f75844SAndroid Build Coastguard Worker   static constexpr size_t kMaximumNumberOfChannels = 8;
44*d9f75844SAndroid Build Coastguard Worker   static constexpr size_t kMaximumChannelSize = 48 * 10;
45*d9f75844SAndroid Build Coastguard Worker 
46*d9f75844SAndroid Build Coastguard Worker   using MixingBuffer = std::array<std::array<float, kMaximumChannelSize>,
47*d9f75844SAndroid Build Coastguard Worker                                   kMaximumNumberOfChannels>;
48*d9f75844SAndroid Build Coastguard Worker 
49*d9f75844SAndroid Build Coastguard Worker  private:
50*d9f75844SAndroid Build Coastguard Worker   std::unique_ptr<ApmDataDumper> data_dumper_;
51*d9f75844SAndroid Build Coastguard Worker   std::unique_ptr<MixingBuffer> mixing_buffer_;
52*d9f75844SAndroid Build Coastguard Worker   Limiter limiter_;
53*d9f75844SAndroid Build Coastguard Worker   const bool use_limiter_;
54*d9f75844SAndroid Build Coastguard Worker };
55*d9f75844SAndroid Build Coastguard Worker }  // namespace webrtc
56*d9f75844SAndroid Build Coastguard Worker 
57*d9f75844SAndroid Build Coastguard Worker #endif  // MODULES_AUDIO_MIXER_FRAME_COMBINER_H_
58