From 83dc9220bca38cc681b0a310c93ab6b3fc5babe2 Mon Sep 17 00:00:00 2001
From: xuxiuxi <xuxiuxi@454eff88-639b-444f-9e54-f578c98de674>
Date: 星期二, 25 四月 2017 10:50:15 +0800
Subject: [PATCH]
---
RtspFace/PL_H264Encoder.cpp | 44 ++++++++++++++++++++++++++++----------------
1 files changed, 28 insertions(+), 16 deletions(-)
diff --git a/RtspFace/PL_H264Encoder.cpp b/RtspFace/PL_H264Encoder.cpp
index 1b3123a..68c502c 100644
--- a/RtspFace/PL_H264Encoder.cpp
+++ b/RtspFace/PL_H264Encoder.cpp
@@ -9,17 +9,17 @@
#include <libavutil/imgutils.h>
#include <libavutil/opt.h>
#include <libavformat/avformat.h>
-
- #include <libyuv.h>
}
+
+#include <libyuv.h>
PL_H264Encoder_Config::PL_H264Encoder_Config() :
inBufferSize(2*1024*1024), // 2MByte
resetPTS(false),
bytesBufferImageWidth(0), bytesBufferImageHeight(0),
avc_bit_rate(1*1024*1024*8), //1Mbit
- avc_fps(25), avc_gop(25), avc_max_b_frames(0), avc_profile(FF_PROFILE_H264_MAIN),
- av_opt_preset("superfast"), av_opt_tune("")
+ avc_fps(25), avc_gop(25), avc_max_b_frames(0), avc_profile(FF_PROFILE_H264_BASELINE),
+ av_opt_preset("superfast"), av_opt_tune(""), avc_profile_str("")
{
// av_opt_tune: zerolatency
}
@@ -47,6 +47,8 @@
~H264Encoder_Internal()
{
+ delete[] buffer;
+ buffer = nullptr;
}
void reset()
@@ -67,8 +69,10 @@
pAVFormatContext = nullptr;
if (buffer != nullptr)
+ {
delete[] buffer;
- buffer = new uint8_t[config.inBufferSize];
+ buffer = nullptr;
+ }
}
};
@@ -98,6 +102,8 @@
in->config = *config;
}
+ in->buffer = new uint8_t[in->config.inBufferSize];
+
return true;
}
@@ -116,7 +122,7 @@
if (!avCodec)
{
- LOG_ERROR << "codec not found!";
+ LOG_ERROR << "codec not found!" << std::endl;
return false;
}
@@ -129,13 +135,16 @@
in->pAVCodecContext->time_base.den = in->config.avc_fps;
in->pAVCodecContext->gop_size = in->config.avc_gop;
in->pAVCodecContext->max_b_frames = in->config.avc_max_b_frames;
- in->pAVCodecContext->profile = in->config.avc_profile;
in->pAVCodecContext->pix_fmt = AV_PIX_FMT_YUV420P;
if (!in->config.av_opt_preset.empty())
av_opt_set(in->pAVCodecContext->priv_data, "preset", in->config.av_opt_preset.c_str(), 0);
if (!in->config.av_opt_tune.empty())
av_opt_set(in->pAVCodecContext->priv_data, "tune", in->config.av_opt_tune.c_str(), 0);
+ if (!in->config.avc_profile_str.empty())
+ av_opt_set(in->pAVCodecContext->priv_data, "profile", in->config.avc_profile_str.c_str(), 0);
+ else
+ in->pAVCodecContext->profile = in->config.avc_profile;
if(avcodec_open2(in->pAVCodecContext, avCodec, NULL) >= 0)
{
@@ -150,13 +159,13 @@
in->pAVCodecContext->pix_fmt, 16);
if (ret < 0)
{
- LOG_ERROR << "av_image_alloc error";
+ LOG_ERROR << "av_image_alloc error" << std::endl;
return false;
}
}
else
{
- LOG_ERROR << "avcodec_open2 error";
+ LOG_ERROR << "avcodec_open2 error" << std::endl;
return false;
}
@@ -227,7 +236,7 @@
int ret = avcodec_encode_video2(in->pAVCodecContext, &pAVPacket, in->pAVFrame, &gotPacket);
if (ret < 0)
{
- LOG_WARN << "avcodec_encode_video2 (1) error=" << ret;
+ LOG_WARN << "avcodec_encode_video2 (1) error=" << ret << std::endl;
return false;
}
@@ -235,7 +244,7 @@
{
in->frameCount++;
LOGP(DEBUG, "Succeed to encode (1) frame=%d, size=%d", in->frameCount, pAVPacket.size);
- memcpy(in->buffer, pAVPacket.data, pAVPacket.size);
+ memcpy(in->buffer, pAVPacket.data, pAVPacket.size);//#todo check inBufferSize
in->buffSize = pAVPacket.size;
av_free_packet(&pAVPacket);
}
@@ -321,7 +330,7 @@
bool ret = initH264EncoderEnv(in);
if (!ret)
{
- LOG_ERROR << "initH264EncoderEnv error";
+ LOG_ERROR << "initH264EncoderEnv error" << std::endl;
return false;
}
else
@@ -330,7 +339,7 @@
if (pm.type != PipeMaterial::PMT_FRAME)
{
- LOG_ERROR << "Only support PMT_FRAME";
+ LOG_ERROR << "Only support PMT_FRAME" << std::endl;
return false;
}
@@ -347,7 +356,7 @@
ret = encodeH264(in, (uint8_t*)(frame->buffer), frame->pts);
else
{
- LOG_ERROR << "Only support MBFT_PTR_AVFRAME / MBFT_YUV420";
+ LOG_ERROR << "Only support MBFT_PTR_AVFRAME / MBFT_YUV420" << std::endl;
in->payError = true;
return false;
}
@@ -361,8 +370,11 @@
in->lastFrame.buffSize = in->buffSize;
in->lastFrame.width = frame->width;
in->lastFrame.height = frame->height;
- in->lastFrame.pts = frame->pts;
- //#todo resetPts
+
+ if (in->config.resetPTS)
+ gettimeofday(&(in->lastFrame.pts),NULL);
+ else
+ in->lastFrame.pts = frame->pts;
}
return ret;
--
Gitblit v1.8.0