1 /*
2 * Copyright (c) 2023-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 #if defined(VIDEO_SUPPORT)
16
17 #define HST_LOG_TAG "CodecBuffer"
18
19 #include "codec_buffer.h"
20 #include "codec_utils.h"
21 #include "hdf_base.h"
22 #include "plugin/common/surface_memory.h"
23
24 namespace OHOS {
25 namespace Media {
26 namespace Plugin {
27 namespace CodecAdapter {
28 using namespace CodecHDI;
CodecBuffer(std::shared_ptr<Buffer> & buffer,CompVerInfo & verInfo,bool isInput,uint32_t bufferSize,MemoryType bufMemType)29 CodecBuffer::CodecBuffer(std::shared_ptr<Buffer>& buffer, CompVerInfo& verInfo,
30 bool isInput, uint32_t bufferSize, MemoryType bufMemType)
31 : buffer_(buffer), verInfo_(verInfo)
32 {
33 Init(isInput, bufferSize, bufMemType);
34 }
35
Init(bool isInput,uint32_t bufferSize,MemoryType bufMemType)36 void CodecBuffer::Init(bool isInput, uint32_t bufferSize, MemoryType bufMemType)
37 {
38 MEDIA_LOG_DD("CodecBuffer Init Start");
39 omxBuffer_ = std::make_shared<OmxCodecBuffer>();
40 omxBuffer_->size = sizeof(OmxCodecBuffer);
41 omxBuffer_->version.version.majorVersion = verInfo_.compVersion.version.majorVersion;
42 omxBuffer_->fenceFd = -1; // check use -1 first with no window
43 omxBuffer_->pts = 0;
44 omxBuffer_->flag = 0;
45 omxBuffer_->bufferType = GetOmxBufferType(bufMemType, isInput);
46 omxBuffer_->allocLen = bufferSize;
47 omxBuffer_->filledLen = 0;
48 omxBuffer_->bufferhandle = nullptr;
49 memory_ = buffer_->GetMemory();
50 FALSE_RETURN(memory_ != nullptr);
51 omxBuffer_->allocLen = memory_->GetCapacity();
52 switch (bufMemType) {
53 case MemoryType::SURFACE_BUFFER: {
54 BufferHandle* bufferHandle =
55 std::static_pointer_cast<Plugin::SurfaceMemory>(memory_)->GetSurfaceBuffer()->GetBufferHandle();
56 FALSE_LOG_MSG(bufferHandle != nullptr, "bufferHandle is null");
57 if (bufferHandle == NULL) {
58 return;
59 }
60 omxBuffer_->filledLen =
61 sizeof(BufferHandle) + (sizeof(int32_t) * (bufferHandle->reserveFds + bufferHandle->reserveInts));
62 omxBuffer_->bufferhandle = new NativeBuffer(bufferHandle);
63 break;
64 }
65 case MemoryType::SHARE_MEMORY: {
66 omxBuffer_->filledLen = sizeof(int);
67 omxBuffer_->type = isInput ? READ_ONLY_TYPE : READ_WRITE_TYPE;
68 omxBuffer_->fd = std::static_pointer_cast<Plugin::ShareMemory>(memory_)->GetShareMemoryFd();
69 MEDIA_LOG_D("share memory fd: " PUBLIC_LOG_D32,
70 std::static_pointer_cast<Plugin::ShareMemory>(memory_)->GetShareMemoryFd());
71 break;
72 }
73 default:
74 MEDIA_LOG_W("UnKnow MemoryType: " PUBLIC_LOG_D32, (int)memory_->GetMemoryType());
75 break;
76 }
77 }
78
GetOmxBuffer()79 std::shared_ptr<OmxCodecBuffer> CodecBuffer::GetOmxBuffer()
80 {
81 return omxBuffer_;
82 }
83
GetBufferId() const84 uint32_t CodecBuffer::GetBufferId() const
85 {
86 return omxBuffer_->bufferId;
87 }
88
Copy(const std::shared_ptr<Plugin::Buffer> & pluginBuffer)89 Status CodecBuffer::Copy(const std::shared_ptr<Plugin::Buffer>& pluginBuffer)
90 {
91 omxBuffer_->flag = Translate2omxFlagSet(pluginBuffer->flag);
92 omxBuffer_->pts = pluginBuffer->pts;
93 MEDIA_LOG_DD("plugin flag: " PUBLIC_LOG_U32 ", pts: " PUBLIC_LOG_D64, omxBuffer_->flag, omxBuffer_->pts);
94 if (pluginBuffer->flag & BUFFER_FLAG_EOS) {
95 MEDIA_LOG_D("EOS flag receive, return");
96 return Status::OK;
97 }
98 auto mem = pluginBuffer->GetMemory();
99 if (mem == nullptr) {
100 MEDIA_LOG_DD("pluginBuffer->GetMemory() return nullptr");
101 return Status::ERROR_INVALID_DATA;
102 }
103 const uint8_t* memAddr = mem->GetReadOnlyData();
104 if (memAddr == nullptr) {
105 MEDIA_LOG_DD("mem->GetReadOnlyData() return nullptr");
106 return Status::ERROR_INVALID_DATA;
107 }
108 memory_->Write(memAddr, mem->GetSize(), 0);
109 omxBuffer_->offset = 0;
110 omxBuffer_->filledLen = mem->GetSize();
111 MEDIA_LOG_DD("CopyBuffer end, bufferId: " PUBLIC_LOG_U32, omxBuffer_->bufferId);
112 return Status::OK;
113 }
114
Rebind(const std::shared_ptr<Plugin::Buffer> & pluginBuffer)115 Status CodecBuffer::Rebind(const std::shared_ptr<Plugin::Buffer>& pluginBuffer)
116 {
117 MEDIA_LOG_DD("Rebind Start");
118 omxBuffer_->flag = Translate2omxFlagSet(pluginBuffer->flag);
119 omxBuffer_->pts = pluginBuffer->pts;
120 MEDIA_LOG_DD("plugin flag: " PUBLIC_LOG_U32 ", pts: " PUBLIC_LOG_D64, omxBuffer_->flag, omxBuffer_->pts);
121 memory_ = pluginBuffer->GetMemory();
122 FALSE_RETURN_V_MSG_E(memory_ != nullptr, Status::ERROR_NULL_POINTER, "Call pluginBuffer->GetMemory() failed.");
123 switch (memory_->GetMemoryType()) {
124 case MemoryType::SURFACE_BUFFER: {
125 auto outMem = std::static_pointer_cast<Plugin::SurfaceMemory>(memory_);
126 auto surfaceBuf = outMem->GetSurfaceBuffer();
127 FALSE_RETURN_V_MSG_E(surfaceBuf != nullptr, Status::ERROR_NULL_POINTER, "GetSurfaceBuffer failed");
128 BufferHandle* bufferHandle = surfaceBuf->GetBufferHandle();
129 FALSE_RETURN_V_MSG_E(bufferHandle != nullptr, Status::ERROR_NULL_POINTER, "GetBufferHandle failed");
130 omxBuffer_->filledLen =
131 sizeof(BufferHandle) + sizeof(int32_t) * (bufferHandle->reserveFds + bufferHandle->reserveInts);
132 omxBuffer_->bufferhandle = new NativeBuffer(bufferHandle);
133 break;
134 }
135 case MemoryType::SHARE_MEMORY:
136 omxBuffer_->filledLen = sizeof(int);
137 omxBuffer_->fd = std::static_pointer_cast<Plugin::ShareMemory>(memory_)->GetShareMemoryFd();
138 omxBuffer_->offset = 0;
139 omxBuffer_->filledLen = 0;
140 break;
141 case MemoryType::VIRTUAL_ADDR:
142 MEDIA_LOG_E("Rebind pluginBuffer failed, MemoryType is VIRTUAL_ADDR");
143 break;
144 default:
145 MEDIA_LOG_E("MemoryType invalid!");
146 break;
147 }
148 // 这里buffer需要保存一下，为了方便往下一节点传数据，通过GetBuffer()获取
149 buffer_ = pluginBuffer;
150 MEDIA_LOG_DD("Rebind end, omxBufferId: " PUBLIC_LOG_U32, omxBuffer_->bufferId);
151 return Status::OK;
152 }
153
154 /**
155 * Receive buffer_ with data contained in CodecBuffer for transmission to the filter
156 * @param buffer Copy the value of buffer_
157 * @param omxBuffer
158 * @return
159 */
Unbind(std::shared_ptr<Plugin::Buffer> & buffer,const OmxCodecBuffer & omxBuffer)160 Status CodecBuffer::Unbind(std::shared_ptr<Plugin::Buffer>& buffer, const OmxCodecBuffer& omxBuffer)
161 {
162 // 因为Rebind()里面用buffer_保存了PluginBuf，所以这里的buffer_需要主动释放，减少智能指针的引用计数
163 // decoder 时，PluginBuf 的真正释放时机应该是在sink节点，该数据送显后才能释放
164 FALSE_RETURN_V_MSG_E(memory_ != nullptr, Status::ERROR_NULL_POINTER, "Param memory_ is nullptr");
165 if (memory_->GetMemoryType() == MemoryType::SHARE_MEMORY) {
166 memory_->UpdateDataSize(static_cast<size_t>(omxBuffer.filledLen - omxBuffer.offset), 0);
167 }
168 buffer = buffer_;
169 buffer->flag = Translate2PluginFlagSet(omxBuffer.flag);
170 buffer->pts = omxBuffer.pts;
171 buffer_ = nullptr;
172 return Status::OK;
173 }
174 } // namespace CodecAdapter
175 } // namespace Plugin
176 } // namespace Media
177 } // namespace OHOS
178 #endif