ATSParser.cpp revision 32f3cefa373cd55e63deda36ca9d07c7fe22eaaf
1cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber/* 2cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber * Copyright (C) 2010 The Android Open Source Project 3cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber * 4cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber * Licensed under the Apache License, Version 2.0 (the "License"); 5cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber * you may not use this file except in compliance with the License. 6cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber * You may obtain a copy of the License at 7cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber * 8cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber * http://www.apache.org/licenses/LICENSE-2.0 9cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber * 10cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber * Unless required by applicable law or agreed to in writing, software 11cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber * distributed under the License is distributed on an "AS IS" BASIS, 12cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 13cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber * See the License for the specific language governing permissions and 14cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber * limitations under the License. 15cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber */ 16cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 176e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber//#define LOG_NDEBUG 0 186e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber#define LOG_TAG "ATSParser" 196e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber#include <utils/Log.h> 206e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber 21cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber#include "ATSParser.h" 22cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 23cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber#include "AnotherPacketSource.h" 246a63a939601645404fd98f58c19cc38ca818d99eAndreas Huber#include "ESQueue.h" 2585f12e9b9062402d6110df3f7099707912040edbAndreas Huber#include "include/avc_utils.h" 26cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 2785f12e9b9062402d6110df3f7099707912040edbAndreas Huber#include <media/stagefright/foundation/ABitReader.h> 28cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber#include <media/stagefright/foundation/ABuffer.h> 29cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber#include <media/stagefright/foundation/ADebug.h> 30cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber#include <media/stagefright/foundation/AMessage.h> 31cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber#include <media/stagefright/foundation/hexdump.h> 32cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber#include <media/stagefright/MediaDefs.h> 33cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber#include <media/stagefright/MediaErrors.h> 34cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber#include <media/stagefright/MetaData.h> 3532f3cefa373cd55e63deda36ca9d07c7fe22eaafAndreas Huber#include <media/IStreamSource.h> 36cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber#include <utils/KeyedVector.h> 37cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 38cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Hubernamespace android { 39cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 406e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber// I want the expression "y" evaluated even if verbose logging is off. 416e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber#define MY_LOGV(x, y) \ 426e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber do { unsigned tmp = y; LOGV(x, tmp); } while (0) 436e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber 44cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huberstatic const size_t kTSPacketSize = 188; 45cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 46cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huberstruct ATSParser::Program : public RefBase { 47f933441648ef6a71dee783d733aac17b9508b452Andreas Huber Program(ATSParser *parser, unsigned programMapPID); 48cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 49cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber bool parsePID( 50cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned pid, unsigned payload_unit_start_indicator, 51cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber ABitReader *br); 52cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 5332f3cefa373cd55e63deda36ca9d07c7fe22eaafAndreas Huber void signalDiscontinuity( 5432f3cefa373cd55e63deda36ca9d07c7fe22eaafAndreas Huber DiscontinuityType type, const sp<AMessage> &extra); 5532f3cefa373cd55e63deda36ca9d07c7fe22eaafAndreas Huber 56f933441648ef6a71dee783d733aac17b9508b452Andreas Huber void signalEOS(status_t finalResult); 572a4d22d79e927f2245537921e10fc5fda1c47a29Andreas Huber 58cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber sp<MediaSource> getSource(SourceType type); 59cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 60bff07d0b22a5ee2d9f044f6cb5e4be1532017ab0Andreas Huber int64_t convertPTSToTimestamp(uint64_t PTS); 61bff07d0b22a5ee2d9f044f6cb5e4be1532017ab0Andreas Huber 6243c3e6ce02215ca99d506458f596cb1211639f29Andreas Huber bool PTSTimeDeltaEstablished() const { 6343c3e6ce02215ca99d506458f596cb1211639f29Andreas Huber return mFirstPTSValid; 6443c3e6ce02215ca99d506458f596cb1211639f29Andreas Huber } 6543c3e6ce02215ca99d506458f596cb1211639f29Andreas Huber 66cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huberprivate: 67f933441648ef6a71dee783d733aac17b9508b452Andreas Huber ATSParser *mParser; 68cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned mProgramMapPID; 69cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber KeyedVector<unsigned, sp<Stream> > mStreams; 70bff07d0b22a5ee2d9f044f6cb5e4be1532017ab0Andreas Huber bool mFirstPTSValid; 71bff07d0b22a5ee2d9f044f6cb5e4be1532017ab0Andreas Huber uint64_t mFirstPTS; 72cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 73cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber void parseProgramMap(ABitReader *br); 74cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 75cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber DISALLOW_EVIL_CONSTRUCTORS(Program); 76cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber}; 77cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 78cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huberstruct ATSParser::Stream : public RefBase { 79bff07d0b22a5ee2d9f044f6cb5e4be1532017ab0Andreas Huber Stream(Program *program, unsigned elementaryPID, unsigned streamType); 80cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 81bc7f5b2e56107cfeaeeab13cf8979379e3c2f139Andreas Huber unsigned type() const { return mStreamType; } 82bc7f5b2e56107cfeaeeab13cf8979379e3c2f139Andreas Huber unsigned pid() const { return mElementaryPID; } 83bc7f5b2e56107cfeaeeab13cf8979379e3c2f139Andreas Huber void setPID(unsigned pid) { mElementaryPID = pid; } 84bc7f5b2e56107cfeaeeab13cf8979379e3c2f139Andreas Huber 85cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber void parse( 86cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned payload_unit_start_indicator, 87cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber ABitReader *br); 88cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 8932f3cefa373cd55e63deda36ca9d07c7fe22eaafAndreas Huber void signalDiscontinuity( 9032f3cefa373cd55e63deda36ca9d07c7fe22eaafAndreas Huber DiscontinuityType type, const sp<AMessage> &extra); 9132f3cefa373cd55e63deda36ca9d07c7fe22eaafAndreas Huber 92f933441648ef6a71dee783d733aac17b9508b452Andreas Huber void signalEOS(status_t finalResult); 932a4d22d79e927f2245537921e10fc5fda1c47a29Andreas Huber 94cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber sp<MediaSource> getSource(SourceType type); 95cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 96cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huberprotected: 97cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber virtual ~Stream(); 98cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 99cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huberprivate: 100bff07d0b22a5ee2d9f044f6cb5e4be1532017ab0Andreas Huber Program *mProgram; 101cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned mElementaryPID; 102cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned mStreamType; 103cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 104cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber sp<ABuffer> mBuffer; 105cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber sp<AnotherPacketSource> mSource; 106cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber bool mPayloadStarted; 107bc7f5b2e56107cfeaeeab13cf8979379e3c2f139Andreas Huber DiscontinuityType mPendingDiscontinuity; 10832f3cefa373cd55e63deda36ca9d07c7fe22eaafAndreas Huber sp<AMessage> mPendingDiscontinuityExtra; 109cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 1106a63a939601645404fd98f58c19cc38ca818d99eAndreas Huber ElementaryStreamQueue mQueue; 1116a63a939601645404fd98f58c19cc38ca818d99eAndreas Huber 112cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber void flush(); 113cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber void parsePES(ABitReader *br); 114cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 115cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber void onPayloadData( 116cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned PTS_DTS_flags, uint64_t PTS, uint64_t DTS, 117cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber const uint8_t *data, size_t size); 118cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 11982f7321b03eec1e40af9d681370f754ee0279582Andreas Huber void extractAACFrames(const sp<ABuffer> &buffer); 12082f7321b03eec1e40af9d681370f754ee0279582Andreas Huber 12132f3cefa373cd55e63deda36ca9d07c7fe22eaafAndreas Huber void deferDiscontinuity( 12232f3cefa373cd55e63deda36ca9d07c7fe22eaafAndreas Huber DiscontinuityType type, const sp<AMessage> &extra); 123bc7f5b2e56107cfeaeeab13cf8979379e3c2f139Andreas Huber 124cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber DISALLOW_EVIL_CONSTRUCTORS(Stream); 125cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber}; 126cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 127cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber//////////////////////////////////////////////////////////////////////////////// 128cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 129f933441648ef6a71dee783d733aac17b9508b452Andreas HuberATSParser::Program::Program(ATSParser *parser, unsigned programMapPID) 130f933441648ef6a71dee783d733aac17b9508b452Andreas Huber : mParser(parser), 131f933441648ef6a71dee783d733aac17b9508b452Andreas Huber mProgramMapPID(programMapPID), 132bff07d0b22a5ee2d9f044f6cb5e4be1532017ab0Andreas Huber mFirstPTSValid(false), 133bff07d0b22a5ee2d9f044f6cb5e4be1532017ab0Andreas Huber mFirstPTS(0) { 134cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber} 135cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 136cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huberbool ATSParser::Program::parsePID( 137cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned pid, unsigned payload_unit_start_indicator, 138cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber ABitReader *br) { 139cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (pid == mProgramMapPID) { 140cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (payload_unit_start_indicator) { 141cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned skip = br->getBits(8); 142cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber br->skipBits(skip * 8); 143cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 144cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 145cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber parseProgramMap(br); 146cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber return true; 147cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 148cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 149cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber ssize_t index = mStreams.indexOfKey(pid); 150cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (index < 0) { 151cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber return false; 152cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 153cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 154cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber mStreams.editValueAt(index)->parse( 155cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber payload_unit_start_indicator, br); 156cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 157cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber return true; 158cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber} 159cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 16032f3cefa373cd55e63deda36ca9d07c7fe22eaafAndreas Hubervoid ATSParser::Program::signalDiscontinuity( 16132f3cefa373cd55e63deda36ca9d07c7fe22eaafAndreas Huber DiscontinuityType type, const sp<AMessage> &extra) { 1622a4d22d79e927f2245537921e10fc5fda1c47a29Andreas Huber for (size_t i = 0; i < mStreams.size(); ++i) { 16332f3cefa373cd55e63deda36ca9d07c7fe22eaafAndreas Huber mStreams.editValueAt(i)->signalDiscontinuity(type, extra); 164f933441648ef6a71dee783d733aac17b9508b452Andreas Huber } 165f933441648ef6a71dee783d733aac17b9508b452Andreas Huber} 166f933441648ef6a71dee783d733aac17b9508b452Andreas Huber 167f933441648ef6a71dee783d733aac17b9508b452Andreas Hubervoid ATSParser::Program::signalEOS(status_t finalResult) { 168f933441648ef6a71dee783d733aac17b9508b452Andreas Huber for (size_t i = 0; i < mStreams.size(); ++i) { 169f933441648ef6a71dee783d733aac17b9508b452Andreas Huber mStreams.editValueAt(i)->signalEOS(finalResult); 1702a4d22d79e927f2245537921e10fc5fda1c47a29Andreas Huber } 1712a4d22d79e927f2245537921e10fc5fda1c47a29Andreas Huber} 1722a4d22d79e927f2245537921e10fc5fda1c47a29Andreas Huber 173bc7f5b2e56107cfeaeeab13cf8979379e3c2f139Andreas Huberstruct StreamInfo { 174bc7f5b2e56107cfeaeeab13cf8979379e3c2f139Andreas Huber unsigned mType; 175bc7f5b2e56107cfeaeeab13cf8979379e3c2f139Andreas Huber unsigned mPID; 176bc7f5b2e56107cfeaeeab13cf8979379e3c2f139Andreas Huber}; 177bc7f5b2e56107cfeaeeab13cf8979379e3c2f139Andreas Huber 178cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Hubervoid ATSParser::Program::parseProgramMap(ABitReader *br) { 179cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned table_id = br->getBits(8); 1806e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber LOGV(" table_id = %u", table_id); 181cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(table_id, 0x02u); 182cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 1836e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber unsigned section_syntax_indicator = br->getBits(1); 1846e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber LOGV(" section_syntax_indicator = %u", section_syntax_indicator); 1856e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber CHECK_EQ(section_syntax_indicator, 1u); 186cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 187cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(br->getBits(1), 0u); 1886e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber MY_LOGV(" reserved = %u", br->getBits(2)); 189cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 190cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned section_length = br->getBits(12); 1916e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber LOGV(" section_length = %u", section_length); 192f933441648ef6a71dee783d733aac17b9508b452Andreas Huber CHECK_EQ(section_length & 0xc00, 0u); 193cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_LE(section_length, 1021u); 194cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 1956e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber MY_LOGV(" program_number = %u", br->getBits(16)); 1966e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber MY_LOGV(" reserved = %u", br->getBits(2)); 1976e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber MY_LOGV(" version_number = %u", br->getBits(5)); 1986e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber MY_LOGV(" current_next_indicator = %u", br->getBits(1)); 1996e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber MY_LOGV(" section_number = %u", br->getBits(8)); 2006e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber MY_LOGV(" last_section_number = %u", br->getBits(8)); 2016e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber MY_LOGV(" reserved = %u", br->getBits(3)); 2026e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber MY_LOGV(" PCR_PID = 0x%04x", br->getBits(13)); 2036e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber MY_LOGV(" reserved = %u", br->getBits(4)); 204cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 205cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned program_info_length = br->getBits(12); 2066e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber LOGV(" program_info_length = %u", program_info_length); 207f933441648ef6a71dee783d733aac17b9508b452Andreas Huber CHECK_EQ(program_info_length & 0xc00, 0u); 208cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 209cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber br->skipBits(program_info_length * 8); // skip descriptors 210cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 211bc7f5b2e56107cfeaeeab13cf8979379e3c2f139Andreas Huber Vector<StreamInfo> infos; 212bc7f5b2e56107cfeaeeab13cf8979379e3c2f139Andreas Huber 213cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber // infoBytesRemaining is the number of bytes that make up the 214cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber // variable length section of ES_infos. It does not include the 215cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber // final CRC. 216cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber size_t infoBytesRemaining = section_length - 9 - program_info_length - 4; 217cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 218cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber while (infoBytesRemaining > 0) { 219cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_GE(infoBytesRemaining, 5u); 220cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 221cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned streamType = br->getBits(8); 2226e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber LOGV(" stream_type = 0x%02x", streamType); 223cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 2246e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber MY_LOGV(" reserved = %u", br->getBits(3)); 225cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 226cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned elementaryPID = br->getBits(13); 2276e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber LOGV(" elementary_PID = 0x%04x", elementaryPID); 228cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 2296e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber MY_LOGV(" reserved = %u", br->getBits(4)); 230cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 231cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned ES_info_length = br->getBits(12); 2326e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber LOGV(" ES_info_length = %u", ES_info_length); 233f933441648ef6a71dee783d733aac17b9508b452Andreas Huber CHECK_EQ(ES_info_length & 0xc00, 0u); 234cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 235cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_GE(infoBytesRemaining - 5, ES_info_length); 236cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 237cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber#if 0 238cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber br->skipBits(ES_info_length * 8); // skip descriptors 239cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber#else 240cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned info_bytes_remaining = ES_info_length; 241cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber while (info_bytes_remaining >= 2) { 2426e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber MY_LOGV(" tag = 0x%02x", br->getBits(8)); 243cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 244cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned descLength = br->getBits(8); 2456e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber LOGV(" len = %u", descLength); 246cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 247cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_GE(info_bytes_remaining, 2 + descLength); 248cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 249cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber br->skipBits(descLength * 8); 250cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 251cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber info_bytes_remaining -= descLength + 2; 252cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 253cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(info_bytes_remaining, 0u); 254cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber#endif 255cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 256bc7f5b2e56107cfeaeeab13cf8979379e3c2f139Andreas Huber StreamInfo info; 257bc7f5b2e56107cfeaeeab13cf8979379e3c2f139Andreas Huber info.mType = streamType; 258bc7f5b2e56107cfeaeeab13cf8979379e3c2f139Andreas Huber info.mPID = elementaryPID; 259bc7f5b2e56107cfeaeeab13cf8979379e3c2f139Andreas Huber infos.push(info); 260cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 261cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber infoBytesRemaining -= 5 + ES_info_length; 262cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 263cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 264cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(infoBytesRemaining, 0u); 2656e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber MY_LOGV(" CRC = 0x%08x", br->getBits(32)); 266bc7f5b2e56107cfeaeeab13cf8979379e3c2f139Andreas Huber 267bc7f5b2e56107cfeaeeab13cf8979379e3c2f139Andreas Huber bool PIDsChanged = false; 268bc7f5b2e56107cfeaeeab13cf8979379e3c2f139Andreas Huber for (size_t i = 0; i < infos.size(); ++i) { 269bc7f5b2e56107cfeaeeab13cf8979379e3c2f139Andreas Huber StreamInfo &info = infos.editItemAt(i); 270bc7f5b2e56107cfeaeeab13cf8979379e3c2f139Andreas Huber 271bc7f5b2e56107cfeaeeab13cf8979379e3c2f139Andreas Huber ssize_t index = mStreams.indexOfKey(info.mPID); 272bc7f5b2e56107cfeaeeab13cf8979379e3c2f139Andreas Huber 273bc7f5b2e56107cfeaeeab13cf8979379e3c2f139Andreas Huber if (index >= 0 && mStreams.editValueAt(index)->type() != info.mType) { 274bc7f5b2e56107cfeaeeab13cf8979379e3c2f139Andreas Huber LOGI("uh oh. stream PIDs have changed."); 275bc7f5b2e56107cfeaeeab13cf8979379e3c2f139Andreas Huber PIDsChanged = true; 276bc7f5b2e56107cfeaeeab13cf8979379e3c2f139Andreas Huber break; 277bc7f5b2e56107cfeaeeab13cf8979379e3c2f139Andreas Huber } 278bc7f5b2e56107cfeaeeab13cf8979379e3c2f139Andreas Huber } 279bc7f5b2e56107cfeaeeab13cf8979379e3c2f139Andreas Huber 280bc7f5b2e56107cfeaeeab13cf8979379e3c2f139Andreas Huber if (PIDsChanged) { 281bc7f5b2e56107cfeaeeab13cf8979379e3c2f139Andreas Huber mStreams.clear(); 282bc7f5b2e56107cfeaeeab13cf8979379e3c2f139Andreas Huber } 283bc7f5b2e56107cfeaeeab13cf8979379e3c2f139Andreas Huber 284bc7f5b2e56107cfeaeeab13cf8979379e3c2f139Andreas Huber for (size_t i = 0; i < infos.size(); ++i) { 285bc7f5b2e56107cfeaeeab13cf8979379e3c2f139Andreas Huber StreamInfo &info = infos.editItemAt(i); 286bc7f5b2e56107cfeaeeab13cf8979379e3c2f139Andreas Huber 287bc7f5b2e56107cfeaeeab13cf8979379e3c2f139Andreas Huber ssize_t index = mStreams.indexOfKey(info.mPID); 288bc7f5b2e56107cfeaeeab13cf8979379e3c2f139Andreas Huber 289bc7f5b2e56107cfeaeeab13cf8979379e3c2f139Andreas Huber if (index < 0) { 290bc7f5b2e56107cfeaeeab13cf8979379e3c2f139Andreas Huber sp<Stream> stream = new Stream(this, info.mPID, info.mType); 291bc7f5b2e56107cfeaeeab13cf8979379e3c2f139Andreas Huber mStreams.add(info.mPID, stream); 292bc7f5b2e56107cfeaeeab13cf8979379e3c2f139Andreas Huber 293bc7f5b2e56107cfeaeeab13cf8979379e3c2f139Andreas Huber if (PIDsChanged) { 29432f3cefa373cd55e63deda36ca9d07c7fe22eaafAndreas Huber sp<AMessage> extra; 29532f3cefa373cd55e63deda36ca9d07c7fe22eaafAndreas Huber stream->signalDiscontinuity(DISCONTINUITY_FORMATCHANGE, extra); 296bc7f5b2e56107cfeaeeab13cf8979379e3c2f139Andreas Huber } 297bc7f5b2e56107cfeaeeab13cf8979379e3c2f139Andreas Huber } 298bc7f5b2e56107cfeaeeab13cf8979379e3c2f139Andreas Huber } 299cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber} 300cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 301cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Hubersp<MediaSource> ATSParser::Program::getSource(SourceType type) { 302decd96988e495133e4a1728f612d4c9fdb4d218eAndreas Huber size_t index = (type == MPEG2ADTS_AUDIO) ? 0 : 0; 303decd96988e495133e4a1728f612d4c9fdb4d218eAndreas Huber 304cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber for (size_t i = 0; i < mStreams.size(); ++i) { 305cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber sp<MediaSource> source = mStreams.editValueAt(i)->getSource(type); 306cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (source != NULL) { 307decd96988e495133e4a1728f612d4c9fdb4d218eAndreas Huber if (index == 0) { 308decd96988e495133e4a1728f612d4c9fdb4d218eAndreas Huber return source; 309decd96988e495133e4a1728f612d4c9fdb4d218eAndreas Huber } 310decd96988e495133e4a1728f612d4c9fdb4d218eAndreas Huber --index; 311cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 312cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 313cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 314cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber return NULL; 315cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber} 316cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 317bff07d0b22a5ee2d9f044f6cb5e4be1532017ab0Andreas Huberint64_t ATSParser::Program::convertPTSToTimestamp(uint64_t PTS) { 318bff07d0b22a5ee2d9f044f6cb5e4be1532017ab0Andreas Huber if (!mFirstPTSValid) { 319bff07d0b22a5ee2d9f044f6cb5e4be1532017ab0Andreas Huber mFirstPTSValid = true; 320bff07d0b22a5ee2d9f044f6cb5e4be1532017ab0Andreas Huber mFirstPTS = PTS; 321bff07d0b22a5ee2d9f044f6cb5e4be1532017ab0Andreas Huber PTS = 0; 322bff07d0b22a5ee2d9f044f6cb5e4be1532017ab0Andreas Huber } else if (PTS < mFirstPTS) { 323bff07d0b22a5ee2d9f044f6cb5e4be1532017ab0Andreas Huber PTS = 0; 324bff07d0b22a5ee2d9f044f6cb5e4be1532017ab0Andreas Huber } else { 325bff07d0b22a5ee2d9f044f6cb5e4be1532017ab0Andreas Huber PTS -= mFirstPTS; 326bff07d0b22a5ee2d9f044f6cb5e4be1532017ab0Andreas Huber } 327bff07d0b22a5ee2d9f044f6cb5e4be1532017ab0Andreas Huber 328bff07d0b22a5ee2d9f044f6cb5e4be1532017ab0Andreas Huber return (PTS * 100) / 9; 329bff07d0b22a5ee2d9f044f6cb5e4be1532017ab0Andreas Huber} 330bff07d0b22a5ee2d9f044f6cb5e4be1532017ab0Andreas Huber 331cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber//////////////////////////////////////////////////////////////////////////////// 332cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 333bff07d0b22a5ee2d9f044f6cb5e4be1532017ab0Andreas HuberATSParser::Stream::Stream( 334bff07d0b22a5ee2d9f044f6cb5e4be1532017ab0Andreas Huber Program *program, unsigned elementaryPID, unsigned streamType) 335bff07d0b22a5ee2d9f044f6cb5e4be1532017ab0Andreas Huber : mProgram(program), 336bff07d0b22a5ee2d9f044f6cb5e4be1532017ab0Andreas Huber mElementaryPID(elementaryPID), 337cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber mStreamType(streamType), 3383831a066bcf1019864a94d2bc7b4c9241efc5c22Andreas Huber mBuffer(new ABuffer(192 * 1024)), 3396a63a939601645404fd98f58c19cc38ca818d99eAndreas Huber mPayloadStarted(false), 340bc7f5b2e56107cfeaeeab13cf8979379e3c2f139Andreas Huber mPendingDiscontinuity(DISCONTINUITY_NONE), 3416a63a939601645404fd98f58c19cc38ca818d99eAndreas Huber mQueue(streamType == 0x1b 3426a63a939601645404fd98f58c19cc38ca818d99eAndreas Huber ? ElementaryStreamQueue::H264 : ElementaryStreamQueue::AAC) { 343cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber mBuffer->setRange(0, 0); 344e332a9181cf6a3155ed1a0fd2afc212ccb1f2753Andreas Huber 345e332a9181cf6a3155ed1a0fd2afc212ccb1f2753Andreas Huber LOGV("new stream PID 0x%02x, type 0x%02x", elementaryPID, streamType); 346cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber} 347cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 348cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas HuberATSParser::Stream::~Stream() { 349cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber} 350cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 351cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Hubervoid ATSParser::Stream::parse( 352cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned payload_unit_start_indicator, ABitReader *br) { 353cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (payload_unit_start_indicator) { 354cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (mPayloadStarted) { 355cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber // Otherwise we run the danger of receiving the trailing bytes 356cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber // of a PES packet that we never saw the start of and assuming 357cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber // we have a a complete PES packet. 358cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 359cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber flush(); 360cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 361cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 362cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber mPayloadStarted = true; 363cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 364cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 365cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (!mPayloadStarted) { 366cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber return; 367cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 368cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 369cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber size_t payloadSizeBits = br->numBitsLeft(); 370f933441648ef6a71dee783d733aac17b9508b452Andreas Huber CHECK_EQ(payloadSizeBits % 8, 0u); 371cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 372cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_LE(mBuffer->size() + payloadSizeBits / 8, mBuffer->capacity()); 373cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 374cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber memcpy(mBuffer->data() + mBuffer->size(), br->data(), payloadSizeBits / 8); 375cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber mBuffer->setRange(0, mBuffer->size() + payloadSizeBits / 8); 376cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber} 377cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 37832f3cefa373cd55e63deda36ca9d07c7fe22eaafAndreas Hubervoid ATSParser::Stream::signalDiscontinuity( 37932f3cefa373cd55e63deda36ca9d07c7fe22eaafAndreas Huber DiscontinuityType type, const sp<AMessage> &extra) { 3802a4d22d79e927f2245537921e10fc5fda1c47a29Andreas Huber mPayloadStarted = false; 3812a4d22d79e927f2245537921e10fc5fda1c47a29Andreas Huber mBuffer->setRange(0, 0); 3822a4d22d79e927f2245537921e10fc5fda1c47a29Andreas Huber 383f933441648ef6a71dee783d733aac17b9508b452Andreas Huber switch (type) { 384f933441648ef6a71dee783d733aac17b9508b452Andreas Huber case DISCONTINUITY_SEEK: 385f933441648ef6a71dee783d733aac17b9508b452Andreas Huber case DISCONTINUITY_FORMATCHANGE: 386f933441648ef6a71dee783d733aac17b9508b452Andreas Huber { 387f933441648ef6a71dee783d733aac17b9508b452Andreas Huber bool isASeek = (type == DISCONTINUITY_SEEK); 388bff07d0b22a5ee2d9f044f6cb5e4be1532017ab0Andreas Huber 389f933441648ef6a71dee783d733aac17b9508b452Andreas Huber mQueue.clear(!isASeek); 390f933441648ef6a71dee783d733aac17b9508b452Andreas Huber 39132f3cefa373cd55e63deda36ca9d07c7fe22eaafAndreas Huber uint64_t resumeAtPTS; 39232f3cefa373cd55e63deda36ca9d07c7fe22eaafAndreas Huber if (extra != NULL 39332f3cefa373cd55e63deda36ca9d07c7fe22eaafAndreas Huber && extra->findInt64( 39432f3cefa373cd55e63deda36ca9d07c7fe22eaafAndreas Huber IStreamListener::kKeyResumeAtPTS, 39532f3cefa373cd55e63deda36ca9d07c7fe22eaafAndreas Huber (int64_t *)&resumeAtPTS)) { 39632f3cefa373cd55e63deda36ca9d07c7fe22eaafAndreas Huber int64_t resumeAtMediaTimeUs = 39732f3cefa373cd55e63deda36ca9d07c7fe22eaafAndreas Huber mProgram->convertPTSToTimestamp(resumeAtPTS); 39832f3cefa373cd55e63deda36ca9d07c7fe22eaafAndreas Huber 39932f3cefa373cd55e63deda36ca9d07c7fe22eaafAndreas Huber extra->setInt64("resume-at-mediatimeUs", resumeAtMediaTimeUs); 40032f3cefa373cd55e63deda36ca9d07c7fe22eaafAndreas Huber } 40132f3cefa373cd55e63deda36ca9d07c7fe22eaafAndreas Huber 402f933441648ef6a71dee783d733aac17b9508b452Andreas Huber if (mSource != NULL) { 40332f3cefa373cd55e63deda36ca9d07c7fe22eaafAndreas Huber mSource->queueDiscontinuity(type, extra); 404bc7f5b2e56107cfeaeeab13cf8979379e3c2f139Andreas Huber } else { 40532f3cefa373cd55e63deda36ca9d07c7fe22eaafAndreas Huber deferDiscontinuity(type, extra); 406f933441648ef6a71dee783d733aac17b9508b452Andreas Huber } 407f933441648ef6a71dee783d733aac17b9508b452Andreas Huber break; 408e332a9181cf6a3155ed1a0fd2afc212ccb1f2753Andreas Huber } 409f933441648ef6a71dee783d733aac17b9508b452Andreas Huber 410f933441648ef6a71dee783d733aac17b9508b452Andreas Huber default: 411f933441648ef6a71dee783d733aac17b9508b452Andreas Huber TRESPASS(); 412f933441648ef6a71dee783d733aac17b9508b452Andreas Huber break; 413bff07d0b22a5ee2d9f044f6cb5e4be1532017ab0Andreas Huber } 414f933441648ef6a71dee783d733aac17b9508b452Andreas Huber} 415bff07d0b22a5ee2d9f044f6cb5e4be1532017ab0Andreas Huber 41632f3cefa373cd55e63deda36ca9d07c7fe22eaafAndreas Hubervoid ATSParser::Stream::deferDiscontinuity( 41732f3cefa373cd55e63deda36ca9d07c7fe22eaafAndreas Huber DiscontinuityType type, const sp<AMessage> &extra) { 418bc7f5b2e56107cfeaeeab13cf8979379e3c2f139Andreas Huber if (type > mPendingDiscontinuity) { 419bc7f5b2e56107cfeaeeab13cf8979379e3c2f139Andreas Huber // Only upgrade discontinuities. 420bc7f5b2e56107cfeaeeab13cf8979379e3c2f139Andreas Huber mPendingDiscontinuity = type; 42132f3cefa373cd55e63deda36ca9d07c7fe22eaafAndreas Huber mPendingDiscontinuityExtra = extra; 422bc7f5b2e56107cfeaeeab13cf8979379e3c2f139Andreas Huber } 423bc7f5b2e56107cfeaeeab13cf8979379e3c2f139Andreas Huber} 424bc7f5b2e56107cfeaeeab13cf8979379e3c2f139Andreas Huber 425f933441648ef6a71dee783d733aac17b9508b452Andreas Hubervoid ATSParser::Stream::signalEOS(status_t finalResult) { 426f933441648ef6a71dee783d733aac17b9508b452Andreas Huber if (mSource != NULL) { 427f933441648ef6a71dee783d733aac17b9508b452Andreas Huber mSource->signalEOS(finalResult); 4282a4d22d79e927f2245537921e10fc5fda1c47a29Andreas Huber } 4292a4d22d79e927f2245537921e10fc5fda1c47a29Andreas Huber} 4302a4d22d79e927f2245537921e10fc5fda1c47a29Andreas Huber 431cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Hubervoid ATSParser::Stream::parsePES(ABitReader *br) { 432cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned packet_startcode_prefix = br->getBits(24); 433cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 4346e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber LOGV("packet_startcode_prefix = 0x%08x", packet_startcode_prefix); 435cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 436cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(packet_startcode_prefix, 0x000001u); 437cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 438cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned stream_id = br->getBits(8); 4396e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber LOGV("stream_id = 0x%02x", stream_id); 440cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 441cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned PES_packet_length = br->getBits(16); 4426e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber LOGV("PES_packet_length = %u", PES_packet_length); 443cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 444cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (stream_id != 0xbc // program_stream_map 445cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber && stream_id != 0xbe // padding_stream 446cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber && stream_id != 0xbf // private_stream_2 447cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber && stream_id != 0xf0 // ECM 448cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber && stream_id != 0xf1 // EMM 449cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber && stream_id != 0xff // program_stream_directory 450cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber && stream_id != 0xf2 // DSMCC 451cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber && stream_id != 0xf8) { // H.222.1 type E 452cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(br->getBits(2), 2u); 453cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 4546e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber MY_LOGV("PES_scrambling_control = %u", br->getBits(2)); 4556e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber MY_LOGV("PES_priority = %u", br->getBits(1)); 4566e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber MY_LOGV("data_alignment_indicator = %u", br->getBits(1)); 4576e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber MY_LOGV("copyright = %u", br->getBits(1)); 4586e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber MY_LOGV("original_or_copy = %u", br->getBits(1)); 459cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 460cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned PTS_DTS_flags = br->getBits(2); 4616e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber LOGV("PTS_DTS_flags = %u", PTS_DTS_flags); 462cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 463cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned ESCR_flag = br->getBits(1); 4646e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber LOGV("ESCR_flag = %u", ESCR_flag); 465cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 466cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned ES_rate_flag = br->getBits(1); 4676e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber LOGV("ES_rate_flag = %u", ES_rate_flag); 468cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 469cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned DSM_trick_mode_flag = br->getBits(1); 4706e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber LOGV("DSM_trick_mode_flag = %u", DSM_trick_mode_flag); 471cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 472cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned additional_copy_info_flag = br->getBits(1); 4736e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber LOGV("additional_copy_info_flag = %u", additional_copy_info_flag); 474cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 4756e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber MY_LOGV("PES_CRC_flag = %u", br->getBits(1)); 4766e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber MY_LOGV("PES_extension_flag = %u", br->getBits(1)); 477cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 478cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned PES_header_data_length = br->getBits(8); 4796e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber LOGV("PES_header_data_length = %u", PES_header_data_length); 480cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 481cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned optional_bytes_remaining = PES_header_data_length; 482cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 483cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber uint64_t PTS = 0, DTS = 0; 484cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 485cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (PTS_DTS_flags == 2 || PTS_DTS_flags == 3) { 486cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_GE(optional_bytes_remaining, 5u); 487cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 488cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(br->getBits(4), PTS_DTS_flags); 489cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 490cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber PTS = ((uint64_t)br->getBits(3)) << 30; 491cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(br->getBits(1), 1u); 492cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber PTS |= ((uint64_t)br->getBits(15)) << 15; 493cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(br->getBits(1), 1u); 494cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber PTS |= br->getBits(15); 495cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(br->getBits(1), 1u); 496cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 4976e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber LOGV("PTS = %llu", PTS); 4986e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber // LOGI("PTS = %.2f secs", PTS / 90000.0f); 499cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 500cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber optional_bytes_remaining -= 5; 501cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 502cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (PTS_DTS_flags == 3) { 503cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_GE(optional_bytes_remaining, 5u); 504cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 505cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(br->getBits(4), 1u); 506cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 507cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber DTS = ((uint64_t)br->getBits(3)) << 30; 508cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(br->getBits(1), 1u); 509cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber DTS |= ((uint64_t)br->getBits(15)) << 15; 510cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(br->getBits(1), 1u); 511cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber DTS |= br->getBits(15); 512cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(br->getBits(1), 1u); 513cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 5146e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber LOGV("DTS = %llu", DTS); 515cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 516cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber optional_bytes_remaining -= 5; 517cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 518cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 519cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 520cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (ESCR_flag) { 521cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_GE(optional_bytes_remaining, 6u); 522cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 523cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber br->getBits(2); 524cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 525cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber uint64_t ESCR = ((uint64_t)br->getBits(3)) << 30; 526cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(br->getBits(1), 1u); 527cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber ESCR |= ((uint64_t)br->getBits(15)) << 15; 528cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(br->getBits(1), 1u); 529cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber ESCR |= br->getBits(15); 530cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(br->getBits(1), 1u); 531cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 5326e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber LOGV("ESCR = %llu", ESCR); 5336e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber MY_LOGV("ESCR_extension = %u", br->getBits(9)); 534cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 535cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(br->getBits(1), 1u); 536cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 537cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber optional_bytes_remaining -= 6; 538cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 539cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 540cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (ES_rate_flag) { 541cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_GE(optional_bytes_remaining, 3u); 542cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 543cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(br->getBits(1), 1u); 5446e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber MY_LOGV("ES_rate = %u", br->getBits(22)); 545cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(br->getBits(1), 1u); 546cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 547cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber optional_bytes_remaining -= 3; 548cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 549cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 550cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber br->skipBits(optional_bytes_remaining * 8); 551cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 552cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber // ES data follows. 553cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 554cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (PES_packet_length != 0) { 555cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_GE(PES_packet_length, PES_header_data_length + 3); 556cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 557cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned dataLength = 558cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber PES_packet_length - 3 - PES_header_data_length; 559cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 5600da4dab0a45a2bc1d95cbc6ef6a4850ed2569584Andreas Huber CHECK_GE(br->numBitsLeft(), dataLength * 8); 5610da4dab0a45a2bc1d95cbc6ef6a4850ed2569584Andreas Huber 5620da4dab0a45a2bc1d95cbc6ef6a4850ed2569584Andreas Huber onPayloadData( 5630da4dab0a45a2bc1d95cbc6ef6a4850ed2569584Andreas Huber PTS_DTS_flags, PTS, DTS, br->data(), dataLength); 564cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 565cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber br->skipBits(dataLength * 8); 566cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } else { 5670da4dab0a45a2bc1d95cbc6ef6a4850ed2569584Andreas Huber onPayloadData( 5680da4dab0a45a2bc1d95cbc6ef6a4850ed2569584Andreas Huber PTS_DTS_flags, PTS, DTS, 5690da4dab0a45a2bc1d95cbc6ef6a4850ed2569584Andreas Huber br->data(), br->numBitsLeft() / 8); 5700da4dab0a45a2bc1d95cbc6ef6a4850ed2569584Andreas Huber 571cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber size_t payloadSizeBits = br->numBitsLeft(); 572f933441648ef6a71dee783d733aac17b9508b452Andreas Huber CHECK_EQ(payloadSizeBits % 8, 0u); 573cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 5746e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber LOGV("There's %d bytes of payload.", payloadSizeBits / 8); 575cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 576cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } else if (stream_id == 0xbe) { // padding_stream 577cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_NE(PES_packet_length, 0u); 578cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber br->skipBits(PES_packet_length * 8); 579cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } else { 580cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_NE(PES_packet_length, 0u); 581cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber br->skipBits(PES_packet_length * 8); 582cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 583cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber} 584cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 585cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Hubervoid ATSParser::Stream::flush() { 586cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (mBuffer->size() == 0) { 587cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber return; 588cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 589cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 5906e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber LOGV("flushing stream 0x%04x size = %d", mElementaryPID, mBuffer->size()); 591cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 592cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber ABitReader br(mBuffer->data(), mBuffer->size()); 593cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber parsePES(&br); 594cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 595cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber mBuffer->setRange(0, 0); 596cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber} 597cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 598cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Hubervoid ATSParser::Stream::onPayloadData( 599cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned PTS_DTS_flags, uint64_t PTS, uint64_t DTS, 600cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber const uint8_t *data, size_t size) { 6016e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber LOGV("onPayloadData mStreamType=0x%02x", mStreamType); 602cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 603cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK(PTS_DTS_flags == 2 || PTS_DTS_flags == 3); 604bff07d0b22a5ee2d9f044f6cb5e4be1532017ab0Andreas Huber int64_t timeUs = mProgram->convertPTSToTimestamp(PTS); 60582f7321b03eec1e40af9d681370f754ee0279582Andreas Huber 6066a63a939601645404fd98f58c19cc38ca818d99eAndreas Huber status_t err = mQueue.appendData(data, size, timeUs); 607decd96988e495133e4a1728f612d4c9fdb4d218eAndreas Huber 608decd96988e495133e4a1728f612d4c9fdb4d218eAndreas Huber if (err != OK) { 609decd96988e495133e4a1728f612d4c9fdb4d218eAndreas Huber return; 610decd96988e495133e4a1728f612d4c9fdb4d218eAndreas Huber } 61182f7321b03eec1e40af9d681370f754ee0279582Andreas Huber 6126a63a939601645404fd98f58c19cc38ca818d99eAndreas Huber sp<ABuffer> accessUnit; 6136a63a939601645404fd98f58c19cc38ca818d99eAndreas Huber while ((accessUnit = mQueue.dequeueAccessUnit()) != NULL) { 6146a63a939601645404fd98f58c19cc38ca818d99eAndreas Huber if (mSource == NULL) { 6156a63a939601645404fd98f58c19cc38ca818d99eAndreas Huber sp<MetaData> meta = mQueue.getFormat(); 61682f7321b03eec1e40af9d681370f754ee0279582Andreas Huber 6176a63a939601645404fd98f58c19cc38ca818d99eAndreas Huber if (meta != NULL) { 6186a63a939601645404fd98f58c19cc38ca818d99eAndreas Huber LOGV("created source!"); 6196a63a939601645404fd98f58c19cc38ca818d99eAndreas Huber mSource = new AnotherPacketSource(meta); 620f933441648ef6a71dee783d733aac17b9508b452Andreas Huber 621bc7f5b2e56107cfeaeeab13cf8979379e3c2f139Andreas Huber if (mPendingDiscontinuity != DISCONTINUITY_NONE) { 62232f3cefa373cd55e63deda36ca9d07c7fe22eaafAndreas Huber mSource->queueDiscontinuity( 62332f3cefa373cd55e63deda36ca9d07c7fe22eaafAndreas Huber mPendingDiscontinuity, mPendingDiscontinuityExtra); 624bc7f5b2e56107cfeaeeab13cf8979379e3c2f139Andreas Huber mPendingDiscontinuity = DISCONTINUITY_NONE; 62532f3cefa373cd55e63deda36ca9d07c7fe22eaafAndreas Huber mPendingDiscontinuityExtra.clear(); 626bc7f5b2e56107cfeaeeab13cf8979379e3c2f139Andreas Huber } 627bc7f5b2e56107cfeaeeab13cf8979379e3c2f139Andreas Huber 6286a63a939601645404fd98f58c19cc38ca818d99eAndreas Huber mSource->queueAccessUnit(accessUnit); 62982f7321b03eec1e40af9d681370f754ee0279582Andreas Huber } 6302a4d22d79e927f2245537921e10fc5fda1c47a29Andreas Huber } else if (mQueue.getFormat() != NULL) { 6312a4d22d79e927f2245537921e10fc5fda1c47a29Andreas Huber // After a discontinuity we invalidate the queue's format 6322a4d22d79e927f2245537921e10fc5fda1c47a29Andreas Huber // and won't enqueue any access units to the source until 6332a4d22d79e927f2245537921e10fc5fda1c47a29Andreas Huber // the queue has reestablished the new format. 6345bc087c573c70c84c6a39946457590b42d392a33Andreas Huber 6355bc087c573c70c84c6a39946457590b42d392a33Andreas Huber if (mSource->getFormat() == NULL) { 6365bc087c573c70c84c6a39946457590b42d392a33Andreas Huber mSource->setFormat(mQueue.getFormat()); 6375bc087c573c70c84c6a39946457590b42d392a33Andreas Huber } 6386a63a939601645404fd98f58c19cc38ca818d99eAndreas Huber mSource->queueAccessUnit(accessUnit); 63982f7321b03eec1e40af9d681370f754ee0279582Andreas Huber } 64082f7321b03eec1e40af9d681370f754ee0279582Andreas Huber } 64182f7321b03eec1e40af9d681370f754ee0279582Andreas Huber} 64282f7321b03eec1e40af9d681370f754ee0279582Andreas Huber 643cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Hubersp<MediaSource> ATSParser::Stream::getSource(SourceType type) { 644cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if ((type == AVC_VIDEO && mStreamType == 0x1b) 645cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber || (type == MPEG2ADTS_AUDIO && mStreamType == 0x0f)) { 646cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber return mSource; 647cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 648cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 649cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber return NULL; 650cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber} 651cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 652cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber//////////////////////////////////////////////////////////////////////////////// 653cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 654cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas HuberATSParser::ATSParser() { 655cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber} 656cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 657cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas HuberATSParser::~ATSParser() { 658cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber} 659cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 660cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Hubervoid ATSParser::feedTSPacket(const void *data, size_t size) { 661cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(size, kTSPacketSize); 662cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 663cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber ABitReader br((const uint8_t *)data, kTSPacketSize); 664cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber parseTS(&br); 665cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber} 666cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 66732f3cefa373cd55e63deda36ca9d07c7fe22eaafAndreas Hubervoid ATSParser::signalDiscontinuity( 66832f3cefa373cd55e63deda36ca9d07c7fe22eaafAndreas Huber DiscontinuityType type, const sp<AMessage> &extra) { 669f933441648ef6a71dee783d733aac17b9508b452Andreas Huber for (size_t i = 0; i < mPrograms.size(); ++i) { 67032f3cefa373cd55e63deda36ca9d07c7fe22eaafAndreas Huber mPrograms.editItemAt(i)->signalDiscontinuity(type, extra); 671f933441648ef6a71dee783d733aac17b9508b452Andreas Huber } 672f933441648ef6a71dee783d733aac17b9508b452Andreas Huber} 673f933441648ef6a71dee783d733aac17b9508b452Andreas Huber 674f933441648ef6a71dee783d733aac17b9508b452Andreas Hubervoid ATSParser::signalEOS(status_t finalResult) { 675f933441648ef6a71dee783d733aac17b9508b452Andreas Huber CHECK_NE(finalResult, (status_t)OK); 676f933441648ef6a71dee783d733aac17b9508b452Andreas Huber 6772a4d22d79e927f2245537921e10fc5fda1c47a29Andreas Huber for (size_t i = 0; i < mPrograms.size(); ++i) { 678f933441648ef6a71dee783d733aac17b9508b452Andreas Huber mPrograms.editItemAt(i)->signalEOS(finalResult); 6792a4d22d79e927f2245537921e10fc5fda1c47a29Andreas Huber } 6802a4d22d79e927f2245537921e10fc5fda1c47a29Andreas Huber} 6812a4d22d79e927f2245537921e10fc5fda1c47a29Andreas Huber 682cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Hubervoid ATSParser::parseProgramAssociationTable(ABitReader *br) { 683cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned table_id = br->getBits(8); 6846e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber LOGV(" table_id = %u", table_id); 685cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(table_id, 0x00u); 686cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 687cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned section_syntax_indictor = br->getBits(1); 6886e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber LOGV(" section_syntax_indictor = %u", section_syntax_indictor); 689cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(section_syntax_indictor, 1u); 690cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 691cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(br->getBits(1), 0u); 6926e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber MY_LOGV(" reserved = %u", br->getBits(2)); 693cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 694cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned section_length = br->getBits(12); 6956e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber LOGV(" section_length = %u", section_length); 696f933441648ef6a71dee783d733aac17b9508b452Andreas Huber CHECK_EQ(section_length & 0xc00, 0u); 697cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 6986e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber MY_LOGV(" transport_stream_id = %u", br->getBits(16)); 6996e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber MY_LOGV(" reserved = %u", br->getBits(2)); 7006e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber MY_LOGV(" version_number = %u", br->getBits(5)); 7016e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber MY_LOGV(" current_next_indicator = %u", br->getBits(1)); 7026e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber MY_LOGV(" section_number = %u", br->getBits(8)); 7036e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber MY_LOGV(" last_section_number = %u", br->getBits(8)); 704cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 705cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber size_t numProgramBytes = (section_length - 5 /* header */ - 4 /* crc */); 706cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ((numProgramBytes % 4), 0u); 707cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 708cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber for (size_t i = 0; i < numProgramBytes / 4; ++i) { 709cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned program_number = br->getBits(16); 7106e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber LOGV(" program_number = %u", program_number); 711cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 7126e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber MY_LOGV(" reserved = %u", br->getBits(3)); 713cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 714cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (program_number == 0) { 7156e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber MY_LOGV(" network_PID = 0x%04x", br->getBits(13)); 716cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } else { 717cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned programMapPID = br->getBits(13); 718cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 7196e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber LOGV(" program_map_PID = 0x%04x", programMapPID); 720cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 721f933441648ef6a71dee783d733aac17b9508b452Andreas Huber mPrograms.push(new Program(this, programMapPID)); 722cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 723cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 724cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 7256e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber MY_LOGV(" CRC = 0x%08x", br->getBits(32)); 726cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber} 727cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 728cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Hubervoid ATSParser::parsePID( 729cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber ABitReader *br, unsigned PID, 730cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned payload_unit_start_indicator) { 731cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (PID == 0) { 732cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (payload_unit_start_indicator) { 733cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned skip = br->getBits(8); 734cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber br->skipBits(skip * 8); 735cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 736cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber parseProgramAssociationTable(br); 737cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber return; 738cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 739cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 740cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber bool handled = false; 741cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber for (size_t i = 0; i < mPrograms.size(); ++i) { 742cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (mPrograms.editItemAt(i)->parsePID( 743cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber PID, payload_unit_start_indicator, br)) { 744cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber handled = true; 745cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber break; 746cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 747cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 748cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 749cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (!handled) { 7506e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber LOGV("PID 0x%04x not handled.", PID); 751cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 752cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber} 753cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 754cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Hubervoid ATSParser::parseAdaptationField(ABitReader *br) { 755cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned adaptation_field_length = br->getBits(8); 756cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (adaptation_field_length > 0) { 757cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber br->skipBits(adaptation_field_length * 8); // XXX 758cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 759cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber} 760cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 761cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Hubervoid ATSParser::parseTS(ABitReader *br) { 7626e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber LOGV("---"); 763cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 764cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned sync_byte = br->getBits(8); 765cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(sync_byte, 0x47u); 766cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 7676e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber MY_LOGV("transport_error_indicator = %u", br->getBits(1)); 768cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 769cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned payload_unit_start_indicator = br->getBits(1); 7706e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber LOGV("payload_unit_start_indicator = %u", payload_unit_start_indicator); 771cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 7726e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber MY_LOGV("transport_priority = %u", br->getBits(1)); 773cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 774cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned PID = br->getBits(13); 7756e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber LOGV("PID = 0x%04x", PID); 776cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 7776e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber MY_LOGV("transport_scrambling_control = %u", br->getBits(2)); 778cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 779cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned adaptation_field_control = br->getBits(2); 7806e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber LOGV("adaptation_field_control = %u", adaptation_field_control); 781cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 7820da4dab0a45a2bc1d95cbc6ef6a4850ed2569584Andreas Huber unsigned continuity_counter = br->getBits(4); 7830da4dab0a45a2bc1d95cbc6ef6a4850ed2569584Andreas Huber LOGV("continuity_counter = %u", continuity_counter); 7840da4dab0a45a2bc1d95cbc6ef6a4850ed2569584Andreas Huber 7850da4dab0a45a2bc1d95cbc6ef6a4850ed2569584Andreas Huber // LOGI("PID = 0x%04x, continuity_counter = %u", PID, continuity_counter); 786cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 787cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (adaptation_field_control == 2 || adaptation_field_control == 3) { 788cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber parseAdaptationField(br); 789cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 790cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 791cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (adaptation_field_control == 1 || adaptation_field_control == 3) { 792cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber parsePID(br, PID, payload_unit_start_indicator); 793cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 794cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber} 795cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 796cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Hubersp<MediaSource> ATSParser::getSource(SourceType type) { 797cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber for (size_t i = 0; i < mPrograms.size(); ++i) { 798cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber sp<MediaSource> source = mPrograms.editItemAt(i)->getSource(type); 799cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 800cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (source != NULL) { 801cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber return source; 802cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 803cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 804cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 805cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber return NULL; 806cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber} 807cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 80843c3e6ce02215ca99d506458f596cb1211639f29Andreas Huberbool ATSParser::PTSTimeDeltaEstablished() { 80943c3e6ce02215ca99d506458f596cb1211639f29Andreas Huber if (mPrograms.isEmpty()) { 81043c3e6ce02215ca99d506458f596cb1211639f29Andreas Huber return false; 81143c3e6ce02215ca99d506458f596cb1211639f29Andreas Huber } 81243c3e6ce02215ca99d506458f596cb1211639f29Andreas Huber 81343c3e6ce02215ca99d506458f596cb1211639f29Andreas Huber return mPrograms.editItemAt(0)->PTSTimeDeltaEstablished(); 81443c3e6ce02215ca99d506458f596cb1211639f29Andreas Huber} 81543c3e6ce02215ca99d506458f596cb1211639f29Andreas Huber 816cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber} // namespace android 817