xref: /aosp_15_r20/frameworks/av/media/libmedia/OMXBuffer.cpp (revision ec779b8e0859a360c3d303172224686826e6e0e1)
1*ec779b8eSAndroid Build Coastguard Worker /*
2*ec779b8eSAndroid Build Coastguard Worker  * Copyright 2016, The Android Open Source Project
3*ec779b8eSAndroid Build Coastguard Worker  *
4*ec779b8eSAndroid Build Coastguard Worker  * Licensed under the Apache License, Version 2.0 (the "License");
5*ec779b8eSAndroid Build Coastguard Worker  * you may not use this file except in compliance with the License.
6*ec779b8eSAndroid Build Coastguard Worker  * You may obtain a copy of the License at
7*ec779b8eSAndroid Build Coastguard Worker  *
8*ec779b8eSAndroid Build Coastguard Worker  *      http://www.apache.org/licenses/LICENSE-2.0
9*ec779b8eSAndroid Build Coastguard Worker  *
10*ec779b8eSAndroid Build Coastguard Worker  * Unless required by applicable law or agreed to in writing, software
11*ec779b8eSAndroid Build Coastguard Worker  * distributed under the License is distributed on an "AS IS" BASIS,
12*ec779b8eSAndroid Build Coastguard Worker  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13*ec779b8eSAndroid Build Coastguard Worker  * See the License for the specific language governing permissions and
14*ec779b8eSAndroid Build Coastguard Worker  * limitations under the License.
15*ec779b8eSAndroid Build Coastguard Worker  */
16*ec779b8eSAndroid Build Coastguard Worker 
17*ec779b8eSAndroid Build Coastguard Worker //#define LOG_NDEBUG 0
18*ec779b8eSAndroid Build Coastguard Worker #define LOG_TAG "OMXBuffer"
19*ec779b8eSAndroid Build Coastguard Worker 
20*ec779b8eSAndroid Build Coastguard Worker #include <media/stagefright/foundation/ADebug.h>
21*ec779b8eSAndroid Build Coastguard Worker #include <media/MediaCodecBuffer.h>
22*ec779b8eSAndroid Build Coastguard Worker #include <media/OMXBuffer.h>
23*ec779b8eSAndroid Build Coastguard Worker #include <binder/IMemory.h>
24*ec779b8eSAndroid Build Coastguard Worker #include <binder/Parcel.h>
25*ec779b8eSAndroid Build Coastguard Worker #include <ui/GraphicBuffer.h>
26*ec779b8eSAndroid Build Coastguard Worker #include <utils/NativeHandle.h>
27*ec779b8eSAndroid Build Coastguard Worker 
28*ec779b8eSAndroid Build Coastguard Worker namespace android {
29*ec779b8eSAndroid Build Coastguard Worker 
30*ec779b8eSAndroid Build Coastguard Worker //static
31*ec779b8eSAndroid Build Coastguard Worker OMXBuffer OMXBuffer::sPreset(static_cast<sp<MediaCodecBuffer> >(NULL));
32*ec779b8eSAndroid Build Coastguard Worker 
OMXBuffer()33*ec779b8eSAndroid Build Coastguard Worker OMXBuffer::OMXBuffer()
34*ec779b8eSAndroid Build Coastguard Worker     : mBufferType(kBufferTypeInvalid) {
35*ec779b8eSAndroid Build Coastguard Worker }
36*ec779b8eSAndroid Build Coastguard Worker 
OMXBuffer(const sp<MediaCodecBuffer> & codecBuffer)37*ec779b8eSAndroid Build Coastguard Worker OMXBuffer::OMXBuffer(const sp<MediaCodecBuffer>& codecBuffer)
38*ec779b8eSAndroid Build Coastguard Worker     : mBufferType(kBufferTypePreset),
39*ec779b8eSAndroid Build Coastguard Worker       mRangeOffset(codecBuffer != NULL ? codecBuffer->offset() : 0),
40*ec779b8eSAndroid Build Coastguard Worker       mRangeLength(codecBuffer != NULL ? codecBuffer->size() : 0) {
41*ec779b8eSAndroid Build Coastguard Worker }
42*ec779b8eSAndroid Build Coastguard Worker 
OMXBuffer(OMX_U32 rangeOffset,OMX_U32 rangeLength)43*ec779b8eSAndroid Build Coastguard Worker OMXBuffer::OMXBuffer(OMX_U32 rangeOffset, OMX_U32 rangeLength)
44*ec779b8eSAndroid Build Coastguard Worker     : mBufferType(kBufferTypePreset),
45*ec779b8eSAndroid Build Coastguard Worker       mRangeOffset(rangeOffset),
46*ec779b8eSAndroid Build Coastguard Worker       mRangeLength(rangeLength) {
47*ec779b8eSAndroid Build Coastguard Worker }
48*ec779b8eSAndroid Build Coastguard Worker 
OMXBuffer(const sp<IMemory> & mem)49*ec779b8eSAndroid Build Coastguard Worker OMXBuffer::OMXBuffer(const sp<IMemory> &mem)
50*ec779b8eSAndroid Build Coastguard Worker     : mBufferType(kBufferTypeSharedMem),
51*ec779b8eSAndroid Build Coastguard Worker       mMem(mem) {
52*ec779b8eSAndroid Build Coastguard Worker }
53*ec779b8eSAndroid Build Coastguard Worker 
OMXBuffer(const sp<GraphicBuffer> & gbuf)54*ec779b8eSAndroid Build Coastguard Worker OMXBuffer::OMXBuffer(const sp<GraphicBuffer> &gbuf)
55*ec779b8eSAndroid Build Coastguard Worker     : mBufferType(kBufferTypeANWBuffer),
56*ec779b8eSAndroid Build Coastguard Worker       mGraphicBuffer(gbuf) {
57*ec779b8eSAndroid Build Coastguard Worker }
58*ec779b8eSAndroid Build Coastguard Worker 
OMXBuffer(const sp<NativeHandle> & handle)59*ec779b8eSAndroid Build Coastguard Worker OMXBuffer::OMXBuffer(const sp<NativeHandle> &handle)
60*ec779b8eSAndroid Build Coastguard Worker     : mBufferType(kBufferTypeNativeHandle),
61*ec779b8eSAndroid Build Coastguard Worker       mNativeHandle(handle) {
62*ec779b8eSAndroid Build Coastguard Worker }
63*ec779b8eSAndroid Build Coastguard Worker 
OMXBuffer(const hidl_memory & hidlMemory)64*ec779b8eSAndroid Build Coastguard Worker OMXBuffer::OMXBuffer(const hidl_memory &hidlMemory)
65*ec779b8eSAndroid Build Coastguard Worker     : mBufferType(kBufferTypeHidlMemory),
66*ec779b8eSAndroid Build Coastguard Worker       mHidlMemory(hidlMemory) {
67*ec779b8eSAndroid Build Coastguard Worker }
68*ec779b8eSAndroid Build Coastguard Worker 
~OMXBuffer()69*ec779b8eSAndroid Build Coastguard Worker OMXBuffer::~OMXBuffer() {
70*ec779b8eSAndroid Build Coastguard Worker }
71*ec779b8eSAndroid Build Coastguard Worker 
writeToParcel(Parcel * parcel) const72*ec779b8eSAndroid Build Coastguard Worker status_t OMXBuffer::writeToParcel(Parcel *parcel) const {
73*ec779b8eSAndroid Build Coastguard Worker     CHECK(mBufferType != kBufferTypeHidlMemory);
74*ec779b8eSAndroid Build Coastguard Worker     parcel->writeInt32(mBufferType);
75*ec779b8eSAndroid Build Coastguard Worker 
76*ec779b8eSAndroid Build Coastguard Worker     switch(mBufferType) {
77*ec779b8eSAndroid Build Coastguard Worker         case kBufferTypePreset:
78*ec779b8eSAndroid Build Coastguard Worker         {
79*ec779b8eSAndroid Build Coastguard Worker             status_t err = parcel->writeUint32(mRangeOffset);
80*ec779b8eSAndroid Build Coastguard Worker             if (err != OK) {
81*ec779b8eSAndroid Build Coastguard Worker                 return err;
82*ec779b8eSAndroid Build Coastguard Worker             }
83*ec779b8eSAndroid Build Coastguard Worker             return parcel->writeUint32(mRangeLength);
84*ec779b8eSAndroid Build Coastguard Worker         }
85*ec779b8eSAndroid Build Coastguard Worker 
86*ec779b8eSAndroid Build Coastguard Worker         case kBufferTypeSharedMem:
87*ec779b8eSAndroid Build Coastguard Worker         {
88*ec779b8eSAndroid Build Coastguard Worker             return parcel->writeStrongBinder(IInterface::asBinder(mMem));
89*ec779b8eSAndroid Build Coastguard Worker         }
90*ec779b8eSAndroid Build Coastguard Worker 
91*ec779b8eSAndroid Build Coastguard Worker         case kBufferTypeANWBuffer:
92*ec779b8eSAndroid Build Coastguard Worker         {
93*ec779b8eSAndroid Build Coastguard Worker             if (mGraphicBuffer == NULL) {
94*ec779b8eSAndroid Build Coastguard Worker                 return parcel->writeBool(false);
95*ec779b8eSAndroid Build Coastguard Worker             }
96*ec779b8eSAndroid Build Coastguard Worker             status_t err = parcel->writeBool(true);
97*ec779b8eSAndroid Build Coastguard Worker             if (err != OK) {
98*ec779b8eSAndroid Build Coastguard Worker                 return err;
99*ec779b8eSAndroid Build Coastguard Worker             }
100*ec779b8eSAndroid Build Coastguard Worker             return parcel->write(*mGraphicBuffer);
101*ec779b8eSAndroid Build Coastguard Worker         }
102*ec779b8eSAndroid Build Coastguard Worker 
103*ec779b8eSAndroid Build Coastguard Worker         case kBufferTypeNativeHandle:
104*ec779b8eSAndroid Build Coastguard Worker         {
105*ec779b8eSAndroid Build Coastguard Worker             return parcel->writeNativeHandle(mNativeHandle->handle());
106*ec779b8eSAndroid Build Coastguard Worker         }
107*ec779b8eSAndroid Build Coastguard Worker 
108*ec779b8eSAndroid Build Coastguard Worker         default:
109*ec779b8eSAndroid Build Coastguard Worker             return BAD_VALUE;
110*ec779b8eSAndroid Build Coastguard Worker     }
111*ec779b8eSAndroid Build Coastguard Worker     return BAD_VALUE;
112*ec779b8eSAndroid Build Coastguard Worker }
113*ec779b8eSAndroid Build Coastguard Worker 
readFromParcel(const Parcel * parcel)114*ec779b8eSAndroid Build Coastguard Worker status_t OMXBuffer::readFromParcel(const Parcel *parcel) {
115*ec779b8eSAndroid Build Coastguard Worker     BufferType bufferType = (BufferType) parcel->readInt32();
116*ec779b8eSAndroid Build Coastguard Worker     CHECK(bufferType != kBufferTypeHidlMemory);
117*ec779b8eSAndroid Build Coastguard Worker 
118*ec779b8eSAndroid Build Coastguard Worker     switch(bufferType) {
119*ec779b8eSAndroid Build Coastguard Worker         case kBufferTypePreset:
120*ec779b8eSAndroid Build Coastguard Worker         {
121*ec779b8eSAndroid Build Coastguard Worker             status_t err = parcel->readUint32(&mRangeOffset);
122*ec779b8eSAndroid Build Coastguard Worker             if (err != OK) {
123*ec779b8eSAndroid Build Coastguard Worker                 return err;
124*ec779b8eSAndroid Build Coastguard Worker             }
125*ec779b8eSAndroid Build Coastguard Worker             err = parcel->readUint32(&mRangeLength);
126*ec779b8eSAndroid Build Coastguard Worker             if (err != OK) {
127*ec779b8eSAndroid Build Coastguard Worker                 return err;
128*ec779b8eSAndroid Build Coastguard Worker             }
129*ec779b8eSAndroid Build Coastguard Worker             break;
130*ec779b8eSAndroid Build Coastguard Worker         }
131*ec779b8eSAndroid Build Coastguard Worker 
132*ec779b8eSAndroid Build Coastguard Worker         case kBufferTypeSharedMem:
133*ec779b8eSAndroid Build Coastguard Worker         {
134*ec779b8eSAndroid Build Coastguard Worker             mMem = interface_cast<IMemory>(parcel->readStrongBinder());
135*ec779b8eSAndroid Build Coastguard Worker             break;
136*ec779b8eSAndroid Build Coastguard Worker         }
137*ec779b8eSAndroid Build Coastguard Worker 
138*ec779b8eSAndroid Build Coastguard Worker         case kBufferTypeANWBuffer:
139*ec779b8eSAndroid Build Coastguard Worker         {
140*ec779b8eSAndroid Build Coastguard Worker             bool notNull;
141*ec779b8eSAndroid Build Coastguard Worker             status_t err = parcel->readBool(&notNull);
142*ec779b8eSAndroid Build Coastguard Worker             if (err != OK) {
143*ec779b8eSAndroid Build Coastguard Worker                 return err;
144*ec779b8eSAndroid Build Coastguard Worker             }
145*ec779b8eSAndroid Build Coastguard Worker             if (notNull) {
146*ec779b8eSAndroid Build Coastguard Worker                 sp<GraphicBuffer> buffer = new GraphicBuffer();
147*ec779b8eSAndroid Build Coastguard Worker                 status_t err = parcel->read(*buffer);
148*ec779b8eSAndroid Build Coastguard Worker                 if (err != OK) {
149*ec779b8eSAndroid Build Coastguard Worker                     return err;
150*ec779b8eSAndroid Build Coastguard Worker                 }
151*ec779b8eSAndroid Build Coastguard Worker                 mGraphicBuffer = buffer;
152*ec779b8eSAndroid Build Coastguard Worker             } else {
153*ec779b8eSAndroid Build Coastguard Worker                 mGraphicBuffer = nullptr;
154*ec779b8eSAndroid Build Coastguard Worker             }
155*ec779b8eSAndroid Build Coastguard Worker             break;
156*ec779b8eSAndroid Build Coastguard Worker         }
157*ec779b8eSAndroid Build Coastguard Worker 
158*ec779b8eSAndroid Build Coastguard Worker         case kBufferTypeNativeHandle:
159*ec779b8eSAndroid Build Coastguard Worker         {
160*ec779b8eSAndroid Build Coastguard Worker             sp<NativeHandle> handle = NativeHandle::create(
161*ec779b8eSAndroid Build Coastguard Worker                     parcel->readNativeHandle(), true /* ownsHandle */);
162*ec779b8eSAndroid Build Coastguard Worker 
163*ec779b8eSAndroid Build Coastguard Worker             mNativeHandle = handle;
164*ec779b8eSAndroid Build Coastguard Worker             break;
165*ec779b8eSAndroid Build Coastguard Worker         }
166*ec779b8eSAndroid Build Coastguard Worker 
167*ec779b8eSAndroid Build Coastguard Worker         default:
168*ec779b8eSAndroid Build Coastguard Worker             return BAD_VALUE;
169*ec779b8eSAndroid Build Coastguard Worker     }
170*ec779b8eSAndroid Build Coastguard Worker 
171*ec779b8eSAndroid Build Coastguard Worker     mBufferType = bufferType;
172*ec779b8eSAndroid Build Coastguard Worker     return OK;
173*ec779b8eSAndroid Build Coastguard Worker }
174*ec779b8eSAndroid Build Coastguard Worker 
operator =(OMXBuffer && source)175*ec779b8eSAndroid Build Coastguard Worker OMXBuffer& OMXBuffer::operator=(OMXBuffer&& source) noexcept {
176*ec779b8eSAndroid Build Coastguard Worker     mBufferType = std::move(source.mBufferType);
177*ec779b8eSAndroid Build Coastguard Worker     mRangeOffset = std::move(source.mRangeOffset);
178*ec779b8eSAndroid Build Coastguard Worker     mRangeLength = std::move(source.mRangeLength);
179*ec779b8eSAndroid Build Coastguard Worker     mMem = std::move(source.mMem);
180*ec779b8eSAndroid Build Coastguard Worker     mGraphicBuffer = std::move(source.mGraphicBuffer);
181*ec779b8eSAndroid Build Coastguard Worker     mNativeHandle = std::move(source.mNativeHandle);
182*ec779b8eSAndroid Build Coastguard Worker     mHidlMemory = std::move(source.mHidlMemory);
183*ec779b8eSAndroid Build Coastguard Worker     return *this;
184*ec779b8eSAndroid Build Coastguard Worker }
185*ec779b8eSAndroid Build Coastguard Worker 
186*ec779b8eSAndroid Build Coastguard Worker } // namespace android
187*ec779b8eSAndroid Build Coastguard Worker 
188*ec779b8eSAndroid Build Coastguard Worker 
189*ec779b8eSAndroid Build Coastguard Worker 
190*ec779b8eSAndroid Build Coastguard Worker 
191