ATSParser.cpp revision 6a63a939601645404fd98f58c19cc38ca818d99e
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> 35cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber#include <utils/KeyedVector.h> 36cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 37cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Hubernamespace android { 38cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 396e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber// I want the expression "y" evaluated even if verbose logging is off. 406e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber#define MY_LOGV(x, y) \ 416e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber do { unsigned tmp = y; LOGV(x, tmp); } while (0) 426e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber 43cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huberstatic const size_t kTSPacketSize = 188; 44cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 45cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huberstruct ATSParser::Program : public RefBase { 46cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber Program(unsigned programMapPID); 47cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 48cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber bool parsePID( 49cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned pid, unsigned payload_unit_start_indicator, 50cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber ABitReader *br); 51cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 52cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber sp<MediaSource> getSource(SourceType type); 53cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 54cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huberprivate: 55cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned mProgramMapPID; 56cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber KeyedVector<unsigned, sp<Stream> > mStreams; 57cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 58cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber void parseProgramMap(ABitReader *br); 59cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 60cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber DISALLOW_EVIL_CONSTRUCTORS(Program); 61cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber}; 62cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 63cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huberstruct ATSParser::Stream : public RefBase { 64cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber Stream(unsigned elementaryPID, unsigned streamType); 65cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 66cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber void parse( 67cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned payload_unit_start_indicator, 68cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber ABitReader *br); 69cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 70cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber sp<MediaSource> getSource(SourceType type); 71cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 72cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huberprotected: 73cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber virtual ~Stream(); 74cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 75cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huberprivate: 76cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned mElementaryPID; 77cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned mStreamType; 78cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 79cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber sp<ABuffer> mBuffer; 80cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber sp<AnotherPacketSource> mSource; 81cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber bool mPayloadStarted; 82cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 836a63a939601645404fd98f58c19cc38ca818d99eAndreas Huber ElementaryStreamQueue mQueue; 846a63a939601645404fd98f58c19cc38ca818d99eAndreas Huber 85cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber void flush(); 86cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber void parsePES(ABitReader *br); 87cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 88cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber void onPayloadData( 89cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned PTS_DTS_flags, uint64_t PTS, uint64_t DTS, 90cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber const uint8_t *data, size_t size); 91cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 9282f7321b03eec1e40af9d681370f754ee0279582Andreas Huber void extractAACFrames(const sp<ABuffer> &buffer); 9382f7321b03eec1e40af9d681370f754ee0279582Andreas Huber 94cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber DISALLOW_EVIL_CONSTRUCTORS(Stream); 95cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber}; 96cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 97cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber//////////////////////////////////////////////////////////////////////////////// 98cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 99cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas HuberATSParser::Program::Program(unsigned programMapPID) 100cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber : mProgramMapPID(programMapPID) { 101cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber} 102cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 103cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huberbool ATSParser::Program::parsePID( 104cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned pid, unsigned payload_unit_start_indicator, 105cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber ABitReader *br) { 106cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (pid == mProgramMapPID) { 107cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (payload_unit_start_indicator) { 108cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned skip = br->getBits(8); 109cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber br->skipBits(skip * 8); 110cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 111cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 112cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber parseProgramMap(br); 113cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber return true; 114cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 115cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 116cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber ssize_t index = mStreams.indexOfKey(pid); 117cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (index < 0) { 118cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber return false; 119cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 120cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 121cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber mStreams.editValueAt(index)->parse( 122cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber payload_unit_start_indicator, br); 123cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 124cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber return true; 125cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber} 126cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 127cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Hubervoid ATSParser::Program::parseProgramMap(ABitReader *br) { 128cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned table_id = br->getBits(8); 1296e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber LOGV(" table_id = %u", table_id); 130cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(table_id, 0x02u); 131cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 1326e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber unsigned section_syntax_indicator = br->getBits(1); 1336e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber LOGV(" section_syntax_indicator = %u", section_syntax_indicator); 1346e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber CHECK_EQ(section_syntax_indicator, 1u); 135cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 136cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(br->getBits(1), 0u); 1376e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber MY_LOGV(" reserved = %u", br->getBits(2)); 138cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 139cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned section_length = br->getBits(12); 1406e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber LOGV(" section_length = %u", section_length); 141cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK((section_length & 0xc00) == 0); 142cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_LE(section_length, 1021u); 143cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 1446e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber MY_LOGV(" program_number = %u", br->getBits(16)); 1456e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber MY_LOGV(" reserved = %u", br->getBits(2)); 1466e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber MY_LOGV(" version_number = %u", br->getBits(5)); 1476e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber MY_LOGV(" current_next_indicator = %u", br->getBits(1)); 1486e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber MY_LOGV(" section_number = %u", br->getBits(8)); 1496e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber MY_LOGV(" last_section_number = %u", br->getBits(8)); 1506e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber MY_LOGV(" reserved = %u", br->getBits(3)); 1516e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber MY_LOGV(" PCR_PID = 0x%04x", br->getBits(13)); 1526e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber MY_LOGV(" reserved = %u", br->getBits(4)); 153cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 154cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned program_info_length = br->getBits(12); 1556e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber LOGV(" program_info_length = %u", program_info_length); 156cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK((program_info_length & 0xc00) == 0); 157cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 158cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber br->skipBits(program_info_length * 8); // skip descriptors 159cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 160cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber // infoBytesRemaining is the number of bytes that make up the 161cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber // variable length section of ES_infos. It does not include the 162cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber // final CRC. 163cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber size_t infoBytesRemaining = section_length - 9 - program_info_length - 4; 164cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 165cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber while (infoBytesRemaining > 0) { 166cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_GE(infoBytesRemaining, 5u); 167cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 168cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned streamType = br->getBits(8); 1696e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber LOGV(" stream_type = 0x%02x", streamType); 170cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 1716e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber MY_LOGV(" reserved = %u", br->getBits(3)); 172cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 173cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned elementaryPID = br->getBits(13); 1746e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber LOGV(" elementary_PID = 0x%04x", elementaryPID); 175cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 1766e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber MY_LOGV(" reserved = %u", br->getBits(4)); 177cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 178cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned ES_info_length = br->getBits(12); 1796e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber LOGV(" ES_info_length = %u", ES_info_length); 180cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK((ES_info_length & 0xc00) == 0); 181cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 182cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_GE(infoBytesRemaining - 5, ES_info_length); 183cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 184cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber#if 0 185cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber br->skipBits(ES_info_length * 8); // skip descriptors 186cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber#else 187cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned info_bytes_remaining = ES_info_length; 188cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber while (info_bytes_remaining >= 2) { 1896e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber MY_LOGV(" tag = 0x%02x", br->getBits(8)); 190cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 191cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned descLength = br->getBits(8); 1926e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber LOGV(" len = %u", descLength); 193cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 194cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_GE(info_bytes_remaining, 2 + descLength); 195cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 196cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber br->skipBits(descLength * 8); 197cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 198cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber info_bytes_remaining -= descLength + 2; 199cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 200cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(info_bytes_remaining, 0u); 201cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber#endif 202cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 203cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber ssize_t index = mStreams.indexOfKey(elementaryPID); 204cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber#if 0 // XXX revisit 205cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_LT(index, 0); 206cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber mStreams.add(elementaryPID, new Stream(elementaryPID, streamType)); 207cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber#else 208cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (index < 0) { 209cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber mStreams.add(elementaryPID, new Stream(elementaryPID, streamType)); 210cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 211cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber#endif 212cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 213cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber infoBytesRemaining -= 5 + ES_info_length; 214cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 215cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 216cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(infoBytesRemaining, 0u); 217cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 2186e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber MY_LOGV(" CRC = 0x%08x", br->getBits(32)); 219cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber} 220cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 221cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Hubersp<MediaSource> ATSParser::Program::getSource(SourceType type) { 222cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber for (size_t i = 0; i < mStreams.size(); ++i) { 223cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber sp<MediaSource> source = mStreams.editValueAt(i)->getSource(type); 224cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (source != NULL) { 225cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber return source; 226cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 227cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 228cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 229cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber return NULL; 230cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber} 231cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 232cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber//////////////////////////////////////////////////////////////////////////////// 233cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 234cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas HuberATSParser::Stream::Stream(unsigned elementaryPID, unsigned streamType) 235cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber : mElementaryPID(elementaryPID), 236cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber mStreamType(streamType), 23782f7321b03eec1e40af9d681370f754ee0279582Andreas Huber mBuffer(new ABuffer(128 * 1024)), 2386a63a939601645404fd98f58c19cc38ca818d99eAndreas Huber mPayloadStarted(false), 2396a63a939601645404fd98f58c19cc38ca818d99eAndreas Huber mQueue(streamType == 0x1b 2406a63a939601645404fd98f58c19cc38ca818d99eAndreas Huber ? ElementaryStreamQueue::H264 : ElementaryStreamQueue::AAC) { 241cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber mBuffer->setRange(0, 0); 242cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber} 243cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 244cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas HuberATSParser::Stream::~Stream() { 245cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber} 246cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 247cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Hubervoid ATSParser::Stream::parse( 248cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned payload_unit_start_indicator, ABitReader *br) { 249cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (payload_unit_start_indicator) { 250cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (mPayloadStarted) { 251cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber // Otherwise we run the danger of receiving the trailing bytes 252cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber // of a PES packet that we never saw the start of and assuming 253cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber // we have a a complete PES packet. 254cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 255cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber flush(); 256cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 257cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 258cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber mPayloadStarted = true; 259cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 260cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 261cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (!mPayloadStarted) { 262cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber return; 263cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 264cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 265cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber size_t payloadSizeBits = br->numBitsLeft(); 2666e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber CHECK((payloadSizeBits % 8) == 0); 267cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 268cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_LE(mBuffer->size() + payloadSizeBits / 8, mBuffer->capacity()); 269cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 270cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber memcpy(mBuffer->data() + mBuffer->size(), br->data(), payloadSizeBits / 8); 271cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber mBuffer->setRange(0, mBuffer->size() + payloadSizeBits / 8); 272cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber} 273cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 274cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Hubervoid ATSParser::Stream::parsePES(ABitReader *br) { 275cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned packet_startcode_prefix = br->getBits(24); 276cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 2776e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber LOGV("packet_startcode_prefix = 0x%08x", packet_startcode_prefix); 278cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 279cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(packet_startcode_prefix, 0x000001u); 280cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 281cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned stream_id = br->getBits(8); 2826e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber LOGV("stream_id = 0x%02x", stream_id); 283cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 284cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned PES_packet_length = br->getBits(16); 2856e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber LOGV("PES_packet_length = %u", PES_packet_length); 286cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 287cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (stream_id != 0xbc // program_stream_map 288cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber && stream_id != 0xbe // padding_stream 289cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber && stream_id != 0xbf // private_stream_2 290cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber && stream_id != 0xf0 // ECM 291cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber && stream_id != 0xf1 // EMM 292cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber && stream_id != 0xff // program_stream_directory 293cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber && stream_id != 0xf2 // DSMCC 294cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber && stream_id != 0xf8) { // H.222.1 type E 295cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(br->getBits(2), 2u); 296cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 2976e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber MY_LOGV("PES_scrambling_control = %u", br->getBits(2)); 2986e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber MY_LOGV("PES_priority = %u", br->getBits(1)); 2996e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber MY_LOGV("data_alignment_indicator = %u", br->getBits(1)); 3006e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber MY_LOGV("copyright = %u", br->getBits(1)); 3016e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber MY_LOGV("original_or_copy = %u", br->getBits(1)); 302cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 303cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned PTS_DTS_flags = br->getBits(2); 3046e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber LOGV("PTS_DTS_flags = %u", PTS_DTS_flags); 305cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 306cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned ESCR_flag = br->getBits(1); 3076e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber LOGV("ESCR_flag = %u", ESCR_flag); 308cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 309cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned ES_rate_flag = br->getBits(1); 3106e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber LOGV("ES_rate_flag = %u", ES_rate_flag); 311cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 312cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned DSM_trick_mode_flag = br->getBits(1); 3136e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber LOGV("DSM_trick_mode_flag = %u", DSM_trick_mode_flag); 314cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 315cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned additional_copy_info_flag = br->getBits(1); 3166e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber LOGV("additional_copy_info_flag = %u", additional_copy_info_flag); 317cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 3186e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber MY_LOGV("PES_CRC_flag = %u", br->getBits(1)); 3196e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber MY_LOGV("PES_extension_flag = %u", br->getBits(1)); 320cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 321cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned PES_header_data_length = br->getBits(8); 3226e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber LOGV("PES_header_data_length = %u", PES_header_data_length); 323cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 324cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned optional_bytes_remaining = PES_header_data_length; 325cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 326cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber uint64_t PTS = 0, DTS = 0; 327cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 328cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (PTS_DTS_flags == 2 || PTS_DTS_flags == 3) { 329cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_GE(optional_bytes_remaining, 5u); 330cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 331cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(br->getBits(4), PTS_DTS_flags); 332cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 333cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber PTS = ((uint64_t)br->getBits(3)) << 30; 334cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(br->getBits(1), 1u); 335cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber PTS |= ((uint64_t)br->getBits(15)) << 15; 336cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(br->getBits(1), 1u); 337cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber PTS |= br->getBits(15); 338cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(br->getBits(1), 1u); 339cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 3406e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber LOGV("PTS = %llu", PTS); 3416e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber // LOGI("PTS = %.2f secs", PTS / 90000.0f); 342cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 343cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber optional_bytes_remaining -= 5; 344cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 345cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (PTS_DTS_flags == 3) { 346cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_GE(optional_bytes_remaining, 5u); 347cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 348cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(br->getBits(4), 1u); 349cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 350cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber DTS = ((uint64_t)br->getBits(3)) << 30; 351cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(br->getBits(1), 1u); 352cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber DTS |= ((uint64_t)br->getBits(15)) << 15; 353cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(br->getBits(1), 1u); 354cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber DTS |= br->getBits(15); 355cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(br->getBits(1), 1u); 356cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 3576e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber LOGV("DTS = %llu", DTS); 358cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 359cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber optional_bytes_remaining -= 5; 360cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 361cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 362cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 363cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (ESCR_flag) { 364cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_GE(optional_bytes_remaining, 6u); 365cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 366cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber br->getBits(2); 367cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 368cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber uint64_t ESCR = ((uint64_t)br->getBits(3)) << 30; 369cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(br->getBits(1), 1u); 370cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber ESCR |= ((uint64_t)br->getBits(15)) << 15; 371cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(br->getBits(1), 1u); 372cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber ESCR |= br->getBits(15); 373cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(br->getBits(1), 1u); 374cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 3756e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber LOGV("ESCR = %llu", ESCR); 3766e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber MY_LOGV("ESCR_extension = %u", br->getBits(9)); 377cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 378cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(br->getBits(1), 1u); 379cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 380cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber optional_bytes_remaining -= 6; 381cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 382cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 383cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (ES_rate_flag) { 384cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_GE(optional_bytes_remaining, 3u); 385cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 386cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(br->getBits(1), 1u); 3876e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber MY_LOGV("ES_rate = %u", br->getBits(22)); 388cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(br->getBits(1), 1u); 389cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 390cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber optional_bytes_remaining -= 3; 391cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 392cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 393cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber br->skipBits(optional_bytes_remaining * 8); 394cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 395cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber // ES data follows. 396cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 397cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (PES_packet_length != 0) { 398cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_GE(PES_packet_length, PES_header_data_length + 3); 399cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 400cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned dataLength = 401cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber PES_packet_length - 3 - PES_header_data_length; 402cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 4030da4dab0a45a2bc1d95cbc6ef6a4850ed2569584Andreas Huber CHECK_GE(br->numBitsLeft(), dataLength * 8); 4040da4dab0a45a2bc1d95cbc6ef6a4850ed2569584Andreas Huber 4050da4dab0a45a2bc1d95cbc6ef6a4850ed2569584Andreas Huber onPayloadData( 4060da4dab0a45a2bc1d95cbc6ef6a4850ed2569584Andreas Huber PTS_DTS_flags, PTS, DTS, br->data(), dataLength); 407cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 408cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber br->skipBits(dataLength * 8); 409cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } else { 4100da4dab0a45a2bc1d95cbc6ef6a4850ed2569584Andreas Huber onPayloadData( 4110da4dab0a45a2bc1d95cbc6ef6a4850ed2569584Andreas Huber PTS_DTS_flags, PTS, DTS, 4120da4dab0a45a2bc1d95cbc6ef6a4850ed2569584Andreas Huber br->data(), br->numBitsLeft() / 8); 4130da4dab0a45a2bc1d95cbc6ef6a4850ed2569584Andreas Huber 414cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber size_t payloadSizeBits = br->numBitsLeft(); 415cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK((payloadSizeBits % 8) == 0); 416cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 4176e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber LOGV("There's %d bytes of payload.", payloadSizeBits / 8); 418cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 419cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } else if (stream_id == 0xbe) { // padding_stream 420cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_NE(PES_packet_length, 0u); 421cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber br->skipBits(PES_packet_length * 8); 422cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } else { 423cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_NE(PES_packet_length, 0u); 424cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber br->skipBits(PES_packet_length * 8); 425cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 426cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber} 427cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 428cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Hubervoid ATSParser::Stream::flush() { 429cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (mBuffer->size() == 0) { 430cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber return; 431cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 432cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 4336e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber LOGV("flushing stream 0x%04x size = %d", mElementaryPID, mBuffer->size()); 434cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 435cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber ABitReader br(mBuffer->data(), mBuffer->size()); 436cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber parsePES(&br); 437cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 438cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber mBuffer->setRange(0, 0); 439cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber} 440cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 441cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Hubervoid ATSParser::Stream::onPayloadData( 442cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned PTS_DTS_flags, uint64_t PTS, uint64_t DTS, 443cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber const uint8_t *data, size_t size) { 4446e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber LOGV("onPayloadData mStreamType=0x%02x", mStreamType); 445cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 446cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK(PTS_DTS_flags == 2 || PTS_DTS_flags == 3); 4476a63a939601645404fd98f58c19cc38ca818d99eAndreas Huber int64_t timeUs = (PTS * 100) / 9; 44882f7321b03eec1e40af9d681370f754ee0279582Andreas Huber 4496a63a939601645404fd98f58c19cc38ca818d99eAndreas Huber status_t err = mQueue.appendData(data, size, timeUs); 4506a63a939601645404fd98f58c19cc38ca818d99eAndreas Huber CHECK_EQ(err, (status_t)OK); 45182f7321b03eec1e40af9d681370f754ee0279582Andreas Huber 4526a63a939601645404fd98f58c19cc38ca818d99eAndreas Huber sp<ABuffer> accessUnit; 4536a63a939601645404fd98f58c19cc38ca818d99eAndreas Huber while ((accessUnit = mQueue.dequeueAccessUnit()) != NULL) { 4546a63a939601645404fd98f58c19cc38ca818d99eAndreas Huber if (mSource == NULL) { 4556a63a939601645404fd98f58c19cc38ca818d99eAndreas Huber sp<MetaData> meta = mQueue.getFormat(); 45682f7321b03eec1e40af9d681370f754ee0279582Andreas Huber 4576a63a939601645404fd98f58c19cc38ca818d99eAndreas Huber if (meta != NULL) { 4586a63a939601645404fd98f58c19cc38ca818d99eAndreas Huber LOGV("created source!"); 4596a63a939601645404fd98f58c19cc38ca818d99eAndreas Huber mSource = new AnotherPacketSource(meta); 4606a63a939601645404fd98f58c19cc38ca818d99eAndreas Huber mSource->queueAccessUnit(accessUnit); 46182f7321b03eec1e40af9d681370f754ee0279582Andreas Huber } 46282f7321b03eec1e40af9d681370f754ee0279582Andreas Huber } else { 4636a63a939601645404fd98f58c19cc38ca818d99eAndreas Huber mSource->queueAccessUnit(accessUnit); 46482f7321b03eec1e40af9d681370f754ee0279582Andreas Huber } 46582f7321b03eec1e40af9d681370f754ee0279582Andreas Huber } 46682f7321b03eec1e40af9d681370f754ee0279582Andreas Huber} 46782f7321b03eec1e40af9d681370f754ee0279582Andreas Huber 468cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Hubersp<MediaSource> ATSParser::Stream::getSource(SourceType type) { 469cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if ((type == AVC_VIDEO && mStreamType == 0x1b) 470cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber || (type == MPEG2ADTS_AUDIO && mStreamType == 0x0f)) { 471cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber return mSource; 472cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 473cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 474cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber return NULL; 475cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber} 476cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 477cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber//////////////////////////////////////////////////////////////////////////////// 478cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 479cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas HuberATSParser::ATSParser() { 480cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber} 481cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 482cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas HuberATSParser::~ATSParser() { 483cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber} 484cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 485cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Hubervoid ATSParser::feedTSPacket(const void *data, size_t size) { 486cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(size, kTSPacketSize); 487cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 488cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber ABitReader br((const uint8_t *)data, kTSPacketSize); 489cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber parseTS(&br); 490cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber} 491cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 492cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Hubervoid ATSParser::parseProgramAssociationTable(ABitReader *br) { 493cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned table_id = br->getBits(8); 4946e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber LOGV(" table_id = %u", table_id); 495cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(table_id, 0x00u); 496cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 497cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned section_syntax_indictor = br->getBits(1); 4986e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber LOGV(" section_syntax_indictor = %u", section_syntax_indictor); 499cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(section_syntax_indictor, 1u); 500cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 501cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(br->getBits(1), 0u); 5026e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber MY_LOGV(" reserved = %u", br->getBits(2)); 503cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 504cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned section_length = br->getBits(12); 5056e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber LOGV(" section_length = %u", section_length); 506cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK((section_length & 0xc00) == 0); 507cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 5086e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber MY_LOGV(" transport_stream_id = %u", br->getBits(16)); 5096e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber MY_LOGV(" reserved = %u", br->getBits(2)); 5106e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber MY_LOGV(" version_number = %u", br->getBits(5)); 5116e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber MY_LOGV(" current_next_indicator = %u", br->getBits(1)); 5126e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber MY_LOGV(" section_number = %u", br->getBits(8)); 5136e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber MY_LOGV(" last_section_number = %u", br->getBits(8)); 514cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 515cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber size_t numProgramBytes = (section_length - 5 /* header */ - 4 /* crc */); 516cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ((numProgramBytes % 4), 0u); 517cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 518cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber for (size_t i = 0; i < numProgramBytes / 4; ++i) { 519cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned program_number = br->getBits(16); 5206e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber LOGV(" program_number = %u", program_number); 521cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 5226e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber MY_LOGV(" reserved = %u", br->getBits(3)); 523cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 524cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (program_number == 0) { 5256e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber MY_LOGV(" network_PID = 0x%04x", br->getBits(13)); 526cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } else { 527cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned programMapPID = br->getBits(13); 528cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 5296e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber LOGV(" program_map_PID = 0x%04x", programMapPID); 530cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 531cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber mPrograms.push(new Program(programMapPID)); 532cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 533cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 534cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 5356e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber MY_LOGV(" CRC = 0x%08x", br->getBits(32)); 536cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber} 537cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 538cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Hubervoid ATSParser::parsePID( 539cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber ABitReader *br, unsigned PID, 540cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned payload_unit_start_indicator) { 541cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (PID == 0) { 542cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (payload_unit_start_indicator) { 543cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned skip = br->getBits(8); 544cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber br->skipBits(skip * 8); 545cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 546cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber parseProgramAssociationTable(br); 547cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber return; 548cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 549cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 550cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber bool handled = false; 551cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber for (size_t i = 0; i < mPrograms.size(); ++i) { 552cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (mPrograms.editItemAt(i)->parsePID( 553cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber PID, payload_unit_start_indicator, br)) { 554cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber handled = true; 555cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber break; 556cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 557cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 558cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 559cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (!handled) { 5606e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber LOGV("PID 0x%04x not handled.", PID); 561cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 562cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber} 563cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 564cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Hubervoid ATSParser::parseAdaptationField(ABitReader *br) { 565cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned adaptation_field_length = br->getBits(8); 566cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (adaptation_field_length > 0) { 567cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber br->skipBits(adaptation_field_length * 8); // XXX 568cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 569cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber} 570cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 571cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Hubervoid ATSParser::parseTS(ABitReader *br) { 5726e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber LOGV("---"); 573cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 574cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned sync_byte = br->getBits(8); 575cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(sync_byte, 0x47u); 576cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 5776e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber MY_LOGV("transport_error_indicator = %u", br->getBits(1)); 578cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 579cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned payload_unit_start_indicator = br->getBits(1); 5806e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber LOGV("payload_unit_start_indicator = %u", payload_unit_start_indicator); 581cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 5826e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber MY_LOGV("transport_priority = %u", br->getBits(1)); 583cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 584cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned PID = br->getBits(13); 5856e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber LOGV("PID = 0x%04x", PID); 586cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 5876e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber MY_LOGV("transport_scrambling_control = %u", br->getBits(2)); 588cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 589cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned adaptation_field_control = br->getBits(2); 5906e4c5c499999c04c2477b987f9e64f3ff2bf1a06Andreas Huber LOGV("adaptation_field_control = %u", adaptation_field_control); 591cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 5920da4dab0a45a2bc1d95cbc6ef6a4850ed2569584Andreas Huber unsigned continuity_counter = br->getBits(4); 5930da4dab0a45a2bc1d95cbc6ef6a4850ed2569584Andreas Huber LOGV("continuity_counter = %u", continuity_counter); 5940da4dab0a45a2bc1d95cbc6ef6a4850ed2569584Andreas Huber 5950da4dab0a45a2bc1d95cbc6ef6a4850ed2569584Andreas Huber // LOGI("PID = 0x%04x, continuity_counter = %u", PID, continuity_counter); 596cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 597cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (adaptation_field_control == 2 || adaptation_field_control == 3) { 598cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber parseAdaptationField(br); 599cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 600cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 601cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (adaptation_field_control == 1 || adaptation_field_control == 3) { 602cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber parsePID(br, PID, payload_unit_start_indicator); 603cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 604cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber} 605cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 606cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Hubersp<MediaSource> ATSParser::getSource(SourceType type) { 607cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber for (size_t i = 0; i < mPrograms.size(); ++i) { 608cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber sp<MediaSource> source = mPrograms.editItemAt(i)->getSource(type); 609cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 610cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (source != NULL) { 611cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber return source; 612cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 613cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 614cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 615cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber return NULL; 616cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber} 617cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 618cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber} // namespace android 619