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