1 /* 2 * Copyright (c) 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 16 #ifndef I_CODEC_BUFFER_H 17 #define I_CODEC_BUFFER_H 18 #include <OMX_Component.h> 19 #include <OMX_Core.h> 20 #include <OMX_Types.h> 21 #include <buffer_handle.h> 22 #include <buffer_handle_utils.h> 23 #include <memory> 24 #include <refbase.h> 25 #include "codec_omx_ext.h" 26 #include "v4_0/codec_types.h" 27 #include "codec_buffer_wrapper.h" 28 29 constexpr uint32_t TIME_WAIT_MS = 10; 30 namespace OHOS { 31 namespace Codec { 32 namespace Omx { 33 struct DynamicBuffer { 34 int32_t type = 0; 35 BufferHandle *bufferHandle = nullptr; 36 }; 37 38 class ICodecBuffer : public RefBase { 39 public: 40 virtual ~ICodecBuffer(); 41 static sptr<ICodecBuffer> UseBuffer(OMX_HANDLETYPE comp, uint32_t portIndex, 42 OmxCodecBuffer &codecBuffer, OMX_BUFFERHEADERTYPE *&header, bool doCopy); 43 static sptr<ICodecBuffer> AllocateBuffer(OMX_HANDLETYPE comp, uint32_t portIndex, 44 OmxCodecBuffer &codecBuffer, OMX_BUFFERHEADERTYPE *&header); 45 virtual int32_t EmptyThisBuffer(OmxCodecBuffer &codecBuffer); 46 virtual int32_t FillThisBuffer(OmxCodecBuffer &codecBuffer); 47 int32_t EmptyBufferDone(OMX_BUFFERHEADERTYPE &omxBuffer, OmxCodecBuffer& codecBuffer); 48 virtual int32_t FillBufferDone(OMX_BUFFERHEADERTYPE &omxBuffer, OmxCodecBuffer& codecBuffer); 49 void FreeBuffer(); 50 51 protected: 52 struct InitInfo { 53 OMX_HANDLETYPE comp; 54 uint32_t portIndex; 55 OmxCodecBuffer codecBuf; 56 OMX_BUFFERHEADERTYPE* omxHeader; 57 }; ICodecBuffer(const InitInfo & info)58 ICodecBuffer(const InitInfo& info) : comp_(info.comp), portIndex_(info.portIndex), 59 codecBuffer_(info.codecBuf), omxBufHeader_(info.omxHeader) {} 60 int32_t SyncWait(int fd, uint32_t timeout); 61 protected: 62 OMX_HANDLETYPE comp_ = nullptr; 63 uint32_t portIndex_; 64 struct OmxCodecBuffer codecBuffer_; 65 OMX_BUFFERHEADERTYPE *omxBufHeader_ = nullptr; 66 }; 67 } // namespace Omx 68 } // namespace Codec 69 } // namespace OHOS 70 #endif