codec.cpp revision 5aaeb0d64fc98f9b019a4378eb39f0ee49ee6ec4
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 235778822d86b0337407514b9372562b86edfa91cdAndreas Huber#include <binder/ProcessState.h> 245778822d86b0337407514b9372562b86edfa91cdAndreas Huber 255778822d86b0337407514b9372562b86edfa91cdAndreas Huber#include <media/stagefright/foundation/ABuffer.h> 265778822d86b0337407514b9372562b86edfa91cdAndreas Huber#include <media/stagefright/foundation/ADebug.h> 275778822d86b0337407514b9372562b86edfa91cdAndreas Huber#include <media/stagefright/foundation/ALooper.h> 285778822d86b0337407514b9372562b86edfa91cdAndreas Huber#include <media/stagefright/foundation/AMessage.h> 295778822d86b0337407514b9372562b86edfa91cdAndreas Huber#include <media/stagefright/DataSource.h> 305778822d86b0337407514b9372562b86edfa91cdAndreas Huber#include <media/stagefright/MediaCodec.h> 315778822d86b0337407514b9372562b86edfa91cdAndreas Huber#include <media/stagefright/MediaDefs.h> 325778822d86b0337407514b9372562b86edfa91cdAndreas Huber#include <media/stagefright/NuMediaExtractor.h> 33df712ea86e6350f7005a02ab0e1c60c28a343ed0Mathias Agopian#include <gui/SurfaceComposerClient.h> 345778822d86b0337407514b9372562b86edfa91cdAndreas Huber 355778822d86b0337407514b9372562b86edfa91cdAndreas Huberstatic void usage(const char *me) { 365778822d86b0337407514b9372562b86edfa91cdAndreas Huber fprintf(stderr, "usage: %s [-a] use audio\n" 375778822d86b0337407514b9372562b86edfa91cdAndreas Huber "\t\t[-v] use video\n" 38bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber "\t\t[-p] playback\n" 39bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber "\t\t[-S] allocate buffers from a surface\n", me); 405778822d86b0337407514b9372562b86edfa91cdAndreas Huber 415778822d86b0337407514b9372562b86edfa91cdAndreas Huber exit(1); 425778822d86b0337407514b9372562b86edfa91cdAndreas Huber} 435778822d86b0337407514b9372562b86edfa91cdAndreas Huber 445778822d86b0337407514b9372562b86edfa91cdAndreas Hubernamespace android { 455778822d86b0337407514b9372562b86edfa91cdAndreas Huber 465778822d86b0337407514b9372562b86edfa91cdAndreas Huberstruct CodecState { 475778822d86b0337407514b9372562b86edfa91cdAndreas Huber sp<MediaCodec> mCodec; 485778822d86b0337407514b9372562b86edfa91cdAndreas Huber Vector<sp<ABuffer> > mCSD; 495778822d86b0337407514b9372562b86edfa91cdAndreas Huber size_t mCSDIndex; 505778822d86b0337407514b9372562b86edfa91cdAndreas Huber Vector<sp<ABuffer> > mInBuffers; 515778822d86b0337407514b9372562b86edfa91cdAndreas Huber Vector<sp<ABuffer> > mOutBuffers; 525aaeb0d64fc98f9b019a4378eb39f0ee49ee6ec4Andreas Huber bool mSignalledInputEOS; 535778822d86b0337407514b9372562b86edfa91cdAndreas Huber bool mSawOutputEOS; 54bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber int64_t mNumBuffersDecoded; 55bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber int64_t mNumBytesDecoded; 56bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber bool mIsAudio; 575778822d86b0337407514b9372562b86edfa91cdAndreas Huber}; 585778822d86b0337407514b9372562b86edfa91cdAndreas Huber 595778822d86b0337407514b9372562b86edfa91cdAndreas Huber} // namespace android 605778822d86b0337407514b9372562b86edfa91cdAndreas Huber 615778822d86b0337407514b9372562b86edfa91cdAndreas Huberstatic int decode( 625778822d86b0337407514b9372562b86edfa91cdAndreas Huber const android::sp<android::ALooper> &looper, 635778822d86b0337407514b9372562b86edfa91cdAndreas Huber const char *path, 645778822d86b0337407514b9372562b86edfa91cdAndreas Huber bool useAudio, 65bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber bool useVideo, 66bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber const android::sp<android::Surface> &surface) { 675778822d86b0337407514b9372562b86edfa91cdAndreas Huber using namespace android; 685778822d86b0337407514b9372562b86edfa91cdAndreas Huber 69bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber static int64_t kTimeout = 500ll; 70bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber 715778822d86b0337407514b9372562b86edfa91cdAndreas Huber sp<NuMediaExtractor> extractor = new NuMediaExtractor; 725778822d86b0337407514b9372562b86edfa91cdAndreas Huber if (extractor->setDataSource(path) != OK) { 735778822d86b0337407514b9372562b86edfa91cdAndreas Huber fprintf(stderr, "unable to instantiate extractor.\n"); 745778822d86b0337407514b9372562b86edfa91cdAndreas Huber return 1; 755778822d86b0337407514b9372562b86edfa91cdAndreas Huber } 765778822d86b0337407514b9372562b86edfa91cdAndreas Huber 775778822d86b0337407514b9372562b86edfa91cdAndreas Huber KeyedVector<size_t, CodecState> stateByTrack; 785778822d86b0337407514b9372562b86edfa91cdAndreas Huber 795778822d86b0337407514b9372562b86edfa91cdAndreas Huber bool haveAudio = false; 805778822d86b0337407514b9372562b86edfa91cdAndreas Huber bool haveVideo = false; 815778822d86b0337407514b9372562b86edfa91cdAndreas Huber for (size_t i = 0; i < extractor->countTracks(); ++i) { 825778822d86b0337407514b9372562b86edfa91cdAndreas Huber sp<AMessage> format; 835778822d86b0337407514b9372562b86edfa91cdAndreas Huber status_t err = extractor->getTrackFormat(i, &format); 845778822d86b0337407514b9372562b86edfa91cdAndreas Huber CHECK_EQ(err, (status_t)OK); 855778822d86b0337407514b9372562b86edfa91cdAndreas Huber 865778822d86b0337407514b9372562b86edfa91cdAndreas Huber AString mime; 875778822d86b0337407514b9372562b86edfa91cdAndreas Huber CHECK(format->findString("mime", &mime)); 885778822d86b0337407514b9372562b86edfa91cdAndreas Huber 89bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber bool isAudio = !strncasecmp(mime.c_str(), "audio/", 6); 90bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber bool isVideo = !strncasecmp(mime.c_str(), "video/", 6); 91bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber 92bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber if (useAudio && !haveAudio && isAudio) { 935778822d86b0337407514b9372562b86edfa91cdAndreas Huber haveAudio = true; 94bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber } else if (useVideo && !haveVideo && isVideo) { 955778822d86b0337407514b9372562b86edfa91cdAndreas Huber haveVideo = true; 965778822d86b0337407514b9372562b86edfa91cdAndreas Huber } else { 975778822d86b0337407514b9372562b86edfa91cdAndreas Huber continue; 985778822d86b0337407514b9372562b86edfa91cdAndreas Huber } 995778822d86b0337407514b9372562b86edfa91cdAndreas Huber 1005778822d86b0337407514b9372562b86edfa91cdAndreas Huber ALOGV("selecting track %d", i); 1015778822d86b0337407514b9372562b86edfa91cdAndreas Huber 1025778822d86b0337407514b9372562b86edfa91cdAndreas Huber err = extractor->selectTrack(i); 1035778822d86b0337407514b9372562b86edfa91cdAndreas Huber CHECK_EQ(err, (status_t)OK); 1045778822d86b0337407514b9372562b86edfa91cdAndreas Huber 1055778822d86b0337407514b9372562b86edfa91cdAndreas Huber CodecState *state = 1065778822d86b0337407514b9372562b86edfa91cdAndreas Huber &stateByTrack.editValueAt(stateByTrack.add(i, CodecState())); 1075778822d86b0337407514b9372562b86edfa91cdAndreas Huber 108bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber state->mNumBytesDecoded = 0; 109bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber state->mNumBuffersDecoded = 0; 110bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber state->mIsAudio = isAudio; 111bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber 1125778822d86b0337407514b9372562b86edfa91cdAndreas Huber state->mCodec = MediaCodec::CreateByType( 1135778822d86b0337407514b9372562b86edfa91cdAndreas Huber looper, mime.c_str(), false /* encoder */); 1145778822d86b0337407514b9372562b86edfa91cdAndreas Huber 1155778822d86b0337407514b9372562b86edfa91cdAndreas Huber CHECK(state->mCodec != NULL); 1165778822d86b0337407514b9372562b86edfa91cdAndreas Huber 1175778822d86b0337407514b9372562b86edfa91cdAndreas Huber err = state->mCodec->configure( 118bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber format, isVideo ? surface : NULL, 0 /* flags */); 1195778822d86b0337407514b9372562b86edfa91cdAndreas Huber 1205778822d86b0337407514b9372562b86edfa91cdAndreas Huber CHECK_EQ(err, (status_t)OK); 1215778822d86b0337407514b9372562b86edfa91cdAndreas Huber 1225778822d86b0337407514b9372562b86edfa91cdAndreas Huber size_t j = 0; 123bf6c85a013fb14960bac147c1ffd0a02a8d5f148Andreas Huber sp<ABuffer> buffer; 124bf6c85a013fb14960bac147c1ffd0a02a8d5f148Andreas Huber while (format->findBuffer(StringPrintf("csd-%d", j).c_str(), &buffer)) { 1255778822d86b0337407514b9372562b86edfa91cdAndreas Huber state->mCSD.push_back(buffer); 1265778822d86b0337407514b9372562b86edfa91cdAndreas Huber 1275778822d86b0337407514b9372562b86edfa91cdAndreas Huber ++j; 1285778822d86b0337407514b9372562b86edfa91cdAndreas Huber } 1295778822d86b0337407514b9372562b86edfa91cdAndreas Huber 1305778822d86b0337407514b9372562b86edfa91cdAndreas Huber state->mCSDIndex = 0; 1315aaeb0d64fc98f9b019a4378eb39f0ee49ee6ec4Andreas Huber state->mSignalledInputEOS = false; 1325778822d86b0337407514b9372562b86edfa91cdAndreas Huber state->mSawOutputEOS = false; 1335778822d86b0337407514b9372562b86edfa91cdAndreas Huber 1345778822d86b0337407514b9372562b86edfa91cdAndreas Huber ALOGV("got %d pieces of codec specific data.", state->mCSD.size()); 1355778822d86b0337407514b9372562b86edfa91cdAndreas Huber } 1365778822d86b0337407514b9372562b86edfa91cdAndreas Huber 1375778822d86b0337407514b9372562b86edfa91cdAndreas Huber CHECK(!stateByTrack.isEmpty()); 1385778822d86b0337407514b9372562b86edfa91cdAndreas Huber 139bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber int64_t startTimeUs = ALooper::GetNowUs(); 140bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber 1415778822d86b0337407514b9372562b86edfa91cdAndreas Huber for (size_t i = 0; i < stateByTrack.size(); ++i) { 1425778822d86b0337407514b9372562b86edfa91cdAndreas Huber CodecState *state = &stateByTrack.editValueAt(i); 1435778822d86b0337407514b9372562b86edfa91cdAndreas Huber 1445778822d86b0337407514b9372562b86edfa91cdAndreas Huber sp<MediaCodec> codec = state->mCodec; 1455778822d86b0337407514b9372562b86edfa91cdAndreas Huber 1465778822d86b0337407514b9372562b86edfa91cdAndreas Huber CHECK_EQ((status_t)OK, codec->start()); 1475778822d86b0337407514b9372562b86edfa91cdAndreas Huber 1485778822d86b0337407514b9372562b86edfa91cdAndreas Huber CHECK_EQ((status_t)OK, codec->getInputBuffers(&state->mInBuffers)); 1495778822d86b0337407514b9372562b86edfa91cdAndreas Huber CHECK_EQ((status_t)OK, codec->getOutputBuffers(&state->mOutBuffers)); 1505778822d86b0337407514b9372562b86edfa91cdAndreas Huber 1515778822d86b0337407514b9372562b86edfa91cdAndreas Huber ALOGV("got %d input and %d output buffers", 1525778822d86b0337407514b9372562b86edfa91cdAndreas Huber state->mInBuffers.size(), state->mOutBuffers.size()); 1535778822d86b0337407514b9372562b86edfa91cdAndreas Huber 1545778822d86b0337407514b9372562b86edfa91cdAndreas Huber while (state->mCSDIndex < state->mCSD.size()) { 1555778822d86b0337407514b9372562b86edfa91cdAndreas Huber size_t index; 156bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber status_t err = codec->dequeueInputBuffer(&index, -1ll); 1575778822d86b0337407514b9372562b86edfa91cdAndreas Huber CHECK_EQ(err, (status_t)OK); 1585778822d86b0337407514b9372562b86edfa91cdAndreas Huber 1595778822d86b0337407514b9372562b86edfa91cdAndreas Huber const sp<ABuffer> &srcBuffer = 1605778822d86b0337407514b9372562b86edfa91cdAndreas Huber state->mCSD.itemAt(state->mCSDIndex++); 1615778822d86b0337407514b9372562b86edfa91cdAndreas Huber 1625778822d86b0337407514b9372562b86edfa91cdAndreas Huber const sp<ABuffer> &buffer = state->mInBuffers.itemAt(index); 1635778822d86b0337407514b9372562b86edfa91cdAndreas Huber 1645778822d86b0337407514b9372562b86edfa91cdAndreas Huber memcpy(buffer->data(), srcBuffer->data(), srcBuffer->size()); 1655778822d86b0337407514b9372562b86edfa91cdAndreas Huber 1665778822d86b0337407514b9372562b86edfa91cdAndreas Huber err = codec->queueInputBuffer( 1675778822d86b0337407514b9372562b86edfa91cdAndreas Huber index, 1685778822d86b0337407514b9372562b86edfa91cdAndreas Huber 0 /* offset */, 1695778822d86b0337407514b9372562b86edfa91cdAndreas Huber srcBuffer->size(), 1705778822d86b0337407514b9372562b86edfa91cdAndreas Huber 0ll /* timeUs */, 1715778822d86b0337407514b9372562b86edfa91cdAndreas Huber MediaCodec::BUFFER_FLAG_CODECCONFIG); 1725778822d86b0337407514b9372562b86edfa91cdAndreas Huber 1735778822d86b0337407514b9372562b86edfa91cdAndreas Huber CHECK_EQ(err, (status_t)OK); 1745778822d86b0337407514b9372562b86edfa91cdAndreas Huber } 1755778822d86b0337407514b9372562b86edfa91cdAndreas Huber } 1765778822d86b0337407514b9372562b86edfa91cdAndreas Huber 1775778822d86b0337407514b9372562b86edfa91cdAndreas Huber bool sawInputEOS = false; 1785778822d86b0337407514b9372562b86edfa91cdAndreas Huber 1795778822d86b0337407514b9372562b86edfa91cdAndreas Huber for (;;) { 1805778822d86b0337407514b9372562b86edfa91cdAndreas Huber if (!sawInputEOS) { 1815778822d86b0337407514b9372562b86edfa91cdAndreas Huber size_t trackIndex; 1825778822d86b0337407514b9372562b86edfa91cdAndreas Huber status_t err = extractor->getSampleTrackIndex(&trackIndex); 1835778822d86b0337407514b9372562b86edfa91cdAndreas Huber 1845778822d86b0337407514b9372562b86edfa91cdAndreas Huber if (err != OK) { 1855aaeb0d64fc98f9b019a4378eb39f0ee49ee6ec4Andreas Huber ALOGV("saw input eos"); 1865778822d86b0337407514b9372562b86edfa91cdAndreas Huber sawInputEOS = true; 1875778822d86b0337407514b9372562b86edfa91cdAndreas Huber } else { 1885778822d86b0337407514b9372562b86edfa91cdAndreas Huber CodecState *state = &stateByTrack.editValueFor(trackIndex); 1895778822d86b0337407514b9372562b86edfa91cdAndreas Huber 1905778822d86b0337407514b9372562b86edfa91cdAndreas Huber size_t index; 191bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber err = state->mCodec->dequeueInputBuffer(&index, kTimeout); 1925778822d86b0337407514b9372562b86edfa91cdAndreas Huber 1935778822d86b0337407514b9372562b86edfa91cdAndreas Huber if (err == OK) { 1945778822d86b0337407514b9372562b86edfa91cdAndreas Huber ALOGV("filling input buffer %d", index); 1955778822d86b0337407514b9372562b86edfa91cdAndreas Huber 1965778822d86b0337407514b9372562b86edfa91cdAndreas Huber const sp<ABuffer> &buffer = state->mInBuffers.itemAt(index); 1975778822d86b0337407514b9372562b86edfa91cdAndreas Huber 1985778822d86b0337407514b9372562b86edfa91cdAndreas Huber err = extractor->readSampleData(buffer); 1995778822d86b0337407514b9372562b86edfa91cdAndreas Huber CHECK_EQ(err, (status_t)OK); 2005778822d86b0337407514b9372562b86edfa91cdAndreas Huber 2015778822d86b0337407514b9372562b86edfa91cdAndreas Huber int64_t timeUs; 2025778822d86b0337407514b9372562b86edfa91cdAndreas Huber err = extractor->getSampleTime(&timeUs); 2035778822d86b0337407514b9372562b86edfa91cdAndreas Huber CHECK_EQ(err, (status_t)OK); 2045778822d86b0337407514b9372562b86edfa91cdAndreas Huber 2055778822d86b0337407514b9372562b86edfa91cdAndreas Huber err = state->mCodec->queueInputBuffer( 2065778822d86b0337407514b9372562b86edfa91cdAndreas Huber index, 2075778822d86b0337407514b9372562b86edfa91cdAndreas Huber 0 /* offset */, 2085778822d86b0337407514b9372562b86edfa91cdAndreas Huber buffer->size(), 2095778822d86b0337407514b9372562b86edfa91cdAndreas Huber timeUs, 2105778822d86b0337407514b9372562b86edfa91cdAndreas Huber 0 /* flags */); 2115778822d86b0337407514b9372562b86edfa91cdAndreas Huber 2125778822d86b0337407514b9372562b86edfa91cdAndreas Huber CHECK_EQ(err, (status_t)OK); 2135778822d86b0337407514b9372562b86edfa91cdAndreas Huber 2145778822d86b0337407514b9372562b86edfa91cdAndreas Huber extractor->advance(); 2155778822d86b0337407514b9372562b86edfa91cdAndreas Huber } else { 2165778822d86b0337407514b9372562b86edfa91cdAndreas Huber CHECK_EQ(err, -EAGAIN); 2175778822d86b0337407514b9372562b86edfa91cdAndreas Huber } 2185778822d86b0337407514b9372562b86edfa91cdAndreas Huber } 2195aaeb0d64fc98f9b019a4378eb39f0ee49ee6ec4Andreas Huber } else { 2205aaeb0d64fc98f9b019a4378eb39f0ee49ee6ec4Andreas Huber for (size_t i = 0; i < stateByTrack.size(); ++i) { 2215aaeb0d64fc98f9b019a4378eb39f0ee49ee6ec4Andreas Huber CodecState *state = &stateByTrack.editValueAt(i); 2225aaeb0d64fc98f9b019a4378eb39f0ee49ee6ec4Andreas Huber 2235aaeb0d64fc98f9b019a4378eb39f0ee49ee6ec4Andreas Huber if (!state->mSignalledInputEOS) { 2245aaeb0d64fc98f9b019a4378eb39f0ee49ee6ec4Andreas Huber size_t index; 2255aaeb0d64fc98f9b019a4378eb39f0ee49ee6ec4Andreas Huber status_t err = 2265aaeb0d64fc98f9b019a4378eb39f0ee49ee6ec4Andreas Huber state->mCodec->dequeueInputBuffer(&index, kTimeout); 2275aaeb0d64fc98f9b019a4378eb39f0ee49ee6ec4Andreas Huber 2285aaeb0d64fc98f9b019a4378eb39f0ee49ee6ec4Andreas Huber if (err == OK) { 2295aaeb0d64fc98f9b019a4378eb39f0ee49ee6ec4Andreas Huber ALOGV("signalling input EOS on track %d", i); 2305aaeb0d64fc98f9b019a4378eb39f0ee49ee6ec4Andreas Huber 2315aaeb0d64fc98f9b019a4378eb39f0ee49ee6ec4Andreas Huber err = state->mCodec->queueInputBuffer( 2325aaeb0d64fc98f9b019a4378eb39f0ee49ee6ec4Andreas Huber index, 2335aaeb0d64fc98f9b019a4378eb39f0ee49ee6ec4Andreas Huber 0 /* offset */, 2345aaeb0d64fc98f9b019a4378eb39f0ee49ee6ec4Andreas Huber 0 /* size */, 2355aaeb0d64fc98f9b019a4378eb39f0ee49ee6ec4Andreas Huber 0ll /* timeUs */, 2365aaeb0d64fc98f9b019a4378eb39f0ee49ee6ec4Andreas Huber MediaCodec::BUFFER_FLAG_EOS); 2375aaeb0d64fc98f9b019a4378eb39f0ee49ee6ec4Andreas Huber 2385aaeb0d64fc98f9b019a4378eb39f0ee49ee6ec4Andreas Huber CHECK_EQ(err, (status_t)OK); 2395aaeb0d64fc98f9b019a4378eb39f0ee49ee6ec4Andreas Huber 2405aaeb0d64fc98f9b019a4378eb39f0ee49ee6ec4Andreas Huber state->mSignalledInputEOS = true; 2415aaeb0d64fc98f9b019a4378eb39f0ee49ee6ec4Andreas Huber } else { 2425aaeb0d64fc98f9b019a4378eb39f0ee49ee6ec4Andreas Huber CHECK_EQ(err, -EAGAIN); 2435aaeb0d64fc98f9b019a4378eb39f0ee49ee6ec4Andreas Huber } 2445aaeb0d64fc98f9b019a4378eb39f0ee49ee6ec4Andreas Huber } 2455aaeb0d64fc98f9b019a4378eb39f0ee49ee6ec4Andreas Huber } 2465778822d86b0337407514b9372562b86edfa91cdAndreas Huber } 2475778822d86b0337407514b9372562b86edfa91cdAndreas Huber 2485778822d86b0337407514b9372562b86edfa91cdAndreas Huber bool sawOutputEOSOnAllTracks = true; 2495778822d86b0337407514b9372562b86edfa91cdAndreas Huber for (size_t i = 0; i < stateByTrack.size(); ++i) { 2505778822d86b0337407514b9372562b86edfa91cdAndreas Huber CodecState *state = &stateByTrack.editValueAt(i); 2515778822d86b0337407514b9372562b86edfa91cdAndreas Huber if (!state->mSawOutputEOS) { 2525778822d86b0337407514b9372562b86edfa91cdAndreas Huber sawOutputEOSOnAllTracks = false; 2535778822d86b0337407514b9372562b86edfa91cdAndreas Huber break; 2545778822d86b0337407514b9372562b86edfa91cdAndreas Huber } 2555778822d86b0337407514b9372562b86edfa91cdAndreas Huber } 2565778822d86b0337407514b9372562b86edfa91cdAndreas Huber 2575778822d86b0337407514b9372562b86edfa91cdAndreas Huber if (sawOutputEOSOnAllTracks) { 2585778822d86b0337407514b9372562b86edfa91cdAndreas Huber break; 2595778822d86b0337407514b9372562b86edfa91cdAndreas Huber } 2605778822d86b0337407514b9372562b86edfa91cdAndreas Huber 2615778822d86b0337407514b9372562b86edfa91cdAndreas Huber for (size_t i = 0; i < stateByTrack.size(); ++i) { 2625778822d86b0337407514b9372562b86edfa91cdAndreas Huber CodecState *state = &stateByTrack.editValueAt(i); 2635778822d86b0337407514b9372562b86edfa91cdAndreas Huber 2645778822d86b0337407514b9372562b86edfa91cdAndreas Huber if (state->mSawOutputEOS) { 2655778822d86b0337407514b9372562b86edfa91cdAndreas Huber continue; 2665778822d86b0337407514b9372562b86edfa91cdAndreas Huber } 2675778822d86b0337407514b9372562b86edfa91cdAndreas Huber 2685778822d86b0337407514b9372562b86edfa91cdAndreas Huber size_t index; 2695778822d86b0337407514b9372562b86edfa91cdAndreas Huber size_t offset; 2705778822d86b0337407514b9372562b86edfa91cdAndreas Huber size_t size; 2715778822d86b0337407514b9372562b86edfa91cdAndreas Huber int64_t presentationTimeUs; 2725778822d86b0337407514b9372562b86edfa91cdAndreas Huber uint32_t flags; 2735778822d86b0337407514b9372562b86edfa91cdAndreas Huber status_t err = state->mCodec->dequeueOutputBuffer( 2745778822d86b0337407514b9372562b86edfa91cdAndreas Huber &index, &offset, &size, &presentationTimeUs, &flags, 275bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber kTimeout); 2765778822d86b0337407514b9372562b86edfa91cdAndreas Huber 2775778822d86b0337407514b9372562b86edfa91cdAndreas Huber if (err == OK) { 2785778822d86b0337407514b9372562b86edfa91cdAndreas Huber ALOGV("draining output buffer %d, time = %lld us", 2795778822d86b0337407514b9372562b86edfa91cdAndreas Huber index, presentationTimeUs); 2805778822d86b0337407514b9372562b86edfa91cdAndreas Huber 281bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber ++state->mNumBuffersDecoded; 282bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber state->mNumBytesDecoded += size; 283bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber 2845778822d86b0337407514b9372562b86edfa91cdAndreas Huber err = state->mCodec->releaseOutputBuffer(index); 2855778822d86b0337407514b9372562b86edfa91cdAndreas Huber CHECK_EQ(err, (status_t)OK); 2865778822d86b0337407514b9372562b86edfa91cdAndreas Huber 2875778822d86b0337407514b9372562b86edfa91cdAndreas Huber if (flags & MediaCodec::BUFFER_FLAG_EOS) { 2885778822d86b0337407514b9372562b86edfa91cdAndreas Huber ALOGV("reached EOS on output."); 2895778822d86b0337407514b9372562b86edfa91cdAndreas Huber 2905778822d86b0337407514b9372562b86edfa91cdAndreas Huber state->mSawOutputEOS = true; 2915778822d86b0337407514b9372562b86edfa91cdAndreas Huber } 2925778822d86b0337407514b9372562b86edfa91cdAndreas Huber } else if (err == INFO_OUTPUT_BUFFERS_CHANGED) { 2935778822d86b0337407514b9372562b86edfa91cdAndreas Huber ALOGV("INFO_OUTPUT_BUFFERS_CHANGED"); 2945778822d86b0337407514b9372562b86edfa91cdAndreas Huber CHECK_EQ((status_t)OK, 2955778822d86b0337407514b9372562b86edfa91cdAndreas Huber state->mCodec->getOutputBuffers(&state->mOutBuffers)); 2965778822d86b0337407514b9372562b86edfa91cdAndreas Huber 2975778822d86b0337407514b9372562b86edfa91cdAndreas Huber ALOGV("got %d output buffers", state->mOutBuffers.size()); 2985778822d86b0337407514b9372562b86edfa91cdAndreas Huber } else if (err == INFO_FORMAT_CHANGED) { 2995778822d86b0337407514b9372562b86edfa91cdAndreas Huber sp<AMessage> format; 3005778822d86b0337407514b9372562b86edfa91cdAndreas Huber CHECK_EQ((status_t)OK, state->mCodec->getOutputFormat(&format)); 3015778822d86b0337407514b9372562b86edfa91cdAndreas Huber 3025778822d86b0337407514b9372562b86edfa91cdAndreas Huber ALOGV("INFO_FORMAT_CHANGED: %s", format->debugString().c_str()); 3035778822d86b0337407514b9372562b86edfa91cdAndreas Huber } else { 3045778822d86b0337407514b9372562b86edfa91cdAndreas Huber CHECK_EQ(err, -EAGAIN); 3055778822d86b0337407514b9372562b86edfa91cdAndreas Huber } 3065778822d86b0337407514b9372562b86edfa91cdAndreas Huber } 3075778822d86b0337407514b9372562b86edfa91cdAndreas Huber } 3085778822d86b0337407514b9372562b86edfa91cdAndreas Huber 309bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber int64_t elapsedTimeUs = ALooper::GetNowUs() - startTimeUs; 310bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber 3115778822d86b0337407514b9372562b86edfa91cdAndreas Huber for (size_t i = 0; i < stateByTrack.size(); ++i) { 3125778822d86b0337407514b9372562b86edfa91cdAndreas Huber CodecState *state = &stateByTrack.editValueAt(i); 3135778822d86b0337407514b9372562b86edfa91cdAndreas Huber 314c95c2ddcdfc974f42408a377fbe2de51b94a8c94Andreas Huber CHECK_EQ((status_t)OK, state->mCodec->release()); 315bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber 316bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber if (state->mIsAudio) { 317bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber printf("track %d: %lld bytes received. %.2f KB/sec\n", 318bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber i, 319bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber state->mNumBytesDecoded, 320bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber state->mNumBytesDecoded * 1E6 / 1024 / elapsedTimeUs); 321bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber } else { 322bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber printf("track %d: %lld frames decoded, %.2f fps. %lld bytes " 323bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber "received. %.2f KB/sec\n", 324bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber i, 325bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber state->mNumBuffersDecoded, 326bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber state->mNumBuffersDecoded * 1E6 / elapsedTimeUs, 327bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber state->mNumBytesDecoded, 328bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber state->mNumBytesDecoded * 1E6 / 1024 / elapsedTimeUs); 329bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber } 3305778822d86b0337407514b9372562b86edfa91cdAndreas Huber } 3315778822d86b0337407514b9372562b86edfa91cdAndreas Huber 3325778822d86b0337407514b9372562b86edfa91cdAndreas Huber return 0; 3335778822d86b0337407514b9372562b86edfa91cdAndreas Huber} 3345778822d86b0337407514b9372562b86edfa91cdAndreas Huber 3355778822d86b0337407514b9372562b86edfa91cdAndreas Huberint main(int argc, char **argv) { 3365778822d86b0337407514b9372562b86edfa91cdAndreas Huber using namespace android; 3375778822d86b0337407514b9372562b86edfa91cdAndreas Huber 3385778822d86b0337407514b9372562b86edfa91cdAndreas Huber const char *me = argv[0]; 3395778822d86b0337407514b9372562b86edfa91cdAndreas Huber 3405778822d86b0337407514b9372562b86edfa91cdAndreas Huber bool useAudio = false; 3415778822d86b0337407514b9372562b86edfa91cdAndreas Huber bool useVideo = false; 3425778822d86b0337407514b9372562b86edfa91cdAndreas Huber bool playback = false; 343bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber bool useSurface = false; 3445778822d86b0337407514b9372562b86edfa91cdAndreas Huber 3455778822d86b0337407514b9372562b86edfa91cdAndreas Huber int res; 346bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber while ((res = getopt(argc, argv, "havpS")) >= 0) { 3475778822d86b0337407514b9372562b86edfa91cdAndreas Huber switch (res) { 3485778822d86b0337407514b9372562b86edfa91cdAndreas Huber case 'a': 3495778822d86b0337407514b9372562b86edfa91cdAndreas Huber { 3505778822d86b0337407514b9372562b86edfa91cdAndreas Huber useAudio = true; 3515778822d86b0337407514b9372562b86edfa91cdAndreas Huber break; 3525778822d86b0337407514b9372562b86edfa91cdAndreas Huber } 3535778822d86b0337407514b9372562b86edfa91cdAndreas Huber 3545778822d86b0337407514b9372562b86edfa91cdAndreas Huber case 'v': 3555778822d86b0337407514b9372562b86edfa91cdAndreas Huber { 3565778822d86b0337407514b9372562b86edfa91cdAndreas Huber useVideo = true; 3575778822d86b0337407514b9372562b86edfa91cdAndreas Huber break; 3585778822d86b0337407514b9372562b86edfa91cdAndreas Huber } 3595778822d86b0337407514b9372562b86edfa91cdAndreas Huber 3605778822d86b0337407514b9372562b86edfa91cdAndreas Huber case 'p': 3615778822d86b0337407514b9372562b86edfa91cdAndreas Huber { 3625778822d86b0337407514b9372562b86edfa91cdAndreas Huber playback = true; 3635778822d86b0337407514b9372562b86edfa91cdAndreas Huber break; 3645778822d86b0337407514b9372562b86edfa91cdAndreas Huber } 3655778822d86b0337407514b9372562b86edfa91cdAndreas Huber 366bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber case 'S': 367bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber { 368bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber useSurface = true; 369bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber break; 370bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber } 371bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber 3725778822d86b0337407514b9372562b86edfa91cdAndreas Huber case '?': 3735778822d86b0337407514b9372562b86edfa91cdAndreas Huber case 'h': 3745778822d86b0337407514b9372562b86edfa91cdAndreas Huber default: 3755778822d86b0337407514b9372562b86edfa91cdAndreas Huber { 3765778822d86b0337407514b9372562b86edfa91cdAndreas Huber usage(me); 3775778822d86b0337407514b9372562b86edfa91cdAndreas Huber } 3785778822d86b0337407514b9372562b86edfa91cdAndreas Huber } 3795778822d86b0337407514b9372562b86edfa91cdAndreas Huber } 3805778822d86b0337407514b9372562b86edfa91cdAndreas Huber 3815778822d86b0337407514b9372562b86edfa91cdAndreas Huber argc -= optind; 3825778822d86b0337407514b9372562b86edfa91cdAndreas Huber argv += optind; 3835778822d86b0337407514b9372562b86edfa91cdAndreas Huber 3845778822d86b0337407514b9372562b86edfa91cdAndreas Huber if (argc != 1) { 3855778822d86b0337407514b9372562b86edfa91cdAndreas Huber usage(me); 3865778822d86b0337407514b9372562b86edfa91cdAndreas Huber } 3875778822d86b0337407514b9372562b86edfa91cdAndreas Huber 3885778822d86b0337407514b9372562b86edfa91cdAndreas Huber if (!useAudio && !useVideo) { 3895778822d86b0337407514b9372562b86edfa91cdAndreas Huber useAudio = useVideo = true; 3905778822d86b0337407514b9372562b86edfa91cdAndreas Huber } 3915778822d86b0337407514b9372562b86edfa91cdAndreas Huber 3925778822d86b0337407514b9372562b86edfa91cdAndreas Huber ProcessState::self()->startThreadPool(); 3935778822d86b0337407514b9372562b86edfa91cdAndreas Huber 3945778822d86b0337407514b9372562b86edfa91cdAndreas Huber DataSource::RegisterDefaultSniffers(); 3955778822d86b0337407514b9372562b86edfa91cdAndreas Huber 3965778822d86b0337407514b9372562b86edfa91cdAndreas Huber sp<ALooper> looper = new ALooper; 3975778822d86b0337407514b9372562b86edfa91cdAndreas Huber looper->start(); 3985778822d86b0337407514b9372562b86edfa91cdAndreas Huber 399bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber sp<SurfaceComposerClient> composerClient; 400bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber sp<SurfaceControl> control; 401bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber sp<Surface> surface; 402bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber 403bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber if (playback || (useSurface && useVideo)) { 404bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber composerClient = new SurfaceComposerClient; 4055778822d86b0337407514b9372562b86edfa91cdAndreas Huber CHECK_EQ(composerClient->initCheck(), (status_t)OK); 4065778822d86b0337407514b9372562b86edfa91cdAndreas Huber 4075778822d86b0337407514b9372562b86edfa91cdAndreas Huber ssize_t displayWidth = composerClient->getDisplayWidth(0); 4085778822d86b0337407514b9372562b86edfa91cdAndreas Huber ssize_t displayHeight = composerClient->getDisplayHeight(0); 4095778822d86b0337407514b9372562b86edfa91cdAndreas Huber 4105778822d86b0337407514b9372562b86edfa91cdAndreas Huber ALOGV("display is %ld x %ld\n", displayWidth, displayHeight); 4115778822d86b0337407514b9372562b86edfa91cdAndreas Huber 412bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber control = composerClient->createSurface( 413bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber String8("A Surface"), 414bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber 0, 415bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber displayWidth, 416bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber displayHeight, 417bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber PIXEL_FORMAT_RGB_565, 418bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber 0); 4195778822d86b0337407514b9372562b86edfa91cdAndreas Huber 4205778822d86b0337407514b9372562b86edfa91cdAndreas Huber CHECK(control != NULL); 4215778822d86b0337407514b9372562b86edfa91cdAndreas Huber CHECK(control->isValid()); 4225778822d86b0337407514b9372562b86edfa91cdAndreas Huber 4235778822d86b0337407514b9372562b86edfa91cdAndreas Huber SurfaceComposerClient::openGlobalTransaction(); 4245778822d86b0337407514b9372562b86edfa91cdAndreas Huber CHECK_EQ(control->setLayer(INT_MAX), (status_t)OK); 4255778822d86b0337407514b9372562b86edfa91cdAndreas Huber CHECK_EQ(control->show(), (status_t)OK); 4265778822d86b0337407514b9372562b86edfa91cdAndreas Huber SurfaceComposerClient::closeGlobalTransaction(); 4275778822d86b0337407514b9372562b86edfa91cdAndreas Huber 428bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber surface = control->getSurface(); 4295778822d86b0337407514b9372562b86edfa91cdAndreas Huber CHECK(surface != NULL); 430bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber } 4315778822d86b0337407514b9372562b86edfa91cdAndreas Huber 432bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber if (playback) { 4335778822d86b0337407514b9372562b86edfa91cdAndreas Huber sp<SimplePlayer> player = new SimplePlayer; 4345778822d86b0337407514b9372562b86edfa91cdAndreas Huber looper->registerHandler(player); 4355778822d86b0337407514b9372562b86edfa91cdAndreas Huber 4365778822d86b0337407514b9372562b86edfa91cdAndreas Huber player->setDataSource(argv[0]); 4375778822d86b0337407514b9372562b86edfa91cdAndreas Huber player->setSurface(surface->getSurfaceTexture()); 4385778822d86b0337407514b9372562b86edfa91cdAndreas Huber player->start(); 439e98f8c04faf27df3b1829d336299ad51dad569cfAndreas Huber sleep(60); 4405778822d86b0337407514b9372562b86edfa91cdAndreas Huber player->stop(); 4415778822d86b0337407514b9372562b86edfa91cdAndreas Huber player->reset(); 442bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber } else { 443bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber decode(looper, argv[0], useAudio, useVideo, surface); 444bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber } 4455778822d86b0337407514b9372562b86edfa91cdAndreas Huber 446bae6f72d16a1cfc2122b4ce9b484c026ecd896b1Andreas Huber if (playback || (useSurface && useVideo)) { 4475778822d86b0337407514b9372562b86edfa91cdAndreas Huber composerClient->dispose(); 4485778822d86b0337407514b9372562b86edfa91cdAndreas Huber } 4495778822d86b0337407514b9372562b86edfa91cdAndreas Huber 4505778822d86b0337407514b9372562b86edfa91cdAndreas Huber looper->stop(); 4515778822d86b0337407514b9372562b86edfa91cdAndreas Huber 4525778822d86b0337407514b9372562b86edfa91cdAndreas Huber return 0; 4535778822d86b0337407514b9372562b86edfa91cdAndreas Huber} 454