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 "v3_0/codec_types.h" 27 using OHOS::HDI::Codec::V3_0::OmxCodecBuffer; 28 constexpr uint32_t TIME_WAIT_MS = 10; 29 namespace OHOS { 30 namespace Codec { 31 namespace Omx { 32 struct DynamicBuffer { 33 int32_t type = 0; 34 BufferHandle *bufferHandle = nullptr; 35 }; 36 37 class ICodecBuffer : public RefBase { 38 public: 39 ICodecBuffer(struct OmxCodecBuffer &codecBuffer); 40 virtual ~ICodecBuffer(); 41 sptr<ICodecBuffer> static CreateCodeBuffer(struct OmxCodecBuffer &codecBuffer); 42 sptr<ICodecBuffer> static AllocateCodecBuffer(struct OmxCodecBuffer &codecBuffer, OMX_BUFFERHEADERTYPE &omxBuffer); 43 virtual int32_t FillOmxBuffer(struct OmxCodecBuffer &codecBuffer, OMX_BUFFERHEADERTYPE &omxBuffer); 44 virtual int32_t EmptyOmxBuffer(struct OmxCodecBuffer &codecBuffer, OMX_BUFFERHEADERTYPE &omxBuffer); 45 virtual int32_t FreeBuffer(struct OmxCodecBuffer &codecBuffer); 46 virtual int32_t EmptyOmxBufferDone(OMX_BUFFERHEADERTYPE &omxBuffer); 47 virtual int32_t FillOmxBufferDone(OMX_BUFFERHEADERTYPE &omxBuffer); 48 virtual uint8_t *GetBuffer(); 49 struct OmxCodecBuffer &GetCodecBuffer(); 50 void SetBufferId(int32_t bufferId); 51 52 protected: ICodecBuffer()53 ICodecBuffer() 54 {} 55 virtual bool CheckInvalid(struct OmxCodecBuffer &codecBuffer); 56 int32_t SyncWait(int fd, uint32_t timeout); 57 protected: 58 struct OmxCodecBuffer codecBuffer_; 59 }; 60 } // namespace Omx 61 } // namespace Codec 62 } // namespace OHOS 63 #endif