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