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 SAMPLE_TABLE_H_
18
19#define SAMPLE_TABLE_H_
20
21#include <sys/types.h>
22#include <stdint.h>
23
24#include <media/stagefright/MediaErrors.h>
25#include <utils/RefBase.h>
26#include <utils/threads.h>
27
28namespace android {
29
30class DataSourceBase;
31struct SampleIterator;
32
33class SampleTable : public RefBase {
34public:
35    explicit SampleTable(DataSourceBase *source);
36
37    bool isValid() const;
38
39    // type can be 'stco' or 'co64'.
40    status_t setChunkOffsetParams(
41            uint32_t type, off64_t data_offset, size_t data_size);
42
43    status_t setSampleToChunkParams(off64_t data_offset, size_t data_size);
44
45    // type can be 'stsz' or 'stz2'.
46    status_t setSampleSizeParams(
47            uint32_t type, off64_t data_offset, size_t data_size);
48
49    status_t setTimeToSampleParams(off64_t data_offset, size_t data_size);
50
51    status_t setCompositionTimeToSampleParams(
52            off64_t data_offset, size_t data_size);
53
54    status_t setSyncSampleParams(off64_t data_offset, size_t data_size);
55
56    ////////////////////////////////////////////////////////////////////////////
57
58    uint32_t countChunkOffsets() const;
59
60    uint32_t countSamples() const;
61
62    status_t getMaxSampleSize(size_t *size);
63
64    status_t getMetaDataForSample(
65            uint32_t sampleIndex,
66            off64_t *offset,
67            size_t *size,
68            uint32_t *compositionTime,
69            bool *isSyncSample = NULL,
70            uint32_t *sampleDuration = NULL);
71
72    enum {
73        kFlagBefore,
74        kFlagAfter,
75        kFlagClosest,
76        kFlagFrameIndex,
77    };
78    status_t findSampleAtTime(
79            uint64_t req_time, uint64_t scale_num, uint64_t scale_den,
80            uint32_t *sample_index, uint32_t flags);
81
82    status_t findSyncSampleNear(
83            uint32_t start_sample_index, uint32_t *sample_index,
84            uint32_t flags);
85
86    status_t findThumbnailSample(uint32_t *sample_index);
87
88protected:
89    ~SampleTable();
90
91private:
92    struct CompositionDeltaLookup;
93
94    static const uint32_t kChunkOffsetType32;
95    static const uint32_t kChunkOffsetType64;
96    static const uint32_t kSampleSizeType32;
97    static const uint32_t kSampleSizeTypeCompact;
98
99    // Limit the total size of all internal tables to 200MiB.
100    static const size_t kMaxTotalSize = 200 * (1 << 20);
101
102    DataSourceBase *mDataSource;
103    Mutex mLock;
104
105    off64_t mChunkOffsetOffset;
106    uint32_t mChunkOffsetType;
107    uint32_t mNumChunkOffsets;
108
109    off64_t mSampleToChunkOffset;
110    uint32_t mNumSampleToChunkOffsets;
111
112    off64_t mSampleSizeOffset;
113    uint32_t mSampleSizeFieldSize;
114    uint32_t mDefaultSampleSize;
115    uint32_t mNumSampleSizes;
116
117    bool mHasTimeToSample;
118    uint32_t mTimeToSampleCount;
119    uint32_t* mTimeToSample;
120
121    struct SampleTimeEntry {
122        uint32_t mSampleIndex;
123        uint32_t mCompositionTime;
124    };
125    SampleTimeEntry *mSampleTimeEntries;
126
127    int32_t *mCompositionTimeDeltaEntries;
128    size_t mNumCompositionTimeDeltaEntries;
129    CompositionDeltaLookup *mCompositionDeltaLookup;
130
131    off64_t mSyncSampleOffset;
132    uint32_t mNumSyncSamples;
133    uint32_t *mSyncSamples;
134    size_t mLastSyncSampleIndex;
135
136    SampleIterator *mSampleIterator;
137
138    struct SampleToChunkEntry {
139        uint32_t startChunk;
140        uint32_t samplesPerChunk;
141        uint32_t chunkDesc;
142    };
143    SampleToChunkEntry *mSampleToChunkEntries;
144
145    // Approximate size of all tables combined.
146    uint64_t mTotalSize;
147
148    friend struct SampleIterator;
149
150    // normally we don't round
151    inline uint64_t getSampleTime(
152            size_t sample_index, uint64_t scale_num, uint64_t scale_den) const {
153        return (sample_index < (size_t)mNumSampleSizes && mSampleTimeEntries != NULL
154                && scale_den != 0)
155                ? (mSampleTimeEntries[sample_index].mCompositionTime * scale_num) / scale_den : 0;
156    }
157
158    status_t getSampleSize_l(uint32_t sample_index, size_t *sample_size);
159    int32_t getCompositionTimeOffset(uint32_t sampleIndex);
160
161    static int CompareIncreasingTime(const void *, const void *);
162
163    void buildSampleEntriesTable();
164
165    SampleTable(const SampleTable &);
166    SampleTable &operator=(const SampleTable &);
167};
168
169}  // namespace android
170
171#endif  // SAMPLE_TABLE_H_
172