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 "avcodec_video_encoder_impl.h"
17 #include "avcodec_errors.h"
18 #include "avcodec_log.h"
19 #include "avcodec_trace.h"
20 #include "i_avcodec_service.h"
21
22 namespace {
23 constexpr OHOS::HiviewDFX::HiLogLabel LABEL = {LOG_CORE, LOG_DOMAIN_FRAMEWORK, "AVCodecVideoEncoderImpl"};
24 }
25
26 namespace OHOS {
27 namespace MediaAVCodec {
CreateByMime(const std::string & mime)28 std::shared_ptr<AVCodecVideoEncoder> VideoEncoderFactory::CreateByMime(const std::string &mime)
29 {
30 std::shared_ptr<AVCodecVideoEncoder> impl = nullptr;
31 Format format;
32
33 int32_t ret = CreateByMime(mime, format, impl);
34 CHECK_AND_RETURN_RET_LOG(ret == AVCS_ERR_OK || impl != nullptr, nullptr,
35 "AVCodec video encoder impl init failed, %{public}s",
36 AVCSErrorToString(static_cast<AVCodecServiceErrCode>(ret)).c_str());
37
38 return impl;
39 }
40
CreateByName(const std::string & name)41 std::shared_ptr<AVCodecVideoEncoder> VideoEncoderFactory::CreateByName(const std::string &name)
42 {
43 std::shared_ptr<AVCodecVideoEncoder> impl = nullptr;
44 Format format;
45
46 int32_t ret = CreateByName(name, format, impl);
47 CHECK_AND_RETURN_RET_LOG(ret == AVCS_ERR_OK || impl != nullptr, nullptr,
48 "AVCodec video encoder impl init failed, %{public}s",
49 AVCSErrorToString(static_cast<AVCodecServiceErrCode>(ret)).c_str());
50
51 return impl;
52 }
53
CreateByMime(const std::string & mime,Format & format,std::shared_ptr<AVCodecVideoEncoder> & encoder)54 int32_t VideoEncoderFactory::CreateByMime(const std::string &mime, Format &format,
55 std::shared_ptr<AVCodecVideoEncoder> &encoder)
56 {
57 auto impl = std::make_shared<AVCodecVideoEncoderImpl>();
58
59 int32_t ret = impl->Init(AVCODEC_TYPE_VIDEO_ENCODER, true, mime, format);
60 CHECK_AND_RETURN_RET_LOG(ret == AVCS_ERR_OK, ret, "AVCodec video encoder impl init failed, %{public}s",
61 AVCSErrorToString(static_cast<AVCodecServiceErrCode>(ret)).c_str());
62
63 encoder = impl;
64 return AVCS_ERR_OK;
65 }
66
CreateByName(const std::string & name,Format & format,std::shared_ptr<AVCodecVideoEncoder> & encoder)67 int32_t VideoEncoderFactory::CreateByName(const std::string &name, Format &format,
68 std::shared_ptr<AVCodecVideoEncoder> &encoder)
69 {
70 auto impl = std::make_shared<AVCodecVideoEncoderImpl>();
71
72 int32_t ret = impl->Init(AVCODEC_TYPE_VIDEO_ENCODER, false, name, format);
73 CHECK_AND_RETURN_RET_LOG(ret == AVCS_ERR_OK, ret, "AVCodec video encoder impl init failed, %{public}s",
74 AVCSErrorToString(static_cast<AVCodecServiceErrCode>(ret)).c_str());
75
76 encoder = impl;
77 return AVCS_ERR_OK;
78 }
79
Init(AVCodecType type,bool isMimeType,const std::string & name,Format & format)80 int32_t AVCodecVideoEncoderImpl::Init(AVCodecType type, bool isMimeType, const std::string &name, Format &format)
81 {
82 AVCODEC_FUNC_TRACE_WITH_TAG_CLIENT;
83
84 int32_t ret = AVCodecServiceFactory::GetInstance().CreateCodecService(codecClient_);
85 CHECK_AND_RETURN_RET_LOG_WITH_TAG(codecClient_ != nullptr, ret, "Codec client create failed");
86 ret = codecClient_->Init(type, isMimeType, name, *format.GetMeta());
87 this->SetTag(codecClient_->GetTag()); // execute with CodecServiceProxy initialization completed
88 return ret;
89 }
90
AVCodecVideoEncoderImpl()91 AVCodecVideoEncoderImpl::AVCodecVideoEncoderImpl()
92 {
93 AVCODEC_LOGD_WITH_TAG("AVCodecVideoEncoderImpl:0x%{public}06" PRIXPTR " Instances create", FAKE_POINTER(this));
94 }
95
~AVCodecVideoEncoderImpl()96 AVCodecVideoEncoderImpl::~AVCodecVideoEncoderImpl()
97 {
98 if (codecClient_ != nullptr) {
99 (void)AVCodecServiceFactory::GetInstance().DestroyCodecService(codecClient_);
100 codecClient_ = nullptr;
101 }
102 AVCODEC_LOGD_WITH_TAG("AVCodecVideoEncoderImpl:0x%{public}06" PRIXPTR " Instances destroy", FAKE_POINTER(this));
103 }
104
Configure(const Format & format)105 int32_t AVCodecVideoEncoderImpl::Configure(const Format &format)
106 {
107 CHECK_AND_RETURN_RET_LOG_WITH_TAG(codecClient_ != nullptr, AVCS_ERR_INVALID_OPERATION, "Codec service is nullptr");
108
109 AVCODEC_FUNC_TRACE_WITH_TAG_CLIENT;
110 return codecClient_->Configure(format);
111 }
112
SetCustomBuffer(std::shared_ptr<AVBuffer> buffer)113 int32_t AVCodecVideoEncoderImpl::SetCustomBuffer(std::shared_ptr<AVBuffer> buffer)
114 {
115 CHECK_AND_RETURN_RET_LOG_WITH_TAG(codecClient_ != nullptr, AVCS_ERR_INVALID_OPERATION, "Codec service is nullptr");
116
117 AVCODEC_FUNC_TRACE_WITH_TAG_CLIENT;
118 return codecClient_->SetCustomBuffer(buffer);
119 }
120
Prepare()121 int32_t AVCodecVideoEncoderImpl::Prepare()
122 {
123 CHECK_AND_RETURN_RET_LOG_WITH_TAG(codecClient_ != nullptr, AVCS_ERR_INVALID_OPERATION, "Codec service is nullptr");
124
125 AVCODEC_FUNC_TRACE_WITH_TAG_CLIENT;
126 return AVCS_ERR_OK;
127 }
128
Start()129 int32_t AVCodecVideoEncoderImpl::Start()
130 {
131 CHECK_AND_RETURN_RET_LOG_WITH_TAG(codecClient_ != nullptr, AVCS_ERR_INVALID_OPERATION, "Codec service is nullptr");
132
133 AVCODEC_FUNC_TRACE_WITH_TAG_CLIENT;
134 return codecClient_->Start();
135 }
136
Stop()137 int32_t AVCodecVideoEncoderImpl::Stop()
138 {
139 CHECK_AND_RETURN_RET_LOG_WITH_TAG(codecClient_ != nullptr, AVCS_ERR_INVALID_OPERATION, "Codec service is nullptr");
140
141 AVCODEC_FUNC_TRACE_WITH_TAG_CLIENT;
142 return codecClient_->Stop();
143 }
144
Flush()145 int32_t AVCodecVideoEncoderImpl::Flush()
146 {
147 CHECK_AND_RETURN_RET_LOG_WITH_TAG(codecClient_ != nullptr, AVCS_ERR_INVALID_OPERATION, "Codec service is nullptr");
148
149 AVCODEC_FUNC_TRACE_WITH_TAG_CLIENT;
150 return codecClient_->Flush();
151 }
152
NotifyEos()153 int32_t AVCodecVideoEncoderImpl::NotifyEos()
154 {
155 CHECK_AND_RETURN_RET_LOG_WITH_TAG(codecClient_ != nullptr, AVCS_ERR_INVALID_OPERATION, "Codec service is nullptr");
156
157 AVCODEC_FUNC_TRACE_WITH_TAG_CLIENT;
158 return codecClient_->NotifyEos();
159 }
160
Reset()161 int32_t AVCodecVideoEncoderImpl::Reset()
162 {
163 CHECK_AND_RETURN_RET_LOG_WITH_TAG(codecClient_ != nullptr, AVCS_ERR_INVALID_OPERATION, "Codec service is nullptr");
164
165 AVCODEC_FUNC_TRACE_WITH_TAG_CLIENT;
166 return codecClient_->Reset();
167 }
168
Release()169 int32_t AVCodecVideoEncoderImpl::Release()
170 {
171 CHECK_AND_RETURN_RET_LOG_WITH_TAG(codecClient_ != nullptr, AVCS_ERR_INVALID_OPERATION, "Codec service is nullptr");
172
173 AVCODEC_FUNC_TRACE_WITH_TAG_CLIENT;
174 return codecClient_->Release();
175 }
176
CreateInputSurface()177 sptr<Surface> AVCodecVideoEncoderImpl::CreateInputSurface()
178 {
179 CHECK_AND_RETURN_RET_LOG_WITH_TAG(codecClient_ != nullptr, nullptr, "Codec service is nullptr");
180
181 AVCODEC_FUNC_TRACE_WITH_TAG_CLIENT;
182 return codecClient_->CreateInputSurface();
183 }
184
QueueInputBuffer(uint32_t index,AVCodecBufferInfo info,AVCodecBufferFlag flag)185 int32_t AVCodecVideoEncoderImpl::QueueInputBuffer(uint32_t index, AVCodecBufferInfo info, AVCodecBufferFlag flag)
186 {
187 CHECK_AND_RETURN_RET_LOG_WITH_TAG(codecClient_ != nullptr, AVCS_ERR_INVALID_OPERATION, "Codec service is nullptr");
188
189 AVCODEC_FUNC_TRACE_WITH_TAG_CLIENT;
190 return codecClient_->QueueInputBuffer(index, info, flag);
191 }
192
QueueInputBuffer(uint32_t index)193 int32_t AVCodecVideoEncoderImpl::QueueInputBuffer(uint32_t index)
194 {
195 CHECK_AND_RETURN_RET_LOG_WITH_TAG(codecClient_ != nullptr, AVCS_ERR_INVALID_OPERATION, "Codec service is nullptr");
196
197 AVCODEC_FUNC_TRACE_WITH_TAG_CLIENT;
198 return codecClient_->QueueInputBuffer(index);
199 }
200
QueueInputParameter(uint32_t index)201 int32_t AVCodecVideoEncoderImpl::QueueInputParameter(uint32_t index)
202 {
203 CHECK_AND_RETURN_RET_LOG_WITH_TAG(codecClient_ != nullptr, AVCS_ERR_INVALID_OPERATION, "Codec service is nullptr");
204
205 AVCODEC_FUNC_TRACE_WITH_TAG_CLIENT;
206 return codecClient_->QueueInputParameter(index);
207 }
208
GetOutputFormat(Format & format)209 int32_t AVCodecVideoEncoderImpl::GetOutputFormat(Format &format)
210 {
211 CHECK_AND_RETURN_RET_LOG_WITH_TAG(codecClient_ != nullptr, AVCS_ERR_INVALID_OPERATION, "Codec service is nullptr");
212
213 AVCODEC_FUNC_TRACE_WITH_TAG_CLIENT;
214 return codecClient_->GetOutputFormat(format);
215 }
216
ReleaseOutputBuffer(uint32_t index)217 int32_t AVCodecVideoEncoderImpl::ReleaseOutputBuffer(uint32_t index)
218 {
219 CHECK_AND_RETURN_RET_LOG_WITH_TAG(codecClient_ != nullptr, AVCS_ERR_INVALID_OPERATION, "Codec service is nullptr");
220
221 AVCODEC_FUNC_TRACE_WITH_TAG_CLIENT;
222 return codecClient_->ReleaseOutputBuffer(index);
223 }
224
QueryInputBuffer(uint32_t & index,int64_t timeoutUs)225 int32_t AVCodecVideoEncoderImpl::QueryInputBuffer(uint32_t &index, int64_t timeoutUs)
226 {
227 CHECK_AND_RETURN_RET_LOG_WITH_TAG(codecClient_ != nullptr, AVCS_ERR_INVALID_OPERATION, "Codec service is nullptr");
228
229 AVCODEC_FUNC_TRACE_WITH_TAG_CLIENT;
230 return codecClient_->QueryInputBuffer(index, timeoutUs);
231 }
232
QueryOutputBuffer(uint32_t & index,int64_t timeoutUs)233 int32_t AVCodecVideoEncoderImpl::QueryOutputBuffer(uint32_t &index, int64_t timeoutUs)
234 {
235 CHECK_AND_RETURN_RET_LOG_WITH_TAG(codecClient_ != nullptr, AVCS_ERR_INVALID_OPERATION, "Codec service is nullptr");
236
237 AVCODEC_FUNC_TRACE_WITH_TAG_CLIENT;
238 return codecClient_->QueryOutputBuffer(index, timeoutUs);
239 }
240
GetInputBuffer(uint32_t index)241 std::shared_ptr<AVBuffer> AVCodecVideoEncoderImpl::GetInputBuffer(uint32_t index)
242 {
243 CHECK_AND_RETURN_RET_LOG_WITH_TAG(codecClient_ != nullptr, nullptr, "Codec service is nullptr");
244
245 AVCODEC_FUNC_TRACE_WITH_TAG_CLIENT;
246 return codecClient_->GetInputBuffer(index);
247 }
248
GetOutputBuffer(uint32_t index)249 std::shared_ptr<AVBuffer> AVCodecVideoEncoderImpl::GetOutputBuffer(uint32_t index)
250 {
251 CHECK_AND_RETURN_RET_LOG_WITH_TAG(codecClient_ != nullptr, nullptr, "Codec service is nullptr");
252
253 AVCODEC_FUNC_TRACE_WITH_TAG_CLIENT;
254 return codecClient_->GetOutputBuffer(index);
255 }
256
SetParameter(const Format & format)257 int32_t AVCodecVideoEncoderImpl::SetParameter(const Format &format)
258 {
259 CHECK_AND_RETURN_RET_LOG_WITH_TAG(codecClient_ != nullptr, AVCS_ERR_INVALID_OPERATION, "Codec service is nullptr");
260
261 AVCODEC_FUNC_TRACE_WITH_TAG_CLIENT;
262 return codecClient_->SetParameter(format);
263 }
264
SetCallback(const std::shared_ptr<AVCodecCallback> & callback)265 int32_t AVCodecVideoEncoderImpl::SetCallback(const std::shared_ptr<AVCodecCallback> &callback)
266 {
267 CHECK_AND_RETURN_RET_LOG_WITH_TAG(codecClient_ != nullptr, AVCS_ERR_INVALID_OPERATION, "Codec service is nullptr");
268 CHECK_AND_RETURN_RET_LOG_WITH_TAG(callback != nullptr, AVCS_ERR_INVALID_VAL, "Callback is nullptr");
269 AVCODEC_FUNC_TRACE_WITH_TAG_CLIENT;
270 return codecClient_->SetCallback(callback);
271 }
272
SetCallback(const std::shared_ptr<MediaCodecCallback> & callback)273 int32_t AVCodecVideoEncoderImpl::SetCallback(const std::shared_ptr<MediaCodecCallback> &callback)
274 {
275 CHECK_AND_RETURN_RET_LOG_WITH_TAG(codecClient_ != nullptr, AVCS_ERR_INVALID_OPERATION, "service died");
276 CHECK_AND_RETURN_RET_LOG_WITH_TAG(callback != nullptr, AVCS_ERR_INVALID_VAL, "callback is nullptr");
277
278 AVCODEC_FUNC_TRACE_WITH_TAG_CLIENT;
279 return codecClient_->SetCallback(callback);
280 }
281
SetCallback(const std::shared_ptr<MediaCodecParameterCallback> & callback)282 int32_t AVCodecVideoEncoderImpl::SetCallback(const std::shared_ptr<MediaCodecParameterCallback> &callback)
283 {
284 CHECK_AND_RETURN_RET_LOG_WITH_TAG(codecClient_ != nullptr, AVCS_ERR_INVALID_OPERATION, "service died");
285 CHECK_AND_RETURN_RET_LOG_WITH_TAG(callback != nullptr, AVCS_ERR_INVALID_VAL, "callback is nullptr");
286
287 AVCODEC_FUNC_TRACE_WITH_TAG_CLIENT;
288 return codecClient_->SetCallback(callback);
289 }
290
SetCallback(const std::shared_ptr<MediaCodecParameterWithAttrCallback> & callback)291 int32_t AVCodecVideoEncoderImpl::SetCallback(const std::shared_ptr<MediaCodecParameterWithAttrCallback> &callback)
292 {
293 CHECK_AND_RETURN_RET_LOG_WITH_TAG(codecClient_ != nullptr, AVCS_ERR_INVALID_OPERATION, "service died");
294 CHECK_AND_RETURN_RET_LOG_WITH_TAG(callback != nullptr, AVCS_ERR_INVALID_VAL, "callback is nullptr");
295
296 AVCODEC_FUNC_TRACE_WITH_TAG_CLIENT;
297 return codecClient_->SetCallback(callback);
298 }
299
GetInputFormat(Format & format)300 int32_t AVCodecVideoEncoderImpl::GetInputFormat(Format &format)
301 {
302 CHECK_AND_RETURN_RET_LOG_WITH_TAG(codecClient_ != nullptr, AVCS_ERR_INVALID_OPERATION, "Codec service is nullptr");
303
304 AVCODEC_FUNC_TRACE_WITH_TAG_CLIENT;
305 return codecClient_->GetInputFormat(format);
306 }
307 } // namespace MediaAVCodec
308 } // namespace OHOS
309