1 /*
2 * Copyright (C) 2023 Huawei Device Co., Ltd.
3 * Licensed under the Apache License, Version 2.0 (the "License");
4 * you may not use this file except in compliance with the License.
5 * You may obtain a copy of the License at
6 *
7 * http://www.apache.org/licenses/LICENSE-2.0
8 *
9 * Unless required by applicable law or agreed to in writing, software
10 * distributed under the License is distributed on an "AS IS" BASIS,
11 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12 * See the License for the specific language governing permissions and
13 * limitations under the License.
14 */
15
16 #include "audio_buffers_manager.h"
17 #include "avcodec_log.h"
18
19 namespace {
20 constexpr OHOS::HiviewDFX::HiLogLabel LABEL = {LOG_CORE, LOG_DOMAIN, "AvCodec-AudioBuffersManager"};
21 constexpr uint8_t LOGD_FREQUENCY = 5;
22 } // namespace
23
24 namespace OHOS {
25 namespace MediaAVCodec {
26 constexpr short DEFAULT_SLEEP_TIME = 500;
27
~AudioBuffersManager()28 AudioBuffersManager::~AudioBuffersManager() {}
29
AudioBuffersManager(const uint32_t bufferSize,const std::string_view & name,const uint16_t count,const uint32_t metaSize)30 AudioBuffersManager::AudioBuffersManager(const uint32_t bufferSize, const std::string_view &name, const uint16_t count,
31 const uint32_t metaSize)
32 : isRunning_(true),
33 inBufIndexExist(count, false),
34 bufferCount_(count),
35 bufferSize_(bufferSize),
36 metaSize_(metaSize),
37 name_(name),
38 bufferInfo_(count)
39 {
40 initBuffers();
41 }
42
getMemory(const uint32_t & index) const43 std::shared_ptr<AudioBufferInfo> AudioBuffersManager::getMemory(const uint32_t &index) const noexcept
44 {
45 if (index >= bufferInfo_.size()) {
46 return nullptr;
47 }
48 AVCODEC_LOGD_LIMIT(LOGD_FREQUENCY, "start get memory,name:%{public}s,index:%{public}u", name_.data(), index);
49 std::lock_guard<std::mutex> lock(stateMutex_);
50 return bufferInfo_[index];
51 }
52
SetBufferBusy(const uint32_t & index)53 bool AudioBuffersManager::SetBufferBusy(const uint32_t &index)
54 {
55 if (index < bufferInfo_.size()) {
56 std::lock_guard<std::mutex> lock(stateMutex_);
57 bufferInfo_[index]->SetBufferOwned();
58 return true;
59 }
60 return false;
61 }
62
initBuffers()63 void AudioBuffersManager::initBuffers()
64 {
65 std::lock_guard<std::mutex> lock(stateMutex_);
66 AVCODEC_LOGD_LIMIT(LOGD_FREQUENCY, "start allocate %{public}s buffers,each buffer size:%{public}d",
67 name_.data(), bufferSize_);
68 for (size_t i = 0; i < bufferCount_; i++) {
69 bufferInfo_[i] = std::make_shared<AudioBufferInfo>(bufferSize_, name_, metaSize_);
70 inBufIndexQue_.emplace(i);
71 inBufIndexExist[i] = true;
72 }
73 }
74
RequestNewBuffer(uint32_t & index,std::shared_ptr<AudioBufferInfo> & buffer)75 bool AudioBuffersManager::RequestNewBuffer(uint32_t &index, std::shared_ptr<AudioBufferInfo> &buffer)
76 {
77 buffer = createNewBuffer();
78 if (buffer == nullptr) {
79 return false;
80 }
81 index = bufferInfo_.size() - 1;
82 inBufIndexExist.emplace_back(false);
83 return true;
84 }
85
RequestAvailableIndex(uint32_t & index)86 bool AudioBuffersManager::RequestAvailableIndex(uint32_t &index)
87 {
88 while (inBufIndexQue_.empty() && isRunning_) {
89 AVCODEC_LOGD("Request empty %{public}s buffer", name_.data());
90 std::unique_lock aLock(availableMutex_);
91 availableCondition_.wait_for(aLock, std::chrono::milliseconds(DEFAULT_SLEEP_TIME),
92 [this] { return !inBufIndexQue_.empty() || !isRunning_; });
93 }
94
95 if (!isRunning_) {
96 return false;
97 }
98 {
99 std::lock_guard<std::mutex> lock(stateMutex_);
100 index = inBufIndexQue_.front();
101 inBufIndexQue_.pop();
102 }
103 if (index >= bufferInfo_.size()) {
104 AVCODEC_LOGW("Request %{public}s buffer index is invalidate ,index:%{public}u.", name_.data(), index);
105 return false;
106 }
107 AVCODEC_LOGD_LIMIT(LOGD_FREQUENCY, "Request %{public}s buffer successful,index:%{public}u", name_.data(), index);
108 inBufIndexExist[index] = false;
109 bufferInfo_[index]->SetBufferOwned();
110 return true;
111 }
112
ReleaseAll()113 void AudioBuffersManager::ReleaseAll()
114 {
115 {
116 std::lock_guard<std::mutex> lock(stateMutex_);
117 while (!inBufIndexQue_.empty()) {
118 inBufIndexQue_.pop();
119 }
120 }
121 for (uint32_t i = 0; i < bufferInfo_.size(); ++i) {
122 bufferInfo_[i]->ResetBuffer();
123 inBufIndexQue_.emplace(i);
124 inBufIndexExist[i] = true;
125 }
126 AVCODEC_LOGD_LIMIT(LOGD_FREQUENCY, "release all %{public}s buffer.", name_.data());
127 }
128
SetRunning()129 void AudioBuffersManager::SetRunning()
130 {
131 isRunning_ = true;
132 }
133
DisableRunning()134 void AudioBuffersManager::DisableRunning()
135 {
136 isRunning_ = false;
137 availableCondition_.notify_all();
138 }
139
ReleaseBuffer(const uint32_t & index)140 bool AudioBuffersManager::ReleaseBuffer(const uint32_t &index)
141 {
142 if (index < bufferInfo_.size()) {
143 AVCODEC_LOGD_LIMIT(LOGD_FREQUENCY, "ReleaseBuffer %{public}s buffer,index:%{public}u", name_.data(), index);
144 std::lock_guard<std::mutex> lock(stateMutex_);
145 bufferInfo_[index]->ResetBuffer();
146 if (!inBufIndexExist[index]) {
147 inBufIndexQue_.emplace(index);
148 inBufIndexExist[index] = true;
149 }
150 availableCondition_.notify_all();
151 return true;
152 }
153 return false;
154 }
155
createNewBuffer()156 std::shared_ptr<AudioBufferInfo> AudioBuffersManager::createNewBuffer()
157 {
158 std::lock_guard<std::mutex> lock(stateMutex_);
159 std::shared_ptr<AudioBufferInfo> buffer = std::make_shared<AudioBufferInfo>(bufferSize_, name_, metaSize_);
160 bufferInfo_.emplace_back(buffer);
161 return buffer;
162 }
163 } // namespace MediaAVCodec
164 } // namespace OHOS