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