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 CODEC_HANDLE_BUFFER_H
17 #define CODEC_HANDLE_BUFFER_H
18 #include <buffer_handle.h>
19 #include "icodec_buffer.h"
20 
21 namespace OHOS {
22 namespace Codec {
23 namespace Omx {
24 class CodecHandleBuffer : ICodecBuffer {
25 public:
26     ~CodecHandleBuffer();
27     sptr<ICodecBuffer> static Create(struct OmxCodecBuffer &codecBuffer);
28     int32_t FillOmxBuffer(struct OmxCodecBuffer &codecBuffer, OMX_BUFFERHEADERTYPE &omxBuffer) override;
29     int32_t EmptyOmxBuffer(struct OmxCodecBuffer &codecBuffer, OMX_BUFFERHEADERTYPE &omxBuffer) override;
30     int32_t FreeBuffer(struct OmxCodecBuffer &codecBuffer) override;
31     int32_t EmptyOmxBufferDone(OMX_BUFFERHEADERTYPE &omxBuffer) override;
32     int32_t FillOmxBufferDone(OMX_BUFFERHEADERTYPE &omxBuffer) override;
33     uint8_t *GetBuffer() override;
34 
35 protected:
36     CodecHandleBuffer(struct OmxCodecBuffer &codecBuffer, BufferHandle *bufferHandle);
37     bool CheckInvalid(struct OmxCodecBuffer &codecBuffer) override;
38 
39 private:
40     void ResetBuffer(struct OmxCodecBuffer &codecBuffer, OMX_BUFFERHEADERTYPE &omxBuffer);
41 
42 private:
43     BufferHandle *bufferHandle_;
44 };
45 }  // namespace Omx
46 }  // namespace Codec
47 }  // namespace OHOS
48 #endif  // CODEC_HANDLE_BUFFER_H