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