virtualenv/FFmpeg
1
0
Fork 0
mirror of https://github.com/FFmpeg/FFmpeg.git synced 2024-12-28 20:53:54 +02:00
Code Issues Releases Activity

doc/examples/qsvdec: simplify this example via hw_device_ctx interface

Browse Source 
Signed-off-by: James Almer <jamrial@gmail.com>
This commit is contained in:
Haihao Xiang 2021-08-11 14:01:55 +08:00 committed by James Almer
parent ecee3b07cd
commit c3ef9e0277
1 changed files with 9 additions and 36 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

45
doc/examples/qsvdec.c
View File

@ -44,38 +44,10 @@
#include "libavutil/hwcontext_qsv.h" #include "libavutil/hwcontext_qsv.h"
#include "libavutil/mem.h" #include "libavutil/mem.h"
typedef struct DecodeContext {
AVBufferRef *hw_device_ref;
} DecodeContext;
static int get_format(AVCodecContext *avctx, const enum AVPixelFormat *pix_fmts) static int get_format(AVCodecContext *avctx, const enum AVPixelFormat *pix_fmts)
{ {
while (*pix_fmts != AV_PIX_FMT_NONE) { while (*pix_fmts != AV_PIX_FMT_NONE) {
if (*pix_fmts == AV_PIX_FMT_QSV) { if (*pix_fmts == AV_PIX_FMT_QSV) {
DecodeContext *decode = avctx->opaque;
AVHWFramesContext *frames_ctx;
AVQSVFramesContext *frames_hwctx;
int ret;
/* create a pool of surfaces to be used by the decoder */
avctx->hw_frames_ctx = av_hwframe_ctx_alloc(decode->hw_device_ref);
if (!avctx->hw_frames_ctx)
return AV_PIX_FMT_NONE;
frames_ctx = (AVHWFramesContext*)avctx->hw_frames_ctx->data;
frames_hwctx = frames_ctx->hwctx;
frames_ctx->format = AV_PIX_FMT_QSV;
frames_ctx->sw_format = avctx->sw_pix_fmt;
frames_ctx->width = FFALIGN(avctx->coded_width, 32);
frames_ctx->height = FFALIGN(avctx->coded_height, 32);
frames_ctx->initial_pool_size = 32;
frames_hwctx->frame_type = MFX_MEMTYPE_VIDEO_MEMORY_DECODER_TARGET;
ret = av_hwframe_ctx_init(avctx->hw_frames_ctx);
if (ret < 0)
return AV_PIX_FMT_NONE;
return AV_PIX_FMT_QSV; return AV_PIX_FMT_QSV;
} }
@ -87,7 +59,7 @@ static int get_format(AVCodecContext *avctx, const enum AVPixelFormat *pix_fmts)
return AV_PIX_FMT_NONE; return AV_PIX_FMT_NONE;
} }
static int decode_packet(DecodeContext *decode, AVCodecContext *decoder_ctx, static int decode_packet(AVCodecContext *decoder_ctx,
AVFrame *frame, AVFrame *sw_frame, AVFrame *frame, AVFrame *sw_frame,
AVPacket *pkt, AVIOContext *output_ctx) AVPacket *pkt, AVIOContext *output_ctx)
{ {
@ -144,12 +116,12 @@ int main(int argc, char **argv)
AVPacket pkt = { 0 }; AVPacket pkt = { 0 };
AVFrame *frame = NULL, *sw_frame = NULL; AVFrame *frame = NULL, *sw_frame = NULL;
DecodeContext decode = { NULL };
AVIOContext *output_ctx = NULL; AVIOContext *output_ctx = NULL;
int ret, i; int ret, i;
AVBufferRef *device_ref = NULL;
if (argc < 3) { if (argc < 3) {
fprintf(stderr, "Usage: %s <input file> <output file>\n", argv[0]); fprintf(stderr, "Usage: %s <input file> <output file>\n", argv[0]);
return 1; return 1;
@ -177,7 +149,7 @@ int main(int argc, char **argv)
} }
/* open the hardware device */ /* open the hardware device */
ret = av_hwdevice_ctx_create(&decode.hw_device_ref, AV_HWDEVICE_TYPE_QSV, ret = av_hwdevice_ctx_create(&device_ref, AV_HWDEVICE_TYPE_QSV,
"auto", NULL, 0); "auto", NULL, 0);
if (ret < 0) { if (ret < 0) {
fprintf(stderr, "Cannot open the hardware device\n"); fprintf(stderr, "Cannot open the hardware device\n");
@ -209,7 +181,8 @@ int main(int argc, char **argv)
decoder_ctx->extradata_size = video_st->codecpar->extradata_size; decoder_ctx->extradata_size = video_st->codecpar->extradata_size;
} }
decoder_ctx->opaque = &decode;
decoder_ctx->hw_device_ctx = av_buffer_ref(device_ref);
decoder_ctx->get_format = get_format; decoder_ctx->get_format = get_format;
ret = avcodec_open2(decoder_ctx, NULL, NULL); ret = avcodec_open2(decoder_ctx, NULL, NULL);
@ -239,7 +212,7 @@ int main(int argc, char **argv)
break; break;
if (pkt.stream_index == video_st->index) if (pkt.stream_index == video_st->index)
ret = decode_packet(&decode, decoder_ctx, frame, sw_frame, &pkt, output_ctx); ret = decode_packet(decoder_ctx, frame, sw_frame, &pkt, output_ctx);
av_packet_unref(&pkt); av_packet_unref(&pkt);
} }
@ -247,7 +220,7 @@ int main(int argc, char **argv)
/* flush the decoder */ /* flush the decoder */
pkt.data = NULL; pkt.data = NULL;
pkt.size = 0; pkt.size = 0;
ret = decode_packet(&decode, decoder_ctx, frame, sw_frame, &pkt, output_ctx); ret = decode_packet(decoder_ctx, frame, sw_frame, &pkt, output_ctx);
finish: finish:
if (ret < 0) { if (ret < 0) {
@ -263,7 +236,7 @@ finish:
avcodec_free_context(&decoder_ctx); avcodec_free_context(&decoder_ctx);
av_buffer_unref(&decode.hw_device_ref); av_buffer_unref(&device_ref);
avio_close(output_ctx); avio_close(output_ctx);