OLD | NEW |
---|---|
(Empty) | |
1 /* | |
2 * Copyright (c) 2016 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 | |
11 #include "webrtc/modules/audio_processing/aec3/frame_blocker.h" | |
12 | |
13 #include <algorithm> | |
14 | |
15 #include "webrtc/base/checks.h" | |
16 #include "webrtc/modules/audio_processing/aec3/aec3_constants.h" | |
17 | |
18 namespace webrtc { | |
19 | |
20 FrameBlocker::FrameBlocker(size_t num_bands) | |
21 : num_bands_(num_bands), buffer_(num_bands_) { | |
22 for (auto& b : buffer_) { | |
23 b.reserve(kBlockSize); | |
24 RTC_DCHECK(b.empty()); | |
25 } | |
26 } | |
27 | |
28 FrameBlocker::~FrameBlocker() = default; | |
29 | |
30 void FrameBlocker::InsertSubFrameAndExtractBlock( | |
31 const std::vector<rtc::ArrayView<float>>& sub_frame, | |
32 std::vector<std::vector<float>>* block) { | |
33 RTC_DCHECK_EQ(num_bands_, block->size()); | |
34 RTC_DCHECK_EQ(num_bands_, sub_frame.size()); | |
35 for (size_t i = 0; i < num_bands_; ++i) { | |
36 RTC_DCHECK_GE(kBlockSize - 16, buffer_[i].size()); | |
37 RTC_DCHECK_EQ(kBlockSize, (*block)[i].size()); | |
38 RTC_DCHECK_EQ(kSubFrameLength, sub_frame[i].size()); | |
39 const int samples_to_block = kBlockSize - buffer_[i].size(); | |
40 (*block)[i].resize(0); | |
hlundin-webrtc
2016/12/22 10:23:57
clear
peah-webrtc
2016/12/22 16:40:04
Afaics, there is no guarantee from the standard th
| |
41 (*block)[i].insert((*block)[i].begin(), buffer_[i].begin(), | |
42 buffer_[i].end()); | |
43 (*block)[i].insert((*block)[i].begin() + buffer_[i].size(), | |
44 sub_frame[i].begin(), | |
45 sub_frame[i].begin() + samples_to_block); | |
46 buffer_[i].resize(0); | |
hlundin-webrtc
2016/12/22 10:23:57
clear
peah-webrtc
2016/12/22 16:40:04
Afaics, there is no guarantee from the standard th
| |
47 buffer_[i].insert(buffer_[i].begin(), | |
48 sub_frame[i].begin() + samples_to_block, | |
49 sub_frame[i].end()); | |
50 } | |
51 } | |
52 | |
53 bool FrameBlocker::IsBlockAvailable() const { | |
54 return kBlockSize == buffer_[0].size(); | |
55 } | |
56 | |
57 void FrameBlocker::ExtractBlock(std::vector<std::vector<float>>* block) { | |
58 RTC_DCHECK_EQ(num_bands_, block->size()); | |
59 RTC_DCHECK(IsBlockAvailable()); | |
60 for (size_t i = 0; i < num_bands_; ++i) { | |
61 RTC_DCHECK_EQ(kBlockSize, buffer_[i].size()); | |
62 RTC_DCHECK_EQ(kBlockSize, (*block)[i].size()); | |
63 (*block)[i].resize(0); | |
hlundin-webrtc
2016/12/22 10:23:57
clear
peah-webrtc
2016/12/22 16:40:04
Afaics, there is no guarantee from the standard th
| |
64 (*block)[i].insert((*block)[i].begin(), buffer_[i].begin(), | |
65 buffer_[i].end()); | |
66 buffer_[i].resize(0); | |
67 } | |
68 } | |
69 | |
70 } // namespace webrtc | |
OLD | NEW |