From 48a70bb5b8b4d2daf24462d94fcc1bc3e3f85858 Mon Sep 17 00:00:00 2001
From: zhangmeng <775834166@qq.com>
Date: 星期四, 17 九月 2020 09:27:19 +0800
Subject: [PATCH] gpu reserved 896M
---
csrc/ffmpeg/format/FormatIn.cpp | 75 +++++++++++++++++++++++--------------
1 files changed, 47 insertions(+), 28 deletions(-)
diff --git a/csrc/ffmpeg/format/FormatIn.cpp b/csrc/ffmpeg/format/FormatIn.cpp
index 59f3f23..daa55fa 100644
--- a/csrc/ffmpeg/format/FormatIn.cpp
+++ b/csrc/ffmpeg/format/FormatIn.cpp
@@ -28,6 +28,7 @@
,dec_ctx_(NULL)
,vs_idx_(-1)
,as_idx_(-1)
+ ,prop_(NULL)
,hw_accl_(hw)
,io_ctx_(NULL)
,read_io_buff_(NULL)
@@ -36,32 +37,46 @@
,fps_(25.0)
{}
+ FormatIn::FormatIn(const VideoProp &prop, bool hw/*=true*/)
+ :ctx_(NULL)
+ ,dec_ctx_(NULL)
+ ,vs_idx_(-1)
+ ,as_idx_(-1)
+ ,prop_(NULL)
+ ,hw_accl_(hw)
+ ,io_ctx_(NULL)
+ ,read_io_buff_(NULL)
+ ,read_io_buff_size_(32768)
+ ,handle_gb28181(NULL)
+ ,fps_(25.0)
+ {
+ prop_ = new VideoProp;
+ *prop_ = prop;
+ }
+
FormatIn::~FormatIn()
{
- if(ctx_){
- if (!(ctx_->flags & AVFMT_FLAG_CUSTOM_IO)){
- avformat_close_input(&ctx_);
- }else{
- avformat_free_context(ctx_);
- }
+ logIt("free format in");
+ if (prop_) delete prop_;
+ if(dec_ctx_){
+ avcodec_close(dec_ctx_);
+ avcodec_free_context(&dec_ctx_);
+ dec_ctx_ = NULL;
+ }
+
+ if(ctx_){
+ avformat_close_input(&ctx_);
ctx_ = NULL;
- if(dec_ctx_){
- avcodec_close(dec_ctx_);
- dec_ctx_ = NULL;
- }
-
}
if (handle_gb28181){
delete handle_gb28181;
}
- if(read_io_buff_){
- av_free(read_io_buff_);
- read_io_buff_ = NULL;
- }
+
if(io_ctx_){
- avio_context_free(&io_ctx_);
+ av_freep(&io_ctx_->buffer);
+ av_freep(&io_ctx_);
io_ctx_ = NULL;
}
@@ -89,11 +104,11 @@
ctx_->pb = io_ctx_;
- auto err = av_probe_input_buffer(ctx_->pb, &ctx_->iformat, NULL, NULL, 0, 0);
- if(err != 0){
- logIt("open with custom io prob input buffer error:%d err: %s\n", err, getAVErrorDesc(err).c_str());
- return -1;
- }
+ // auto err = av_probe_input_buffer(ctx_->pb, &ctx_->iformat, NULL, NULL, 0, 0);
+ // if(err != 0){
+ // logIt("open with custom io prob input buffer error:%d err: %s\n", err, getAVErrorDesc(err).c_str());
+ // return -1;
+ // }
return 0;
}
@@ -164,7 +179,7 @@
}else if(in->avg_frame_rate.num >=1 && in->avg_frame_rate.den >= 1){
fps_ = av_q2d(in->avg_frame_rate);
}
- logIt("in stream video fps %f, time_base: %d : %d", fps_, in->time_base.num, in->time_base.den);
+ logIt("in stream video fps %f, time_base: %d : %d, size: %dx%d", fps_, in->time_base.num, in->time_base.den, in->codecpar->width, in->codecpar->height);
}
if (type == AVMEDIA_TYPE_AUDIO){
auto in = ctx_->streams[i];
@@ -199,17 +214,21 @@
for (int i = 0; i < 2; ++i)
{
if(hw_accl_){
- idle_gpu = gpu::getGPU(100);
+ idle_gpu = gpu::getGPUPrior(350, 896, 0);
+ // idle_gpu = gpu::getGPU(300);
+ if (prop_->gpu_index_ > -1){
+ idle_gpu = prop_->gpu_index_;
+ }
if(idle_gpu < 0){
logIt("NO GPU RESOURCE TO DECODE");
hw_accl_ = false;
continue;
}
- if(codecpar->codec_id == AV_CODEC_ID_H264){
- dec = avcodec_find_decoder_by_name("h264_cuvid");
- }else if(codecpar->codec_id == AV_CODEC_ID_H265){
- dec = avcodec_find_decoder_by_name("hevc_cuvid");
- }
+
+ std::string codec_name(avcodec_get_name(codecpar->codec_id));
+ codec_name += "_cuvid";
+ dec = avcodec_find_decoder_by_name(codec_name.c_str());
+
if(!dec){
hw_accl_ = false;
continue;
--
Gitblit v1.8.0