ACodec.cpp revision 3cb576166020bfdc1522b33919fe5ff8d211d0a3
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", CodecBase::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_CodingHEVC: 2039 err = setupHEVCEncoderParameters(msg); 2040 break; 2041 2042 case OMX_VIDEO_CodingVP8: 2043 case OMX_VIDEO_CodingVP9: 2044 err = setupVPXEncoderParameters(msg); 2045 break; 2046 2047 default: 2048 break; 2049 } 2050 2051 ALOGI("setupVideoEncoder succeeded"); 2052 2053 return err; 2054} 2055 2056status_t ACodec::setCyclicIntraMacroblockRefresh(const sp<AMessage> &msg, int32_t mode) { 2057 OMX_VIDEO_PARAM_INTRAREFRESHTYPE params; 2058 InitOMXParams(¶ms); 2059 params.nPortIndex = kPortIndexOutput; 2060 2061 params.eRefreshMode = static_cast<OMX_VIDEO_INTRAREFRESHTYPE>(mode); 2062 2063 if (params.eRefreshMode == OMX_VIDEO_IntraRefreshCyclic || 2064 params.eRefreshMode == OMX_VIDEO_IntraRefreshBoth) { 2065 int32_t mbs; 2066 if (!msg->findInt32("intra-refresh-CIR-mbs", &mbs)) { 2067 return INVALID_OPERATION; 2068 } 2069 params.nCirMBs = mbs; 2070 } 2071 2072 if (params.eRefreshMode == OMX_VIDEO_IntraRefreshAdaptive || 2073 params.eRefreshMode == OMX_VIDEO_IntraRefreshBoth) { 2074 int32_t mbs; 2075 if (!msg->findInt32("intra-refresh-AIR-mbs", &mbs)) { 2076 return INVALID_OPERATION; 2077 } 2078 params.nAirMBs = mbs; 2079 2080 int32_t ref; 2081 if (!msg->findInt32("intra-refresh-AIR-ref", &ref)) { 2082 return INVALID_OPERATION; 2083 } 2084 params.nAirRef = ref; 2085 } 2086 2087 status_t err = mOMX->setParameter( 2088 mNode, OMX_IndexParamVideoIntraRefresh, 2089 ¶ms, sizeof(params)); 2090 return err; 2091} 2092 2093static OMX_U32 setPFramesSpacing(int32_t iFramesInterval, int32_t frameRate) { 2094 if (iFramesInterval < 0) { 2095 return 0xFFFFFFFF; 2096 } else if (iFramesInterval == 0) { 2097 return 0; 2098 } 2099 OMX_U32 ret = frameRate * iFramesInterval; 2100 CHECK(ret > 1); 2101 return ret; 2102} 2103 2104static OMX_VIDEO_CONTROLRATETYPE getBitrateMode(const sp<AMessage> &msg) { 2105 int32_t tmp; 2106 if (!msg->findInt32("bitrate-mode", &tmp)) { 2107 return OMX_Video_ControlRateVariable; 2108 } 2109 2110 return static_cast<OMX_VIDEO_CONTROLRATETYPE>(tmp); 2111} 2112 2113status_t ACodec::setupMPEG4EncoderParameters(const sp<AMessage> &msg) { 2114 int32_t bitrate, iFrameInterval; 2115 if (!msg->findInt32("bitrate", &bitrate) 2116 || !msg->findInt32("i-frame-interval", &iFrameInterval)) { 2117 return INVALID_OPERATION; 2118 } 2119 2120 OMX_VIDEO_CONTROLRATETYPE bitrateMode = getBitrateMode(msg); 2121 2122 float frameRate; 2123 if (!msg->findFloat("frame-rate", &frameRate)) { 2124 int32_t tmp; 2125 if (!msg->findInt32("frame-rate", &tmp)) { 2126 return INVALID_OPERATION; 2127 } 2128 frameRate = (float)tmp; 2129 } 2130 2131 OMX_VIDEO_PARAM_MPEG4TYPE mpeg4type; 2132 InitOMXParams(&mpeg4type); 2133 mpeg4type.nPortIndex = kPortIndexOutput; 2134 2135 status_t err = mOMX->getParameter( 2136 mNode, OMX_IndexParamVideoMpeg4, &mpeg4type, sizeof(mpeg4type)); 2137 2138 if (err != OK) { 2139 return err; 2140 } 2141 2142 mpeg4type.nSliceHeaderSpacing = 0; 2143 mpeg4type.bSVH = OMX_FALSE; 2144 mpeg4type.bGov = OMX_FALSE; 2145 2146 mpeg4type.nAllowedPictureTypes = 2147 OMX_VIDEO_PictureTypeI | OMX_VIDEO_PictureTypeP; 2148 2149 mpeg4type.nPFrames = setPFramesSpacing(iFrameInterval, frameRate); 2150 if (mpeg4type.nPFrames == 0) { 2151 mpeg4type.nAllowedPictureTypes = OMX_VIDEO_PictureTypeI; 2152 } 2153 mpeg4type.nBFrames = 0; 2154 mpeg4type.nIDCVLCThreshold = 0; 2155 mpeg4type.bACPred = OMX_TRUE; 2156 mpeg4type.nMaxPacketSize = 256; 2157 mpeg4type.nTimeIncRes = 1000; 2158 mpeg4type.nHeaderExtension = 0; 2159 mpeg4type.bReversibleVLC = OMX_FALSE; 2160 2161 int32_t profile; 2162 if (msg->findInt32("profile", &profile)) { 2163 int32_t level; 2164 if (!msg->findInt32("level", &level)) { 2165 return INVALID_OPERATION; 2166 } 2167 2168 err = verifySupportForProfileAndLevel(profile, level); 2169 2170 if (err != OK) { 2171 return err; 2172 } 2173 2174 mpeg4type.eProfile = static_cast<OMX_VIDEO_MPEG4PROFILETYPE>(profile); 2175 mpeg4type.eLevel = static_cast<OMX_VIDEO_MPEG4LEVELTYPE>(level); 2176 } 2177 2178 err = mOMX->setParameter( 2179 mNode, OMX_IndexParamVideoMpeg4, &mpeg4type, sizeof(mpeg4type)); 2180 2181 if (err != OK) { 2182 return err; 2183 } 2184 2185 err = configureBitrate(bitrate, bitrateMode); 2186 2187 if (err != OK) { 2188 return err; 2189 } 2190 2191 return setupErrorCorrectionParameters(); 2192} 2193 2194status_t ACodec::setupH263EncoderParameters(const sp<AMessage> &msg) { 2195 int32_t bitrate, iFrameInterval; 2196 if (!msg->findInt32("bitrate", &bitrate) 2197 || !msg->findInt32("i-frame-interval", &iFrameInterval)) { 2198 return INVALID_OPERATION; 2199 } 2200 2201 OMX_VIDEO_CONTROLRATETYPE bitrateMode = getBitrateMode(msg); 2202 2203 float frameRate; 2204 if (!msg->findFloat("frame-rate", &frameRate)) { 2205 int32_t tmp; 2206 if (!msg->findInt32("frame-rate", &tmp)) { 2207 return INVALID_OPERATION; 2208 } 2209 frameRate = (float)tmp; 2210 } 2211 2212 OMX_VIDEO_PARAM_H263TYPE h263type; 2213 InitOMXParams(&h263type); 2214 h263type.nPortIndex = kPortIndexOutput; 2215 2216 status_t err = mOMX->getParameter( 2217 mNode, OMX_IndexParamVideoH263, &h263type, sizeof(h263type)); 2218 2219 if (err != OK) { 2220 return err; 2221 } 2222 2223 h263type.nAllowedPictureTypes = 2224 OMX_VIDEO_PictureTypeI | OMX_VIDEO_PictureTypeP; 2225 2226 h263type.nPFrames = setPFramesSpacing(iFrameInterval, frameRate); 2227 if (h263type.nPFrames == 0) { 2228 h263type.nAllowedPictureTypes = OMX_VIDEO_PictureTypeI; 2229 } 2230 h263type.nBFrames = 0; 2231 2232 int32_t profile; 2233 if (msg->findInt32("profile", &profile)) { 2234 int32_t level; 2235 if (!msg->findInt32("level", &level)) { 2236 return INVALID_OPERATION; 2237 } 2238 2239 err = verifySupportForProfileAndLevel(profile, level); 2240 2241 if (err != OK) { 2242 return err; 2243 } 2244 2245 h263type.eProfile = static_cast<OMX_VIDEO_H263PROFILETYPE>(profile); 2246 h263type.eLevel = static_cast<OMX_VIDEO_H263LEVELTYPE>(level); 2247 } 2248 2249 h263type.bPLUSPTYPEAllowed = OMX_FALSE; 2250 h263type.bForceRoundingTypeToZero = OMX_FALSE; 2251 h263type.nPictureHeaderRepetition = 0; 2252 h263type.nGOBHeaderInterval = 0; 2253 2254 err = mOMX->setParameter( 2255 mNode, OMX_IndexParamVideoH263, &h263type, sizeof(h263type)); 2256 2257 if (err != OK) { 2258 return err; 2259 } 2260 2261 err = configureBitrate(bitrate, bitrateMode); 2262 2263 if (err != OK) { 2264 return err; 2265 } 2266 2267 return setupErrorCorrectionParameters(); 2268} 2269 2270status_t ACodec::setupAVCEncoderParameters(const sp<AMessage> &msg) { 2271 int32_t bitrate, iFrameInterval; 2272 if (!msg->findInt32("bitrate", &bitrate) 2273 || !msg->findInt32("i-frame-interval", &iFrameInterval)) { 2274 return INVALID_OPERATION; 2275 } 2276 2277 OMX_VIDEO_CONTROLRATETYPE bitrateMode = getBitrateMode(msg); 2278 2279 float frameRate; 2280 if (!msg->findFloat("frame-rate", &frameRate)) { 2281 int32_t tmp; 2282 if (!msg->findInt32("frame-rate", &tmp)) { 2283 return INVALID_OPERATION; 2284 } 2285 frameRate = (float)tmp; 2286 } 2287 2288 status_t err = OK; 2289 int32_t intraRefreshMode = 0; 2290 if (msg->findInt32("intra-refresh-mode", &intraRefreshMode)) { 2291 err = setCyclicIntraMacroblockRefresh(msg, intraRefreshMode); 2292 if (err != OK) { 2293 ALOGE("Setting intra macroblock refresh mode (%d) failed: 0x%x", 2294 err, intraRefreshMode); 2295 return err; 2296 } 2297 } 2298 2299 OMX_VIDEO_PARAM_AVCTYPE h264type; 2300 InitOMXParams(&h264type); 2301 h264type.nPortIndex = kPortIndexOutput; 2302 2303 err = mOMX->getParameter( 2304 mNode, OMX_IndexParamVideoAvc, &h264type, sizeof(h264type)); 2305 2306 if (err != OK) { 2307 return err; 2308 } 2309 2310 h264type.nAllowedPictureTypes = 2311 OMX_VIDEO_PictureTypeI | OMX_VIDEO_PictureTypeP; 2312 2313 int32_t profile; 2314 if (msg->findInt32("profile", &profile)) { 2315 int32_t level; 2316 if (!msg->findInt32("level", &level)) { 2317 return INVALID_OPERATION; 2318 } 2319 2320 err = verifySupportForProfileAndLevel(profile, level); 2321 2322 if (err != OK) { 2323 return err; 2324 } 2325 2326 h264type.eProfile = static_cast<OMX_VIDEO_AVCPROFILETYPE>(profile); 2327 h264type.eLevel = static_cast<OMX_VIDEO_AVCLEVELTYPE>(level); 2328 } 2329 2330 // XXX 2331 if (h264type.eProfile != OMX_VIDEO_AVCProfileBaseline) { 2332 ALOGW("Use baseline profile instead of %d for AVC recording", 2333 h264type.eProfile); 2334 h264type.eProfile = OMX_VIDEO_AVCProfileBaseline; 2335 } 2336 2337 if (h264type.eProfile == OMX_VIDEO_AVCProfileBaseline) { 2338 h264type.nSliceHeaderSpacing = 0; 2339 h264type.bUseHadamard = OMX_TRUE; 2340 h264type.nRefFrames = 1; 2341 h264type.nBFrames = 0; 2342 h264type.nPFrames = setPFramesSpacing(iFrameInterval, frameRate); 2343 if (h264type.nPFrames == 0) { 2344 h264type.nAllowedPictureTypes = OMX_VIDEO_PictureTypeI; 2345 } 2346 h264type.nRefIdx10ActiveMinus1 = 0; 2347 h264type.nRefIdx11ActiveMinus1 = 0; 2348 h264type.bEntropyCodingCABAC = OMX_FALSE; 2349 h264type.bWeightedPPrediction = OMX_FALSE; 2350 h264type.bconstIpred = OMX_FALSE; 2351 h264type.bDirect8x8Inference = OMX_FALSE; 2352 h264type.bDirectSpatialTemporal = OMX_FALSE; 2353 h264type.nCabacInitIdc = 0; 2354 } 2355 2356 if (h264type.nBFrames != 0) { 2357 h264type.nAllowedPictureTypes |= OMX_VIDEO_PictureTypeB; 2358 } 2359 2360 h264type.bEnableUEP = OMX_FALSE; 2361 h264type.bEnableFMO = OMX_FALSE; 2362 h264type.bEnableASO = OMX_FALSE; 2363 h264type.bEnableRS = OMX_FALSE; 2364 h264type.bFrameMBsOnly = OMX_TRUE; 2365 h264type.bMBAFF = OMX_FALSE; 2366 h264type.eLoopFilterMode = OMX_VIDEO_AVCLoopFilterEnable; 2367 2368 err = mOMX->setParameter( 2369 mNode, OMX_IndexParamVideoAvc, &h264type, sizeof(h264type)); 2370 2371 if (err != OK) { 2372 return err; 2373 } 2374 2375 return configureBitrate(bitrate, bitrateMode); 2376} 2377 2378status_t ACodec::setupHEVCEncoderParameters(const sp<AMessage> &msg) { 2379 int32_t bitrate, iFrameInterval; 2380 if (!msg->findInt32("bitrate", &bitrate) 2381 || !msg->findInt32("i-frame-interval", &iFrameInterval)) { 2382 return INVALID_OPERATION; 2383 } 2384 2385 OMX_VIDEO_CONTROLRATETYPE bitrateMode = getBitrateMode(msg); 2386 2387 float frameRate; 2388 if (!msg->findFloat("frame-rate", &frameRate)) { 2389 int32_t tmp; 2390 if (!msg->findInt32("frame-rate", &tmp)) { 2391 return INVALID_OPERATION; 2392 } 2393 frameRate = (float)tmp; 2394 } 2395 2396 OMX_VIDEO_PARAM_HEVCTYPE hevcType; 2397 InitOMXParams(&hevcType); 2398 hevcType.nPortIndex = kPortIndexOutput; 2399 2400 status_t err = OK; 2401 err = mOMX->getParameter( 2402 mNode, (OMX_INDEXTYPE)OMX_IndexParamVideoHevc, &hevcType, sizeof(hevcType)); 2403 if (err != OK) { 2404 return err; 2405 } 2406 2407 int32_t profile; 2408 if (msg->findInt32("profile", &profile)) { 2409 int32_t level; 2410 if (!msg->findInt32("level", &level)) { 2411 return INVALID_OPERATION; 2412 } 2413 2414 err = verifySupportForProfileAndLevel(profile, level); 2415 if (err != OK) { 2416 return err; 2417 } 2418 2419 hevcType.eProfile = static_cast<OMX_VIDEO_HEVCPROFILETYPE>(profile); 2420 hevcType.eLevel = static_cast<OMX_VIDEO_HEVCLEVELTYPE>(level); 2421 } 2422 2423 // TODO: Need OMX structure definition for setting iFrameInterval 2424 2425 err = mOMX->setParameter( 2426 mNode, (OMX_INDEXTYPE)OMX_IndexParamVideoHevc, &hevcType, sizeof(hevcType)); 2427 if (err != OK) { 2428 return err; 2429 } 2430 2431 return configureBitrate(bitrate, bitrateMode); 2432} 2433 2434status_t ACodec::setupVPXEncoderParameters(const sp<AMessage> &msg) { 2435 int32_t bitrate; 2436 int32_t iFrameInterval = 0; 2437 size_t tsLayers = 0; 2438 OMX_VIDEO_ANDROID_VPXTEMPORALLAYERPATTERNTYPE pattern = 2439 OMX_VIDEO_VPXTemporalLayerPatternNone; 2440 static const uint32_t kVp8LayerRateAlloction 2441 [OMX_VIDEO_ANDROID_MAXVP8TEMPORALLAYERS] 2442 [OMX_VIDEO_ANDROID_MAXVP8TEMPORALLAYERS] = { 2443 {100, 100, 100}, // 1 layer 2444 { 60, 100, 100}, // 2 layers {60%, 40%} 2445 { 40, 60, 100}, // 3 layers {40%, 20%, 40%} 2446 }; 2447 if (!msg->findInt32("bitrate", &bitrate)) { 2448 return INVALID_OPERATION; 2449 } 2450 msg->findInt32("i-frame-interval", &iFrameInterval); 2451 2452 OMX_VIDEO_CONTROLRATETYPE bitrateMode = getBitrateMode(msg); 2453 2454 float frameRate; 2455 if (!msg->findFloat("frame-rate", &frameRate)) { 2456 int32_t tmp; 2457 if (!msg->findInt32("frame-rate", &tmp)) { 2458 return INVALID_OPERATION; 2459 } 2460 frameRate = (float)tmp; 2461 } 2462 2463 AString tsSchema; 2464 if (msg->findString("ts-schema", &tsSchema)) { 2465 if (tsSchema == "webrtc.vp8.1-layer") { 2466 pattern = OMX_VIDEO_VPXTemporalLayerPatternWebRTC; 2467 tsLayers = 1; 2468 } else if (tsSchema == "webrtc.vp8.2-layer") { 2469 pattern = OMX_VIDEO_VPXTemporalLayerPatternWebRTC; 2470 tsLayers = 2; 2471 } else if (tsSchema == "webrtc.vp8.3-layer") { 2472 pattern = OMX_VIDEO_VPXTemporalLayerPatternWebRTC; 2473 tsLayers = 3; 2474 } else { 2475 ALOGW("Unsupported ts-schema [%s]", tsSchema.c_str()); 2476 } 2477 } 2478 2479 OMX_VIDEO_PARAM_ANDROID_VP8ENCODERTYPE vp8type; 2480 InitOMXParams(&vp8type); 2481 vp8type.nPortIndex = kPortIndexOutput; 2482 status_t err = mOMX->getParameter( 2483 mNode, (OMX_INDEXTYPE)OMX_IndexParamVideoAndroidVp8Encoder, 2484 &vp8type, sizeof(vp8type)); 2485 2486 if (err == OK) { 2487 if (iFrameInterval > 0) { 2488 vp8type.nKeyFrameInterval = setPFramesSpacing(iFrameInterval, frameRate); 2489 } 2490 vp8type.eTemporalPattern = pattern; 2491 vp8type.nTemporalLayerCount = tsLayers; 2492 if (tsLayers > 0) { 2493 for (size_t i = 0; i < OMX_VIDEO_ANDROID_MAXVP8TEMPORALLAYERS; i++) { 2494 vp8type.nTemporalLayerBitrateRatio[i] = 2495 kVp8LayerRateAlloction[tsLayers - 1][i]; 2496 } 2497 } 2498 if (bitrateMode == OMX_Video_ControlRateConstant) { 2499 vp8type.nMinQuantizer = 2; 2500 vp8type.nMaxQuantizer = 63; 2501 } 2502 2503 err = mOMX->setParameter( 2504 mNode, (OMX_INDEXTYPE)OMX_IndexParamVideoAndroidVp8Encoder, 2505 &vp8type, sizeof(vp8type)); 2506 if (err != OK) { 2507 ALOGW("Extended VP8 parameters set failed: %d", err); 2508 } 2509 } 2510 2511 return configureBitrate(bitrate, bitrateMode); 2512} 2513 2514status_t ACodec::verifySupportForProfileAndLevel( 2515 int32_t profile, int32_t level) { 2516 OMX_VIDEO_PARAM_PROFILELEVELTYPE params; 2517 InitOMXParams(¶ms); 2518 params.nPortIndex = kPortIndexOutput; 2519 2520 for (params.nProfileIndex = 0;; ++params.nProfileIndex) { 2521 status_t err = mOMX->getParameter( 2522 mNode, 2523 OMX_IndexParamVideoProfileLevelQuerySupported, 2524 ¶ms, 2525 sizeof(params)); 2526 2527 if (err != OK) { 2528 return err; 2529 } 2530 2531 int32_t supportedProfile = static_cast<int32_t>(params.eProfile); 2532 int32_t supportedLevel = static_cast<int32_t>(params.eLevel); 2533 2534 if (profile == supportedProfile && level <= supportedLevel) { 2535 return OK; 2536 } 2537 } 2538} 2539 2540status_t ACodec::configureBitrate( 2541 int32_t bitrate, OMX_VIDEO_CONTROLRATETYPE bitrateMode) { 2542 OMX_VIDEO_PARAM_BITRATETYPE bitrateType; 2543 InitOMXParams(&bitrateType); 2544 bitrateType.nPortIndex = kPortIndexOutput; 2545 2546 status_t err = mOMX->getParameter( 2547 mNode, OMX_IndexParamVideoBitrate, 2548 &bitrateType, sizeof(bitrateType)); 2549 2550 if (err != OK) { 2551 return err; 2552 } 2553 2554 bitrateType.eControlRate = bitrateMode; 2555 bitrateType.nTargetBitrate = bitrate; 2556 2557 return mOMX->setParameter( 2558 mNode, OMX_IndexParamVideoBitrate, 2559 &bitrateType, sizeof(bitrateType)); 2560} 2561 2562status_t ACodec::setupErrorCorrectionParameters() { 2563 OMX_VIDEO_PARAM_ERRORCORRECTIONTYPE errorCorrectionType; 2564 InitOMXParams(&errorCorrectionType); 2565 errorCorrectionType.nPortIndex = kPortIndexOutput; 2566 2567 status_t err = mOMX->getParameter( 2568 mNode, OMX_IndexParamVideoErrorCorrection, 2569 &errorCorrectionType, sizeof(errorCorrectionType)); 2570 2571 if (err != OK) { 2572 return OK; // Optional feature. Ignore this failure 2573 } 2574 2575 errorCorrectionType.bEnableHEC = OMX_FALSE; 2576 errorCorrectionType.bEnableResync = OMX_TRUE; 2577 errorCorrectionType.nResynchMarkerSpacing = 256; 2578 errorCorrectionType.bEnableDataPartitioning = OMX_FALSE; 2579 errorCorrectionType.bEnableRVLC = OMX_FALSE; 2580 2581 return mOMX->setParameter( 2582 mNode, OMX_IndexParamVideoErrorCorrection, 2583 &errorCorrectionType, sizeof(errorCorrectionType)); 2584} 2585 2586status_t ACodec::setVideoFormatOnPort( 2587 OMX_U32 portIndex, 2588 int32_t width, int32_t height, OMX_VIDEO_CODINGTYPE compressionFormat) { 2589 OMX_PARAM_PORTDEFINITIONTYPE def; 2590 InitOMXParams(&def); 2591 def.nPortIndex = portIndex; 2592 2593 OMX_VIDEO_PORTDEFINITIONTYPE *video_def = &def.format.video; 2594 2595 status_t err = mOMX->getParameter( 2596 mNode, OMX_IndexParamPortDefinition, &def, sizeof(def)); 2597 2598 CHECK_EQ(err, (status_t)OK); 2599 2600 if (portIndex == kPortIndexInput) { 2601 // XXX Need a (much) better heuristic to compute input buffer sizes. 2602 const size_t X = 64 * 1024; 2603 if (def.nBufferSize < X) { 2604 def.nBufferSize = X; 2605 } 2606 } 2607 2608 CHECK_EQ((int)def.eDomain, (int)OMX_PortDomainVideo); 2609 2610 video_def->nFrameWidth = width; 2611 video_def->nFrameHeight = height; 2612 2613 if (portIndex == kPortIndexInput) { 2614 video_def->eCompressionFormat = compressionFormat; 2615 video_def->eColorFormat = OMX_COLOR_FormatUnused; 2616 } 2617 2618 err = mOMX->setParameter( 2619 mNode, OMX_IndexParamPortDefinition, &def, sizeof(def)); 2620 2621 return err; 2622} 2623 2624status_t ACodec::initNativeWindow() { 2625 if (mNativeWindow != NULL) { 2626 return mOMX->enableGraphicBuffers(mNode, kPortIndexOutput, OMX_TRUE); 2627 } 2628 2629 mOMX->enableGraphicBuffers(mNode, kPortIndexOutput, OMX_FALSE); 2630 return OK; 2631} 2632 2633size_t ACodec::countBuffersOwnedByComponent(OMX_U32 portIndex) const { 2634 size_t n = 0; 2635 2636 for (size_t i = 0; i < mBuffers[portIndex].size(); ++i) { 2637 const BufferInfo &info = mBuffers[portIndex].itemAt(i); 2638 2639 if (info.mStatus == BufferInfo::OWNED_BY_COMPONENT) { 2640 ++n; 2641 } 2642 } 2643 2644 return n; 2645} 2646 2647size_t ACodec::countBuffersOwnedByNativeWindow() const { 2648 size_t n = 0; 2649 2650 for (size_t i = 0; i < mBuffers[kPortIndexOutput].size(); ++i) { 2651 const BufferInfo &info = mBuffers[kPortIndexOutput].itemAt(i); 2652 2653 if (info.mStatus == BufferInfo::OWNED_BY_NATIVE_WINDOW) { 2654 ++n; 2655 } 2656 } 2657 2658 return n; 2659} 2660 2661void ACodec::waitUntilAllPossibleNativeWindowBuffersAreReturnedToUs() { 2662 if (mNativeWindow == NULL) { 2663 return; 2664 } 2665 2666 while (countBuffersOwnedByNativeWindow() > mNumUndequeuedBuffers 2667 && dequeueBufferFromNativeWindow() != NULL) { 2668 // these buffers will be submitted as regular buffers; account for this 2669 if (mStoreMetaDataInOutputBuffers && mMetaDataBuffersToSubmit > 0) { 2670 --mMetaDataBuffersToSubmit; 2671 } 2672 } 2673} 2674 2675bool ACodec::allYourBuffersAreBelongToUs( 2676 OMX_U32 portIndex) { 2677 for (size_t i = 0; i < mBuffers[portIndex].size(); ++i) { 2678 BufferInfo *info = &mBuffers[portIndex].editItemAt(i); 2679 2680 if (info->mStatus != BufferInfo::OWNED_BY_US 2681 && info->mStatus != BufferInfo::OWNED_BY_NATIVE_WINDOW) { 2682 ALOGV("[%s] Buffer %u on port %u still has status %d", 2683 mComponentName.c_str(), 2684 info->mBufferID, portIndex, info->mStatus); 2685 return false; 2686 } 2687 } 2688 2689 return true; 2690} 2691 2692bool ACodec::allYourBuffersAreBelongToUs() { 2693 return allYourBuffersAreBelongToUs(kPortIndexInput) 2694 && allYourBuffersAreBelongToUs(kPortIndexOutput); 2695} 2696 2697void ACodec::deferMessage(const sp<AMessage> &msg) { 2698 bool wasEmptyBefore = mDeferredQueue.empty(); 2699 mDeferredQueue.push_back(msg); 2700} 2701 2702void ACodec::processDeferredMessages() { 2703 List<sp<AMessage> > queue = mDeferredQueue; 2704 mDeferredQueue.clear(); 2705 2706 List<sp<AMessage> >::iterator it = queue.begin(); 2707 while (it != queue.end()) { 2708 onMessageReceived(*it++); 2709 } 2710} 2711 2712status_t ACodec::getPortFormat(OMX_U32 portIndex, sp<AMessage> ¬ify) { 2713 // TODO: catch errors an return them instead of using CHECK 2714 OMX_PARAM_PORTDEFINITIONTYPE def; 2715 InitOMXParams(&def); 2716 def.nPortIndex = portIndex; 2717 2718 CHECK_EQ(mOMX->getParameter( 2719 mNode, OMX_IndexParamPortDefinition, &def, sizeof(def)), 2720 (status_t)OK); 2721 2722 CHECK_EQ((int)def.eDir, 2723 (int)(portIndex == kPortIndexOutput ? OMX_DirOutput : OMX_DirInput)); 2724 2725 switch (def.eDomain) { 2726 case OMX_PortDomainVideo: 2727 { 2728 OMX_VIDEO_PORTDEFINITIONTYPE *videoDef = &def.format.video; 2729 switch ((int)videoDef->eCompressionFormat) { 2730 case OMX_VIDEO_CodingUnused: 2731 { 2732 CHECK(mIsEncoder ^ (portIndex == kPortIndexOutput)); 2733 notify->setString("mime", MEDIA_MIMETYPE_VIDEO_RAW); 2734 2735 notify->setInt32("stride", videoDef->nStride); 2736 notify->setInt32("slice-height", videoDef->nSliceHeight); 2737 notify->setInt32("color-format", videoDef->eColorFormat); 2738 2739 OMX_CONFIG_RECTTYPE rect; 2740 InitOMXParams(&rect); 2741 rect.nPortIndex = kPortIndexOutput; 2742 2743 if (mOMX->getConfig( 2744 mNode, OMX_IndexConfigCommonOutputCrop, 2745 &rect, sizeof(rect)) != OK) { 2746 rect.nLeft = 0; 2747 rect.nTop = 0; 2748 rect.nWidth = videoDef->nFrameWidth; 2749 rect.nHeight = videoDef->nFrameHeight; 2750 } 2751 2752 CHECK_GE(rect.nLeft, 0); 2753 CHECK_GE(rect.nTop, 0); 2754 CHECK_GE(rect.nWidth, 0u); 2755 CHECK_GE(rect.nHeight, 0u); 2756 CHECK_LE(rect.nLeft + rect.nWidth - 1, videoDef->nFrameWidth); 2757 CHECK_LE(rect.nTop + rect.nHeight - 1, videoDef->nFrameHeight); 2758 2759 notify->setRect( 2760 "crop", 2761 rect.nLeft, 2762 rect.nTop, 2763 rect.nLeft + rect.nWidth - 1, 2764 rect.nTop + rect.nHeight - 1); 2765 2766 break; 2767 } 2768 default: 2769 { 2770 CHECK(mIsEncoder ^ (portIndex == kPortIndexInput)); 2771 AString mime; 2772 if (GetMimeTypeForVideoCoding( 2773 videoDef->eCompressionFormat, &mime) != OK) { 2774 notify->setString("mime", "application/octet-stream"); 2775 } else { 2776 notify->setString("mime", mime.c_str()); 2777 } 2778 break; 2779 } 2780 } 2781 2782 notify->setInt32("width", videoDef->nFrameWidth); 2783 notify->setInt32("height", videoDef->nFrameHeight); 2784 break; 2785 } 2786 2787 case OMX_PortDomainAudio: 2788 { 2789 OMX_AUDIO_PORTDEFINITIONTYPE *audioDef = &def.format.audio; 2790 2791 switch ((int)audioDef->eEncoding) { 2792 case OMX_AUDIO_CodingPCM: 2793 { 2794 OMX_AUDIO_PARAM_PCMMODETYPE params; 2795 InitOMXParams(¶ms); 2796 params.nPortIndex = portIndex; 2797 2798 CHECK_EQ(mOMX->getParameter( 2799 mNode, OMX_IndexParamAudioPcm, 2800 ¶ms, sizeof(params)), 2801 (status_t)OK); 2802 2803 CHECK_GT(params.nChannels, 0); 2804 CHECK(params.nChannels == 1 || params.bInterleaved); 2805 CHECK_EQ(params.nBitPerSample, 16u); 2806 2807 CHECK_EQ((int)params.eNumData, 2808 (int)OMX_NumericalDataSigned); 2809 2810 CHECK_EQ((int)params.ePCMMode, 2811 (int)OMX_AUDIO_PCMModeLinear); 2812 2813 notify->setString("mime", MEDIA_MIMETYPE_AUDIO_RAW); 2814 notify->setInt32("channel-count", params.nChannels); 2815 notify->setInt32("sample-rate", params.nSamplingRate); 2816 2817 if (mChannelMaskPresent) { 2818 notify->setInt32("channel-mask", mChannelMask); 2819 } 2820 break; 2821 } 2822 2823 case OMX_AUDIO_CodingAAC: 2824 { 2825 OMX_AUDIO_PARAM_AACPROFILETYPE params; 2826 InitOMXParams(¶ms); 2827 params.nPortIndex = portIndex; 2828 2829 CHECK_EQ(mOMX->getParameter( 2830 mNode, OMX_IndexParamAudioAac, 2831 ¶ms, sizeof(params)), 2832 (status_t)OK); 2833 2834 notify->setString("mime", MEDIA_MIMETYPE_AUDIO_AAC); 2835 notify->setInt32("channel-count", params.nChannels); 2836 notify->setInt32("sample-rate", params.nSampleRate); 2837 break; 2838 } 2839 2840 case OMX_AUDIO_CodingAMR: 2841 { 2842 OMX_AUDIO_PARAM_AMRTYPE params; 2843 InitOMXParams(¶ms); 2844 params.nPortIndex = portIndex; 2845 2846 CHECK_EQ(mOMX->getParameter( 2847 mNode, OMX_IndexParamAudioAmr, 2848 ¶ms, sizeof(params)), 2849 (status_t)OK); 2850 2851 notify->setInt32("channel-count", 1); 2852 if (params.eAMRBandMode >= OMX_AUDIO_AMRBandModeWB0) { 2853 notify->setString( 2854 "mime", MEDIA_MIMETYPE_AUDIO_AMR_WB); 2855 2856 notify->setInt32("sample-rate", 16000); 2857 } else { 2858 notify->setString( 2859 "mime", MEDIA_MIMETYPE_AUDIO_AMR_NB); 2860 2861 notify->setInt32("sample-rate", 8000); 2862 } 2863 break; 2864 } 2865 2866 case OMX_AUDIO_CodingFLAC: 2867 { 2868 OMX_AUDIO_PARAM_FLACTYPE params; 2869 InitOMXParams(¶ms); 2870 params.nPortIndex = portIndex; 2871 2872 CHECK_EQ(mOMX->getParameter( 2873 mNode, OMX_IndexParamAudioFlac, 2874 ¶ms, sizeof(params)), 2875 (status_t)OK); 2876 2877 notify->setString("mime", MEDIA_MIMETYPE_AUDIO_FLAC); 2878 notify->setInt32("channel-count", params.nChannels); 2879 notify->setInt32("sample-rate", params.nSampleRate); 2880 break; 2881 } 2882 2883 case OMX_AUDIO_CodingMP3: 2884 { 2885 OMX_AUDIO_PARAM_MP3TYPE params; 2886 InitOMXParams(¶ms); 2887 params.nPortIndex = portIndex; 2888 2889 CHECK_EQ(mOMX->getParameter( 2890 mNode, OMX_IndexParamAudioMp3, 2891 ¶ms, sizeof(params)), 2892 (status_t)OK); 2893 2894 notify->setString("mime", MEDIA_MIMETYPE_AUDIO_MPEG); 2895 notify->setInt32("channel-count", params.nChannels); 2896 notify->setInt32("sample-rate", params.nSampleRate); 2897 break; 2898 } 2899 2900 case OMX_AUDIO_CodingVORBIS: 2901 { 2902 OMX_AUDIO_PARAM_VORBISTYPE params; 2903 InitOMXParams(¶ms); 2904 params.nPortIndex = portIndex; 2905 2906 CHECK_EQ(mOMX->getParameter( 2907 mNode, OMX_IndexParamAudioVorbis, 2908 ¶ms, sizeof(params)), 2909 (status_t)OK); 2910 2911 notify->setString("mime", MEDIA_MIMETYPE_AUDIO_VORBIS); 2912 notify->setInt32("channel-count", params.nChannels); 2913 notify->setInt32("sample-rate", params.nSampleRate); 2914 break; 2915 } 2916 2917 case OMX_AUDIO_CodingAndroidAC3: 2918 { 2919 OMX_AUDIO_PARAM_ANDROID_AC3TYPE params; 2920 InitOMXParams(¶ms); 2921 params.nPortIndex = portIndex; 2922 2923 CHECK_EQ((status_t)OK, mOMX->getParameter( 2924 mNode, 2925 (OMX_INDEXTYPE)OMX_IndexParamAudioAndroidAc3, 2926 ¶ms, 2927 sizeof(params))); 2928 2929 notify->setString("mime", MEDIA_MIMETYPE_AUDIO_AC3); 2930 notify->setInt32("channel-count", params.nChannels); 2931 notify->setInt32("sample-rate", params.nSampleRate); 2932 break; 2933 } 2934 2935 case OMX_AUDIO_CodingAndroidOPUS: 2936 { 2937 OMX_AUDIO_PARAM_ANDROID_OPUSTYPE params; 2938 InitOMXParams(¶ms); 2939 params.nPortIndex = portIndex; 2940 2941 CHECK_EQ((status_t)OK, mOMX->getParameter( 2942 mNode, 2943 (OMX_INDEXTYPE)OMX_IndexParamAudioAndroidOpus, 2944 ¶ms, 2945 sizeof(params))); 2946 2947 notify->setString("mime", MEDIA_MIMETYPE_AUDIO_OPUS); 2948 notify->setInt32("channel-count", params.nChannels); 2949 notify->setInt32("sample-rate", params.nSampleRate); 2950 break; 2951 } 2952 2953 default: 2954 ALOGE("UNKNOWN AUDIO CODING: %d\n", audioDef->eEncoding); 2955 TRESPASS(); 2956 } 2957 break; 2958 } 2959 2960 default: 2961 TRESPASS(); 2962 } 2963 2964 return OK; 2965} 2966 2967void ACodec::sendFormatChange(const sp<AMessage> &reply) { 2968 sp<AMessage> notify = mNotify->dup(); 2969 notify->setInt32("what", kWhatOutputFormatChanged); 2970 2971 CHECK_EQ(getPortFormat(kPortIndexOutput, notify), (status_t)OK); 2972 2973 AString mime; 2974 CHECK(notify->findString("mime", &mime)); 2975 2976 int32_t left, top, right, bottom; 2977 if (mime == MEDIA_MIMETYPE_VIDEO_RAW && 2978 mNativeWindow != NULL && 2979 notify->findRect("crop", &left, &top, &right, &bottom)) { 2980 // notify renderer of the crop change 2981 // NOTE: native window uses extended right-bottom coordinate 2982 reply->setRect("crop", left, top, right + 1, bottom + 1); 2983 } else if (mime == MEDIA_MIMETYPE_AUDIO_RAW && 2984 (mEncoderDelay || mEncoderPadding)) { 2985 int32_t channelCount; 2986 CHECK(notify->findInt32("channel-count", &channelCount)); 2987 size_t frameSize = channelCount * sizeof(int16_t); 2988 if (mSkipCutBuffer != NULL) { 2989 size_t prevbufsize = mSkipCutBuffer->size(); 2990 if (prevbufsize != 0) { 2991 ALOGW("Replacing SkipCutBuffer holding %d " 2992 "bytes", 2993 prevbufsize); 2994 } 2995 } 2996 mSkipCutBuffer = new SkipCutBuffer( 2997 mEncoderDelay * frameSize, 2998 mEncoderPadding * frameSize); 2999 } 3000 3001 notify->post(); 3002 3003 mSentFormat = true; 3004} 3005 3006void ACodec::signalError(OMX_ERRORTYPE error, status_t internalError) { 3007 sp<AMessage> notify = mNotify->dup(); 3008 notify->setInt32("what", CodecBase::kWhatError); 3009 notify->setInt32("omx-error", error); 3010 notify->setInt32("err", internalError); 3011 notify->post(); 3012} 3013 3014status_t ACodec::pushBlankBuffersToNativeWindow() { 3015 status_t err = NO_ERROR; 3016 ANativeWindowBuffer* anb = NULL; 3017 int numBufs = 0; 3018 int minUndequeuedBufs = 0; 3019 3020 // We need to reconnect to the ANativeWindow as a CPU client to ensure that 3021 // no frames get dropped by SurfaceFlinger assuming that these are video 3022 // frames. 3023 err = native_window_api_disconnect(mNativeWindow.get(), 3024 NATIVE_WINDOW_API_MEDIA); 3025 if (err != NO_ERROR) { 3026 ALOGE("error pushing blank frames: api_disconnect failed: %s (%d)", 3027 strerror(-err), -err); 3028 return err; 3029 } 3030 3031 err = native_window_api_connect(mNativeWindow.get(), 3032 NATIVE_WINDOW_API_CPU); 3033 if (err != NO_ERROR) { 3034 ALOGE("error pushing blank frames: api_connect failed: %s (%d)", 3035 strerror(-err), -err); 3036 return err; 3037 } 3038 3039 err = native_window_set_buffers_geometry(mNativeWindow.get(), 1, 1, 3040 HAL_PIXEL_FORMAT_RGBX_8888); 3041 if (err != NO_ERROR) { 3042 ALOGE("error pushing blank frames: set_buffers_geometry failed: %s (%d)", 3043 strerror(-err), -err); 3044 goto error; 3045 } 3046 3047 err = native_window_set_scaling_mode(mNativeWindow.get(), 3048 NATIVE_WINDOW_SCALING_MODE_SCALE_TO_WINDOW); 3049 if (err != NO_ERROR) { 3050 ALOGE("error pushing blank_frames: set_scaling_mode failed: %s (%d)", 3051 strerror(-err), -err); 3052 goto error; 3053 } 3054 3055 err = native_window_set_usage(mNativeWindow.get(), 3056 GRALLOC_USAGE_SW_WRITE_OFTEN); 3057 if (err != NO_ERROR) { 3058 ALOGE("error pushing blank frames: set_usage failed: %s (%d)", 3059 strerror(-err), -err); 3060 goto error; 3061 } 3062 3063 err = mNativeWindow->query(mNativeWindow.get(), 3064 NATIVE_WINDOW_MIN_UNDEQUEUED_BUFFERS, &minUndequeuedBufs); 3065 if (err != NO_ERROR) { 3066 ALOGE("error pushing blank frames: MIN_UNDEQUEUED_BUFFERS query " 3067 "failed: %s (%d)", strerror(-err), -err); 3068 goto error; 3069 } 3070 3071 numBufs = minUndequeuedBufs + 1; 3072 err = native_window_set_buffer_count(mNativeWindow.get(), numBufs); 3073 if (err != NO_ERROR) { 3074 ALOGE("error pushing blank frames: set_buffer_count failed: %s (%d)", 3075 strerror(-err), -err); 3076 goto error; 3077 } 3078 3079 // We push numBufs + 1 buffers to ensure that we've drawn into the same 3080 // buffer twice. This should guarantee that the buffer has been displayed 3081 // on the screen and then been replaced, so an previous video frames are 3082 // guaranteed NOT to be currently displayed. 3083 for (int i = 0; i < numBufs + 1; i++) { 3084 int fenceFd = -1; 3085 err = native_window_dequeue_buffer_and_wait(mNativeWindow.get(), &anb); 3086 if (err != NO_ERROR) { 3087 ALOGE("error pushing blank frames: dequeueBuffer failed: %s (%d)", 3088 strerror(-err), -err); 3089 goto error; 3090 } 3091 3092 sp<GraphicBuffer> buf(new GraphicBuffer(anb, false)); 3093 3094 // Fill the buffer with the a 1x1 checkerboard pattern ;) 3095 uint32_t* img = NULL; 3096 err = buf->lock(GRALLOC_USAGE_SW_WRITE_OFTEN, (void**)(&img)); 3097 if (err != NO_ERROR) { 3098 ALOGE("error pushing blank frames: lock failed: %s (%d)", 3099 strerror(-err), -err); 3100 goto error; 3101 } 3102 3103 *img = 0; 3104 3105 err = buf->unlock(); 3106 if (err != NO_ERROR) { 3107 ALOGE("error pushing blank frames: unlock failed: %s (%d)", 3108 strerror(-err), -err); 3109 goto error; 3110 } 3111 3112 err = mNativeWindow->queueBuffer(mNativeWindow.get(), 3113 buf->getNativeBuffer(), -1); 3114 if (err != NO_ERROR) { 3115 ALOGE("error pushing blank frames: queueBuffer failed: %s (%d)", 3116 strerror(-err), -err); 3117 goto error; 3118 } 3119 3120 anb = NULL; 3121 } 3122 3123error: 3124 3125 if (err != NO_ERROR) { 3126 // Clean up after an error. 3127 if (anb != NULL) { 3128 mNativeWindow->cancelBuffer(mNativeWindow.get(), anb, -1); 3129 } 3130 3131 native_window_api_disconnect(mNativeWindow.get(), 3132 NATIVE_WINDOW_API_CPU); 3133 native_window_api_connect(mNativeWindow.get(), 3134 NATIVE_WINDOW_API_MEDIA); 3135 3136 return err; 3137 } else { 3138 // Clean up after success. 3139 err = native_window_api_disconnect(mNativeWindow.get(), 3140 NATIVE_WINDOW_API_CPU); 3141 if (err != NO_ERROR) { 3142 ALOGE("error pushing blank frames: api_disconnect failed: %s (%d)", 3143 strerror(-err), -err); 3144 return err; 3145 } 3146 3147 err = native_window_api_connect(mNativeWindow.get(), 3148 NATIVE_WINDOW_API_MEDIA); 3149 if (err != NO_ERROR) { 3150 ALOGE("error pushing blank frames: api_connect failed: %s (%d)", 3151 strerror(-err), -err); 3152 return err; 3153 } 3154 3155 return NO_ERROR; 3156 } 3157} 3158 3159//////////////////////////////////////////////////////////////////////////////// 3160 3161ACodec::PortDescription::PortDescription() { 3162} 3163 3164status_t ACodec::requestIDRFrame() { 3165 if (!mIsEncoder) { 3166 return ERROR_UNSUPPORTED; 3167 } 3168 3169 OMX_CONFIG_INTRAREFRESHVOPTYPE params; 3170 InitOMXParams(¶ms); 3171 3172 params.nPortIndex = kPortIndexOutput; 3173 params.IntraRefreshVOP = OMX_TRUE; 3174 3175 return mOMX->setConfig( 3176 mNode, 3177 OMX_IndexConfigVideoIntraVOPRefresh, 3178 ¶ms, 3179 sizeof(params)); 3180} 3181 3182void ACodec::PortDescription::addBuffer( 3183 IOMX::buffer_id id, const sp<ABuffer> &buffer) { 3184 mBufferIDs.push_back(id); 3185 mBuffers.push_back(buffer); 3186} 3187 3188size_t ACodec::PortDescription::countBuffers() { 3189 return mBufferIDs.size(); 3190} 3191 3192IOMX::buffer_id ACodec::PortDescription::bufferIDAt(size_t index) const { 3193 return mBufferIDs.itemAt(index); 3194} 3195 3196sp<ABuffer> ACodec::PortDescription::bufferAt(size_t index) const { 3197 return mBuffers.itemAt(index); 3198} 3199 3200//////////////////////////////////////////////////////////////////////////////// 3201 3202ACodec::BaseState::BaseState(ACodec *codec, const sp<AState> &parentState) 3203 : AState(parentState), 3204 mCodec(codec) { 3205} 3206 3207ACodec::BaseState::PortMode ACodec::BaseState::getPortMode( 3208 OMX_U32 /* portIndex */) { 3209 return KEEP_BUFFERS; 3210} 3211 3212bool ACodec::BaseState::onMessageReceived(const sp<AMessage> &msg) { 3213 switch (msg->what()) { 3214 case kWhatInputBufferFilled: 3215 { 3216 onInputBufferFilled(msg); 3217 break; 3218 } 3219 3220 case kWhatOutputBufferDrained: 3221 { 3222 onOutputBufferDrained(msg); 3223 break; 3224 } 3225 3226 case ACodec::kWhatOMXMessage: 3227 { 3228 return onOMXMessage(msg); 3229 } 3230 3231 case ACodec::kWhatCreateInputSurface: 3232 case ACodec::kWhatSignalEndOfInputStream: 3233 { 3234 ALOGE("Message 0x%x was not handled", msg->what()); 3235 mCodec->signalError(OMX_ErrorUndefined, INVALID_OPERATION); 3236 return true; 3237 } 3238 3239 case ACodec::kWhatOMXDied: 3240 { 3241 ALOGE("OMX/mediaserver died, signalling error!"); 3242 mCodec->signalError(OMX_ErrorResourcesLost, DEAD_OBJECT); 3243 break; 3244 } 3245 3246 default: 3247 return false; 3248 } 3249 3250 return true; 3251} 3252 3253bool ACodec::BaseState::onOMXMessage(const sp<AMessage> &msg) { 3254 int32_t type; 3255 CHECK(msg->findInt32("type", &type)); 3256 3257 // there is a possibility that this is an outstanding message for a 3258 // codec that we have already destroyed 3259 if (mCodec->mNode == NULL) { 3260 ALOGI("ignoring message as already freed component: %s", 3261 msg->debugString().c_str()); 3262 return true; 3263 } 3264 3265 IOMX::node_id nodeID; 3266 CHECK(msg->findInt32("node", (int32_t*)&nodeID)); 3267 CHECK_EQ(nodeID, mCodec->mNode); 3268 3269 switch (type) { 3270 case omx_message::EVENT: 3271 { 3272 int32_t event, data1, data2; 3273 CHECK(msg->findInt32("event", &event)); 3274 CHECK(msg->findInt32("data1", &data1)); 3275 CHECK(msg->findInt32("data2", &data2)); 3276 3277 if (event == OMX_EventCmdComplete 3278 && data1 == OMX_CommandFlush 3279 && data2 == (int32_t)OMX_ALL) { 3280 // Use of this notification is not consistent across 3281 // implementations. We'll drop this notification and rely 3282 // on flush-complete notifications on the individual port 3283 // indices instead. 3284 3285 return true; 3286 } 3287 3288 return onOMXEvent( 3289 static_cast<OMX_EVENTTYPE>(event), 3290 static_cast<OMX_U32>(data1), 3291 static_cast<OMX_U32>(data2)); 3292 } 3293 3294 case omx_message::EMPTY_BUFFER_DONE: 3295 { 3296 IOMX::buffer_id bufferID; 3297 CHECK(msg->findInt32("buffer", (int32_t*)&bufferID)); 3298 3299 return onOMXEmptyBufferDone(bufferID); 3300 } 3301 3302 case omx_message::FILL_BUFFER_DONE: 3303 { 3304 IOMX::buffer_id bufferID; 3305 CHECK(msg->findInt32("buffer", (int32_t*)&bufferID)); 3306 3307 int32_t rangeOffset, rangeLength, flags; 3308 int64_t timeUs; 3309 3310 CHECK(msg->findInt32("range_offset", &rangeOffset)); 3311 CHECK(msg->findInt32("range_length", &rangeLength)); 3312 CHECK(msg->findInt32("flags", &flags)); 3313 CHECK(msg->findInt64("timestamp", &timeUs)); 3314 3315 return onOMXFillBufferDone( 3316 bufferID, 3317 (size_t)rangeOffset, (size_t)rangeLength, 3318 (OMX_U32)flags, 3319 timeUs); 3320 } 3321 3322 default: 3323 TRESPASS(); 3324 break; 3325 } 3326} 3327 3328bool ACodec::BaseState::onOMXEvent( 3329 OMX_EVENTTYPE event, OMX_U32 data1, OMX_U32 data2) { 3330 if (event != OMX_EventError) { 3331 ALOGV("[%s] EVENT(%d, 0x%08lx, 0x%08lx)", 3332 mCodec->mComponentName.c_str(), event, data1, data2); 3333 3334 return false; 3335 } 3336 3337 ALOGE("[%s] ERROR(0x%08lx)", mCodec->mComponentName.c_str(), data1); 3338 3339 mCodec->signalError((OMX_ERRORTYPE)data1); 3340 3341 return true; 3342} 3343 3344bool ACodec::BaseState::onOMXEmptyBufferDone(IOMX::buffer_id bufferID) { 3345 ALOGV("[%s] onOMXEmptyBufferDone %p", 3346 mCodec->mComponentName.c_str(), bufferID); 3347 3348 BufferInfo *info = 3349 mCodec->findBufferByID(kPortIndexInput, bufferID); 3350 3351 CHECK_EQ((int)info->mStatus, (int)BufferInfo::OWNED_BY_COMPONENT); 3352 info->mStatus = BufferInfo::OWNED_BY_US; 3353 3354 const sp<AMessage> &bufferMeta = info->mData->meta(); 3355 void *mediaBuffer; 3356 if (bufferMeta->findPointer("mediaBuffer", &mediaBuffer) 3357 && mediaBuffer != NULL) { 3358 // We're in "store-metadata-in-buffers" mode, the underlying 3359 // OMX component had access to data that's implicitly refcounted 3360 // by this "mediaBuffer" object. Now that the OMX component has 3361 // told us that it's done with the input buffer, we can decrement 3362 // the mediaBuffer's reference count. 3363 3364 ALOGV("releasing mbuf %p", mediaBuffer); 3365 3366 ((MediaBuffer *)mediaBuffer)->release(); 3367 mediaBuffer = NULL; 3368 3369 bufferMeta->setPointer("mediaBuffer", NULL); 3370 } 3371 3372 PortMode mode = getPortMode(kPortIndexInput); 3373 3374 switch (mode) { 3375 case KEEP_BUFFERS: 3376 break; 3377 3378 case RESUBMIT_BUFFERS: 3379 postFillThisBuffer(info); 3380 break; 3381 3382 default: 3383 { 3384 CHECK_EQ((int)mode, (int)FREE_BUFFERS); 3385 TRESPASS(); // Not currently used 3386 break; 3387 } 3388 } 3389 3390 return true; 3391} 3392 3393void ACodec::BaseState::postFillThisBuffer(BufferInfo *info) { 3394 if (mCodec->mPortEOS[kPortIndexInput]) { 3395 return; 3396 } 3397 3398 CHECK_EQ((int)info->mStatus, (int)BufferInfo::OWNED_BY_US); 3399 3400 sp<AMessage> notify = mCodec->mNotify->dup(); 3401 notify->setInt32("what", CodecBase::kWhatFillThisBuffer); 3402 notify->setInt32("buffer-id", info->mBufferID); 3403 3404 info->mData->meta()->clear(); 3405 notify->setBuffer("buffer", info->mData); 3406 3407 sp<AMessage> reply = new AMessage(kWhatInputBufferFilled, mCodec->id()); 3408 reply->setInt32("buffer-id", info->mBufferID); 3409 3410 notify->setMessage("reply", reply); 3411 3412 notify->post(); 3413 3414 info->mStatus = BufferInfo::OWNED_BY_UPSTREAM; 3415} 3416 3417void ACodec::BaseState::onInputBufferFilled(const sp<AMessage> &msg) { 3418 IOMX::buffer_id bufferID; 3419 CHECK(msg->findInt32("buffer-id", (int32_t*)&bufferID)); 3420 sp<ABuffer> buffer; 3421 int32_t err = OK; 3422 bool eos = false; 3423 PortMode mode = getPortMode(kPortIndexInput); 3424 3425 if (!msg->findBuffer("buffer", &buffer)) { 3426 /* these are unfilled buffers returned by client */ 3427 CHECK(msg->findInt32("err", &err)); 3428 3429 if (err == OK) { 3430 /* buffers with no errors are returned on MediaCodec.flush */ 3431 mode = KEEP_BUFFERS; 3432 } else { 3433 ALOGV("[%s] saw error %d instead of an input buffer", 3434 mCodec->mComponentName.c_str(), err); 3435 eos = true; 3436 } 3437 3438 buffer.clear(); 3439 } 3440 3441 int32_t tmp; 3442 if (buffer != NULL && buffer->meta()->findInt32("eos", &tmp) && tmp) { 3443 eos = true; 3444 err = ERROR_END_OF_STREAM; 3445 } 3446 3447 BufferInfo *info = mCodec->findBufferByID(kPortIndexInput, bufferID); 3448 CHECK_EQ((int)info->mStatus, (int)BufferInfo::OWNED_BY_UPSTREAM); 3449 3450 info->mStatus = BufferInfo::OWNED_BY_US; 3451 3452 switch (mode) { 3453 case KEEP_BUFFERS: 3454 { 3455 if (eos) { 3456 if (!mCodec->mPortEOS[kPortIndexInput]) { 3457 mCodec->mPortEOS[kPortIndexInput] = true; 3458 mCodec->mInputEOSResult = err; 3459 } 3460 } 3461 break; 3462 } 3463 3464 case RESUBMIT_BUFFERS: 3465 { 3466 if (buffer != NULL && !mCodec->mPortEOS[kPortIndexInput]) { 3467 int64_t timeUs; 3468 CHECK(buffer->meta()->findInt64("timeUs", &timeUs)); 3469 3470 OMX_U32 flags = OMX_BUFFERFLAG_ENDOFFRAME; 3471 3472 int32_t isCSD; 3473 if (buffer->meta()->findInt32("csd", &isCSD) && isCSD != 0) { 3474 flags |= OMX_BUFFERFLAG_CODECCONFIG; 3475 } 3476 3477 if (eos) { 3478 flags |= OMX_BUFFERFLAG_EOS; 3479 } 3480 3481 if (buffer != info->mData) { 3482 ALOGV("[%s] Needs to copy input data for buffer %p. (%p != %p)", 3483 mCodec->mComponentName.c_str(), 3484 bufferID, 3485 buffer.get(), info->mData.get()); 3486 3487 CHECK_LE(buffer->size(), info->mData->capacity()); 3488 memcpy(info->mData->data(), buffer->data(), buffer->size()); 3489 } 3490 3491 if (flags & OMX_BUFFERFLAG_CODECCONFIG) { 3492 ALOGV("[%s] calling emptyBuffer %p w/ codec specific data", 3493 mCodec->mComponentName.c_str(), bufferID); 3494 } else if (flags & OMX_BUFFERFLAG_EOS) { 3495 ALOGV("[%s] calling emptyBuffer %p w/ EOS", 3496 mCodec->mComponentName.c_str(), bufferID); 3497 } else { 3498#if TRACK_BUFFER_TIMING 3499 ALOGI("[%s] calling emptyBuffer %p w/ time %lld us", 3500 mCodec->mComponentName.c_str(), bufferID, timeUs); 3501#else 3502 ALOGV("[%s] calling emptyBuffer %p w/ time %lld us", 3503 mCodec->mComponentName.c_str(), bufferID, timeUs); 3504#endif 3505 } 3506 3507#if TRACK_BUFFER_TIMING 3508 ACodec::BufferStats stats; 3509 stats.mEmptyBufferTimeUs = ALooper::GetNowUs(); 3510 stats.mFillBufferDoneTimeUs = -1ll; 3511 mCodec->mBufferStats.add(timeUs, stats); 3512#endif 3513 3514 if (mCodec->mStoreMetaDataInOutputBuffers) { 3515 // try to submit an output buffer for each input buffer 3516 PortMode outputMode = getPortMode(kPortIndexOutput); 3517 3518 ALOGV("MetaDataBuffersToSubmit=%u portMode=%s", 3519 mCodec->mMetaDataBuffersToSubmit, 3520 (outputMode == FREE_BUFFERS ? "FREE" : 3521 outputMode == KEEP_BUFFERS ? "KEEP" : "RESUBMIT")); 3522 if (outputMode == RESUBMIT_BUFFERS) { 3523 CHECK_EQ(mCodec->submitOutputMetaDataBuffer(), 3524 (status_t)OK); 3525 } 3526 } 3527 3528 CHECK_EQ(mCodec->mOMX->emptyBuffer( 3529 mCodec->mNode, 3530 bufferID, 3531 0, 3532 buffer->size(), 3533 flags, 3534 timeUs), 3535 (status_t)OK); 3536 3537 info->mStatus = BufferInfo::OWNED_BY_COMPONENT; 3538 3539 if (!eos) { 3540 getMoreInputDataIfPossible(); 3541 } else { 3542 ALOGV("[%s] Signalled EOS on the input port", 3543 mCodec->mComponentName.c_str()); 3544 3545 mCodec->mPortEOS[kPortIndexInput] = true; 3546 mCodec->mInputEOSResult = err; 3547 } 3548 } else if (!mCodec->mPortEOS[kPortIndexInput]) { 3549 if (err != ERROR_END_OF_STREAM) { 3550 ALOGV("[%s] Signalling EOS on the input port " 3551 "due to error %d", 3552 mCodec->mComponentName.c_str(), err); 3553 } else { 3554 ALOGV("[%s] Signalling EOS on the input port", 3555 mCodec->mComponentName.c_str()); 3556 } 3557 3558 ALOGV("[%s] calling emptyBuffer %p signalling EOS", 3559 mCodec->mComponentName.c_str(), bufferID); 3560 3561 CHECK_EQ(mCodec->mOMX->emptyBuffer( 3562 mCodec->mNode, 3563 bufferID, 3564 0, 3565 0, 3566 OMX_BUFFERFLAG_EOS, 3567 0), 3568 (status_t)OK); 3569 3570 info->mStatus = BufferInfo::OWNED_BY_COMPONENT; 3571 3572 mCodec->mPortEOS[kPortIndexInput] = true; 3573 mCodec->mInputEOSResult = err; 3574 } 3575 break; 3576 } 3577 3578 default: 3579 CHECK_EQ((int)mode, (int)FREE_BUFFERS); 3580 break; 3581 } 3582} 3583 3584void ACodec::BaseState::getMoreInputDataIfPossible() { 3585 if (mCodec->mPortEOS[kPortIndexInput]) { 3586 return; 3587 } 3588 3589 BufferInfo *eligible = NULL; 3590 3591 for (size_t i = 0; i < mCodec->mBuffers[kPortIndexInput].size(); ++i) { 3592 BufferInfo *info = &mCodec->mBuffers[kPortIndexInput].editItemAt(i); 3593 3594#if 0 3595 if (info->mStatus == BufferInfo::OWNED_BY_UPSTREAM) { 3596 // There's already a "read" pending. 3597 return; 3598 } 3599#endif 3600 3601 if (info->mStatus == BufferInfo::OWNED_BY_US) { 3602 eligible = info; 3603 } 3604 } 3605 3606 if (eligible == NULL) { 3607 return; 3608 } 3609 3610 postFillThisBuffer(eligible); 3611} 3612 3613bool ACodec::BaseState::onOMXFillBufferDone( 3614 IOMX::buffer_id bufferID, 3615 size_t rangeOffset, size_t rangeLength, 3616 OMX_U32 flags, 3617 int64_t timeUs) { 3618 ALOGV("[%s] onOMXFillBufferDone %u time %" PRId64 " us, flags = 0x%08x", 3619 mCodec->mComponentName.c_str(), bufferID, timeUs, flags); 3620 3621 ssize_t index; 3622 3623#if TRACK_BUFFER_TIMING 3624 index = mCodec->mBufferStats.indexOfKey(timeUs); 3625 if (index >= 0) { 3626 ACodec::BufferStats *stats = &mCodec->mBufferStats.editValueAt(index); 3627 stats->mFillBufferDoneTimeUs = ALooper::GetNowUs(); 3628 3629 ALOGI("frame PTS %lld: %lld", 3630 timeUs, 3631 stats->mFillBufferDoneTimeUs - stats->mEmptyBufferTimeUs); 3632 3633 mCodec->mBufferStats.removeItemsAt(index); 3634 stats = NULL; 3635 } 3636#endif 3637 3638 BufferInfo *info = 3639 mCodec->findBufferByID(kPortIndexOutput, bufferID, &index); 3640 3641 CHECK_EQ((int)info->mStatus, (int)BufferInfo::OWNED_BY_COMPONENT); 3642 3643 info->mDequeuedAt = ++mCodec->mDequeueCounter; 3644 info->mStatus = BufferInfo::OWNED_BY_US; 3645 3646 PortMode mode = getPortMode(kPortIndexOutput); 3647 3648 switch (mode) { 3649 case KEEP_BUFFERS: 3650 break; 3651 3652 case RESUBMIT_BUFFERS: 3653 { 3654 if (rangeLength == 0 && !(flags & OMX_BUFFERFLAG_EOS)) { 3655 ALOGV("[%s] calling fillBuffer %u", 3656 mCodec->mComponentName.c_str(), info->mBufferID); 3657 3658 CHECK_EQ(mCodec->mOMX->fillBuffer( 3659 mCodec->mNode, info->mBufferID), 3660 (status_t)OK); 3661 3662 info->mStatus = BufferInfo::OWNED_BY_COMPONENT; 3663 break; 3664 } 3665 3666 sp<AMessage> reply = 3667 new AMessage(kWhatOutputBufferDrained, mCodec->id()); 3668 3669 if (!mCodec->mSentFormat && rangeLength > 0) { 3670 mCodec->sendFormatChange(reply); 3671 } 3672 3673 if (mCodec->mUseMetadataOnEncoderOutput) { 3674 native_handle_t* handle = 3675 *(native_handle_t**)(info->mData->data() + 4); 3676 info->mData->meta()->setPointer("handle", handle); 3677 info->mData->meta()->setInt32("rangeOffset", rangeOffset); 3678 info->mData->meta()->setInt32("rangeLength", rangeLength); 3679 } else { 3680 info->mData->setRange(rangeOffset, rangeLength); 3681 } 3682#if 0 3683 if (mCodec->mNativeWindow == NULL) { 3684 if (IsIDR(info->mData)) { 3685 ALOGI("IDR frame"); 3686 } 3687 } 3688#endif 3689 3690 if (mCodec->mSkipCutBuffer != NULL) { 3691 mCodec->mSkipCutBuffer->submit(info->mData); 3692 } 3693 info->mData->meta()->setInt64("timeUs", timeUs); 3694 3695 sp<AMessage> notify = mCodec->mNotify->dup(); 3696 notify->setInt32("what", CodecBase::kWhatDrainThisBuffer); 3697 notify->setInt32("buffer-id", info->mBufferID); 3698 notify->setBuffer("buffer", info->mData); 3699 notify->setInt32("flags", flags); 3700 3701 reply->setInt32("buffer-id", info->mBufferID); 3702 3703 notify->setMessage("reply", reply); 3704 3705 notify->post(); 3706 3707 info->mStatus = BufferInfo::OWNED_BY_DOWNSTREAM; 3708 3709 if (flags & OMX_BUFFERFLAG_EOS) { 3710 ALOGV("[%s] saw output EOS", mCodec->mComponentName.c_str()); 3711 3712 sp<AMessage> notify = mCodec->mNotify->dup(); 3713 notify->setInt32("what", CodecBase::kWhatEOS); 3714 notify->setInt32("err", mCodec->mInputEOSResult); 3715 notify->post(); 3716 3717 mCodec->mPortEOS[kPortIndexOutput] = true; 3718 } 3719 break; 3720 } 3721 3722 default: 3723 { 3724 CHECK_EQ((int)mode, (int)FREE_BUFFERS); 3725 3726 CHECK_EQ((status_t)OK, 3727 mCodec->freeBuffer(kPortIndexOutput, index)); 3728 break; 3729 } 3730 } 3731 3732 return true; 3733} 3734 3735void ACodec::BaseState::onOutputBufferDrained(const sp<AMessage> &msg) { 3736 IOMX::buffer_id bufferID; 3737 CHECK(msg->findInt32("buffer-id", (int32_t*)&bufferID)); 3738 ssize_t index; 3739 BufferInfo *info = 3740 mCodec->findBufferByID(kPortIndexOutput, bufferID, &index); 3741 CHECK_EQ((int)info->mStatus, (int)BufferInfo::OWNED_BY_DOWNSTREAM); 3742 3743 android_native_rect_t crop; 3744 if (msg->findRect("crop", 3745 &crop.left, &crop.top, &crop.right, &crop.bottom)) { 3746 CHECK_EQ(0, native_window_set_crop( 3747 mCodec->mNativeWindow.get(), &crop)); 3748 } 3749 3750 int32_t render; 3751 if (mCodec->mNativeWindow != NULL 3752 && msg->findInt32("render", &render) && render != 0 3753 && info->mData != NULL && info->mData->size() != 0) { 3754 ATRACE_NAME("render"); 3755 // The client wants this buffer to be rendered. 3756 3757 int64_t timestampNs = 0; 3758 if (!msg->findInt64("timestampNs", ×tampNs)) { 3759 // TODO: it seems like we should use the timestamp 3760 // in the (media)buffer as it potentially came from 3761 // an input surface, but we did not propagate it prior to 3762 // API 20. Perhaps check for target SDK version. 3763#if 0 3764 if (info->mData->meta()->findInt64("timeUs", ×tampNs)) { 3765 ALOGV("using buffer PTS of %" PRId64, timestampNs); 3766 timestampNs *= 1000; 3767 } 3768#endif 3769 } 3770 3771 status_t err; 3772 err = native_window_set_buffers_timestamp(mCodec->mNativeWindow.get(), timestampNs); 3773 if (err != OK) { 3774 ALOGW("failed to set buffer timestamp: %d", err); 3775 } 3776 3777 if ((err = mCodec->mNativeWindow->queueBuffer( 3778 mCodec->mNativeWindow.get(), 3779 info->mGraphicBuffer.get(), -1)) == OK) { 3780 info->mStatus = BufferInfo::OWNED_BY_NATIVE_WINDOW; 3781 } else { 3782 mCodec->signalError(OMX_ErrorUndefined, err); 3783 info->mStatus = BufferInfo::OWNED_BY_US; 3784 } 3785 } else { 3786 if (mCodec->mNativeWindow != NULL && 3787 (info->mData == NULL || info->mData->size() != 0)) { 3788 ATRACE_NAME("frame-drop"); 3789 } 3790 info->mStatus = BufferInfo::OWNED_BY_US; 3791 } 3792 3793 PortMode mode = getPortMode(kPortIndexOutput); 3794 3795 switch (mode) { 3796 case KEEP_BUFFERS: 3797 { 3798 // XXX fishy, revisit!!! What about the FREE_BUFFERS case below? 3799 3800 if (info->mStatus == BufferInfo::OWNED_BY_NATIVE_WINDOW) { 3801 // We cannot resubmit the buffer we just rendered, dequeue 3802 // the spare instead. 3803 3804 info = mCodec->dequeueBufferFromNativeWindow(); 3805 } 3806 break; 3807 } 3808 3809 case RESUBMIT_BUFFERS: 3810 { 3811 if (!mCodec->mPortEOS[kPortIndexOutput]) { 3812 if (info->mStatus == BufferInfo::OWNED_BY_NATIVE_WINDOW) { 3813 // We cannot resubmit the buffer we just rendered, dequeue 3814 // the spare instead. 3815 3816 info = mCodec->dequeueBufferFromNativeWindow(); 3817 } 3818 3819 if (info != NULL) { 3820 ALOGV("[%s] calling fillBuffer %u", 3821 mCodec->mComponentName.c_str(), info->mBufferID); 3822 3823 CHECK_EQ(mCodec->mOMX->fillBuffer(mCodec->mNode, info->mBufferID), 3824 (status_t)OK); 3825 3826 info->mStatus = BufferInfo::OWNED_BY_COMPONENT; 3827 } 3828 } 3829 break; 3830 } 3831 3832 default: 3833 { 3834 CHECK_EQ((int)mode, (int)FREE_BUFFERS); 3835 3836 CHECK_EQ((status_t)OK, 3837 mCodec->freeBuffer(kPortIndexOutput, index)); 3838 break; 3839 } 3840 } 3841} 3842 3843//////////////////////////////////////////////////////////////////////////////// 3844 3845ACodec::UninitializedState::UninitializedState(ACodec *codec) 3846 : BaseState(codec) { 3847} 3848 3849void ACodec::UninitializedState::stateEntered() { 3850 ALOGV("Now uninitialized"); 3851 3852 if (mDeathNotifier != NULL) { 3853 mCodec->mOMX->asBinder()->unlinkToDeath(mDeathNotifier); 3854 mDeathNotifier.clear(); 3855 } 3856 3857 mCodec->mNativeWindow.clear(); 3858 mCodec->mNode = NULL; 3859 mCodec->mOMX.clear(); 3860 mCodec->mQuirks = 0; 3861 mCodec->mFlags = 0; 3862 mCodec->mUseMetadataOnEncoderOutput = 0; 3863 mCodec->mComponentName.clear(); 3864} 3865 3866bool ACodec::UninitializedState::onMessageReceived(const sp<AMessage> &msg) { 3867 bool handled = false; 3868 3869 switch (msg->what()) { 3870 case ACodec::kWhatSetup: 3871 { 3872 onSetup(msg); 3873 3874 handled = true; 3875 break; 3876 } 3877 3878 case ACodec::kWhatAllocateComponent: 3879 { 3880 onAllocateComponent(msg); 3881 handled = true; 3882 break; 3883 } 3884 3885 case ACodec::kWhatShutdown: 3886 { 3887 int32_t keepComponentAllocated; 3888 CHECK(msg->findInt32( 3889 "keepComponentAllocated", &keepComponentAllocated)); 3890 ALOGW_IF(keepComponentAllocated, 3891 "cannot keep component allocated on shutdown in Uninitialized state"); 3892 3893 sp<AMessage> notify = mCodec->mNotify->dup(); 3894 notify->setInt32("what", CodecBase::kWhatShutdownCompleted); 3895 notify->post(); 3896 3897 handled = true; 3898 break; 3899 } 3900 3901 case ACodec::kWhatFlush: 3902 { 3903 sp<AMessage> notify = mCodec->mNotify->dup(); 3904 notify->setInt32("what", CodecBase::kWhatFlushCompleted); 3905 notify->post(); 3906 3907 handled = true; 3908 break; 3909 } 3910 3911 default: 3912 return BaseState::onMessageReceived(msg); 3913 } 3914 3915 return handled; 3916} 3917 3918void ACodec::UninitializedState::onSetup( 3919 const sp<AMessage> &msg) { 3920 if (onAllocateComponent(msg) 3921 && mCodec->mLoadedState->onConfigureComponent(msg)) { 3922 mCodec->mLoadedState->onStart(); 3923 } 3924} 3925 3926bool ACodec::UninitializedState::onAllocateComponent(const sp<AMessage> &msg) { 3927 ALOGV("onAllocateComponent"); 3928 3929 CHECK(mCodec->mNode == NULL); 3930 3931 OMXClient client; 3932 CHECK_EQ(client.connect(), (status_t)OK); 3933 3934 sp<IOMX> omx = client.interface(); 3935 3936 sp<AMessage> notify = new AMessage(kWhatOMXDied, mCodec->id()); 3937 3938 mDeathNotifier = new DeathNotifier(notify); 3939 if (omx->asBinder()->linkToDeath(mDeathNotifier) != OK) { 3940 // This was a local binder, if it dies so do we, we won't care 3941 // about any notifications in the afterlife. 3942 mDeathNotifier.clear(); 3943 } 3944 3945 Vector<OMXCodec::CodecNameAndQuirks> matchingCodecs; 3946 3947 AString mime; 3948 3949 AString componentName; 3950 uint32_t quirks = 0; 3951 int32_t encoder = false; 3952 if (msg->findString("componentName", &componentName)) { 3953 ssize_t index = matchingCodecs.add(); 3954 OMXCodec::CodecNameAndQuirks *entry = &matchingCodecs.editItemAt(index); 3955 entry->mName = String8(componentName.c_str()); 3956 3957 if (!OMXCodec::findCodecQuirks( 3958 componentName.c_str(), &entry->mQuirks)) { 3959 entry->mQuirks = 0; 3960 } 3961 } else { 3962 CHECK(msg->findString("mime", &mime)); 3963 3964 if (!msg->findInt32("encoder", &encoder)) { 3965 encoder = false; 3966 } 3967 3968 OMXCodec::findMatchingCodecs( 3969 mime.c_str(), 3970 encoder, // createEncoder 3971 NULL, // matchComponentName 3972 0, // flags 3973 &matchingCodecs); 3974 } 3975 3976 sp<CodecObserver> observer = new CodecObserver; 3977 IOMX::node_id node = NULL; 3978 3979 for (size_t matchIndex = 0; matchIndex < matchingCodecs.size(); 3980 ++matchIndex) { 3981 componentName = matchingCodecs.itemAt(matchIndex).mName.string(); 3982 quirks = matchingCodecs.itemAt(matchIndex).mQuirks; 3983 3984 pid_t tid = androidGetTid(); 3985 int prevPriority = androidGetThreadPriority(tid); 3986 androidSetThreadPriority(tid, ANDROID_PRIORITY_FOREGROUND); 3987 status_t err = omx->allocateNode(componentName.c_str(), observer, &node); 3988 androidSetThreadPriority(tid, prevPriority); 3989 3990 if (err == OK) { 3991 break; 3992 } else { 3993 ALOGW("Allocating component '%s' failed, try next one.", componentName.c_str()); 3994 } 3995 3996 node = NULL; 3997 } 3998 3999 if (node == NULL) { 4000 if (!mime.empty()) { 4001 ALOGE("Unable to instantiate a %scoder for type '%s'.", 4002 encoder ? "en" : "de", mime.c_str()); 4003 } else { 4004 ALOGE("Unable to instantiate codec '%s'.", componentName.c_str()); 4005 } 4006 4007 mCodec->signalError(OMX_ErrorComponentNotFound); 4008 return false; 4009 } 4010 4011 notify = new AMessage(kWhatOMXMessage, mCodec->id()); 4012 observer->setNotificationMessage(notify); 4013 4014 mCodec->mComponentName = componentName; 4015 mCodec->mFlags = 0; 4016 4017 if (componentName.endsWith(".secure")) { 4018 mCodec->mFlags |= kFlagIsSecure; 4019 mCodec->mFlags |= kFlagPushBlankBuffersToNativeWindowOnShutdown; 4020 } 4021 4022 mCodec->mQuirks = quirks; 4023 mCodec->mOMX = omx; 4024 mCodec->mNode = node; 4025 4026 { 4027 sp<AMessage> notify = mCodec->mNotify->dup(); 4028 notify->setInt32("what", CodecBase::kWhatComponentAllocated); 4029 notify->setString("componentName", mCodec->mComponentName.c_str()); 4030 notify->post(); 4031 } 4032 4033 mCodec->changeState(mCodec->mLoadedState); 4034 4035 return true; 4036} 4037 4038//////////////////////////////////////////////////////////////////////////////// 4039 4040ACodec::LoadedState::LoadedState(ACodec *codec) 4041 : BaseState(codec) { 4042} 4043 4044void ACodec::LoadedState::stateEntered() { 4045 ALOGV("[%s] Now Loaded", mCodec->mComponentName.c_str()); 4046 4047 mCodec->mPortEOS[kPortIndexInput] = 4048 mCodec->mPortEOS[kPortIndexOutput] = false; 4049 4050 mCodec->mInputEOSResult = OK; 4051 4052 mCodec->mDequeueCounter = 0; 4053 mCodec->mMetaDataBuffersToSubmit = 0; 4054 mCodec->mRepeatFrameDelayUs = -1ll; 4055 mCodec->mInputFormat.clear(); 4056 mCodec->mOutputFormat.clear(); 4057 4058 if (mCodec->mShutdownInProgress) { 4059 bool keepComponentAllocated = mCodec->mKeepComponentAllocated; 4060 4061 mCodec->mShutdownInProgress = false; 4062 mCodec->mKeepComponentAllocated = false; 4063 4064 onShutdown(keepComponentAllocated); 4065 } 4066 mCodec->mExplicitShutdown = false; 4067} 4068 4069void ACodec::LoadedState::onShutdown(bool keepComponentAllocated) { 4070 if (!keepComponentAllocated) { 4071 CHECK_EQ(mCodec->mOMX->freeNode(mCodec->mNode), (status_t)OK); 4072 4073 mCodec->changeState(mCodec->mUninitializedState); 4074 } 4075 4076 if (mCodec->mExplicitShutdown) { 4077 sp<AMessage> notify = mCodec->mNotify->dup(); 4078 notify->setInt32("what", CodecBase::kWhatShutdownCompleted); 4079 notify->post(); 4080 mCodec->mExplicitShutdown = false; 4081 } 4082} 4083 4084bool ACodec::LoadedState::onMessageReceived(const sp<AMessage> &msg) { 4085 bool handled = false; 4086 4087 switch (msg->what()) { 4088 case ACodec::kWhatConfigureComponent: 4089 { 4090 onConfigureComponent(msg); 4091 handled = true; 4092 break; 4093 } 4094 4095 case ACodec::kWhatCreateInputSurface: 4096 { 4097 onCreateInputSurface(msg); 4098 handled = true; 4099 break; 4100 } 4101 4102 case ACodec::kWhatStart: 4103 { 4104 onStart(); 4105 handled = true; 4106 break; 4107 } 4108 4109 case ACodec::kWhatShutdown: 4110 { 4111 int32_t keepComponentAllocated; 4112 CHECK(msg->findInt32( 4113 "keepComponentAllocated", &keepComponentAllocated)); 4114 4115 mCodec->mExplicitShutdown = true; 4116 onShutdown(keepComponentAllocated); 4117 4118 handled = true; 4119 break; 4120 } 4121 4122 case ACodec::kWhatFlush: 4123 { 4124 sp<AMessage> notify = mCodec->mNotify->dup(); 4125 notify->setInt32("what", CodecBase::kWhatFlushCompleted); 4126 notify->post(); 4127 4128 handled = true; 4129 break; 4130 } 4131 4132 default: 4133 return BaseState::onMessageReceived(msg); 4134 } 4135 4136 return handled; 4137} 4138 4139bool ACodec::LoadedState::onConfigureComponent( 4140 const sp<AMessage> &msg) { 4141 ALOGV("onConfigureComponent"); 4142 4143 CHECK(mCodec->mNode != NULL); 4144 4145 AString mime; 4146 CHECK(msg->findString("mime", &mime)); 4147 4148 status_t err = mCodec->configureCodec(mime.c_str(), msg); 4149 4150 if (err != OK) { 4151 ALOGE("[%s] configureCodec returning error %d", 4152 mCodec->mComponentName.c_str(), err); 4153 4154 mCodec->signalError(OMX_ErrorUndefined, err); 4155 return false; 4156 } 4157 4158 sp<RefBase> obj; 4159 if (msg->findObject("native-window", &obj) 4160 && strncmp("OMX.google.", mCodec->mComponentName.c_str(), 11)) { 4161 sp<NativeWindowWrapper> nativeWindow( 4162 static_cast<NativeWindowWrapper *>(obj.get())); 4163 CHECK(nativeWindow != NULL); 4164 mCodec->mNativeWindow = nativeWindow->getNativeWindow(); 4165 4166 native_window_set_scaling_mode( 4167 mCodec->mNativeWindow.get(), 4168 NATIVE_WINDOW_SCALING_MODE_SCALE_TO_WINDOW); 4169 } 4170 CHECK_EQ((status_t)OK, mCodec->initNativeWindow()); 4171 4172 { 4173 sp<AMessage> notify = mCodec->mNotify->dup(); 4174 notify->setInt32("what", CodecBase::kWhatComponentConfigured); 4175 notify->setMessage("input-format", mCodec->mInputFormat); 4176 notify->setMessage("output-format", mCodec->mOutputFormat); 4177 notify->post(); 4178 } 4179 4180 return true; 4181} 4182 4183void ACodec::LoadedState::onCreateInputSurface( 4184 const sp<AMessage> & /* msg */) { 4185 ALOGV("onCreateInputSurface"); 4186 4187 sp<AMessage> notify = mCodec->mNotify->dup(); 4188 notify->setInt32("what", CodecBase::kWhatInputSurfaceCreated); 4189 4190 sp<IGraphicBufferProducer> bufferProducer; 4191 status_t err; 4192 4193 err = mCodec->mOMX->createInputSurface(mCodec->mNode, kPortIndexInput, 4194 &bufferProducer); 4195 4196 if (err == OK && mCodec->mRepeatFrameDelayUs > 0ll) { 4197 err = mCodec->mOMX->setInternalOption( 4198 mCodec->mNode, 4199 kPortIndexInput, 4200 IOMX::INTERNAL_OPTION_REPEAT_PREVIOUS_FRAME_DELAY, 4201 &mCodec->mRepeatFrameDelayUs, 4202 sizeof(mCodec->mRepeatFrameDelayUs)); 4203 4204 if (err != OK) { 4205 ALOGE("[%s] Unable to configure option to repeat previous " 4206 "frames (err %d)", 4207 mCodec->mComponentName.c_str(), 4208 err); 4209 } 4210 } 4211 4212 if (err == OK && mCodec->mMaxPtsGapUs > 0ll) { 4213 err = mCodec->mOMX->setInternalOption( 4214 mCodec->mNode, 4215 kPortIndexInput, 4216 IOMX::INTERNAL_OPTION_MAX_TIMESTAMP_GAP, 4217 &mCodec->mMaxPtsGapUs, 4218 sizeof(mCodec->mMaxPtsGapUs)); 4219 4220 if (err != OK) { 4221 ALOGE("[%s] Unable to configure max timestamp gap (err %d)", 4222 mCodec->mComponentName.c_str(), 4223 err); 4224 } 4225 } 4226 4227 if (err == OK && mCodec->mTimePerCaptureUs > 0ll 4228 && mCodec->mTimePerFrameUs > 0ll) { 4229 int64_t timeLapse[2]; 4230 timeLapse[0] = mCodec->mTimePerFrameUs; 4231 timeLapse[1] = mCodec->mTimePerCaptureUs; 4232 err = mCodec->mOMX->setInternalOption( 4233 mCodec->mNode, 4234 kPortIndexInput, 4235 IOMX::INTERNAL_OPTION_TIME_LAPSE, 4236 &timeLapse[0], 4237 sizeof(timeLapse)); 4238 4239 if (err != OK) { 4240 ALOGE("[%s] Unable to configure time lapse (err %d)", 4241 mCodec->mComponentName.c_str(), 4242 err); 4243 } 4244 } 4245 4246 if (err == OK && mCodec->mCreateInputBuffersSuspended) { 4247 bool suspend = true; 4248 err = mCodec->mOMX->setInternalOption( 4249 mCodec->mNode, 4250 kPortIndexInput, 4251 IOMX::INTERNAL_OPTION_SUSPEND, 4252 &suspend, 4253 sizeof(suspend)); 4254 4255 if (err != OK) { 4256 ALOGE("[%s] Unable to configure option to suspend (err %d)", 4257 mCodec->mComponentName.c_str(), 4258 err); 4259 } 4260 } 4261 4262 if (err == OK) { 4263 notify->setObject("input-surface", 4264 new BufferProducerWrapper(bufferProducer)); 4265 } else { 4266 // Can't use mCodec->signalError() here -- MediaCodec won't forward 4267 // the error through because it's in the "configured" state. We 4268 // send a kWhatInputSurfaceCreated with an error value instead. 4269 ALOGE("[%s] onCreateInputSurface returning error %d", 4270 mCodec->mComponentName.c_str(), err); 4271 notify->setInt32("err", err); 4272 } 4273 notify->post(); 4274} 4275 4276void ACodec::LoadedState::onStart() { 4277 ALOGV("onStart"); 4278 4279 CHECK_EQ(mCodec->mOMX->sendCommand( 4280 mCodec->mNode, OMX_CommandStateSet, OMX_StateIdle), 4281 (status_t)OK); 4282 4283 mCodec->changeState(mCodec->mLoadedToIdleState); 4284} 4285 4286//////////////////////////////////////////////////////////////////////////////// 4287 4288ACodec::LoadedToIdleState::LoadedToIdleState(ACodec *codec) 4289 : BaseState(codec) { 4290} 4291 4292void ACodec::LoadedToIdleState::stateEntered() { 4293 ALOGV("[%s] Now Loaded->Idle", mCodec->mComponentName.c_str()); 4294 4295 status_t err; 4296 if ((err = allocateBuffers()) != OK) { 4297 ALOGE("Failed to allocate buffers after transitioning to IDLE state " 4298 "(error 0x%08x)", 4299 err); 4300 4301 mCodec->signalError(OMX_ErrorUndefined, err); 4302 4303 mCodec->changeState(mCodec->mLoadedState); 4304 } 4305} 4306 4307status_t ACodec::LoadedToIdleState::allocateBuffers() { 4308 status_t err = mCodec->allocateBuffersOnPort(kPortIndexInput); 4309 4310 if (err != OK) { 4311 return err; 4312 } 4313 4314 return mCodec->allocateBuffersOnPort(kPortIndexOutput); 4315} 4316 4317bool ACodec::LoadedToIdleState::onMessageReceived(const sp<AMessage> &msg) { 4318 switch (msg->what()) { 4319 case kWhatSetParameters: 4320 case kWhatShutdown: 4321 { 4322 mCodec->deferMessage(msg); 4323 return true; 4324 } 4325 4326 case kWhatSignalEndOfInputStream: 4327 { 4328 mCodec->onSignalEndOfInputStream(); 4329 return true; 4330 } 4331 4332 case kWhatResume: 4333 { 4334 // We'll be active soon enough. 4335 return true; 4336 } 4337 4338 case kWhatFlush: 4339 { 4340 // We haven't even started yet, so we're flushed alright... 4341 sp<AMessage> notify = mCodec->mNotify->dup(); 4342 notify->setInt32("what", CodecBase::kWhatFlushCompleted); 4343 notify->post(); 4344 return true; 4345 } 4346 4347 default: 4348 return BaseState::onMessageReceived(msg); 4349 } 4350} 4351 4352bool ACodec::LoadedToIdleState::onOMXEvent( 4353 OMX_EVENTTYPE event, OMX_U32 data1, OMX_U32 data2) { 4354 switch (event) { 4355 case OMX_EventCmdComplete: 4356 { 4357 CHECK_EQ(data1, (OMX_U32)OMX_CommandStateSet); 4358 CHECK_EQ(data2, (OMX_U32)OMX_StateIdle); 4359 4360 CHECK_EQ(mCodec->mOMX->sendCommand( 4361 mCodec->mNode, OMX_CommandStateSet, OMX_StateExecuting), 4362 (status_t)OK); 4363 4364 mCodec->changeState(mCodec->mIdleToExecutingState); 4365 4366 return true; 4367 } 4368 4369 default: 4370 return BaseState::onOMXEvent(event, data1, data2); 4371 } 4372} 4373 4374//////////////////////////////////////////////////////////////////////////////// 4375 4376ACodec::IdleToExecutingState::IdleToExecutingState(ACodec *codec) 4377 : BaseState(codec) { 4378} 4379 4380void ACodec::IdleToExecutingState::stateEntered() { 4381 ALOGV("[%s] Now Idle->Executing", mCodec->mComponentName.c_str()); 4382} 4383 4384bool ACodec::IdleToExecutingState::onMessageReceived(const sp<AMessage> &msg) { 4385 switch (msg->what()) { 4386 case kWhatSetParameters: 4387 case kWhatShutdown: 4388 { 4389 mCodec->deferMessage(msg); 4390 return true; 4391 } 4392 4393 case kWhatResume: 4394 { 4395 // We'll be active soon enough. 4396 return true; 4397 } 4398 4399 case kWhatFlush: 4400 { 4401 // We haven't even started yet, so we're flushed alright... 4402 sp<AMessage> notify = mCodec->mNotify->dup(); 4403 notify->setInt32("what", CodecBase::kWhatFlushCompleted); 4404 notify->post(); 4405 4406 return true; 4407 } 4408 4409 case kWhatSignalEndOfInputStream: 4410 { 4411 mCodec->onSignalEndOfInputStream(); 4412 return true; 4413 } 4414 4415 default: 4416 return BaseState::onMessageReceived(msg); 4417 } 4418} 4419 4420bool ACodec::IdleToExecutingState::onOMXEvent( 4421 OMX_EVENTTYPE event, OMX_U32 data1, OMX_U32 data2) { 4422 switch (event) { 4423 case OMX_EventCmdComplete: 4424 { 4425 CHECK_EQ(data1, (OMX_U32)OMX_CommandStateSet); 4426 CHECK_EQ(data2, (OMX_U32)OMX_StateExecuting); 4427 4428 mCodec->mExecutingState->resume(); 4429 mCodec->changeState(mCodec->mExecutingState); 4430 4431 return true; 4432 } 4433 4434 default: 4435 return BaseState::onOMXEvent(event, data1, data2); 4436 } 4437} 4438 4439//////////////////////////////////////////////////////////////////////////////// 4440 4441ACodec::ExecutingState::ExecutingState(ACodec *codec) 4442 : BaseState(codec), 4443 mActive(false) { 4444} 4445 4446ACodec::BaseState::PortMode ACodec::ExecutingState::getPortMode( 4447 OMX_U32 /* portIndex */) { 4448 return RESUBMIT_BUFFERS; 4449} 4450 4451void ACodec::ExecutingState::submitOutputMetaBuffers() { 4452 // submit as many buffers as there are input buffers with the codec 4453 // in case we are in port reconfiguring 4454 for (size_t i = 0; i < mCodec->mBuffers[kPortIndexInput].size(); ++i) { 4455 BufferInfo *info = &mCodec->mBuffers[kPortIndexInput].editItemAt(i); 4456 4457 if (info->mStatus == BufferInfo::OWNED_BY_COMPONENT) { 4458 if (mCodec->submitOutputMetaDataBuffer() != OK) 4459 break; 4460 } 4461 } 4462 4463 // *** NOTE: THE FOLLOWING WORKAROUND WILL BE REMOVED *** 4464 mCodec->signalSubmitOutputMetaDataBufferIfEOS_workaround(); 4465} 4466 4467void ACodec::ExecutingState::submitRegularOutputBuffers() { 4468 for (size_t i = 0; i < mCodec->mBuffers[kPortIndexOutput].size(); ++i) { 4469 BufferInfo *info = &mCodec->mBuffers[kPortIndexOutput].editItemAt(i); 4470 4471 if (mCodec->mNativeWindow != NULL) { 4472 CHECK(info->mStatus == BufferInfo::OWNED_BY_US 4473 || info->mStatus == BufferInfo::OWNED_BY_NATIVE_WINDOW); 4474 4475 if (info->mStatus == BufferInfo::OWNED_BY_NATIVE_WINDOW) { 4476 continue; 4477 } 4478 } else { 4479 CHECK_EQ((int)info->mStatus, (int)BufferInfo::OWNED_BY_US); 4480 } 4481 4482 ALOGV("[%s] calling fillBuffer %p", 4483 mCodec->mComponentName.c_str(), info->mBufferID); 4484 4485 CHECK_EQ(mCodec->mOMX->fillBuffer(mCodec->mNode, info->mBufferID), 4486 (status_t)OK); 4487 4488 info->mStatus = BufferInfo::OWNED_BY_COMPONENT; 4489 } 4490} 4491 4492void ACodec::ExecutingState::submitOutputBuffers() { 4493 submitRegularOutputBuffers(); 4494 if (mCodec->mStoreMetaDataInOutputBuffers) { 4495 submitOutputMetaBuffers(); 4496 } 4497} 4498 4499void ACodec::ExecutingState::resume() { 4500 if (mActive) { 4501 ALOGV("[%s] We're already active, no need to resume.", 4502 mCodec->mComponentName.c_str()); 4503 4504 return; 4505 } 4506 4507 submitOutputBuffers(); 4508 4509 // Post all available input buffers 4510 CHECK_GT(mCodec->mBuffers[kPortIndexInput].size(), 0u); 4511 for (size_t i = 0; i < mCodec->mBuffers[kPortIndexInput].size(); i++) { 4512 BufferInfo *info = &mCodec->mBuffers[kPortIndexInput].editItemAt(i); 4513 if (info->mStatus == BufferInfo::OWNED_BY_US) { 4514 postFillThisBuffer(info); 4515 } 4516 } 4517 4518 mActive = true; 4519} 4520 4521void ACodec::ExecutingState::stateEntered() { 4522 ALOGV("[%s] Now Executing", mCodec->mComponentName.c_str()); 4523 4524 mCodec->processDeferredMessages(); 4525} 4526 4527bool ACodec::ExecutingState::onMessageReceived(const sp<AMessage> &msg) { 4528 bool handled = false; 4529 4530 switch (msg->what()) { 4531 case kWhatShutdown: 4532 { 4533 int32_t keepComponentAllocated; 4534 CHECK(msg->findInt32( 4535 "keepComponentAllocated", &keepComponentAllocated)); 4536 4537 mCodec->mShutdownInProgress = true; 4538 mCodec->mExplicitShutdown = true; 4539 mCodec->mKeepComponentAllocated = keepComponentAllocated; 4540 4541 mActive = false; 4542 4543 CHECK_EQ(mCodec->mOMX->sendCommand( 4544 mCodec->mNode, OMX_CommandStateSet, OMX_StateIdle), 4545 (status_t)OK); 4546 4547 mCodec->changeState(mCodec->mExecutingToIdleState); 4548 4549 handled = true; 4550 break; 4551 } 4552 4553 case kWhatFlush: 4554 { 4555 ALOGV("[%s] ExecutingState flushing now " 4556 "(codec owns %d/%d input, %d/%d output).", 4557 mCodec->mComponentName.c_str(), 4558 mCodec->countBuffersOwnedByComponent(kPortIndexInput), 4559 mCodec->mBuffers[kPortIndexInput].size(), 4560 mCodec->countBuffersOwnedByComponent(kPortIndexOutput), 4561 mCodec->mBuffers[kPortIndexOutput].size()); 4562 4563 mActive = false; 4564 4565 CHECK_EQ(mCodec->mOMX->sendCommand( 4566 mCodec->mNode, OMX_CommandFlush, OMX_ALL), 4567 (status_t)OK); 4568 4569 mCodec->changeState(mCodec->mFlushingState); 4570 handled = true; 4571 break; 4572 } 4573 4574 case kWhatResume: 4575 { 4576 resume(); 4577 4578 handled = true; 4579 break; 4580 } 4581 4582 case kWhatRequestIDRFrame: 4583 { 4584 status_t err = mCodec->requestIDRFrame(); 4585 if (err != OK) { 4586 ALOGW("Requesting an IDR frame failed."); 4587 } 4588 4589 handled = true; 4590 break; 4591 } 4592 4593 case kWhatSetParameters: 4594 { 4595 sp<AMessage> params; 4596 CHECK(msg->findMessage("params", ¶ms)); 4597 4598 status_t err = mCodec->setParameters(params); 4599 4600 sp<AMessage> reply; 4601 if (msg->findMessage("reply", &reply)) { 4602 reply->setInt32("err", err); 4603 reply->post(); 4604 } 4605 4606 handled = true; 4607 break; 4608 } 4609 4610 case ACodec::kWhatSignalEndOfInputStream: 4611 { 4612 mCodec->onSignalEndOfInputStream(); 4613 handled = true; 4614 break; 4615 } 4616 4617 // *** NOTE: THE FOLLOWING WORKAROUND WILL BE REMOVED *** 4618 case kWhatSubmitOutputMetaDataBufferIfEOS: 4619 { 4620 if (mCodec->mPortEOS[kPortIndexInput] && 4621 !mCodec->mPortEOS[kPortIndexOutput]) { 4622 status_t err = mCodec->submitOutputMetaDataBuffer(); 4623 if (err == OK) { 4624 mCodec->signalSubmitOutputMetaDataBufferIfEOS_workaround(); 4625 } 4626 } 4627 return true; 4628 } 4629 4630 default: 4631 handled = BaseState::onMessageReceived(msg); 4632 break; 4633 } 4634 4635 return handled; 4636} 4637 4638status_t ACodec::setParameters(const sp<AMessage> ¶ms) { 4639 int32_t videoBitrate; 4640 if (params->findInt32("video-bitrate", &videoBitrate)) { 4641 OMX_VIDEO_CONFIG_BITRATETYPE configParams; 4642 InitOMXParams(&configParams); 4643 configParams.nPortIndex = kPortIndexOutput; 4644 configParams.nEncodeBitrate = videoBitrate; 4645 4646 status_t err = mOMX->setConfig( 4647 mNode, 4648 OMX_IndexConfigVideoBitrate, 4649 &configParams, 4650 sizeof(configParams)); 4651 4652 if (err != OK) { 4653 ALOGE("setConfig(OMX_IndexConfigVideoBitrate, %d) failed w/ err %d", 4654 videoBitrate, err); 4655 4656 return err; 4657 } 4658 } 4659 4660 int64_t skipFramesBeforeUs; 4661 if (params->findInt64("skip-frames-before", &skipFramesBeforeUs)) { 4662 status_t err = 4663 mOMX->setInternalOption( 4664 mNode, 4665 kPortIndexInput, 4666 IOMX::INTERNAL_OPTION_START_TIME, 4667 &skipFramesBeforeUs, 4668 sizeof(skipFramesBeforeUs)); 4669 4670 if (err != OK) { 4671 ALOGE("Failed to set parameter 'skip-frames-before' (err %d)", err); 4672 return err; 4673 } 4674 } 4675 4676 int32_t dropInputFrames; 4677 if (params->findInt32("drop-input-frames", &dropInputFrames)) { 4678 bool suspend = dropInputFrames != 0; 4679 4680 status_t err = 4681 mOMX->setInternalOption( 4682 mNode, 4683 kPortIndexInput, 4684 IOMX::INTERNAL_OPTION_SUSPEND, 4685 &suspend, 4686 sizeof(suspend)); 4687 4688 if (err != OK) { 4689 ALOGE("Failed to set parameter 'drop-input-frames' (err %d)", err); 4690 return err; 4691 } 4692 } 4693 4694 int32_t dummy; 4695 if (params->findInt32("request-sync", &dummy)) { 4696 status_t err = requestIDRFrame(); 4697 4698 if (err != OK) { 4699 ALOGE("Requesting a sync frame failed w/ err %d", err); 4700 return err; 4701 } 4702 } 4703 4704 return OK; 4705} 4706 4707void ACodec::onSignalEndOfInputStream() { 4708 sp<AMessage> notify = mNotify->dup(); 4709 notify->setInt32("what", CodecBase::kWhatSignaledInputEOS); 4710 4711 status_t err = mOMX->signalEndOfInputStream(mNode); 4712 if (err != OK) { 4713 notify->setInt32("err", err); 4714 } 4715 notify->post(); 4716} 4717 4718bool ACodec::ExecutingState::onOMXEvent( 4719 OMX_EVENTTYPE event, OMX_U32 data1, OMX_U32 data2) { 4720 switch (event) { 4721 case OMX_EventPortSettingsChanged: 4722 { 4723 CHECK_EQ(data1, (OMX_U32)kPortIndexOutput); 4724 4725 if (data2 == 0 || data2 == OMX_IndexParamPortDefinition) { 4726 mCodec->mMetaDataBuffersToSubmit = 0; 4727 CHECK_EQ(mCodec->mOMX->sendCommand( 4728 mCodec->mNode, 4729 OMX_CommandPortDisable, kPortIndexOutput), 4730 (status_t)OK); 4731 4732 mCodec->freeOutputBuffersNotOwnedByComponent(); 4733 4734 mCodec->changeState(mCodec->mOutputPortSettingsChangedState); 4735 } else if (data2 == OMX_IndexConfigCommonOutputCrop) { 4736 mCodec->mSentFormat = false; 4737 } else { 4738 ALOGV("[%s] OMX_EventPortSettingsChanged 0x%08lx", 4739 mCodec->mComponentName.c_str(), data2); 4740 } 4741 4742 return true; 4743 } 4744 4745 case OMX_EventBufferFlag: 4746 { 4747 return true; 4748 } 4749 4750 default: 4751 return BaseState::onOMXEvent(event, data1, data2); 4752 } 4753} 4754 4755//////////////////////////////////////////////////////////////////////////////// 4756 4757ACodec::OutputPortSettingsChangedState::OutputPortSettingsChangedState( 4758 ACodec *codec) 4759 : BaseState(codec) { 4760} 4761 4762ACodec::BaseState::PortMode ACodec::OutputPortSettingsChangedState::getPortMode( 4763 OMX_U32 portIndex) { 4764 if (portIndex == kPortIndexOutput) { 4765 return FREE_BUFFERS; 4766 } 4767 4768 CHECK_EQ(portIndex, (OMX_U32)kPortIndexInput); 4769 4770 return RESUBMIT_BUFFERS; 4771} 4772 4773bool ACodec::OutputPortSettingsChangedState::onMessageReceived( 4774 const sp<AMessage> &msg) { 4775 bool handled = false; 4776 4777 switch (msg->what()) { 4778 case kWhatFlush: 4779 case kWhatShutdown: 4780 case kWhatResume: 4781 { 4782 if (msg->what() == kWhatResume) { 4783 ALOGV("[%s] Deferring resume", mCodec->mComponentName.c_str()); 4784 } 4785 4786 mCodec->deferMessage(msg); 4787 handled = true; 4788 break; 4789 } 4790 4791 default: 4792 handled = BaseState::onMessageReceived(msg); 4793 break; 4794 } 4795 4796 return handled; 4797} 4798 4799void ACodec::OutputPortSettingsChangedState::stateEntered() { 4800 ALOGV("[%s] Now handling output port settings change", 4801 mCodec->mComponentName.c_str()); 4802} 4803 4804bool ACodec::OutputPortSettingsChangedState::onOMXEvent( 4805 OMX_EVENTTYPE event, OMX_U32 data1, OMX_U32 data2) { 4806 switch (event) { 4807 case OMX_EventCmdComplete: 4808 { 4809 if (data1 == (OMX_U32)OMX_CommandPortDisable) { 4810 CHECK_EQ(data2, (OMX_U32)kPortIndexOutput); 4811 4812 ALOGV("[%s] Output port now disabled.", 4813 mCodec->mComponentName.c_str()); 4814 4815 CHECK(mCodec->mBuffers[kPortIndexOutput].isEmpty()); 4816 mCodec->mDealer[kPortIndexOutput].clear(); 4817 4818 CHECK_EQ(mCodec->mOMX->sendCommand( 4819 mCodec->mNode, OMX_CommandPortEnable, kPortIndexOutput), 4820 (status_t)OK); 4821 4822 status_t err; 4823 if ((err = mCodec->allocateBuffersOnPort( 4824 kPortIndexOutput)) != OK) { 4825 ALOGE("Failed to allocate output port buffers after " 4826 "port reconfiguration (error 0x%08x)", 4827 err); 4828 4829 mCodec->signalError(OMX_ErrorUndefined, err); 4830 4831 // This is technically not correct, but appears to be 4832 // the only way to free the component instance. 4833 // Controlled transitioning from excecuting->idle 4834 // and idle->loaded seem impossible probably because 4835 // the output port never finishes re-enabling. 4836 mCodec->mShutdownInProgress = true; 4837 mCodec->mKeepComponentAllocated = false; 4838 mCodec->changeState(mCodec->mLoadedState); 4839 } 4840 4841 return true; 4842 } else if (data1 == (OMX_U32)OMX_CommandPortEnable) { 4843 CHECK_EQ(data2, (OMX_U32)kPortIndexOutput); 4844 4845 mCodec->mSentFormat = false; 4846 4847 ALOGV("[%s] Output port now reenabled.", 4848 mCodec->mComponentName.c_str()); 4849 4850 if (mCodec->mExecutingState->active()) { 4851 mCodec->mExecutingState->submitOutputBuffers(); 4852 } 4853 4854 mCodec->changeState(mCodec->mExecutingState); 4855 4856 return true; 4857 } 4858 4859 return false; 4860 } 4861 4862 default: 4863 return false; 4864 } 4865} 4866 4867//////////////////////////////////////////////////////////////////////////////// 4868 4869ACodec::ExecutingToIdleState::ExecutingToIdleState(ACodec *codec) 4870 : BaseState(codec), 4871 mComponentNowIdle(false) { 4872} 4873 4874bool ACodec::ExecutingToIdleState::onMessageReceived(const sp<AMessage> &msg) { 4875 bool handled = false; 4876 4877 switch (msg->what()) { 4878 case kWhatFlush: 4879 { 4880 // Don't send me a flush request if you previously wanted me 4881 // to shutdown. 4882 TRESPASS(); 4883 break; 4884 } 4885 4886 case kWhatShutdown: 4887 { 4888 // We're already doing that... 4889 4890 handled = true; 4891 break; 4892 } 4893 4894 default: 4895 handled = BaseState::onMessageReceived(msg); 4896 break; 4897 } 4898 4899 return handled; 4900} 4901 4902void ACodec::ExecutingToIdleState::stateEntered() { 4903 ALOGV("[%s] Now Executing->Idle", mCodec->mComponentName.c_str()); 4904 4905 mComponentNowIdle = false; 4906 mCodec->mSentFormat = false; 4907} 4908 4909bool ACodec::ExecutingToIdleState::onOMXEvent( 4910 OMX_EVENTTYPE event, OMX_U32 data1, OMX_U32 data2) { 4911 switch (event) { 4912 case OMX_EventCmdComplete: 4913 { 4914 CHECK_EQ(data1, (OMX_U32)OMX_CommandStateSet); 4915 CHECK_EQ(data2, (OMX_U32)OMX_StateIdle); 4916 4917 mComponentNowIdle = true; 4918 4919 changeStateIfWeOwnAllBuffers(); 4920 4921 return true; 4922 } 4923 4924 case OMX_EventPortSettingsChanged: 4925 case OMX_EventBufferFlag: 4926 { 4927 // We're shutting down and don't care about this anymore. 4928 return true; 4929 } 4930 4931 default: 4932 return BaseState::onOMXEvent(event, data1, data2); 4933 } 4934} 4935 4936void ACodec::ExecutingToIdleState::changeStateIfWeOwnAllBuffers() { 4937 if (mComponentNowIdle && mCodec->allYourBuffersAreBelongToUs()) { 4938 CHECK_EQ(mCodec->mOMX->sendCommand( 4939 mCodec->mNode, OMX_CommandStateSet, OMX_StateLoaded), 4940 (status_t)OK); 4941 4942 CHECK_EQ(mCodec->freeBuffersOnPort(kPortIndexInput), (status_t)OK); 4943 CHECK_EQ(mCodec->freeBuffersOnPort(kPortIndexOutput), (status_t)OK); 4944 4945 if ((mCodec->mFlags & kFlagPushBlankBuffersToNativeWindowOnShutdown) 4946 && mCodec->mNativeWindow != NULL) { 4947 // We push enough 1x1 blank buffers to ensure that one of 4948 // them has made it to the display. This allows the OMX 4949 // component teardown to zero out any protected buffers 4950 // without the risk of scanning out one of those buffers. 4951 mCodec->pushBlankBuffersToNativeWindow(); 4952 } 4953 4954 mCodec->changeState(mCodec->mIdleToLoadedState); 4955 } 4956} 4957 4958void ACodec::ExecutingToIdleState::onInputBufferFilled( 4959 const sp<AMessage> &msg) { 4960 BaseState::onInputBufferFilled(msg); 4961 4962 changeStateIfWeOwnAllBuffers(); 4963} 4964 4965void ACodec::ExecutingToIdleState::onOutputBufferDrained( 4966 const sp<AMessage> &msg) { 4967 BaseState::onOutputBufferDrained(msg); 4968 4969 changeStateIfWeOwnAllBuffers(); 4970} 4971 4972//////////////////////////////////////////////////////////////////////////////// 4973 4974ACodec::IdleToLoadedState::IdleToLoadedState(ACodec *codec) 4975 : BaseState(codec) { 4976} 4977 4978bool ACodec::IdleToLoadedState::onMessageReceived(const sp<AMessage> &msg) { 4979 bool handled = false; 4980 4981 switch (msg->what()) { 4982 case kWhatShutdown: 4983 { 4984 // We're already doing that... 4985 4986 handled = true; 4987 break; 4988 } 4989 4990 case kWhatFlush: 4991 { 4992 // Don't send me a flush request if you previously wanted me 4993 // to shutdown. 4994 TRESPASS(); 4995 break; 4996 } 4997 4998 default: 4999 handled = BaseState::onMessageReceived(msg); 5000 break; 5001 } 5002 5003 return handled; 5004} 5005 5006void ACodec::IdleToLoadedState::stateEntered() { 5007 ALOGV("[%s] Now Idle->Loaded", mCodec->mComponentName.c_str()); 5008} 5009 5010bool ACodec::IdleToLoadedState::onOMXEvent( 5011 OMX_EVENTTYPE event, OMX_U32 data1, OMX_U32 data2) { 5012 switch (event) { 5013 case OMX_EventCmdComplete: 5014 { 5015 CHECK_EQ(data1, (OMX_U32)OMX_CommandStateSet); 5016 CHECK_EQ(data2, (OMX_U32)OMX_StateLoaded); 5017 5018 mCodec->changeState(mCodec->mLoadedState); 5019 5020 return true; 5021 } 5022 5023 default: 5024 return BaseState::onOMXEvent(event, data1, data2); 5025 } 5026} 5027 5028//////////////////////////////////////////////////////////////////////////////// 5029 5030ACodec::FlushingState::FlushingState(ACodec *codec) 5031 : BaseState(codec) { 5032} 5033 5034void ACodec::FlushingState::stateEntered() { 5035 ALOGV("[%s] Now Flushing", mCodec->mComponentName.c_str()); 5036 5037 mFlushComplete[kPortIndexInput] = mFlushComplete[kPortIndexOutput] = false; 5038} 5039 5040bool ACodec::FlushingState::onMessageReceived(const sp<AMessage> &msg) { 5041 bool handled = false; 5042 5043 switch (msg->what()) { 5044 case kWhatShutdown: 5045 { 5046 mCodec->deferMessage(msg); 5047 break; 5048 } 5049 5050 case kWhatFlush: 5051 { 5052 // We're already doing this right now. 5053 handled = true; 5054 break; 5055 } 5056 5057 default: 5058 handled = BaseState::onMessageReceived(msg); 5059 break; 5060 } 5061 5062 return handled; 5063} 5064 5065bool ACodec::FlushingState::onOMXEvent( 5066 OMX_EVENTTYPE event, OMX_U32 data1, OMX_U32 data2) { 5067 ALOGV("[%s] FlushingState onOMXEvent(%d,%ld)", 5068 mCodec->mComponentName.c_str(), event, data1); 5069 5070 switch (event) { 5071 case OMX_EventCmdComplete: 5072 { 5073 CHECK_EQ(data1, (OMX_U32)OMX_CommandFlush); 5074 5075 if (data2 == kPortIndexInput || data2 == kPortIndexOutput) { 5076 CHECK(!mFlushComplete[data2]); 5077 mFlushComplete[data2] = true; 5078 5079 if (mFlushComplete[kPortIndexInput] 5080 && mFlushComplete[kPortIndexOutput]) { 5081 changeStateIfWeOwnAllBuffers(); 5082 } 5083 } else { 5084 CHECK_EQ(data2, OMX_ALL); 5085 CHECK(mFlushComplete[kPortIndexInput]); 5086 CHECK(mFlushComplete[kPortIndexOutput]); 5087 5088 changeStateIfWeOwnAllBuffers(); 5089 } 5090 5091 return true; 5092 } 5093 5094 case OMX_EventPortSettingsChanged: 5095 { 5096 sp<AMessage> msg = new AMessage(kWhatOMXMessage, mCodec->id()); 5097 msg->setInt32("type", omx_message::EVENT); 5098 msg->setInt32("node", mCodec->mNode); 5099 msg->setInt32("event", event); 5100 msg->setInt32("data1", data1); 5101 msg->setInt32("data2", data2); 5102 5103 ALOGV("[%s] Deferring OMX_EventPortSettingsChanged", 5104 mCodec->mComponentName.c_str()); 5105 5106 mCodec->deferMessage(msg); 5107 5108 return true; 5109 } 5110 5111 default: 5112 return BaseState::onOMXEvent(event, data1, data2); 5113 } 5114 5115 return true; 5116} 5117 5118void ACodec::FlushingState::onOutputBufferDrained(const sp<AMessage> &msg) { 5119 BaseState::onOutputBufferDrained(msg); 5120 5121 changeStateIfWeOwnAllBuffers(); 5122} 5123 5124void ACodec::FlushingState::onInputBufferFilled(const sp<AMessage> &msg) { 5125 BaseState::onInputBufferFilled(msg); 5126 5127 changeStateIfWeOwnAllBuffers(); 5128} 5129 5130void ACodec::FlushingState::changeStateIfWeOwnAllBuffers() { 5131 if (mFlushComplete[kPortIndexInput] 5132 && mFlushComplete[kPortIndexOutput] 5133 && mCodec->allYourBuffersAreBelongToUs()) { 5134 // We now own all buffers except possibly those still queued with 5135 // the native window for rendering. Let's get those back as well. 5136 mCodec->waitUntilAllPossibleNativeWindowBuffersAreReturnedToUs(); 5137 5138 sp<AMessage> notify = mCodec->mNotify->dup(); 5139 notify->setInt32("what", CodecBase::kWhatFlushCompleted); 5140 notify->post(); 5141 5142 mCodec->mPortEOS[kPortIndexInput] = 5143 mCodec->mPortEOS[kPortIndexOutput] = false; 5144 5145 mCodec->mInputEOSResult = OK; 5146 5147 if (mCodec->mSkipCutBuffer != NULL) { 5148 mCodec->mSkipCutBuffer->clear(); 5149 } 5150 5151 mCodec->changeState(mCodec->mExecutingState); 5152 } 5153} 5154 5155} // namespace android 5156