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