codec.cpp revision f69e53033f23f9f70fcdb28a3c2e650de0147459
15778822d86b0337407514b9372562b86edfa91cdAndreas Huber/* 25778822d86b0337407514b9372562b86edfa91cdAndreas Huber * Copyright (C) 2012 The Android Open Source Project 35778822d86b0337407514b9372562b86edfa91cdAndreas Huber * 45778822d86b0337407514b9372562b86edfa91cdAndreas Huber * Licensed under the Apache License, Version 2.0 (the "License"); 55778822d86b0337407514b9372562b86edfa91cdAndreas Huber * you may not use this file except in compliance with the License. 65778822d86b0337407514b9372562b86edfa91cdAndreas Huber * You may obtain a copy of the License at 75778822d86b0337407514b9372562b86edfa91cdAndreas Huber * 85778822d86b0337407514b9372562b86edfa91cdAndreas Huber * http://www.apache.org/licenses/LICENSE-2.0 95778822d86b0337407514b9372562b86edfa91cdAndreas Huber * 105778822d86b0337407514b9372562b86edfa91cdAndreas Huber * Unless required by applicable law or agreed to in writing, software 115778822d86b0337407514b9372562b86edfa91cdAndreas Huber * distributed under the License is distributed on an "AS IS" BASIS, 125778822d86b0337407514b9372562b86edfa91cdAndreas Huber * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 135778822d86b0337407514b9372562b86edfa91cdAndreas Huber * See the License for the specific language governing permissions and 145778822d86b0337407514b9372562b86edfa91cdAndreas Huber * limitations under the License. 155778822d86b0337407514b9372562b86edfa91cdAndreas Huber */ 165778822d86b0337407514b9372562b86edfa91cdAndreas Huber 175778822d86b0337407514b9372562b86edfa91cdAndreas Huber//#define LOG_NDEBUG 0 185778822d86b0337407514b9372562b86edfa91cdAndreas Huber#define LOG_TAG "codec" 195778822d86b0337407514b9372562b86edfa91cdAndreas Huber#include <utils/Log.h> 205778822d86b0337407514b9372562b86edfa91cdAndreas Huber 215778822d86b0337407514b9372562b86edfa91cdAndreas Huber#include "SimplePlayer.h" 225778822d86b0337407514b9372562b86edfa91cdAndreas Huber 231bd139a2a68690e80398b70b27ca59550fea0e65Andreas Huber#include <binder/IServiceManager.h> 245778822d86b0337407514b9372562b86edfa91cdAndreas Huber#include <binder/ProcessState.h> 251bd139a2a68690e80398b70b27ca59550fea0e65Andreas Huber#include <media/ICrypto.h> 261bd139a2a68690e80398b70b27ca59550fea0e65Andreas Huber#include <media/IMediaPlayerService.h> 275778822d86b0337407514b9372562b86edfa91cdAndreas Huber#include <media/stagefright/foundation/ABuffer.h> 285778822d86b0337407514b9372562b86edfa91cdAndreas Huber#include <media/stagefright/foundation/ADebug.h> 295778822d86b0337407514b9372562b86edfa91cdAndreas Huber#include <media/stagefright/foundation/ALooper.h> 305778822d86b0337407514b9372562b86edfa91cdAndreas Huber#include <media/stagefright/foundation/AMessage.h> 315778822d86b0337407514b9372562b86edfa91cdAndreas Huber#include <media/stagefright/DataSource.h> 325778822d86b0337407514b9372562b86edfa91cdAndreas Huber#include <media/stagefright/MediaCodec.h> 33ed3e3e046840d5bf1ca84a8c0cc097425e89d6d6Andreas Huber#include <media/stagefright/MediaCodecList.h> 345778822d86b0337407514b9372562b86edfa91cdAndreas Huber#include <media/stagefright/MediaDefs.h> 355778822d86b0337407514b9372562b86edfa91cdAndreas Huber#include <media/stagefright/NuMediaExtractor.h> 36df712ea86e6350f7005a02ab0e1c60c28a343ed0Mathias Agopian#include <gui/SurfaceComposerClient.h> 375778822d86b0337407514b9372562b86edfa91cdAndreas Huber 385778822d86b0337407514b9372562b86edfa91cdAndreas Huberstatic void usage(const char *me) { 395778822d86b0337407514b9372562b86edfa91cdAndreas Huber fprintf(stderr, "usage: %s [-a] use audio\n" 405778822d86b0337407514b9372562b86edfa91cdAndreas Huber "\t\t[-v] use video\n" 41bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber "\t\t[-p] playback\n" 42f69e53033f23f9f70fcdb28a3c2e650de0147459Andreas Huber "\t\t[-S] allocate buffers from a surface\n", 43ed3e3e046840d5bf1ca84a8c0cc097425e89d6d6Andreas Huber me); 445778822d86b0337407514b9372562b86edfa91cdAndreas Huber 455778822d86b0337407514b9372562b86edfa91cdAndreas Huber exit(1); 465778822d86b0337407514b9372562b86edfa91cdAndreas Huber} 475778822d86b0337407514b9372562b86edfa91cdAndreas Huber 485778822d86b0337407514b9372562b86edfa91cdAndreas Hubernamespace android { 495778822d86b0337407514b9372562b86edfa91cdAndreas Huber 505778822d86b0337407514b9372562b86edfa91cdAndreas Huberstruct CodecState { 515778822d86b0337407514b9372562b86edfa91cdAndreas Huber sp<MediaCodec> mCodec; 525778822d86b0337407514b9372562b86edfa91cdAndreas Huber Vector<sp<ABuffer> > mCSD; 535778822d86b0337407514b9372562b86edfa91cdAndreas Huber size_t mCSDIndex; 545778822d86b0337407514b9372562b86edfa91cdAndreas Huber Vector<sp<ABuffer> > mInBuffers; 555778822d86b0337407514b9372562b86edfa91cdAndreas Huber Vector<sp<ABuffer> > mOutBuffers; 565aaeb0d64fc98f9b019a4378eb39f0ee49ee6ec4Andreas Huber bool mSignalledInputEOS; 575778822d86b0337407514b9372562b86edfa91cdAndreas Huber bool mSawOutputEOS; 58bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber int64_t mNumBuffersDecoded; 59bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber int64_t mNumBytesDecoded; 60bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber bool mIsAudio; 615778822d86b0337407514b9372562b86edfa91cdAndreas Huber}; 625778822d86b0337407514b9372562b86edfa91cdAndreas Huber 635778822d86b0337407514b9372562b86edfa91cdAndreas Huber} // namespace android 645778822d86b0337407514b9372562b86edfa91cdAndreas Huber 655778822d86b0337407514b9372562b86edfa91cdAndreas Huberstatic int decode( 665778822d86b0337407514b9372562b86edfa91cdAndreas Huber const android::sp<android::ALooper> &looper, 675778822d86b0337407514b9372562b86edfa91cdAndreas Huber const char *path, 685778822d86b0337407514b9372562b86edfa91cdAndreas Huber bool useAudio, 69bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber bool useVideo, 70f69e53033f23f9f70fcdb28a3c2e650de0147459Andreas Huber const android::sp<android::Surface> &surface) { 715778822d86b0337407514b9372562b86edfa91cdAndreas Huber using namespace android; 725778822d86b0337407514b9372562b86edfa91cdAndreas Huber 73bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber static int64_t kTimeout = 500ll; 74bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber 755778822d86b0337407514b9372562b86edfa91cdAndreas Huber sp<NuMediaExtractor> extractor = new NuMediaExtractor; 765778822d86b0337407514b9372562b86edfa91cdAndreas Huber if (extractor->setDataSource(path) != OK) { 775778822d86b0337407514b9372562b86edfa91cdAndreas Huber fprintf(stderr, "unable to instantiate extractor.\n"); 785778822d86b0337407514b9372562b86edfa91cdAndreas Huber return 1; 795778822d86b0337407514b9372562b86edfa91cdAndreas Huber } 805778822d86b0337407514b9372562b86edfa91cdAndreas Huber 815778822d86b0337407514b9372562b86edfa91cdAndreas Huber KeyedVector<size_t, CodecState> stateByTrack; 825778822d86b0337407514b9372562b86edfa91cdAndreas Huber 835778822d86b0337407514b9372562b86edfa91cdAndreas Huber bool haveAudio = false; 845778822d86b0337407514b9372562b86edfa91cdAndreas Huber bool haveVideo = false; 855778822d86b0337407514b9372562b86edfa91cdAndreas Huber for (size_t i = 0; i < extractor->countTracks(); ++i) { 865778822d86b0337407514b9372562b86edfa91cdAndreas Huber sp<AMessage> format; 875778822d86b0337407514b9372562b86edfa91cdAndreas Huber status_t err = extractor->getTrackFormat(i, &format); 885778822d86b0337407514b9372562b86edfa91cdAndreas Huber CHECK_EQ(err, (status_t)OK); 895778822d86b0337407514b9372562b86edfa91cdAndreas Huber 905778822d86b0337407514b9372562b86edfa91cdAndreas Huber AString mime; 915778822d86b0337407514b9372562b86edfa91cdAndreas Huber CHECK(format->findString("mime", &mime)); 925778822d86b0337407514b9372562b86edfa91cdAndreas Huber 93bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber bool isAudio = !strncasecmp(mime.c_str(), "audio/", 6); 94bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber bool isVideo = !strncasecmp(mime.c_str(), "video/", 6); 95bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber 96bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber if (useAudio && !haveAudio && isAudio) { 975778822d86b0337407514b9372562b86edfa91cdAndreas Huber haveAudio = true; 98bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber } else if (useVideo && !haveVideo && isVideo) { 995778822d86b0337407514b9372562b86edfa91cdAndreas Huber haveVideo = true; 1005778822d86b0337407514b9372562b86edfa91cdAndreas Huber } else { 1015778822d86b0337407514b9372562b86edfa91cdAndreas Huber continue; 1025778822d86b0337407514b9372562b86edfa91cdAndreas Huber } 1035778822d86b0337407514b9372562b86edfa91cdAndreas Huber 1045778822d86b0337407514b9372562b86edfa91cdAndreas Huber ALOGV("selecting track %d", i); 1055778822d86b0337407514b9372562b86edfa91cdAndreas Huber 1065778822d86b0337407514b9372562b86edfa91cdAndreas Huber err = extractor->selectTrack(i); 1075778822d86b0337407514b9372562b86edfa91cdAndreas Huber CHECK_EQ(err, (status_t)OK); 1085778822d86b0337407514b9372562b86edfa91cdAndreas Huber 1095778822d86b0337407514b9372562b86edfa91cdAndreas Huber CodecState *state = 1105778822d86b0337407514b9372562b86edfa91cdAndreas Huber &stateByTrack.editValueAt(stateByTrack.add(i, CodecState())); 1115778822d86b0337407514b9372562b86edfa91cdAndreas Huber 112bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber state->mNumBytesDecoded = 0; 113bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber state->mNumBuffersDecoded = 0; 114bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber state->mIsAudio = isAudio; 115bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber 116f69e53033f23f9f70fcdb28a3c2e650de0147459Andreas Huber state->mCodec = MediaCodec::CreateByType( 117f69e53033f23f9f70fcdb28a3c2e650de0147459Andreas Huber looper, mime.c_str(), false /* encoder */); 1185778822d86b0337407514b9372562b86edfa91cdAndreas Huber 1195778822d86b0337407514b9372562b86edfa91cdAndreas Huber CHECK(state->mCodec != NULL); 1205778822d86b0337407514b9372562b86edfa91cdAndreas Huber 1215778822d86b0337407514b9372562b86edfa91cdAndreas Huber err = state->mCodec->configure( 122ed3e3e046840d5bf1ca84a8c0cc097425e89d6d6Andreas Huber format, isVideo ? surface : NULL, 123f69e53033f23f9f70fcdb28a3c2e650de0147459Andreas Huber NULL /* crypto */, 1241bd139a2a68690e80398b70b27ca59550fea0e65Andreas Huber 0 /* flags */); 1255778822d86b0337407514b9372562b86edfa91cdAndreas Huber 1265778822d86b0337407514b9372562b86edfa91cdAndreas Huber CHECK_EQ(err, (status_t)OK); 1275778822d86b0337407514b9372562b86edfa91cdAndreas Huber 1285778822d86b0337407514b9372562b86edfa91cdAndreas Huber size_t j = 0; 129bf6c85a013fb14960bac147c1ffd0a02a8d5f148Andreas Huber sp<ABuffer> buffer; 130bf6c85a013fb14960bac147c1ffd0a02a8d5f148Andreas Huber while (format->findBuffer(StringPrintf("csd-%d", j).c_str(), &buffer)) { 1315778822d86b0337407514b9372562b86edfa91cdAndreas Huber state->mCSD.push_back(buffer); 1325778822d86b0337407514b9372562b86edfa91cdAndreas Huber 1335778822d86b0337407514b9372562b86edfa91cdAndreas Huber ++j; 1345778822d86b0337407514b9372562b86edfa91cdAndreas Huber } 1355778822d86b0337407514b9372562b86edfa91cdAndreas Huber 1365778822d86b0337407514b9372562b86edfa91cdAndreas Huber state->mCSDIndex = 0; 1375aaeb0d64fc98f9b019a4378eb39f0ee49ee6ec4Andreas Huber state->mSignalledInputEOS = false; 1385778822d86b0337407514b9372562b86edfa91cdAndreas Huber state->mSawOutputEOS = false; 1395778822d86b0337407514b9372562b86edfa91cdAndreas Huber 1405778822d86b0337407514b9372562b86edfa91cdAndreas Huber ALOGV("got %d pieces of codec specific data.", state->mCSD.size()); 1415778822d86b0337407514b9372562b86edfa91cdAndreas Huber } 1425778822d86b0337407514b9372562b86edfa91cdAndreas Huber 1435778822d86b0337407514b9372562b86edfa91cdAndreas Huber CHECK(!stateByTrack.isEmpty()); 1445778822d86b0337407514b9372562b86edfa91cdAndreas Huber 145bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber int64_t startTimeUs = ALooper::GetNowUs(); 146bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber 1475778822d86b0337407514b9372562b86edfa91cdAndreas Huber for (size_t i = 0; i < stateByTrack.size(); ++i) { 1485778822d86b0337407514b9372562b86edfa91cdAndreas Huber CodecState *state = &stateByTrack.editValueAt(i); 1495778822d86b0337407514b9372562b86edfa91cdAndreas Huber 1505778822d86b0337407514b9372562b86edfa91cdAndreas Huber sp<MediaCodec> codec = state->mCodec; 1515778822d86b0337407514b9372562b86edfa91cdAndreas Huber 1525778822d86b0337407514b9372562b86edfa91cdAndreas Huber CHECK_EQ((status_t)OK, codec->start()); 1535778822d86b0337407514b9372562b86edfa91cdAndreas Huber 1545778822d86b0337407514b9372562b86edfa91cdAndreas Huber CHECK_EQ((status_t)OK, codec->getInputBuffers(&state->mInBuffers)); 1555778822d86b0337407514b9372562b86edfa91cdAndreas Huber CHECK_EQ((status_t)OK, codec->getOutputBuffers(&state->mOutBuffers)); 1565778822d86b0337407514b9372562b86edfa91cdAndreas Huber 1575778822d86b0337407514b9372562b86edfa91cdAndreas Huber ALOGV("got %d input and %d output buffers", 1585778822d86b0337407514b9372562b86edfa91cdAndreas Huber state->mInBuffers.size(), state->mOutBuffers.size()); 1595778822d86b0337407514b9372562b86edfa91cdAndreas Huber 1605778822d86b0337407514b9372562b86edfa91cdAndreas Huber while (state->mCSDIndex < state->mCSD.size()) { 1615778822d86b0337407514b9372562b86edfa91cdAndreas Huber size_t index; 162bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber status_t err = codec->dequeueInputBuffer(&index, -1ll); 1635778822d86b0337407514b9372562b86edfa91cdAndreas Huber CHECK_EQ(err, (status_t)OK); 1645778822d86b0337407514b9372562b86edfa91cdAndreas Huber 1655778822d86b0337407514b9372562b86edfa91cdAndreas Huber const sp<ABuffer> &srcBuffer = 1665778822d86b0337407514b9372562b86edfa91cdAndreas Huber state->mCSD.itemAt(state->mCSDIndex++); 1675778822d86b0337407514b9372562b86edfa91cdAndreas Huber 1685778822d86b0337407514b9372562b86edfa91cdAndreas Huber const sp<ABuffer> &buffer = state->mInBuffers.itemAt(index); 1695778822d86b0337407514b9372562b86edfa91cdAndreas Huber 1705778822d86b0337407514b9372562b86edfa91cdAndreas Huber memcpy(buffer->data(), srcBuffer->data(), srcBuffer->size()); 1715778822d86b0337407514b9372562b86edfa91cdAndreas Huber 1725778822d86b0337407514b9372562b86edfa91cdAndreas Huber err = codec->queueInputBuffer( 1735778822d86b0337407514b9372562b86edfa91cdAndreas Huber index, 1745778822d86b0337407514b9372562b86edfa91cdAndreas Huber 0 /* offset */, 1755778822d86b0337407514b9372562b86edfa91cdAndreas Huber srcBuffer->size(), 1765778822d86b0337407514b9372562b86edfa91cdAndreas Huber 0ll /* timeUs */, 1775778822d86b0337407514b9372562b86edfa91cdAndreas Huber MediaCodec::BUFFER_FLAG_CODECCONFIG); 1785778822d86b0337407514b9372562b86edfa91cdAndreas Huber 1795778822d86b0337407514b9372562b86edfa91cdAndreas Huber CHECK_EQ(err, (status_t)OK); 1805778822d86b0337407514b9372562b86edfa91cdAndreas Huber } 1815778822d86b0337407514b9372562b86edfa91cdAndreas Huber } 1825778822d86b0337407514b9372562b86edfa91cdAndreas Huber 1835778822d86b0337407514b9372562b86edfa91cdAndreas Huber bool sawInputEOS = false; 1845778822d86b0337407514b9372562b86edfa91cdAndreas Huber 1855778822d86b0337407514b9372562b86edfa91cdAndreas Huber for (;;) { 1865778822d86b0337407514b9372562b86edfa91cdAndreas Huber if (!sawInputEOS) { 1875778822d86b0337407514b9372562b86edfa91cdAndreas Huber size_t trackIndex; 1885778822d86b0337407514b9372562b86edfa91cdAndreas Huber status_t err = extractor->getSampleTrackIndex(&trackIndex); 1895778822d86b0337407514b9372562b86edfa91cdAndreas Huber 1905778822d86b0337407514b9372562b86edfa91cdAndreas Huber if (err != OK) { 1915aaeb0d64fc98f9b019a4378eb39f0ee49ee6ec4Andreas Huber ALOGV("saw input eos"); 1925778822d86b0337407514b9372562b86edfa91cdAndreas Huber sawInputEOS = true; 1935778822d86b0337407514b9372562b86edfa91cdAndreas Huber } else { 1945778822d86b0337407514b9372562b86edfa91cdAndreas Huber CodecState *state = &stateByTrack.editValueFor(trackIndex); 1955778822d86b0337407514b9372562b86edfa91cdAndreas Huber 1965778822d86b0337407514b9372562b86edfa91cdAndreas Huber size_t index; 197bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber err = state->mCodec->dequeueInputBuffer(&index, kTimeout); 1985778822d86b0337407514b9372562b86edfa91cdAndreas Huber 1995778822d86b0337407514b9372562b86edfa91cdAndreas Huber if (err == OK) { 2005778822d86b0337407514b9372562b86edfa91cdAndreas Huber ALOGV("filling input buffer %d", index); 2015778822d86b0337407514b9372562b86edfa91cdAndreas Huber 2025778822d86b0337407514b9372562b86edfa91cdAndreas Huber const sp<ABuffer> &buffer = state->mInBuffers.itemAt(index); 2035778822d86b0337407514b9372562b86edfa91cdAndreas Huber 2045778822d86b0337407514b9372562b86edfa91cdAndreas Huber err = extractor->readSampleData(buffer); 2055778822d86b0337407514b9372562b86edfa91cdAndreas Huber CHECK_EQ(err, (status_t)OK); 2065778822d86b0337407514b9372562b86edfa91cdAndreas Huber 2075778822d86b0337407514b9372562b86edfa91cdAndreas Huber int64_t timeUs; 2085778822d86b0337407514b9372562b86edfa91cdAndreas Huber err = extractor->getSampleTime(&timeUs); 2095778822d86b0337407514b9372562b86edfa91cdAndreas Huber CHECK_EQ(err, (status_t)OK); 2105778822d86b0337407514b9372562b86edfa91cdAndreas Huber 211ed3e3e046840d5bf1ca84a8c0cc097425e89d6d6Andreas Huber uint32_t bufferFlags = 0; 212ed3e3e046840d5bf1ca84a8c0cc097425e89d6d6Andreas Huber 213f69e53033f23f9f70fcdb28a3c2e650de0147459Andreas Huber err = state->mCodec->queueInputBuffer( 214f69e53033f23f9f70fcdb28a3c2e650de0147459Andreas Huber index, 215f69e53033f23f9f70fcdb28a3c2e650de0147459Andreas Huber 0 /* offset */, 216f69e53033f23f9f70fcdb28a3c2e650de0147459Andreas Huber buffer->size(), 217f69e53033f23f9f70fcdb28a3c2e650de0147459Andreas Huber timeUs, 218f69e53033f23f9f70fcdb28a3c2e650de0147459Andreas Huber bufferFlags); 2195778822d86b0337407514b9372562b86edfa91cdAndreas Huber 2205778822d86b0337407514b9372562b86edfa91cdAndreas Huber CHECK_EQ(err, (status_t)OK); 2215778822d86b0337407514b9372562b86edfa91cdAndreas Huber 2225778822d86b0337407514b9372562b86edfa91cdAndreas Huber extractor->advance(); 2235778822d86b0337407514b9372562b86edfa91cdAndreas Huber } else { 2245778822d86b0337407514b9372562b86edfa91cdAndreas Huber CHECK_EQ(err, -EAGAIN); 2255778822d86b0337407514b9372562b86edfa91cdAndreas Huber } 2265778822d86b0337407514b9372562b86edfa91cdAndreas Huber } 2275aaeb0d64fc98f9b019a4378eb39f0ee49ee6ec4Andreas Huber } else { 2285aaeb0d64fc98f9b019a4378eb39f0ee49ee6ec4Andreas Huber for (size_t i = 0; i < stateByTrack.size(); ++i) { 2295aaeb0d64fc98f9b019a4378eb39f0ee49ee6ec4Andreas Huber CodecState *state = &stateByTrack.editValueAt(i); 2305aaeb0d64fc98f9b019a4378eb39f0ee49ee6ec4Andreas Huber 2315aaeb0d64fc98f9b019a4378eb39f0ee49ee6ec4Andreas Huber if (!state->mSignalledInputEOS) { 2325aaeb0d64fc98f9b019a4378eb39f0ee49ee6ec4Andreas Huber size_t index; 2335aaeb0d64fc98f9b019a4378eb39f0ee49ee6ec4Andreas Huber status_t err = 2345aaeb0d64fc98f9b019a4378eb39f0ee49ee6ec4Andreas Huber state->mCodec->dequeueInputBuffer(&index, kTimeout); 2355aaeb0d64fc98f9b019a4378eb39f0ee49ee6ec4Andreas Huber 2365aaeb0d64fc98f9b019a4378eb39f0ee49ee6ec4Andreas Huber if (err == OK) { 2375aaeb0d64fc98f9b019a4378eb39f0ee49ee6ec4Andreas Huber ALOGV("signalling input EOS on track %d", i); 2385aaeb0d64fc98f9b019a4378eb39f0ee49ee6ec4Andreas Huber 2395aaeb0d64fc98f9b019a4378eb39f0ee49ee6ec4Andreas Huber err = state->mCodec->queueInputBuffer( 2405aaeb0d64fc98f9b019a4378eb39f0ee49ee6ec4Andreas Huber index, 2415aaeb0d64fc98f9b019a4378eb39f0ee49ee6ec4Andreas Huber 0 /* offset */, 2425aaeb0d64fc98f9b019a4378eb39f0ee49ee6ec4Andreas Huber 0 /* size */, 2435aaeb0d64fc98f9b019a4378eb39f0ee49ee6ec4Andreas Huber 0ll /* timeUs */, 2445aaeb0d64fc98f9b019a4378eb39f0ee49ee6ec4Andreas Huber MediaCodec::BUFFER_FLAG_EOS); 2455aaeb0d64fc98f9b019a4378eb39f0ee49ee6ec4Andreas Huber 2465aaeb0d64fc98f9b019a4378eb39f0ee49ee6ec4Andreas Huber CHECK_EQ(err, (status_t)OK); 2475aaeb0d64fc98f9b019a4378eb39f0ee49ee6ec4Andreas Huber 2485aaeb0d64fc98f9b019a4378eb39f0ee49ee6ec4Andreas Huber state->mSignalledInputEOS = true; 2495aaeb0d64fc98f9b019a4378eb39f0ee49ee6ec4Andreas Huber } else { 2505aaeb0d64fc98f9b019a4378eb39f0ee49ee6ec4Andreas Huber CHECK_EQ(err, -EAGAIN); 2515aaeb0d64fc98f9b019a4378eb39f0ee49ee6ec4Andreas Huber } 2525aaeb0d64fc98f9b019a4378eb39f0ee49ee6ec4Andreas Huber } 2535aaeb0d64fc98f9b019a4378eb39f0ee49ee6ec4Andreas Huber } 2545778822d86b0337407514b9372562b86edfa91cdAndreas Huber } 2555778822d86b0337407514b9372562b86edfa91cdAndreas Huber 2565778822d86b0337407514b9372562b86edfa91cdAndreas Huber bool sawOutputEOSOnAllTracks = true; 2575778822d86b0337407514b9372562b86edfa91cdAndreas Huber for (size_t i = 0; i < stateByTrack.size(); ++i) { 2585778822d86b0337407514b9372562b86edfa91cdAndreas Huber CodecState *state = &stateByTrack.editValueAt(i); 2595778822d86b0337407514b9372562b86edfa91cdAndreas Huber if (!state->mSawOutputEOS) { 2605778822d86b0337407514b9372562b86edfa91cdAndreas Huber sawOutputEOSOnAllTracks = false; 2615778822d86b0337407514b9372562b86edfa91cdAndreas Huber break; 2625778822d86b0337407514b9372562b86edfa91cdAndreas Huber } 2635778822d86b0337407514b9372562b86edfa91cdAndreas Huber } 2645778822d86b0337407514b9372562b86edfa91cdAndreas Huber 2655778822d86b0337407514b9372562b86edfa91cdAndreas Huber if (sawOutputEOSOnAllTracks) { 2665778822d86b0337407514b9372562b86edfa91cdAndreas Huber break; 2675778822d86b0337407514b9372562b86edfa91cdAndreas Huber } 2685778822d86b0337407514b9372562b86edfa91cdAndreas Huber 2695778822d86b0337407514b9372562b86edfa91cdAndreas Huber for (size_t i = 0; i < stateByTrack.size(); ++i) { 2705778822d86b0337407514b9372562b86edfa91cdAndreas Huber CodecState *state = &stateByTrack.editValueAt(i); 2715778822d86b0337407514b9372562b86edfa91cdAndreas Huber 2725778822d86b0337407514b9372562b86edfa91cdAndreas Huber if (state->mSawOutputEOS) { 2735778822d86b0337407514b9372562b86edfa91cdAndreas Huber continue; 2745778822d86b0337407514b9372562b86edfa91cdAndreas Huber } 2755778822d86b0337407514b9372562b86edfa91cdAndreas Huber 2765778822d86b0337407514b9372562b86edfa91cdAndreas Huber size_t index; 2775778822d86b0337407514b9372562b86edfa91cdAndreas Huber size_t offset; 2785778822d86b0337407514b9372562b86edfa91cdAndreas Huber size_t size; 2795778822d86b0337407514b9372562b86edfa91cdAndreas Huber int64_t presentationTimeUs; 2805778822d86b0337407514b9372562b86edfa91cdAndreas Huber uint32_t flags; 2815778822d86b0337407514b9372562b86edfa91cdAndreas Huber status_t err = state->mCodec->dequeueOutputBuffer( 2825778822d86b0337407514b9372562b86edfa91cdAndreas Huber &index, &offset, &size, &presentationTimeUs, &flags, 283bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber kTimeout); 2845778822d86b0337407514b9372562b86edfa91cdAndreas Huber 2855778822d86b0337407514b9372562b86edfa91cdAndreas Huber if (err == OK) { 2865778822d86b0337407514b9372562b86edfa91cdAndreas Huber ALOGV("draining output buffer %d, time = %lld us", 2875778822d86b0337407514b9372562b86edfa91cdAndreas Huber index, presentationTimeUs); 2885778822d86b0337407514b9372562b86edfa91cdAndreas Huber 289bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber ++state->mNumBuffersDecoded; 290bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber state->mNumBytesDecoded += size; 291bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber 2925778822d86b0337407514b9372562b86edfa91cdAndreas Huber err = state->mCodec->releaseOutputBuffer(index); 2935778822d86b0337407514b9372562b86edfa91cdAndreas Huber CHECK_EQ(err, (status_t)OK); 2945778822d86b0337407514b9372562b86edfa91cdAndreas Huber 2955778822d86b0337407514b9372562b86edfa91cdAndreas Huber if (flags & MediaCodec::BUFFER_FLAG_EOS) { 2965778822d86b0337407514b9372562b86edfa91cdAndreas Huber ALOGV("reached EOS on output."); 2975778822d86b0337407514b9372562b86edfa91cdAndreas Huber 2985778822d86b0337407514b9372562b86edfa91cdAndreas Huber state->mSawOutputEOS = true; 2995778822d86b0337407514b9372562b86edfa91cdAndreas Huber } 3005778822d86b0337407514b9372562b86edfa91cdAndreas Huber } else if (err == INFO_OUTPUT_BUFFERS_CHANGED) { 3015778822d86b0337407514b9372562b86edfa91cdAndreas Huber ALOGV("INFO_OUTPUT_BUFFERS_CHANGED"); 3025778822d86b0337407514b9372562b86edfa91cdAndreas Huber CHECK_EQ((status_t)OK, 3035778822d86b0337407514b9372562b86edfa91cdAndreas Huber state->mCodec->getOutputBuffers(&state->mOutBuffers)); 3045778822d86b0337407514b9372562b86edfa91cdAndreas Huber 3055778822d86b0337407514b9372562b86edfa91cdAndreas Huber ALOGV("got %d output buffers", state->mOutBuffers.size()); 3065778822d86b0337407514b9372562b86edfa91cdAndreas Huber } else if (err == INFO_FORMAT_CHANGED) { 3075778822d86b0337407514b9372562b86edfa91cdAndreas Huber sp<AMessage> format; 3085778822d86b0337407514b9372562b86edfa91cdAndreas Huber CHECK_EQ((status_t)OK, state->mCodec->getOutputFormat(&format)); 3095778822d86b0337407514b9372562b86edfa91cdAndreas Huber 3105778822d86b0337407514b9372562b86edfa91cdAndreas Huber ALOGV("INFO_FORMAT_CHANGED: %s", format->debugString().c_str()); 3115778822d86b0337407514b9372562b86edfa91cdAndreas Huber } else { 3125778822d86b0337407514b9372562b86edfa91cdAndreas Huber CHECK_EQ(err, -EAGAIN); 3135778822d86b0337407514b9372562b86edfa91cdAndreas Huber } 3145778822d86b0337407514b9372562b86edfa91cdAndreas Huber } 3155778822d86b0337407514b9372562b86edfa91cdAndreas Huber } 3165778822d86b0337407514b9372562b86edfa91cdAndreas Huber 317bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber int64_t elapsedTimeUs = ALooper::GetNowUs() - startTimeUs; 318bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber 3195778822d86b0337407514b9372562b86edfa91cdAndreas Huber for (size_t i = 0; i < stateByTrack.size(); ++i) { 3205778822d86b0337407514b9372562b86edfa91cdAndreas Huber CodecState *state = &stateByTrack.editValueAt(i); 3215778822d86b0337407514b9372562b86edfa91cdAndreas Huber 322c95c2ddcdfc974f42408a377fbe2de51b94a8c94Andreas Huber CHECK_EQ((status_t)OK, state->mCodec->release()); 323bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber 324bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber if (state->mIsAudio) { 325bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber printf("track %d: %lld bytes received. %.2f KB/sec\n", 326bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber i, 327bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber state->mNumBytesDecoded, 328bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber state->mNumBytesDecoded * 1E6 / 1024 / elapsedTimeUs); 329bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber } else { 330bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber printf("track %d: %lld frames decoded, %.2f fps. %lld bytes " 331bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber "received. %.2f KB/sec\n", 332bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber i, 333bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber state->mNumBuffersDecoded, 334bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber state->mNumBuffersDecoded * 1E6 / elapsedTimeUs, 335bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber state->mNumBytesDecoded, 336bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber state->mNumBytesDecoded * 1E6 / 1024 / elapsedTimeUs); 337bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber } 3385778822d86b0337407514b9372562b86edfa91cdAndreas Huber } 3395778822d86b0337407514b9372562b86edfa91cdAndreas Huber 3405778822d86b0337407514b9372562b86edfa91cdAndreas Huber return 0; 3415778822d86b0337407514b9372562b86edfa91cdAndreas Huber} 3425778822d86b0337407514b9372562b86edfa91cdAndreas Huber 3435778822d86b0337407514b9372562b86edfa91cdAndreas Huberint main(int argc, char **argv) { 3445778822d86b0337407514b9372562b86edfa91cdAndreas Huber using namespace android; 3455778822d86b0337407514b9372562b86edfa91cdAndreas Huber 3465778822d86b0337407514b9372562b86edfa91cdAndreas Huber const char *me = argv[0]; 3475778822d86b0337407514b9372562b86edfa91cdAndreas Huber 3485778822d86b0337407514b9372562b86edfa91cdAndreas Huber bool useAudio = false; 3495778822d86b0337407514b9372562b86edfa91cdAndreas Huber bool useVideo = false; 3505778822d86b0337407514b9372562b86edfa91cdAndreas Huber bool playback = false; 351bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber bool useSurface = false; 3525778822d86b0337407514b9372562b86edfa91cdAndreas Huber 3535778822d86b0337407514b9372562b86edfa91cdAndreas Huber int res; 354ed3e3e046840d5bf1ca84a8c0cc097425e89d6d6Andreas Huber while ((res = getopt(argc, argv, "havpSD")) >= 0) { 3555778822d86b0337407514b9372562b86edfa91cdAndreas Huber switch (res) { 3565778822d86b0337407514b9372562b86edfa91cdAndreas Huber case 'a': 3575778822d86b0337407514b9372562b86edfa91cdAndreas Huber { 3585778822d86b0337407514b9372562b86edfa91cdAndreas Huber useAudio = true; 3595778822d86b0337407514b9372562b86edfa91cdAndreas Huber break; 3605778822d86b0337407514b9372562b86edfa91cdAndreas Huber } 3615778822d86b0337407514b9372562b86edfa91cdAndreas Huber 3625778822d86b0337407514b9372562b86edfa91cdAndreas Huber case 'v': 3635778822d86b0337407514b9372562b86edfa91cdAndreas Huber { 3645778822d86b0337407514b9372562b86edfa91cdAndreas Huber useVideo = true; 3655778822d86b0337407514b9372562b86edfa91cdAndreas Huber break; 3665778822d86b0337407514b9372562b86edfa91cdAndreas Huber } 3675778822d86b0337407514b9372562b86edfa91cdAndreas Huber 3685778822d86b0337407514b9372562b86edfa91cdAndreas Huber case 'p': 3695778822d86b0337407514b9372562b86edfa91cdAndreas Huber { 3705778822d86b0337407514b9372562b86edfa91cdAndreas Huber playback = true; 3715778822d86b0337407514b9372562b86edfa91cdAndreas Huber break; 3725778822d86b0337407514b9372562b86edfa91cdAndreas Huber } 3735778822d86b0337407514b9372562b86edfa91cdAndreas Huber 374bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber case 'S': 375bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber { 376bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber useSurface = true; 377bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber break; 378bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber } 379bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber 3805778822d86b0337407514b9372562b86edfa91cdAndreas Huber case '?': 3815778822d86b0337407514b9372562b86edfa91cdAndreas Huber case 'h': 3825778822d86b0337407514b9372562b86edfa91cdAndreas Huber default: 3835778822d86b0337407514b9372562b86edfa91cdAndreas Huber { 3845778822d86b0337407514b9372562b86edfa91cdAndreas Huber usage(me); 3855778822d86b0337407514b9372562b86edfa91cdAndreas Huber } 3865778822d86b0337407514b9372562b86edfa91cdAndreas Huber } 3875778822d86b0337407514b9372562b86edfa91cdAndreas Huber } 3885778822d86b0337407514b9372562b86edfa91cdAndreas Huber 3895778822d86b0337407514b9372562b86edfa91cdAndreas Huber argc -= optind; 3905778822d86b0337407514b9372562b86edfa91cdAndreas Huber argv += optind; 3915778822d86b0337407514b9372562b86edfa91cdAndreas Huber 3925778822d86b0337407514b9372562b86edfa91cdAndreas Huber if (argc != 1) { 3935778822d86b0337407514b9372562b86edfa91cdAndreas Huber usage(me); 3945778822d86b0337407514b9372562b86edfa91cdAndreas Huber } 3955778822d86b0337407514b9372562b86edfa91cdAndreas Huber 3965778822d86b0337407514b9372562b86edfa91cdAndreas Huber if (!useAudio && !useVideo) { 3975778822d86b0337407514b9372562b86edfa91cdAndreas Huber useAudio = useVideo = true; 3985778822d86b0337407514b9372562b86edfa91cdAndreas Huber } 3995778822d86b0337407514b9372562b86edfa91cdAndreas Huber 4005778822d86b0337407514b9372562b86edfa91cdAndreas Huber ProcessState::self()->startThreadPool(); 4015778822d86b0337407514b9372562b86edfa91cdAndreas Huber 4025778822d86b0337407514b9372562b86edfa91cdAndreas Huber DataSource::RegisterDefaultSniffers(); 4035778822d86b0337407514b9372562b86edfa91cdAndreas Huber 4045778822d86b0337407514b9372562b86edfa91cdAndreas Huber sp<ALooper> looper = new ALooper; 4055778822d86b0337407514b9372562b86edfa91cdAndreas Huber looper->start(); 4065778822d86b0337407514b9372562b86edfa91cdAndreas Huber 407bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber sp<SurfaceComposerClient> composerClient; 408bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber sp<SurfaceControl> control; 409bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber sp<Surface> surface; 410bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber 411bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber if (playback || (useSurface && useVideo)) { 412bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber composerClient = new SurfaceComposerClient; 4135778822d86b0337407514b9372562b86edfa91cdAndreas Huber CHECK_EQ(composerClient->initCheck(), (status_t)OK); 4145778822d86b0337407514b9372562b86edfa91cdAndreas Huber 4155778822d86b0337407514b9372562b86edfa91cdAndreas Huber ssize_t displayWidth = composerClient->getDisplayWidth(0); 4165778822d86b0337407514b9372562b86edfa91cdAndreas Huber ssize_t displayHeight = composerClient->getDisplayHeight(0); 4175778822d86b0337407514b9372562b86edfa91cdAndreas Huber 4185778822d86b0337407514b9372562b86edfa91cdAndreas Huber ALOGV("display is %ld x %ld\n", displayWidth, displayHeight); 4195778822d86b0337407514b9372562b86edfa91cdAndreas Huber 420bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber control = composerClient->createSurface( 421bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber String8("A Surface"), 422bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber 0, 423bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber displayWidth, 424bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber displayHeight, 425bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber PIXEL_FORMAT_RGB_565, 426bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber 0); 4275778822d86b0337407514b9372562b86edfa91cdAndreas Huber 4285778822d86b0337407514b9372562b86edfa91cdAndreas Huber CHECK(control != NULL); 4295778822d86b0337407514b9372562b86edfa91cdAndreas Huber CHECK(control->isValid()); 4305778822d86b0337407514b9372562b86edfa91cdAndreas Huber 4315778822d86b0337407514b9372562b86edfa91cdAndreas Huber SurfaceComposerClient::openGlobalTransaction(); 4325778822d86b0337407514b9372562b86edfa91cdAndreas Huber CHECK_EQ(control->setLayer(INT_MAX), (status_t)OK); 4335778822d86b0337407514b9372562b86edfa91cdAndreas Huber CHECK_EQ(control->show(), (status_t)OK); 4345778822d86b0337407514b9372562b86edfa91cdAndreas Huber SurfaceComposerClient::closeGlobalTransaction(); 4355778822d86b0337407514b9372562b86edfa91cdAndreas Huber 436bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber surface = control->getSurface(); 4375778822d86b0337407514b9372562b86edfa91cdAndreas Huber CHECK(surface != NULL); 438bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber } 4395778822d86b0337407514b9372562b86edfa91cdAndreas Huber 440bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber if (playback) { 4415778822d86b0337407514b9372562b86edfa91cdAndreas Huber sp<SimplePlayer> player = new SimplePlayer; 4425778822d86b0337407514b9372562b86edfa91cdAndreas Huber looper->registerHandler(player); 4435778822d86b0337407514b9372562b86edfa91cdAndreas Huber 4445778822d86b0337407514b9372562b86edfa91cdAndreas Huber player->setDataSource(argv[0]); 4455778822d86b0337407514b9372562b86edfa91cdAndreas Huber player->setSurface(surface->getSurfaceTexture()); 4465778822d86b0337407514b9372562b86edfa91cdAndreas Huber player->start(); 447e98f8c04faf27df3b1829d336299ad51dad569cfAndreas Huber sleep(60); 4485778822d86b0337407514b9372562b86edfa91cdAndreas Huber player->stop(); 4495778822d86b0337407514b9372562b86edfa91cdAndreas Huber player->reset(); 450bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber } else { 451f69e53033f23f9f70fcdb28a3c2e650de0147459Andreas Huber decode(looper, argv[0], useAudio, useVideo, surface); 452bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber } 4535778822d86b0337407514b9372562b86edfa91cdAndreas Huber 454bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber if (playback || (useSurface && useVideo)) { 4555778822d86b0337407514b9372562b86edfa91cdAndreas Huber composerClient->dispose(); 4565778822d86b0337407514b9372562b86edfa91cdAndreas Huber } 4575778822d86b0337407514b9372562b86edfa91cdAndreas Huber 4585778822d86b0337407514b9372562b86edfa91cdAndreas Huber looper->stop(); 4595778822d86b0337407514b9372562b86edfa91cdAndreas Huber 4605778822d86b0337407514b9372562b86edfa91cdAndreas Huber return 0; 4615778822d86b0337407514b9372562b86edfa91cdAndreas Huber} 462