MPEG2TSWriter.h revision 59b7dc39ea8332d3418a599e51447d7edb612ac4 
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    virtual status_t addSource(const sp<MediaSource> &source);
35    virtual status_t start(MetaData *param = NULL);
36    virtual status_t stop();
37    virtual status_t pause();
38    virtual bool reachedEOS();
39    virtual status_t dump(int fd, const Vector<String16>& args);
40
41    void onMessageReceived(const sp<AMessage> &msg);
42
43protected:
44    virtual ~MPEG2TSWriter();
45
46private:
47    enum {
48        kWhatSourceNotify = 'noti'
49    };
50
51    struct SourceInfo;
52
53    FILE *mFile;
54    sp<ALooper> mLooper;
55    sp<AHandlerReflector<MPEG2TSWriter> > mReflector;
56
57    bool mStarted;
58
59    Vector<sp<SourceInfo> > mSources;
60    size_t mNumSourcesDone;
61
62    int64_t mNumTSPacketsWritten;
63    int64_t mNumTSPacketsBeforeMeta;
64
65    void init();
66
67    void writeTS();
68    void writeProgramAssociationTable();
69    void writeProgramMap();
70    void writeAccessUnit(int32_t sourceIndex, const sp<ABuffer> &buffer);
71
72    DISALLOW_EVIL_CONSTRUCTORS(MPEG2TSWriter);
73};
74
75}  // namespace android
76
77#endif  // MPEG2TS_WRITER_H_
78