YangWX/crossdesk
1
0
Fork 0
mirror of https://github.com/kunkundi/crossdesk.git synced 2025-10-27 04:35:34 +08:00
Code Issues Packages Projects Releases Wiki Activity

Use OpenH264 as default ffmpeg codec

Browse Source
This commit is contained in:
dijunkun
2023-11-07 00:11:23 -08:00
parent e456ae3577
commit 99360f5da2
2 changed files with 15 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

34
src/media/video/encode/ffmpeg/ffmpeg_video_encoder.cpp
View File

@@ -65,14 +65,13 @@ int FFmpegVideoEncoder::Init() {
if (!codec_) { if (!codec_) {
LOG_ERROR("Failed to find H.264 encoder"); LOG_ERROR("Failed to find H.264 encoder");
return -1; return -1;
} else {
if (0 == strcmp(codec_->name, "openh264")) {
use_openh264_ = true;
LOG_INFO("Use H264 encoder [OpenH264]");
}
} }
// else { use_openh264_ = true;
// LOG_INFO("Use H264 encoder [{}]", codec_->name);
// if (0 == strcmp(codec_->name, "libx264")) {
// use_libx264_ = true;
// }
// }
// use_libx264_ = true;
codec_ctx_ = avcodec_alloc_context3(codec_); codec_ctx_ = avcodec_alloc_context3(codec_);
if (!codec_ctx_) { if (!codec_ctx_) {
@@ -86,7 +85,7 @@ int FFmpegVideoEncoder::Init() {
codec_ctx_->height = frame_height; codec_ctx_->height = frame_height;
codec_ctx_->time_base.num = 1; codec_ctx_->time_base.num = 1;
codec_ctx_->time_base.den = fps_; codec_ctx_->time_base.den = fps_;
if (use_libx264_) { if (use_openh264_) {
codec_ctx_->pix_fmt = AV_PIX_FMT_YUV420P; codec_ctx_->pix_fmt = AV_PIX_FMT_YUV420P;
} else { } else {
codec_ctx_->pix_fmt = AV_PIX_FMT_NV12; codec_ctx_->pix_fmt = AV_PIX_FMT_NV12;
@@ -99,7 +98,9 @@ int FFmpegVideoEncoder::Init() {
av_opt_set_int(codec_ctx_->priv_data, "qp", 51, 0); av_opt_set_int(codec_ctx_->priv_data, "qp", 51, 0);
av_opt_set_int(codec_ctx_->priv_data, "crf", 23, 0); av_opt_set_int(codec_ctx_->priv_data, "crf", 23, 0);
av_opt_set(codec_ctx_->priv_data, "profile", "baseline", 0); if (!use_openh264_) {
av_opt_set(codec_ctx_->priv_data, "profile", "baseline", 0);
}
av_opt_set(codec_ctx_->priv_data, "preset", "ultrafast", 0); av_opt_set(codec_ctx_->priv_data, "preset", "ultrafast", 0);
av_opt_set(codec_ctx_->priv_data, "tune", "zerolatency", 0); av_opt_set(codec_ctx_->priv_data, "tune", "zerolatency", 0);
@@ -146,20 +147,13 @@ int FFmpegVideoEncoder::Encode(
return -1; return -1;
} }
if (use_libx264_) { if (use_openh264_) {
NV12ToYUV420PFFmpeg((unsigned char *)pData, frame_->width, frame_->height, NV12ToYUV420PFFmpeg((unsigned char *)pData, frame_->width, frame_->height,
(unsigned char *)yuv420p_buffer); (unsigned char *)yuv420p_buffer);
memcpy(frame_->data[0], yuv420p_buffer, frame_->width * frame_->height);
memcpy(frame_->data[1], yuv420p_buffer + frame_->width * frame_->height,
frame_->width * frame_->height / 4);
memcpy(frame_->data[2],
yuv420p_buffer + frame_->width * frame_->height * 5 / 4,
frame_->width * frame_->height / 4);
// frame_->data[0] = yuv420p_buffer; frame_->data[0] = yuv420p_buffer;
// frame_->data[1] = yuv420p_buffer + frame_->width * frame_->height; frame_->data[1] = yuv420p_buffer + frame_->width * frame_->height;
// frame_->data[2] = yuv420p_buffer + frame_->width * frame_->height * 3 / frame_->data[2] = yuv420p_buffer + frame_->width * frame_->height * 5 / 4;
// 2;
if (SAVE_NV12_STREAM) { if (SAVE_NV12_STREAM) {
fwrite(yuv420p_buffer, 1, nSize, file_nv12_); fwrite(yuv420p_buffer, 1, nSize, file_nv12_);

2
src/media/video/encode/ffmpeg/ffmpeg_video_encoder.h
View File

@@ -58,7 +58,7 @@ class FFmpegVideoEncoder : public VideoEncoder {
FILE* file_nv12_ = nullptr; FILE* file_nv12_ = nullptr;
unsigned char* nv12_data_ = nullptr; unsigned char* nv12_data_ = nullptr;
unsigned int seq_ = 0; unsigned int seq_ = 0;
bool use_libx264_ = false; bool use_openh264_ = false;
const AVCodec* codec_ = nullptr; const AVCodec* codec_ = nullptr;
AVCodecContext* codec_ctx_ = nullptr; AVCodecContext* codec_ctx_ = nullptr;