1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
|
/*
* Copyright 2018, The Android Open Source Project
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#ifndef C2_OMX_NODE_H_
#define C2_OMX_NODE_H_
#include <atomic>
#include <android/IOMXBufferSource.h>
#include <codec2/hidl/client.h>
#include <media/stagefright/foundation/Mutexed.h>
#include <media/IOMX.h>
#include <media/OMXBuffer.h>
namespace android {
/**
* IOmxNode implementation around codec 2.0 component, only to be used in
* IGraphicBufferSource::configure. Only subset of IOmxNode API is implemented
* and others are left as stub. As a result, one cannot expect this IOmxNode
* to work in any other usage than IGraphicBufferSource.
*/
struct C2OMXNode : public BnOMXNode {
explicit C2OMXNode(const std::shared_ptr<Codec2Client::Component> &comp);
~C2OMXNode() override = default;
// IOMXNode
status_t freeNode() override;
status_t sendCommand(OMX_COMMANDTYPE cmd, OMX_S32 param) override;
status_t getParameter(
OMX_INDEXTYPE index, void *params, size_t size) override;
status_t setParameter(
OMX_INDEXTYPE index, const void *params, size_t size) override;
status_t getConfig(
OMX_INDEXTYPE index, void *params, size_t size) override;
status_t setConfig(
OMX_INDEXTYPE index, const void *params, size_t size) override;
status_t setPortMode(OMX_U32 port_index, IOMX::PortMode mode) override;
status_t prepareForAdaptivePlayback(
OMX_U32 portIndex, OMX_BOOL enable,
OMX_U32 maxFrameWidth, OMX_U32 maxFrameHeight) override;
status_t configureVideoTunnelMode(
OMX_U32 portIndex, OMX_BOOL tunneled,
OMX_U32 audioHwSync, native_handle_t **sidebandHandle) override;
status_t getGraphicBufferUsage(
OMX_U32 port_index, OMX_U32* usage) override;
status_t setInputSurface(
const sp<IOMXBufferSource> &bufferSource) override;
status_t allocateSecureBuffer(
OMX_U32 port_index, size_t size, buffer_id *buffer,
void **buffer_data, sp<NativeHandle> *native_handle) override;
status_t useBuffer(
OMX_U32 port_index, const OMXBuffer &omxBuf, buffer_id *buffer) override;
status_t freeBuffer(
OMX_U32 port_index, buffer_id buffer) override;
status_t fillBuffer(
buffer_id buffer, const OMXBuffer &omxBuf, int fenceFd) override;
status_t emptyBuffer(
buffer_id buffer, const OMXBuffer &omxBuf,
OMX_U32 flags, OMX_TICKS timestamp, int fenceFd) override;
status_t getExtensionIndex(
const char *parameter_name,
OMX_INDEXTYPE *index) override;
status_t dispatchMessage(const omx_message &msg) override;
/**
* Returns underlying IOMXBufferSource object.
*/
sp<IOMXBufferSource> getSource();
/**
* Configure the frame size.
*/
void setFrameSize(uint32_t width, uint32_t height);
/**
* Clean up work item reference.
*
* \param index input work index
*/
void onInputBufferDone(c2_cntr64_t index);
private:
std::weak_ptr<Codec2Client::Component> mComp;
sp<IOMXBufferSource> mBufferSource;
std::shared_ptr<C2Allocator> mAllocator;
std::atomic_uint64_t mFrameIndex;
uint32_t mWidth;
uint32_t mHeight;
uint64_t mUsage;
// WORKAROUND: timestamp adjustment
// if >0: this is the max timestamp gap, if <0: this is -1 times the fixed timestamp gap
// if 0: no timestamp adjustment is made
// note that C2OMXNode can be recycled between encoding sessions.
int32_t mAdjustTimestampGapUs;
bool mFirstInputFrame; // true for first input
c2_cntr64_t mPrevInputTimestamp; // input timestamp for previous frame
c2_cntr64_t mPrevCodecTimestamp; // adjusted (codec) timestamp for previous frame
Mutexed<std::map<uint64_t, buffer_id>> mBufferIdsInUse;
class QueueThread;
sp<QueueThread> mQueueThread;
};
} // namespace android
#endif // C2_OMX_NODE_H_
|
