MPEG2TSWriter.h revision 983995747ed3adaf25035c433df1ccbaa056a898
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 MPEG2TS_WRITER_H_ 18 19#define MPEG2TS_WRITER_H_ 20 21#include <media/stagefright/foundation/ABase.h> 22#include <media/stagefright/foundation/AHandlerReflector.h> 23#include <media/stagefright/foundation/ALooper.h> 24#include <media/stagefright/MediaWriter.h> 25 26namespace android { 27 28struct ABuffer; 29 30struct MPEG2TSWriter : public MediaWriter { 31 MPEG2TSWriter(int fd); 32 MPEG2TSWriter(const char *filename); 33 34 MPEG2TSWriter( 35 void *cookie, 36 ssize_t (*write)(void *cookie, const void *data, size_t size)); 37 38 virtual status_t addSource(const sp<MediaSource> &source); 39 virtual status_t start(MetaData *param = NULL); 40 virtual status_t stop() { return reset(); } 41 virtual status_t pause(); 42 virtual bool reachedEOS(); 43 virtual status_t dump(int fd, const Vector<String16>& args); 44 45 void onMessageReceived(const sp<AMessage> &msg); 46 47protected: 48 virtual ~MPEG2TSWriter(); 49 50private: 51 enum { 52 kWhatSourceNotify = 'noti' 53 }; 54 55 struct SourceInfo; 56 57 FILE *mFile; 58 59 void *mWriteCookie; 60 ssize_t (*mWriteFunc)(void *cookie, const void *data, size_t size); 61 62 sp<ALooper> mLooper; 63 sp<AHandlerReflector<MPEG2TSWriter> > mReflector; 64 65 bool mStarted; 66 67 Vector<sp<SourceInfo> > mSources; 68 size_t mNumSourcesDone; 69 70 int64_t mNumTSPacketsWritten; 71 int64_t mNumTSPacketsBeforeMeta; 72 int mPATContinuityCounter; 73 int mPMTContinuityCounter; 74 uint32_t mCrcTable[256]; 75 76 void init(); 77 78 void writeTS(); 79 void writeProgramAssociationTable(); 80 void writeProgramMap(); 81 void writeAccessUnit(int32_t sourceIndex, const sp<ABuffer> &buffer); 82 void initCrcTable(); 83 uint32_t crc32(const uint8_t *start, size_t length); 84 85 ssize_t internalWrite(const void *data, size_t size); 86 status_t reset(); 87 88 DISALLOW_EVIL_CONSTRUCTORS(MPEG2TSWriter); 89}; 90 91} // namespace android 92 93#endif // MPEG2TS_WRITER_H_ 94