51 lines
1.2 KiB
C++
51 lines
1.2 KiB
C++
/*
|
|
* 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_ */
|