OMX.cpp revision 81dbf71f4488ee1a4cb612f2aba341f365fb546a
1/* 2 * Copyright (C) 2009 The Android Open Source Project 3 * 4 * Licensed under the Apache License, Version 2.0 (the "License"); 5 * you may not use this file except in compliance with the License. 6 * You may obtain a copy of the License at 7 * 8 * http://www.apache.org/licenses/LICENSE-2.0 9 * 10 * Unless required by applicable law or agreed to in writing, software 11 * distributed under the License is distributed on an "AS IS" BASIS, 12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13 * See the License for the specific language governing permissions and 14 * limitations under the License. 15 */ 16 17//#define LOG_NDEBUG 0 18#define LOG_TAG "OMX" 19#include <utils/Log.h> 20 21#include <dlfcn.h> 22 23#include "../include/OMX.h" 24#include "OMXRenderer.h" 25 26#include "../include/OMXNodeInstance.h" 27#include "../include/SoftwareRenderer.h" 28 29#include <binder/IMemory.h> 30#include <media/stagefright/MediaDebug.h> 31#include <media/stagefright/VideoRenderer.h> 32 33#include "OMXMaster.h" 34 35#include <OMX_Component.h> 36 37namespace android { 38 39//////////////////////////////////////////////////////////////////////////////// 40 41struct OMX::CallbackDispatcher : public RefBase { 42 CallbackDispatcher(OMX *owner); 43 44 void post(const omx_message &msg); 45 46protected: 47 virtual ~CallbackDispatcher(); 48 49private: 50 Mutex mLock; 51 52 OMX *mOwner; 53 bool mDone; 54 Condition mQueueChanged; 55 List<omx_message> mQueue; 56 57 pthread_t mThread; 58 59 void dispatch(const omx_message &msg); 60 61 static void *ThreadWrapper(void *me); 62 void threadEntry(); 63 64 CallbackDispatcher(const CallbackDispatcher &); 65 CallbackDispatcher &operator=(const CallbackDispatcher &); 66}; 67 68OMX::CallbackDispatcher::CallbackDispatcher(OMX *owner) 69 : mOwner(owner), 70 mDone(false) { 71 pthread_attr_t attr; 72 pthread_attr_init(&attr); 73 pthread_attr_setdetachstate(&attr, PTHREAD_CREATE_JOINABLE); 74 75 pthread_create(&mThread, &attr, ThreadWrapper, this); 76 77 pthread_attr_destroy(&attr); 78} 79 80OMX::CallbackDispatcher::~CallbackDispatcher() { 81 { 82 Mutex::Autolock autoLock(mLock); 83 84 mDone = true; 85 mQueueChanged.signal(); 86 } 87 88 void *dummy; 89 pthread_join(mThread, &dummy); 90} 91 92void OMX::CallbackDispatcher::post(const omx_message &msg) { 93 Mutex::Autolock autoLock(mLock); 94 mQueue.push_back(msg); 95 mQueueChanged.signal(); 96} 97 98void OMX::CallbackDispatcher::dispatch(const omx_message &msg) { 99 OMXNodeInstance *instance = mOwner->findInstance(msg.node); 100 if (instance == NULL) { 101 LOGV("Would have dispatched a message to a node that's already gone."); 102 return; 103 } 104 instance->onMessage(msg); 105} 106 107// static 108void *OMX::CallbackDispatcher::ThreadWrapper(void *me) { 109 static_cast<CallbackDispatcher *>(me)->threadEntry(); 110 111 return NULL; 112} 113 114void OMX::CallbackDispatcher::threadEntry() { 115 for (;;) { 116 omx_message msg; 117 118 { 119 Mutex::Autolock autoLock(mLock); 120 while (!mDone && mQueue.empty()) { 121 mQueueChanged.wait(mLock); 122 } 123 124 if (mDone) { 125 break; 126 } 127 128 msg = *mQueue.begin(); 129 mQueue.erase(mQueue.begin()); 130 } 131 132 dispatch(msg); 133 } 134} 135 136//////////////////////////////////////////////////////////////////////////////// 137 138OMX::OMX() 139 : mMaster(new OMXMaster), 140 mDispatcher(new CallbackDispatcher(this)), 141 mNodeCounter(0) { 142} 143 144OMX::~OMX() { 145 delete mMaster; 146 mMaster = NULL; 147} 148 149void OMX::binderDied(const wp<IBinder> &the_late_who) { 150 OMXNodeInstance *instance; 151 152 { 153 Mutex::Autolock autoLock(mLock); 154 155 ssize_t index = mLiveNodes.indexOfKey(the_late_who); 156 CHECK(index >= 0); 157 158 instance = mLiveNodes.editValueAt(index); 159 mLiveNodes.removeItemsAt(index); 160 161 invalidateNodeID_l(instance->nodeID()); 162 } 163 164 instance->onObserverDied(mMaster); 165} 166 167status_t OMX::listNodes(List<ComponentInfo> *list) { 168 list->clear(); 169 170 OMX_U32 index = 0; 171 char componentName[256]; 172 while (mMaster->enumerateComponents( 173 componentName, sizeof(componentName), index) == OMX_ErrorNone) { 174 list->push_back(ComponentInfo()); 175 ComponentInfo &info = *--list->end(); 176 177 info.mName = componentName; 178 179 Vector<String8> roles; 180 OMX_ERRORTYPE err = 181 mMaster->getRolesOfComponent(componentName, &roles); 182 183 if (err == OMX_ErrorNone) { 184 for (OMX_U32 i = 0; i < roles.size(); ++i) { 185 info.mRoles.push_back(roles[i]); 186 } 187 } 188 189 ++index; 190 } 191 192 return OK; 193} 194 195status_t OMX::allocateNode( 196 const char *name, const sp<IOMXObserver> &observer, node_id *node) { 197 Mutex::Autolock autoLock(mLock); 198 199 *node = 0; 200 201 OMXNodeInstance *instance = new OMXNodeInstance(this, observer); 202 203 OMX_COMPONENTTYPE *handle; 204 OMX_ERRORTYPE err = mMaster->makeComponentInstance( 205 name, &OMXNodeInstance::kCallbacks, 206 instance, &handle); 207 208 if (err != OMX_ErrorNone) { 209 LOGV("FAILED to allocate omx component '%s'", name); 210 211 instance->onGetHandleFailed(); 212 213 return UNKNOWN_ERROR; 214 } 215 216 *node = makeNodeID(instance); 217 218 instance->setHandle(*node, handle); 219 220 mLiveNodes.add(observer->asBinder(), instance); 221 observer->asBinder()->linkToDeath(this); 222 223 return OK; 224} 225 226status_t OMX::freeNode(node_id node) { 227 OMXNodeInstance *instance = findInstance(node); 228 229 ssize_t index = mLiveNodes.indexOfKey(instance->observer()->asBinder()); 230 CHECK(index >= 0); 231 mLiveNodes.removeItemsAt(index); 232 instance->observer()->asBinder()->unlinkToDeath(this); 233 234 return instance->freeNode(mMaster); 235} 236 237status_t OMX::sendCommand( 238 node_id node, OMX_COMMANDTYPE cmd, OMX_S32 param) { 239 return findInstance(node)->sendCommand(cmd, param); 240} 241 242status_t OMX::getParameter( 243 node_id node, OMX_INDEXTYPE index, 244 void *params, size_t size) { 245 return findInstance(node)->getParameter( 246 index, params, size); 247} 248 249status_t OMX::setParameter( 250 node_id node, OMX_INDEXTYPE index, 251 const void *params, size_t size) { 252 return findInstance(node)->setParameter( 253 index, params, size); 254} 255 256status_t OMX::getConfig( 257 node_id node, OMX_INDEXTYPE index, 258 void *params, size_t size) { 259 return findInstance(node)->getConfig( 260 index, params, size); 261} 262 263status_t OMX::setConfig( 264 node_id node, OMX_INDEXTYPE index, 265 const void *params, size_t size) { 266 return findInstance(node)->setConfig( 267 index, params, size); 268} 269 270status_t OMX::useBuffer( 271 node_id node, OMX_U32 port_index, const sp<IMemory> ¶ms, 272 buffer_id *buffer) { 273 return findInstance(node)->useBuffer( 274 port_index, params, buffer); 275} 276 277status_t OMX::allocateBuffer( 278 node_id node, OMX_U32 port_index, size_t size, 279 buffer_id *buffer) { 280 return findInstance(node)->allocateBuffer( 281 port_index, size, buffer); 282} 283 284status_t OMX::allocateBufferWithBackup( 285 node_id node, OMX_U32 port_index, const sp<IMemory> ¶ms, 286 buffer_id *buffer) { 287 return findInstance(node)->allocateBufferWithBackup( 288 port_index, params, buffer); 289} 290 291status_t OMX::freeBuffer(node_id node, OMX_U32 port_index, buffer_id buffer) { 292 return findInstance(node)->freeBuffer( 293 port_index, buffer); 294} 295 296status_t OMX::fillBuffer(node_id node, buffer_id buffer) { 297 return findInstance(node)->fillBuffer(buffer); 298} 299 300status_t OMX::emptyBuffer( 301 node_id node, 302 buffer_id buffer, 303 OMX_U32 range_offset, OMX_U32 range_length, 304 OMX_U32 flags, OMX_TICKS timestamp) { 305 return findInstance(node)->emptyBuffer( 306 buffer, range_offset, range_length, flags, timestamp); 307} 308 309status_t OMX::getExtensionIndex( 310 node_id node, 311 const char *parameter_name, 312 OMX_INDEXTYPE *index) { 313 return findInstance(node)->getExtensionIndex( 314 parameter_name, index); 315} 316 317OMX_ERRORTYPE OMX::OnEvent( 318 node_id node, 319 OMX_IN OMX_EVENTTYPE eEvent, 320 OMX_IN OMX_U32 nData1, 321 OMX_IN OMX_U32 nData2, 322 OMX_IN OMX_PTR pEventData) { 323 LOGV("OnEvent(%d, %ld, %ld)", eEvent, nData1, nData2); 324 325 omx_message msg; 326 msg.type = omx_message::EVENT; 327 msg.node = node; 328 msg.u.event_data.event = eEvent; 329 msg.u.event_data.data1 = nData1; 330 msg.u.event_data.data2 = nData2; 331 332 mDispatcher->post(msg); 333 334 return OMX_ErrorNone; 335} 336 337OMX_ERRORTYPE OMX::OnEmptyBufferDone( 338 node_id node, OMX_IN OMX_BUFFERHEADERTYPE *pBuffer) { 339 LOGV("OnEmptyBufferDone buffer=%p", pBuffer); 340 341 omx_message msg; 342 msg.type = omx_message::EMPTY_BUFFER_DONE; 343 msg.node = node; 344 msg.u.buffer_data.buffer = pBuffer; 345 346 mDispatcher->post(msg); 347 348 return OMX_ErrorNone; 349} 350 351OMX_ERRORTYPE OMX::OnFillBufferDone( 352 node_id node, OMX_IN OMX_BUFFERHEADERTYPE *pBuffer) { 353 LOGV("OnFillBufferDone buffer=%p", pBuffer); 354 355 omx_message msg; 356 msg.type = omx_message::FILL_BUFFER_DONE; 357 msg.node = node; 358 msg.u.extended_buffer_data.buffer = pBuffer; 359 msg.u.extended_buffer_data.range_offset = pBuffer->nOffset; 360 msg.u.extended_buffer_data.range_length = pBuffer->nFilledLen; 361 msg.u.extended_buffer_data.flags = pBuffer->nFlags; 362 msg.u.extended_buffer_data.timestamp = pBuffer->nTimeStamp; 363 msg.u.extended_buffer_data.platform_private = pBuffer->pPlatformPrivate; 364 365 mDispatcher->post(msg); 366 367 return OMX_ErrorNone; 368} 369 370OMX::node_id OMX::makeNodeID(OMXNodeInstance *instance) { 371 // mLock is already held. 372 373 node_id node = (node_id)++mNodeCounter; 374 mNodeIDToInstance.add(node, instance); 375 376 return node; 377} 378 379OMXNodeInstance *OMX::findInstance(node_id node) { 380 Mutex::Autolock autoLock(mLock); 381 382 ssize_t index = mNodeIDToInstance.indexOfKey(node); 383 384 return index < 0 ? NULL : mNodeIDToInstance.valueAt(index); 385} 386 387void OMX::invalidateNodeID(node_id node) { 388 Mutex::Autolock autoLock(mLock); 389 invalidateNodeID_l(node); 390} 391 392void OMX::invalidateNodeID_l(node_id node) { 393 // mLock is held. 394 mNodeIDToInstance.removeItem(node); 395} 396 397//////////////////////////////////////////////////////////////////////////////// 398 399sp<IOMXRenderer> OMX::createRenderer( 400 const sp<ISurface> &surface, 401 const char *componentName, 402 OMX_COLOR_FORMATTYPE colorFormat, 403 size_t encodedWidth, size_t encodedHeight, 404 size_t displayWidth, size_t displayHeight) { 405 Mutex::Autolock autoLock(mLock); 406 407 VideoRenderer *impl = NULL; 408 409 static void *libHandle = NULL; 410 411 if (!libHandle) { 412 libHandle = dlopen("libstagefrighthw.so", RTLD_NOW); 413 } 414 415 if (libHandle) { 416 typedef VideoRenderer *(*CreateRendererFunc)( 417 const sp<ISurface> &surface, 418 const char *componentName, 419 OMX_COLOR_FORMATTYPE colorFormat, 420 size_t displayWidth, size_t displayHeight, 421 size_t decodedWidth, size_t decodedHeight); 422 423 CreateRendererFunc func = 424 (CreateRendererFunc)dlsym( 425 libHandle, 426 "_Z14createRendererRKN7android2spINS_8ISurfaceEEEPKc20" 427 "OMX_COLOR_FORMATTYPEjjjj"); 428 429 if (func) { 430 impl = (*func)(surface, componentName, colorFormat, 431 displayWidth, displayHeight, encodedWidth, encodedHeight); 432 } 433 } 434 435 if (!impl) { 436 LOGW("Using software renderer."); 437 impl = new SoftwareRenderer( 438 colorFormat, 439 surface, 440 displayWidth, displayHeight, 441 encodedWidth, encodedHeight); 442 } 443 444 return new OMXRenderer(impl); 445} 446 447OMXRenderer::OMXRenderer(VideoRenderer *impl) 448 : mImpl(impl) { 449} 450 451OMXRenderer::~OMXRenderer() { 452 delete mImpl; 453 mImpl = NULL; 454} 455 456void OMXRenderer::render(IOMX::buffer_id buffer) { 457 OMX_BUFFERHEADERTYPE *header = (OMX_BUFFERHEADERTYPE *)buffer; 458 459 mImpl->render( 460 header->pBuffer + header->nOffset, 461 header->nFilledLen, 462 header->pPlatformPrivate); 463} 464 465} // namespace android 466 467