1/*
2 * Copyright 2018 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 C2_SOFT_MPEG4_ENC_H__
18#define C2_SOFT_MPEG4_ENC_H__
19
20#include <map>
21
22#include <Codec2BufferUtils.h>
23#include <SimpleC2Component.h>
24
25#include "mp4enc_api.h"
26
27namespace android {
28
29struct C2SoftMpeg4Enc : public SimpleC2Component {
30    class IntfImpl;
31
32    C2SoftMpeg4Enc(const char* name, c2_node_id_t id,
33                   const std::shared_ptr<IntfImpl>& intfImpl);
34
35    // From SimpleC2Component
36    c2_status_t onInit() override;
37    c2_status_t onStop() override;
38    void onReset() override;
39    void onRelease() override;
40    c2_status_t onFlush_sm() override;
41    void process(
42            const std::unique_ptr<C2Work> &work,
43            const std::shared_ptr<C2BlockPool> &pool) override;
44    c2_status_t drain(
45            uint32_t drainMode,
46            const std::shared_ptr<C2BlockPool> &pool) override;
47
48protected:
49
50    virtual ~C2SoftMpeg4Enc();
51
52private:
53    std::shared_ptr<IntfImpl> mIntf;
54
55    tagvideoEncControls *mHandle;
56    tagvideoEncOptions  *mEncParams;
57
58    bool     mStarted;
59    bool     mSignalledOutputEos;
60    bool     mSignalledError;
61
62    uint32_t mOutBufferSize;
63    // 1: all I-frames, <0: infinite
64    int32_t  mKeyFrameInterval;
65    int64_t  mNumInputFrames;
66    MP4EncodingMode mEncodeMode;
67
68    MemoryBlockPool mConversionBuffers;
69    std::map<void *, MemoryBlock> mConversionBuffersInUse;
70
71    c2_status_t initEncParams();
72    c2_status_t initEncoder();
73
74    C2_DO_NOT_COPY(C2SoftMpeg4Enc);
75};
76
77}  // namespace android
78
79#endif  // C2_SOFT_MPEG4_ENC_H__
80