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 AVC_ENCODER_H_
18
19#define AVC_ENCODER_H_
20
21#include <media/stagefright/MediaBuffer.h>
22#include <media/stagefright/MediaSource.h>
23#include <utils/Vector.h>
24
25struct tagAVCHandle;
26struct tagAVCEncParam;
27
28namespace android {
29
30struct MediaBuffer;
31struct MediaBufferGroup;
32
33struct AVCEncoder : public MediaSource,
34                    public MediaBufferObserver {
35    AVCEncoder(const sp<MediaSource> &source,
36            const sp<MetaData>& meta);
37
38    virtual status_t start(MetaData *params);
39    virtual status_t stop();
40
41    virtual sp<MetaData> getFormat();
42
43    virtual status_t read(
44            MediaBuffer **buffer, const ReadOptions *options);
45
46    virtual void signalBufferReturned(MediaBuffer *buffer);
47
48    // Callbacks required by the encoder
49    int32_t allocOutputBuffers(unsigned int sizeInMbs, unsigned int numBuffers);
50    void    unbindOutputBuffer(int32_t index);
51    int32_t bindOutputBuffer(int32_t index, uint8_t **yuv);
52
53protected:
54    virtual ~AVCEncoder();
55
56private:
57    sp<MediaSource> mSource;
58    sp<MetaData>    mFormat;
59    sp<MetaData>    mMeta;
60
61    int32_t  mVideoWidth;
62    int32_t  mVideoHeight;
63    int32_t  mVideoFrameRate;
64    int32_t  mVideoBitRate;
65    int32_t  mVideoColorFormat;
66    int64_t  mNumInputFrames;
67    int64_t  mPrevTimestampUs;
68    status_t mInitCheck;
69    bool     mStarted;
70    bool     mSpsPpsHeaderReceived;
71    bool     mReadyForNextFrame;
72    int32_t  mIsIDRFrame;  // for set kKeyIsSyncFrame
73
74    tagAVCHandle          *mHandle;
75    tagAVCEncParam        *mEncParams;
76    MediaBuffer           *mInputBuffer;
77    uint8_t               *mInputFrameData;
78    MediaBufferGroup      *mGroup;
79    Vector<MediaBuffer *> mOutputBuffers;
80
81
82    status_t initCheck(const sp<MetaData>& meta);
83    void releaseOutputBuffers();
84
85    AVCEncoder(const AVCEncoder &);
86    AVCEncoder &operator=(const AVCEncoder &);
87};
88
89}  // namespace android
90
91#endif  // AVC_ENCODER_H_
92