MatroskaExtractor.h revision 2f46e8152fb881d3a1d7afd223f1ed51f6e358b8
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#ifndef MATROSKA_EXTRACTOR_H_ 18 19#define MATROSKA_EXTRACTOR_H_ 20 21#include "mkvparser.hpp" 22 23#include <media/stagefright/MediaExtractor.h> 24#include <utils/Vector.h> 25#include <utils/threads.h> 26 27namespace android { 28 29struct AMessage; 30class String8; 31 32struct DataSourceReader; 33struct MatroskaSource; 34 35struct MatroskaExtractor : public MediaExtractor { 36 MatroskaExtractor(const sp<DataSource> &source); 37 38 virtual size_t countTracks(); 39 40 virtual sp<MediaSource> getTrack(size_t index); 41 42 virtual sp<MetaData> getTrackMetaData( 43 size_t index, uint32_t flags); 44 45 virtual sp<MetaData> getMetaData(); 46 47 virtual uint32_t flags() const; 48 49protected: 50 virtual ~MatroskaExtractor(); 51 52private: 53 friend struct MatroskaSource; 54 friend struct BlockIterator; 55 56 struct TrackInfo { 57 unsigned long mTrackNum; 58 sp<MetaData> mMeta; 59 const MatroskaExtractor *mExtractor; 60 Vector<const mkvparser::CuePoint*> mCuePoints; 61 62 const mkvparser::Track* getTrack() const; 63 const mkvparser::CuePoint::TrackPosition *find(long long timeNs) const; 64 }; 65 66 Mutex mLock; 67 Vector<TrackInfo> mTracks; 68 69 sp<DataSource> mDataSource; 70 DataSourceReader *mReader; 71 mkvparser::Segment *mSegment; 72 bool mExtractedThumbnails; 73 bool mIsLiveStreaming; 74 bool mIsWebm; 75 int64_t mSeekPreRollNs; 76 77 void addTracks(); 78 void findThumbnails(); 79 80 bool isLiveStreaming() const; 81 82 MatroskaExtractor(const MatroskaExtractor &); 83 MatroskaExtractor &operator=(const MatroskaExtractor &); 84}; 85 86bool SniffMatroska( 87 const sp<DataSource> &source, String8 *mimeType, float *confidence, 88 sp<AMessage> *); 89 90} // namespace android 91 92#endif // MATROSKA_EXTRACTOR_H_ 93