From 77513d0db1242f30821f94ae97e272055e5ae0bc Mon Sep 17 00:00:00 2001
From: houxiao <houxiao@454eff88-639b-444f-9e54-f578c98de674>
Date: 星期四, 23 二月 2017 17:02:25 +0800
Subject: [PATCH] 更新产品原型
---
RtspFace/PL_H264Encoder.cpp | 35 ++++++++++++++++++++++-------------
1 files changed, 22 insertions(+), 13 deletions(-)
diff --git a/RtspFace/PL_H264Encoder.cpp b/RtspFace/PL_H264Encoder.cpp
index 1b3123a..2af9e3e 100644
--- a/RtspFace/PL_H264Encoder.cpp
+++ b/RtspFace/PL_H264Encoder.cpp
@@ -9,9 +9,9 @@
#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
@@ -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;
}
@@ -150,13 +156,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 +233,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 +241,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 +327,7 @@
bool ret = initH264EncoderEnv(in);
if (!ret)
{
- LOG_ERROR << "initH264EncoderEnv error";
+ LOG_ERROR << "initH264EncoderEnv error" << std::endl;
return false;
}
else
@@ -330,7 +336,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 +353,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 +367,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