ACodec.cpp revision 2242fbd1d9327dff9c25ac2aba0fa7d5d471bb88
1/* 2 * Copyright (C) 2010 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 "ACodec" 19 20#include <media/stagefright/ACodec.h> 21 22#include <binder/MemoryDealer.h> 23 24#include <media/stagefright/foundation/hexdump.h> 25#include <media/stagefright/foundation/ABuffer.h> 26#include <media/stagefright/foundation/ADebug.h> 27#include <media/stagefright/foundation/AMessage.h> 28 29#include <media/stagefright/BufferProducerWrapper.h> 30#include <media/stagefright/MediaCodecList.h> 31#include <media/stagefright/MediaDefs.h> 32#include <media/stagefright/NativeWindowWrapper.h> 33#include <media/stagefright/OMXClient.h> 34#include <media/stagefright/OMXCodec.h> 35 36#include <media/hardware/HardwareAPI.h> 37 38#include <OMX_Component.h> 39 40#include "include/avc_utils.h" 41 42namespace android { 43 44template<class T> 45static void InitOMXParams(T *params) { 46 params->nSize = sizeof(T); 47 params->nVersion.s.nVersionMajor = 1; 48 params->nVersion.s.nVersionMinor = 0; 49 params->nVersion.s.nRevision = 0; 50 params->nVersion.s.nStep = 0; 51} 52 53struct CodecObserver : public BnOMXObserver { 54 CodecObserver() {} 55 56 void setNotificationMessage(const sp<AMessage> &msg) { 57 mNotify = msg; 58 } 59 60 // from IOMXObserver 61 virtual void onMessage(const omx_message &omx_msg) { 62 sp<AMessage> msg = mNotify->dup(); 63 64 msg->setInt32("type", omx_msg.type); 65 msg->setPointer("node", omx_msg.node); 66 67 switch (omx_msg.type) { 68 case omx_message::EVENT: 69 { 70 msg->setInt32("event", omx_msg.u.event_data.event); 71 msg->setInt32("data1", omx_msg.u.event_data.data1); 72 msg->setInt32("data2", omx_msg.u.event_data.data2); 73 break; 74 } 75 76 case omx_message::EMPTY_BUFFER_DONE: 77 { 78 msg->setPointer("buffer", omx_msg.u.buffer_data.buffer); 79 break; 80 } 81 82 case omx_message::FILL_BUFFER_DONE: 83 { 84 msg->setPointer( 85 "buffer", omx_msg.u.extended_buffer_data.buffer); 86 msg->setInt32( 87 "range_offset", 88 omx_msg.u.extended_buffer_data.range_offset); 89 msg->setInt32( 90 "range_length", 91 omx_msg.u.extended_buffer_data.range_length); 92 msg->setInt32( 93 "flags", 94 omx_msg.u.extended_buffer_data.flags); 95 msg->setInt64( 96 "timestamp", 97 omx_msg.u.extended_buffer_data.timestamp); 98 msg->setPointer( 99 "platform_private", 100 omx_msg.u.extended_buffer_data.platform_private); 101 msg->setPointer( 102 "data_ptr", 103 omx_msg.u.extended_buffer_data.data_ptr); 104 break; 105 } 106 107 default: 108 TRESPASS(); 109 break; 110 } 111 112 msg->post(); 113 } 114 115protected: 116 virtual ~CodecObserver() {} 117 118private: 119 sp<AMessage> mNotify; 120 121 DISALLOW_EVIL_CONSTRUCTORS(CodecObserver); 122}; 123 124//////////////////////////////////////////////////////////////////////////////// 125 126struct ACodec::BaseState : public AState { 127 BaseState(ACodec *codec, const sp<AState> &parentState = NULL); 128 129protected: 130 enum PortMode { 131 KEEP_BUFFERS, 132 RESUBMIT_BUFFERS, 133 FREE_BUFFERS, 134 }; 135 136 ACodec *mCodec; 137 138 virtual PortMode getPortMode(OMX_U32 portIndex); 139 140 virtual bool onMessageReceived(const sp<AMessage> &msg); 141 142 virtual bool onOMXEvent(OMX_EVENTTYPE event, OMX_U32 data1, OMX_U32 data2); 143 144 virtual void onOutputBufferDrained(const sp<AMessage> &msg); 145 virtual void onInputBufferFilled(const sp<AMessage> &msg); 146 147 void postFillThisBuffer(BufferInfo *info); 148 149private: 150 bool onOMXMessage(const sp<AMessage> &msg); 151 152 bool onOMXEmptyBufferDone(IOMX::buffer_id bufferID); 153 154 bool onOMXFillBufferDone( 155 IOMX::buffer_id bufferID, 156 size_t rangeOffset, size_t rangeLength, 157 OMX_U32 flags, 158 int64_t timeUs, 159 void *platformPrivate, 160 void *dataPtr); 161 162 void getMoreInputDataIfPossible(); 163 164 DISALLOW_EVIL_CONSTRUCTORS(BaseState); 165}; 166 167//////////////////////////////////////////////////////////////////////////////// 168 169struct ACodec::DeathNotifier : public IBinder::DeathRecipient { 170 DeathNotifier(const sp<AMessage> ¬ify) 171 : mNotify(notify) { 172 } 173 174 virtual void binderDied(const wp<IBinder> &) { 175 mNotify->post(); 176 } 177 178protected: 179 virtual ~DeathNotifier() {} 180 181private: 182 sp<AMessage> mNotify; 183 184 DISALLOW_EVIL_CONSTRUCTORS(DeathNotifier); 185}; 186 187struct ACodec::UninitializedState : public ACodec::BaseState { 188 UninitializedState(ACodec *codec); 189 190protected: 191 virtual bool onMessageReceived(const sp<AMessage> &msg); 192 virtual void stateEntered(); 193 194private: 195 void onSetup(const sp<AMessage> &msg); 196 bool onAllocateComponent(const sp<AMessage> &msg); 197 198 sp<DeathNotifier> mDeathNotifier; 199 200 DISALLOW_EVIL_CONSTRUCTORS(UninitializedState); 201}; 202 203//////////////////////////////////////////////////////////////////////////////// 204 205struct ACodec::LoadedState : public ACodec::BaseState { 206 LoadedState(ACodec *codec); 207 208protected: 209 virtual bool onMessageReceived(const sp<AMessage> &msg); 210 virtual void stateEntered(); 211 212private: 213 friend struct ACodec::UninitializedState; 214 215 bool onConfigureComponent(const sp<AMessage> &msg); 216 void onCreateInputSurface(const sp<AMessage> &msg); 217 void onStart(); 218 void onShutdown(bool keepComponentAllocated); 219 220 DISALLOW_EVIL_CONSTRUCTORS(LoadedState); 221}; 222 223//////////////////////////////////////////////////////////////////////////////// 224 225struct ACodec::LoadedToIdleState : public ACodec::BaseState { 226 LoadedToIdleState(ACodec *codec); 227 228protected: 229 virtual bool onMessageReceived(const sp<AMessage> &msg); 230 virtual bool onOMXEvent(OMX_EVENTTYPE event, OMX_U32 data1, OMX_U32 data2); 231 virtual void stateEntered(); 232 233private: 234 status_t allocateBuffers(); 235 236 DISALLOW_EVIL_CONSTRUCTORS(LoadedToIdleState); 237}; 238 239//////////////////////////////////////////////////////////////////////////////// 240 241struct ACodec::IdleToExecutingState : public ACodec::BaseState { 242 IdleToExecutingState(ACodec *codec); 243 244protected: 245 virtual bool onMessageReceived(const sp<AMessage> &msg); 246 virtual bool onOMXEvent(OMX_EVENTTYPE event, OMX_U32 data1, OMX_U32 data2); 247 virtual void stateEntered(); 248 249private: 250 DISALLOW_EVIL_CONSTRUCTORS(IdleToExecutingState); 251}; 252 253//////////////////////////////////////////////////////////////////////////////// 254 255struct ACodec::ExecutingState : public ACodec::BaseState { 256 ExecutingState(ACodec *codec); 257 258 void submitOutputBuffers(); 259 260 // Submit output buffers to the decoder, submit input buffers to client 261 // to fill with data. 262 void resume(); 263 264 // Returns true iff input and output buffers are in play. 265 bool active() const { return mActive; } 266 267protected: 268 virtual PortMode getPortMode(OMX_U32 portIndex); 269 virtual bool onMessageReceived(const sp<AMessage> &msg); 270 virtual void stateEntered(); 271 272 virtual bool onOMXEvent(OMX_EVENTTYPE event, OMX_U32 data1, OMX_U32 data2); 273 274private: 275 bool mActive; 276 277 DISALLOW_EVIL_CONSTRUCTORS(ExecutingState); 278}; 279 280//////////////////////////////////////////////////////////////////////////////// 281 282struct ACodec::OutputPortSettingsChangedState : public ACodec::BaseState { 283 OutputPortSettingsChangedState(ACodec *codec); 284 285protected: 286 virtual PortMode getPortMode(OMX_U32 portIndex); 287 virtual bool onMessageReceived(const sp<AMessage> &msg); 288 virtual void stateEntered(); 289 290 virtual bool onOMXEvent(OMX_EVENTTYPE event, OMX_U32 data1, OMX_U32 data2); 291 292private: 293 DISALLOW_EVIL_CONSTRUCTORS(OutputPortSettingsChangedState); 294}; 295 296//////////////////////////////////////////////////////////////////////////////// 297 298struct ACodec::ExecutingToIdleState : public ACodec::BaseState { 299 ExecutingToIdleState(ACodec *codec); 300 301protected: 302 virtual bool onMessageReceived(const sp<AMessage> &msg); 303 virtual void stateEntered(); 304 305 virtual bool onOMXEvent(OMX_EVENTTYPE event, OMX_U32 data1, OMX_U32 data2); 306 307 virtual void onOutputBufferDrained(const sp<AMessage> &msg); 308 virtual void onInputBufferFilled(const sp<AMessage> &msg); 309 310private: 311 void changeStateIfWeOwnAllBuffers(); 312 313 bool mComponentNowIdle; 314 315 DISALLOW_EVIL_CONSTRUCTORS(ExecutingToIdleState); 316}; 317 318//////////////////////////////////////////////////////////////////////////////// 319 320struct ACodec::IdleToLoadedState : public ACodec::BaseState { 321 IdleToLoadedState(ACodec *codec); 322 323protected: 324 virtual bool onMessageReceived(const sp<AMessage> &msg); 325 virtual void stateEntered(); 326 327 virtual bool onOMXEvent(OMX_EVENTTYPE event, OMX_U32 data1, OMX_U32 data2); 328 329private: 330 DISALLOW_EVIL_CONSTRUCTORS(IdleToLoadedState); 331}; 332 333//////////////////////////////////////////////////////////////////////////////// 334 335struct ACodec::FlushingState : public ACodec::BaseState { 336 FlushingState(ACodec *codec); 337 338protected: 339 virtual bool onMessageReceived(const sp<AMessage> &msg); 340 virtual void stateEntered(); 341 342 virtual bool onOMXEvent(OMX_EVENTTYPE event, OMX_U32 data1, OMX_U32 data2); 343 344 virtual void onOutputBufferDrained(const sp<AMessage> &msg); 345 virtual void onInputBufferFilled(const sp<AMessage> &msg); 346 347private: 348 bool mFlushComplete[2]; 349 350 void changeStateIfWeOwnAllBuffers(); 351 352 DISALLOW_EVIL_CONSTRUCTORS(FlushingState); 353}; 354 355//////////////////////////////////////////////////////////////////////////////// 356 357ACodec::ACodec() 358 : mQuirks(0), 359 mNode(NULL), 360 mSentFormat(false), 361 mIsEncoder(false), 362 mShutdownInProgress(false), 363 mEncoderDelay(0), 364 mEncoderPadding(0), 365 mChannelMaskPresent(false), 366 mChannelMask(0) { 367 mUninitializedState = new UninitializedState(this); 368 mLoadedState = new LoadedState(this); 369 mLoadedToIdleState = new LoadedToIdleState(this); 370 mIdleToExecutingState = new IdleToExecutingState(this); 371 mExecutingState = new ExecutingState(this); 372 373 mOutputPortSettingsChangedState = 374 new OutputPortSettingsChangedState(this); 375 376 mExecutingToIdleState = new ExecutingToIdleState(this); 377 mIdleToLoadedState = new IdleToLoadedState(this); 378 mFlushingState = new FlushingState(this); 379 380 mPortEOS[kPortIndexInput] = mPortEOS[kPortIndexOutput] = false; 381 mInputEOSResult = OK; 382 383 changeState(mUninitializedState); 384} 385 386ACodec::~ACodec() { 387} 388 389void ACodec::setNotificationMessage(const sp<AMessage> &msg) { 390 mNotify = msg; 391} 392 393void ACodec::initiateSetup(const sp<AMessage> &msg) { 394 msg->setWhat(kWhatSetup); 395 msg->setTarget(id()); 396 msg->post(); 397} 398 399void ACodec::signalSetParameters(const sp<AMessage> ¶ms) { 400 sp<AMessage> msg = new AMessage(kWhatSetParameters, id()); 401 msg->setMessage("params", params); 402 msg->post(); 403} 404 405void ACodec::initiateAllocateComponent(const sp<AMessage> &msg) { 406 msg->setWhat(kWhatAllocateComponent); 407 msg->setTarget(id()); 408 msg->post(); 409} 410 411void ACodec::initiateConfigureComponent(const sp<AMessage> &msg) { 412 msg->setWhat(kWhatConfigureComponent); 413 msg->setTarget(id()); 414 msg->post(); 415} 416 417void ACodec::initiateCreateInputSurface() { 418 (new AMessage(kWhatCreateInputSurface, id()))->post(); 419} 420 421void ACodec::signalEndOfInputStream() { 422 (new AMessage(kWhatSignalEndOfInputStream, id()))->post(); 423} 424 425void ACodec::initiateStart() { 426 (new AMessage(kWhatStart, id()))->post(); 427} 428 429void ACodec::signalFlush() { 430 ALOGV("[%s] signalFlush", mComponentName.c_str()); 431 (new AMessage(kWhatFlush, id()))->post(); 432} 433 434void ACodec::signalResume() { 435 (new AMessage(kWhatResume, id()))->post(); 436} 437 438void ACodec::initiateShutdown(bool keepComponentAllocated) { 439 sp<AMessage> msg = new AMessage(kWhatShutdown, id()); 440 msg->setInt32("keepComponentAllocated", keepComponentAllocated); 441 msg->post(); 442} 443 444void ACodec::signalRequestIDRFrame() { 445 (new AMessage(kWhatRequestIDRFrame, id()))->post(); 446} 447 448status_t ACodec::allocateBuffersOnPort(OMX_U32 portIndex) { 449 CHECK(portIndex == kPortIndexInput || portIndex == kPortIndexOutput); 450 451 CHECK(mDealer[portIndex] == NULL); 452 CHECK(mBuffers[portIndex].isEmpty()); 453 454 status_t err; 455 if (mNativeWindow != NULL && portIndex == kPortIndexOutput) { 456 err = allocateOutputBuffersFromNativeWindow(); 457 } else { 458 OMX_PARAM_PORTDEFINITIONTYPE def; 459 InitOMXParams(&def); 460 def.nPortIndex = portIndex; 461 462 err = mOMX->getParameter( 463 mNode, OMX_IndexParamPortDefinition, &def, sizeof(def)); 464 465 if (err == OK) { 466 ALOGV("[%s] Allocating %lu buffers of size %lu on %s port", 467 mComponentName.c_str(), 468 def.nBufferCountActual, def.nBufferSize, 469 portIndex == kPortIndexInput ? "input" : "output"); 470 471 size_t totalSize = def.nBufferCountActual * def.nBufferSize; 472 mDealer[portIndex] = new MemoryDealer(totalSize, "ACodec"); 473 474 for (OMX_U32 i = 0; i < def.nBufferCountActual; ++i) { 475 sp<IMemory> mem = mDealer[portIndex]->allocate(def.nBufferSize); 476 CHECK(mem.get() != NULL); 477 478 BufferInfo info; 479 info.mStatus = BufferInfo::OWNED_BY_US; 480 481 uint32_t requiresAllocateBufferBit = 482 (portIndex == kPortIndexInput) 483 ? OMXCodec::kRequiresAllocateBufferOnInputPorts 484 : OMXCodec::kRequiresAllocateBufferOnOutputPorts; 485 486 if (portIndex == kPortIndexInput && (mFlags & kFlagIsSecure)) { 487 mem.clear(); 488 489 void *ptr; 490 err = mOMX->allocateBuffer( 491 mNode, portIndex, def.nBufferSize, &info.mBufferID, 492 &ptr); 493 494 info.mData = new ABuffer(ptr, def.nBufferSize); 495 } else if (mQuirks & requiresAllocateBufferBit) { 496 err = mOMX->allocateBufferWithBackup( 497 mNode, portIndex, mem, &info.mBufferID); 498 } else { 499 err = mOMX->useBuffer(mNode, portIndex, mem, &info.mBufferID); 500 } 501 502 if (mem != NULL) { 503 info.mData = new ABuffer(mem->pointer(), def.nBufferSize); 504 } 505 506 mBuffers[portIndex].push(info); 507 } 508 } 509 } 510 511 if (err != OK) { 512 return err; 513 } 514 515 sp<AMessage> notify = mNotify->dup(); 516 notify->setInt32("what", ACodec::kWhatBuffersAllocated); 517 518 notify->setInt32("portIndex", portIndex); 519 520 sp<PortDescription> desc = new PortDescription; 521 522 for (size_t i = 0; i < mBuffers[portIndex].size(); ++i) { 523 const BufferInfo &info = mBuffers[portIndex][i]; 524 525 desc->addBuffer(info.mBufferID, info.mData); 526 } 527 528 notify->setObject("portDesc", desc); 529 notify->post(); 530 531 return OK; 532} 533 534status_t ACodec::allocateOutputBuffersFromNativeWindow() { 535 OMX_PARAM_PORTDEFINITIONTYPE def; 536 InitOMXParams(&def); 537 def.nPortIndex = kPortIndexOutput; 538 539 status_t err = mOMX->getParameter( 540 mNode, OMX_IndexParamPortDefinition, &def, sizeof(def)); 541 542 if (err != OK) { 543 return err; 544 } 545 546 err = native_window_set_buffers_geometry( 547 mNativeWindow.get(), 548 def.format.video.nFrameWidth, 549 def.format.video.nFrameHeight, 550 def.format.video.eColorFormat); 551 552 if (err != 0) { 553 ALOGE("native_window_set_buffers_geometry failed: %s (%d)", 554 strerror(-err), -err); 555 return err; 556 } 557 558 // Set up the native window. 559 OMX_U32 usage = 0; 560 err = mOMX->getGraphicBufferUsage(mNode, kPortIndexOutput, &usage); 561 if (err != 0) { 562 ALOGW("querying usage flags from OMX IL component failed: %d", err); 563 // XXX: Currently this error is logged, but not fatal. 564 usage = 0; 565 } 566 567 if (mFlags & kFlagIsSecure) { 568 usage |= GRALLOC_USAGE_PROTECTED; 569 } 570 571 // Make sure to check whether either Stagefright or the video decoder 572 // requested protected buffers. 573 if (usage & GRALLOC_USAGE_PROTECTED) { 574 // Verify that the ANativeWindow sends images directly to 575 // SurfaceFlinger. 576 int queuesToNativeWindow = 0; 577 err = mNativeWindow->query( 578 mNativeWindow.get(), NATIVE_WINDOW_QUEUES_TO_WINDOW_COMPOSER, 579 &queuesToNativeWindow); 580 if (err != 0) { 581 ALOGE("error authenticating native window: %d", err); 582 return err; 583 } 584 if (queuesToNativeWindow != 1) { 585 ALOGE("native window could not be authenticated"); 586 return PERMISSION_DENIED; 587 } 588 } 589 590 err = native_window_set_usage( 591 mNativeWindow.get(), 592 usage | GRALLOC_USAGE_HW_TEXTURE | GRALLOC_USAGE_EXTERNAL_DISP); 593 594 if (err != 0) { 595 ALOGE("native_window_set_usage failed: %s (%d)", strerror(-err), -err); 596 return err; 597 } 598 599 int minUndequeuedBufs = 0; 600 err = mNativeWindow->query( 601 mNativeWindow.get(), NATIVE_WINDOW_MIN_UNDEQUEUED_BUFFERS, 602 &minUndequeuedBufs); 603 604 if (err != 0) { 605 ALOGE("NATIVE_WINDOW_MIN_UNDEQUEUED_BUFFERS query failed: %s (%d)", 606 strerror(-err), -err); 607 return err; 608 } 609 610 // XXX: Is this the right logic to use? It's not clear to me what the OMX 611 // buffer counts refer to - how do they account for the renderer holding on 612 // to buffers? 613 if (def.nBufferCountActual < def.nBufferCountMin + minUndequeuedBufs) { 614 OMX_U32 newBufferCount = def.nBufferCountMin + minUndequeuedBufs; 615 def.nBufferCountActual = newBufferCount; 616 err = mOMX->setParameter( 617 mNode, OMX_IndexParamPortDefinition, &def, sizeof(def)); 618 619 if (err != OK) { 620 ALOGE("[%s] setting nBufferCountActual to %lu failed: %d", 621 mComponentName.c_str(), newBufferCount, err); 622 return err; 623 } 624 } 625 626 err = native_window_set_buffer_count( 627 mNativeWindow.get(), def.nBufferCountActual); 628 629 if (err != 0) { 630 ALOGE("native_window_set_buffer_count failed: %s (%d)", strerror(-err), 631 -err); 632 return err; 633 } 634 635 ALOGV("[%s] Allocating %lu buffers from a native window of size %lu on " 636 "output port", 637 mComponentName.c_str(), def.nBufferCountActual, def.nBufferSize); 638 639 // Dequeue buffers and send them to OMX 640 for (OMX_U32 i = 0; i < def.nBufferCountActual; i++) { 641 ANativeWindowBuffer *buf; 642 err = native_window_dequeue_buffer_and_wait(mNativeWindow.get(), &buf); 643 if (err != 0) { 644 ALOGE("dequeueBuffer failed: %s (%d)", strerror(-err), -err); 645 break; 646 } 647 648 sp<GraphicBuffer> graphicBuffer(new GraphicBuffer(buf, false)); 649 BufferInfo info; 650 info.mStatus = BufferInfo::OWNED_BY_US; 651 info.mData = new ABuffer(NULL /* data */, def.nBufferSize /* capacity */); 652 info.mGraphicBuffer = graphicBuffer; 653 mBuffers[kPortIndexOutput].push(info); 654 655 IOMX::buffer_id bufferId; 656 err = mOMX->useGraphicBuffer(mNode, kPortIndexOutput, graphicBuffer, 657 &bufferId); 658 if (err != 0) { 659 ALOGE("registering GraphicBuffer %lu with OMX IL component failed: " 660 "%d", i, err); 661 break; 662 } 663 664 mBuffers[kPortIndexOutput].editItemAt(i).mBufferID = bufferId; 665 666 ALOGV("[%s] Registered graphic buffer with ID %p (pointer = %p)", 667 mComponentName.c_str(), 668 bufferId, graphicBuffer.get()); 669 } 670 671 OMX_U32 cancelStart; 672 OMX_U32 cancelEnd; 673 674 if (err != 0) { 675 // If an error occurred while dequeuing we need to cancel any buffers 676 // that were dequeued. 677 cancelStart = 0; 678 cancelEnd = mBuffers[kPortIndexOutput].size(); 679 } else { 680 // Return the last two buffers to the native window. 681 cancelStart = def.nBufferCountActual - minUndequeuedBufs; 682 cancelEnd = def.nBufferCountActual; 683 } 684 685 for (OMX_U32 i = cancelStart; i < cancelEnd; i++) { 686 BufferInfo *info = &mBuffers[kPortIndexOutput].editItemAt(i); 687 cancelBufferToNativeWindow(info); 688 } 689 690 return err; 691} 692 693status_t ACodec::cancelBufferToNativeWindow(BufferInfo *info) { 694 CHECK_EQ((int)info->mStatus, (int)BufferInfo::OWNED_BY_US); 695 696 ALOGV("[%s] Calling cancelBuffer on buffer %p", 697 mComponentName.c_str(), info->mBufferID); 698 699 int err = mNativeWindow->cancelBuffer( 700 mNativeWindow.get(), info->mGraphicBuffer.get(), -1); 701 702 CHECK_EQ(err, 0); 703 704 info->mStatus = BufferInfo::OWNED_BY_NATIVE_WINDOW; 705 706 return OK; 707} 708 709ACodec::BufferInfo *ACodec::dequeueBufferFromNativeWindow() { 710 ANativeWindowBuffer *buf; 711 int fenceFd = -1; 712 if (native_window_dequeue_buffer_and_wait(mNativeWindow.get(), &buf) != 0) { 713 ALOGE("dequeueBuffer failed."); 714 return NULL; 715 } 716 717 for (size_t i = mBuffers[kPortIndexOutput].size(); i-- > 0;) { 718 BufferInfo *info = 719 &mBuffers[kPortIndexOutput].editItemAt(i); 720 721 if (info->mGraphicBuffer->handle == buf->handle) { 722 CHECK_EQ((int)info->mStatus, 723 (int)BufferInfo::OWNED_BY_NATIVE_WINDOW); 724 725 info->mStatus = BufferInfo::OWNED_BY_US; 726 727 return info; 728 } 729 } 730 731 TRESPASS(); 732 733 return NULL; 734} 735 736status_t ACodec::freeBuffersOnPort(OMX_U32 portIndex) { 737 for (size_t i = mBuffers[portIndex].size(); i-- > 0;) { 738 CHECK_EQ((status_t)OK, freeBuffer(portIndex, i)); 739 } 740 741 mDealer[portIndex].clear(); 742 743 return OK; 744} 745 746status_t ACodec::freeOutputBuffersNotOwnedByComponent() { 747 for (size_t i = mBuffers[kPortIndexOutput].size(); i-- > 0;) { 748 BufferInfo *info = 749 &mBuffers[kPortIndexOutput].editItemAt(i); 750 751 if (info->mStatus != 752 BufferInfo::OWNED_BY_COMPONENT) { 753 // We shouldn't have sent out any buffers to the client at this 754 // point. 755 CHECK_NE((int)info->mStatus, (int)BufferInfo::OWNED_BY_DOWNSTREAM); 756 757 CHECK_EQ((status_t)OK, freeBuffer(kPortIndexOutput, i)); 758 } 759 } 760 761 return OK; 762} 763 764status_t ACodec::freeBuffer(OMX_U32 portIndex, size_t i) { 765 BufferInfo *info = &mBuffers[portIndex].editItemAt(i); 766 767 CHECK(info->mStatus == BufferInfo::OWNED_BY_US 768 || info->mStatus == BufferInfo::OWNED_BY_NATIVE_WINDOW); 769 770 if (portIndex == kPortIndexOutput && mNativeWindow != NULL 771 && info->mStatus == BufferInfo::OWNED_BY_US) { 772 CHECK_EQ((status_t)OK, cancelBufferToNativeWindow(info)); 773 } 774 775 CHECK_EQ(mOMX->freeBuffer( 776 mNode, portIndex, info->mBufferID), 777 (status_t)OK); 778 779 mBuffers[portIndex].removeAt(i); 780 781 return OK; 782} 783 784ACodec::BufferInfo *ACodec::findBufferByID( 785 uint32_t portIndex, IOMX::buffer_id bufferID, 786 ssize_t *index) { 787 for (size_t i = 0; i < mBuffers[portIndex].size(); ++i) { 788 BufferInfo *info = &mBuffers[portIndex].editItemAt(i); 789 790 if (info->mBufferID == bufferID) { 791 if (index != NULL) { 792 *index = i; 793 } 794 return info; 795 } 796 } 797 798 TRESPASS(); 799 800 return NULL; 801} 802 803status_t ACodec::setComponentRole( 804 bool isEncoder, const char *mime) { 805 struct MimeToRole { 806 const char *mime; 807 const char *decoderRole; 808 const char *encoderRole; 809 }; 810 811 static const MimeToRole kMimeToRole[] = { 812 { MEDIA_MIMETYPE_AUDIO_MPEG, 813 "audio_decoder.mp3", "audio_encoder.mp3" }, 814 { MEDIA_MIMETYPE_AUDIO_MPEG_LAYER_I, 815 "audio_decoder.mp1", "audio_encoder.mp1" }, 816 { MEDIA_MIMETYPE_AUDIO_MPEG_LAYER_II, 817 "audio_decoder.mp2", "audio_encoder.mp2" }, 818 { MEDIA_MIMETYPE_AUDIO_AMR_NB, 819 "audio_decoder.amrnb", "audio_encoder.amrnb" }, 820 { MEDIA_MIMETYPE_AUDIO_AMR_WB, 821 "audio_decoder.amrwb", "audio_encoder.amrwb" }, 822 { MEDIA_MIMETYPE_AUDIO_AAC, 823 "audio_decoder.aac", "audio_encoder.aac" }, 824 { MEDIA_MIMETYPE_AUDIO_VORBIS, 825 "audio_decoder.vorbis", "audio_encoder.vorbis" }, 826 { MEDIA_MIMETYPE_AUDIO_G711_MLAW, 827 "audio_decoder.g711mlaw", "audio_encoder.g711mlaw" }, 828 { MEDIA_MIMETYPE_AUDIO_G711_ALAW, 829 "audio_decoder.g711alaw", "audio_encoder.g711alaw" }, 830 { MEDIA_MIMETYPE_VIDEO_AVC, 831 "video_decoder.avc", "video_encoder.avc" }, 832 { MEDIA_MIMETYPE_VIDEO_MPEG4, 833 "video_decoder.mpeg4", "video_encoder.mpeg4" }, 834 { MEDIA_MIMETYPE_VIDEO_H263, 835 "video_decoder.h263", "video_encoder.h263" }, 836 { MEDIA_MIMETYPE_VIDEO_VPX, 837 "video_decoder.vpx", "video_encoder.vpx" }, 838 { MEDIA_MIMETYPE_AUDIO_RAW, 839 "audio_decoder.raw", "audio_encoder.raw" }, 840 { MEDIA_MIMETYPE_AUDIO_FLAC, 841 "audio_decoder.flac", "audio_encoder.flac" }, 842 { MEDIA_MIMETYPE_AUDIO_MSGSM, 843 "audio_decoder.gsm", "audio_encoder.gsm" }, 844 }; 845 846 static const size_t kNumMimeToRole = 847 sizeof(kMimeToRole) / sizeof(kMimeToRole[0]); 848 849 size_t i; 850 for (i = 0; i < kNumMimeToRole; ++i) { 851 if (!strcasecmp(mime, kMimeToRole[i].mime)) { 852 break; 853 } 854 } 855 856 if (i == kNumMimeToRole) { 857 return ERROR_UNSUPPORTED; 858 } 859 860 const char *role = 861 isEncoder ? kMimeToRole[i].encoderRole 862 : kMimeToRole[i].decoderRole; 863 864 if (role != NULL) { 865 OMX_PARAM_COMPONENTROLETYPE roleParams; 866 InitOMXParams(&roleParams); 867 868 strncpy((char *)roleParams.cRole, 869 role, OMX_MAX_STRINGNAME_SIZE - 1); 870 871 roleParams.cRole[OMX_MAX_STRINGNAME_SIZE - 1] = '\0'; 872 873 status_t err = mOMX->setParameter( 874 mNode, OMX_IndexParamStandardComponentRole, 875 &roleParams, sizeof(roleParams)); 876 877 if (err != OK) { 878 ALOGW("[%s] Failed to set standard component role '%s'.", 879 mComponentName.c_str(), role); 880 881 return err; 882 } 883 } 884 885 return OK; 886} 887 888status_t ACodec::configureCodec( 889 const char *mime, const sp<AMessage> &msg) { 890 int32_t encoder; 891 if (!msg->findInt32("encoder", &encoder)) { 892 encoder = false; 893 } 894 895 mIsEncoder = encoder; 896 897 status_t err = setComponentRole(encoder /* isEncoder */, mime); 898 899 if (err != OK) { 900 return err; 901 } 902 903 int32_t bitRate = 0; 904 // FLAC encoder doesn't need a bitrate, other encoders do 905 if (encoder && strcasecmp(mime, MEDIA_MIMETYPE_AUDIO_FLAC) 906 && !msg->findInt32("bitrate", &bitRate)) { 907 return INVALID_OPERATION; 908 } 909 910 int32_t storeMeta; 911 if (encoder 912 && msg->findInt32("store-metadata-in-buffers", &storeMeta) 913 && storeMeta != 0) { 914 err = mOMX->storeMetaDataInBuffers(mNode, kPortIndexInput, OMX_TRUE); 915 916 if (err != OK) { 917 ALOGE("[%s] storeMetaDataInBuffers failed w/ err %d", 918 mComponentName.c_str(), err); 919 920 return err; 921 } 922 } 923 924 int32_t prependSPSPPS; 925 if (encoder 926 && msg->findInt32("prepend-sps-pps-to-idr-frames", &prependSPSPPS) 927 && prependSPSPPS != 0) { 928 OMX_INDEXTYPE index; 929 err = mOMX->getExtensionIndex( 930 mNode, 931 "OMX.google.android.index.prependSPSPPSToIDRFrames", 932 &index); 933 934 if (err == OK) { 935 PrependSPSPPSToIDRFramesParams params; 936 InitOMXParams(¶ms); 937 params.bEnable = OMX_TRUE; 938 939 err = mOMX->setParameter( 940 mNode, index, ¶ms, sizeof(params)); 941 } 942 943 if (err != OK) { 944 ALOGE("Encoder could not be configured to emit SPS/PPS before " 945 "IDR frames. (err %d)", err); 946 947 return err; 948 } 949 } 950 951 if (!strncasecmp(mime, "video/", 6)) { 952 if (encoder) { 953 err = setupVideoEncoder(mime, msg); 954 } else { 955 int32_t width, height; 956 if (!msg->findInt32("width", &width) 957 || !msg->findInt32("height", &height)) { 958 err = INVALID_OPERATION; 959 } else { 960 err = setupVideoDecoder(mime, width, height); 961 } 962 } 963 } else if (!strcasecmp(mime, MEDIA_MIMETYPE_AUDIO_MPEG)) { 964 int32_t numChannels, sampleRate; 965 if (!msg->findInt32("channel-count", &numChannels) 966 || !msg->findInt32("sample-rate", &sampleRate)) { 967 // Since we did not always check for these, leave them optional 968 // and have the decoder figure it all out. 969 err = OK; 970 } else { 971 err = setupRawAudioFormat( 972 encoder ? kPortIndexInput : kPortIndexOutput, 973 sampleRate, 974 numChannels); 975 } 976 } else if (!strcasecmp(mime, MEDIA_MIMETYPE_AUDIO_AAC)) { 977 int32_t numChannels, sampleRate; 978 if (!msg->findInt32("channel-count", &numChannels) 979 || !msg->findInt32("sample-rate", &sampleRate)) { 980 err = INVALID_OPERATION; 981 } else { 982 int32_t isADTS, aacProfile; 983 if (!msg->findInt32("is-adts", &isADTS)) { 984 isADTS = 0; 985 } 986 if (!msg->findInt32("aac-profile", &aacProfile)) { 987 aacProfile = OMX_AUDIO_AACObjectNull; 988 } 989 990 err = setupAACCodec( 991 encoder, numChannels, sampleRate, bitRate, aacProfile, 992 isADTS != 0); 993 } 994 } else if (!strcasecmp(mime, MEDIA_MIMETYPE_AUDIO_AMR_NB)) { 995 err = setupAMRCodec(encoder, false /* isWAMR */, bitRate); 996 } else if (!strcasecmp(mime, MEDIA_MIMETYPE_AUDIO_AMR_WB)) { 997 err = setupAMRCodec(encoder, true /* isWAMR */, bitRate); 998 } else if (!strcasecmp(mime, MEDIA_MIMETYPE_AUDIO_G711_ALAW) 999 || !strcasecmp(mime, MEDIA_MIMETYPE_AUDIO_G711_MLAW)) { 1000 // These are PCM-like formats with a fixed sample rate but 1001 // a variable number of channels. 1002 1003 int32_t numChannels; 1004 if (!msg->findInt32("channel-count", &numChannels)) { 1005 err = INVALID_OPERATION; 1006 } else { 1007 err = setupG711Codec(encoder, numChannels); 1008 } 1009 } else if (!strcasecmp(mime, MEDIA_MIMETYPE_AUDIO_FLAC)) { 1010 int32_t numChannels, sampleRate, compressionLevel = -1; 1011 if (encoder && 1012 (!msg->findInt32("channel-count", &numChannels) 1013 || !msg->findInt32("sample-rate", &sampleRate))) { 1014 ALOGE("missing channel count or sample rate for FLAC encoder"); 1015 err = INVALID_OPERATION; 1016 } else { 1017 if (encoder) { 1018 if (!msg->findInt32( 1019 "flac-compression-level", &compressionLevel)) { 1020 compressionLevel = 5;// default FLAC compression level 1021 } else if (compressionLevel < 0) { 1022 ALOGW("compression level %d outside [0..8] range, " 1023 "using 0", 1024 compressionLevel); 1025 compressionLevel = 0; 1026 } else if (compressionLevel > 8) { 1027 ALOGW("compression level %d outside [0..8] range, " 1028 "using 8", 1029 compressionLevel); 1030 compressionLevel = 8; 1031 } 1032 } 1033 err = setupFlacCodec( 1034 encoder, numChannels, sampleRate, compressionLevel); 1035 } 1036 } else if (!strcasecmp(mime, MEDIA_MIMETYPE_AUDIO_RAW)) { 1037 int32_t numChannels, sampleRate; 1038 if (encoder 1039 || !msg->findInt32("channel-count", &numChannels) 1040 || !msg->findInt32("sample-rate", &sampleRate)) { 1041 err = INVALID_OPERATION; 1042 } else { 1043 err = setupRawAudioFormat(kPortIndexInput, sampleRate, numChannels); 1044 } 1045 } 1046 1047 if (err != OK) { 1048 return err; 1049 } 1050 1051 if (!msg->findInt32("encoder-delay", &mEncoderDelay)) { 1052 mEncoderDelay = 0; 1053 } 1054 1055 if (!msg->findInt32("encoder-padding", &mEncoderPadding)) { 1056 mEncoderPadding = 0; 1057 } 1058 1059 if (msg->findInt32("channel-mask", &mChannelMask)) { 1060 mChannelMaskPresent = true; 1061 } else { 1062 mChannelMaskPresent = false; 1063 } 1064 1065 int32_t maxInputSize; 1066 if (msg->findInt32("max-input-size", &maxInputSize)) { 1067 err = setMinBufferSize(kPortIndexInput, (size_t)maxInputSize); 1068 } else if (!strcmp("OMX.Nvidia.aac.decoder", mComponentName.c_str())) { 1069 err = setMinBufferSize(kPortIndexInput, 8192); // XXX 1070 } 1071 1072 return err; 1073} 1074 1075status_t ACodec::setMinBufferSize(OMX_U32 portIndex, size_t size) { 1076 OMX_PARAM_PORTDEFINITIONTYPE def; 1077 InitOMXParams(&def); 1078 def.nPortIndex = portIndex; 1079 1080 status_t err = mOMX->getParameter( 1081 mNode, OMX_IndexParamPortDefinition, &def, sizeof(def)); 1082 1083 if (err != OK) { 1084 return err; 1085 } 1086 1087 if (def.nBufferSize >= size) { 1088 return OK; 1089 } 1090 1091 def.nBufferSize = size; 1092 1093 err = mOMX->setParameter( 1094 mNode, OMX_IndexParamPortDefinition, &def, sizeof(def)); 1095 1096 if (err != OK) { 1097 return err; 1098 } 1099 1100 err = mOMX->getParameter( 1101 mNode, OMX_IndexParamPortDefinition, &def, sizeof(def)); 1102 1103 if (err != OK) { 1104 return err; 1105 } 1106 1107 CHECK(def.nBufferSize >= size); 1108 1109 return OK; 1110} 1111 1112status_t ACodec::selectAudioPortFormat( 1113 OMX_U32 portIndex, OMX_AUDIO_CODINGTYPE desiredFormat) { 1114 OMX_AUDIO_PARAM_PORTFORMATTYPE format; 1115 InitOMXParams(&format); 1116 1117 format.nPortIndex = portIndex; 1118 for (OMX_U32 index = 0;; ++index) { 1119 format.nIndex = index; 1120 1121 status_t err = mOMX->getParameter( 1122 mNode, OMX_IndexParamAudioPortFormat, 1123 &format, sizeof(format)); 1124 1125 if (err != OK) { 1126 return err; 1127 } 1128 1129 if (format.eEncoding == desiredFormat) { 1130 break; 1131 } 1132 } 1133 1134 return mOMX->setParameter( 1135 mNode, OMX_IndexParamAudioPortFormat, &format, sizeof(format)); 1136} 1137 1138status_t ACodec::setupAACCodec( 1139 bool encoder, int32_t numChannels, int32_t sampleRate, 1140 int32_t bitRate, int32_t aacProfile, bool isADTS) { 1141 if (encoder && isADTS) { 1142 return -EINVAL; 1143 } 1144 1145 status_t err = setupRawAudioFormat( 1146 encoder ? kPortIndexInput : kPortIndexOutput, 1147 sampleRate, 1148 numChannels); 1149 1150 if (err != OK) { 1151 return err; 1152 } 1153 1154 if (encoder) { 1155 err = selectAudioPortFormat(kPortIndexOutput, OMX_AUDIO_CodingAAC); 1156 1157 if (err != OK) { 1158 return err; 1159 } 1160 1161 OMX_PARAM_PORTDEFINITIONTYPE def; 1162 InitOMXParams(&def); 1163 def.nPortIndex = kPortIndexOutput; 1164 1165 err = mOMX->getParameter( 1166 mNode, OMX_IndexParamPortDefinition, &def, sizeof(def)); 1167 1168 if (err != OK) { 1169 return err; 1170 } 1171 1172 def.format.audio.bFlagErrorConcealment = OMX_TRUE; 1173 def.format.audio.eEncoding = OMX_AUDIO_CodingAAC; 1174 1175 err = mOMX->setParameter( 1176 mNode, OMX_IndexParamPortDefinition, &def, sizeof(def)); 1177 1178 if (err != OK) { 1179 return err; 1180 } 1181 1182 OMX_AUDIO_PARAM_AACPROFILETYPE profile; 1183 InitOMXParams(&profile); 1184 profile.nPortIndex = kPortIndexOutput; 1185 1186 err = mOMX->getParameter( 1187 mNode, OMX_IndexParamAudioAac, &profile, sizeof(profile)); 1188 1189 if (err != OK) { 1190 return err; 1191 } 1192 1193 profile.nChannels = numChannels; 1194 1195 profile.eChannelMode = 1196 (numChannels == 1) 1197 ? OMX_AUDIO_ChannelModeMono: OMX_AUDIO_ChannelModeStereo; 1198 1199 profile.nSampleRate = sampleRate; 1200 profile.nBitRate = bitRate; 1201 profile.nAudioBandWidth = 0; 1202 profile.nFrameLength = 0; 1203 profile.nAACtools = OMX_AUDIO_AACToolAll; 1204 profile.nAACERtools = OMX_AUDIO_AACERNone; 1205 profile.eAACProfile = (OMX_AUDIO_AACPROFILETYPE) aacProfile; 1206 profile.eAACStreamFormat = OMX_AUDIO_AACStreamFormatMP4FF; 1207 1208 err = mOMX->setParameter( 1209 mNode, OMX_IndexParamAudioAac, &profile, sizeof(profile)); 1210 1211 if (err != OK) { 1212 return err; 1213 } 1214 1215 return err; 1216 } 1217 1218 OMX_AUDIO_PARAM_AACPROFILETYPE profile; 1219 InitOMXParams(&profile); 1220 profile.nPortIndex = kPortIndexInput; 1221 1222 err = mOMX->getParameter( 1223 mNode, OMX_IndexParamAudioAac, &profile, sizeof(profile)); 1224 1225 if (err != OK) { 1226 return err; 1227 } 1228 1229 profile.nChannels = numChannels; 1230 profile.nSampleRate = sampleRate; 1231 1232 profile.eAACStreamFormat = 1233 isADTS 1234 ? OMX_AUDIO_AACStreamFormatMP4ADTS 1235 : OMX_AUDIO_AACStreamFormatMP4FF; 1236 1237 return mOMX->setParameter( 1238 mNode, OMX_IndexParamAudioAac, &profile, sizeof(profile)); 1239} 1240 1241static OMX_AUDIO_AMRBANDMODETYPE pickModeFromBitRate( 1242 bool isAMRWB, int32_t bps) { 1243 if (isAMRWB) { 1244 if (bps <= 6600) { 1245 return OMX_AUDIO_AMRBandModeWB0; 1246 } else if (bps <= 8850) { 1247 return OMX_AUDIO_AMRBandModeWB1; 1248 } else if (bps <= 12650) { 1249 return OMX_AUDIO_AMRBandModeWB2; 1250 } else if (bps <= 14250) { 1251 return OMX_AUDIO_AMRBandModeWB3; 1252 } else if (bps <= 15850) { 1253 return OMX_AUDIO_AMRBandModeWB4; 1254 } else if (bps <= 18250) { 1255 return OMX_AUDIO_AMRBandModeWB5; 1256 } else if (bps <= 19850) { 1257 return OMX_AUDIO_AMRBandModeWB6; 1258 } else if (bps <= 23050) { 1259 return OMX_AUDIO_AMRBandModeWB7; 1260 } 1261 1262 // 23850 bps 1263 return OMX_AUDIO_AMRBandModeWB8; 1264 } else { // AMRNB 1265 if (bps <= 4750) { 1266 return OMX_AUDIO_AMRBandModeNB0; 1267 } else if (bps <= 5150) { 1268 return OMX_AUDIO_AMRBandModeNB1; 1269 } else if (bps <= 5900) { 1270 return OMX_AUDIO_AMRBandModeNB2; 1271 } else if (bps <= 6700) { 1272 return OMX_AUDIO_AMRBandModeNB3; 1273 } else if (bps <= 7400) { 1274 return OMX_AUDIO_AMRBandModeNB4; 1275 } else if (bps <= 7950) { 1276 return OMX_AUDIO_AMRBandModeNB5; 1277 } else if (bps <= 10200) { 1278 return OMX_AUDIO_AMRBandModeNB6; 1279 } 1280 1281 // 12200 bps 1282 return OMX_AUDIO_AMRBandModeNB7; 1283 } 1284} 1285 1286status_t ACodec::setupAMRCodec(bool encoder, bool isWAMR, int32_t bitrate) { 1287 OMX_AUDIO_PARAM_AMRTYPE def; 1288 InitOMXParams(&def); 1289 def.nPortIndex = encoder ? kPortIndexOutput : kPortIndexInput; 1290 1291 status_t err = 1292 mOMX->getParameter(mNode, OMX_IndexParamAudioAmr, &def, sizeof(def)); 1293 1294 if (err != OK) { 1295 return err; 1296 } 1297 1298 def.eAMRFrameFormat = OMX_AUDIO_AMRFrameFormatFSF; 1299 def.eAMRBandMode = pickModeFromBitRate(isWAMR, bitrate); 1300 1301 err = mOMX->setParameter( 1302 mNode, OMX_IndexParamAudioAmr, &def, sizeof(def)); 1303 1304 if (err != OK) { 1305 return err; 1306 } 1307 1308 return setupRawAudioFormat( 1309 encoder ? kPortIndexInput : kPortIndexOutput, 1310 isWAMR ? 16000 : 8000 /* sampleRate */, 1311 1 /* numChannels */); 1312} 1313 1314status_t ACodec::setupG711Codec(bool encoder, int32_t numChannels) { 1315 CHECK(!encoder); // XXX TODO 1316 1317 return setupRawAudioFormat( 1318 kPortIndexInput, 8000 /* sampleRate */, numChannels); 1319} 1320 1321status_t ACodec::setupFlacCodec( 1322 bool encoder, int32_t numChannels, int32_t sampleRate, int32_t compressionLevel) { 1323 1324 if (encoder) { 1325 OMX_AUDIO_PARAM_FLACTYPE def; 1326 InitOMXParams(&def); 1327 def.nPortIndex = kPortIndexOutput; 1328 1329 // configure compression level 1330 status_t err = mOMX->getParameter(mNode, OMX_IndexParamAudioFlac, &def, sizeof(def)); 1331 if (err != OK) { 1332 ALOGE("setupFlacCodec(): Error %d getting OMX_IndexParamAudioFlac parameter", err); 1333 return err; 1334 } 1335 def.nCompressionLevel = compressionLevel; 1336 err = mOMX->setParameter(mNode, OMX_IndexParamAudioFlac, &def, sizeof(def)); 1337 if (err != OK) { 1338 ALOGE("setupFlacCodec(): Error %d setting OMX_IndexParamAudioFlac parameter", err); 1339 return err; 1340 } 1341 } 1342 1343 return setupRawAudioFormat( 1344 encoder ? kPortIndexInput : kPortIndexOutput, 1345 sampleRate, 1346 numChannels); 1347} 1348 1349status_t ACodec::setupRawAudioFormat( 1350 OMX_U32 portIndex, int32_t sampleRate, int32_t numChannels) { 1351 OMX_PARAM_PORTDEFINITIONTYPE def; 1352 InitOMXParams(&def); 1353 def.nPortIndex = portIndex; 1354 1355 status_t err = mOMX->getParameter( 1356 mNode, OMX_IndexParamPortDefinition, &def, sizeof(def)); 1357 1358 if (err != OK) { 1359 return err; 1360 } 1361 1362 def.format.audio.eEncoding = OMX_AUDIO_CodingPCM; 1363 1364 err = mOMX->setParameter( 1365 mNode, OMX_IndexParamPortDefinition, &def, sizeof(def)); 1366 1367 if (err != OK) { 1368 return err; 1369 } 1370 1371 OMX_AUDIO_PARAM_PCMMODETYPE pcmParams; 1372 InitOMXParams(&pcmParams); 1373 pcmParams.nPortIndex = portIndex; 1374 1375 err = mOMX->getParameter( 1376 mNode, OMX_IndexParamAudioPcm, &pcmParams, sizeof(pcmParams)); 1377 1378 if (err != OK) { 1379 return err; 1380 } 1381 1382 pcmParams.nChannels = numChannels; 1383 pcmParams.eNumData = OMX_NumericalDataSigned; 1384 pcmParams.bInterleaved = OMX_TRUE; 1385 pcmParams.nBitPerSample = 16; 1386 pcmParams.nSamplingRate = sampleRate; 1387 pcmParams.ePCMMode = OMX_AUDIO_PCMModeLinear; 1388 1389 if (getOMXChannelMapping(numChannels, pcmParams.eChannelMapping) != OK) { 1390 return OMX_ErrorNone; 1391 } 1392 1393 return mOMX->setParameter( 1394 mNode, OMX_IndexParamAudioPcm, &pcmParams, sizeof(pcmParams)); 1395} 1396 1397status_t ACodec::setVideoPortFormatType( 1398 OMX_U32 portIndex, 1399 OMX_VIDEO_CODINGTYPE compressionFormat, 1400 OMX_COLOR_FORMATTYPE colorFormat) { 1401 OMX_VIDEO_PARAM_PORTFORMATTYPE format; 1402 InitOMXParams(&format); 1403 format.nPortIndex = portIndex; 1404 format.nIndex = 0; 1405 bool found = false; 1406 1407 OMX_U32 index = 0; 1408 for (;;) { 1409 format.nIndex = index; 1410 status_t err = mOMX->getParameter( 1411 mNode, OMX_IndexParamVideoPortFormat, 1412 &format, sizeof(format)); 1413 1414 if (err != OK) { 1415 return err; 1416 } 1417 1418 // The following assertion is violated by TI's video decoder. 1419 // CHECK_EQ(format.nIndex, index); 1420 1421 if (!strcmp("OMX.TI.Video.encoder", mComponentName.c_str())) { 1422 if (portIndex == kPortIndexInput 1423 && colorFormat == format.eColorFormat) { 1424 // eCompressionFormat does not seem right. 1425 found = true; 1426 break; 1427 } 1428 if (portIndex == kPortIndexOutput 1429 && compressionFormat == format.eCompressionFormat) { 1430 // eColorFormat does not seem right. 1431 found = true; 1432 break; 1433 } 1434 } 1435 1436 if (format.eCompressionFormat == compressionFormat 1437 && format.eColorFormat == colorFormat) { 1438 found = true; 1439 break; 1440 } 1441 1442 ++index; 1443 } 1444 1445 if (!found) { 1446 return UNKNOWN_ERROR; 1447 } 1448 1449 status_t err = mOMX->setParameter( 1450 mNode, OMX_IndexParamVideoPortFormat, 1451 &format, sizeof(format)); 1452 1453 return err; 1454} 1455 1456status_t ACodec::setSupportedOutputFormat() { 1457 OMX_VIDEO_PARAM_PORTFORMATTYPE format; 1458 InitOMXParams(&format); 1459 format.nPortIndex = kPortIndexOutput; 1460 format.nIndex = 0; 1461 1462 status_t err = mOMX->getParameter( 1463 mNode, OMX_IndexParamVideoPortFormat, 1464 &format, sizeof(format)); 1465 CHECK_EQ(err, (status_t)OK); 1466 CHECK_EQ((int)format.eCompressionFormat, (int)OMX_VIDEO_CodingUnused); 1467 1468 return mOMX->setParameter( 1469 mNode, OMX_IndexParamVideoPortFormat, 1470 &format, sizeof(format)); 1471} 1472 1473static status_t GetVideoCodingTypeFromMime( 1474 const char *mime, OMX_VIDEO_CODINGTYPE *codingType) { 1475 if (!strcasecmp(MEDIA_MIMETYPE_VIDEO_AVC, mime)) { 1476 *codingType = OMX_VIDEO_CodingAVC; 1477 } else if (!strcasecmp(MEDIA_MIMETYPE_VIDEO_MPEG4, mime)) { 1478 *codingType = OMX_VIDEO_CodingMPEG4; 1479 } else if (!strcasecmp(MEDIA_MIMETYPE_VIDEO_H263, mime)) { 1480 *codingType = OMX_VIDEO_CodingH263; 1481 } else if (!strcasecmp(MEDIA_MIMETYPE_VIDEO_MPEG2, mime)) { 1482 *codingType = OMX_VIDEO_CodingMPEG2; 1483 } else if (!strcasecmp(MEDIA_MIMETYPE_VIDEO_VPX, mime)) { 1484 *codingType = OMX_VIDEO_CodingVPX; 1485 } else { 1486 *codingType = OMX_VIDEO_CodingUnused; 1487 return ERROR_UNSUPPORTED; 1488 } 1489 1490 return OK; 1491} 1492 1493status_t ACodec::setupVideoDecoder( 1494 const char *mime, int32_t width, int32_t height) { 1495 OMX_VIDEO_CODINGTYPE compressionFormat; 1496 status_t err = GetVideoCodingTypeFromMime(mime, &compressionFormat); 1497 1498 if (err != OK) { 1499 return err; 1500 } 1501 1502 err = setVideoPortFormatType( 1503 kPortIndexInput, compressionFormat, OMX_COLOR_FormatUnused); 1504 1505 if (err != OK) { 1506 return err; 1507 } 1508 1509 err = setSupportedOutputFormat(); 1510 1511 if (err != OK) { 1512 return err; 1513 } 1514 1515 err = setVideoFormatOnPort( 1516 kPortIndexInput, width, height, compressionFormat); 1517 1518 if (err != OK) { 1519 return err; 1520 } 1521 1522 err = setVideoFormatOnPort( 1523 kPortIndexOutput, width, height, OMX_VIDEO_CodingUnused); 1524 1525 if (err != OK) { 1526 return err; 1527 } 1528 1529 return OK; 1530} 1531 1532status_t ACodec::setupVideoEncoder(const char *mime, const sp<AMessage> &msg) { 1533 int32_t tmp; 1534 if (!msg->findInt32("color-format", &tmp)) { 1535 return INVALID_OPERATION; 1536 } 1537 1538 OMX_COLOR_FORMATTYPE colorFormat = 1539 static_cast<OMX_COLOR_FORMATTYPE>(tmp); 1540 1541 status_t err = setVideoPortFormatType( 1542 kPortIndexInput, OMX_VIDEO_CodingUnused, colorFormat); 1543 1544 if (err != OK) { 1545 ALOGE("[%s] does not support color format %d", 1546 mComponentName.c_str(), colorFormat); 1547 1548 return err; 1549 } 1550 1551 /* Input port configuration */ 1552 1553 OMX_PARAM_PORTDEFINITIONTYPE def; 1554 InitOMXParams(&def); 1555 1556 OMX_VIDEO_PORTDEFINITIONTYPE *video_def = &def.format.video; 1557 1558 def.nPortIndex = kPortIndexInput; 1559 1560 err = mOMX->getParameter( 1561 mNode, OMX_IndexParamPortDefinition, &def, sizeof(def)); 1562 1563 if (err != OK) { 1564 return err; 1565 } 1566 1567 int32_t width, height, bitrate; 1568 if (!msg->findInt32("width", &width) 1569 || !msg->findInt32("height", &height) 1570 || !msg->findInt32("bitrate", &bitrate)) { 1571 return INVALID_OPERATION; 1572 } 1573 1574 video_def->nFrameWidth = width; 1575 video_def->nFrameHeight = height; 1576 1577 int32_t stride; 1578 if (!msg->findInt32("stride", &stride)) { 1579 stride = width; 1580 } 1581 1582 video_def->nStride = stride; 1583 1584 int32_t sliceHeight; 1585 if (!msg->findInt32("slice-height", &sliceHeight)) { 1586 sliceHeight = height; 1587 } 1588 1589 video_def->nSliceHeight = sliceHeight; 1590 1591 def.nBufferSize = (video_def->nStride * video_def->nSliceHeight * 3) / 2; 1592 1593 float frameRate; 1594 if (!msg->findFloat("frame-rate", &frameRate)) { 1595 int32_t tmp; 1596 if (!msg->findInt32("frame-rate", &tmp)) { 1597 return INVALID_OPERATION; 1598 } 1599 frameRate = (float)tmp; 1600 } 1601 1602 video_def->xFramerate = (OMX_U32)(frameRate * 65536.0f); 1603 video_def->eCompressionFormat = OMX_VIDEO_CodingUnused; 1604 video_def->eColorFormat = colorFormat; 1605 1606 err = mOMX->setParameter( 1607 mNode, OMX_IndexParamPortDefinition, &def, sizeof(def)); 1608 1609 if (err != OK) { 1610 ALOGE("[%s] failed to set input port definition parameters.", 1611 mComponentName.c_str()); 1612 1613 return err; 1614 } 1615 1616 /* Output port configuration */ 1617 1618 OMX_VIDEO_CODINGTYPE compressionFormat; 1619 err = GetVideoCodingTypeFromMime(mime, &compressionFormat); 1620 1621 if (err != OK) { 1622 return err; 1623 } 1624 1625 err = setVideoPortFormatType( 1626 kPortIndexOutput, compressionFormat, OMX_COLOR_FormatUnused); 1627 1628 if (err != OK) { 1629 ALOGE("[%s] does not support compression format %d", 1630 mComponentName.c_str(), compressionFormat); 1631 1632 return err; 1633 } 1634 1635 def.nPortIndex = kPortIndexOutput; 1636 1637 err = mOMX->getParameter( 1638 mNode, OMX_IndexParamPortDefinition, &def, sizeof(def)); 1639 1640 if (err != OK) { 1641 return err; 1642 } 1643 1644 video_def->nFrameWidth = width; 1645 video_def->nFrameHeight = height; 1646 video_def->xFramerate = 0; 1647 video_def->nBitrate = bitrate; 1648 video_def->eCompressionFormat = compressionFormat; 1649 video_def->eColorFormat = OMX_COLOR_FormatUnused; 1650 1651 err = mOMX->setParameter( 1652 mNode, OMX_IndexParamPortDefinition, &def, sizeof(def)); 1653 1654 if (err != OK) { 1655 ALOGE("[%s] failed to set output port definition parameters.", 1656 mComponentName.c_str()); 1657 1658 return err; 1659 } 1660 1661 switch (compressionFormat) { 1662 case OMX_VIDEO_CodingMPEG4: 1663 err = setupMPEG4EncoderParameters(msg); 1664 break; 1665 1666 case OMX_VIDEO_CodingH263: 1667 err = setupH263EncoderParameters(msg); 1668 break; 1669 1670 case OMX_VIDEO_CodingAVC: 1671 err = setupAVCEncoderParameters(msg); 1672 break; 1673 1674 default: 1675 break; 1676 } 1677 1678 ALOGI("setupVideoEncoder succeeded"); 1679 1680 return err; 1681} 1682 1683status_t ACodec::setCyclicIntraMacroblockRefresh(const sp<AMessage> &msg, int32_t mode) { 1684 OMX_VIDEO_PARAM_INTRAREFRESHTYPE params; 1685 InitOMXParams(¶ms); 1686 params.nPortIndex = kPortIndexOutput; 1687 1688 params.eRefreshMode = static_cast<OMX_VIDEO_INTRAREFRESHTYPE>(mode); 1689 1690 if (params.eRefreshMode == OMX_VIDEO_IntraRefreshCyclic || 1691 params.eRefreshMode == OMX_VIDEO_IntraRefreshBoth) { 1692 int32_t mbs; 1693 if (!msg->findInt32("intra-refresh-CIR-mbs", &mbs)) { 1694 return INVALID_OPERATION; 1695 } 1696 params.nCirMBs = mbs; 1697 } 1698 1699 if (params.eRefreshMode == OMX_VIDEO_IntraRefreshAdaptive || 1700 params.eRefreshMode == OMX_VIDEO_IntraRefreshBoth) { 1701 int32_t mbs; 1702 if (!msg->findInt32("intra-refresh-AIR-mbs", &mbs)) { 1703 return INVALID_OPERATION; 1704 } 1705 params.nAirMBs = mbs; 1706 1707 int32_t ref; 1708 if (!msg->findInt32("intra-refresh-AIR-ref", &ref)) { 1709 return INVALID_OPERATION; 1710 } 1711 params.nAirRef = ref; 1712 } 1713 1714 status_t err = mOMX->setParameter( 1715 mNode, OMX_IndexParamVideoIntraRefresh, 1716 ¶ms, sizeof(params)); 1717 return err; 1718} 1719 1720static OMX_U32 setPFramesSpacing(int32_t iFramesInterval, int32_t frameRate) { 1721 if (iFramesInterval < 0) { 1722 return 0xFFFFFFFF; 1723 } else if (iFramesInterval == 0) { 1724 return 0; 1725 } 1726 OMX_U32 ret = frameRate * iFramesInterval; 1727 CHECK(ret > 1); 1728 return ret; 1729} 1730 1731static OMX_VIDEO_CONTROLRATETYPE getBitrateMode(const sp<AMessage> &msg) { 1732 int32_t tmp; 1733 if (!msg->findInt32("bitrate-mode", &tmp)) { 1734 return OMX_Video_ControlRateVariable; 1735 } 1736 1737 return static_cast<OMX_VIDEO_CONTROLRATETYPE>(tmp); 1738} 1739 1740status_t ACodec::setupMPEG4EncoderParameters(const sp<AMessage> &msg) { 1741 int32_t bitrate, iFrameInterval; 1742 if (!msg->findInt32("bitrate", &bitrate) 1743 || !msg->findInt32("i-frame-interval", &iFrameInterval)) { 1744 return INVALID_OPERATION; 1745 } 1746 1747 OMX_VIDEO_CONTROLRATETYPE bitrateMode = getBitrateMode(msg); 1748 1749 float frameRate; 1750 if (!msg->findFloat("frame-rate", &frameRate)) { 1751 int32_t tmp; 1752 if (!msg->findInt32("frame-rate", &tmp)) { 1753 return INVALID_OPERATION; 1754 } 1755 frameRate = (float)tmp; 1756 } 1757 1758 OMX_VIDEO_PARAM_MPEG4TYPE mpeg4type; 1759 InitOMXParams(&mpeg4type); 1760 mpeg4type.nPortIndex = kPortIndexOutput; 1761 1762 status_t err = mOMX->getParameter( 1763 mNode, OMX_IndexParamVideoMpeg4, &mpeg4type, sizeof(mpeg4type)); 1764 1765 if (err != OK) { 1766 return err; 1767 } 1768 1769 mpeg4type.nSliceHeaderSpacing = 0; 1770 mpeg4type.bSVH = OMX_FALSE; 1771 mpeg4type.bGov = OMX_FALSE; 1772 1773 mpeg4type.nAllowedPictureTypes = 1774 OMX_VIDEO_PictureTypeI | OMX_VIDEO_PictureTypeP; 1775 1776 mpeg4type.nPFrames = setPFramesSpacing(iFrameInterval, frameRate); 1777 if (mpeg4type.nPFrames == 0) { 1778 mpeg4type.nAllowedPictureTypes = OMX_VIDEO_PictureTypeI; 1779 } 1780 mpeg4type.nBFrames = 0; 1781 mpeg4type.nIDCVLCThreshold = 0; 1782 mpeg4type.bACPred = OMX_TRUE; 1783 mpeg4type.nMaxPacketSize = 256; 1784 mpeg4type.nTimeIncRes = 1000; 1785 mpeg4type.nHeaderExtension = 0; 1786 mpeg4type.bReversibleVLC = OMX_FALSE; 1787 1788 int32_t profile; 1789 if (msg->findInt32("profile", &profile)) { 1790 int32_t level; 1791 if (!msg->findInt32("level", &level)) { 1792 return INVALID_OPERATION; 1793 } 1794 1795 err = verifySupportForProfileAndLevel(profile, level); 1796 1797 if (err != OK) { 1798 return err; 1799 } 1800 1801 mpeg4type.eProfile = static_cast<OMX_VIDEO_MPEG4PROFILETYPE>(profile); 1802 mpeg4type.eLevel = static_cast<OMX_VIDEO_MPEG4LEVELTYPE>(level); 1803 } 1804 1805 err = mOMX->setParameter( 1806 mNode, OMX_IndexParamVideoMpeg4, &mpeg4type, sizeof(mpeg4type)); 1807 1808 if (err != OK) { 1809 return err; 1810 } 1811 1812 err = configureBitrate(bitrate, bitrateMode); 1813 1814 if (err != OK) { 1815 return err; 1816 } 1817 1818 return setupErrorCorrectionParameters(); 1819} 1820 1821status_t ACodec::setupH263EncoderParameters(const sp<AMessage> &msg) { 1822 int32_t bitrate, iFrameInterval; 1823 if (!msg->findInt32("bitrate", &bitrate) 1824 || !msg->findInt32("i-frame-interval", &iFrameInterval)) { 1825 return INVALID_OPERATION; 1826 } 1827 1828 OMX_VIDEO_CONTROLRATETYPE bitrateMode = getBitrateMode(msg); 1829 1830 float frameRate; 1831 if (!msg->findFloat("frame-rate", &frameRate)) { 1832 int32_t tmp; 1833 if (!msg->findInt32("frame-rate", &tmp)) { 1834 return INVALID_OPERATION; 1835 } 1836 frameRate = (float)tmp; 1837 } 1838 1839 OMX_VIDEO_PARAM_H263TYPE h263type; 1840 InitOMXParams(&h263type); 1841 h263type.nPortIndex = kPortIndexOutput; 1842 1843 status_t err = mOMX->getParameter( 1844 mNode, OMX_IndexParamVideoH263, &h263type, sizeof(h263type)); 1845 1846 if (err != OK) { 1847 return err; 1848 } 1849 1850 h263type.nAllowedPictureTypes = 1851 OMX_VIDEO_PictureTypeI | OMX_VIDEO_PictureTypeP; 1852 1853 h263type.nPFrames = setPFramesSpacing(iFrameInterval, frameRate); 1854 if (h263type.nPFrames == 0) { 1855 h263type.nAllowedPictureTypes = OMX_VIDEO_PictureTypeI; 1856 } 1857 h263type.nBFrames = 0; 1858 1859 int32_t profile; 1860 if (msg->findInt32("profile", &profile)) { 1861 int32_t level; 1862 if (!msg->findInt32("level", &level)) { 1863 return INVALID_OPERATION; 1864 } 1865 1866 err = verifySupportForProfileAndLevel(profile, level); 1867 1868 if (err != OK) { 1869 return err; 1870 } 1871 1872 h263type.eProfile = static_cast<OMX_VIDEO_H263PROFILETYPE>(profile); 1873 h263type.eLevel = static_cast<OMX_VIDEO_H263LEVELTYPE>(level); 1874 } 1875 1876 h263type.bPLUSPTYPEAllowed = OMX_FALSE; 1877 h263type.bForceRoundingTypeToZero = OMX_FALSE; 1878 h263type.nPictureHeaderRepetition = 0; 1879 h263type.nGOBHeaderInterval = 0; 1880 1881 err = mOMX->setParameter( 1882 mNode, OMX_IndexParamVideoH263, &h263type, sizeof(h263type)); 1883 1884 if (err != OK) { 1885 return err; 1886 } 1887 1888 err = configureBitrate(bitrate, bitrateMode); 1889 1890 if (err != OK) { 1891 return err; 1892 } 1893 1894 return setupErrorCorrectionParameters(); 1895} 1896 1897status_t ACodec::setupAVCEncoderParameters(const sp<AMessage> &msg) { 1898 int32_t bitrate, iFrameInterval; 1899 if (!msg->findInt32("bitrate", &bitrate) 1900 || !msg->findInt32("i-frame-interval", &iFrameInterval)) { 1901 return INVALID_OPERATION; 1902 } 1903 1904 OMX_VIDEO_CONTROLRATETYPE bitrateMode = getBitrateMode(msg); 1905 1906 float frameRate; 1907 if (!msg->findFloat("frame-rate", &frameRate)) { 1908 int32_t tmp; 1909 if (!msg->findInt32("frame-rate", &tmp)) { 1910 return INVALID_OPERATION; 1911 } 1912 frameRate = (float)tmp; 1913 } 1914 1915 status_t err = OK; 1916 int32_t intraRefreshMode = 0; 1917 if (msg->findInt32("intra-refresh-mode", &intraRefreshMode)) { 1918 err = setCyclicIntraMacroblockRefresh(msg, intraRefreshMode); 1919 if (err != OK) { 1920 ALOGE("Setting intra macroblock refresh mode (%d) failed: 0x%x", 1921 err, intraRefreshMode); 1922 return err; 1923 } 1924 } 1925 1926 OMX_VIDEO_PARAM_AVCTYPE h264type; 1927 InitOMXParams(&h264type); 1928 h264type.nPortIndex = kPortIndexOutput; 1929 1930 err = mOMX->getParameter( 1931 mNode, OMX_IndexParamVideoAvc, &h264type, sizeof(h264type)); 1932 1933 if (err != OK) { 1934 return err; 1935 } 1936 1937 h264type.nAllowedPictureTypes = 1938 OMX_VIDEO_PictureTypeI | OMX_VIDEO_PictureTypeP; 1939 1940 int32_t profile; 1941 if (msg->findInt32("profile", &profile)) { 1942 int32_t level; 1943 if (!msg->findInt32("level", &level)) { 1944 return INVALID_OPERATION; 1945 } 1946 1947 err = verifySupportForProfileAndLevel(profile, level); 1948 1949 if (err != OK) { 1950 return err; 1951 } 1952 1953 h264type.eProfile = static_cast<OMX_VIDEO_AVCPROFILETYPE>(profile); 1954 h264type.eLevel = static_cast<OMX_VIDEO_AVCLEVELTYPE>(level); 1955 } 1956 1957 // XXX 1958 if (h264type.eProfile != OMX_VIDEO_AVCProfileBaseline) { 1959 ALOGW("Use baseline profile instead of %d for AVC recording", 1960 h264type.eProfile); 1961 h264type.eProfile = OMX_VIDEO_AVCProfileBaseline; 1962 } 1963 1964 if (h264type.eProfile == OMX_VIDEO_AVCProfileBaseline) { 1965 h264type.nSliceHeaderSpacing = 0; 1966 h264type.bUseHadamard = OMX_TRUE; 1967 h264type.nRefFrames = 1; 1968 h264type.nBFrames = 0; 1969 h264type.nPFrames = setPFramesSpacing(iFrameInterval, frameRate); 1970 if (h264type.nPFrames == 0) { 1971 h264type.nAllowedPictureTypes = OMX_VIDEO_PictureTypeI; 1972 } 1973 h264type.nRefIdx10ActiveMinus1 = 0; 1974 h264type.nRefIdx11ActiveMinus1 = 0; 1975 h264type.bEntropyCodingCABAC = OMX_FALSE; 1976 h264type.bWeightedPPrediction = OMX_FALSE; 1977 h264type.bconstIpred = OMX_FALSE; 1978 h264type.bDirect8x8Inference = OMX_FALSE; 1979 h264type.bDirectSpatialTemporal = OMX_FALSE; 1980 h264type.nCabacInitIdc = 0; 1981 } 1982 1983 if (h264type.nBFrames != 0) { 1984 h264type.nAllowedPictureTypes |= OMX_VIDEO_PictureTypeB; 1985 } 1986 1987 h264type.bEnableUEP = OMX_FALSE; 1988 h264type.bEnableFMO = OMX_FALSE; 1989 h264type.bEnableASO = OMX_FALSE; 1990 h264type.bEnableRS = OMX_FALSE; 1991 h264type.bFrameMBsOnly = OMX_TRUE; 1992 h264type.bMBAFF = OMX_FALSE; 1993 h264type.eLoopFilterMode = OMX_VIDEO_AVCLoopFilterEnable; 1994 1995 err = mOMX->setParameter( 1996 mNode, OMX_IndexParamVideoAvc, &h264type, sizeof(h264type)); 1997 1998 if (err != OK) { 1999 return err; 2000 } 2001 2002 return configureBitrate(bitrate, bitrateMode); 2003} 2004 2005status_t ACodec::verifySupportForProfileAndLevel( 2006 int32_t profile, int32_t level) { 2007 OMX_VIDEO_PARAM_PROFILELEVELTYPE params; 2008 InitOMXParams(¶ms); 2009 params.nPortIndex = kPortIndexOutput; 2010 2011 for (params.nProfileIndex = 0;; ++params.nProfileIndex) { 2012 status_t err = mOMX->getParameter( 2013 mNode, 2014 OMX_IndexParamVideoProfileLevelQuerySupported, 2015 ¶ms, 2016 sizeof(params)); 2017 2018 if (err != OK) { 2019 return err; 2020 } 2021 2022 int32_t supportedProfile = static_cast<int32_t>(params.eProfile); 2023 int32_t supportedLevel = static_cast<int32_t>(params.eLevel); 2024 2025 if (profile == supportedProfile && level <= supportedLevel) { 2026 return OK; 2027 } 2028 } 2029} 2030 2031status_t ACodec::configureBitrate( 2032 int32_t bitrate, OMX_VIDEO_CONTROLRATETYPE bitrateMode) { 2033 OMX_VIDEO_PARAM_BITRATETYPE bitrateType; 2034 InitOMXParams(&bitrateType); 2035 bitrateType.nPortIndex = kPortIndexOutput; 2036 2037 status_t err = mOMX->getParameter( 2038 mNode, OMX_IndexParamVideoBitrate, 2039 &bitrateType, sizeof(bitrateType)); 2040 2041 if (err != OK) { 2042 return err; 2043 } 2044 2045 bitrateType.eControlRate = bitrateMode; 2046 bitrateType.nTargetBitrate = bitrate; 2047 2048 return mOMX->setParameter( 2049 mNode, OMX_IndexParamVideoBitrate, 2050 &bitrateType, sizeof(bitrateType)); 2051} 2052 2053status_t ACodec::setupErrorCorrectionParameters() { 2054 OMX_VIDEO_PARAM_ERRORCORRECTIONTYPE errorCorrectionType; 2055 InitOMXParams(&errorCorrectionType); 2056 errorCorrectionType.nPortIndex = kPortIndexOutput; 2057 2058 status_t err = mOMX->getParameter( 2059 mNode, OMX_IndexParamVideoErrorCorrection, 2060 &errorCorrectionType, sizeof(errorCorrectionType)); 2061 2062 if (err != OK) { 2063 return OK; // Optional feature. Ignore this failure 2064 } 2065 2066 errorCorrectionType.bEnableHEC = OMX_FALSE; 2067 errorCorrectionType.bEnableResync = OMX_TRUE; 2068 errorCorrectionType.nResynchMarkerSpacing = 256; 2069 errorCorrectionType.bEnableDataPartitioning = OMX_FALSE; 2070 errorCorrectionType.bEnableRVLC = OMX_FALSE; 2071 2072 return mOMX->setParameter( 2073 mNode, OMX_IndexParamVideoErrorCorrection, 2074 &errorCorrectionType, sizeof(errorCorrectionType)); 2075} 2076 2077status_t ACodec::setVideoFormatOnPort( 2078 OMX_U32 portIndex, 2079 int32_t width, int32_t height, OMX_VIDEO_CODINGTYPE compressionFormat) { 2080 OMX_PARAM_PORTDEFINITIONTYPE def; 2081 InitOMXParams(&def); 2082 def.nPortIndex = portIndex; 2083 2084 OMX_VIDEO_PORTDEFINITIONTYPE *video_def = &def.format.video; 2085 2086 status_t err = mOMX->getParameter( 2087 mNode, OMX_IndexParamPortDefinition, &def, sizeof(def)); 2088 2089 CHECK_EQ(err, (status_t)OK); 2090 2091 if (portIndex == kPortIndexInput) { 2092 // XXX Need a (much) better heuristic to compute input buffer sizes. 2093 const size_t X = 64 * 1024; 2094 if (def.nBufferSize < X) { 2095 def.nBufferSize = X; 2096 } 2097 } 2098 2099 CHECK_EQ((int)def.eDomain, (int)OMX_PortDomainVideo); 2100 2101 video_def->nFrameWidth = width; 2102 video_def->nFrameHeight = height; 2103 2104 if (portIndex == kPortIndexInput) { 2105 video_def->eCompressionFormat = compressionFormat; 2106 video_def->eColorFormat = OMX_COLOR_FormatUnused; 2107 } 2108 2109 err = mOMX->setParameter( 2110 mNode, OMX_IndexParamPortDefinition, &def, sizeof(def)); 2111 2112 return err; 2113} 2114 2115status_t ACodec::initNativeWindow() { 2116 if (mNativeWindow != NULL) { 2117 return mOMX->enableGraphicBuffers(mNode, kPortIndexOutput, OMX_TRUE); 2118 } 2119 2120 mOMX->enableGraphicBuffers(mNode, kPortIndexOutput, OMX_FALSE); 2121 return OK; 2122} 2123 2124size_t ACodec::countBuffersOwnedByComponent(OMX_U32 portIndex) const { 2125 size_t n = 0; 2126 2127 for (size_t i = 0; i < mBuffers[portIndex].size(); ++i) { 2128 const BufferInfo &info = mBuffers[portIndex].itemAt(i); 2129 2130 if (info.mStatus == BufferInfo::OWNED_BY_COMPONENT) { 2131 ++n; 2132 } 2133 } 2134 2135 return n; 2136} 2137 2138bool ACodec::allYourBuffersAreBelongToUs( 2139 OMX_U32 portIndex) { 2140 for (size_t i = 0; i < mBuffers[portIndex].size(); ++i) { 2141 BufferInfo *info = &mBuffers[portIndex].editItemAt(i); 2142 2143 if (info->mStatus != BufferInfo::OWNED_BY_US 2144 && info->mStatus != BufferInfo::OWNED_BY_NATIVE_WINDOW) { 2145 ALOGV("[%s] Buffer %p on port %ld still has status %d", 2146 mComponentName.c_str(), 2147 info->mBufferID, portIndex, info->mStatus); 2148 return false; 2149 } 2150 } 2151 2152 return true; 2153} 2154 2155bool ACodec::allYourBuffersAreBelongToUs() { 2156 return allYourBuffersAreBelongToUs(kPortIndexInput) 2157 && allYourBuffersAreBelongToUs(kPortIndexOutput); 2158} 2159 2160void ACodec::deferMessage(const sp<AMessage> &msg) { 2161 bool wasEmptyBefore = mDeferredQueue.empty(); 2162 mDeferredQueue.push_back(msg); 2163} 2164 2165void ACodec::processDeferredMessages() { 2166 List<sp<AMessage> > queue = mDeferredQueue; 2167 mDeferredQueue.clear(); 2168 2169 List<sp<AMessage> >::iterator it = queue.begin(); 2170 while (it != queue.end()) { 2171 onMessageReceived(*it++); 2172 } 2173} 2174 2175void ACodec::sendFormatChange() { 2176 sp<AMessage> notify = mNotify->dup(); 2177 notify->setInt32("what", kWhatOutputFormatChanged); 2178 2179 OMX_PARAM_PORTDEFINITIONTYPE def; 2180 InitOMXParams(&def); 2181 def.nPortIndex = kPortIndexOutput; 2182 2183 CHECK_EQ(mOMX->getParameter( 2184 mNode, OMX_IndexParamPortDefinition, &def, sizeof(def)), 2185 (status_t)OK); 2186 2187 CHECK_EQ((int)def.eDir, (int)OMX_DirOutput); 2188 2189 switch (def.eDomain) { 2190 case OMX_PortDomainVideo: 2191 { 2192 OMX_VIDEO_PORTDEFINITIONTYPE *videoDef = &def.format.video; 2193 2194 notify->setString("mime", MEDIA_MIMETYPE_VIDEO_RAW); 2195 notify->setInt32("width", videoDef->nFrameWidth); 2196 notify->setInt32("height", videoDef->nFrameHeight); 2197 notify->setInt32("stride", videoDef->nStride); 2198 notify->setInt32("slice-height", videoDef->nSliceHeight); 2199 notify->setInt32("color-format", videoDef->eColorFormat); 2200 2201 OMX_CONFIG_RECTTYPE rect; 2202 InitOMXParams(&rect); 2203 rect.nPortIndex = kPortIndexOutput; 2204 2205 if (mOMX->getConfig( 2206 mNode, OMX_IndexConfigCommonOutputCrop, 2207 &rect, sizeof(rect)) != OK) { 2208 rect.nLeft = 0; 2209 rect.nTop = 0; 2210 rect.nWidth = videoDef->nFrameWidth; 2211 rect.nHeight = videoDef->nFrameHeight; 2212 } 2213 2214 CHECK_GE(rect.nLeft, 0); 2215 CHECK_GE(rect.nTop, 0); 2216 CHECK_GE(rect.nWidth, 0u); 2217 CHECK_GE(rect.nHeight, 0u); 2218 CHECK_LE(rect.nLeft + rect.nWidth - 1, videoDef->nFrameWidth); 2219 CHECK_LE(rect.nTop + rect.nHeight - 1, videoDef->nFrameHeight); 2220 2221 notify->setRect( 2222 "crop", 2223 rect.nLeft, 2224 rect.nTop, 2225 rect.nLeft + rect.nWidth - 1, 2226 rect.nTop + rect.nHeight - 1); 2227 2228 if (mNativeWindow != NULL) { 2229 android_native_rect_t crop; 2230 crop.left = rect.nLeft; 2231 crop.top = rect.nTop; 2232 crop.right = rect.nLeft + rect.nWidth; 2233 crop.bottom = rect.nTop + rect.nHeight; 2234 2235 CHECK_EQ(0, native_window_set_crop( 2236 mNativeWindow.get(), &crop)); 2237 } 2238 break; 2239 } 2240 2241 case OMX_PortDomainAudio: 2242 { 2243 OMX_AUDIO_PORTDEFINITIONTYPE *audioDef = &def.format.audio; 2244 CHECK_EQ((int)audioDef->eEncoding, (int)OMX_AUDIO_CodingPCM); 2245 2246 OMX_AUDIO_PARAM_PCMMODETYPE params; 2247 InitOMXParams(¶ms); 2248 params.nPortIndex = kPortIndexOutput; 2249 2250 CHECK_EQ(mOMX->getParameter( 2251 mNode, OMX_IndexParamAudioPcm, 2252 ¶ms, sizeof(params)), 2253 (status_t)OK); 2254 2255 CHECK(params.nChannels == 1 || params.bInterleaved); 2256 CHECK_EQ(params.nBitPerSample, 16u); 2257 CHECK_EQ((int)params.eNumData, (int)OMX_NumericalDataSigned); 2258 CHECK_EQ((int)params.ePCMMode, (int)OMX_AUDIO_PCMModeLinear); 2259 2260 notify->setString("mime", MEDIA_MIMETYPE_AUDIO_RAW); 2261 notify->setInt32("channel-count", params.nChannels); 2262 notify->setInt32("sample-rate", params.nSamplingRate); 2263 if (mEncoderDelay + mEncoderPadding) { 2264 size_t frameSize = params.nChannels * sizeof(int16_t); 2265 if (mSkipCutBuffer != NULL) { 2266 size_t prevbufsize = mSkipCutBuffer->size(); 2267 if (prevbufsize != 0) { 2268 ALOGW("Replacing SkipCutBuffer holding %d bytes", prevbufsize); 2269 } 2270 } 2271 mSkipCutBuffer = new SkipCutBuffer(mEncoderDelay * frameSize, 2272 mEncoderPadding * frameSize); 2273 } 2274 2275 if (mChannelMaskPresent) { 2276 notify->setInt32("channel-mask", mChannelMask); 2277 } 2278 2279 break; 2280 } 2281 2282 default: 2283 TRESPASS(); 2284 } 2285 2286 notify->post(); 2287 2288 mSentFormat = true; 2289} 2290 2291void ACodec::signalError(OMX_ERRORTYPE error, status_t internalError) { 2292 sp<AMessage> notify = mNotify->dup(); 2293 notify->setInt32("what", ACodec::kWhatError); 2294 notify->setInt32("omx-error", error); 2295 notify->setInt32("err", internalError); 2296 notify->post(); 2297} 2298 2299status_t ACodec::pushBlankBuffersToNativeWindow() { 2300 status_t err = NO_ERROR; 2301 ANativeWindowBuffer* anb = NULL; 2302 int numBufs = 0; 2303 int minUndequeuedBufs = 0; 2304 2305 // We need to reconnect to the ANativeWindow as a CPU client to ensure that 2306 // no frames get dropped by SurfaceFlinger assuming that these are video 2307 // frames. 2308 err = native_window_api_disconnect(mNativeWindow.get(), 2309 NATIVE_WINDOW_API_MEDIA); 2310 if (err != NO_ERROR) { 2311 ALOGE("error pushing blank frames: api_disconnect failed: %s (%d)", 2312 strerror(-err), -err); 2313 return err; 2314 } 2315 2316 err = native_window_api_connect(mNativeWindow.get(), 2317 NATIVE_WINDOW_API_CPU); 2318 if (err != NO_ERROR) { 2319 ALOGE("error pushing blank frames: api_connect failed: %s (%d)", 2320 strerror(-err), -err); 2321 return err; 2322 } 2323 2324 err = native_window_set_buffers_geometry(mNativeWindow.get(), 1, 1, 2325 HAL_PIXEL_FORMAT_RGBX_8888); 2326 if (err != NO_ERROR) { 2327 ALOGE("error pushing blank frames: set_buffers_geometry failed: %s (%d)", 2328 strerror(-err), -err); 2329 goto error; 2330 } 2331 2332 err = native_window_set_usage(mNativeWindow.get(), 2333 GRALLOC_USAGE_SW_WRITE_OFTEN); 2334 if (err != NO_ERROR) { 2335 ALOGE("error pushing blank frames: set_usage failed: %s (%d)", 2336 strerror(-err), -err); 2337 goto error; 2338 } 2339 2340 err = mNativeWindow->query(mNativeWindow.get(), 2341 NATIVE_WINDOW_MIN_UNDEQUEUED_BUFFERS, &minUndequeuedBufs); 2342 if (err != NO_ERROR) { 2343 ALOGE("error pushing blank frames: MIN_UNDEQUEUED_BUFFERS query " 2344 "failed: %s (%d)", strerror(-err), -err); 2345 goto error; 2346 } 2347 2348 numBufs = minUndequeuedBufs + 1; 2349 err = native_window_set_buffer_count(mNativeWindow.get(), numBufs); 2350 if (err != NO_ERROR) { 2351 ALOGE("error pushing blank frames: set_buffer_count failed: %s (%d)", 2352 strerror(-err), -err); 2353 goto error; 2354 } 2355 2356 // We push numBufs + 1 buffers to ensure that we've drawn into the same 2357 // buffer twice. This should guarantee that the buffer has been displayed 2358 // on the screen and then been replaced, so an previous video frames are 2359 // guaranteed NOT to be currently displayed. 2360 for (int i = 0; i < numBufs + 1; i++) { 2361 int fenceFd = -1; 2362 err = native_window_dequeue_buffer_and_wait(mNativeWindow.get(), &anb); 2363 if (err != NO_ERROR) { 2364 ALOGE("error pushing blank frames: dequeueBuffer failed: %s (%d)", 2365 strerror(-err), -err); 2366 goto error; 2367 } 2368 2369 sp<GraphicBuffer> buf(new GraphicBuffer(anb, false)); 2370 2371 // Fill the buffer with the a 1x1 checkerboard pattern ;) 2372 uint32_t* img = NULL; 2373 err = buf->lock(GRALLOC_USAGE_SW_WRITE_OFTEN, (void**)(&img)); 2374 if (err != NO_ERROR) { 2375 ALOGE("error pushing blank frames: lock failed: %s (%d)", 2376 strerror(-err), -err); 2377 goto error; 2378 } 2379 2380 *img = 0; 2381 2382 err = buf->unlock(); 2383 if (err != NO_ERROR) { 2384 ALOGE("error pushing blank frames: unlock failed: %s (%d)", 2385 strerror(-err), -err); 2386 goto error; 2387 } 2388 2389 err = mNativeWindow->queueBuffer(mNativeWindow.get(), 2390 buf->getNativeBuffer(), -1); 2391 if (err != NO_ERROR) { 2392 ALOGE("error pushing blank frames: queueBuffer failed: %s (%d)", 2393 strerror(-err), -err); 2394 goto error; 2395 } 2396 2397 anb = NULL; 2398 } 2399 2400error: 2401 2402 if (err != NO_ERROR) { 2403 // Clean up after an error. 2404 if (anb != NULL) { 2405 mNativeWindow->cancelBuffer(mNativeWindow.get(), anb, -1); 2406 } 2407 2408 native_window_api_disconnect(mNativeWindow.get(), 2409 NATIVE_WINDOW_API_CPU); 2410 native_window_api_connect(mNativeWindow.get(), 2411 NATIVE_WINDOW_API_MEDIA); 2412 2413 return err; 2414 } else { 2415 // Clean up after success. 2416 err = native_window_api_disconnect(mNativeWindow.get(), 2417 NATIVE_WINDOW_API_CPU); 2418 if (err != NO_ERROR) { 2419 ALOGE("error pushing blank frames: api_disconnect failed: %s (%d)", 2420 strerror(-err), -err); 2421 return err; 2422 } 2423 2424 err = native_window_api_connect(mNativeWindow.get(), 2425 NATIVE_WINDOW_API_MEDIA); 2426 if (err != NO_ERROR) { 2427 ALOGE("error pushing blank frames: api_connect failed: %s (%d)", 2428 strerror(-err), -err); 2429 return err; 2430 } 2431 2432 return NO_ERROR; 2433 } 2434} 2435 2436//////////////////////////////////////////////////////////////////////////////// 2437 2438ACodec::PortDescription::PortDescription() { 2439} 2440 2441status_t ACodec::requestIDRFrame() { 2442 if (!mIsEncoder) { 2443 return ERROR_UNSUPPORTED; 2444 } 2445 2446 OMX_CONFIG_INTRAREFRESHVOPTYPE params; 2447 InitOMXParams(¶ms); 2448 2449 params.nPortIndex = kPortIndexOutput; 2450 params.IntraRefreshVOP = OMX_TRUE; 2451 2452 return mOMX->setConfig( 2453 mNode, 2454 OMX_IndexConfigVideoIntraVOPRefresh, 2455 ¶ms, 2456 sizeof(params)); 2457} 2458 2459void ACodec::PortDescription::addBuffer( 2460 IOMX::buffer_id id, const sp<ABuffer> &buffer) { 2461 mBufferIDs.push_back(id); 2462 mBuffers.push_back(buffer); 2463} 2464 2465size_t ACodec::PortDescription::countBuffers() { 2466 return mBufferIDs.size(); 2467} 2468 2469IOMX::buffer_id ACodec::PortDescription::bufferIDAt(size_t index) const { 2470 return mBufferIDs.itemAt(index); 2471} 2472 2473sp<ABuffer> ACodec::PortDescription::bufferAt(size_t index) const { 2474 return mBuffers.itemAt(index); 2475} 2476 2477//////////////////////////////////////////////////////////////////////////////// 2478 2479ACodec::BaseState::BaseState(ACodec *codec, const sp<AState> &parentState) 2480 : AState(parentState), 2481 mCodec(codec) { 2482} 2483 2484ACodec::BaseState::PortMode ACodec::BaseState::getPortMode(OMX_U32 portIndex) { 2485 return KEEP_BUFFERS; 2486} 2487 2488bool ACodec::BaseState::onMessageReceived(const sp<AMessage> &msg) { 2489 switch (msg->what()) { 2490 case kWhatInputBufferFilled: 2491 { 2492 onInputBufferFilled(msg); 2493 break; 2494 } 2495 2496 case kWhatOutputBufferDrained: 2497 { 2498 onOutputBufferDrained(msg); 2499 break; 2500 } 2501 2502 case ACodec::kWhatOMXMessage: 2503 { 2504 return onOMXMessage(msg); 2505 } 2506 2507 case ACodec::kWhatCreateInputSurface: 2508 case ACodec::kWhatSignalEndOfInputStream: 2509 { 2510 ALOGE("Message 0x%x was not handled", msg->what()); 2511 mCodec->signalError(OMX_ErrorUndefined, INVALID_OPERATION); 2512 return true; 2513 } 2514 2515 case ACodec::kWhatOMXDied: 2516 { 2517 ALOGE("OMX/mediaserver died, signalling error!"); 2518 mCodec->signalError(OMX_ErrorResourcesLost, DEAD_OBJECT); 2519 break; 2520 } 2521 2522 default: 2523 return false; 2524 } 2525 2526 return true; 2527} 2528 2529bool ACodec::BaseState::onOMXMessage(const sp<AMessage> &msg) { 2530 int32_t type; 2531 CHECK(msg->findInt32("type", &type)); 2532 2533 IOMX::node_id nodeID; 2534 CHECK(msg->findPointer("node", &nodeID)); 2535 CHECK_EQ(nodeID, mCodec->mNode); 2536 2537 switch (type) { 2538 case omx_message::EVENT: 2539 { 2540 int32_t event, data1, data2; 2541 CHECK(msg->findInt32("event", &event)); 2542 CHECK(msg->findInt32("data1", &data1)); 2543 CHECK(msg->findInt32("data2", &data2)); 2544 2545 if (event == OMX_EventCmdComplete 2546 && data1 == OMX_CommandFlush 2547 && data2 == (int32_t)OMX_ALL) { 2548 // Use of this notification is not consistent across 2549 // implementations. We'll drop this notification and rely 2550 // on flush-complete notifications on the individual port 2551 // indices instead. 2552 2553 return true; 2554 } 2555 2556 return onOMXEvent( 2557 static_cast<OMX_EVENTTYPE>(event), 2558 static_cast<OMX_U32>(data1), 2559 static_cast<OMX_U32>(data2)); 2560 } 2561 2562 case omx_message::EMPTY_BUFFER_DONE: 2563 { 2564 IOMX::buffer_id bufferID; 2565 CHECK(msg->findPointer("buffer", &bufferID)); 2566 2567 return onOMXEmptyBufferDone(bufferID); 2568 } 2569 2570 case omx_message::FILL_BUFFER_DONE: 2571 { 2572 IOMX::buffer_id bufferID; 2573 CHECK(msg->findPointer("buffer", &bufferID)); 2574 2575 int32_t rangeOffset, rangeLength, flags; 2576 int64_t timeUs; 2577 void *platformPrivate; 2578 void *dataPtr; 2579 2580 CHECK(msg->findInt32("range_offset", &rangeOffset)); 2581 CHECK(msg->findInt32("range_length", &rangeLength)); 2582 CHECK(msg->findInt32("flags", &flags)); 2583 CHECK(msg->findInt64("timestamp", &timeUs)); 2584 CHECK(msg->findPointer("platform_private", &platformPrivate)); 2585 CHECK(msg->findPointer("data_ptr", &dataPtr)); 2586 2587 return onOMXFillBufferDone( 2588 bufferID, 2589 (size_t)rangeOffset, (size_t)rangeLength, 2590 (OMX_U32)flags, 2591 timeUs, 2592 platformPrivate, 2593 dataPtr); 2594 } 2595 2596 default: 2597 TRESPASS(); 2598 break; 2599 } 2600} 2601 2602bool ACodec::BaseState::onOMXEvent( 2603 OMX_EVENTTYPE event, OMX_U32 data1, OMX_U32 data2) { 2604 if (event != OMX_EventError) { 2605 ALOGV("[%s] EVENT(%d, 0x%08lx, 0x%08lx)", 2606 mCodec->mComponentName.c_str(), event, data1, data2); 2607 2608 return false; 2609 } 2610 2611 ALOGE("[%s] ERROR(0x%08lx)", mCodec->mComponentName.c_str(), data1); 2612 2613 mCodec->signalError((OMX_ERRORTYPE)data1); 2614 2615 return true; 2616} 2617 2618bool ACodec::BaseState::onOMXEmptyBufferDone(IOMX::buffer_id bufferID) { 2619 ALOGV("[%s] onOMXEmptyBufferDone %p", 2620 mCodec->mComponentName.c_str(), bufferID); 2621 2622 BufferInfo *info = 2623 mCodec->findBufferByID(kPortIndexInput, bufferID); 2624 2625 CHECK_EQ((int)info->mStatus, (int)BufferInfo::OWNED_BY_COMPONENT); 2626 info->mStatus = BufferInfo::OWNED_BY_US; 2627 2628 const sp<AMessage> &bufferMeta = info->mData->meta(); 2629 void *mediaBuffer; 2630 if (bufferMeta->findPointer("mediaBuffer", &mediaBuffer) 2631 && mediaBuffer != NULL) { 2632 // We're in "store-metadata-in-buffers" mode, the underlying 2633 // OMX component had access to data that's implicitly refcounted 2634 // by this "mediaBuffer" object. Now that the OMX component has 2635 // told us that it's done with the input buffer, we can decrement 2636 // the mediaBuffer's reference count. 2637 2638 ALOGV("releasing mbuf %p", mediaBuffer); 2639 2640 ((MediaBuffer *)mediaBuffer)->release(); 2641 mediaBuffer = NULL; 2642 2643 bufferMeta->setPointer("mediaBuffer", NULL); 2644 } 2645 2646 PortMode mode = getPortMode(kPortIndexInput); 2647 2648 switch (mode) { 2649 case KEEP_BUFFERS: 2650 break; 2651 2652 case RESUBMIT_BUFFERS: 2653 postFillThisBuffer(info); 2654 break; 2655 2656 default: 2657 { 2658 CHECK_EQ((int)mode, (int)FREE_BUFFERS); 2659 TRESPASS(); // Not currently used 2660 break; 2661 } 2662 } 2663 2664 return true; 2665} 2666 2667void ACodec::BaseState::postFillThisBuffer(BufferInfo *info) { 2668 if (mCodec->mPortEOS[kPortIndexInput]) { 2669 return; 2670 } 2671 2672 CHECK_EQ((int)info->mStatus, (int)BufferInfo::OWNED_BY_US); 2673 2674 sp<AMessage> notify = mCodec->mNotify->dup(); 2675 notify->setInt32("what", ACodec::kWhatFillThisBuffer); 2676 notify->setPointer("buffer-id", info->mBufferID); 2677 2678 info->mData->meta()->clear(); 2679 notify->setBuffer("buffer", info->mData); 2680 2681 sp<AMessage> reply = new AMessage(kWhatInputBufferFilled, mCodec->id()); 2682 reply->setPointer("buffer-id", info->mBufferID); 2683 2684 notify->setMessage("reply", reply); 2685 2686 notify->post(); 2687 2688 info->mStatus = BufferInfo::OWNED_BY_UPSTREAM; 2689} 2690 2691void ACodec::BaseState::onInputBufferFilled(const sp<AMessage> &msg) { 2692 IOMX::buffer_id bufferID; 2693 CHECK(msg->findPointer("buffer-id", &bufferID)); 2694 2695 sp<ABuffer> buffer; 2696 int32_t err = OK; 2697 bool eos = false; 2698 2699 if (!msg->findBuffer("buffer", &buffer)) { 2700 CHECK(msg->findInt32("err", &err)); 2701 2702 ALOGV("[%s] saw error %d instead of an input buffer", 2703 mCodec->mComponentName.c_str(), err); 2704 2705 buffer.clear(); 2706 2707 eos = true; 2708 } 2709 2710 int32_t tmp; 2711 if (buffer != NULL && buffer->meta()->findInt32("eos", &tmp) && tmp) { 2712 eos = true; 2713 err = ERROR_END_OF_STREAM; 2714 } 2715 2716 BufferInfo *info = mCodec->findBufferByID(kPortIndexInput, bufferID); 2717 CHECK_EQ((int)info->mStatus, (int)BufferInfo::OWNED_BY_UPSTREAM); 2718 2719 info->mStatus = BufferInfo::OWNED_BY_US; 2720 2721 PortMode mode = getPortMode(kPortIndexInput); 2722 2723 switch (mode) { 2724 case KEEP_BUFFERS: 2725 { 2726 if (eos) { 2727 if (!mCodec->mPortEOS[kPortIndexInput]) { 2728 mCodec->mPortEOS[kPortIndexInput] = true; 2729 mCodec->mInputEOSResult = err; 2730 } 2731 } 2732 break; 2733 } 2734 2735 case RESUBMIT_BUFFERS: 2736 { 2737 if (buffer != NULL && !mCodec->mPortEOS[kPortIndexInput]) { 2738 int64_t timeUs; 2739 CHECK(buffer->meta()->findInt64("timeUs", &timeUs)); 2740 2741 OMX_U32 flags = OMX_BUFFERFLAG_ENDOFFRAME; 2742 2743 int32_t isCSD; 2744 if (buffer->meta()->findInt32("csd", &isCSD) && isCSD != 0) { 2745 flags |= OMX_BUFFERFLAG_CODECCONFIG; 2746 } 2747 2748 if (eos) { 2749 flags |= OMX_BUFFERFLAG_EOS; 2750 } 2751 2752 if (buffer != info->mData) { 2753 ALOGV("[%s] Needs to copy input data for buffer %p. (%p != %p)", 2754 mCodec->mComponentName.c_str(), 2755 bufferID, 2756 buffer.get(), info->mData.get()); 2757 2758 CHECK_LE(buffer->size(), info->mData->capacity()); 2759 memcpy(info->mData->data(), buffer->data(), buffer->size()); 2760 } 2761 2762 if (flags & OMX_BUFFERFLAG_CODECCONFIG) { 2763 ALOGV("[%s] calling emptyBuffer %p w/ codec specific data", 2764 mCodec->mComponentName.c_str(), bufferID); 2765 } else if (flags & OMX_BUFFERFLAG_EOS) { 2766 ALOGV("[%s] calling emptyBuffer %p w/ EOS", 2767 mCodec->mComponentName.c_str(), bufferID); 2768 } else { 2769#if TRACK_BUFFER_TIMING 2770 ALOGI("[%s] calling emptyBuffer %p w/ time %lld us", 2771 mCodec->mComponentName.c_str(), bufferID, timeUs); 2772#else 2773 ALOGV("[%s] calling emptyBuffer %p w/ time %lld us", 2774 mCodec->mComponentName.c_str(), bufferID, timeUs); 2775#endif 2776 } 2777 2778#if TRACK_BUFFER_TIMING 2779 ACodec::BufferStats stats; 2780 stats.mEmptyBufferTimeUs = ALooper::GetNowUs(); 2781 stats.mFillBufferDoneTimeUs = -1ll; 2782 mCodec->mBufferStats.add(timeUs, stats); 2783#endif 2784 2785 CHECK_EQ(mCodec->mOMX->emptyBuffer( 2786 mCodec->mNode, 2787 bufferID, 2788 0, 2789 buffer->size(), 2790 flags, 2791 timeUs), 2792 (status_t)OK); 2793 2794 info->mStatus = BufferInfo::OWNED_BY_COMPONENT; 2795 2796 if (!eos) { 2797 getMoreInputDataIfPossible(); 2798 } else { 2799 ALOGV("[%s] Signalled EOS on the input port", 2800 mCodec->mComponentName.c_str()); 2801 2802 mCodec->mPortEOS[kPortIndexInput] = true; 2803 mCodec->mInputEOSResult = err; 2804 } 2805 } else if (!mCodec->mPortEOS[kPortIndexInput]) { 2806 if (err != ERROR_END_OF_STREAM) { 2807 ALOGV("[%s] Signalling EOS on the input port " 2808 "due to error %d", 2809 mCodec->mComponentName.c_str(), err); 2810 } else { 2811 ALOGV("[%s] Signalling EOS on the input port", 2812 mCodec->mComponentName.c_str()); 2813 } 2814 2815 ALOGV("[%s] calling emptyBuffer %p signalling EOS", 2816 mCodec->mComponentName.c_str(), bufferID); 2817 2818 CHECK_EQ(mCodec->mOMX->emptyBuffer( 2819 mCodec->mNode, 2820 bufferID, 2821 0, 2822 0, 2823 OMX_BUFFERFLAG_EOS, 2824 0), 2825 (status_t)OK); 2826 2827 info->mStatus = BufferInfo::OWNED_BY_COMPONENT; 2828 2829 mCodec->mPortEOS[kPortIndexInput] = true; 2830 mCodec->mInputEOSResult = err; 2831 } 2832 break; 2833 2834 default: 2835 CHECK_EQ((int)mode, (int)FREE_BUFFERS); 2836 break; 2837 } 2838 } 2839} 2840 2841void ACodec::BaseState::getMoreInputDataIfPossible() { 2842 if (mCodec->mPortEOS[kPortIndexInput]) { 2843 return; 2844 } 2845 2846 BufferInfo *eligible = NULL; 2847 2848 for (size_t i = 0; i < mCodec->mBuffers[kPortIndexInput].size(); ++i) { 2849 BufferInfo *info = &mCodec->mBuffers[kPortIndexInput].editItemAt(i); 2850 2851#if 0 2852 if (info->mStatus == BufferInfo::OWNED_BY_UPSTREAM) { 2853 // There's already a "read" pending. 2854 return; 2855 } 2856#endif 2857 2858 if (info->mStatus == BufferInfo::OWNED_BY_US) { 2859 eligible = info; 2860 } 2861 } 2862 2863 if (eligible == NULL) { 2864 return; 2865 } 2866 2867 postFillThisBuffer(eligible); 2868} 2869 2870bool ACodec::BaseState::onOMXFillBufferDone( 2871 IOMX::buffer_id bufferID, 2872 size_t rangeOffset, size_t rangeLength, 2873 OMX_U32 flags, 2874 int64_t timeUs, 2875 void *platformPrivate, 2876 void *dataPtr) { 2877 ALOGV("[%s] onOMXFillBufferDone %p time %lld us, flags = 0x%08lx", 2878 mCodec->mComponentName.c_str(), bufferID, timeUs, flags); 2879 2880 ssize_t index; 2881 2882#if TRACK_BUFFER_TIMING 2883 index = mCodec->mBufferStats.indexOfKey(timeUs); 2884 if (index >= 0) { 2885 ACodec::BufferStats *stats = &mCodec->mBufferStats.editValueAt(index); 2886 stats->mFillBufferDoneTimeUs = ALooper::GetNowUs(); 2887 2888 ALOGI("frame PTS %lld: %lld", 2889 timeUs, 2890 stats->mFillBufferDoneTimeUs - stats->mEmptyBufferTimeUs); 2891 2892 mCodec->mBufferStats.removeItemsAt(index); 2893 stats = NULL; 2894 } 2895#endif 2896 2897 BufferInfo *info = 2898 mCodec->findBufferByID(kPortIndexOutput, bufferID, &index); 2899 2900 CHECK_EQ((int)info->mStatus, (int)BufferInfo::OWNED_BY_COMPONENT); 2901 2902 info->mStatus = BufferInfo::OWNED_BY_US; 2903 2904 PortMode mode = getPortMode(kPortIndexOutput); 2905 2906 switch (mode) { 2907 case KEEP_BUFFERS: 2908 break; 2909 2910 case RESUBMIT_BUFFERS: 2911 { 2912 if (rangeLength == 0 && !(flags & OMX_BUFFERFLAG_EOS)) { 2913 ALOGV("[%s] calling fillBuffer %p", 2914 mCodec->mComponentName.c_str(), info->mBufferID); 2915 2916 CHECK_EQ(mCodec->mOMX->fillBuffer( 2917 mCodec->mNode, info->mBufferID), 2918 (status_t)OK); 2919 2920 info->mStatus = BufferInfo::OWNED_BY_COMPONENT; 2921 break; 2922 } 2923 2924 if (!mCodec->mIsEncoder && !mCodec->mSentFormat) { 2925 mCodec->sendFormatChange(); 2926 } 2927 2928 info->mData->setRange(rangeOffset, rangeLength); 2929#if 0 2930 if (mCodec->mNativeWindow == NULL) { 2931 if (IsIDR(info->mData)) { 2932 ALOGI("IDR frame"); 2933 } 2934 } 2935#endif 2936 2937 if (mCodec->mSkipCutBuffer != NULL) { 2938 mCodec->mSkipCutBuffer->submit(info->mData); 2939 } 2940 info->mData->meta()->setInt64("timeUs", timeUs); 2941 2942 sp<AMessage> notify = mCodec->mNotify->dup(); 2943 notify->setInt32("what", ACodec::kWhatDrainThisBuffer); 2944 notify->setPointer("buffer-id", info->mBufferID); 2945 notify->setBuffer("buffer", info->mData); 2946 notify->setInt32("flags", flags); 2947 2948 sp<AMessage> reply = 2949 new AMessage(kWhatOutputBufferDrained, mCodec->id()); 2950 2951 reply->setPointer("buffer-id", info->mBufferID); 2952 2953 notify->setMessage("reply", reply); 2954 2955 notify->post(); 2956 2957 info->mStatus = BufferInfo::OWNED_BY_DOWNSTREAM; 2958 2959 if (flags & OMX_BUFFERFLAG_EOS) { 2960 ALOGV("[%s] saw output EOS", mCodec->mComponentName.c_str()); 2961 2962 sp<AMessage> notify = mCodec->mNotify->dup(); 2963 notify->setInt32("what", ACodec::kWhatEOS); 2964 notify->setInt32("err", mCodec->mInputEOSResult); 2965 notify->post(); 2966 2967 mCodec->mPortEOS[kPortIndexOutput] = true; 2968 } 2969 break; 2970 } 2971 2972 default: 2973 { 2974 CHECK_EQ((int)mode, (int)FREE_BUFFERS); 2975 2976 CHECK_EQ((status_t)OK, 2977 mCodec->freeBuffer(kPortIndexOutput, index)); 2978 break; 2979 } 2980 } 2981 2982 return true; 2983} 2984 2985void ACodec::BaseState::onOutputBufferDrained(const sp<AMessage> &msg) { 2986 IOMX::buffer_id bufferID; 2987 CHECK(msg->findPointer("buffer-id", &bufferID)); 2988 2989 ssize_t index; 2990 BufferInfo *info = 2991 mCodec->findBufferByID(kPortIndexOutput, bufferID, &index); 2992 CHECK_EQ((int)info->mStatus, (int)BufferInfo::OWNED_BY_DOWNSTREAM); 2993 2994 int32_t render; 2995 if (mCodec->mNativeWindow != NULL 2996 && msg->findInt32("render", &render) && render != 0) { 2997 // The client wants this buffer to be rendered. 2998 2999 status_t err; 3000 if ((err = mCodec->mNativeWindow->queueBuffer( 3001 mCodec->mNativeWindow.get(), 3002 info->mGraphicBuffer.get(), -1)) == OK) { 3003 info->mStatus = BufferInfo::OWNED_BY_NATIVE_WINDOW; 3004 } else { 3005 mCodec->signalError(OMX_ErrorUndefined, err); 3006 info->mStatus = BufferInfo::OWNED_BY_US; 3007 } 3008 } else { 3009 info->mStatus = BufferInfo::OWNED_BY_US; 3010 } 3011 3012 PortMode mode = getPortMode(kPortIndexOutput); 3013 3014 switch (mode) { 3015 case KEEP_BUFFERS: 3016 { 3017 // XXX fishy, revisit!!! What about the FREE_BUFFERS case below? 3018 3019 if (info->mStatus == BufferInfo::OWNED_BY_NATIVE_WINDOW) { 3020 // We cannot resubmit the buffer we just rendered, dequeue 3021 // the spare instead. 3022 3023 info = mCodec->dequeueBufferFromNativeWindow(); 3024 } 3025 break; 3026 } 3027 3028 case RESUBMIT_BUFFERS: 3029 { 3030 if (!mCodec->mPortEOS[kPortIndexOutput]) { 3031 if (info->mStatus == BufferInfo::OWNED_BY_NATIVE_WINDOW) { 3032 // We cannot resubmit the buffer we just rendered, dequeue 3033 // the spare instead. 3034 3035 info = mCodec->dequeueBufferFromNativeWindow(); 3036 } 3037 3038 if (info != NULL) { 3039 ALOGV("[%s] calling fillBuffer %p", 3040 mCodec->mComponentName.c_str(), info->mBufferID); 3041 3042 CHECK_EQ(mCodec->mOMX->fillBuffer(mCodec->mNode, info->mBufferID), 3043 (status_t)OK); 3044 3045 info->mStatus = BufferInfo::OWNED_BY_COMPONENT; 3046 } 3047 } 3048 break; 3049 } 3050 3051 default: 3052 { 3053 CHECK_EQ((int)mode, (int)FREE_BUFFERS); 3054 3055 CHECK_EQ((status_t)OK, 3056 mCodec->freeBuffer(kPortIndexOutput, index)); 3057 break; 3058 } 3059 } 3060} 3061 3062//////////////////////////////////////////////////////////////////////////////// 3063 3064ACodec::UninitializedState::UninitializedState(ACodec *codec) 3065 : BaseState(codec) { 3066} 3067 3068void ACodec::UninitializedState::stateEntered() { 3069 ALOGV("Now uninitialized"); 3070 3071 if (mDeathNotifier != NULL) { 3072 mCodec->mOMX->asBinder()->unlinkToDeath(mDeathNotifier); 3073 mDeathNotifier.clear(); 3074 } 3075 3076 mCodec->mNativeWindow.clear(); 3077 mCodec->mNode = NULL; 3078 mCodec->mOMX.clear(); 3079 mCodec->mQuirks = 0; 3080 mCodec->mFlags = 0; 3081 mCodec->mComponentName.clear(); 3082} 3083 3084bool ACodec::UninitializedState::onMessageReceived(const sp<AMessage> &msg) { 3085 bool handled = false; 3086 3087 switch (msg->what()) { 3088 case ACodec::kWhatSetup: 3089 { 3090 onSetup(msg); 3091 3092 handled = true; 3093 break; 3094 } 3095 3096 case ACodec::kWhatAllocateComponent: 3097 { 3098 onAllocateComponent(msg); 3099 handled = true; 3100 break; 3101 } 3102 3103 case ACodec::kWhatShutdown: 3104 { 3105 int32_t keepComponentAllocated; 3106 CHECK(msg->findInt32( 3107 "keepComponentAllocated", &keepComponentAllocated)); 3108 CHECK(!keepComponentAllocated); 3109 3110 sp<AMessage> notify = mCodec->mNotify->dup(); 3111 notify->setInt32("what", ACodec::kWhatShutdownCompleted); 3112 notify->post(); 3113 3114 handled = true; 3115 break; 3116 } 3117 3118 case ACodec::kWhatFlush: 3119 { 3120 sp<AMessage> notify = mCodec->mNotify->dup(); 3121 notify->setInt32("what", ACodec::kWhatFlushCompleted); 3122 notify->post(); 3123 3124 handled = true; 3125 break; 3126 } 3127 3128 default: 3129 return BaseState::onMessageReceived(msg); 3130 } 3131 3132 return handled; 3133} 3134 3135void ACodec::UninitializedState::onSetup( 3136 const sp<AMessage> &msg) { 3137 if (onAllocateComponent(msg) 3138 && mCodec->mLoadedState->onConfigureComponent(msg)) { 3139 mCodec->mLoadedState->onStart(); 3140 } 3141} 3142 3143bool ACodec::UninitializedState::onAllocateComponent(const sp<AMessage> &msg) { 3144 ALOGV("onAllocateComponent"); 3145 3146 CHECK(mCodec->mNode == NULL); 3147 3148 OMXClient client; 3149 CHECK_EQ(client.connect(), (status_t)OK); 3150 3151 sp<IOMX> omx = client.interface(); 3152 3153 sp<AMessage> notify = new AMessage(kWhatOMXDied, mCodec->id()); 3154 3155 mDeathNotifier = new DeathNotifier(notify); 3156 if (omx->asBinder()->linkToDeath(mDeathNotifier) != OK) { 3157 // This was a local binder, if it dies so do we, we won't care 3158 // about any notifications in the afterlife. 3159 mDeathNotifier.clear(); 3160 } 3161 3162 Vector<OMXCodec::CodecNameAndQuirks> matchingCodecs; 3163 3164 AString mime; 3165 3166 AString componentName; 3167 uint32_t quirks = 0; 3168 if (msg->findString("componentName", &componentName)) { 3169 ssize_t index = matchingCodecs.add(); 3170 OMXCodec::CodecNameAndQuirks *entry = &matchingCodecs.editItemAt(index); 3171 entry->mName = String8(componentName.c_str()); 3172 3173 if (!OMXCodec::findCodecQuirks( 3174 componentName.c_str(), &entry->mQuirks)) { 3175 entry->mQuirks = 0; 3176 } 3177 } else { 3178 CHECK(msg->findString("mime", &mime)); 3179 3180 int32_t encoder; 3181 if (!msg->findInt32("encoder", &encoder)) { 3182 encoder = false; 3183 } 3184 3185 OMXCodec::findMatchingCodecs( 3186 mime.c_str(), 3187 encoder, // createEncoder 3188 NULL, // matchComponentName 3189 0, // flags 3190 &matchingCodecs); 3191 } 3192 3193 sp<CodecObserver> observer = new CodecObserver; 3194 IOMX::node_id node = NULL; 3195 3196 for (size_t matchIndex = 0; matchIndex < matchingCodecs.size(); 3197 ++matchIndex) { 3198 componentName = matchingCodecs.itemAt(matchIndex).mName.string(); 3199 quirks = matchingCodecs.itemAt(matchIndex).mQuirks; 3200 3201 pid_t tid = androidGetTid(); 3202 int prevPriority = androidGetThreadPriority(tid); 3203 androidSetThreadPriority(tid, ANDROID_PRIORITY_FOREGROUND); 3204 status_t err = omx->allocateNode(componentName.c_str(), observer, &node); 3205 androidSetThreadPriority(tid, prevPriority); 3206 3207 if (err == OK) { 3208 break; 3209 } 3210 3211 node = NULL; 3212 } 3213 3214 if (node == NULL) { 3215 if (!mime.empty()) { 3216 ALOGE("Unable to instantiate a decoder for type '%s'.", 3217 mime.c_str()); 3218 } else { 3219 ALOGE("Unable to instantiate decoder '%s'.", componentName.c_str()); 3220 } 3221 3222 mCodec->signalError(OMX_ErrorComponentNotFound); 3223 return false; 3224 } 3225 3226 notify = new AMessage(kWhatOMXMessage, mCodec->id()); 3227 observer->setNotificationMessage(notify); 3228 3229 mCodec->mComponentName = componentName; 3230 mCodec->mFlags = 0; 3231 3232 if (componentName.endsWith(".secure")) { 3233 mCodec->mFlags |= kFlagIsSecure; 3234 } 3235 3236 mCodec->mQuirks = quirks; 3237 mCodec->mOMX = omx; 3238 mCodec->mNode = node; 3239 3240 { 3241 sp<AMessage> notify = mCodec->mNotify->dup(); 3242 notify->setInt32("what", ACodec::kWhatComponentAllocated); 3243 notify->setString("componentName", mCodec->mComponentName.c_str()); 3244 notify->post(); 3245 } 3246 3247 mCodec->changeState(mCodec->mLoadedState); 3248 3249 return true; 3250} 3251 3252//////////////////////////////////////////////////////////////////////////////// 3253 3254ACodec::LoadedState::LoadedState(ACodec *codec) 3255 : BaseState(codec) { 3256} 3257 3258void ACodec::LoadedState::stateEntered() { 3259 ALOGV("[%s] Now Loaded", mCodec->mComponentName.c_str()); 3260 3261 mCodec->mPortEOS[kPortIndexInput] = 3262 mCodec->mPortEOS[kPortIndexOutput] = false; 3263 3264 mCodec->mInputEOSResult = OK; 3265 3266 if (mCodec->mShutdownInProgress) { 3267 bool keepComponentAllocated = mCodec->mKeepComponentAllocated; 3268 3269 mCodec->mShutdownInProgress = false; 3270 mCodec->mKeepComponentAllocated = false; 3271 3272 onShutdown(keepComponentAllocated); 3273 } 3274} 3275 3276void ACodec::LoadedState::onShutdown(bool keepComponentAllocated) { 3277 if (!keepComponentAllocated) { 3278 CHECK_EQ(mCodec->mOMX->freeNode(mCodec->mNode), (status_t)OK); 3279 3280 mCodec->changeState(mCodec->mUninitializedState); 3281 } 3282 3283 sp<AMessage> notify = mCodec->mNotify->dup(); 3284 notify->setInt32("what", ACodec::kWhatShutdownCompleted); 3285 notify->post(); 3286} 3287 3288bool ACodec::LoadedState::onMessageReceived(const sp<AMessage> &msg) { 3289 bool handled = false; 3290 3291 switch (msg->what()) { 3292 case ACodec::kWhatConfigureComponent: 3293 { 3294 onConfigureComponent(msg); 3295 handled = true; 3296 break; 3297 } 3298 3299 case ACodec::kWhatCreateInputSurface: 3300 { 3301 onCreateInputSurface(msg); 3302 handled = true; 3303 break; 3304 } 3305 3306 case ACodec::kWhatStart: 3307 { 3308 onStart(); 3309 handled = true; 3310 break; 3311 } 3312 3313 case ACodec::kWhatShutdown: 3314 { 3315 int32_t keepComponentAllocated; 3316 CHECK(msg->findInt32( 3317 "keepComponentAllocated", &keepComponentAllocated)); 3318 3319 onShutdown(keepComponentAllocated); 3320 3321 handled = true; 3322 break; 3323 } 3324 3325 case ACodec::kWhatFlush: 3326 { 3327 sp<AMessage> notify = mCodec->mNotify->dup(); 3328 notify->setInt32("what", ACodec::kWhatFlushCompleted); 3329 notify->post(); 3330 3331 handled = true; 3332 break; 3333 } 3334 3335 default: 3336 return BaseState::onMessageReceived(msg); 3337 } 3338 3339 return handled; 3340} 3341 3342bool ACodec::LoadedState::onConfigureComponent( 3343 const sp<AMessage> &msg) { 3344 ALOGV("onConfigureComponent"); 3345 3346 CHECK(mCodec->mNode != NULL); 3347 3348 AString mime; 3349 CHECK(msg->findString("mime", &mime)); 3350 3351 status_t err = mCodec->configureCodec(mime.c_str(), msg); 3352 3353 if (err != OK) { 3354 ALOGE("[%s] configureCodec returning error %d", 3355 mCodec->mComponentName.c_str(), err); 3356 3357 mCodec->signalError(OMX_ErrorUndefined, err); 3358 return false; 3359 } 3360 3361 sp<RefBase> obj; 3362 if (msg->findObject("native-window", &obj) 3363 && strncmp("OMX.google.", mCodec->mComponentName.c_str(), 11)) { 3364 sp<NativeWindowWrapper> nativeWindow( 3365 static_cast<NativeWindowWrapper *>(obj.get())); 3366 CHECK(nativeWindow != NULL); 3367 mCodec->mNativeWindow = nativeWindow->getNativeWindow(); 3368 3369 native_window_set_scaling_mode( 3370 mCodec->mNativeWindow.get(), 3371 NATIVE_WINDOW_SCALING_MODE_SCALE_TO_WINDOW); 3372 } 3373 CHECK_EQ((status_t)OK, mCodec->initNativeWindow()); 3374 3375 { 3376 sp<AMessage> notify = mCodec->mNotify->dup(); 3377 notify->setInt32("what", ACodec::kWhatComponentConfigured); 3378 notify->post(); 3379 } 3380 3381 return true; 3382} 3383 3384void ACodec::LoadedState::onCreateInputSurface( 3385 const sp<AMessage> &msg) { 3386 ALOGV("onCreateInputSurface"); 3387 3388 sp<AMessage> notify = mCodec->mNotify->dup(); 3389 notify->setInt32("what", ACodec::kWhatInputSurfaceCreated); 3390 3391 sp<IGraphicBufferProducer> bufferProducer; 3392 status_t err; 3393 3394 err = mCodec->mOMX->createInputSurface(mCodec->mNode, kPortIndexInput, 3395 &bufferProducer); 3396 if (err == OK) { 3397 notify->setObject("input-surface", 3398 new BufferProducerWrapper(bufferProducer)); 3399 } else { 3400 // Can't use mCodec->signalError() here -- MediaCodec won't forward 3401 // the error through because it's in the "configured" state. We 3402 // send a kWhatInputSurfaceCreated with an error value instead. 3403 ALOGE("[%s] onCreateInputSurface returning error %d", 3404 mCodec->mComponentName.c_str(), err); 3405 notify->setInt32("err", err); 3406 } 3407 notify->post(); 3408} 3409 3410void ACodec::LoadedState::onStart() { 3411 ALOGV("onStart"); 3412 3413 CHECK_EQ(mCodec->mOMX->sendCommand( 3414 mCodec->mNode, OMX_CommandStateSet, OMX_StateIdle), 3415 (status_t)OK); 3416 3417 mCodec->changeState(mCodec->mLoadedToIdleState); 3418} 3419 3420//////////////////////////////////////////////////////////////////////////////// 3421 3422ACodec::LoadedToIdleState::LoadedToIdleState(ACodec *codec) 3423 : BaseState(codec) { 3424} 3425 3426void ACodec::LoadedToIdleState::stateEntered() { 3427 ALOGV("[%s] Now Loaded->Idle", mCodec->mComponentName.c_str()); 3428 3429 status_t err; 3430 if ((err = allocateBuffers()) != OK) { 3431 ALOGE("Failed to allocate buffers after transitioning to IDLE state " 3432 "(error 0x%08x)", 3433 err); 3434 3435 mCodec->signalError(OMX_ErrorUndefined, err); 3436 3437 mCodec->changeState(mCodec->mLoadedState); 3438 } 3439} 3440 3441status_t ACodec::LoadedToIdleState::allocateBuffers() { 3442 status_t err = mCodec->allocateBuffersOnPort(kPortIndexInput); 3443 3444 if (err != OK) { 3445 return err; 3446 } 3447 3448 return mCodec->allocateBuffersOnPort(kPortIndexOutput); 3449} 3450 3451bool ACodec::LoadedToIdleState::onMessageReceived(const sp<AMessage> &msg) { 3452 switch (msg->what()) { 3453 case kWhatShutdown: 3454 { 3455 mCodec->deferMessage(msg); 3456 return true; 3457 } 3458 3459 case kWhatSignalEndOfInputStream: 3460 { 3461 mCodec->onSignalEndOfInputStream(); 3462 return true; 3463 } 3464 3465 case kWhatResume: 3466 { 3467 // We'll be active soon enough. 3468 return true; 3469 } 3470 3471 case kWhatFlush: 3472 { 3473 // We haven't even started yet, so we're flushed alright... 3474 sp<AMessage> notify = mCodec->mNotify->dup(); 3475 notify->setInt32("what", ACodec::kWhatFlushCompleted); 3476 notify->post(); 3477 return true; 3478 } 3479 3480 default: 3481 return BaseState::onMessageReceived(msg); 3482 } 3483} 3484 3485bool ACodec::LoadedToIdleState::onOMXEvent( 3486 OMX_EVENTTYPE event, OMX_U32 data1, OMX_U32 data2) { 3487 switch (event) { 3488 case OMX_EventCmdComplete: 3489 { 3490 CHECK_EQ(data1, (OMX_U32)OMX_CommandStateSet); 3491 CHECK_EQ(data2, (OMX_U32)OMX_StateIdle); 3492 3493 CHECK_EQ(mCodec->mOMX->sendCommand( 3494 mCodec->mNode, OMX_CommandStateSet, OMX_StateExecuting), 3495 (status_t)OK); 3496 3497 mCodec->changeState(mCodec->mIdleToExecutingState); 3498 3499 return true; 3500 } 3501 3502 default: 3503 return BaseState::onOMXEvent(event, data1, data2); 3504 } 3505} 3506 3507//////////////////////////////////////////////////////////////////////////////// 3508 3509ACodec::IdleToExecutingState::IdleToExecutingState(ACodec *codec) 3510 : BaseState(codec) { 3511} 3512 3513void ACodec::IdleToExecutingState::stateEntered() { 3514 ALOGV("[%s] Now Idle->Executing", mCodec->mComponentName.c_str()); 3515} 3516 3517bool ACodec::IdleToExecutingState::onMessageReceived(const sp<AMessage> &msg) { 3518 switch (msg->what()) { 3519 case kWhatShutdown: 3520 { 3521 mCodec->deferMessage(msg); 3522 return true; 3523 } 3524 3525 case kWhatResume: 3526 { 3527 // We'll be active soon enough. 3528 return true; 3529 } 3530 3531 case kWhatFlush: 3532 { 3533 // We haven't even started yet, so we're flushed alright... 3534 sp<AMessage> notify = mCodec->mNotify->dup(); 3535 notify->setInt32("what", ACodec::kWhatFlushCompleted); 3536 notify->post(); 3537 3538 return true; 3539 } 3540 3541 case kWhatSignalEndOfInputStream: 3542 { 3543 mCodec->onSignalEndOfInputStream(); 3544 return true; 3545 } 3546 3547 default: 3548 return BaseState::onMessageReceived(msg); 3549 } 3550} 3551 3552bool ACodec::IdleToExecutingState::onOMXEvent( 3553 OMX_EVENTTYPE event, OMX_U32 data1, OMX_U32 data2) { 3554 switch (event) { 3555 case OMX_EventCmdComplete: 3556 { 3557 CHECK_EQ(data1, (OMX_U32)OMX_CommandStateSet); 3558 CHECK_EQ(data2, (OMX_U32)OMX_StateExecuting); 3559 3560 mCodec->mExecutingState->resume(); 3561 mCodec->changeState(mCodec->mExecutingState); 3562 3563 return true; 3564 } 3565 3566 default: 3567 return BaseState::onOMXEvent(event, data1, data2); 3568 } 3569} 3570 3571//////////////////////////////////////////////////////////////////////////////// 3572 3573ACodec::ExecutingState::ExecutingState(ACodec *codec) 3574 : BaseState(codec), 3575 mActive(false) { 3576} 3577 3578ACodec::BaseState::PortMode ACodec::ExecutingState::getPortMode( 3579 OMX_U32 portIndex) { 3580 return RESUBMIT_BUFFERS; 3581} 3582 3583void ACodec::ExecutingState::submitOutputBuffers() { 3584 for (size_t i = 0; i < mCodec->mBuffers[kPortIndexOutput].size(); ++i) { 3585 BufferInfo *info = &mCodec->mBuffers[kPortIndexOutput].editItemAt(i); 3586 3587 if (mCodec->mNativeWindow != NULL) { 3588 CHECK(info->mStatus == BufferInfo::OWNED_BY_US 3589 || info->mStatus == BufferInfo::OWNED_BY_NATIVE_WINDOW); 3590 3591 if (info->mStatus == BufferInfo::OWNED_BY_NATIVE_WINDOW) { 3592 continue; 3593 } 3594 } else { 3595 CHECK_EQ((int)info->mStatus, (int)BufferInfo::OWNED_BY_US); 3596 } 3597 3598 ALOGV("[%s] calling fillBuffer %p", 3599 mCodec->mComponentName.c_str(), info->mBufferID); 3600 3601 CHECK_EQ(mCodec->mOMX->fillBuffer(mCodec->mNode, info->mBufferID), 3602 (status_t)OK); 3603 3604 info->mStatus = BufferInfo::OWNED_BY_COMPONENT; 3605 } 3606} 3607 3608void ACodec::ExecutingState::resume() { 3609 if (mActive) { 3610 ALOGV("[%s] We're already active, no need to resume.", 3611 mCodec->mComponentName.c_str()); 3612 3613 return; 3614 } 3615 3616 submitOutputBuffers(); 3617 3618 // Post the first input buffer. 3619 CHECK_GT(mCodec->mBuffers[kPortIndexInput].size(), 0u); 3620 BufferInfo *info = &mCodec->mBuffers[kPortIndexInput].editItemAt(0); 3621 3622 postFillThisBuffer(info); 3623 3624 mActive = true; 3625} 3626 3627void ACodec::ExecutingState::stateEntered() { 3628 ALOGV("[%s] Now Executing", mCodec->mComponentName.c_str()); 3629 3630 mCodec->processDeferredMessages(); 3631} 3632 3633bool ACodec::ExecutingState::onMessageReceived(const sp<AMessage> &msg) { 3634 bool handled = false; 3635 3636 switch (msg->what()) { 3637 case kWhatShutdown: 3638 { 3639 int32_t keepComponentAllocated; 3640 CHECK(msg->findInt32( 3641 "keepComponentAllocated", &keepComponentAllocated)); 3642 3643 mCodec->mShutdownInProgress = true; 3644 mCodec->mKeepComponentAllocated = keepComponentAllocated; 3645 3646 mActive = false; 3647 3648 CHECK_EQ(mCodec->mOMX->sendCommand( 3649 mCodec->mNode, OMX_CommandStateSet, OMX_StateIdle), 3650 (status_t)OK); 3651 3652 mCodec->changeState(mCodec->mExecutingToIdleState); 3653 3654 handled = true; 3655 break; 3656 } 3657 3658 case kWhatFlush: 3659 { 3660 ALOGV("[%s] ExecutingState flushing now " 3661 "(codec owns %d/%d input, %d/%d output).", 3662 mCodec->mComponentName.c_str(), 3663 mCodec->countBuffersOwnedByComponent(kPortIndexInput), 3664 mCodec->mBuffers[kPortIndexInput].size(), 3665 mCodec->countBuffersOwnedByComponent(kPortIndexOutput), 3666 mCodec->mBuffers[kPortIndexOutput].size()); 3667 3668 mActive = false; 3669 3670 CHECK_EQ(mCodec->mOMX->sendCommand( 3671 mCodec->mNode, OMX_CommandFlush, OMX_ALL), 3672 (status_t)OK); 3673 3674 mCodec->changeState(mCodec->mFlushingState); 3675 handled = true; 3676 break; 3677 } 3678 3679 case kWhatResume: 3680 { 3681 resume(); 3682 3683 handled = true; 3684 break; 3685 } 3686 3687 case kWhatRequestIDRFrame: 3688 { 3689 status_t err = mCodec->requestIDRFrame(); 3690 if (err != OK) { 3691 ALOGW("Requesting an IDR frame failed."); 3692 } 3693 3694 handled = true; 3695 break; 3696 } 3697 3698 case kWhatSetParameters: 3699 { 3700 sp<AMessage> params; 3701 CHECK(msg->findMessage("params", ¶ms)); 3702 3703 status_t err = mCodec->setParameters(params); 3704 3705 sp<AMessage> reply; 3706 if (msg->findMessage("reply", &reply)) { 3707 reply->setInt32("err", err); 3708 reply->post(); 3709 } 3710 3711 handled = true; 3712 break; 3713 } 3714 3715 case ACodec::kWhatSignalEndOfInputStream: 3716 { 3717 mCodec->onSignalEndOfInputStream(); 3718 handled = true; 3719 break; 3720 } 3721 3722 default: 3723 handled = BaseState::onMessageReceived(msg); 3724 break; 3725 } 3726 3727 return handled; 3728} 3729 3730status_t ACodec::setParameters(const sp<AMessage> ¶ms) { 3731 int32_t videoBitrate; 3732 if (params->findInt32("videoBitrate", &videoBitrate)) { 3733 OMX_VIDEO_CONFIG_BITRATETYPE configParams; 3734 InitOMXParams(&configParams); 3735 configParams.nPortIndex = kPortIndexOutput; 3736 configParams.nEncodeBitrate = videoBitrate; 3737 3738 status_t err = mOMX->setConfig( 3739 mNode, 3740 OMX_IndexConfigVideoBitrate, 3741 &configParams, 3742 sizeof(configParams)); 3743 3744 if (err != OK) { 3745 ALOGE("setConfig(OMX_IndexConfigVideoBitrate, %d) failed w/ err %d", 3746 videoBitrate, err); 3747 3748 return err; 3749 } 3750 } 3751 3752 return OK; 3753} 3754 3755void ACodec::onSignalEndOfInputStream() { 3756 sp<AMessage> notify = mNotify->dup(); 3757 notify->setInt32("what", ACodec::kWhatSignaledInputEOS); 3758 3759 status_t err = mOMX->signalEndOfInputStream(mNode); 3760 if (err != OK) { 3761 notify->setInt32("err", err); 3762 } 3763 notify->post(); 3764} 3765 3766bool ACodec::ExecutingState::onOMXEvent( 3767 OMX_EVENTTYPE event, OMX_U32 data1, OMX_U32 data2) { 3768 switch (event) { 3769 case OMX_EventPortSettingsChanged: 3770 { 3771 CHECK_EQ(data1, (OMX_U32)kPortIndexOutput); 3772 3773 if (data2 == 0 || data2 == OMX_IndexParamPortDefinition) { 3774 CHECK_EQ(mCodec->mOMX->sendCommand( 3775 mCodec->mNode, 3776 OMX_CommandPortDisable, kPortIndexOutput), 3777 (status_t)OK); 3778 3779 mCodec->freeOutputBuffersNotOwnedByComponent(); 3780 3781 mCodec->changeState(mCodec->mOutputPortSettingsChangedState); 3782 } else if (data2 == OMX_IndexConfigCommonOutputCrop) { 3783 mCodec->mSentFormat = false; 3784 } else { 3785 ALOGV("[%s] OMX_EventPortSettingsChanged 0x%08lx", 3786 mCodec->mComponentName.c_str(), data2); 3787 } 3788 3789 return true; 3790 } 3791 3792 case OMX_EventBufferFlag: 3793 { 3794 return true; 3795 } 3796 3797 default: 3798 return BaseState::onOMXEvent(event, data1, data2); 3799 } 3800} 3801 3802//////////////////////////////////////////////////////////////////////////////// 3803 3804ACodec::OutputPortSettingsChangedState::OutputPortSettingsChangedState( 3805 ACodec *codec) 3806 : BaseState(codec) { 3807} 3808 3809ACodec::BaseState::PortMode ACodec::OutputPortSettingsChangedState::getPortMode( 3810 OMX_U32 portIndex) { 3811 if (portIndex == kPortIndexOutput) { 3812 return FREE_BUFFERS; 3813 } 3814 3815 CHECK_EQ(portIndex, (OMX_U32)kPortIndexInput); 3816 3817 return RESUBMIT_BUFFERS; 3818} 3819 3820bool ACodec::OutputPortSettingsChangedState::onMessageReceived( 3821 const sp<AMessage> &msg) { 3822 bool handled = false; 3823 3824 switch (msg->what()) { 3825 case kWhatFlush: 3826 case kWhatShutdown: 3827 case kWhatResume: 3828 { 3829 if (msg->what() == kWhatResume) { 3830 ALOGV("[%s] Deferring resume", mCodec->mComponentName.c_str()); 3831 } 3832 3833 mCodec->deferMessage(msg); 3834 handled = true; 3835 break; 3836 } 3837 3838 default: 3839 handled = BaseState::onMessageReceived(msg); 3840 break; 3841 } 3842 3843 return handled; 3844} 3845 3846void ACodec::OutputPortSettingsChangedState::stateEntered() { 3847 ALOGV("[%s] Now handling output port settings change", 3848 mCodec->mComponentName.c_str()); 3849} 3850 3851bool ACodec::OutputPortSettingsChangedState::onOMXEvent( 3852 OMX_EVENTTYPE event, OMX_U32 data1, OMX_U32 data2) { 3853 switch (event) { 3854 case OMX_EventCmdComplete: 3855 { 3856 if (data1 == (OMX_U32)OMX_CommandPortDisable) { 3857 CHECK_EQ(data2, (OMX_U32)kPortIndexOutput); 3858 3859 ALOGV("[%s] Output port now disabled.", 3860 mCodec->mComponentName.c_str()); 3861 3862 CHECK(mCodec->mBuffers[kPortIndexOutput].isEmpty()); 3863 mCodec->mDealer[kPortIndexOutput].clear(); 3864 3865 CHECK_EQ(mCodec->mOMX->sendCommand( 3866 mCodec->mNode, OMX_CommandPortEnable, kPortIndexOutput), 3867 (status_t)OK); 3868 3869 status_t err; 3870 if ((err = mCodec->allocateBuffersOnPort( 3871 kPortIndexOutput)) != OK) { 3872 ALOGE("Failed to allocate output port buffers after " 3873 "port reconfiguration (error 0x%08x)", 3874 err); 3875 3876 mCodec->signalError(OMX_ErrorUndefined, err); 3877 3878 // This is technically not correct, but appears to be 3879 // the only way to free the component instance. 3880 // Controlled transitioning from excecuting->idle 3881 // and idle->loaded seem impossible probably because 3882 // the output port never finishes re-enabling. 3883 mCodec->mShutdownInProgress = true; 3884 mCodec->mKeepComponentAllocated = false; 3885 mCodec->changeState(mCodec->mLoadedState); 3886 } 3887 3888 return true; 3889 } else if (data1 == (OMX_U32)OMX_CommandPortEnable) { 3890 CHECK_EQ(data2, (OMX_U32)kPortIndexOutput); 3891 3892 mCodec->mSentFormat = false; 3893 3894 ALOGV("[%s] Output port now reenabled.", 3895 mCodec->mComponentName.c_str()); 3896 3897 if (mCodec->mExecutingState->active()) { 3898 mCodec->mExecutingState->submitOutputBuffers(); 3899 } 3900 3901 mCodec->changeState(mCodec->mExecutingState); 3902 3903 return true; 3904 } 3905 3906 return false; 3907 } 3908 3909 default: 3910 return false; 3911 } 3912} 3913 3914//////////////////////////////////////////////////////////////////////////////// 3915 3916ACodec::ExecutingToIdleState::ExecutingToIdleState(ACodec *codec) 3917 : BaseState(codec), 3918 mComponentNowIdle(false) { 3919} 3920 3921bool ACodec::ExecutingToIdleState::onMessageReceived(const sp<AMessage> &msg) { 3922 bool handled = false; 3923 3924 switch (msg->what()) { 3925 case kWhatFlush: 3926 { 3927 // Don't send me a flush request if you previously wanted me 3928 // to shutdown. 3929 TRESPASS(); 3930 break; 3931 } 3932 3933 case kWhatShutdown: 3934 { 3935 // We're already doing that... 3936 3937 handled = true; 3938 break; 3939 } 3940 3941 default: 3942 handled = BaseState::onMessageReceived(msg); 3943 break; 3944 } 3945 3946 return handled; 3947} 3948 3949void ACodec::ExecutingToIdleState::stateEntered() { 3950 ALOGV("[%s] Now Executing->Idle", mCodec->mComponentName.c_str()); 3951 3952 mComponentNowIdle = false; 3953 mCodec->mSentFormat = false; 3954} 3955 3956bool ACodec::ExecutingToIdleState::onOMXEvent( 3957 OMX_EVENTTYPE event, OMX_U32 data1, OMX_U32 data2) { 3958 switch (event) { 3959 case OMX_EventCmdComplete: 3960 { 3961 CHECK_EQ(data1, (OMX_U32)OMX_CommandStateSet); 3962 CHECK_EQ(data2, (OMX_U32)OMX_StateIdle); 3963 3964 mComponentNowIdle = true; 3965 3966 changeStateIfWeOwnAllBuffers(); 3967 3968 return true; 3969 } 3970 3971 case OMX_EventPortSettingsChanged: 3972 case OMX_EventBufferFlag: 3973 { 3974 // We're shutting down and don't care about this anymore. 3975 return true; 3976 } 3977 3978 default: 3979 return BaseState::onOMXEvent(event, data1, data2); 3980 } 3981} 3982 3983void ACodec::ExecutingToIdleState::changeStateIfWeOwnAllBuffers() { 3984 if (mComponentNowIdle && mCodec->allYourBuffersAreBelongToUs()) { 3985 CHECK_EQ(mCodec->mOMX->sendCommand( 3986 mCodec->mNode, OMX_CommandStateSet, OMX_StateLoaded), 3987 (status_t)OK); 3988 3989 CHECK_EQ(mCodec->freeBuffersOnPort(kPortIndexInput), (status_t)OK); 3990 CHECK_EQ(mCodec->freeBuffersOnPort(kPortIndexOutput), (status_t)OK); 3991 3992 if (mCodec->mFlags & kFlagIsSecure && mCodec->mNativeWindow != NULL) { 3993 // We push enough 1x1 blank buffers to ensure that one of 3994 // them has made it to the display. This allows the OMX 3995 // component teardown to zero out any protected buffers 3996 // without the risk of scanning out one of those buffers. 3997 mCodec->pushBlankBuffersToNativeWindow(); 3998 } 3999 4000 mCodec->changeState(mCodec->mIdleToLoadedState); 4001 } 4002} 4003 4004void ACodec::ExecutingToIdleState::onInputBufferFilled( 4005 const sp<AMessage> &msg) { 4006 BaseState::onInputBufferFilled(msg); 4007 4008 changeStateIfWeOwnAllBuffers(); 4009} 4010 4011void ACodec::ExecutingToIdleState::onOutputBufferDrained( 4012 const sp<AMessage> &msg) { 4013 BaseState::onOutputBufferDrained(msg); 4014 4015 changeStateIfWeOwnAllBuffers(); 4016} 4017 4018//////////////////////////////////////////////////////////////////////////////// 4019 4020ACodec::IdleToLoadedState::IdleToLoadedState(ACodec *codec) 4021 : BaseState(codec) { 4022} 4023 4024bool ACodec::IdleToLoadedState::onMessageReceived(const sp<AMessage> &msg) { 4025 bool handled = false; 4026 4027 switch (msg->what()) { 4028 case kWhatShutdown: 4029 { 4030 // We're already doing that... 4031 4032 handled = true; 4033 break; 4034 } 4035 4036 case kWhatFlush: 4037 { 4038 // Don't send me a flush request if you previously wanted me 4039 // to shutdown. 4040 TRESPASS(); 4041 break; 4042 } 4043 4044 default: 4045 handled = BaseState::onMessageReceived(msg); 4046 break; 4047 } 4048 4049 return handled; 4050} 4051 4052void ACodec::IdleToLoadedState::stateEntered() { 4053 ALOGV("[%s] Now Idle->Loaded", mCodec->mComponentName.c_str()); 4054} 4055 4056bool ACodec::IdleToLoadedState::onOMXEvent( 4057 OMX_EVENTTYPE event, OMX_U32 data1, OMX_U32 data2) { 4058 switch (event) { 4059 case OMX_EventCmdComplete: 4060 { 4061 CHECK_EQ(data1, (OMX_U32)OMX_CommandStateSet); 4062 CHECK_EQ(data2, (OMX_U32)OMX_StateLoaded); 4063 4064 mCodec->changeState(mCodec->mLoadedState); 4065 4066 return true; 4067 } 4068 4069 default: 4070 return BaseState::onOMXEvent(event, data1, data2); 4071 } 4072} 4073 4074//////////////////////////////////////////////////////////////////////////////// 4075 4076ACodec::FlushingState::FlushingState(ACodec *codec) 4077 : BaseState(codec) { 4078} 4079 4080void ACodec::FlushingState::stateEntered() { 4081 ALOGV("[%s] Now Flushing", mCodec->mComponentName.c_str()); 4082 4083 mFlushComplete[kPortIndexInput] = mFlushComplete[kPortIndexOutput] = false; 4084} 4085 4086bool ACodec::FlushingState::onMessageReceived(const sp<AMessage> &msg) { 4087 bool handled = false; 4088 4089 switch (msg->what()) { 4090 case kWhatShutdown: 4091 { 4092 mCodec->deferMessage(msg); 4093 break; 4094 } 4095 4096 case kWhatFlush: 4097 { 4098 // We're already doing this right now. 4099 handled = true; 4100 break; 4101 } 4102 4103 default: 4104 handled = BaseState::onMessageReceived(msg); 4105 break; 4106 } 4107 4108 return handled; 4109} 4110 4111bool ACodec::FlushingState::onOMXEvent( 4112 OMX_EVENTTYPE event, OMX_U32 data1, OMX_U32 data2) { 4113 ALOGV("[%s] FlushingState onOMXEvent(%d,%ld)", 4114 mCodec->mComponentName.c_str(), event, data1); 4115 4116 switch (event) { 4117 case OMX_EventCmdComplete: 4118 { 4119 CHECK_EQ(data1, (OMX_U32)OMX_CommandFlush); 4120 4121 if (data2 == kPortIndexInput || data2 == kPortIndexOutput) { 4122 CHECK(!mFlushComplete[data2]); 4123 mFlushComplete[data2] = true; 4124 4125 if (mFlushComplete[kPortIndexInput] 4126 && mFlushComplete[kPortIndexOutput]) { 4127 changeStateIfWeOwnAllBuffers(); 4128 } 4129 } else { 4130 CHECK_EQ(data2, OMX_ALL); 4131 CHECK(mFlushComplete[kPortIndexInput]); 4132 CHECK(mFlushComplete[kPortIndexOutput]); 4133 4134 changeStateIfWeOwnAllBuffers(); 4135 } 4136 4137 return true; 4138 } 4139 4140 case OMX_EventPortSettingsChanged: 4141 { 4142 sp<AMessage> msg = new AMessage(kWhatOMXMessage, mCodec->id()); 4143 msg->setInt32("type", omx_message::EVENT); 4144 msg->setPointer("node", mCodec->mNode); 4145 msg->setInt32("event", event); 4146 msg->setInt32("data1", data1); 4147 msg->setInt32("data2", data2); 4148 4149 ALOGV("[%s] Deferring OMX_EventPortSettingsChanged", 4150 mCodec->mComponentName.c_str()); 4151 4152 mCodec->deferMessage(msg); 4153 4154 return true; 4155 } 4156 4157 default: 4158 return BaseState::onOMXEvent(event, data1, data2); 4159 } 4160 4161 return true; 4162} 4163 4164void ACodec::FlushingState::onOutputBufferDrained(const sp<AMessage> &msg) { 4165 BaseState::onOutputBufferDrained(msg); 4166 4167 changeStateIfWeOwnAllBuffers(); 4168} 4169 4170void ACodec::FlushingState::onInputBufferFilled(const sp<AMessage> &msg) { 4171 BaseState::onInputBufferFilled(msg); 4172 4173 changeStateIfWeOwnAllBuffers(); 4174} 4175 4176void ACodec::FlushingState::changeStateIfWeOwnAllBuffers() { 4177 if (mFlushComplete[kPortIndexInput] 4178 && mFlushComplete[kPortIndexOutput] 4179 && mCodec->allYourBuffersAreBelongToUs()) { 4180 sp<AMessage> notify = mCodec->mNotify->dup(); 4181 notify->setInt32("what", ACodec::kWhatFlushCompleted); 4182 notify->post(); 4183 4184 mCodec->mPortEOS[kPortIndexInput] = 4185 mCodec->mPortEOS[kPortIndexOutput] = false; 4186 4187 mCodec->mInputEOSResult = OK; 4188 4189 if (mCodec->mSkipCutBuffer != NULL) { 4190 mCodec->mSkipCutBuffer->clear(); 4191 } 4192 4193 mCodec->changeState(mCodec->mExecutingState); 4194 } 4195} 4196 4197} // namespace android 4198