1 /*
2 * Copyright 2022 Shenzhen Kaihong DID 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 #include "icodec_buffer.h"
16 #include <hdf_base.h>
17 #include <hdf_log.h>
18 #include <poll.h>
19 #include <securec.h>
20 #include "codec_dyna_buffer.h"
21 #include "codec_handle_buffer.h"
22 #include "codec_share_buffer.h"
23
24 namespace OHOS {
25 namespace Codec {
26 namespace Omx {
ICodecBuffer(struct OmxCodecBuffer & codecBuffer)27 ICodecBuffer::ICodecBuffer(struct OmxCodecBuffer &codecBuffer)
28 {
29 codecBuffer_ = codecBuffer;
30 }
~ICodecBuffer()31 ICodecBuffer::~ICodecBuffer()
32 {}
33
CreateCodeBuffer(struct OmxCodecBuffer & codecBuffer)34 sptr<ICodecBuffer> ICodecBuffer::CreateCodeBuffer(struct OmxCodecBuffer &codecBuffer)
35 {
36 sptr<ICodecBuffer> buffer = nullptr;
37 switch (codecBuffer.bufferType) {
38 case CODEC_BUFFER_TYPE_AVSHARE_MEM_FD:
39 buffer = CodecShareBuffer::Create(codecBuffer);
40 break;
41 case CODEC_BUFFER_TYPE_HANDLE:
42 buffer = CodecHandleBuffer::Create(codecBuffer);
43 break;
44 case CODEC_BUFFER_TYPE_DYNAMIC_HANDLE:
45 buffer = CodecDynaBuffer::Create(codecBuffer);
46 break;
47 default:
48 HDF_LOGE("%s: bufferType[%{public}d] is unexpected", __func__, codecBuffer.bufferType);
49 break;
50 }
51 return buffer;
52 }
53
AllocateCodecBuffer(struct OmxCodecBuffer & codecBuffer)54 sptr<ICodecBuffer> ICodecBuffer::AllocateCodecBuffer(struct OmxCodecBuffer &codecBuffer)
55 {
56 sptr<ICodecBuffer> buffer = nullptr;
57 switch (codecBuffer.bufferType) {
58 case CODEC_BUFFER_TYPE_AVSHARE_MEM_FD:
59 buffer = CodecShareBuffer::Allocate(codecBuffer);
60 break;
61 default:
62 HDF_LOGE("%s: bufferType[%{public}d] is unexpected", __func__, codecBuffer.bufferType);
63 break;
64 }
65 return buffer;
66 }
67
GetCodecBuffer()68 struct OmxCodecBuffer &ICodecBuffer::GetCodecBuffer()
69 {
70 return codecBuffer_;
71 }
72
SetBufferId(int32_t bufferId)73 void ICodecBuffer::SetBufferId(int32_t bufferId)
74 {
75 codecBuffer_.bufferId = static_cast<uint32_t>(bufferId);
76 }
77
CheckInvalid(struct OmxCodecBuffer & codecBuffer)78 bool ICodecBuffer::CheckInvalid(struct OmxCodecBuffer &codecBuffer)
79 {
80 if (codecBuffer_.type != codecBuffer.type) {
81 HDF_LOGE("%{public}s :input buffer type [%{public}d], but expect type [%{public}d]", __func__,
82 codecBuffer.bufferType, codecBuffer_.bufferType);
83 return false;
84 }
85 return true;
86 }
87
SyncWait(int fd,uint32_t timeout)88 int32_t ICodecBuffer::SyncWait(int fd, uint32_t timeout)
89 {
90 int retCode = -EPERM;
91 if (fd < 0) {
92 HDF_LOGE("%{public}s The fence id is invalid.", __func__);
93 return retCode;
94 }
95
96 struct pollfd pollfds = {0};
97 pollfds.fd = fd;
98 pollfds.events = POLLIN;
99
100 do {
101 retCode = poll(&pollfds, 1, timeout);
102 } while (retCode == -EPERM && (errno == EINTR || errno == EAGAIN));
103
104 if (retCode == 0) {
105 retCode = -EPERM;
106 errno = ETIME;
107 } else if (retCode > 0) {
108 if (static_cast<uint32_t>(pollfds.revents) & (POLLERR | POLLNVAL)) {
109 retCode = -EPERM;
110 errno = EINVAL;
111 }
112 }
113 return retCode < 0 ? -errno : EOK;
114 }
115
FillOmxBuffer(struct OmxCodecBuffer & codecBuffer,OMX_BUFFERHEADERTYPE & omxBuffer)116 int32_t ICodecBuffer::FillOmxBuffer(struct OmxCodecBuffer &codecBuffer, OMX_BUFFERHEADERTYPE &omxBuffer)
117 {
118 omxBuffer.nOffset = codecBuffer.offset;
119 omxBuffer.nFilledLen = codecBuffer.filledLen;
120 omxBuffer.nFlags = codecBuffer.flag;
121 return HDF_SUCCESS;
122 }
123
EmptyOmxBuffer(struct OmxCodecBuffer & codecBuffer,OMX_BUFFERHEADERTYPE & omxBuffer)124 int32_t ICodecBuffer::EmptyOmxBuffer(struct OmxCodecBuffer &codecBuffer, OMX_BUFFERHEADERTYPE &omxBuffer)
125 {
126 omxBuffer.nOffset = codecBuffer.offset;
127 omxBuffer.nFilledLen = codecBuffer.filledLen;
128 omxBuffer.nFlags = codecBuffer.flag;
129 omxBuffer.nTimeStamp = codecBuffer.pts;
130 return HDF_SUCCESS;
131 }
132
EmptyOmxBufferDone(OMX_BUFFERHEADERTYPE & omxBuffer)133 int32_t ICodecBuffer::EmptyOmxBufferDone(OMX_BUFFERHEADERTYPE &omxBuffer)
134 {
135 codecBuffer_.offset = omxBuffer.nOffset;
136 codecBuffer_.filledLen = omxBuffer.nFilledLen;
137 return HDF_SUCCESS;
138 }
139
FillOmxBufferDone(OMX_BUFFERHEADERTYPE & omxBuffer)140 int32_t ICodecBuffer::FillOmxBufferDone(OMX_BUFFERHEADERTYPE &omxBuffer)
141 {
142 codecBuffer_.offset = omxBuffer.nOffset;
143 codecBuffer_.filledLen = omxBuffer.nFilledLen;
144 codecBuffer_.flag = omxBuffer.nFlags;
145 codecBuffer_.pts = omxBuffer.nTimeStamp;
146 return HDF_SUCCESS;
147 }
148 } // namespace Omx
149 } // namespace Codec
150 } // namespace OHOS