hwaccel_vaapi.c File Reference

This file contains functions related to hw acceleration. More...

#include <libavcodec/version.h>
#include <libavutil/pixdesc.h>
#include <libavutil/hwcontext_vaapi.h>
#include "debug.h"
#include "hwaccel_libav_common.h"
#include "hwaccel_vaapi.h"
#include "libavcodec/lavc_common.h"

Include dependency graph for hwaccel_vaapi.c:

Go to the source code of this file.

Classes
struct	vaapi_ctx

Macros
#define	DEFAULT_SURFACES   20
#define	MOD_NAME   "[vaapi] "

Functions
static enum AVPixelFormat	get_sw_format (VADisplay display, AVBufferRef *device_ref, enum AVPixelFormat fallback_fmt)
int	vaapi_init (struct AVCodecContext *s, struct hw_accel_state *state, codec_t out_codec)
void	vaapi_uninit (struct hw_accel_state *s)

Detailed Description

This file contains functions related to hw acceleration.

Author

Martin Piatka piatk.nosp@m.a@ce.nosp@m.snet..nosp@m.cz

Definition in file hwaccel_vaapi.c.

Macro Definition Documentation

DEFAULT_SURFACES

#define DEFAULT_SURFACES   20

Definition at line 63 of file hwaccel_vaapi.c.

Referenced by vaapi_init().

MOD_NAME

#define MOD_NAME   "[vaapi] "

Definition at line 64 of file hwaccel_vaapi.c.

Function Documentation

get_sw_format()

static enum AVPixelFormat get_sw_format ( VADisplay display, AVBufferRef * device_ref, enum AVPixelFormat fallback_fmt )

static

Returns first SW format from valid_sw_formats. This is usually AV_PIX_FMT_YUV420P or AV_PIX_FMT_NV12.

The code borrows heavily from mpv https://github.com/mpv-player/mpv/blob/master/video/out/hwdec/hwdec_vaapi.c namely from function try_format_config().

< just some valid size

Definition at line 210 of file hwaccel_vaapi.c.

{
        enum AVPixelFormat     ret       = AV_PIX_FMT_NONE;
        AVVAAPIHWConfig       *hwconfig  = NULL;
        VAConfigID             config_id = 0;
        AVHWFramesConstraints *fc        = NULL;
 
        VAStatus status = vaCreateConfig(
            display, VAProfileNone, VAEntrypointVideoProc, NULL, 0, &config_id);
        if (status != VA_STATUS_SUCCESS) {
                MSG(ERROR, "cannot create config\n");
                goto fail;
        }
        fc = av_hwdevice_get_hwframe_constraints(device_ref, hwconfig);
        if (!fc) {
                MSG(ERROR, "failed to retrieve libavutil frame constraints\n");
                goto fail;
        }
 
        /*
         * We need a hwframe_ctx to be able to get the valid formats, but to
         * initialise it, we need a format, so we get the first format from the
         * hwconfig. We don't care about the other formats in the config because
         * the transfer formats list will already include them.
         */
        AVBufferRef *fref = NULL;
        fref              = av_hwframe_ctx_alloc(device_ref);
        if (!fref) {
                MSG(ERROR, "failed to alloc libavutil frame context\n");
                goto fail;
        }
        AVHWFramesContext *fctx = (void *) fref->data;
        enum {
                INIT_SIZE = 128, 
        };
        fctx->format    = AV_PIX_FMT_VAAPI;
        fctx->sw_format = fc->valid_sw_formats[0];
        fctx->width     = INIT_SIZE;
        fctx->height    = INIT_SIZE;
        if (av_hwframe_ctx_init(fref) < 0) {
                MSG(ERROR, "failed to init libavutil frame context\n");
                goto fail;
        }
 
        enum AVPixelFormat *fmts = NULL;
        int                 rc   = av_hwframe_transfer_get_formats(
            fref, AV_HWFRAME_TRANSFER_DIRECTION_FROM, &fmts, 0);
        if (rc) {
                MSG(ERROR, "failed to get libavutil frame context supported "
                           "formats\n");
                goto fail;
        }
        MSG(DEBUG, "Available HW layouts: %s\n", get_avpixfmts_names(fmts));
        ret = fmts[0];
 
fail:
        av_hwframe_constraints_free(&fc);
        av_buffer_unref(&fref);
        if (ret == AV_PIX_FMT_NONE) {
                MSG(WARNING, "Using fallback HW frames layout: %s\n",
                    av_get_pix_fmt_name(ret));
                ret = fallback_fmt;
        }
        MSG(VERBOSE, "Selected HW frames layout: %s\n",
            av_get_pix_fmt_name(ret));
        return ret;
}

References DEBUG, get_avpixfmts_names(), and MSG.

Referenced by vaapi_init().

Here is the call graph for this function:

vaapi_init()

int vaapi_init	(	struct AVCodecContext *	s,
		struct hw_accel_state *	state,
		codec_t	out_codec )

Definition at line 279 of file hwaccel_vaapi.c.

{
        (void)(out_codec);
        struct vaapi_ctx *ctx = calloc(1, sizeof(struct vaapi_ctx));
        if(!ctx){
                return -1;
        }
 
        int ret = create_hw_device_ctx(AV_HWDEVICE_TYPE_VAAPI, &ctx->device_ref);
        if(ret < 0)
                goto fail;
 
        ctx->device_ctx = (AVHWDeviceContext*)(void *)ctx->device_ref->data;
        ctx->device_vaapi_ctx = ctx->device_ctx->hwctx;
 
#if LIBAVCODEC_VERSION_INT < AV_VERSION_INT(57, 74, 100)
        ret = vaapi_create_context(ctx, s);
        if(ret < 0)
                goto fail;
#endif
 
        int decode_surfaces = DEFAULT_SURFACES;
 
        if (s->active_thread_type & FF_THREAD_FRAME)
                decode_surfaces += s->thread_count;
 
        const enum AVPixelFormat sw_format = get_sw_format(
            ctx->device_vaapi_ctx->display, ctx->device_ref, s->sw_pix_fmt);
        ret = create_hw_frame_ctx(ctx->device_ref,
                        s->coded_width,
                        s->coded_height,
                        AV_PIX_FMT_VAAPI,
                        sw_format,
                        decode_surfaces,
                        &ctx->hw_frames_ctx);
        if(ret < 0)
                goto fail;
 
        ctx->frame_ctx = (AVHWFramesContext *)(void *) (ctx->hw_frames_ctx->data);
 
        s->hw_frames_ctx = ctx->hw_frames_ctx;
 
        state->tmp_frame = av_frame_alloc();
        if(!state->tmp_frame){
                ret = -1;
                goto fail;
        }
        state->type = HWACCEL_VAAPI;
        state->copy = true;
        state->ctx = ctx;
        state->uninit = vaapi_uninit;
 
#if LIBAVCODEC_VERSION_INT < AV_VERSION_INT(57, 74, 100)
        AVVAAPIFramesContext *avfc = ctx->frame_ctx->hwctx;
        VAStatus status = vaCreateContext(ctx->device_vaapi_ctx->display,
                        ctx->va_config, s->coded_width, s->coded_height,
                        VA_PROGRESSIVE,
                        avfc->surface_ids,
                        avfc->nb_surfaces,
                        &ctx->va_context);
 
        if(status != VA_STATUS_SUCCESS){
                log_msg(LOG_LEVEL_ERROR, "[lavd] Create config failed: %d (%s)\n", status, vaErrorStr(status));
                ret = -1;
                goto fail;
        }
 
        ctx->decoder_context.display = ctx->device_vaapi_ctx->display;
        ctx->decoder_context.config_id = ctx->va_config;
        ctx->decoder_context.context_id = ctx->va_context;
 
        s->hwaccel_context = &ctx->decoder_context;
#endif //LIBAVCODEC_VERSION_INT < AV_VERSION_INT(57, 74, 100)
 
        av_buffer_unref(&ctx->device_ref);
        return 0;
 
 
fail:
        av_frame_free(&state->tmp_frame);
        av_buffer_unref(&ctx->hw_frames_ctx);
#if LIBAVCODEC_VERSION_INT < AV_VERSION_INT(57, 74, 100)
        if(ctx->device_vaapi_ctx)
                vaDestroyConfig(ctx->device_vaapi_ctx->display, ctx->va_config);
#endif
        av_buffer_unref(&ctx->device_ref);
        free(ctx);
        return ret;
}

References DEFAULT_SURFACES, vaapi_ctx::device_ctx, vaapi_ctx::device_ref, vaapi_ctx::device_vaapi_ctx, vaapi_ctx::frame_ctx, get_sw_format(), vaapi_ctx::hw_frames_ctx, HWACCEL_VAAPI, LOG_LEVEL_ERROR, log_msg(), state, and vaapi_uninit().

Referenced by get_format_callback().

Here is the call graph for this function:

vaapi_uninit()

void vaapi_uninit

(

struct hw_accel_state *

s

)

Definition at line 84 of file hwaccel_vaapi.c.

                                           {
 
        free(s->ctx);
}

References hw_accel_state::ctx.

Referenced by vaapi_init().