1 /*
2  * Copyright (c) 2022-2023 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_component_type.h"
26 #include "codec_omx_ext.h"
27 constexpr uint32_t TIME_WAIT_MS = 10;
28 namespace OHOS {
29 namespace Codec {
30 namespace Omx {
31 struct DynamicBuffer {
32     int32_t type;
33     BufferHandle *bufferHandle;
34 
DynamicBufferDynamicBuffer35     DynamicBuffer() : type(0), bufferHandle(nullptr)
36     {}
~DynamicBufferDynamicBuffer37     ~DynamicBuffer()
38     {
39         type = 0;
40         if (bufferHandle != nullptr) {
41             FreeBufferHandle(bufferHandle);
42         }
43         bufferHandle = nullptr;
44     }
45 };
46 
47 class ICodecBuffer : public RefBase {
48 public:
49     explicit ICodecBuffer(struct OmxCodecBuffer &codecBuffer);
50     virtual ~ICodecBuffer() override;
51     static sptr<ICodecBuffer> CreateCodeBuffer(struct OmxCodecBuffer &codecBuffer);
52     static sptr<ICodecBuffer> AllocateCodecBuffer(struct OmxCodecBuffer &codecBuffer);
53     virtual int32_t FillOmxBuffer(struct OmxCodecBuffer &codecBuffer, OMX_BUFFERHEADERTYPE &omxBuffer);
54     virtual int32_t EmptyOmxBuffer(struct OmxCodecBuffer &codecBuffer, OMX_BUFFERHEADERTYPE &omxBuffer);
55     virtual int32_t FreeBuffer(struct OmxCodecBuffer &codecBuffer) = 0;
56     virtual int32_t EmptyOmxBufferDone(OMX_BUFFERHEADERTYPE &omxBuffer);
57     virtual int32_t FillOmxBufferDone(OMX_BUFFERHEADERTYPE &omxBuffer);
58     virtual uint8_t *GetBuffer() = 0;
59     struct OmxCodecBuffer &GetCodecBuffer();
60     void SetBufferId(int32_t bufferId);
61 
62 protected:
ICodecBuffer()63     ICodecBuffer()
64     {}
65     virtual bool CheckInvalid(struct OmxCodecBuffer &codecBuffer);
66     int32_t SyncWait(int fd, uint32_t timeout);
67 protected:
68     struct OmxCodecBuffer codecBuffer_;
69 };
70 }  // namespace Omx
71 }  // namespace Codec
72 }  // namespace OHOS
73 #endif