MediaCodec.cpp revision 8140a2b7ef0650e8cce6be16efcbfce56d0ed1e4
1/* 2 * Copyright 2012, 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 "MediaCodec" 19#include <inttypes.h> 20 21#include "include/avc_utils.h" 22#include "include/SoftwareRenderer.h" 23 24#include <binder/IBatteryStats.h> 25#include <binder/IServiceManager.h> 26#include <gui/Surface.h> 27#include <media/ICrypto.h> 28#include <media/stagefright/foundation/ABuffer.h> 29#include <media/stagefright/foundation/ADebug.h> 30#include <media/stagefright/foundation/AMessage.h> 31#include <media/stagefright/foundation/AString.h> 32#include <media/stagefright/foundation/hexdump.h> 33#include <media/stagefright/ACodec.h> 34#include <media/stagefright/BufferProducerWrapper.h> 35#include <media/stagefright/MediaCodec.h> 36#include <media/stagefright/MediaCodecList.h> 37#include <media/stagefright/MediaDefs.h> 38#include <media/stagefright/MediaErrors.h> 39#include <media/stagefright/MetaData.h> 40#include <media/stagefright/NativeWindowWrapper.h> 41#include <private/android_filesystem_config.h> 42#include <utils/Log.h> 43#include <utils/Singleton.h> 44 45namespace android { 46 47struct MediaCodec::BatteryNotifier : public Singleton<BatteryNotifier> { 48 BatteryNotifier(); 49 50 void noteStartVideo(); 51 void noteStopVideo(); 52 void noteStartAudio(); 53 void noteStopAudio(); 54 55private: 56 int32_t mVideoRefCount; 57 int32_t mAudioRefCount; 58 sp<IBatteryStats> mBatteryStatService; 59}; 60 61ANDROID_SINGLETON_STATIC_INSTANCE(MediaCodec::BatteryNotifier) 62 63MediaCodec::BatteryNotifier::BatteryNotifier() : 64 mVideoRefCount(0), 65 mAudioRefCount(0) { 66 // get battery service 67 const sp<IServiceManager> sm(defaultServiceManager()); 68 if (sm != NULL) { 69 const String16 name("batterystats"); 70 mBatteryStatService = interface_cast<IBatteryStats>(sm->getService(name)); 71 if (mBatteryStatService == NULL) { 72 ALOGE("batterystats service unavailable!"); 73 } 74 } 75} 76 77void MediaCodec::BatteryNotifier::noteStartVideo() { 78 if (mVideoRefCount == 0 && mBatteryStatService != NULL) { 79 mBatteryStatService->noteStartVideo(AID_MEDIA); 80 } 81 mVideoRefCount++; 82} 83 84void MediaCodec::BatteryNotifier::noteStopVideo() { 85 if (mVideoRefCount == 0) { 86 ALOGW("BatteryNotifier::noteStop(): video refcount is broken!"); 87 return; 88 } 89 90 mVideoRefCount--; 91 if (mVideoRefCount == 0 && mBatteryStatService != NULL) { 92 mBatteryStatService->noteStopVideo(AID_MEDIA); 93 } 94} 95 96void MediaCodec::BatteryNotifier::noteStartAudio() { 97 if (mAudioRefCount == 0 && mBatteryStatService != NULL) { 98 mBatteryStatService->noteStartAudio(AID_MEDIA); 99 } 100 mAudioRefCount++; 101} 102 103void MediaCodec::BatteryNotifier::noteStopAudio() { 104 if (mAudioRefCount == 0) { 105 ALOGW("BatteryNotifier::noteStop(): audio refcount is broken!"); 106 return; 107 } 108 109 mAudioRefCount--; 110 if (mAudioRefCount == 0 && mBatteryStatService != NULL) { 111 mBatteryStatService->noteStopAudio(AID_MEDIA); 112 } 113} 114// static 115sp<MediaCodec> MediaCodec::CreateByType( 116 const sp<ALooper> &looper, const char *mime, bool encoder, status_t *err) { 117 sp<MediaCodec> codec = new MediaCodec(looper); 118 119 const status_t ret = codec->init(mime, true /* nameIsType */, encoder); 120 if (err != NULL) { 121 *err = ret; 122 } 123 return ret == OK ? codec : NULL; // NULL deallocates codec. 124} 125 126// static 127sp<MediaCodec> MediaCodec::CreateByComponentName( 128 const sp<ALooper> &looper, const char *name, status_t *err) { 129 sp<MediaCodec> codec = new MediaCodec(looper); 130 131 const status_t ret = codec->init(name, false /* nameIsType */, false /* encoder */); 132 if (err != NULL) { 133 *err = ret; 134 } 135 return ret == OK ? codec : NULL; // NULL deallocates codec. 136} 137 138MediaCodec::MediaCodec(const sp<ALooper> &looper) 139 : mState(UNINITIALIZED), 140 mLooper(looper), 141 mCodec(NULL), 142 mReplyID(0), 143 mFlags(0), 144 mStickyError(OK), 145 mSoftRenderer(NULL), 146 mBatteryStatNotified(false), 147 mIsVideo(false), 148 mDequeueInputTimeoutGeneration(0), 149 mDequeueInputReplyID(0), 150 mDequeueOutputTimeoutGeneration(0), 151 mDequeueOutputReplyID(0), 152 mHaveInputSurface(false) { 153} 154 155MediaCodec::~MediaCodec() { 156 CHECK_EQ(mState, UNINITIALIZED); 157} 158 159// static 160status_t MediaCodec::PostAndAwaitResponse( 161 const sp<AMessage> &msg, sp<AMessage> *response) { 162 status_t err = msg->postAndAwaitResponse(response); 163 164 if (err != OK) { 165 return err; 166 } 167 168 if (!(*response)->findInt32("err", &err)) { 169 err = OK; 170 } 171 172 return err; 173} 174 175// static 176void MediaCodec::PostReplyWithError(int32_t replyID, int32_t err) { 177 sp<AMessage> response = new AMessage; 178 response->setInt32("err", err); 179 response->postReply(replyID); 180} 181 182status_t MediaCodec::init(const AString &name, bool nameIsType, bool encoder) { 183 // save init parameters for reset 184 mInitName = name; 185 mInitNameIsType = nameIsType; 186 mInitIsEncoder = encoder; 187 188 // Current video decoders do not return from OMX_FillThisBuffer 189 // quickly, violating the OpenMAX specs, until that is remedied 190 // we need to invest in an extra looper to free the main event 191 // queue. 192 mCodec = new ACodec; 193 bool needDedicatedLooper = false; 194 if (nameIsType && !strncasecmp(name.c_str(), "video/", 6)) { 195 needDedicatedLooper = true; 196 } else { 197 AString tmp = name; 198 if (tmp.endsWith(".secure")) { 199 tmp.erase(tmp.size() - 7, 7); 200 } 201 const sp<IMediaCodecList> mcl = MediaCodecList::getInstance(); 202 ssize_t codecIdx = mcl->findCodecByName(tmp.c_str()); 203 if (codecIdx >= 0) { 204 const sp<MediaCodecInfo> info = mcl->getCodecInfo(codecIdx); 205 Vector<AString> mimes; 206 info->getSupportedMimes(&mimes); 207 for (size_t i = 0; i < mimes.size(); i++) { 208 if (mimes[i].startsWith("video/")) { 209 needDedicatedLooper = true; 210 break; 211 } 212 } 213 } 214 } 215 216 if (needDedicatedLooper) { 217 if (mCodecLooper == NULL) { 218 mCodecLooper = new ALooper; 219 mCodecLooper->setName("CodecLooper"); 220 mCodecLooper->start(false, false, ANDROID_PRIORITY_AUDIO); 221 } 222 223 mCodecLooper->registerHandler(mCodec); 224 } else { 225 mLooper->registerHandler(mCodec); 226 } 227 228 mLooper->registerHandler(this); 229 230 mCodec->setNotificationMessage(new AMessage(kWhatCodecNotify, id())); 231 232 sp<AMessage> msg = new AMessage(kWhatInit, id()); 233 msg->setString("name", name); 234 msg->setInt32("nameIsType", nameIsType); 235 236 if (nameIsType) { 237 msg->setInt32("encoder", encoder); 238 } 239 240 sp<AMessage> response; 241 return PostAndAwaitResponse(msg, &response); 242} 243 244status_t MediaCodec::setCallback(const sp<AMessage> &callback) { 245 sp<AMessage> msg = new AMessage(kWhatSetCallback, id()); 246 msg->setMessage("callback", callback); 247 248 sp<AMessage> response; 249 return PostAndAwaitResponse(msg, &response); 250} 251 252status_t MediaCodec::configure( 253 const sp<AMessage> &format, 254 const sp<Surface> &nativeWindow, 255 const sp<ICrypto> &crypto, 256 uint32_t flags) { 257 sp<AMessage> msg = new AMessage(kWhatConfigure, id()); 258 259 msg->setMessage("format", format); 260 msg->setInt32("flags", flags); 261 262 if (nativeWindow != NULL) { 263 msg->setObject( 264 "native-window", 265 new NativeWindowWrapper(nativeWindow)); 266 } 267 268 if (crypto != NULL) { 269 msg->setPointer("crypto", crypto.get()); 270 } 271 272 sp<AMessage> response; 273 status_t err = PostAndAwaitResponse(msg, &response); 274 275 if (err != OK && err != INVALID_OPERATION) { 276 // MediaCodec now set state to UNINITIALIZED upon any fatal error. 277 // To maintain backward-compatibility, do a reset() to put codec 278 // back into INITIALIZED state. 279 // But don't reset if the err is INVALID_OPERATION, which means 280 // the configure failure is due to wrong state. 281 282 ALOGE("configure failed with err 0x%08x, resetting...", err); 283 reset(); 284 } 285 286 return err; 287} 288 289status_t MediaCodec::createInputSurface( 290 sp<IGraphicBufferProducer>* bufferProducer) { 291 sp<AMessage> msg = new AMessage(kWhatCreateInputSurface, id()); 292 293 sp<AMessage> response; 294 status_t err = PostAndAwaitResponse(msg, &response); 295 if (err == NO_ERROR) { 296 // unwrap the sp<IGraphicBufferProducer> 297 sp<RefBase> obj; 298 bool found = response->findObject("input-surface", &obj); 299 CHECK(found); 300 sp<BufferProducerWrapper> wrapper( 301 static_cast<BufferProducerWrapper*>(obj.get())); 302 *bufferProducer = wrapper->getBufferProducer(); 303 } else { 304 ALOGW("createInputSurface failed, err=%d", err); 305 } 306 return err; 307} 308 309status_t MediaCodec::start() { 310 sp<AMessage> msg = new AMessage(kWhatStart, id()); 311 312 sp<AMessage> response; 313 return PostAndAwaitResponse(msg, &response); 314} 315 316status_t MediaCodec::stop() { 317 sp<AMessage> msg = new AMessage(kWhatStop, id()); 318 319 sp<AMessage> response; 320 return PostAndAwaitResponse(msg, &response); 321} 322 323status_t MediaCodec::release() { 324 sp<AMessage> msg = new AMessage(kWhatRelease, id()); 325 326 sp<AMessage> response; 327 return PostAndAwaitResponse(msg, &response); 328} 329 330status_t MediaCodec::reset() { 331 /* When external-facing MediaCodec object is created, 332 it is already initialized. Thus, reset is essentially 333 release() followed by init(), plus clearing the state */ 334 335 status_t err = release(); 336 337 // unregister handlers 338 if (mCodec != NULL) { 339 if (mCodecLooper != NULL) { 340 mCodecLooper->unregisterHandler(mCodec->id()); 341 } else { 342 mLooper->unregisterHandler(mCodec->id()); 343 } 344 mCodec = NULL; 345 } 346 mLooper->unregisterHandler(id()); 347 348 mFlags = 0; // clear all flags 349 mStickyError = OK; 350 351 // reset state not reset by setState(UNINITIALIZED) 352 mReplyID = 0; 353 mDequeueInputReplyID = 0; 354 mDequeueOutputReplyID = 0; 355 mDequeueInputTimeoutGeneration = 0; 356 mDequeueOutputTimeoutGeneration = 0; 357 mHaveInputSurface = false; 358 359 if (err == OK) { 360 err = init(mInitName, mInitNameIsType, mInitIsEncoder); 361 } 362 return err; 363} 364 365status_t MediaCodec::queueInputBuffer( 366 size_t index, 367 size_t offset, 368 size_t size, 369 int64_t presentationTimeUs, 370 uint32_t flags, 371 AString *errorDetailMsg) { 372 if (errorDetailMsg != NULL) { 373 errorDetailMsg->clear(); 374 } 375 376 sp<AMessage> msg = new AMessage(kWhatQueueInputBuffer, id()); 377 msg->setSize("index", index); 378 msg->setSize("offset", offset); 379 msg->setSize("size", size); 380 msg->setInt64("timeUs", presentationTimeUs); 381 msg->setInt32("flags", flags); 382 msg->setPointer("errorDetailMsg", errorDetailMsg); 383 384 sp<AMessage> response; 385 return PostAndAwaitResponse(msg, &response); 386} 387 388status_t MediaCodec::queueSecureInputBuffer( 389 size_t index, 390 size_t offset, 391 const CryptoPlugin::SubSample *subSamples, 392 size_t numSubSamples, 393 const uint8_t key[16], 394 const uint8_t iv[16], 395 CryptoPlugin::Mode mode, 396 int64_t presentationTimeUs, 397 uint32_t flags, 398 AString *errorDetailMsg) { 399 if (errorDetailMsg != NULL) { 400 errorDetailMsg->clear(); 401 } 402 403 sp<AMessage> msg = new AMessage(kWhatQueueInputBuffer, id()); 404 msg->setSize("index", index); 405 msg->setSize("offset", offset); 406 msg->setPointer("subSamples", (void *)subSamples); 407 msg->setSize("numSubSamples", numSubSamples); 408 msg->setPointer("key", (void *)key); 409 msg->setPointer("iv", (void *)iv); 410 msg->setInt32("mode", mode); 411 msg->setInt64("timeUs", presentationTimeUs); 412 msg->setInt32("flags", flags); 413 msg->setPointer("errorDetailMsg", errorDetailMsg); 414 415 sp<AMessage> response; 416 status_t err = PostAndAwaitResponse(msg, &response); 417 418 return err; 419} 420 421status_t MediaCodec::dequeueInputBuffer(size_t *index, int64_t timeoutUs) { 422 sp<AMessage> msg = new AMessage(kWhatDequeueInputBuffer, id()); 423 msg->setInt64("timeoutUs", timeoutUs); 424 425 sp<AMessage> response; 426 status_t err; 427 if ((err = PostAndAwaitResponse(msg, &response)) != OK) { 428 return err; 429 } 430 431 CHECK(response->findSize("index", index)); 432 433 return OK; 434} 435 436status_t MediaCodec::dequeueOutputBuffer( 437 size_t *index, 438 size_t *offset, 439 size_t *size, 440 int64_t *presentationTimeUs, 441 uint32_t *flags, 442 int64_t timeoutUs) { 443 sp<AMessage> msg = new AMessage(kWhatDequeueOutputBuffer, id()); 444 msg->setInt64("timeoutUs", timeoutUs); 445 446 sp<AMessage> response; 447 status_t err; 448 if ((err = PostAndAwaitResponse(msg, &response)) != OK) { 449 return err; 450 } 451 452 CHECK(response->findSize("index", index)); 453 CHECK(response->findSize("offset", offset)); 454 CHECK(response->findSize("size", size)); 455 CHECK(response->findInt64("timeUs", presentationTimeUs)); 456 CHECK(response->findInt32("flags", (int32_t *)flags)); 457 458 return OK; 459} 460 461status_t MediaCodec::renderOutputBufferAndRelease(size_t index) { 462 sp<AMessage> msg = new AMessage(kWhatReleaseOutputBuffer, id()); 463 msg->setSize("index", index); 464 msg->setInt32("render", true); 465 466 sp<AMessage> response; 467 return PostAndAwaitResponse(msg, &response); 468} 469 470status_t MediaCodec::renderOutputBufferAndRelease(size_t index, int64_t timestampNs) { 471 sp<AMessage> msg = new AMessage(kWhatReleaseOutputBuffer, id()); 472 msg->setSize("index", index); 473 msg->setInt32("render", true); 474 msg->setInt64("timestampNs", timestampNs); 475 476 sp<AMessage> response; 477 return PostAndAwaitResponse(msg, &response); 478} 479 480status_t MediaCodec::releaseOutputBuffer(size_t index) { 481 sp<AMessage> msg = new AMessage(kWhatReleaseOutputBuffer, id()); 482 msg->setSize("index", index); 483 484 sp<AMessage> response; 485 return PostAndAwaitResponse(msg, &response); 486} 487 488status_t MediaCodec::signalEndOfInputStream() { 489 sp<AMessage> msg = new AMessage(kWhatSignalEndOfInputStream, id()); 490 491 sp<AMessage> response; 492 return PostAndAwaitResponse(msg, &response); 493} 494 495status_t MediaCodec::getOutputFormat(sp<AMessage> *format) const { 496 sp<AMessage> msg = new AMessage(kWhatGetOutputFormat, id()); 497 498 sp<AMessage> response; 499 status_t err; 500 if ((err = PostAndAwaitResponse(msg, &response)) != OK) { 501 return err; 502 } 503 504 CHECK(response->findMessage("format", format)); 505 506 return OK; 507} 508 509status_t MediaCodec::getInputFormat(sp<AMessage> *format) const { 510 sp<AMessage> msg = new AMessage(kWhatGetInputFormat, id()); 511 512 sp<AMessage> response; 513 status_t err; 514 if ((err = PostAndAwaitResponse(msg, &response)) != OK) { 515 return err; 516 } 517 518 CHECK(response->findMessage("format", format)); 519 520 return OK; 521} 522 523status_t MediaCodec::getName(AString *name) const { 524 sp<AMessage> msg = new AMessage(kWhatGetName, id()); 525 526 sp<AMessage> response; 527 status_t err; 528 if ((err = PostAndAwaitResponse(msg, &response)) != OK) { 529 return err; 530 } 531 532 CHECK(response->findString("name", name)); 533 534 return OK; 535} 536 537status_t MediaCodec::getInputBuffers(Vector<sp<ABuffer> > *buffers) const { 538 sp<AMessage> msg = new AMessage(kWhatGetBuffers, id()); 539 msg->setInt32("portIndex", kPortIndexInput); 540 msg->setPointer("buffers", buffers); 541 542 sp<AMessage> response; 543 return PostAndAwaitResponse(msg, &response); 544} 545 546status_t MediaCodec::getOutputBuffers(Vector<sp<ABuffer> > *buffers) const { 547 sp<AMessage> msg = new AMessage(kWhatGetBuffers, id()); 548 msg->setInt32("portIndex", kPortIndexOutput); 549 msg->setPointer("buffers", buffers); 550 551 sp<AMessage> response; 552 return PostAndAwaitResponse(msg, &response); 553} 554 555status_t MediaCodec::getOutputBuffer(size_t index, sp<ABuffer> *buffer) { 556 sp<AMessage> format; 557 return getBufferAndFormat(kPortIndexOutput, index, buffer, &format); 558} 559 560status_t MediaCodec::getOutputFormat(size_t index, sp<AMessage> *format) { 561 sp<ABuffer> buffer; 562 return getBufferAndFormat(kPortIndexOutput, index, &buffer, format); 563} 564 565status_t MediaCodec::getInputBuffer(size_t index, sp<ABuffer> *buffer) { 566 sp<AMessage> format; 567 return getBufferAndFormat(kPortIndexInput, index, buffer, &format); 568} 569 570bool MediaCodec::isExecuting() const { 571 return mState == STARTED || mState == FLUSHED; 572} 573 574status_t MediaCodec::getBufferAndFormat( 575 size_t portIndex, size_t index, 576 sp<ABuffer> *buffer, sp<AMessage> *format) { 577 // use mutex instead of a context switch 578 579 buffer->clear(); 580 format->clear(); 581 if (!isExecuting()) { 582 return INVALID_OPERATION; 583 } 584 585 // we do not want mPortBuffers to change during this section 586 // we also don't want mOwnedByClient to change during this 587 Mutex::Autolock al(mBufferLock); 588 Vector<BufferInfo> *buffers = &mPortBuffers[portIndex]; 589 if (index < buffers->size()) { 590 const BufferInfo &info = buffers->itemAt(index); 591 if (info.mOwnedByClient) { 592 // by the time buffers array is initialized, crypto is set 593 if (portIndex == kPortIndexInput && mCrypto != NULL) { 594 *buffer = info.mEncryptedData; 595 } else { 596 *buffer = info.mData; 597 } 598 *format = info.mFormat; 599 } 600 } 601 return OK; 602} 603 604status_t MediaCodec::flush() { 605 sp<AMessage> msg = new AMessage(kWhatFlush, id()); 606 607 sp<AMessage> response; 608 return PostAndAwaitResponse(msg, &response); 609} 610 611status_t MediaCodec::requestIDRFrame() { 612 (new AMessage(kWhatRequestIDRFrame, id()))->post(); 613 614 return OK; 615} 616 617void MediaCodec::requestActivityNotification(const sp<AMessage> ¬ify) { 618 sp<AMessage> msg = new AMessage(kWhatRequestActivityNotification, id()); 619 msg->setMessage("notify", notify); 620 msg->post(); 621} 622 623//////////////////////////////////////////////////////////////////////////////// 624 625void MediaCodec::cancelPendingDequeueOperations() { 626 if (mFlags & kFlagDequeueInputPending) { 627 PostReplyWithError(mDequeueInputReplyID, INVALID_OPERATION); 628 629 ++mDequeueInputTimeoutGeneration; 630 mDequeueInputReplyID = 0; 631 mFlags &= ~kFlagDequeueInputPending; 632 } 633 634 if (mFlags & kFlagDequeueOutputPending) { 635 PostReplyWithError(mDequeueOutputReplyID, INVALID_OPERATION); 636 637 ++mDequeueOutputTimeoutGeneration; 638 mDequeueOutputReplyID = 0; 639 mFlags &= ~kFlagDequeueOutputPending; 640 } 641} 642 643bool MediaCodec::handleDequeueInputBuffer(uint32_t replyID, bool newRequest) { 644 if (!isExecuting() || (mFlags & kFlagIsAsync) 645 || (newRequest && (mFlags & kFlagDequeueInputPending))) { 646 PostReplyWithError(replyID, INVALID_OPERATION); 647 return true; 648 } else if (mFlags & kFlagStickyError) { 649 PostReplyWithError(replyID, getStickyError()); 650 return true; 651 } 652 653 ssize_t index = dequeuePortBuffer(kPortIndexInput); 654 655 if (index < 0) { 656 CHECK_EQ(index, -EAGAIN); 657 return false; 658 } 659 660 sp<AMessage> response = new AMessage; 661 response->setSize("index", index); 662 response->postReply(replyID); 663 664 return true; 665} 666 667bool MediaCodec::handleDequeueOutputBuffer(uint32_t replyID, bool newRequest) { 668 sp<AMessage> response = new AMessage; 669 670 if (!isExecuting() || (mFlags & kFlagIsAsync) 671 || (newRequest && (mFlags & kFlagDequeueOutputPending))) { 672 response->setInt32("err", INVALID_OPERATION); 673 } else if (mFlags & kFlagStickyError) { 674 response->setInt32("err", getStickyError()); 675 } else if (mFlags & kFlagOutputBuffersChanged) { 676 response->setInt32("err", INFO_OUTPUT_BUFFERS_CHANGED); 677 mFlags &= ~kFlagOutputBuffersChanged; 678 } else if (mFlags & kFlagOutputFormatChanged) { 679 response->setInt32("err", INFO_FORMAT_CHANGED); 680 mFlags &= ~kFlagOutputFormatChanged; 681 } else { 682 ssize_t index = dequeuePortBuffer(kPortIndexOutput); 683 684 if (index < 0) { 685 CHECK_EQ(index, -EAGAIN); 686 return false; 687 } 688 689 const sp<ABuffer> &buffer = 690 mPortBuffers[kPortIndexOutput].itemAt(index).mData; 691 692 response->setSize("index", index); 693 response->setSize("offset", buffer->offset()); 694 response->setSize("size", buffer->size()); 695 696 int64_t timeUs; 697 CHECK(buffer->meta()->findInt64("timeUs", &timeUs)); 698 699 response->setInt64("timeUs", timeUs); 700 701 int32_t omxFlags; 702 CHECK(buffer->meta()->findInt32("omxFlags", &omxFlags)); 703 704 uint32_t flags = 0; 705 if (omxFlags & OMX_BUFFERFLAG_SYNCFRAME) { 706 flags |= BUFFER_FLAG_SYNCFRAME; 707 } 708 if (omxFlags & OMX_BUFFERFLAG_CODECCONFIG) { 709 flags |= BUFFER_FLAG_CODECCONFIG; 710 } 711 if (omxFlags & OMX_BUFFERFLAG_EOS) { 712 flags |= BUFFER_FLAG_EOS; 713 } 714 715 response->setInt32("flags", flags); 716 } 717 718 response->postReply(replyID); 719 720 return true; 721} 722 723void MediaCodec::onMessageReceived(const sp<AMessage> &msg) { 724 switch (msg->what()) { 725 case kWhatCodecNotify: 726 { 727 int32_t what; 728 CHECK(msg->findInt32("what", &what)); 729 730 switch (what) { 731 case CodecBase::kWhatError: 732 { 733 int32_t err, actionCode; 734 CHECK(msg->findInt32("err", &err)); 735 CHECK(msg->findInt32("actionCode", &actionCode)); 736 737 ALOGE("Codec reported err %#x, actionCode %d, while in state %d", 738 err, actionCode, mState); 739 if (err == DEAD_OBJECT) { 740 mFlags |= kFlagSawMediaServerDie; 741 mFlags &= ~kFlagIsComponentAllocated; 742 } 743 744 bool sendErrorResponse = true; 745 746 switch (mState) { 747 case INITIALIZING: 748 { 749 setState(UNINITIALIZED); 750 break; 751 } 752 753 case CONFIGURING: 754 { 755 setState(actionCode == ACTION_CODE_FATAL ? 756 UNINITIALIZED : INITIALIZED); 757 break; 758 } 759 760 case STARTING: 761 { 762 setState(actionCode == ACTION_CODE_FATAL ? 763 UNINITIALIZED : CONFIGURED); 764 break; 765 } 766 767 case STOPPING: 768 case RELEASING: 769 { 770 // Ignore the error, assuming we'll still get 771 // the shutdown complete notification. 772 773 sendErrorResponse = false; 774 775 if (mFlags & kFlagSawMediaServerDie) { 776 // MediaServer died, there definitely won't 777 // be a shutdown complete notification after 778 // all. 779 780 // note that we're directly going from 781 // STOPPING->UNINITIALIZED, instead of the 782 // usual STOPPING->INITIALIZED state. 783 setState(UNINITIALIZED); 784 if (mState == RELEASING) { 785 mComponentName.clear(); 786 } 787 (new AMessage)->postReply(mReplyID); 788 } 789 break; 790 } 791 792 case FLUSHING: 793 { 794 if (actionCode == ACTION_CODE_FATAL) { 795 setState(UNINITIALIZED); 796 } else { 797 setState( 798 (mFlags & kFlagIsAsync) ? FLUSHED : STARTED); 799 } 800 break; 801 } 802 803 case FLUSHED: 804 case STARTED: 805 { 806 sendErrorResponse = false; 807 808 setStickyError(err); 809 postActivityNotificationIfPossible(); 810 811 cancelPendingDequeueOperations(); 812 813 if (mFlags & kFlagIsAsync) { 814 onError(err, actionCode); 815 } 816 switch (actionCode) { 817 case ACTION_CODE_TRANSIENT: 818 break; 819 case ACTION_CODE_RECOVERABLE: 820 setState(INITIALIZED); 821 break; 822 default: 823 setState(UNINITIALIZED); 824 break; 825 } 826 break; 827 } 828 829 default: 830 { 831 sendErrorResponse = false; 832 833 setStickyError(err); 834 postActivityNotificationIfPossible(); 835 836 // actionCode in an uninitialized state is always fatal. 837 if (mState == UNINITIALIZED) { 838 actionCode = ACTION_CODE_FATAL; 839 } 840 if (mFlags & kFlagIsAsync) { 841 onError(err, actionCode); 842 } 843 switch (actionCode) { 844 case ACTION_CODE_TRANSIENT: 845 break; 846 case ACTION_CODE_RECOVERABLE: 847 setState(INITIALIZED); 848 break; 849 default: 850 setState(UNINITIALIZED); 851 break; 852 } 853 break; 854 } 855 } 856 857 if (sendErrorResponse) { 858 PostReplyWithError(mReplyID, err); 859 } 860 break; 861 } 862 863 case CodecBase::kWhatComponentAllocated: 864 { 865 CHECK_EQ(mState, INITIALIZING); 866 setState(INITIALIZED); 867 mFlags |= kFlagIsComponentAllocated; 868 869 CHECK(msg->findString("componentName", &mComponentName)); 870 871 if (mComponentName.startsWith("OMX.google.")) { 872 mFlags |= kFlagIsSoftwareCodec; 873 } else { 874 mFlags &= ~kFlagIsSoftwareCodec; 875 } 876 877 if (mComponentName.endsWith(".secure")) { 878 mFlags |= kFlagIsSecure; 879 } else { 880 mFlags &= ~kFlagIsSecure; 881 } 882 883 (new AMessage)->postReply(mReplyID); 884 break; 885 } 886 887 case CodecBase::kWhatComponentConfigured: 888 { 889 CHECK_EQ(mState, CONFIGURING); 890 891 // reset input surface flag 892 mHaveInputSurface = false; 893 894 CHECK(msg->findMessage("input-format", &mInputFormat)); 895 CHECK(msg->findMessage("output-format", &mOutputFormat)); 896 897 setState(CONFIGURED); 898 (new AMessage)->postReply(mReplyID); 899 break; 900 } 901 902 case CodecBase::kWhatInputSurfaceCreated: 903 { 904 // response to initiateCreateInputSurface() 905 status_t err = NO_ERROR; 906 sp<AMessage> response = new AMessage(); 907 if (!msg->findInt32("err", &err)) { 908 sp<RefBase> obj; 909 msg->findObject("input-surface", &obj); 910 CHECK(obj != NULL); 911 response->setObject("input-surface", obj); 912 mHaveInputSurface = true; 913 } else { 914 response->setInt32("err", err); 915 } 916 response->postReply(mReplyID); 917 break; 918 } 919 920 case CodecBase::kWhatSignaledInputEOS: 921 { 922 // response to signalEndOfInputStream() 923 sp<AMessage> response = new AMessage(); 924 status_t err; 925 if (msg->findInt32("err", &err)) { 926 response->setInt32("err", err); 927 } 928 response->postReply(mReplyID); 929 break; 930 } 931 932 933 case CodecBase::kWhatBuffersAllocated: 934 { 935 Mutex::Autolock al(mBufferLock); 936 int32_t portIndex; 937 CHECK(msg->findInt32("portIndex", &portIndex)); 938 939 ALOGV("%s buffers allocated", 940 portIndex == kPortIndexInput ? "input" : "output"); 941 942 CHECK(portIndex == kPortIndexInput 943 || portIndex == kPortIndexOutput); 944 945 mPortBuffers[portIndex].clear(); 946 947 Vector<BufferInfo> *buffers = &mPortBuffers[portIndex]; 948 949 sp<RefBase> obj; 950 CHECK(msg->findObject("portDesc", &obj)); 951 952 sp<CodecBase::PortDescription> portDesc = 953 static_cast<CodecBase::PortDescription *>(obj.get()); 954 955 size_t numBuffers = portDesc->countBuffers(); 956 957 for (size_t i = 0; i < numBuffers; ++i) { 958 BufferInfo info; 959 info.mBufferID = portDesc->bufferIDAt(i); 960 info.mOwnedByClient = false; 961 info.mData = portDesc->bufferAt(i); 962 963 if (portIndex == kPortIndexInput && mCrypto != NULL) { 964 info.mEncryptedData = 965 new ABuffer(info.mData->capacity()); 966 } 967 968 buffers->push_back(info); 969 } 970 971 if (portIndex == kPortIndexOutput) { 972 if (mState == STARTING) { 973 // We're always allocating output buffers after 974 // allocating input buffers, so this is a good 975 // indication that now all buffers are allocated. 976 setState(STARTED); 977 (new AMessage)->postReply(mReplyID); 978 } else { 979 mFlags |= kFlagOutputBuffersChanged; 980 postActivityNotificationIfPossible(); 981 } 982 } 983 break; 984 } 985 986 case CodecBase::kWhatOutputFormatChanged: 987 { 988 ALOGV("codec output format changed"); 989 990 if (mSoftRenderer == NULL && 991 mNativeWindow != NULL && 992 (mFlags & kFlagIsSoftwareCodec)) { 993 AString mime; 994 CHECK(msg->findString("mime", &mime)); 995 996 if (mime.startsWithIgnoreCase("video/")) { 997 mSoftRenderer = new SoftwareRenderer(mNativeWindow); 998 } 999 } 1000 1001 mOutputFormat = msg; 1002 1003 if (mFlags & kFlagIsEncoder) { 1004 // Before we announce the format change we should 1005 // collect codec specific data and amend the output 1006 // format as necessary. 1007 mFlags |= kFlagGatherCodecSpecificData; 1008 } else if (mFlags & kFlagIsAsync) { 1009 onOutputFormatChanged(); 1010 } else { 1011 mFlags |= kFlagOutputFormatChanged; 1012 postActivityNotificationIfPossible(); 1013 } 1014 1015 // Notify mCrypto of video resolution changes 1016 if (mCrypto != NULL) { 1017 int32_t left, top, right, bottom, width, height; 1018 if (mOutputFormat->findRect("crop", &left, &top, &right, &bottom)) { 1019 mCrypto->notifyResolution(right - left + 1, bottom - top + 1); 1020 } else if (mOutputFormat->findInt32("width", &width) 1021 && mOutputFormat->findInt32("height", &height)) { 1022 mCrypto->notifyResolution(width, height); 1023 } 1024 } 1025 1026 break; 1027 } 1028 1029 case CodecBase::kWhatFillThisBuffer: 1030 { 1031 /* size_t index = */updateBuffers(kPortIndexInput, msg); 1032 1033 if (mState == FLUSHING 1034 || mState == STOPPING 1035 || mState == RELEASING) { 1036 returnBuffersToCodecOnPort(kPortIndexInput); 1037 break; 1038 } 1039 1040 if (!mCSD.empty()) { 1041 ssize_t index = dequeuePortBuffer(kPortIndexInput); 1042 CHECK_GE(index, 0); 1043 1044 // If codec specific data had been specified as 1045 // part of the format in the call to configure and 1046 // if there's more csd left, we submit it here 1047 // clients only get access to input buffers once 1048 // this data has been exhausted. 1049 1050 status_t err = queueCSDInputBuffer(index); 1051 1052 if (err != OK) { 1053 ALOGE("queueCSDInputBuffer failed w/ error %d", 1054 err); 1055 1056 setStickyError(err); 1057 postActivityNotificationIfPossible(); 1058 1059 cancelPendingDequeueOperations(); 1060 } 1061 break; 1062 } 1063 1064 if (mFlags & kFlagIsAsync) { 1065 if (!mHaveInputSurface) { 1066 onInputBufferAvailable(); 1067 } 1068 } else if (mFlags & kFlagDequeueInputPending) { 1069 CHECK(handleDequeueInputBuffer(mDequeueInputReplyID)); 1070 1071 ++mDequeueInputTimeoutGeneration; 1072 mFlags &= ~kFlagDequeueInputPending; 1073 mDequeueInputReplyID = 0; 1074 } else { 1075 postActivityNotificationIfPossible(); 1076 } 1077 break; 1078 } 1079 1080 case CodecBase::kWhatDrainThisBuffer: 1081 { 1082 /* size_t index = */updateBuffers(kPortIndexOutput, msg); 1083 1084 if (mState == FLUSHING 1085 || mState == STOPPING 1086 || mState == RELEASING) { 1087 returnBuffersToCodecOnPort(kPortIndexOutput); 1088 break; 1089 } 1090 1091 sp<ABuffer> buffer; 1092 CHECK(msg->findBuffer("buffer", &buffer)); 1093 1094 int32_t omxFlags; 1095 CHECK(msg->findInt32("flags", &omxFlags)); 1096 1097 buffer->meta()->setInt32("omxFlags", omxFlags); 1098 1099 if (mFlags & kFlagGatherCodecSpecificData) { 1100 // This is the very first output buffer after a 1101 // format change was signalled, it'll either contain 1102 // the one piece of codec specific data we can expect 1103 // or there won't be codec specific data. 1104 if (omxFlags & OMX_BUFFERFLAG_CODECCONFIG) { 1105 status_t err = 1106 amendOutputFormatWithCodecSpecificData(buffer); 1107 1108 if (err != OK) { 1109 ALOGE("Codec spit out malformed codec " 1110 "specific data!"); 1111 } 1112 } 1113 1114 mFlags &= ~kFlagGatherCodecSpecificData; 1115 if (mFlags & kFlagIsAsync) { 1116 onOutputFormatChanged(); 1117 } else { 1118 mFlags |= kFlagOutputFormatChanged; 1119 } 1120 } 1121 1122 if (mFlags & kFlagIsAsync) { 1123 onOutputBufferAvailable(); 1124 } else if (mFlags & kFlagDequeueOutputPending) { 1125 CHECK(handleDequeueOutputBuffer(mDequeueOutputReplyID)); 1126 1127 ++mDequeueOutputTimeoutGeneration; 1128 mFlags &= ~kFlagDequeueOutputPending; 1129 mDequeueOutputReplyID = 0; 1130 } else { 1131 postActivityNotificationIfPossible(); 1132 } 1133 1134 break; 1135 } 1136 1137 case CodecBase::kWhatEOS: 1138 { 1139 // We already notify the client of this by using the 1140 // corresponding flag in "onOutputBufferReady". 1141 break; 1142 } 1143 1144 case CodecBase::kWhatShutdownCompleted: 1145 { 1146 if (mState == STOPPING) { 1147 setState(INITIALIZED); 1148 } else { 1149 CHECK_EQ(mState, RELEASING); 1150 setState(UNINITIALIZED); 1151 mComponentName.clear(); 1152 } 1153 mFlags &= ~kFlagIsComponentAllocated; 1154 1155 (new AMessage)->postReply(mReplyID); 1156 break; 1157 } 1158 1159 case CodecBase::kWhatFlushCompleted: 1160 { 1161 if (mState != FLUSHING) { 1162 ALOGW("received FlushCompleted message in state %d", 1163 mState); 1164 break; 1165 } 1166 1167 if (mFlags & kFlagIsAsync) { 1168 setState(FLUSHED); 1169 } else { 1170 setState(STARTED); 1171 mCodec->signalResume(); 1172 } 1173 1174 (new AMessage)->postReply(mReplyID); 1175 break; 1176 } 1177 1178 default: 1179 TRESPASS(); 1180 } 1181 break; 1182 } 1183 1184 case kWhatInit: 1185 { 1186 uint32_t replyID; 1187 CHECK(msg->senderAwaitsResponse(&replyID)); 1188 1189 if (mState != UNINITIALIZED) { 1190 PostReplyWithError(replyID, INVALID_OPERATION); 1191 break; 1192 } 1193 1194 mReplyID = replyID; 1195 setState(INITIALIZING); 1196 1197 AString name; 1198 CHECK(msg->findString("name", &name)); 1199 1200 int32_t nameIsType; 1201 int32_t encoder = false; 1202 CHECK(msg->findInt32("nameIsType", &nameIsType)); 1203 if (nameIsType) { 1204 CHECK(msg->findInt32("encoder", &encoder)); 1205 } 1206 1207 sp<AMessage> format = new AMessage; 1208 1209 if (nameIsType) { 1210 format->setString("mime", name.c_str()); 1211 format->setInt32("encoder", encoder); 1212 } else { 1213 format->setString("componentName", name.c_str()); 1214 } 1215 1216 mCodec->initiateAllocateComponent(format); 1217 break; 1218 } 1219 1220 case kWhatSetCallback: 1221 { 1222 uint32_t replyID; 1223 CHECK(msg->senderAwaitsResponse(&replyID)); 1224 1225 if (mState == UNINITIALIZED 1226 || mState == INITIALIZING 1227 || isExecuting()) { 1228 // callback can't be set after codec is executing, 1229 // or before it's initialized (as the callback 1230 // will be cleared when it goes to INITIALIZED) 1231 PostReplyWithError(replyID, INVALID_OPERATION); 1232 break; 1233 } 1234 1235 sp<AMessage> callback; 1236 CHECK(msg->findMessage("callback", &callback)); 1237 1238 mCallback = callback; 1239 1240 if (mCallback != NULL) { 1241 ALOGI("MediaCodec will operate in async mode"); 1242 mFlags |= kFlagIsAsync; 1243 } else { 1244 mFlags &= ~kFlagIsAsync; 1245 } 1246 1247 sp<AMessage> response = new AMessage; 1248 response->postReply(replyID); 1249 break; 1250 } 1251 1252 case kWhatConfigure: 1253 { 1254 uint32_t replyID; 1255 CHECK(msg->senderAwaitsResponse(&replyID)); 1256 1257 if (mState != INITIALIZED) { 1258 PostReplyWithError(replyID, INVALID_OPERATION); 1259 break; 1260 } 1261 1262 sp<RefBase> obj; 1263 if (!msg->findObject("native-window", &obj)) { 1264 obj.clear(); 1265 } 1266 1267 sp<AMessage> format; 1268 CHECK(msg->findMessage("format", &format)); 1269 1270 if (obj != NULL) { 1271 format->setObject("native-window", obj); 1272 1273 status_t err = setNativeWindow( 1274 static_cast<NativeWindowWrapper *>(obj.get()) 1275 ->getSurfaceTextureClient()); 1276 1277 if (err != OK) { 1278 PostReplyWithError(replyID, err); 1279 break; 1280 } 1281 } else { 1282 setNativeWindow(NULL); 1283 } 1284 1285 mReplyID = replyID; 1286 setState(CONFIGURING); 1287 1288 void *crypto; 1289 if (!msg->findPointer("crypto", &crypto)) { 1290 crypto = NULL; 1291 } 1292 1293 mCrypto = static_cast<ICrypto *>(crypto); 1294 1295 uint32_t flags; 1296 CHECK(msg->findInt32("flags", (int32_t *)&flags)); 1297 1298 if (flags & CONFIGURE_FLAG_ENCODE) { 1299 format->setInt32("encoder", true); 1300 mFlags |= kFlagIsEncoder; 1301 } 1302 1303 extractCSD(format); 1304 1305 mCodec->initiateConfigureComponent(format); 1306 break; 1307 } 1308 1309 case kWhatCreateInputSurface: 1310 { 1311 uint32_t replyID; 1312 CHECK(msg->senderAwaitsResponse(&replyID)); 1313 1314 // Must be configured, but can't have been started yet. 1315 if (mState != CONFIGURED) { 1316 PostReplyWithError(replyID, INVALID_OPERATION); 1317 break; 1318 } 1319 1320 mReplyID = replyID; 1321 mCodec->initiateCreateInputSurface(); 1322 break; 1323 } 1324 1325 case kWhatStart: 1326 { 1327 uint32_t replyID; 1328 CHECK(msg->senderAwaitsResponse(&replyID)); 1329 1330 if (mState == FLUSHED) { 1331 setState(STARTED); 1332 mCodec->signalResume(); 1333 PostReplyWithError(replyID, OK); 1334 break; 1335 } else if (mState != CONFIGURED) { 1336 PostReplyWithError(replyID, INVALID_OPERATION); 1337 break; 1338 } 1339 1340 mReplyID = replyID; 1341 setState(STARTING); 1342 1343 mCodec->initiateStart(); 1344 break; 1345 } 1346 1347 case kWhatStop: 1348 case kWhatRelease: 1349 { 1350 State targetState = 1351 (msg->what() == kWhatStop) ? INITIALIZED : UNINITIALIZED; 1352 1353 uint32_t replyID; 1354 CHECK(msg->senderAwaitsResponse(&replyID)); 1355 1356 if (!((mFlags & kFlagIsComponentAllocated) && targetState == UNINITIALIZED) // See 1 1357 && mState != INITIALIZED 1358 && mState != CONFIGURED && !isExecuting()) { 1359 // 1) Permit release to shut down the component if allocated. 1360 // 1361 // 2) We may be in "UNINITIALIZED" state already and 1362 // also shutdown the encoder/decoder without the 1363 // client being aware of this if media server died while 1364 // we were being stopped. The client would assume that 1365 // after stop() returned, it would be safe to call release() 1366 // and it should be in this case, no harm to allow a release() 1367 // if we're already uninitialized. 1368 sp<AMessage> response = new AMessage; 1369 status_t err = mState == targetState ? OK : INVALID_OPERATION; 1370 response->setInt32("err", err); 1371 if (err == OK && targetState == UNINITIALIZED) { 1372 mComponentName.clear(); 1373 } 1374 response->postReply(replyID); 1375 break; 1376 } 1377 1378 if (mFlags & kFlagSawMediaServerDie) { 1379 // It's dead, Jim. Don't expect initiateShutdown to yield 1380 // any useful results now... 1381 setState(UNINITIALIZED); 1382 if (targetState == UNINITIALIZED) { 1383 mComponentName.clear(); 1384 } 1385 (new AMessage)->postReply(replyID); 1386 break; 1387 } 1388 1389 mReplyID = replyID; 1390 setState(msg->what() == kWhatStop ? STOPPING : RELEASING); 1391 1392 mCodec->initiateShutdown( 1393 msg->what() == kWhatStop /* keepComponentAllocated */); 1394 1395 returnBuffersToCodec(); 1396 break; 1397 } 1398 1399 case kWhatDequeueInputBuffer: 1400 { 1401 uint32_t replyID; 1402 CHECK(msg->senderAwaitsResponse(&replyID)); 1403 1404 if (mFlags & kFlagIsAsync) { 1405 ALOGE("dequeueOutputBuffer can't be used in async mode"); 1406 PostReplyWithError(replyID, INVALID_OPERATION); 1407 break; 1408 } 1409 1410 if (mHaveInputSurface) { 1411 ALOGE("dequeueInputBuffer can't be used with input surface"); 1412 PostReplyWithError(replyID, INVALID_OPERATION); 1413 break; 1414 } 1415 1416 if (handleDequeueInputBuffer(replyID, true /* new request */)) { 1417 break; 1418 } 1419 1420 int64_t timeoutUs; 1421 CHECK(msg->findInt64("timeoutUs", &timeoutUs)); 1422 1423 if (timeoutUs == 0ll) { 1424 PostReplyWithError(replyID, -EAGAIN); 1425 break; 1426 } 1427 1428 mFlags |= kFlagDequeueInputPending; 1429 mDequeueInputReplyID = replyID; 1430 1431 if (timeoutUs > 0ll) { 1432 sp<AMessage> timeoutMsg = 1433 new AMessage(kWhatDequeueInputTimedOut, id()); 1434 timeoutMsg->setInt32( 1435 "generation", ++mDequeueInputTimeoutGeneration); 1436 timeoutMsg->post(timeoutUs); 1437 } 1438 break; 1439 } 1440 1441 case kWhatDequeueInputTimedOut: 1442 { 1443 int32_t generation; 1444 CHECK(msg->findInt32("generation", &generation)); 1445 1446 if (generation != mDequeueInputTimeoutGeneration) { 1447 // Obsolete 1448 break; 1449 } 1450 1451 CHECK(mFlags & kFlagDequeueInputPending); 1452 1453 PostReplyWithError(mDequeueInputReplyID, -EAGAIN); 1454 1455 mFlags &= ~kFlagDequeueInputPending; 1456 mDequeueInputReplyID = 0; 1457 break; 1458 } 1459 1460 case kWhatQueueInputBuffer: 1461 { 1462 uint32_t replyID; 1463 CHECK(msg->senderAwaitsResponse(&replyID)); 1464 1465 if (!isExecuting()) { 1466 PostReplyWithError(replyID, INVALID_OPERATION); 1467 break; 1468 } else if (mFlags & kFlagStickyError) { 1469 PostReplyWithError(replyID, getStickyError()); 1470 break; 1471 } 1472 1473 status_t err = onQueueInputBuffer(msg); 1474 1475 PostReplyWithError(replyID, err); 1476 break; 1477 } 1478 1479 case kWhatDequeueOutputBuffer: 1480 { 1481 uint32_t replyID; 1482 CHECK(msg->senderAwaitsResponse(&replyID)); 1483 1484 if (mFlags & kFlagIsAsync) { 1485 ALOGE("dequeueOutputBuffer can't be used in async mode"); 1486 PostReplyWithError(replyID, INVALID_OPERATION); 1487 break; 1488 } 1489 1490 if (handleDequeueOutputBuffer(replyID, true /* new request */)) { 1491 break; 1492 } 1493 1494 int64_t timeoutUs; 1495 CHECK(msg->findInt64("timeoutUs", &timeoutUs)); 1496 1497 if (timeoutUs == 0ll) { 1498 PostReplyWithError(replyID, -EAGAIN); 1499 break; 1500 } 1501 1502 mFlags |= kFlagDequeueOutputPending; 1503 mDequeueOutputReplyID = replyID; 1504 1505 if (timeoutUs > 0ll) { 1506 sp<AMessage> timeoutMsg = 1507 new AMessage(kWhatDequeueOutputTimedOut, id()); 1508 timeoutMsg->setInt32( 1509 "generation", ++mDequeueOutputTimeoutGeneration); 1510 timeoutMsg->post(timeoutUs); 1511 } 1512 break; 1513 } 1514 1515 case kWhatDequeueOutputTimedOut: 1516 { 1517 int32_t generation; 1518 CHECK(msg->findInt32("generation", &generation)); 1519 1520 if (generation != mDequeueOutputTimeoutGeneration) { 1521 // Obsolete 1522 break; 1523 } 1524 1525 CHECK(mFlags & kFlagDequeueOutputPending); 1526 1527 PostReplyWithError(mDequeueOutputReplyID, -EAGAIN); 1528 1529 mFlags &= ~kFlagDequeueOutputPending; 1530 mDequeueOutputReplyID = 0; 1531 break; 1532 } 1533 1534 case kWhatReleaseOutputBuffer: 1535 { 1536 uint32_t replyID; 1537 CHECK(msg->senderAwaitsResponse(&replyID)); 1538 1539 if (!isExecuting()) { 1540 PostReplyWithError(replyID, INVALID_OPERATION); 1541 break; 1542 } else if (mFlags & kFlagStickyError) { 1543 PostReplyWithError(replyID, getStickyError()); 1544 break; 1545 } 1546 1547 status_t err = onReleaseOutputBuffer(msg); 1548 1549 PostReplyWithError(replyID, err); 1550 break; 1551 } 1552 1553 case kWhatSignalEndOfInputStream: 1554 { 1555 uint32_t replyID; 1556 CHECK(msg->senderAwaitsResponse(&replyID)); 1557 1558 if (!isExecuting()) { 1559 PostReplyWithError(replyID, INVALID_OPERATION); 1560 break; 1561 } else if (mFlags & kFlagStickyError) { 1562 PostReplyWithError(replyID, getStickyError()); 1563 break; 1564 } 1565 1566 mReplyID = replyID; 1567 mCodec->signalEndOfInputStream(); 1568 break; 1569 } 1570 1571 case kWhatGetBuffers: 1572 { 1573 uint32_t replyID; 1574 CHECK(msg->senderAwaitsResponse(&replyID)); 1575 1576 if (!isExecuting() || (mFlags & kFlagIsAsync)) { 1577 PostReplyWithError(replyID, INVALID_OPERATION); 1578 break; 1579 } else if (mFlags & kFlagStickyError) { 1580 PostReplyWithError(replyID, getStickyError()); 1581 break; 1582 } 1583 1584 int32_t portIndex; 1585 CHECK(msg->findInt32("portIndex", &portIndex)); 1586 1587 Vector<sp<ABuffer> > *dstBuffers; 1588 CHECK(msg->findPointer("buffers", (void **)&dstBuffers)); 1589 1590 dstBuffers->clear(); 1591 const Vector<BufferInfo> &srcBuffers = mPortBuffers[portIndex]; 1592 1593 for (size_t i = 0; i < srcBuffers.size(); ++i) { 1594 const BufferInfo &info = srcBuffers.itemAt(i); 1595 1596 dstBuffers->push_back( 1597 (portIndex == kPortIndexInput && mCrypto != NULL) 1598 ? info.mEncryptedData : info.mData); 1599 } 1600 1601 (new AMessage)->postReply(replyID); 1602 break; 1603 } 1604 1605 case kWhatFlush: 1606 { 1607 uint32_t replyID; 1608 CHECK(msg->senderAwaitsResponse(&replyID)); 1609 1610 if (!isExecuting()) { 1611 PostReplyWithError(replyID, INVALID_OPERATION); 1612 break; 1613 } else if (mFlags & kFlagStickyError) { 1614 PostReplyWithError(replyID, getStickyError()); 1615 break; 1616 } 1617 1618 mReplyID = replyID; 1619 // TODO: skip flushing if already FLUSHED 1620 setState(FLUSHING); 1621 1622 mCodec->signalFlush(); 1623 returnBuffersToCodec(); 1624 break; 1625 } 1626 1627 case kWhatGetInputFormat: 1628 case kWhatGetOutputFormat: 1629 { 1630 sp<AMessage> format = 1631 (msg->what() == kWhatGetOutputFormat ? mOutputFormat : mInputFormat); 1632 1633 uint32_t replyID; 1634 CHECK(msg->senderAwaitsResponse(&replyID)); 1635 1636 if ((mState != CONFIGURED && mState != STARTING && 1637 mState != STARTED && mState != FLUSHING && 1638 mState != FLUSHED) 1639 || format == NULL) { 1640 PostReplyWithError(replyID, INVALID_OPERATION); 1641 break; 1642 } else if (mFlags & kFlagStickyError) { 1643 PostReplyWithError(replyID, getStickyError()); 1644 break; 1645 } 1646 1647 sp<AMessage> response = new AMessage; 1648 response->setMessage("format", format); 1649 response->postReply(replyID); 1650 break; 1651 } 1652 1653 case kWhatRequestIDRFrame: 1654 { 1655 mCodec->signalRequestIDRFrame(); 1656 break; 1657 } 1658 1659 case kWhatRequestActivityNotification: 1660 { 1661 CHECK(mActivityNotify == NULL); 1662 CHECK(msg->findMessage("notify", &mActivityNotify)); 1663 1664 postActivityNotificationIfPossible(); 1665 break; 1666 } 1667 1668 case kWhatGetName: 1669 { 1670 uint32_t replyID; 1671 CHECK(msg->senderAwaitsResponse(&replyID)); 1672 1673 if (mComponentName.empty()) { 1674 PostReplyWithError(replyID, INVALID_OPERATION); 1675 break; 1676 } 1677 1678 sp<AMessage> response = new AMessage; 1679 response->setString("name", mComponentName.c_str()); 1680 response->postReply(replyID); 1681 break; 1682 } 1683 1684 case kWhatSetParameters: 1685 { 1686 uint32_t replyID; 1687 CHECK(msg->senderAwaitsResponse(&replyID)); 1688 1689 sp<AMessage> params; 1690 CHECK(msg->findMessage("params", ¶ms)); 1691 1692 status_t err = onSetParameters(params); 1693 1694 PostReplyWithError(replyID, err); 1695 break; 1696 } 1697 1698 default: 1699 TRESPASS(); 1700 } 1701} 1702 1703void MediaCodec::extractCSD(const sp<AMessage> &format) { 1704 mCSD.clear(); 1705 1706 size_t i = 0; 1707 for (;;) { 1708 sp<ABuffer> csd; 1709 if (!format->findBuffer(StringPrintf("csd-%u", i).c_str(), &csd)) { 1710 break; 1711 } 1712 1713 mCSD.push_back(csd); 1714 ++i; 1715 } 1716 1717 ALOGV("Found %zu pieces of codec specific data.", mCSD.size()); 1718} 1719 1720status_t MediaCodec::queueCSDInputBuffer(size_t bufferIndex) { 1721 CHECK(!mCSD.empty()); 1722 1723 const BufferInfo *info = 1724 &mPortBuffers[kPortIndexInput].itemAt(bufferIndex); 1725 1726 sp<ABuffer> csd = *mCSD.begin(); 1727 mCSD.erase(mCSD.begin()); 1728 1729 const sp<ABuffer> &codecInputData = 1730 (mCrypto != NULL) ? info->mEncryptedData : info->mData; 1731 1732 if (csd->size() > codecInputData->capacity()) { 1733 return -EINVAL; 1734 } 1735 1736 memcpy(codecInputData->data(), csd->data(), csd->size()); 1737 1738 AString errorDetailMsg; 1739 1740 sp<AMessage> msg = new AMessage(kWhatQueueInputBuffer, id()); 1741 msg->setSize("index", bufferIndex); 1742 msg->setSize("offset", 0); 1743 msg->setSize("size", csd->size()); 1744 msg->setInt64("timeUs", 0ll); 1745 msg->setInt32("flags", BUFFER_FLAG_CODECCONFIG); 1746 msg->setPointer("errorDetailMsg", &errorDetailMsg); 1747 1748 return onQueueInputBuffer(msg); 1749} 1750 1751void MediaCodec::setState(State newState) { 1752 if (newState == INITIALIZED || newState == UNINITIALIZED) { 1753 delete mSoftRenderer; 1754 mSoftRenderer = NULL; 1755 1756 mCrypto.clear(); 1757 setNativeWindow(NULL); 1758 1759 mInputFormat.clear(); 1760 mOutputFormat.clear(); 1761 mFlags &= ~kFlagOutputFormatChanged; 1762 mFlags &= ~kFlagOutputBuffersChanged; 1763 mFlags &= ~kFlagStickyError; 1764 mFlags &= ~kFlagIsEncoder; 1765 mFlags &= ~kFlagGatherCodecSpecificData; 1766 mFlags &= ~kFlagIsAsync; 1767 mStickyError = OK; 1768 1769 mActivityNotify.clear(); 1770 mCallback.clear(); 1771 } 1772 1773 if (newState == UNINITIALIZED) { 1774 // return any straggling buffers, e.g. if we got here on an error 1775 returnBuffersToCodec(); 1776 1777 // The component is gone, mediaserver's probably back up already 1778 // but should definitely be back up should we try to instantiate 1779 // another component.. and the cycle continues. 1780 mFlags &= ~kFlagSawMediaServerDie; 1781 } 1782 1783 mState = newState; 1784 1785 cancelPendingDequeueOperations(); 1786 1787 updateBatteryStat(); 1788} 1789 1790void MediaCodec::returnBuffersToCodec() { 1791 returnBuffersToCodecOnPort(kPortIndexInput); 1792 returnBuffersToCodecOnPort(kPortIndexOutput); 1793} 1794 1795void MediaCodec::returnBuffersToCodecOnPort(int32_t portIndex) { 1796 CHECK(portIndex == kPortIndexInput || portIndex == kPortIndexOutput); 1797 Mutex::Autolock al(mBufferLock); 1798 1799 Vector<BufferInfo> *buffers = &mPortBuffers[portIndex]; 1800 1801 for (size_t i = 0; i < buffers->size(); ++i) { 1802 BufferInfo *info = &buffers->editItemAt(i); 1803 1804 if (info->mNotify != NULL) { 1805 sp<AMessage> msg = info->mNotify; 1806 info->mNotify = NULL; 1807 info->mOwnedByClient = false; 1808 1809 if (portIndex == kPortIndexInput) { 1810 /* no error, just returning buffers */ 1811 msg->setInt32("err", OK); 1812 } 1813 msg->post(); 1814 } 1815 } 1816 1817 mAvailPortBuffers[portIndex].clear(); 1818} 1819 1820size_t MediaCodec::updateBuffers( 1821 int32_t portIndex, const sp<AMessage> &msg) { 1822 CHECK(portIndex == kPortIndexInput || portIndex == kPortIndexOutput); 1823 1824 uint32_t bufferID; 1825 CHECK(msg->findInt32("buffer-id", (int32_t*)&bufferID)); 1826 1827 Vector<BufferInfo> *buffers = &mPortBuffers[portIndex]; 1828 1829 for (size_t i = 0; i < buffers->size(); ++i) { 1830 BufferInfo *info = &buffers->editItemAt(i); 1831 1832 if (info->mBufferID == bufferID) { 1833 CHECK(info->mNotify == NULL); 1834 CHECK(msg->findMessage("reply", &info->mNotify)); 1835 1836 info->mFormat = 1837 (portIndex == kPortIndexInput) ? mInputFormat : mOutputFormat; 1838 mAvailPortBuffers[portIndex].push_back(i); 1839 1840 return i; 1841 } 1842 } 1843 1844 TRESPASS(); 1845 1846 return 0; 1847} 1848 1849status_t MediaCodec::onQueueInputBuffer(const sp<AMessage> &msg) { 1850 size_t index; 1851 size_t offset; 1852 size_t size; 1853 int64_t timeUs; 1854 uint32_t flags; 1855 CHECK(msg->findSize("index", &index)); 1856 CHECK(msg->findSize("offset", &offset)); 1857 CHECK(msg->findInt64("timeUs", &timeUs)); 1858 CHECK(msg->findInt32("flags", (int32_t *)&flags)); 1859 1860 const CryptoPlugin::SubSample *subSamples; 1861 size_t numSubSamples; 1862 const uint8_t *key; 1863 const uint8_t *iv; 1864 CryptoPlugin::Mode mode = CryptoPlugin::kMode_Unencrypted; 1865 1866 // We allow the simpler queueInputBuffer API to be used even in 1867 // secure mode, by fabricating a single unencrypted subSample. 1868 CryptoPlugin::SubSample ss; 1869 1870 if (msg->findSize("size", &size)) { 1871 if (mCrypto != NULL) { 1872 ss.mNumBytesOfClearData = size; 1873 ss.mNumBytesOfEncryptedData = 0; 1874 1875 subSamples = &ss; 1876 numSubSamples = 1; 1877 key = NULL; 1878 iv = NULL; 1879 } 1880 } else { 1881 if (mCrypto == NULL) { 1882 return -EINVAL; 1883 } 1884 1885 CHECK(msg->findPointer("subSamples", (void **)&subSamples)); 1886 CHECK(msg->findSize("numSubSamples", &numSubSamples)); 1887 CHECK(msg->findPointer("key", (void **)&key)); 1888 CHECK(msg->findPointer("iv", (void **)&iv)); 1889 1890 int32_t tmp; 1891 CHECK(msg->findInt32("mode", &tmp)); 1892 1893 mode = (CryptoPlugin::Mode)tmp; 1894 1895 size = 0; 1896 for (size_t i = 0; i < numSubSamples; ++i) { 1897 size += subSamples[i].mNumBytesOfClearData; 1898 size += subSamples[i].mNumBytesOfEncryptedData; 1899 } 1900 } 1901 1902 if (index >= mPortBuffers[kPortIndexInput].size()) { 1903 return -ERANGE; 1904 } 1905 1906 BufferInfo *info = &mPortBuffers[kPortIndexInput].editItemAt(index); 1907 1908 if (info->mNotify == NULL || !info->mOwnedByClient) { 1909 return -EACCES; 1910 } 1911 1912 if (offset + size > info->mData->capacity()) { 1913 return -EINVAL; 1914 } 1915 1916 sp<AMessage> reply = info->mNotify; 1917 info->mData->setRange(offset, size); 1918 info->mData->meta()->setInt64("timeUs", timeUs); 1919 1920 if (flags & BUFFER_FLAG_EOS) { 1921 info->mData->meta()->setInt32("eos", true); 1922 } 1923 1924 if (flags & BUFFER_FLAG_CODECCONFIG) { 1925 info->mData->meta()->setInt32("csd", true); 1926 } 1927 1928 if (mCrypto != NULL) { 1929 if (size > info->mEncryptedData->capacity()) { 1930 return -ERANGE; 1931 } 1932 1933 AString *errorDetailMsg; 1934 CHECK(msg->findPointer("errorDetailMsg", (void **)&errorDetailMsg)); 1935 1936 ssize_t result = mCrypto->decrypt( 1937 (mFlags & kFlagIsSecure) != 0, 1938 key, 1939 iv, 1940 mode, 1941 info->mEncryptedData->base() + offset, 1942 subSamples, 1943 numSubSamples, 1944 info->mData->base(), 1945 errorDetailMsg); 1946 1947 if (result < 0) { 1948 return result; 1949 } 1950 1951 info->mData->setRange(0, result); 1952 } 1953 1954 // synchronization boundary for getBufferAndFormat 1955 { 1956 Mutex::Autolock al(mBufferLock); 1957 info->mOwnedByClient = false; 1958 } 1959 reply->setBuffer("buffer", info->mData); 1960 reply->post(); 1961 1962 info->mNotify = NULL; 1963 1964 return OK; 1965} 1966 1967status_t MediaCodec::onReleaseOutputBuffer(const sp<AMessage> &msg) { 1968 size_t index; 1969 CHECK(msg->findSize("index", &index)); 1970 1971 int32_t render; 1972 if (!msg->findInt32("render", &render)) { 1973 render = 0; 1974 } 1975 1976 if (!isExecuting()) { 1977 return -EINVAL; 1978 } 1979 1980 if (index >= mPortBuffers[kPortIndexOutput].size()) { 1981 return -ERANGE; 1982 } 1983 1984 BufferInfo *info = &mPortBuffers[kPortIndexOutput].editItemAt(index); 1985 1986 if (info->mNotify == NULL || !info->mOwnedByClient) { 1987 return -EACCES; 1988 } 1989 1990 // synchronization boundary for getBufferAndFormat 1991 { 1992 Mutex::Autolock al(mBufferLock); 1993 info->mOwnedByClient = false; 1994 } 1995 1996 if (render && info->mData != NULL && info->mData->size() != 0) { 1997 info->mNotify->setInt32("render", true); 1998 1999 int64_t timestampNs = 0; 2000 if (msg->findInt64("timestampNs", ×tampNs)) { 2001 info->mNotify->setInt64("timestampNs", timestampNs); 2002 } else { 2003 // TODO: it seems like we should use the timestamp 2004 // in the (media)buffer as it potentially came from 2005 // an input surface, but we did not propagate it prior to 2006 // API 20. Perhaps check for target SDK version. 2007#if 0 2008 if (info->mData->meta()->findInt64("timeUs", ×tampNs)) { 2009 ALOGV("using buffer PTS of %" PRId64, timestampNs); 2010 timestampNs *= 1000; 2011 } 2012#endif 2013 } 2014 2015 if (mSoftRenderer != NULL) { 2016 mSoftRenderer->render( 2017 info->mData->data(), info->mData->size(), 2018 timestampNs, NULL, info->mFormat); 2019 } 2020 } 2021 2022 info->mNotify->post(); 2023 info->mNotify = NULL; 2024 2025 return OK; 2026} 2027 2028ssize_t MediaCodec::dequeuePortBuffer(int32_t portIndex) { 2029 CHECK(portIndex == kPortIndexInput || portIndex == kPortIndexOutput); 2030 2031 List<size_t> *availBuffers = &mAvailPortBuffers[portIndex]; 2032 2033 if (availBuffers->empty()) { 2034 return -EAGAIN; 2035 } 2036 2037 size_t index = *availBuffers->begin(); 2038 availBuffers->erase(availBuffers->begin()); 2039 2040 BufferInfo *info = &mPortBuffers[portIndex].editItemAt(index); 2041 CHECK(!info->mOwnedByClient); 2042 { 2043 Mutex::Autolock al(mBufferLock); 2044 info->mOwnedByClient = true; 2045 2046 // set image-data 2047 if (info->mFormat != NULL) { 2048 sp<ABuffer> imageData; 2049 if (info->mFormat->findBuffer("image-data", &imageData)) { 2050 info->mData->meta()->setBuffer("image-data", imageData); 2051 } 2052 int32_t left, top, right, bottom; 2053 if (info->mFormat->findRect("crop", &left, &top, &right, &bottom)) { 2054 info->mData->meta()->setRect("crop-rect", left, top, right, bottom); 2055 } 2056 } 2057 } 2058 2059 return index; 2060} 2061 2062status_t MediaCodec::setNativeWindow( 2063 const sp<Surface> &surfaceTextureClient) { 2064 status_t err; 2065 2066 if (mNativeWindow != NULL) { 2067 err = native_window_api_disconnect( 2068 mNativeWindow.get(), NATIVE_WINDOW_API_MEDIA); 2069 2070 if (err != OK) { 2071 ALOGW("native_window_api_disconnect returned an error: %s (%d)", 2072 strerror(-err), err); 2073 } 2074 2075 mNativeWindow.clear(); 2076 } 2077 2078 if (surfaceTextureClient != NULL) { 2079 err = native_window_api_connect( 2080 surfaceTextureClient.get(), NATIVE_WINDOW_API_MEDIA); 2081 2082 if (err != OK) { 2083 ALOGE("native_window_api_connect returned an error: %s (%d)", 2084 strerror(-err), err); 2085 2086 return err; 2087 } 2088 2089 mNativeWindow = surfaceTextureClient; 2090 } 2091 2092 return OK; 2093} 2094 2095void MediaCodec::onInputBufferAvailable() { 2096 int32_t index; 2097 while ((index = dequeuePortBuffer(kPortIndexInput)) >= 0) { 2098 sp<AMessage> msg = mCallback->dup(); 2099 msg->setInt32("callbackID", CB_INPUT_AVAILABLE); 2100 msg->setInt32("index", index); 2101 msg->post(); 2102 } 2103} 2104 2105void MediaCodec::onOutputBufferAvailable() { 2106 int32_t index; 2107 while ((index = dequeuePortBuffer(kPortIndexOutput)) >= 0) { 2108 const sp<ABuffer> &buffer = 2109 mPortBuffers[kPortIndexOutput].itemAt(index).mData; 2110 sp<AMessage> msg = mCallback->dup(); 2111 msg->setInt32("callbackID", CB_OUTPUT_AVAILABLE); 2112 msg->setInt32("index", index); 2113 msg->setSize("offset", buffer->offset()); 2114 msg->setSize("size", buffer->size()); 2115 2116 int64_t timeUs; 2117 CHECK(buffer->meta()->findInt64("timeUs", &timeUs)); 2118 2119 msg->setInt64("timeUs", timeUs); 2120 2121 int32_t omxFlags; 2122 CHECK(buffer->meta()->findInt32("omxFlags", &omxFlags)); 2123 2124 uint32_t flags = 0; 2125 if (omxFlags & OMX_BUFFERFLAG_SYNCFRAME) { 2126 flags |= BUFFER_FLAG_SYNCFRAME; 2127 } 2128 if (omxFlags & OMX_BUFFERFLAG_CODECCONFIG) { 2129 flags |= BUFFER_FLAG_CODECCONFIG; 2130 } 2131 if (omxFlags & OMX_BUFFERFLAG_EOS) { 2132 flags |= BUFFER_FLAG_EOS; 2133 } 2134 2135 msg->setInt32("flags", flags); 2136 2137 msg->post(); 2138 } 2139} 2140 2141void MediaCodec::onError(status_t err, int32_t actionCode, const char *detail) { 2142 if (mCallback != NULL) { 2143 sp<AMessage> msg = mCallback->dup(); 2144 msg->setInt32("callbackID", CB_ERROR); 2145 msg->setInt32("err", err); 2146 msg->setInt32("actionCode", actionCode); 2147 2148 if (detail != NULL) { 2149 msg->setString("detail", detail); 2150 } 2151 2152 msg->post(); 2153 } 2154} 2155 2156void MediaCodec::onOutputFormatChanged() { 2157 if (mCallback != NULL) { 2158 sp<AMessage> msg = mCallback->dup(); 2159 msg->setInt32("callbackID", CB_OUTPUT_FORMAT_CHANGED); 2160 msg->setMessage("format", mOutputFormat); 2161 msg->post(); 2162 } 2163} 2164 2165 2166void MediaCodec::postActivityNotificationIfPossible() { 2167 if (mActivityNotify == NULL) { 2168 return; 2169 } 2170 2171 bool isErrorOrOutputChanged = 2172 (mFlags & (kFlagStickyError 2173 | kFlagOutputBuffersChanged 2174 | kFlagOutputFormatChanged)); 2175 2176 if (isErrorOrOutputChanged 2177 || !mAvailPortBuffers[kPortIndexInput].empty() 2178 || !mAvailPortBuffers[kPortIndexOutput].empty()) { 2179 mActivityNotify->setInt32("input-buffers", 2180 mAvailPortBuffers[kPortIndexInput].size()); 2181 2182 if (isErrorOrOutputChanged) { 2183 // we want consumer to dequeue as many times as it can 2184 mActivityNotify->setInt32("output-buffers", INT32_MAX); 2185 } else { 2186 mActivityNotify->setInt32("output-buffers", 2187 mAvailPortBuffers[kPortIndexOutput].size()); 2188 } 2189 mActivityNotify->post(); 2190 mActivityNotify.clear(); 2191 } 2192} 2193 2194status_t MediaCodec::setParameters(const sp<AMessage> ¶ms) { 2195 sp<AMessage> msg = new AMessage(kWhatSetParameters, id()); 2196 msg->setMessage("params", params); 2197 2198 sp<AMessage> response; 2199 return PostAndAwaitResponse(msg, &response); 2200} 2201 2202status_t MediaCodec::onSetParameters(const sp<AMessage> ¶ms) { 2203 mCodec->signalSetParameters(params); 2204 2205 return OK; 2206} 2207 2208status_t MediaCodec::amendOutputFormatWithCodecSpecificData( 2209 const sp<ABuffer> &buffer) { 2210 AString mime; 2211 CHECK(mOutputFormat->findString("mime", &mime)); 2212 2213 if (!strcasecmp(mime.c_str(), MEDIA_MIMETYPE_VIDEO_AVC)) { 2214 // Codec specific data should be SPS and PPS in a single buffer, 2215 // each prefixed by a startcode (0x00 0x00 0x00 0x01). 2216 // We separate the two and put them into the output format 2217 // under the keys "csd-0" and "csd-1". 2218 2219 unsigned csdIndex = 0; 2220 2221 const uint8_t *data = buffer->data(); 2222 size_t size = buffer->size(); 2223 2224 const uint8_t *nalStart; 2225 size_t nalSize; 2226 while (getNextNALUnit(&data, &size, &nalStart, &nalSize, true) == OK) { 2227 sp<ABuffer> csd = new ABuffer(nalSize + 4); 2228 memcpy(csd->data(), "\x00\x00\x00\x01", 4); 2229 memcpy(csd->data() + 4, nalStart, nalSize); 2230 2231 mOutputFormat->setBuffer( 2232 StringPrintf("csd-%u", csdIndex).c_str(), csd); 2233 2234 ++csdIndex; 2235 } 2236 2237 if (csdIndex != 2) { 2238 return ERROR_MALFORMED; 2239 } 2240 } else { 2241 // For everything else we just stash the codec specific data into 2242 // the output format as a single piece of csd under "csd-0". 2243 mOutputFormat->setBuffer("csd-0", buffer); 2244 } 2245 2246 return OK; 2247} 2248 2249void MediaCodec::updateBatteryStat() { 2250 if (mState == CONFIGURED && !mBatteryStatNotified) { 2251 AString mime; 2252 CHECK(mOutputFormat != NULL && 2253 mOutputFormat->findString("mime", &mime)); 2254 2255 mIsVideo = mime.startsWithIgnoreCase("video/"); 2256 2257 BatteryNotifier& notifier(BatteryNotifier::getInstance()); 2258 2259 if (mIsVideo) { 2260 notifier.noteStartVideo(); 2261 } else { 2262 notifier.noteStartAudio(); 2263 } 2264 2265 mBatteryStatNotified = true; 2266 } else if (mState == UNINITIALIZED && mBatteryStatNotified) { 2267 BatteryNotifier& notifier(BatteryNotifier::getInstance()); 2268 2269 if (mIsVideo) { 2270 notifier.noteStopVideo(); 2271 } else { 2272 notifier.noteStopAudio(); 2273 } 2274 2275 mBatteryStatNotified = false; 2276 } 2277} 2278 2279} // namespace android 2280