ATSParser.cpp revision cda17c606b0fe3ccda4dc68a6d43882410ea2462
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 17cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber#include "ATSParser.h" 18cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 19cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber#include "ABitReader.h" 20cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber#include "AnotherPacketSource.h" 21cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 22cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber#include <media/stagefright/foundation/ABuffer.h> 23cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber#include <media/stagefright/foundation/ADebug.h> 24cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber#include <media/stagefright/foundation/AMessage.h> 25cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber#include <media/stagefright/foundation/hexdump.h> 26cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber#include <media/stagefright/MediaDefs.h> 27cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber#include <media/stagefright/MediaErrors.h> 28cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber#include <media/stagefright/MetaData.h> 29cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber#include <utils/KeyedVector.h> 30cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 31cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Hubernamespace android { 32cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 33cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huberstatic const size_t kTSPacketSize = 188; 34cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 35cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huberstruct ATSParser::Program : public RefBase { 36cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber Program(unsigned programMapPID); 37cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 38cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber bool parsePID( 39cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned pid, unsigned payload_unit_start_indicator, 40cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber ABitReader *br); 41cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 42cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber sp<MediaSource> getSource(SourceType type); 43cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 44cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huberprivate: 45cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned mProgramMapPID; 46cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber KeyedVector<unsigned, sp<Stream> > mStreams; 47cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 48cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber void parseProgramMap(ABitReader *br); 49cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 50cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber DISALLOW_EVIL_CONSTRUCTORS(Program); 51cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber}; 52cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 53cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huberstruct ATSParser::Stream : public RefBase { 54cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber Stream(unsigned elementaryPID, unsigned streamType); 55cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 56cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber void parse( 57cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned payload_unit_start_indicator, 58cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber ABitReader *br); 59cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 60cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber sp<MediaSource> getSource(SourceType type); 61cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 62cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huberprotected: 63cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber virtual ~Stream(); 64cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 65cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huberprivate: 66cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned mElementaryPID; 67cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned mStreamType; 68cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 69cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber sp<ABuffer> mBuffer; 70cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber sp<AnotherPacketSource> mSource; 71cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber bool mPayloadStarted; 72cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 73cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber void flush(); 74cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber void parsePES(ABitReader *br); 75cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 76cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber void onPayloadData( 77cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned PTS_DTS_flags, uint64_t PTS, uint64_t DTS, 78cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber const uint8_t *data, size_t size); 79cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 80cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber DISALLOW_EVIL_CONSTRUCTORS(Stream); 81cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber}; 82cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 83cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber//////////////////////////////////////////////////////////////////////////////// 84cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 85cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas HuberATSParser::Program::Program(unsigned programMapPID) 86cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber : mProgramMapPID(programMapPID) { 87cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber} 88cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 89cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huberbool ATSParser::Program::parsePID( 90cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned pid, unsigned payload_unit_start_indicator, 91cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber ABitReader *br) { 92cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (pid == mProgramMapPID) { 93cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (payload_unit_start_indicator) { 94cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned skip = br->getBits(8); 95cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber br->skipBits(skip * 8); 96cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 97cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 98cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber parseProgramMap(br); 99cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber return true; 100cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 101cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 102cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber ssize_t index = mStreams.indexOfKey(pid); 103cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (index < 0) { 104cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber return false; 105cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 106cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 107cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber mStreams.editValueAt(index)->parse( 108cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber payload_unit_start_indicator, br); 109cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 110cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber return true; 111cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber} 112cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 113cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Hubervoid ATSParser::Program::parseProgramMap(ABitReader *br) { 114cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned table_id = br->getBits(8); 115cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber LOG(VERBOSE) << " table_id = " << table_id; 116cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(table_id, 0x02u); 117cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 118cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned section_syntax_indictor = br->getBits(1); 119cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber LOG(VERBOSE) << " section_syntax_indictor = " << section_syntax_indictor; 120cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(section_syntax_indictor, 1u); 121cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 122cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(br->getBits(1), 0u); 123cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber LOG(VERBOSE) << " reserved = " << br->getBits(2); 124cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 125cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned section_length = br->getBits(12); 126cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber LOG(VERBOSE) << " section_length = " << section_length; 127cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK((section_length & 0xc00) == 0); 128cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_LE(section_length, 1021u); 129cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 130cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber LOG(VERBOSE) << " program_number = " << br->getBits(16); 131cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber LOG(VERBOSE) << " reserved = " << br->getBits(2); 132cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber LOG(VERBOSE) << " version_number = " << br->getBits(5); 133cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber LOG(VERBOSE) << " current_next_indicator = " << br->getBits(1); 134cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber LOG(VERBOSE) << " section_number = " << br->getBits(8); 135cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber LOG(VERBOSE) << " last_section_number = " << br->getBits(8); 136cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber LOG(VERBOSE) << " reserved = " << br->getBits(3); 137cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 138cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber LOG(VERBOSE) << " PCR_PID = " 139cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber << StringPrintf("0x%04x", br->getBits(13)); 140cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 141cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber LOG(VERBOSE) << " reserved = " << br->getBits(4); 142cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 143cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned program_info_length = br->getBits(12); 144cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber LOG(VERBOSE) << " program_info_length = " << program_info_length; 145cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK((program_info_length & 0xc00) == 0); 146cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 147cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber br->skipBits(program_info_length * 8); // skip descriptors 148cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 149cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber // infoBytesRemaining is the number of bytes that make up the 150cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber // variable length section of ES_infos. It does not include the 151cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber // final CRC. 152cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber size_t infoBytesRemaining = section_length - 9 - program_info_length - 4; 153cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 154cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber while (infoBytesRemaining > 0) { 155cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_GE(infoBytesRemaining, 5u); 156cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 157cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned streamType = br->getBits(8); 158cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber LOG(VERBOSE) << " stream_type = " 159cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber << StringPrintf("0x%02x", streamType); 160cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 161cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber LOG(VERBOSE) << " reserved = " << br->getBits(3); 162cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 163cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned elementaryPID = br->getBits(13); 164cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber LOG(VERBOSE) << " elementary_PID = " 165cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber << StringPrintf("0x%04x", elementaryPID); 166cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 167cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber LOG(VERBOSE) << " reserved = " << br->getBits(4); 168cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 169cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned ES_info_length = br->getBits(12); 170cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber LOG(VERBOSE) << " ES_info_length = " << ES_info_length; 171cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK((ES_info_length & 0xc00) == 0); 172cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 173cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_GE(infoBytesRemaining - 5, ES_info_length); 174cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 175cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber#if 0 176cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber br->skipBits(ES_info_length * 8); // skip descriptors 177cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber#else 178cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned info_bytes_remaining = ES_info_length; 179cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber while (info_bytes_remaining >= 2) { 180cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber LOG(VERBOSE) << " tag = " << StringPrintf("0x%02x", br->getBits(8)); 181cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 182cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned descLength = br->getBits(8); 183cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber LOG(VERBOSE) << " len = " << descLength; 184cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 185cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_GE(info_bytes_remaining, 2 + descLength); 186cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 187cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber br->skipBits(descLength * 8); 188cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 189cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber info_bytes_remaining -= descLength + 2; 190cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 191cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(info_bytes_remaining, 0u); 192cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber#endif 193cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 194cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber ssize_t index = mStreams.indexOfKey(elementaryPID); 195cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber#if 0 // XXX revisit 196cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_LT(index, 0); 197cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber mStreams.add(elementaryPID, new Stream(elementaryPID, streamType)); 198cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber#else 199cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (index < 0) { 200cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber mStreams.add(elementaryPID, new Stream(elementaryPID, streamType)); 201cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 202cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber#endif 203cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 204cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber infoBytesRemaining -= 5 + ES_info_length; 205cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 206cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 207cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(infoBytesRemaining, 0u); 208cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 209cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber LOG(VERBOSE) << " CRC = " << StringPrintf("0x%08x", br->getBits(32)); 210cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber} 211cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 212cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Hubersp<MediaSource> ATSParser::Program::getSource(SourceType type) { 213cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber for (size_t i = 0; i < mStreams.size(); ++i) { 214cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber sp<MediaSource> source = mStreams.editValueAt(i)->getSource(type); 215cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (source != NULL) { 216cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber return source; 217cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 218cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 219cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 220cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber return NULL; 221cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber} 222cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 223cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber//////////////////////////////////////////////////////////////////////////////// 224cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 225cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas HuberATSParser::Stream::Stream(unsigned elementaryPID, unsigned streamType) 226cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber : mElementaryPID(elementaryPID), 227cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber mStreamType(streamType), 228cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber mBuffer(new ABuffer(65536)), 229cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber mPayloadStarted(false) { 230cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber mBuffer->setRange(0, 0); 231cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber} 232cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 233cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas HuberATSParser::Stream::~Stream() { 234cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber} 235cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 236cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Hubervoid ATSParser::Stream::parse( 237cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned payload_unit_start_indicator, ABitReader *br) { 238cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (payload_unit_start_indicator) { 239cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (mPayloadStarted) { 240cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber // Otherwise we run the danger of receiving the trailing bytes 241cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber // of a PES packet that we never saw the start of and assuming 242cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber // we have a a complete PES packet. 243cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 244cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber flush(); 245cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 246cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 247cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber mPayloadStarted = true; 248cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 249cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 250cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (!mPayloadStarted) { 251cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber return; 252cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 253cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 254cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber size_t payloadSizeBits = br->numBitsLeft(); 255cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(payloadSizeBits % 8, 0u); 256cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 257cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_LE(mBuffer->size() + payloadSizeBits / 8, mBuffer->capacity()); 258cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 259cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber memcpy(mBuffer->data() + mBuffer->size(), br->data(), payloadSizeBits / 8); 260cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber mBuffer->setRange(0, mBuffer->size() + payloadSizeBits / 8); 261cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber} 262cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 263cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Hubervoid ATSParser::Stream::parsePES(ABitReader *br) { 264cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned packet_startcode_prefix = br->getBits(24); 265cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 266cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber LOG(VERBOSE) << "packet_startcode_prefix = " 267cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber << StringPrintf("0x%08x", packet_startcode_prefix); 268cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 269cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(packet_startcode_prefix, 0x000001u); 270cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 271cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned stream_id = br->getBits(8); 272cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber LOG(VERBOSE) << "stream_id = " << StringPrintf("0x%02x", stream_id); 273cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 274cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned PES_packet_length = br->getBits(16); 275cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber LOG(VERBOSE) << "PES_packet_length = " << PES_packet_length; 276cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 277cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (stream_id != 0xbc // program_stream_map 278cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber && stream_id != 0xbe // padding_stream 279cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber && stream_id != 0xbf // private_stream_2 280cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber && stream_id != 0xf0 // ECM 281cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber && stream_id != 0xf1 // EMM 282cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber && stream_id != 0xff // program_stream_directory 283cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber && stream_id != 0xf2 // DSMCC 284cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber && stream_id != 0xf8) { // H.222.1 type E 285cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(br->getBits(2), 2u); 286cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 287cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber LOG(VERBOSE) << "PES_scrambling_control = " << br->getBits(2); 288cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber LOG(VERBOSE) << "PES_priority = " << br->getBits(1); 289cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber LOG(VERBOSE) << "data_alignment_indicator = " << br->getBits(1); 290cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber LOG(VERBOSE) << "copyright = " << br->getBits(1); 291cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber LOG(VERBOSE) << "original_or_copy = " << br->getBits(1); 292cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 293cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned PTS_DTS_flags = br->getBits(2); 294cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber LOG(VERBOSE) << "PTS_DTS_flags = " << PTS_DTS_flags; 295cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 296cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned ESCR_flag = br->getBits(1); 297cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber LOG(VERBOSE) << "ESCR_flag = " << ESCR_flag; 298cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 299cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned ES_rate_flag = br->getBits(1); 300cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber LOG(VERBOSE) << "ES_rate_flag = " << ES_rate_flag; 301cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 302cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned DSM_trick_mode_flag = br->getBits(1); 303cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber LOG(VERBOSE) << "DSM_trick_mode_flag = " << DSM_trick_mode_flag; 304cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 305cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned additional_copy_info_flag = br->getBits(1); 306cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber LOG(VERBOSE) << "additional_copy_info_flag = " 307cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber << additional_copy_info_flag; 308cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 309cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber LOG(VERBOSE) << "PES_CRC_flag = " << br->getBits(1); 310cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber LOG(VERBOSE) << "PES_extension_flag = " << br->getBits(1); 311cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 312cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned PES_header_data_length = br->getBits(8); 313cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber LOG(VERBOSE) << "PES_header_data_length = " << PES_header_data_length; 314cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 315cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned optional_bytes_remaining = PES_header_data_length; 316cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 317cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber uint64_t PTS = 0, DTS = 0; 318cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 319cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (PTS_DTS_flags == 2 || PTS_DTS_flags == 3) { 320cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_GE(optional_bytes_remaining, 5u); 321cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 322cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(br->getBits(4), PTS_DTS_flags); 323cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 324cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber PTS = ((uint64_t)br->getBits(3)) << 30; 325cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(br->getBits(1), 1u); 326cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber PTS |= ((uint64_t)br->getBits(15)) << 15; 327cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(br->getBits(1), 1u); 328cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber PTS |= br->getBits(15); 329cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(br->getBits(1), 1u); 330cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 331cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber LOG(VERBOSE) << "PTS = " << PTS; 332cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber // LOG(INFO) << "PTS = " << PTS / 90000.0f << " secs"; 333cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 334cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber optional_bytes_remaining -= 5; 335cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 336cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (PTS_DTS_flags == 3) { 337cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_GE(optional_bytes_remaining, 5u); 338cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 339cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(br->getBits(4), 1u); 340cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 341cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber DTS = ((uint64_t)br->getBits(3)) << 30; 342cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(br->getBits(1), 1u); 343cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber DTS |= ((uint64_t)br->getBits(15)) << 15; 344cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(br->getBits(1), 1u); 345cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber DTS |= br->getBits(15); 346cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(br->getBits(1), 1u); 347cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 348cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber LOG(VERBOSE) << "DTS = " << DTS; 349cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 350cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber optional_bytes_remaining -= 5; 351cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 352cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 353cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 354cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (ESCR_flag) { 355cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_GE(optional_bytes_remaining, 6u); 356cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 357cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber br->getBits(2); 358cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 359cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber uint64_t ESCR = ((uint64_t)br->getBits(3)) << 30; 360cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(br->getBits(1), 1u); 361cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber ESCR |= ((uint64_t)br->getBits(15)) << 15; 362cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(br->getBits(1), 1u); 363cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber ESCR |= br->getBits(15); 364cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(br->getBits(1), 1u); 365cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 366cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber LOG(VERBOSE) << "ESCR = " << ESCR; 367cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber LOG(VERBOSE) << "ESCR_extension = " << br->getBits(9); 368cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 369cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(br->getBits(1), 1u); 370cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 371cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber optional_bytes_remaining -= 6; 372cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 373cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 374cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (ES_rate_flag) { 375cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_GE(optional_bytes_remaining, 3u); 376cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 377cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(br->getBits(1), 1u); 378cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber LOG(VERBOSE) << "ES_rate = " << br->getBits(22); 379cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(br->getBits(1), 1u); 380cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 381cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber optional_bytes_remaining -= 3; 382cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 383cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 384cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber br->skipBits(optional_bytes_remaining * 8); 385cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 386cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber // ES data follows. 387cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 388cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber onPayloadData( 389cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber PTS_DTS_flags, PTS, DTS, 390cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber br->data(), br->numBitsLeft() / 8); 391cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 392cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (PES_packet_length != 0) { 393cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_GE(PES_packet_length, PES_header_data_length + 3); 394cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 395cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned dataLength = 396cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber PES_packet_length - 3 - PES_header_data_length; 397cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 398cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(br->numBitsLeft(), dataLength * 8); 399cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 400cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber br->skipBits(dataLength * 8); 401cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } else { 402cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber size_t payloadSizeBits = br->numBitsLeft(); 403cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK((payloadSizeBits % 8) == 0); 404cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 405cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber LOG(VERBOSE) << "There's " << (payloadSizeBits / 8) 406cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber << " bytes of payload."; 407cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 408cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } else if (stream_id == 0xbe) { // padding_stream 409cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_NE(PES_packet_length, 0u); 410cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber br->skipBits(PES_packet_length * 8); 411cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } else { 412cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_NE(PES_packet_length, 0u); 413cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber br->skipBits(PES_packet_length * 8); 414cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 415cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber} 416cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 417cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Hubervoid ATSParser::Stream::flush() { 418cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (mBuffer->size() == 0) { 419cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber return; 420cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 421cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 422cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber LOG(VERBOSE) << "flushing stream " 423cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber << StringPrintf("0x%04x", mElementaryPID) 424cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber << " size = " << mBuffer->size(); 425cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 426cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber ABitReader br(mBuffer->data(), mBuffer->size()); 427cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber parsePES(&br); 428cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 429cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber mBuffer->setRange(0, 0); 430cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber} 431cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 432cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huberstatic sp<ABuffer> FindNAL( 433cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber const uint8_t *data, size_t size, unsigned nalType, 434cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber size_t *stopOffset) { 435cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber bool foundStart = false; 436cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber size_t startOffset = 0; 437cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 438cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber size_t offset = 0; 439cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber for (;;) { 440cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber while (offset + 3 < size 441cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber && memcmp("\x00\x00\x00\x01", &data[offset], 4)) { 442cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber ++offset; 443cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 444cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 445cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (foundStart) { 446cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber size_t nalSize; 447cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (offset + 3 >= size) { 448cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber nalSize = size - startOffset; 449cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } else { 450cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber nalSize = offset - startOffset; 451cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 452cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 453cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber sp<ABuffer> nal = new ABuffer(nalSize); 454cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber memcpy(nal->data(), &data[startOffset], nalSize); 455cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 456cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (stopOffset != NULL) { 457cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber *stopOffset = startOffset + nalSize; 458cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 459cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 460cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber return nal; 461cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 462cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 463cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (offset + 4 >= size) { 464cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber return NULL; 465cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 466cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 467cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if ((data[offset + 4] & 0x1f) == nalType) { 468cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber foundStart = true; 469cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber startOffset = offset + 4; 470cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 471cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 472cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber offset += 4; 473cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 474cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber} 475cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 476cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huberstatic unsigned parseUE(ABitReader *br) { 477cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned numZeroes = 0; 478cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber while (br->getBits(1) == 0) { 479cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber ++numZeroes; 480cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 481cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 482cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned x = br->getBits(numZeroes); 483cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 484cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber return x + (1u << numZeroes) - 1; 485cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber} 486cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 487cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber// Determine video dimensions from the sequence parameterset. 488cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huberstatic void FindDimensions( 489cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber const sp<ABuffer> seqParamSet, int32_t *width, int32_t *height) { 490cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber ABitReader br(seqParamSet->data() + 1, seqParamSet->size() - 1); 491cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 492cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned profile_idc = br.getBits(8); 493cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber br.skipBits(16); 494cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber parseUE(&br); // seq_parameter_set_id 495cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 496cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (profile_idc == 100 || profile_idc == 110 497cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber || profile_idc == 122 || profile_idc == 144) { 498cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber TRESPASS(); 499cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 500cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 501cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber parseUE(&br); // log2_max_frame_num_minus4 502cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned pic_order_cnt_type = parseUE(&br); 503cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 504cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (pic_order_cnt_type == 0) { 505cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber parseUE(&br); // log2_max_pic_order_cnt_lsb_minus4 506cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } else if (pic_order_cnt_type == 1) { 507cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber br.getBits(1); // delta_pic_order_always_zero_flag 508cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber parseUE(&br); // offset_for_non_ref_pic 509cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber parseUE(&br); // offset_for_top_to_bottom_field 510cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 511cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned num_ref_frames_in_pic_order_cnt_cycle = parseUE(&br); 512cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber for (unsigned i = 0; i < num_ref_frames_in_pic_order_cnt_cycle; ++i) { 513cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber parseUE(&br); // offset_for_ref_frame 514cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 515cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 516cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 517cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber parseUE(&br); // num_ref_frames 518cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber br.getBits(1); // gaps_in_frame_num_value_allowed_flag 519cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 520cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned pic_width_in_mbs_minus1 = parseUE(&br); 521cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned pic_height_in_map_units_minus1 = parseUE(&br); 522cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned frame_mbs_only_flag = br.getBits(1); 523cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 524cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber *width = pic_width_in_mbs_minus1 * 16 + 16; 525cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 526cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber *height = (2 - frame_mbs_only_flag) 527cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber * (pic_height_in_map_units_minus1 * 16 + 16); 528cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber} 529cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 530cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huberstatic sp<ABuffer> MakeAVCCodecSpecificData( 531cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber const sp<ABuffer> &buffer, int32_t *width, int32_t *height) { 532cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber const uint8_t *data = buffer->data(); 533cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber size_t size = buffer->size(); 534cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 535cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber sp<ABuffer> seqParamSet = FindNAL(data, size, 7, NULL); 536cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (seqParamSet == NULL) { 537cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber return NULL; 538cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 539cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 540cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber FindDimensions(seqParamSet, width, height); 541cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 542cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber size_t stopOffset; 543cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber sp<ABuffer> picParamSet = FindNAL(data, size, 8, &stopOffset); 544cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK(picParamSet != NULL); 545cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 546cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber buffer->setRange(stopOffset, size - stopOffset); 547cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber LOG(INFO) << "buffer has " << buffer->size() << " bytes left."; 548cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 549cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber size_t csdSize = 550cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 1 + 3 + 1 + 1 551cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber + 2 * 1 + seqParamSet->size() 552cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber + 1 + 2 * 1 + picParamSet->size(); 553cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 554cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber sp<ABuffer> csd = new ABuffer(csdSize); 555cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber uint8_t *out = csd->data(); 556cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 557cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber *out++ = 0x01; // configurationVersion 558cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber memcpy(out, seqParamSet->data() + 1, 3); // profile/level... 559cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber out += 3; 560cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber *out++ = (0x3f << 2) | 1; // lengthSize == 2 bytes 561cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber *out++ = 0xe0 | 1; 562cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 563cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber *out++ = seqParamSet->size() >> 8; 564cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber *out++ = seqParamSet->size() & 0xff; 565cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber memcpy(out, seqParamSet->data(), seqParamSet->size()); 566cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber out += seqParamSet->size(); 567cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 568cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber *out++ = 1; 569cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 570cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber *out++ = picParamSet->size() >> 8; 571cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber *out++ = picParamSet->size() & 0xff; 572cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber memcpy(out, picParamSet->data(), picParamSet->size()); 573cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 574cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber return csd; 575cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber} 576cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 577cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huberstatic bool getNextNALUnit( 578cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber const uint8_t **_data, size_t *_size, 579cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber const uint8_t **nalStart, size_t *nalSize) { 580cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber const uint8_t *data = *_data; 581cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber size_t size = *_size; 582cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 583cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber *nalStart = NULL; 584cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber *nalSize = 0; 585cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 586cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (size == 0) { 587cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber return false; 588cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 589cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 590cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber size_t offset = 0; 591cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber for (;;) { 592cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_LT(offset + 2, size); 593cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 594cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (!memcmp("\x00\x00\x01", &data[offset], 3)) { 595cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber break; 596cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 597cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 598cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ((unsigned)data[offset], 0x00u); 599cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber ++offset; 600cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 601cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 602cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber offset += 3; 603cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber size_t startOffset = offset; 604cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 605cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber while (offset + 2 < size 606cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber && memcmp("\x00\x00\x00", &data[offset], 3) 607cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber && memcmp("\x00\x00\x01", &data[offset], 3)) { 608cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber ++offset; 609cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 610cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 611cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (offset + 2 >= size) { 612cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber *nalStart = &data[startOffset]; 613cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber *nalSize = size - startOffset; 614cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 615cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber *_data = NULL; 616cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber *_size = 0; 617cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 618cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber return true; 619cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 620cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 621cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber size_t endOffset = offset; 622cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 623cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber while (offset + 2 < size && memcmp("\x00\x00\x01", &data[offset], 3)) { 624cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ((unsigned)data[offset], 0x00u); 625cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber ++offset; 626cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 627cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 628cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_LT(offset + 2, size); 629cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 630cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber *nalStart = &data[startOffset]; 631cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber *nalSize = endOffset - startOffset; 632cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 633cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber *_data = &data[offset]; 634cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber *_size = size - offset; 635cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 636cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber return true; 637cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber} 638cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 639cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Hubersp<ABuffer> MakeCleanAVCData(const uint8_t *data, size_t size) { 640cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber const uint8_t *tmpData = data; 641cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber size_t tmpSize = size; 642cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 643cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber size_t totalSize = 0; 644cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber const uint8_t *nalStart; 645cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber size_t nalSize; 646cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber while (getNextNALUnit(&tmpData, &tmpSize, &nalStart, &nalSize)) { 647cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber totalSize += 4 + nalSize; 648cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 649cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 650cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber sp<ABuffer> buffer = new ABuffer(totalSize); 651cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber size_t offset = 0; 652cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber while (getNextNALUnit(&data, &size, &nalStart, &nalSize)) { 653cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber memcpy(buffer->data() + offset, "\x00\x00\x00\x01", 4); 654cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber memcpy(buffer->data() + offset + 4, nalStart, nalSize); 655cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 656cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber offset += 4 + nalSize; 657cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 658cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 659cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber return buffer; 660cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber} 661cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 662cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huberstatic sp<ABuffer> FindMPEG2ADTSConfig( 663cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber const sp<ABuffer> &buffer, int32_t *sampleRate, int32_t *channelCount) { 664cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber ABitReader br(buffer->data(), buffer->size()); 665cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 666cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(br.getBits(12), 0xfffu); 667cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(br.getBits(1), 0u); 668cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(br.getBits(2), 0u); 669cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber br.getBits(1); // protection_absent 670cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned profile = br.getBits(2); 671cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber LOG(INFO) << "profile = " << profile; 672cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_NE(profile, 3u); 673cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned sampling_freq_index = br.getBits(4); 674cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber br.getBits(1); // private_bit 675cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned channel_configuration = br.getBits(3); 676cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_NE(channel_configuration, 0u); 677cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 678cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber LOG(INFO) << "sampling_freq_index = " << sampling_freq_index; 679cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber LOG(INFO) << "channel_configuration = " << channel_configuration; 680cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 681cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_LE(sampling_freq_index, 11u); 682cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber static const int32_t kSamplingFreq[] = { 683cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 96000, 88200, 64000, 48000, 44100, 32000, 24000, 22050, 684cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 16000, 12000, 11025, 8000 685cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber }; 686cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber *sampleRate = kSamplingFreq[sampling_freq_index]; 687cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 688cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber *channelCount = channel_configuration; 689cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 690cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber static const uint8_t kStaticESDS[] = { 691cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 0x03, 22, 692cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 0x00, 0x00, // ES_ID 693cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 0x00, // streamDependenceFlag, URL_Flag, OCRstreamFlag 694cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 695cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 0x04, 17, 696cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 0x40, // Audio ISO/IEC 14496-3 697cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 0x00, 0x00, 0x00, 0x00, 698cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 0x00, 0x00, 0x00, 0x00, 699cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 0x00, 0x00, 0x00, 0x00, 700cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 701cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 0x05, 2, 702cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber // AudioSpecificInfo follows 703cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 704cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber // oooo offf fccc c000 705cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber // o - audioObjectType 706cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber // f - samplingFreqIndex 707cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber // c - channelConfig 708cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber }; 709cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber sp<ABuffer> csd = new ABuffer(sizeof(kStaticESDS) + 2); 710cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber memcpy(csd->data(), kStaticESDS, sizeof(kStaticESDS)); 711cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 712cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber csd->data()[sizeof(kStaticESDS)] = 713cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber ((profile + 1) << 3) | (sampling_freq_index >> 1); 714cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 715cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber csd->data()[sizeof(kStaticESDS) + 1] = 716cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber ((sampling_freq_index << 7) & 0x80) | (channel_configuration << 3); 717cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 718cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber hexdump(csd->data(), csd->size()); 719cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber return csd; 720cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber} 721cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 722cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Hubervoid ATSParser::Stream::onPayloadData( 723cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned PTS_DTS_flags, uint64_t PTS, uint64_t DTS, 724cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber const uint8_t *data, size_t size) { 725cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber LOG(VERBOSE) << "onPayloadData mStreamType=" 726cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber << StringPrintf("0x%02x", mStreamType); 727cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 728cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber sp<ABuffer> buffer; 729cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 730cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (mStreamType == 0x1b) { 731cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber buffer = MakeCleanAVCData(data, size); 732cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } else { 733cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber // hexdump(data, size); 734cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 735cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber buffer = new ABuffer(size); 736cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber memcpy(buffer->data(), data, size); 737cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 738cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 739cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (mSource == NULL) { 740cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber sp<MetaData> meta = new MetaData; 741cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 742cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (mStreamType == 0x1b) { 743cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber meta->setCString(kKeyMIMEType, MEDIA_MIMETYPE_VIDEO_AVC); 744cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 745cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber int32_t width, height; 746cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber sp<ABuffer> csd = MakeAVCCodecSpecificData(buffer, &width, &height); 747cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 748cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (csd == NULL) { 749cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber return; 750cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 751cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 752cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber meta->setData(kKeyAVCC, 0, csd->data(), csd->size()); 753cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber meta->setInt32(kKeyWidth, width); 754cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber meta->setInt32(kKeyHeight, height); 755cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } else { 756cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(mStreamType, 0x0fu); 757cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 758cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber meta->setCString(kKeyMIMEType, MEDIA_MIMETYPE_AUDIO_AAC); 759cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 760cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber int32_t sampleRate, channelCount; 761cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber sp<ABuffer> csd = 762cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber FindMPEG2ADTSConfig(buffer, &sampleRate, &channelCount); 763cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 764cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber LOG(INFO) << "sampleRate = " << sampleRate; 765cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber LOG(INFO) << "channelCount = " << channelCount; 766cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 767cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber meta->setInt32(kKeySampleRate, sampleRate); 768cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber meta->setInt32(kKeyChannelCount, channelCount); 769cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 770cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber meta->setData(kKeyESDS, 0, csd->data(), csd->size()); 771cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 772cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 773cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber LOG(INFO) << "created source!"; 774cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber mSource = new AnotherPacketSource(meta); 775cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 776cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber // fall through 777cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 778cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 779cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK(PTS_DTS_flags == 2 || PTS_DTS_flags == 3); 780cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber buffer->meta()->setInt64("time", (PTS * 100) / 9); 781cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 782cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (mStreamType == 0x0f) { 783cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber // WHY??? 784cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber buffer->setRange(7, buffer->size() - 7); 785cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 786cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 787cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber mSource->queueAccessUnit(buffer); 788cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber} 789cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 790cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Hubersp<MediaSource> ATSParser::Stream::getSource(SourceType type) { 791cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if ((type == AVC_VIDEO && mStreamType == 0x1b) 792cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber || (type == MPEG2ADTS_AUDIO && mStreamType == 0x0f)) { 793cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber return mSource; 794cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 795cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 796cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber return NULL; 797cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber} 798cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 799cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber//////////////////////////////////////////////////////////////////////////////// 800cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 801cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas HuberATSParser::ATSParser() { 802cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber} 803cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 804cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas HuberATSParser::~ATSParser() { 805cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber} 806cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 807cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Hubervoid ATSParser::feedTSPacket(const void *data, size_t size) { 808cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(size, kTSPacketSize); 809cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 810cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber ABitReader br((const uint8_t *)data, kTSPacketSize); 811cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber parseTS(&br); 812cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber} 813cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 814cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Hubervoid ATSParser::parseProgramAssociationTable(ABitReader *br) { 815cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned table_id = br->getBits(8); 816cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber LOG(VERBOSE) << " table_id = " << table_id; 817cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(table_id, 0x00u); 818cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 819cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned section_syntax_indictor = br->getBits(1); 820cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber LOG(VERBOSE) << " section_syntax_indictor = " << section_syntax_indictor; 821cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(section_syntax_indictor, 1u); 822cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 823cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(br->getBits(1), 0u); 824cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber LOG(VERBOSE) << " reserved = " << br->getBits(2); 825cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 826cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned section_length = br->getBits(12); 827cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber LOG(VERBOSE) << " section_length = " << section_length; 828cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK((section_length & 0xc00) == 0); 829cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 830cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber LOG(VERBOSE) << " transport_stream_id = " << br->getBits(16); 831cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber LOG(VERBOSE) << " reserved = " << br->getBits(2); 832cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber LOG(VERBOSE) << " version_number = " << br->getBits(5); 833cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber LOG(VERBOSE) << " current_next_indicator = " << br->getBits(1); 834cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber LOG(VERBOSE) << " section_number = " << br->getBits(8); 835cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber LOG(VERBOSE) << " last_section_number = " << br->getBits(8); 836cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 837cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber size_t numProgramBytes = (section_length - 5 /* header */ - 4 /* crc */); 838cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ((numProgramBytes % 4), 0u); 839cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 840cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber for (size_t i = 0; i < numProgramBytes / 4; ++i) { 841cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned program_number = br->getBits(16); 842cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber LOG(VERBOSE) << " program_number = " << program_number; 843cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 844cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber LOG(VERBOSE) << " reserved = " << br->getBits(3); 845cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 846cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (program_number == 0) { 847cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber LOG(VERBOSE) << " network_PID = " 848cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber << StringPrintf("0x%04x", br->getBits(13)); 849cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } else { 850cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned programMapPID = br->getBits(13); 851cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 852cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber LOG(VERBOSE) << " program_map_PID = " 853cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber << StringPrintf("0x%04x", programMapPID); 854cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 855cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber mPrograms.push(new Program(programMapPID)); 856cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 857cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 858cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 859cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber LOG(VERBOSE) << " CRC = " << StringPrintf("0x%08x", br->getBits(32)); 860cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber} 861cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 862cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Hubervoid ATSParser::parsePID( 863cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber ABitReader *br, unsigned PID, 864cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned payload_unit_start_indicator) { 865cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (PID == 0) { 866cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (payload_unit_start_indicator) { 867cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned skip = br->getBits(8); 868cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber br->skipBits(skip * 8); 869cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 870cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber parseProgramAssociationTable(br); 871cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber return; 872cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 873cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 874cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber bool handled = false; 875cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber for (size_t i = 0; i < mPrograms.size(); ++i) { 876cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (mPrograms.editItemAt(i)->parsePID( 877cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber PID, payload_unit_start_indicator, br)) { 878cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber handled = true; 879cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber break; 880cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 881cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 882cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 883cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (!handled) { 884cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber LOG(WARNING) << "PID " << StringPrintf("0x%04x", PID) 885cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber << " not handled."; 886cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 887cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber} 888cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 889cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Hubervoid ATSParser::parseAdaptationField(ABitReader *br) { 890cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned adaptation_field_length = br->getBits(8); 891cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (adaptation_field_length > 0) { 892cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber br->skipBits(adaptation_field_length * 8); // XXX 893cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 894cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber} 895cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 896cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Hubervoid ATSParser::parseTS(ABitReader *br) { 897cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber LOG(VERBOSE) << "---"; 898cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 899cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned sync_byte = br->getBits(8); 900cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber CHECK_EQ(sync_byte, 0x47u); 901cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 902cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber LOG(VERBOSE) << "transport_error_indicator = " << br->getBits(1); 903cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 904cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned payload_unit_start_indicator = br->getBits(1); 905cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber LOG(VERBOSE) << "payload_unit_start_indicator = " 906cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber << payload_unit_start_indicator; 907cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 908cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber LOG(VERBOSE) << "transport_priority = " << br->getBits(1); 909cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 910cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned PID = br->getBits(13); 911cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber LOG(VERBOSE) << "PID = " << StringPrintf("0x%04x", PID); 912cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 913cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber LOG(VERBOSE) << "transport_scrambling_control = " << br->getBits(2); 914cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 915cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber unsigned adaptation_field_control = br->getBits(2); 916cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber LOG(VERBOSE) << "adaptation_field_control = " << adaptation_field_control; 917cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 918cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber LOG(VERBOSE) << "continuity_counter = " << br->getBits(4); 919cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 920cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (adaptation_field_control == 2 || adaptation_field_control == 3) { 921cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber parseAdaptationField(br); 922cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 923cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 924cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (adaptation_field_control == 1 || adaptation_field_control == 3) { 925cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber parsePID(br, PID, payload_unit_start_indicator); 926cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 927cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber} 928cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 929cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Hubersp<MediaSource> ATSParser::getSource(SourceType type) { 930cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber for (size_t i = 0; i < mPrograms.size(); ++i) { 931cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber sp<MediaSource> source = mPrograms.editItemAt(i)->getSource(type); 932cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 933cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber if (source != NULL) { 934cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber return source; 935cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 936cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber } 937cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 938cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber return NULL; 939cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber} 940cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber 941cda17c606b0fe3ccda4dc68a6d43882410ea2462Andreas Huber} // namespace android 942