首次提交，可运行。但有两问题：1.画质不佳, 2.无音频
This commit is contained in:
253
VideoDecoder.cpp
Normal file
253
VideoDecoder.cpp
Normal file
@@ -0,0 +1,253 @@
|
||||
extern "C"
|
||||
{
|
||||
#include <libavcodec/avcodec.h>
|
||||
#include <libavformat/avformat.h>
|
||||
}
|
||||
|
||||
#include<iostream>
|
||||
#include"VideoDecoder.h"
|
||||
#include"FrameRecviver.h"
|
||||
|
||||
namespace
|
||||
{
|
||||
struct EncodecName_by_ID
|
||||
{
|
||||
AVCodecID id;
|
||||
char name[32];
|
||||
};
|
||||
|
||||
constexpr EncodecName_by_ID encodec_name_by_id[]=
|
||||
{
|
||||
{AV_CODEC_ID_MJPEG ,"mjpeg_cuvid"},
|
||||
{AV_CODEC_ID_MPEG1VIDEO ,"mpeg1_cuvid"},
|
||||
{AV_CODEC_ID_MPEG2VIDEO ,"mpeg2_cuvid"},
|
||||
{AV_CODEC_ID_MPEG4 ,"mpeg4_cuvid"},
|
||||
{AV_CODEC_ID_H264 ,"h264_cuvid"},
|
||||
{AV_CODEC_ID_HEVC ,"hevc_cuvid"},
|
||||
{AV_CODEC_ID_VC1 ,"vc1_cuvid"},
|
||||
{AV_CODEC_ID_VP8 ,"vp8_cuvid"},
|
||||
{AV_CODEC_ID_VP9 ,"vp9_cuvid"},
|
||||
{AV_CODEC_ID_AV1 ,"av1_cuvid"},
|
||||
|
||||
{AV_CODEC_ID_H264 ,"h264_amf"},
|
||||
{AV_CODEC_ID_HEVC ,"hevc_amf"},
|
||||
|
||||
{AV_CODEC_ID_MJPEG ,"mjpeg_qsv"},
|
||||
{AV_CODEC_ID_MPEG2VIDEO ,"mpeg2_qsv"},
|
||||
{AV_CODEC_ID_H264 ,"h264_qsv"},
|
||||
{AV_CODEC_ID_HEVC ,"hevc_qsv"},
|
||||
{AV_CODEC_ID_VC1 ,"vc1_qsv"},
|
||||
{AV_CODEC_ID_VP8 ,"vp8_qsv"},
|
||||
{AV_CODEC_ID_VP9 ,"vp9_qsv"},
|
||||
{AV_CODEC_ID_AV1 ,"av1_qsv"},
|
||||
|
||||
{AV_CODEC_ID_NONE ,""}
|
||||
};
|
||||
|
||||
const AVCodec *GetAVDecodec(AVCodecID id,bool hardware)
|
||||
{
|
||||
if(hardware)
|
||||
{
|
||||
const AVCodec *codec=nullptr;
|
||||
|
||||
for(auto &c:encodec_name_by_id)
|
||||
{
|
||||
if(c.id==id)
|
||||
{
|
||||
codec=avcodec_find_decoder_by_name(c.name);
|
||||
|
||||
if(codec)
|
||||
{
|
||||
std::cout<<"use decoder: "<<c.name<<std::endl;
|
||||
return codec;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
return avcodec_find_decoder(id);
|
||||
}
|
||||
}//namespace
|
||||
|
||||
class FFmpegVideoDecoder:public VideoDecoder
|
||||
{
|
||||
AVCodecParameters * video_cp;
|
||||
|
||||
AVStream * video_stream;
|
||||
int video_stream_index;
|
||||
|
||||
const AVCodec * video_codec;
|
||||
AVCodecContext * video_ctx;
|
||||
|
||||
FrameRecviver * frame_recviver;
|
||||
AVFrame * frame;
|
||||
|
||||
AVPacket packet;
|
||||
|
||||
public:
|
||||
|
||||
FFmpegVideoDecoder(FrameRecviver *fr)
|
||||
{
|
||||
ctx=nullptr;
|
||||
frame_recviver=fr;
|
||||
|
||||
video_cp=nullptr;
|
||||
video_stream=nullptr;
|
||||
video_stream_index=-1;
|
||||
|
||||
video_codec=nullptr;
|
||||
video_ctx=nullptr;
|
||||
|
||||
frame=nullptr;
|
||||
}
|
||||
|
||||
~FFmpegVideoDecoder()
|
||||
{
|
||||
if(frame)
|
||||
av_free(frame);
|
||||
|
||||
if(video_ctx)
|
||||
avcodec_free_context(&video_ctx);
|
||||
|
||||
if(ctx)
|
||||
avformat_close_input(&ctx);
|
||||
}
|
||||
|
||||
bool Init(const char *filename,const bool use_hardware)
|
||||
{
|
||||
ctx=avformat_alloc_context();
|
||||
|
||||
if(avformat_open_input(&ctx,filename,nullptr,nullptr)!=0)
|
||||
return(false);
|
||||
|
||||
if(avformat_find_stream_info(ctx,nullptr)<0)
|
||||
return(false);
|
||||
|
||||
av_dump_format(ctx,0,filename,0);
|
||||
|
||||
video_cp=nullptr;
|
||||
video_stream=nullptr;
|
||||
video_stream_index=-1;
|
||||
|
||||
for(int i=0;i<ctx->nb_streams;i++)
|
||||
{
|
||||
if(ctx->streams[i]->codecpar->codec_type==AVMEDIA_TYPE_VIDEO)
|
||||
{
|
||||
video_stream=ctx->streams[i];
|
||||
video_stream_index=i;
|
||||
video_cp=video_stream->codecpar;
|
||||
}
|
||||
//if (ctx->streams[i]->codecpar->codec_type == AVMEDIA_TYPE_AUDIO)
|
||||
//{
|
||||
// audio_stream_index = i;
|
||||
//}
|
||||
}
|
||||
|
||||
if(!video_cp)
|
||||
return(false);
|
||||
|
||||
video_codec=GetAVDecodec(video_cp->codec_id,use_hardware);
|
||||
|
||||
if(!video_codec)
|
||||
{
|
||||
printf("Can't find decoder:%d\n",video_cp->codec_id);
|
||||
return(false);
|
||||
}
|
||||
|
||||
video_ctx=avcodec_alloc_context3(video_codec);
|
||||
|
||||
avcodec_parameters_to_context(video_ctx,video_cp);
|
||||
|
||||
if(avcodec_open2(video_ctx,video_codec,nullptr)<0)
|
||||
{
|
||||
printf("Couldn't open video stream");
|
||||
return(false);
|
||||
}
|
||||
|
||||
width=video_cp->width;
|
||||
height=video_cp->height;
|
||||
|
||||
frame=av_frame_alloc();
|
||||
|
||||
frame_recviver->SetFrameRate(video_stream->avg_frame_rate);
|
||||
|
||||
return(true);
|
||||
}
|
||||
|
||||
void Start()
|
||||
{
|
||||
if(!video_ctx)return;
|
||||
if(!video_stream)return;
|
||||
|
||||
av_seek_frame(ctx,video_stream_index,0,AVSEEK_FLAG_FRAME);
|
||||
}
|
||||
|
||||
bool NextFrame()
|
||||
{
|
||||
int ret;
|
||||
|
||||
while((ret=av_read_frame(ctx,&packet))>=0)
|
||||
{
|
||||
if(packet.stream_index==video_stream_index)
|
||||
{
|
||||
ret=avcodec_send_packet(video_ctx,&packet);
|
||||
|
||||
if(ret<0)
|
||||
{
|
||||
std::cerr<<"avcodec_send_packet return "<<ret<<std::endl;
|
||||
return(false);
|
||||
}
|
||||
|
||||
ret=avcodec_receive_frame(video_ctx,frame);
|
||||
|
||||
if(ret==AVERROR(EAGAIN)||ret==AVERROR_EOF)
|
||||
{
|
||||
av_packet_unref(&packet);
|
||||
continue;
|
||||
}
|
||||
|
||||
if(ret<0)
|
||||
{
|
||||
std::cerr<<"avcodec_receive_frame return "<<ret<<std::endl;
|
||||
av_packet_unref(&packet);
|
||||
return(false);
|
||||
}
|
||||
|
||||
frame_recviver->OnFrame(frame);
|
||||
|
||||
av_packet_unref(&packet);
|
||||
return(true);
|
||||
}
|
||||
//else if (packet.stream_index == audio_stream_index)
|
||||
//{
|
||||
// if (audio_decoder)
|
||||
// {
|
||||
// audio_decoder->DecodePackete(&packet);
|
||||
// }
|
||||
//}
|
||||
|
||||
av_packet_unref(&packet);
|
||||
}
|
||||
|
||||
std::cerr<<"av_read_frame return "<<ret<<std::endl;
|
||||
return(false);
|
||||
}
|
||||
|
||||
void Abort()
|
||||
{
|
||||
|
||||
}
|
||||
};//class FFmpegDecoder:public Decoder
|
||||
|
||||
VideoDecoder *CreateVideoDecoder(const char *filename,FrameRecviver *fr,const bool use_hardware)
|
||||
{
|
||||
FFmpegVideoDecoder *dec=new FFmpegVideoDecoder(fr);
|
||||
|
||||
if(dec->Init(filename,use_hardware)==false)
|
||||
{
|
||||
delete dec;
|
||||
return(nullptr);
|
||||
}
|
||||
|
||||
return(dec);
|
||||
}
|
||||
Reference in New Issue
Block a user