ATSParser.cpp revision c4c17d47b674b425fb6c399822c0ab3258543c0a 
1/*
2 * Copyright (C) 2010 The Android Open Source Project
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 *      http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16
17//#define LOG_NDEBUG 0
18#define LOG_TAG "ATSParser"
19#include <utils/Log.h>
20
21#include "ATSParser.h"
22
23#include "AnotherPacketSource.h"
24#include "ESQueue.h"
25#include "include/avc_utils.h"
26
27#include <media/stagefright/foundation/ABitReader.h>
28#include <media/stagefright/foundation/ABuffer.h>
29#include <media/stagefright/foundation/ADebug.h>
30#include <media/stagefright/foundation/AMessage.h>
31#include <media/stagefright/foundation/hexdump.h>
32#include <media/stagefright/MediaDefs.h>
33#include <media/stagefright/MediaErrors.h>
34#include <media/stagefright/MetaData.h>
35#include <media/IStreamSource.h>
36#include <utils/KeyedVector.h>
37
38namespace android {
39
40// I want the expression "y" evaluated even if verbose logging is off.
41#define MY_LOGV(x, y) \
42    do { unsigned tmp = y; LOGV(x, tmp); } while (0)
43
44static const size_t kTSPacketSize = 188;
45
46struct ATSParser::Program : public RefBase {
47    Program(ATSParser *parser, unsigned programNumber, unsigned programMapPID);
48
49    bool parsePID(
50            unsigned pid, unsigned payload_unit_start_indicator,
51            ABitReader *br);
52
53    void signalDiscontinuity(
54            DiscontinuityType type, const sp<AMessage> &extra);
55
56    void signalEOS(status_t finalResult);
57
58    sp<MediaSource> getSource(SourceType type);
59
60    int64_t convertPTSToTimestamp(uint64_t PTS);
61
62    bool PTSTimeDeltaEstablished() const {
63        return mFirstPTSValid;
64    }
65
66    unsigned number() const { return mProgramNumber; }
67
68    void updateProgramMapPID(unsigned programMapPID) {
69        mProgramMapPID = programMapPID;
70    }
71
72private:
73    ATSParser *mParser;
74    unsigned mProgramNumber;
75    unsigned mProgramMapPID;
76    KeyedVector<unsigned, sp<Stream> > mStreams;
77    bool mFirstPTSValid;
78    uint64_t mFirstPTS;
79
80    void parseProgramMap(ABitReader *br);
81
82    DISALLOW_EVIL_CONSTRUCTORS(Program);
83};
84
85struct ATSParser::Stream : public RefBase {
86    Stream(Program *program, unsigned elementaryPID, unsigned streamType);
87
88    unsigned type() const { return mStreamType; }
89    unsigned pid() const { return mElementaryPID; }
90    void setPID(unsigned pid) { mElementaryPID = pid; }
91
92    void parse(
93            unsigned payload_unit_start_indicator,
94            ABitReader *br);
95
96    void signalDiscontinuity(
97            DiscontinuityType type, const sp<AMessage> &extra);
98
99    void signalEOS(status_t finalResult);
100
101    sp<MediaSource> getSource(SourceType type);
102
103protected:
104    virtual ~Stream();
105
106private:
107    Program *mProgram;
108    unsigned mElementaryPID;
109    unsigned mStreamType;
110
111    sp<ABuffer> mBuffer;
112    sp<AnotherPacketSource> mSource;
113    bool mPayloadStarted;
114    DiscontinuityType mPendingDiscontinuity;
115    sp<AMessage> mPendingDiscontinuityExtra;
116
117    ElementaryStreamQueue *mQueue;
118
119    void flush();
120    void parsePES(ABitReader *br);
121
122    void onPayloadData(
123            unsigned PTS_DTS_flags, uint64_t PTS, uint64_t DTS,
124            const uint8_t *data, size_t size);
125
126    void extractAACFrames(const sp<ABuffer> &buffer);
127
128    void deferDiscontinuity(
129            DiscontinuityType type, const sp<AMessage> &extra);
130
131    DISALLOW_EVIL_CONSTRUCTORS(Stream);
132};
133
134////////////////////////////////////////////////////////////////////////////////
135
136ATSParser::Program::Program(
137        ATSParser *parser, unsigned programNumber, unsigned programMapPID)
138    : mParser(parser),
139      mProgramNumber(programNumber),
140      mProgramMapPID(programMapPID),
141      mFirstPTSValid(false),
142      mFirstPTS(0) {
143    LOGV("new program number %u", programNumber);
144}
145
146bool ATSParser::Program::parsePID(
147        unsigned pid, unsigned payload_unit_start_indicator,
148        ABitReader *br) {
149    if (pid == mProgramMapPID) {
150        if (payload_unit_start_indicator) {
151            unsigned skip = br->getBits(8);
152            br->skipBits(skip * 8);
153        }
154
155        parseProgramMap(br);
156        return true;
157    }
158
159    ssize_t index = mStreams.indexOfKey(pid);
160    if (index < 0) {
161        return false;
162    }
163
164    mStreams.editValueAt(index)->parse(
165            payload_unit_start_indicator, br);
166
167    return true;
168}
169
170void ATSParser::Program::signalDiscontinuity(
171        DiscontinuityType type, const sp<AMessage> &extra) {
172    for (size_t i = 0; i < mStreams.size(); ++i) {
173        mStreams.editValueAt(i)->signalDiscontinuity(type, extra);
174    }
175}
176
177void ATSParser::Program::signalEOS(status_t finalResult) {
178    for (size_t i = 0; i < mStreams.size(); ++i) {
179        mStreams.editValueAt(i)->signalEOS(finalResult);
180    }
181}
182
183struct StreamInfo {
184    unsigned mType;
185    unsigned mPID;
186};
187
188void ATSParser::Program::parseProgramMap(ABitReader *br) {
189    unsigned table_id = br->getBits(8);
190    LOGV("  table_id = %u", table_id);
191    CHECK_EQ(table_id, 0x02u);
192
193    unsigned section_syntax_indicator = br->getBits(1);
194    LOGV("  section_syntax_indicator = %u", section_syntax_indicator);
195    CHECK_EQ(section_syntax_indicator, 1u);
196
197    CHECK_EQ(br->getBits(1), 0u);
198    MY_LOGV("  reserved = %u", br->getBits(2));
199
200    unsigned section_length = br->getBits(12);
201    LOGV("  section_length = %u", section_length);
202    CHECK_EQ(section_length & 0xc00, 0u);
203    CHECK_LE(section_length, 1021u);
204
205    MY_LOGV("  program_number = %u", br->getBits(16));
206    MY_LOGV("  reserved = %u", br->getBits(2));
207    MY_LOGV("  version_number = %u", br->getBits(5));
208    MY_LOGV("  current_next_indicator = %u", br->getBits(1));
209    MY_LOGV("  section_number = %u", br->getBits(8));
210    MY_LOGV("  last_section_number = %u", br->getBits(8));
211    MY_LOGV("  reserved = %u", br->getBits(3));
212    MY_LOGV("  PCR_PID = 0x%04x", br->getBits(13));
213    MY_LOGV("  reserved = %u", br->getBits(4));
214
215    unsigned program_info_length = br->getBits(12);
216    LOGV("  program_info_length = %u", program_info_length);
217    CHECK_EQ(program_info_length & 0xc00, 0u);
218
219    br->skipBits(program_info_length * 8);  // skip descriptors
220
221    Vector<StreamInfo> infos;
222
223    // infoBytesRemaining is the number of bytes that make up the
224    // variable length section of ES_infos. It does not include the
225    // final CRC.
226    size_t infoBytesRemaining = section_length - 9 - program_info_length - 4;
227
228    while (infoBytesRemaining > 0) {
229        CHECK_GE(infoBytesRemaining, 5u);
230
231        unsigned streamType = br->getBits(8);
232        LOGV("    stream_type = 0x%02x", streamType);
233
234        MY_LOGV("    reserved = %u", br->getBits(3));
235
236        unsigned elementaryPID = br->getBits(13);
237        LOGV("    elementary_PID = 0x%04x", elementaryPID);
238
239        MY_LOGV("    reserved = %u", br->getBits(4));
240
241        unsigned ES_info_length = br->getBits(12);
242        LOGV("    ES_info_length = %u", ES_info_length);
243        CHECK_EQ(ES_info_length & 0xc00, 0u);
244
245        CHECK_GE(infoBytesRemaining - 5, ES_info_length);
246
247#if 0
248        br->skipBits(ES_info_length * 8);  // skip descriptors
249#else
250        unsigned info_bytes_remaining = ES_info_length;
251        while (info_bytes_remaining >= 2) {
252            MY_LOGV("      tag = 0x%02x", br->getBits(8));
253
254            unsigned descLength = br->getBits(8);
255            LOGV("      len = %u", descLength);
256
257            CHECK_GE(info_bytes_remaining, 2 + descLength);
258
259            br->skipBits(descLength * 8);
260
261            info_bytes_remaining -= descLength + 2;
262        }
263        CHECK_EQ(info_bytes_remaining, 0u);
264#endif
265
266        StreamInfo info;
267        info.mType = streamType;
268        info.mPID = elementaryPID;
269        infos.push(info);
270
271        infoBytesRemaining -= 5 + ES_info_length;
272    }
273
274    CHECK_EQ(infoBytesRemaining, 0u);
275    MY_LOGV("  CRC = 0x%08x", br->getBits(32));
276
277    bool PIDsChanged = false;
278    for (size_t i = 0; i < infos.size(); ++i) {
279        StreamInfo &info = infos.editItemAt(i);
280
281        ssize_t index = mStreams.indexOfKey(info.mPID);
282
283        if (index >= 0 && mStreams.editValueAt(index)->type() != info.mType) {
284            LOGI("uh oh. stream PIDs have changed.");
285            PIDsChanged = true;
286            break;
287        }
288    }
289
290    if (PIDsChanged) {
291        mStreams.clear();
292    }
293
294    for (size_t i = 0; i < infos.size(); ++i) {
295        StreamInfo &info = infos.editItemAt(i);
296
297        ssize_t index = mStreams.indexOfKey(info.mPID);
298
299        if (index < 0) {
300            sp<Stream> stream = new Stream(this, info.mPID, info.mType);
301            mStreams.add(info.mPID, stream);
302
303            if (PIDsChanged) {
304                sp<AMessage> extra;
305                stream->signalDiscontinuity(DISCONTINUITY_FORMATCHANGE, extra);
306            }
307        }
308    }
309}
310
311sp<MediaSource> ATSParser::Program::getSource(SourceType type) {
312    size_t index = (type == AUDIO) ? 0 : 0;
313
314    for (size_t i = 0; i < mStreams.size(); ++i) {
315        sp<MediaSource> source = mStreams.editValueAt(i)->getSource(type);
316        if (source != NULL) {
317            if (index == 0) {
318                return source;
319            }
320            --index;
321        }
322    }
323
324    return NULL;
325}
326
327int64_t ATSParser::Program::convertPTSToTimestamp(uint64_t PTS) {
328    if (!(mParser->mFlags & TS_TIMESTAMPS_ARE_ABSOLUTE)) {
329        if (!mFirstPTSValid) {
330            mFirstPTSValid = true;
331            mFirstPTS = PTS;
332            PTS = 0;
333        } else if (PTS < mFirstPTS) {
334            PTS = 0;
335        } else {
336            PTS -= mFirstPTS;
337        }
338    }
339
340    return (PTS * 100) / 9;
341}
342
343////////////////////////////////////////////////////////////////////////////////
344
345ATSParser::Stream::Stream(
346        Program *program, unsigned elementaryPID, unsigned streamType)
347    : mProgram(program),
348      mElementaryPID(elementaryPID),
349      mStreamType(streamType),
350      mBuffer(new ABuffer(192 * 1024)),
351      mPayloadStarted(false),
352      mPendingDiscontinuity(DISCONTINUITY_NONE),
353      mQueue(NULL) {
354    mBuffer->setRange(0, 0);
355
356    switch (mStreamType) {
357        case STREAMTYPE_H264:
358            mQueue = new ElementaryStreamQueue(ElementaryStreamQueue::H264);
359            break;
360        case STREAMTYPE_MPEG2_AUDIO_ATDS:
361            mQueue = new ElementaryStreamQueue(ElementaryStreamQueue::AAC);
362            break;
363        case STREAMTYPE_MPEG1_AUDIO:
364        case STREAMTYPE_MPEG2_AUDIO:
365            mQueue = new ElementaryStreamQueue(
366                    ElementaryStreamQueue::MPEG_AUDIO);
367            break;
368
369        case STREAMTYPE_MPEG1_VIDEO:
370        case STREAMTYPE_MPEG2_VIDEO:
371            mQueue = new ElementaryStreamQueue(
372                    ElementaryStreamQueue::MPEG_VIDEO);
373            break;
374
375        case STREAMTYPE_MPEG4_VIDEO:
376            mQueue = new ElementaryStreamQueue(
377                    ElementaryStreamQueue::MPEG4_VIDEO);
378            break;
379
380        default:
381            break;
382    }
383
384    LOGV("new stream PID 0x%02x, type 0x%02x", elementaryPID, streamType);
385}
386
387ATSParser::Stream::~Stream() {
388    delete mQueue;
389    mQueue = NULL;
390}
391
392void ATSParser::Stream::parse(
393        unsigned payload_unit_start_indicator, ABitReader *br) {
394    if (payload_unit_start_indicator) {
395        if (mPayloadStarted) {
396            // Otherwise we run the danger of receiving the trailing bytes
397            // of a PES packet that we never saw the start of and assuming
398            // we have a a complete PES packet.
399
400            flush();
401        }
402
403        mPayloadStarted = true;
404    }
405
406    if (!mPayloadStarted) {
407        return;
408    }
409
410    size_t payloadSizeBits = br->numBitsLeft();
411    CHECK_EQ(payloadSizeBits % 8, 0u);
412
413    size_t neededSize = mBuffer->size() + payloadSizeBits / 8;
414    if (mBuffer->capacity() < neededSize) {
415        // Increment in multiples of 64K.
416        neededSize = (neededSize + 65535) & ~65535;
417
418        LOGI("resizing buffer to %d bytes", neededSize);
419
420        sp<ABuffer> newBuffer = new ABuffer(neededSize);
421        memcpy(newBuffer->data(), mBuffer->data(), mBuffer->size());
422        newBuffer->setRange(0, mBuffer->size());
423        mBuffer = newBuffer;
424    }
425
426    memcpy(mBuffer->data() + mBuffer->size(), br->data(), payloadSizeBits / 8);
427    mBuffer->setRange(0, mBuffer->size() + payloadSizeBits / 8);
428}
429
430void ATSParser::Stream::signalDiscontinuity(
431        DiscontinuityType type, const sp<AMessage> &extra) {
432    mPayloadStarted = false;
433    mBuffer->setRange(0, 0);
434
435    switch (type) {
436        case DISCONTINUITY_SEEK:
437        case DISCONTINUITY_FORMATCHANGE:
438        {
439            bool isASeek = (type == DISCONTINUITY_SEEK);
440
441            mQueue->clear(!isASeek);
442
443            uint64_t resumeAtPTS;
444            if (extra != NULL
445                    && extra->findInt64(
446                        IStreamListener::kKeyResumeAtPTS,
447                        (int64_t *)&resumeAtPTS)) {
448                int64_t resumeAtMediaTimeUs =
449                    mProgram->convertPTSToTimestamp(resumeAtPTS);
450
451                extra->setInt64("resume-at-mediatimeUs", resumeAtMediaTimeUs);
452            }
453
454            if (mSource != NULL) {
455                mSource->queueDiscontinuity(type, extra);
456            } else {
457                deferDiscontinuity(type, extra);
458            }
459            break;
460        }
461
462        default:
463            TRESPASS();
464            break;
465    }
466}
467
468void ATSParser::Stream::deferDiscontinuity(
469        DiscontinuityType type, const sp<AMessage> &extra) {
470    if (type > mPendingDiscontinuity) {
471        // Only upgrade discontinuities.
472        mPendingDiscontinuity = type;
473        mPendingDiscontinuityExtra = extra;
474    }
475}
476
477void ATSParser::Stream::signalEOS(status_t finalResult) {
478    if (mSource != NULL) {
479        mSource->signalEOS(finalResult);
480    }
481}
482
483void ATSParser::Stream::parsePES(ABitReader *br) {
484    unsigned packet_startcode_prefix = br->getBits(24);
485
486    LOGV("packet_startcode_prefix = 0x%08x", packet_startcode_prefix);
487
488    if (packet_startcode_prefix != 1) {
489        LOGV("Supposedly payload_unit_start=1 unit does not start "
490             "with startcode.");
491        return;
492    }
493
494    CHECK_EQ(packet_startcode_prefix, 0x000001u);
495
496    unsigned stream_id = br->getBits(8);
497    LOGV("stream_id = 0x%02x", stream_id);
498
499    unsigned PES_packet_length = br->getBits(16);
500    LOGV("PES_packet_length = %u", PES_packet_length);
501
502    if (stream_id != 0xbc  // program_stream_map
503            && stream_id != 0xbe  // padding_stream
504            && stream_id != 0xbf  // private_stream_2
505            && stream_id != 0xf0  // ECM
506            && stream_id != 0xf1  // EMM
507            && stream_id != 0xff  // program_stream_directory
508            && stream_id != 0xf2  // DSMCC
509            && stream_id != 0xf8) {  // H.222.1 type E
510        CHECK_EQ(br->getBits(2), 2u);
511
512        MY_LOGV("PES_scrambling_control = %u", br->getBits(2));
513        MY_LOGV("PES_priority = %u", br->getBits(1));
514        MY_LOGV("data_alignment_indicator = %u", br->getBits(1));
515        MY_LOGV("copyright = %u", br->getBits(1));
516        MY_LOGV("original_or_copy = %u", br->getBits(1));
517
518        unsigned PTS_DTS_flags = br->getBits(2);
519        LOGV("PTS_DTS_flags = %u", PTS_DTS_flags);
520
521        unsigned ESCR_flag = br->getBits(1);
522        LOGV("ESCR_flag = %u", ESCR_flag);
523
524        unsigned ES_rate_flag = br->getBits(1);
525        LOGV("ES_rate_flag = %u", ES_rate_flag);
526
527        unsigned DSM_trick_mode_flag = br->getBits(1);
528        LOGV("DSM_trick_mode_flag = %u", DSM_trick_mode_flag);
529
530        unsigned additional_copy_info_flag = br->getBits(1);
531        LOGV("additional_copy_info_flag = %u", additional_copy_info_flag);
532
533        MY_LOGV("PES_CRC_flag = %u", br->getBits(1));
534        MY_LOGV("PES_extension_flag = %u", br->getBits(1));
535
536        unsigned PES_header_data_length = br->getBits(8);
537        LOGV("PES_header_data_length = %u", PES_header_data_length);
538
539        unsigned optional_bytes_remaining = PES_header_data_length;
540
541        uint64_t PTS = 0, DTS = 0;
542
543        if (PTS_DTS_flags == 2 || PTS_DTS_flags == 3) {
544            CHECK_GE(optional_bytes_remaining, 5u);
545
546            CHECK_EQ(br->getBits(4), PTS_DTS_flags);
547
548            PTS = ((uint64_t)br->getBits(3)) << 30;
549            CHECK_EQ(br->getBits(1), 1u);
550            PTS |= ((uint64_t)br->getBits(15)) << 15;
551            CHECK_EQ(br->getBits(1), 1u);
552            PTS |= br->getBits(15);
553            CHECK_EQ(br->getBits(1), 1u);
554
555            LOGV("PTS = %llu", PTS);
556            // LOGI("PTS = %.2f secs", PTS / 90000.0f);
557
558            optional_bytes_remaining -= 5;
559
560            if (PTS_DTS_flags == 3) {
561                CHECK_GE(optional_bytes_remaining, 5u);
562
563                CHECK_EQ(br->getBits(4), 1u);
564
565                DTS = ((uint64_t)br->getBits(3)) << 30;
566                CHECK_EQ(br->getBits(1), 1u);
567                DTS |= ((uint64_t)br->getBits(15)) << 15;
568                CHECK_EQ(br->getBits(1), 1u);
569                DTS |= br->getBits(15);
570                CHECK_EQ(br->getBits(1), 1u);
571
572                LOGV("DTS = %llu", DTS);
573
574                optional_bytes_remaining -= 5;
575            }
576        }
577
578        if (ESCR_flag) {
579            CHECK_GE(optional_bytes_remaining, 6u);
580
581            br->getBits(2);
582
583            uint64_t ESCR = ((uint64_t)br->getBits(3)) << 30;
584            CHECK_EQ(br->getBits(1), 1u);
585            ESCR |= ((uint64_t)br->getBits(15)) << 15;
586            CHECK_EQ(br->getBits(1), 1u);
587            ESCR |= br->getBits(15);
588            CHECK_EQ(br->getBits(1), 1u);
589
590            LOGV("ESCR = %llu", ESCR);
591            MY_LOGV("ESCR_extension = %u", br->getBits(9));
592
593            CHECK_EQ(br->getBits(1), 1u);
594
595            optional_bytes_remaining -= 6;
596        }
597
598        if (ES_rate_flag) {
599            CHECK_GE(optional_bytes_remaining, 3u);
600
601            CHECK_EQ(br->getBits(1), 1u);
602            MY_LOGV("ES_rate = %u", br->getBits(22));
603            CHECK_EQ(br->getBits(1), 1u);
604
605            optional_bytes_remaining -= 3;
606        }
607
608        br->skipBits(optional_bytes_remaining * 8);
609
610        // ES data follows.
611
612        if (PES_packet_length != 0) {
613            CHECK_GE(PES_packet_length, PES_header_data_length + 3);
614
615            unsigned dataLength =
616                PES_packet_length - 3 - PES_header_data_length;
617
618            CHECK_GE(br->numBitsLeft(), dataLength * 8);
619
620            onPayloadData(
621                    PTS_DTS_flags, PTS, DTS, br->data(), dataLength);
622
623            br->skipBits(dataLength * 8);
624        } else {
625            onPayloadData(
626                    PTS_DTS_flags, PTS, DTS,
627                    br->data(), br->numBitsLeft() / 8);
628
629            size_t payloadSizeBits = br->numBitsLeft();
630            CHECK_EQ(payloadSizeBits % 8, 0u);
631
632            LOGV("There's %d bytes of payload.", payloadSizeBits / 8);
633        }
634    } else if (stream_id == 0xbe) {  // padding_stream
635        CHECK_NE(PES_packet_length, 0u);
636        br->skipBits(PES_packet_length * 8);
637    } else {
638        CHECK_NE(PES_packet_length, 0u);
639        br->skipBits(PES_packet_length * 8);
640    }
641}
642
643void ATSParser::Stream::flush() {
644    if (mBuffer->size() == 0) {
645        return;
646    }
647
648    LOGV("flushing stream 0x%04x size = %d", mElementaryPID, mBuffer->size());
649
650    ABitReader br(mBuffer->data(), mBuffer->size());
651    parsePES(&br);
652
653    mBuffer->setRange(0, 0);
654}
655
656void ATSParser::Stream::onPayloadData(
657        unsigned PTS_DTS_flags, uint64_t PTS, uint64_t DTS,
658        const uint8_t *data, size_t size) {
659    LOGV("onPayloadData mStreamType=0x%02x", mStreamType);
660
661    if (mQueue == NULL) {
662        return;
663    }
664
665    CHECK(PTS_DTS_flags == 2 || PTS_DTS_flags == 3);
666    int64_t timeUs = mProgram->convertPTSToTimestamp(PTS);
667
668    status_t err = mQueue->appendData(data, size, timeUs);
669
670    if (err != OK) {
671        return;
672    }
673
674    sp<ABuffer> accessUnit;
675    while ((accessUnit = mQueue->dequeueAccessUnit()) != NULL) {
676        if (mSource == NULL) {
677            sp<MetaData> meta = mQueue->getFormat();
678
679            if (meta != NULL) {
680                LOGV("Stream PID 0x%08x of type 0x%02x now has data.",
681                     mElementaryPID, mStreamType);
682
683                mSource = new AnotherPacketSource(meta);
684
685                if (mPendingDiscontinuity != DISCONTINUITY_NONE) {
686                    mSource->queueDiscontinuity(
687                            mPendingDiscontinuity, mPendingDiscontinuityExtra);
688                    mPendingDiscontinuity = DISCONTINUITY_NONE;
689                    mPendingDiscontinuityExtra.clear();
690                }
691
692                mSource->queueAccessUnit(accessUnit);
693            }
694        } else if (mQueue->getFormat() != NULL) {
695            // After a discontinuity we invalidate the queue's format
696            // and won't enqueue any access units to the source until
697            // the queue has reestablished the new format.
698
699            if (mSource->getFormat() == NULL) {
700                mSource->setFormat(mQueue->getFormat());
701            }
702            mSource->queueAccessUnit(accessUnit);
703        }
704    }
705}
706
707sp<MediaSource> ATSParser::Stream::getSource(SourceType type) {
708    switch (type) {
709        case VIDEO:
710        {
711            if (mStreamType == STREAMTYPE_H264
712                    || mStreamType == STREAMTYPE_MPEG1_VIDEO
713                    || mStreamType == STREAMTYPE_MPEG2_VIDEO
714                    || mStreamType == STREAMTYPE_MPEG4_VIDEO) {
715                return mSource;
716            }
717            break;
718        }
719
720        case AUDIO:
721        {
722            if (mStreamType == STREAMTYPE_MPEG1_AUDIO
723                    || mStreamType == STREAMTYPE_MPEG2_AUDIO
724                    || mStreamType == STREAMTYPE_MPEG2_AUDIO_ATDS) {
725                return mSource;
726            }
727            break;
728        }
729
730        default:
731            break;
732    }
733
734    return NULL;
735}
736
737////////////////////////////////////////////////////////////////////////////////
738
739ATSParser::ATSParser(uint32_t flags)
740    : mFlags(flags) {
741}
742
743ATSParser::~ATSParser() {
744}
745
746void ATSParser::feedTSPacket(const void *data, size_t size) {
747    CHECK_EQ(size, kTSPacketSize);
748
749    ABitReader br((const uint8_t *)data, kTSPacketSize);
750    parseTS(&br);
751}
752
753void ATSParser::signalDiscontinuity(
754        DiscontinuityType type, const sp<AMessage> &extra) {
755    for (size_t i = 0; i < mPrograms.size(); ++i) {
756        mPrograms.editItemAt(i)->signalDiscontinuity(type, extra);
757    }
758}
759
760void ATSParser::signalEOS(status_t finalResult) {
761    CHECK_NE(finalResult, (status_t)OK);
762
763    for (size_t i = 0; i < mPrograms.size(); ++i) {
764        mPrograms.editItemAt(i)->signalEOS(finalResult);
765    }
766}
767
768void ATSParser::parseProgramAssociationTable(ABitReader *br) {
769    unsigned table_id = br->getBits(8);
770    LOGV("  table_id = %u", table_id);
771    CHECK_EQ(table_id, 0x00u);
772
773    unsigned section_syntax_indictor = br->getBits(1);
774    LOGV("  section_syntax_indictor = %u", section_syntax_indictor);
775    CHECK_EQ(section_syntax_indictor, 1u);
776
777    CHECK_EQ(br->getBits(1), 0u);
778    MY_LOGV("  reserved = %u", br->getBits(2));
779
780    unsigned section_length = br->getBits(12);
781    LOGV("  section_length = %u", section_length);
782    CHECK_EQ(section_length & 0xc00, 0u);
783
784    MY_LOGV("  transport_stream_id = %u", br->getBits(16));
785    MY_LOGV("  reserved = %u", br->getBits(2));
786    MY_LOGV("  version_number = %u", br->getBits(5));
787    MY_LOGV("  current_next_indicator = %u", br->getBits(1));
788    MY_LOGV("  section_number = %u", br->getBits(8));
789    MY_LOGV("  last_section_number = %u", br->getBits(8));
790
791    size_t numProgramBytes = (section_length - 5 /* header */ - 4 /* crc */);
792    CHECK_EQ((numProgramBytes % 4), 0u);
793
794    for (size_t i = 0; i < numProgramBytes / 4; ++i) {
795        unsigned program_number = br->getBits(16);
796        LOGV("    program_number = %u", program_number);
797
798        MY_LOGV("    reserved = %u", br->getBits(3));
799
800        if (program_number == 0) {
801            MY_LOGV("    network_PID = 0x%04x", br->getBits(13));
802        } else {
803            unsigned programMapPID = br->getBits(13);
804
805            LOGV("    program_map_PID = 0x%04x", programMapPID);
806
807            bool found = false;
808            for (size_t index = 0; index < mPrograms.size(); ++index) {
809                const sp<Program> &program = mPrograms.itemAt(index);
810
811                if (program->number() == program_number) {
812                    program->updateProgramMapPID(programMapPID);
813                    found = true;
814                    break;
815                }
816            }
817
818            if (!found) {
819                mPrograms.push(
820                        new Program(this, program_number, programMapPID));
821            }
822        }
823    }
824
825    MY_LOGV("  CRC = 0x%08x", br->getBits(32));
826}
827
828void ATSParser::parsePID(
829        ABitReader *br, unsigned PID,
830        unsigned payload_unit_start_indicator) {
831    if (PID == 0) {
832        if (payload_unit_start_indicator) {
833            unsigned skip = br->getBits(8);
834            br->skipBits(skip * 8);
835        }
836        parseProgramAssociationTable(br);
837        return;
838    }
839
840    bool handled = false;
841    for (size_t i = 0; i < mPrograms.size(); ++i) {
842        if (mPrograms.editItemAt(i)->parsePID(
843                    PID, payload_unit_start_indicator, br)) {
844            handled = true;
845            break;
846        }
847    }
848
849    if (!handled) {
850        LOGV("PID 0x%04x not handled.", PID);
851    }
852}
853
854void ATSParser::parseAdaptationField(ABitReader *br) {
855    unsigned adaptation_field_length = br->getBits(8);
856    if (adaptation_field_length > 0) {
857        br->skipBits(adaptation_field_length * 8);  // XXX
858    }
859}
860
861void ATSParser::parseTS(ABitReader *br) {
862    LOGV("---");
863
864    unsigned sync_byte = br->getBits(8);
865    CHECK_EQ(sync_byte, 0x47u);
866
867    MY_LOGV("transport_error_indicator = %u", br->getBits(1));
868
869    unsigned payload_unit_start_indicator = br->getBits(1);
870    LOGV("payload_unit_start_indicator = %u", payload_unit_start_indicator);
871
872    MY_LOGV("transport_priority = %u", br->getBits(1));
873
874    unsigned PID = br->getBits(13);
875    LOGV("PID = 0x%04x", PID);
876
877    MY_LOGV("transport_scrambling_control = %u", br->getBits(2));
878
879    unsigned adaptation_field_control = br->getBits(2);
880    LOGV("adaptation_field_control = %u", adaptation_field_control);
881
882    unsigned continuity_counter = br->getBits(4);
883    LOGV("continuity_counter = %u", continuity_counter);
884
885    // LOGI("PID = 0x%04x, continuity_counter = %u", PID, continuity_counter);
886
887    if (adaptation_field_control == 2 || adaptation_field_control == 3) {
888        parseAdaptationField(br);
889    }
890
891    if (adaptation_field_control == 1 || adaptation_field_control == 3) {
892        parsePID(br, PID, payload_unit_start_indicator);
893    }
894}
895
896sp<MediaSource> ATSParser::getSource(SourceType type) {
897    int which = -1;  // any
898
899    for (size_t i = 0; i < mPrograms.size(); ++i) {
900        const sp<Program> &program = mPrograms.editItemAt(i);
901
902        if (which >= 0 && (int)program->number() != which) {
903            continue;
904        }
905
906        sp<MediaSource> source = program->getSource(type);
907
908        if (source != NULL) {
909            return source;
910        }
911    }
912
913    return NULL;
914}
915
916bool ATSParser::PTSTimeDeltaEstablished() {
917    if (mPrograms.isEmpty()) {
918        return false;
919    }
920
921    return mPrograms.editItemAt(0)->PTSTimeDeltaEstablished();
922}
923
924}  // namespace android
925