MPEG4Extractor.h revision c7fc37a3dab9bd1f96713649f351b5990e6316ff
1/*
2 * Copyright (C) 2009 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#ifndef MPEG4_EXTRACTOR_H_
18
19#define MPEG4_EXTRACTOR_H_
20
21#include <media/stagefright/MediaExtractor.h>
22#include <utils/Vector.h>
23
24namespace android {
25
26struct AMessage;
27class DataSource;
28class SampleTable;
29class String8;
30
31class MPEG4Extractor : public MediaExtractor {
32public:
33    // Extractor assumes ownership of "source".
34    MPEG4Extractor(const sp<DataSource> &source);
35
36    virtual size_t countTracks();
37    virtual sp<MediaSource> getTrack(size_t index);
38    virtual sp<MetaData> getTrackMetaData(size_t index, uint32_t flags);
39
40    virtual sp<MetaData> getMetaData();
41
42    // for DRM
43    virtual void setDrmFlag(bool flag);
44    virtual char* getDrmTrackInfo(size_t trackID, int *len);
45
46protected:
47    virtual ~MPEG4Extractor();
48
49private:
50    struct Track {
51        Track *next;
52        sp<MetaData> meta;
53        uint32_t timescale;
54        sp<SampleTable> sampleTable;
55        bool includes_expensive_metadata;
56        bool skipTrack;
57    };
58
59    sp<DataSource> mDataSource;
60    bool mHaveMetadata;
61    bool mHasVideo;
62
63    Track *mFirstTrack, *mLastTrack;
64
65    sp<MetaData> mFileMetaData;
66
67    Vector<uint32_t> mPath;
68
69    status_t readMetaData();
70    status_t parseChunk(off64_t *offset, int depth);
71    status_t parseMetaData(off64_t offset, size_t size);
72
73    status_t updateAudioTrackInfoFromESDS_MPEG4Audio(
74            const void *esds_data, size_t esds_size);
75
76    static status_t verifyTrack(Track *track);
77
78    struct SINF {
79        SINF *next;
80        uint16_t trackID;
81        uint8_t IPMPDescriptorID;
82        ssize_t len;
83        char *IPMPData;
84    };
85
86    SINF *mFirstSINF;
87
88    bool mIsDrm;
89    status_t parseDrmSINF(off64_t *offset, off64_t data_offset);
90
91    status_t parseTrackHeader(off64_t data_offset, off64_t data_size);
92
93    MPEG4Extractor(const MPEG4Extractor &);
94    MPEG4Extractor &operator=(const MPEG4Extractor &);
95};
96
97bool SniffMPEG4(
98        const sp<DataSource> &source, String8 *mimeType, float *confidence,
99        sp<AMessage> *);
100
101}  // namespace android
102
103#endif  // MPEG4_EXTRACTOR_H_
104