MPEG4Writer.h revision 22b37fa6e0d7abbfa373e5fdf057fc6dcb2474ff
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_WRITER_H_ 18 19#define MPEG4_WRITER_H_ 20 21#include <stdio.h> 22 23#include <media/stagefright/MediaWriter.h> 24#include <utils/List.h> 25#include <utils/threads.h> 26 27namespace android { 28 29class MediaBuffer; 30class MediaSource; 31class MetaData; 32 33class MPEG4Writer : public MediaWriter { 34public: 35 MPEG4Writer(const char *filename); 36 MPEG4Writer(int fd); 37 38 virtual status_t addSource(const sp<MediaSource> &source); 39 virtual status_t start(MetaData *param = NULL); 40 virtual status_t stop(); 41 virtual status_t pause(); 42 virtual bool reachedEOS(); 43 virtual status_t dump(int fd, const Vector<String16>& args); 44 45 void beginBox(const char *fourcc); 46 void writeInt8(int8_t x); 47 void writeInt16(int16_t x); 48 void writeInt32(int32_t x); 49 void writeInt64(int64_t x); 50 void writeCString(const char *s); 51 void writeFourcc(const char *fourcc); 52 void write(const void *data, size_t size); 53 void endBox(); 54 uint32_t interleaveDuration() const { return mInterleaveDurationUs; } 55 status_t setInterleaveDuration(uint32_t duration); 56 int32_t getTimeScale() const { return mTimeScale; } 57 58protected: 59 virtual ~MPEG4Writer(); 60 61private: 62 class Track; 63 64 FILE *mFile; 65 bool mUse4ByteNalLength; 66 bool mUse32BitOffset; 67 bool mIsFileSizeLimitExplicitlyRequested; 68 bool mPaused; 69 bool mStarted; 70 off_t mOffset; 71 off_t mMdatOffset; 72 uint8_t *mMoovBoxBuffer; 73 off_t mMoovBoxBufferOffset; 74 bool mWriteMoovBoxToMemory; 75 off_t mFreeBoxOffset; 76 bool mStreamableFile; 77 off_t mEstimatedMoovBoxSize; 78 uint32_t mInterleaveDurationUs; 79 int32_t mTimeScale; 80 int64_t mStartTimestampUs; 81 82 Mutex mLock; 83 84 List<Track *> mTracks; 85 86 List<off_t> mBoxes; 87 88 void setStartTimestampUs(int64_t timeUs); 89 int64_t getStartTimestampUs(); // Not const 90 status_t startTracks(MetaData *params); 91 size_t numTracks(); 92 int64_t estimateMoovBoxSize(int32_t bitRate); 93 94 struct Chunk { 95 Track *mTrack; // Owner 96 int64_t mTimeStampUs; // Timestamp of the 1st sample 97 List<MediaBuffer *> mSamples; // Sample data 98 99 // Convenient constructor 100 Chunk(Track *track, int64_t timeUs, List<MediaBuffer *> samples) 101 : mTrack(track), mTimeStampUs(timeUs), mSamples(samples) { 102 } 103 104 }; 105 struct ChunkInfo { 106 Track *mTrack; // Owner 107 List<Chunk> mChunks; // Remaining chunks to be written 108 }; 109 110 bool mIsFirstChunk; 111 volatile bool mDone; // Writer thread is done? 112 pthread_t mThread; // Thread id for the writer 113 List<ChunkInfo> mChunkInfos; // Chunk infos 114 Condition mChunkReadyCondition; // Signal that chunks are available 115 116 // Writer thread handling 117 status_t startWriterThread(); 118 void stopWriterThread(); 119 static void *ThreadWrapper(void *me); 120 void threadFunc(); 121 122 // Buffer a single chunk to be written out later. 123 void bufferChunk(const Chunk& chunk); 124 125 // Write all buffered chunks from all tracks 126 void writeChunks(); 127 128 // Write a chunk if there is one 129 status_t writeOneChunk(); 130 131 // Write the first chunk from the given ChunkInfo. 132 void writeFirstChunk(ChunkInfo* info); 133 134 // Adjust other track media clock (presumably wall clock) 135 // based on audio track media clock with the drift time. 136 int64_t mDriftTimeUs; 137 void setDriftTimeUs(int64_t driftTimeUs); 138 int64_t getDriftTimeUs(); 139 140 // Return whether the nal length is 4 bytes or 2 bytes 141 // Only makes sense for H.264/AVC 142 bool useNalLengthFour(); 143 144 void lock(); 145 void unlock(); 146 147 // Acquire lock before calling these methods 148 off_t addSample_l(MediaBuffer *buffer); 149 off_t addLengthPrefixedSample_l(MediaBuffer *buffer); 150 151 inline size_t write(const void *ptr, size_t size, size_t nmemb, FILE* stream); 152 bool exceedsFileSizeLimit(); 153 bool use32BitFileOffset() const; 154 bool exceedsFileDurationLimit(); 155 bool isFileStreamable() const; 156 void trackProgressStatus(const Track* track, int64_t timeUs, status_t err = OK); 157 158 MPEG4Writer(const MPEG4Writer &); 159 MPEG4Writer &operator=(const MPEG4Writer &); 160}; 161 162} // namespace android 163 164#endif // MPEG4_WRITER_H_ 165