Google Git
Sign in
android / device / google / cuttlefish / f081441ba47ed3f9391ef9236463d8fe6043db19 / . / host / frontend / gcastv2 / libandroid / ATSParser.cpp
blob: ac061bfbe15f985151f99ed52039d2102fdd895d [file] [log] [blame]
/*
* Copyright (C) 2010 The Android Open Source Project
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
//#define LOG_NDEBUG 0
#define LOG_TAG "ATSParser"
#include <utils/Log.h>
#include <media/stagefright/ATSParser.h>
#include "ESQueue.h"
#include <media/stagefright/avc_utils.h>
#include <media/stagefright/foundation/ABitReader.h>
#include <media/stagefright/foundation/ABuffer.h>
#include <media/stagefright/foundation/ADebug.h>
#include <media/stagefright/foundation/AMessage.h>
#include <media/stagefright/foundation/hexdump.h>
#include <media/stagefright/AnotherPacketSource.h>
#include <media/stagefright/MediaDefs.h>
#include <media/stagefright/MediaErrors.h>
#include <media/stagefright/MetaData.h>
#include <media/stagefright/Utils.h>
#include <map>
#include <vector>
namespace android {
// I want the expression "y" evaluated even if verbose logging is off.
#define MY_LOGV(x, y) \
do { unsigned tmp = y; ALOGV(x, tmp); tmp = tmp; } while (0)
static const size_t kTSPacketSize = 188;
struct ATSParser::Program : public RefBase {
Program(ATSParser *parser, unsigned programNumber, unsigned programMapPID);
bool parsePSISection(
unsigned pid, ABitReader *br, status_t *err);
bool parsePID(
unsigned pid, unsigned continuity_counter,
unsigned payload_unit_start_indicator,
ABitReader *br, status_t *err);
void signalDiscontinuity(
DiscontinuityType type, const sp<AMessage> &extra);
void signalEOS(status_t finalResult);
sp<AnotherPacketSource> getSource(SourceType type);
int64_t convertPTSToTimestamp(uint64_t PTS);
bool PTSTimeDeltaEstablished() const {
return mFirstPTSValid;
}
unsigned number() const { return mProgramNumber; }
void updateProgramMapPID(unsigned programMapPID) {
mProgramMapPID = programMapPID;
}
unsigned programMapPID() const {
return mProgramMapPID;
}
uint32_t parserFlags() const {
return mParser->mFlags;
}
private:
ATSParser *mParser;
unsigned mProgramNumber;
unsigned mProgramMapPID;
std::map<unsigned, sp<Stream>> mStreams;
bool mFirstPTSValid;
uint64_t mFirstPTS;
status_t parseProgramMap(ABitReader *br);
DISALLOW_EVIL_CONSTRUCTORS(Program);
};
struct ATSParser::Stream : public RefBase {
Stream(Program *program,
unsigned elementaryPID,
unsigned streamType,
unsigned PCR_PID);
unsigned type() const { return mStreamType; }
unsigned pid() const { return mElementaryPID; }
void setPID(unsigned pid) { mElementaryPID = pid; }
status_t parse(
unsigned continuity_counter,
unsigned payload_unit_start_indicator,
ABitReader *br);
void signalDiscontinuity(
DiscontinuityType type, const sp<AMessage> &extra);
void signalEOS(status_t finalResult);
sp<AnotherPacketSource> getSource(SourceType type);
protected:
virtual ~Stream();
private:
Program *mProgram;
unsigned mElementaryPID;
unsigned mStreamType;
unsigned mPCR_PID;
int32_t mExpectedContinuityCounter;
sp<ABuffer> mBuffer;
sp<AnotherPacketSource> mSource;
bool mPayloadStarted;
uint64_t mPrevPTS;
ElementaryStreamQueue *mQueue;
status_t flush();
status_t parsePES(ABitReader *br);
void onPayloadData(
unsigned PTS_DTS_flags, uint64_t PTS, uint64_t DTS,
const uint8_t *data, size_t size);
void extractAACFrames(const sp<ABuffer> &buffer);
bool isAudio() const;
bool isVideo() const;
DISALLOW_EVIL_CONSTRUCTORS(Stream);
};
struct ATSParser::PSISection : public RefBase {
PSISection();
status_t append(const void *data, size_t size);
void clear();
bool isComplete() const;
bool isEmpty() const;
const uint8_t *data() const;
size_t size() const;
protected:
virtual ~PSISection();
private:
sp<ABuffer> mBuffer;
DISALLOW_EVIL_CONSTRUCTORS(PSISection);
};
////////////////////////////////////////////////////////////////////////////////
ATSParser::Program::Program(
ATSParser *parser, unsigned programNumber, unsigned programMapPID)
: mParser(parser),
mProgramNumber(programNumber),
mProgramMapPID(programMapPID),
mFirstPTSValid(false),
mFirstPTS(0) {
ALOGV("new program number %u", programNumber);
}
bool ATSParser::Program::parsePSISection(
unsigned pid, ABitReader *br, status_t *err) {
*err = OK;
if (pid != mProgramMapPID) {
return false;
}
*err = parseProgramMap(br);
return true;
}
bool ATSParser::Program::parsePID(
unsigned pid, unsigned continuity_counter,
unsigned payload_unit_start_indicator,
ABitReader *br, status_t *err) {
*err = OK;
auto it = mStreams.find(pid);
if (it == mStreams.end()) {
return false;
}
*err = it->second->parse(
continuity_counter, payload_unit_start_indicator, br);
return true;
}
void ATSParser::Program::signalDiscontinuity(
DiscontinuityType type, const sp<AMessage> &extra) {
for (const auto &pair : mStreams) {
pair.second->signalDiscontinuity(type, extra);
}
}
void ATSParser::Program::signalEOS(status_t finalResult) {
for (const auto &pair : mStreams) {
pair.second->signalEOS(finalResult);
}
}
struct StreamInfo {
unsigned mType;
unsigned mPID;
};
status_t ATSParser::Program::parseProgramMap(ABitReader *br) {
unsigned table_id = br->getBits(8);
ALOGV(" table_id = %u", table_id);
CHECK_EQ(table_id, 0x02u);
unsigned section_syntax_indicator = br->getBits(1);
ALOGV(" section_syntax_indicator = %u", section_syntax_indicator);
CHECK_EQ(section_syntax_indicator, 1u);
CHECK_EQ(br->getBits(1), 0u);
MY_LOGV(" reserved = %u", br->getBits(2));
unsigned section_length = br->getBits(12);
ALOGV(" section_length = %u", section_length);
CHECK_EQ(section_length & 0xc00, 0u);
CHECK_LE(section_length, 1021u);
MY_LOGV(" program_number = %u", br->getBits(16));
MY_LOGV(" reserved = %u", br->getBits(2));
MY_LOGV(" version_number = %u", br->getBits(5));
MY_LOGV(" current_next_indicator = %u", br->getBits(1));
MY_LOGV(" section_number = %u", br->getBits(8));
MY_LOGV(" last_section_number = %u", br->getBits(8));
MY_LOGV(" reserved = %u", br->getBits(3));
unsigned PCR_PID = br->getBits(13);
ALOGV(" PCR_PID = 0x%04x", PCR_PID);
MY_LOGV(" reserved = %u", br->getBits(4));
unsigned program_info_length = br->getBits(12);
ALOGV(" program_info_length = %u", program_info_length);
CHECK_EQ(program_info_length & 0xc00, 0u);
br->skipBits(program_info_length * 8); // skip descriptors
std::vector<StreamInfo> infos;
// infoBytesRemaining is the number of bytes that make up the
// variable length section of ES_infos. It does not include the
// final CRC.
size_t infoBytesRemaining = section_length - 9 - program_info_length - 4;
while (infoBytesRemaining > 0) {
CHECK_GE(infoBytesRemaining, 5u);
unsigned streamType = br->getBits(8);
ALOGV(" stream_type = 0x%02x", streamType);
MY_LOGV(" reserved = %u", br->getBits(3));
unsigned elementaryPID = br->getBits(13);
ALOGV(" elementary_PID = 0x%04x", elementaryPID);
MY_LOGV(" reserved = %u", br->getBits(4));
unsigned ES_info_length = br->getBits(12);
ALOGV(" ES_info_length = %u", ES_info_length);
CHECK_EQ(ES_info_length & 0xc00, 0u);
CHECK_GE(infoBytesRemaining - 5, ES_info_length);
unsigned info_bytes_remaining = ES_info_length;
while (info_bytes_remaining >= 2) {
MY_LOGV(" tag = 0x%02x", br->getBits(8));
unsigned descLength = br->getBits(8);
ALOGV(" len = %u", descLength);
CHECK_GE(info_bytes_remaining, 2 + descLength);
br->skipBits(descLength * 8);
info_bytes_remaining -= descLength + 2;
}
CHECK_EQ(info_bytes_remaining, 0u);
StreamInfo info;
info.mType = streamType;
info.mPID = elementaryPID;
infos.push_back(info);
infoBytesRemaining -= 5 + ES_info_length;
}
CHECK_EQ(infoBytesRemaining, 0u);
MY_LOGV(" CRC = 0x%08x", br->getBits(32));
bool PIDsChanged = false;
for (const auto &info : infos) {
auto it = mStreams.find(info.mPID);
if (it != mStreams.end() && it->second->type() != info.mType) {
ALOGI("%s", "uh oh. stream PIDs have changed.");
PIDsChanged = true;
break;
}
}
if (PIDsChanged) {
// The only case we can recover from is if we have two streams
// and they switched PIDs.
bool success = false;
if (mStreams.size() == 2 && infos.size() == 2) {
const StreamInfo &info1 = infos.at(0);
const StreamInfo &info2 = infos.at(1);
sp<Stream> s1 = mStreams.begin()->second;
sp<Stream> s2 = (++mStreams.begin())->second;
bool caseA =
info1.mPID == s1->pid() && info1.mType == s2->type()
&& info2.mPID == s2->pid() && info2.mType == s1->type();
bool caseB =
info1.mPID == s2->pid() && info1.mType == s1->type()
&& info2.mPID == s1->pid() && info2.mType == s2->type();
if (caseA || caseB) {
unsigned pid1 = s1->pid();
unsigned pid2 = s2->pid();
s1->setPID(pid2);
s2->setPID(pid1);
mStreams.clear();
mStreams[s1->pid()] = s1;
mStreams[s2->pid()] = s2;
success = true;
}
}
if (!success) {
ALOGI("%s", "Stream PIDs changed and we cannot recover.");
return ERROR_MALFORMED;
}
}
for (const StreamInfo &info : infos) {
auto it = mStreams.find(info.mPID);
if (it == mStreams.end()) {
sp<Stream> stream = new Stream(
this, info.mPID, info.mType, PCR_PID);
mStreams[info.mPID] = stream;
}
}
return OK;
}
sp<AnotherPacketSource> ATSParser::Program::getSource(SourceType type) {
size_t index = (type == AUDIO) ? 0 : 0;
for (const auto &pair : mStreams) {
sp<AnotherPacketSource> source = pair.second->getSource(type);
if (source != NULL) {
if (index == 0) {
return source;
}
--index;
}
}
return NULL;
}
int64_t ATSParser::Program::convertPTSToTimestamp(uint64_t PTS) {
if (mParser->mFlags & DUMP_PTS) {
ALOGI("PTS = 0x%016llx", PTS);
}
if (!(mParser->mFlags & TS_TIMESTAMPS_ARE_ABSOLUTE)) {
if (!mFirstPTSValid) {
mFirstPTSValid = true;
mFirstPTS = PTS;
PTS = 0;
} else if (PTS < mFirstPTS) {
PTS = 0;
} else {
PTS -= mFirstPTS;
}
}
int64_t timeUs = (PTS * 100) / 9;
if (mParser->mAbsoluteTimeAnchorUs >= 0ll) {
timeUs += mParser->mAbsoluteTimeAnchorUs;
}
if (mParser->mTimeOffsetValid) {
timeUs += mParser->mTimeOffsetUs;
}
return timeUs;
}
////////////////////////////////////////////////////////////////////////////////
ATSParser::Stream::Stream(
Program *program,
unsigned elementaryPID,
unsigned streamType,
unsigned PCR_PID)
: mProgram(program),
mElementaryPID(elementaryPID),
mStreamType(streamType),
mPCR_PID(PCR_PID),
mExpectedContinuityCounter(-1),
mPayloadStarted(false),
mPrevPTS(0),
mQueue(NULL) {
switch (mStreamType) {
case STREAMTYPE_H264:
mQueue = new ElementaryStreamQueue(
ElementaryStreamQueue::H264,
(mProgram->parserFlags() & ALIGNED_VIDEO_DATA)
? ElementaryStreamQueue::kFlag_AlignedData : 0);
break;
case STREAMTYPE_MPEG2_AUDIO_ADTS:
mQueue = new ElementaryStreamQueue(ElementaryStreamQueue::AAC);
break;
case STREAMTYPE_MPEG1_AUDIO:
case STREAMTYPE_MPEG2_AUDIO:
mQueue = new ElementaryStreamQueue(
ElementaryStreamQueue::MPEG_AUDIO);
break;
case STREAMTYPE_MPEG1_VIDEO:
case STREAMTYPE_MPEG2_VIDEO:
mQueue = new ElementaryStreamQueue(
ElementaryStreamQueue::MPEG_VIDEO);
break;
case STREAMTYPE_MPEG4_VIDEO:
mQueue = new ElementaryStreamQueue(
ElementaryStreamQueue::MPEG4_VIDEO);
break;
case STREAMTYPE_PCM_AUDIO:
mQueue = new ElementaryStreamQueue(
ElementaryStreamQueue::PCM_AUDIO);
break;
default:
break;
}
ALOGV("new stream PID 0x%02x, type 0x%02x", elementaryPID, streamType);
if (mQueue != NULL) {
mBuffer = new ABuffer(192 * 1024);
mBuffer->setRange(0, 0);
}
}
ATSParser::Stream::~Stream() {
delete mQueue;
mQueue = NULL;
}
status_t ATSParser::Stream::parse(
unsigned continuity_counter,
unsigned payload_unit_start_indicator, ABitReader *br) {
if (mQueue == NULL) {
return OK;
}
if (mExpectedContinuityCounter >= 0
&& (unsigned)mExpectedContinuityCounter != continuity_counter) {
ALOGI("discontinuity on stream pid 0x%04x", mElementaryPID);
mPayloadStarted = false;
mBuffer->setRange(0, 0);
mExpectedContinuityCounter = -1;
return OK;
}
mExpectedContinuityCounter = (continuity_counter + 1) & 0x0f;
if (payload_unit_start_indicator) {
if (mPayloadStarted) {
// Otherwise we run the danger of receiving the trailing bytes
// of a PES packet that we never saw the start of and assuming
// we have a a complete PES packet.
status_t err = flush();
if (err != OK) {
return err;
}
}
mPayloadStarted = true;
}
if (!mPayloadStarted) {
return OK;
}
size_t payloadSizeBits = br->numBitsLeft();
CHECK_EQ(payloadSizeBits % 8, 0u);
size_t neededSize = mBuffer->size() + payloadSizeBits / 8;
if (mBuffer->capacity() < neededSize) {
// Increment in multiples of 64K.
neededSize = (neededSize + 65535) & ~65535;
ALOGI("resizing buffer to %ld bytes", neededSize);
sp<ABuffer> newBuffer = new ABuffer(neededSize);
memcpy(newBuffer->data(), mBuffer->data(), mBuffer->size());
newBuffer->setRange(0, mBuffer->size());
mBuffer = newBuffer;
}
memcpy(mBuffer->data() + mBuffer->size(), br->data(), payloadSizeBits / 8);
mBuffer->setRange(0, mBuffer->size() + payloadSizeBits / 8);
return OK;
}
bool ATSParser::Stream::isVideo() const {
switch (mStreamType) {
case STREAMTYPE_H264:
case STREAMTYPE_MPEG1_VIDEO:
case STREAMTYPE_MPEG2_VIDEO:
case STREAMTYPE_MPEG4_VIDEO:
return true;
default:
return false;
}
}
bool ATSParser::Stream::isAudio() const {
switch (mStreamType) {
case STREAMTYPE_MPEG1_AUDIO:
case STREAMTYPE_MPEG2_AUDIO:
case STREAMTYPE_MPEG2_AUDIO_ADTS:
case STREAMTYPE_PCM_AUDIO:
return true;
default:
return false;
}
}
void ATSParser::Stream::signalDiscontinuity(
DiscontinuityType type, const sp<AMessage> &extra) {
mExpectedContinuityCounter = -1;
if (mQueue == NULL) {
return;
}
mPayloadStarted = false;
mBuffer->setRange(0, 0);
bool clearFormat = false;
if (isAudio()) {
if (type & DISCONTINUITY_AUDIO_FORMAT) {
clearFormat = true;
}
} else {
if (type & DISCONTINUITY_VIDEO_FORMAT) {
clearFormat = true;
}
}
mQueue->clear(clearFormat);
if (mSource != NULL) {
mSource->queueDiscontinuity(type, extra);
}
}
void ATSParser::Stream::signalEOS(status_t finalResult) {
if (mSource != NULL) {
mSource->signalEOS(finalResult);
}
}
status_t ATSParser::Stream::parsePES(ABitReader *br) {
unsigned packet_startcode_prefix = br->getBits(24);
ALOGV("packet_startcode_prefix = 0x%08x", packet_startcode_prefix);
if (packet_startcode_prefix != 1) {
ALOGV("Supposedly payload_unit_start=1 unit does not start "
"with startcode.");
return ERROR_MALFORMED;
}
CHECK_EQ(packet_startcode_prefix, 0x000001u);
unsigned stream_id = br->getBits(8);
ALOGV("stream_id = 0x%02x", stream_id);
unsigned PES_packet_length = br->getBits(16);
ALOGV("PES_packet_length = %u", PES_packet_length);
if (stream_id != 0xbc // program_stream_map
&& stream_id != 0xbe // padding_stream
&& stream_id != 0xbf // private_stream_2
&& stream_id != 0xf0 // ECM
&& stream_id != 0xf1 // EMM
&& stream_id != 0xff // program_stream_directory
&& stream_id != 0xf2 // DSMCC
&& stream_id != 0xf8) { // H.222.1 type E
CHECK_EQ(br->getBits(2), 2u);
MY_LOGV("PES_scrambling_control = %u", br->getBits(2));
MY_LOGV("PES_priority = %u", br->getBits(1));
MY_LOGV("data_alignment_indicator = %u", br->getBits(1));
MY_LOGV("copyright = %u", br->getBits(1));
MY_LOGV("original_or_copy = %u", br->getBits(1));
unsigned PTS_DTS_flags = br->getBits(2);
ALOGV("PTS_DTS_flags = %u", PTS_DTS_flags);
unsigned ESCR_flag = br->getBits(1);
ALOGV("ESCR_flag = %u", ESCR_flag);
unsigned ES_rate_flag = br->getBits(1);
ALOGV("ES_rate_flag = %u", ES_rate_flag);
MY_LOGV("DSM_trick_mode_flag = %u", br->getBits(1));
MY_LOGV("additional_copy_info_flag = %u", br->getBits(1));
MY_LOGV("PES_CRC_flag = %u", br->getBits(1));
MY_LOGV("PES_extension_flag = %u", br->getBits(1));
unsigned PES_header_data_length = br->getBits(8);
ALOGV("PES_header_data_length = %u", PES_header_data_length);
unsigned optional_bytes_remaining = PES_header_data_length;
uint64_t PTS = 0, DTS = 0;
if (PTS_DTS_flags == 2 || PTS_DTS_flags == 3) {
CHECK_GE(optional_bytes_remaining, 5u);
CHECK_EQ(br->getBits(4), PTS_DTS_flags);
PTS = ((uint64_t)br->getBits(3)) << 30;
CHECK_EQ(br->getBits(1), 1u);
PTS |= ((uint64_t)br->getBits(15)) << 15;
CHECK_EQ(br->getBits(1), 1u);
PTS |= br->getBits(15);
CHECK_EQ(br->getBits(1), 1u);
ALOGV("PTS = 0x%016llx (%.2f)", PTS, PTS / 90000.0);
optional_bytes_remaining -= 5;
if (PTS_DTS_flags == 3) {
CHECK_GE(optional_bytes_remaining, 5u);
CHECK_EQ(br->getBits(4), 1u);
DTS = ((uint64_t)br->getBits(3)) << 30;
CHECK_EQ(br->getBits(1), 1u);
DTS |= ((uint64_t)br->getBits(15)) << 15;
CHECK_EQ(br->getBits(1), 1u);
DTS |= br->getBits(15);
CHECK_EQ(br->getBits(1), 1u);
ALOGV("DTS = %llu", DTS);
optional_bytes_remaining -= 5;
}
}
if (ESCR_flag) {
CHECK_GE(optional_bytes_remaining, 6u);
br->getBits(2);
uint64_t ESCR = ((uint64_t)br->getBits(3)) << 30;
CHECK_EQ(br->getBits(1), 1u);
ESCR |= ((uint64_t)br->getBits(15)) << 15;
CHECK_EQ(br->getBits(1), 1u);
ESCR |= br->getBits(15);
CHECK_EQ(br->getBits(1), 1u);
ALOGV("ESCR = %llu", ESCR);
MY_LOGV("ESCR_extension = %u", br->getBits(9));
CHECK_EQ(br->getBits(1), 1u);
optional_bytes_remaining -= 6;
}
if (ES_rate_flag) {
CHECK_GE(optional_bytes_remaining, 3u);
CHECK_EQ(br->getBits(1), 1u);
MY_LOGV("ES_rate = %u", br->getBits(22));
CHECK_EQ(br->getBits(1), 1u);
optional_bytes_remaining -= 3;
}
br->skipBits(optional_bytes_remaining * 8);
// ES data follows.
if (PES_packet_length != 0) {
CHECK_GE(PES_packet_length, PES_header_data_length + 3);
unsigned dataLength =
PES_packet_length - 3 - PES_header_data_length;
if (br->numBitsLeft() < dataLength * 8) {
ALOGE("PES packet does not carry enough data to contain "
"payload. (numBitsLeft = %ld, required = %d)",
br->numBitsLeft(), dataLength * 8);
return ERROR_MALFORMED;
}
CHECK_GE(br->numBitsLeft(), dataLength * 8);
onPayloadData(
PTS_DTS_flags, PTS, DTS, br->data(), dataLength);
br->skipBits(dataLength * 8);
} else {
onPayloadData(
PTS_DTS_flags, PTS, DTS,
br->data(), br->numBitsLeft() / 8);
size_t payloadSizeBits = br->numBitsLeft();
CHECK_EQ(payloadSizeBits % 8, 0u);
ALOGV("There's %d bytes of payload.", payloadSizeBits / 8);
}
} else if (stream_id == 0xbe) { // padding_stream
CHECK_NE(PES_packet_length, 0u);
br->skipBits(PES_packet_length * 8);
} else {
CHECK_NE(PES_packet_length, 0u);
br->skipBits(PES_packet_length * 8);
}
return OK;
}
status_t ATSParser::Stream::flush() {
if (mBuffer->size() == 0) {
return OK;
}
ALOGV("flushing stream 0x%04x size = %d", mElementaryPID, mBuffer->size());
ABitReader br(mBuffer->data(), mBuffer->size());
status_t err = parsePES(&br);
mBuffer->setRange(0, 0);
return err;
}
void ATSParser::Stream::onPayloadData(
unsigned PTS_DTS_flags, uint64_t PTS, uint64_t /* DTS */,
const uint8_t *data, size_t size) {
ALOGV("onPayloadData mStreamType=0x%02x", mStreamType);
int64_t timeUs = 0ll; // no presentation timestamp available.
if (PTS_DTS_flags == 2 || PTS_DTS_flags == 3) {
timeUs = mProgram->convertPTSToTimestamp(PTS);
}
status_t err = mQueue->appendData(data, size, timeUs);
if (err != OK) {
return;
}
sp<ABuffer> accessUnit;
while ((accessUnit = mQueue->dequeueAccessUnit()) != NULL) {
if (mSource == NULL) {
sp<MetaData> meta = mQueue->getFormat();
if (meta != NULL) {
ALOGV("Stream PID 0x%08x of type 0x%02x now has data.",
mElementaryPID, mStreamType);
mSource = new AnotherPacketSource(meta);
mSource->queueAccessUnit(accessUnit);
}
} else if (mQueue->getFormat() != NULL) {
// After a discontinuity we invalidate the queue's format
// and won't enqueue any access units to the source until
// the queue has reestablished the new format.
if (mSource->getFormat() == NULL) {
mSource->setFormat(mQueue->getFormat());
}
mSource->queueAccessUnit(accessUnit);
}
}
}
sp<AnotherPacketSource> ATSParser::Stream::getSource(SourceType type) {
switch (type) {
case VIDEO:
{
if (isVideo()) {
return mSource;
}
break;
}
case AUDIO:
{
if (isAudio()) {
return mSource;
}
break;
}
default:
break;
}
return NULL;
}
////////////////////////////////////////////////////////////////////////////////
ATSParser::ATSParser(uint32_t flags)
: mFlags(flags),
mAbsoluteTimeAnchorUs(-1ll),
mTimeOffsetValid(false),
mTimeOffsetUs(0ll),
mNumTSPacketsParsed(0),
mNumPCRs(0) {
mPSISections[0 /* PID */] = new PSISection;
}
ATSParser::~ATSParser() {
}
status_t ATSParser::feedTSPacket(const void *data, size_t size) {
CHECK_EQ(size, kTSPacketSize);
ABitReader br((const uint8_t *)data, kTSPacketSize);
return parseTS(&br);
}
void ATSParser::signalDiscontinuity(
DiscontinuityType type, const sp<AMessage> &extra) {
if (type == DISCONTINUITY_ABSOLUTE_TIME) {
int64_t timeUs;
CHECK(extra->findInt64("timeUs", &timeUs));
CHECK(mPrograms.empty());
mAbsoluteTimeAnchorUs = timeUs;
return;
} else if (type == DISCONTINUITY_TIME_OFFSET) {
int64_t offset;
CHECK(extra->findInt64("offset", &offset));
mTimeOffsetValid = true;
mTimeOffsetUs = offset;
return;
}
for (const auto &program : mPrograms) {
program->signalDiscontinuity(type, extra);
}
}
void ATSParser::signalEOS(status_t finalResult) {
CHECK_NE(finalResult, (status_t)OK);
for (const auto &program : mPrograms) {
program->signalEOS(finalResult);
}
}
void ATSParser::parseProgramAssociationTable(ABitReader *br) {
unsigned table_id = br->getBits(8);
ALOGV(" table_id = %u", table_id);
CHECK_EQ(table_id, 0x00u);
unsigned section_syntax_indictor = br->getBits(1);
ALOGV(" section_syntax_indictor = %u", section_syntax_indictor);
CHECK_EQ(section_syntax_indictor, 1u);
CHECK_EQ(br->getBits(1), 0u);
MY_LOGV(" reserved = %u", br->getBits(2));
unsigned section_length = br->getBits(12);
ALOGV(" section_length = %u", section_length);
CHECK_EQ(section_length & 0xc00, 0u);
MY_LOGV(" transport_stream_id = %u", br->getBits(16));
MY_LOGV(" reserved = %u", br->getBits(2));
MY_LOGV(" version_number = %u", br->getBits(5));
MY_LOGV(" current_next_indicator = %u", br->getBits(1));
MY_LOGV(" section_number = %u", br->getBits(8));
MY_LOGV(" last_section_number = %u", br->getBits(8));
size_t numProgramBytes = (section_length - 5 /* header */ - 4 /* crc */);
CHECK_EQ((numProgramBytes % 4), 0u);
for (size_t i = 0; i < numProgramBytes / 4; ++i) {
unsigned program_number = br->getBits(16);
ALOGV(" program_number = %u", program_number);
MY_LOGV(" reserved = %u", br->getBits(3));
if (program_number == 0) {
MY_LOGV(" network_PID = 0x%04x", br->getBits(13));
} else {
unsigned programMapPID = br->getBits(13);
ALOGV(" program_map_PID = 0x%04x", programMapPID);
bool found = false;
for (const auto &program : mPrograms) {
if (program->number() == program_number) {
program->updateProgramMapPID(programMapPID);
found = true;
break;
}
}
if (!found) {
mPrograms.push_back(
new Program(this, program_number, programMapPID));
}
if (mPSISections.find(programMapPID) == mPSISections.end()) {
mPSISections[programMapPID] = new PSISection;
}
}
}
MY_LOGV(" CRC = 0x%08x", br->getBits(32));
}
status_t ATSParser::parsePID(
ABitReader *br, unsigned PID,
unsigned continuity_counter,
unsigned payload_unit_start_indicator) {
auto it = mPSISections.find(PID);
if (it != mPSISections.end()) {
const sp<PSISection> &section = it->second;
if (payload_unit_start_indicator) {
CHECK(section->isEmpty());
unsigned skip = br->getBits(8);
br->skipBits(skip * 8);
}
CHECK((br->numBitsLeft() % 8) == 0);
status_t err = section->append(br->data(), br->numBitsLeft() / 8);
if (err != OK) {
return err;
}
if (!section->isComplete()) {
return OK;
}
ABitReader sectionBits(section->data(), section->size());
if (PID == 0) {
parseProgramAssociationTable(&sectionBits);
} else {
bool handled = false;
for (const auto &program : mPrograms) {
status_t err;
if (!program->parsePSISection(PID, &sectionBits, &err)) {
continue;
}
if (err != OK) {
return err;
}
handled = true;
break;
}
if (!handled) {
mPSISections.erase(it);
}
}
section->clear();
return OK;
}
bool handled = false;
for (const auto &program : mPrograms) {
status_t err;
if (program->parsePID(
PID, continuity_counter, payload_unit_start_indicator,
br, &err)) {
if (err != OK) {
return err;
}
handled = true;
break;
}
}
if (!handled) {
ALOGV("PID 0x%04x not handled.", PID);
}
return OK;
}
void ATSParser::parseAdaptationField(ABitReader *br, unsigned PID) {
unsigned adaptation_field_length = br->getBits(8);
if (adaptation_field_length > 0) {
unsigned discontinuity_indicator = br->getBits(1);
if (discontinuity_indicator) {
ALOGV("PID 0x%04x: discontinuity_indicator = 1 (!!!)", PID);
}
br->skipBits(2);
unsigned PCR_flag = br->getBits(1);
size_t numBitsRead = 4;
if (PCR_flag) {
br->skipBits(4);
uint64_t PCR_base = br->getBits(32);
PCR_base = (PCR_base << 1) | br->getBits(1);
br->skipBits(6);
unsigned PCR_ext = br->getBits(9);
// The number of bytes from the start of the current
// MPEG2 transport stream packet up and including
// the final byte of this PCR_ext field.
size_t byteOffsetFromStartOfTSPacket =
(188 - br->numBitsLeft() / 8);
uint64_t PCR = PCR_base * 300 + PCR_ext;
ALOGV("PID 0x%04x: PCR = 0x%016llx (%.2f)",
PID, PCR, PCR / 27E6);
// The number of bytes received by this parser up to and
// including the final byte of this PCR_ext field.
size_t byteOffsetFromStart =
mNumTSPacketsParsed * 188 + byteOffsetFromStartOfTSPacket;
for (size_t i = 0; i < mPrograms.size(); ++i) {
updatePCR(PID, PCR, byteOffsetFromStart);
}
numBitsRead += 52;
}
CHECK_GE(adaptation_field_length * 8, numBitsRead);
br->skipBits(adaptation_field_length * 8 - numBitsRead);
}
}
status_t ATSParser::parseTS(ABitReader *br) {
ALOGV("---");
unsigned sync_byte = br->getBits(8);
CHECK_EQ(sync_byte, 0x47u);
MY_LOGV("transport_error_indicator = %u", br->getBits(1));
unsigned payload_unit_start_indicator = br->getBits(1);
ALOGV("payload_unit_start_indicator = %u", payload_unit_start_indicator);
MY_LOGV("transport_priority = %u", br->getBits(1));
unsigned PID = br->getBits(13);
ALOGV("PID = 0x%04x", PID);
MY_LOGV("transport_scrambling_control = %u", br->getBits(2));
unsigned adaptation_field_control = br->getBits(2);
ALOGV("adaptation_field_control = %u", adaptation_field_control);
unsigned continuity_counter = br->getBits(4);
ALOGV("PID = 0x%04x, continuity_counter = %u", PID, continuity_counter);
// ALOGI("PID = 0x%04x, continuity_counter = %u", PID, continuity_counter);
if (adaptation_field_control == 2 || adaptation_field_control == 3) {
parseAdaptationField(br, PID);
}
status_t err = OK;
if (adaptation_field_control == 1 || adaptation_field_control == 3) {
err = parsePID(
br, PID, continuity_counter, payload_unit_start_indicator);
}
++mNumTSPacketsParsed;
return err;
}
sp<AnotherPacketSource> ATSParser::getSource(SourceType type) {
int which = -1; // any
for (const auto &program : mPrograms) {
if (which >= 0 && (int)program->number() != which) {
continue;
}
sp<AnotherPacketSource> source = program->getSource(type);
if (source != NULL) {
return source;
}
}
return NULL;
}
bool ATSParser::PTSTimeDeltaEstablished() {
if (mPrograms.empty()) {
return false;
}
return mPrograms.front()->PTSTimeDeltaEstablished();
}
void ATSParser::updatePCR(
unsigned /* PID */, uint64_t PCR, size_t byteOffsetFromStart) {
ALOGV("PCR 0x%016llx @ %d", PCR, byteOffsetFromStart);
if (mNumPCRs == 2) {
mPCR[0] = mPCR[1];
mPCRBytes[0] = mPCRBytes[1];
mSystemTimeUs[0] = mSystemTimeUs[1];
mNumPCRs = 1;
}
mPCR[mNumPCRs] = PCR;
mPCRBytes[mNumPCRs] = byteOffsetFromStart;
mSystemTimeUs[mNumPCRs] = ALooper::GetNowUs();
++mNumPCRs;
}
////////////////////////////////////////////////////////////////////////////////
ATSParser::PSISection::PSISection() {
}
ATSParser::PSISection::~PSISection() {
}
status_t ATSParser::PSISection::append(const void *data, size_t size) {
if (mBuffer == NULL || mBuffer->size() + size > mBuffer->capacity()) {
size_t newCapacity =
(mBuffer == NULL) ? size : mBuffer->capacity() + size;
newCapacity = (newCapacity + 1023) & ~1023;
sp<ABuffer> newBuffer = new ABuffer(newCapacity);
if (mBuffer != NULL) {
memcpy(newBuffer->data(), mBuffer->data(), mBuffer->size());
newBuffer->setRange(0, mBuffer->size());
} else {
newBuffer->setRange(0, 0);
}
mBuffer = newBuffer;
}
memcpy(mBuffer->data() + mBuffer->size(), data, size);
mBuffer->setRange(0, mBuffer->size() + size);
return OK;
}
void ATSParser::PSISection::clear() {
if (mBuffer != NULL) {
mBuffer->setRange(0, 0);
}
}
bool ATSParser::PSISection::isComplete() const {
if (mBuffer == NULL || mBuffer->size() < 3) {
return false;
}
unsigned sectionLength = U16_AT(mBuffer->data() + 1) & 0xfff;
return mBuffer->size() >= sectionLength + 3;
}
bool ATSParser::PSISection::isEmpty() const {
return mBuffer == NULL || mBuffer->size() == 0;
}
const uint8_t *ATSParser::PSISection::data() const {
return mBuffer == NULL ? NULL : mBuffer->data();
}
size_t ATSParser::PSISection::size() const {
return mBuffer == NULL ? 0 : mBuffer->size();
}
} // namespace android