ZLMediaKit/src/Codec/H264Encoder.h

/*
 * Copyright (c) 2016-present The ZLMediaKit project authors. All Rights Reserved.
 *
 * This file is part of ZLMediaKit(https://github.com/ZLMediaKit/ZLMediaKit).
 *
 * Use of this source code is governed by MIT-like license that can be found in the
 * LICENSE file in the root of the source tree. All contributing project authors
 * may be found in the AUTHORS file in the root of the source tree.
 */
#ifndef CODEC_H264ENCODER_H_
#define CODEC_H264ENCODER_H_

#include <cstdint>
#ifdef __cplusplus
extern "C" {
#endif //__cplusplus
#include <x264.h>
#ifdef __cplusplus
}
#endif //__cplusplus

namespace mediakit {

class H264Encoder {
public:
    typedef struct {
        int iType;
        int iLength;
        uint8_t *pucData;

        int64_t dts;
        int64_t pts;
    } H264Frame;

    H264Encoder();
    ~H264Encoder();

    bool init(int iWidth, int iHeight, int iFps, int iBitRate);
    int inputData(char *yuv[3], int linesize[3], int64_t cts, H264Frame **out_frame);

private:
    x264_t *_pX264Handle = nullptr;
    x264_picture_t *_pPicIn = nullptr;
    x264_picture_t *_pPicOut = nullptr;
    H264Frame _aFrames[10];
};

} /* namespace mediakit */

#endif /* CODEC_H264ENCODER_H_ */