vaapi_encode_av1.c File Reference

#include <va/va.h>
#include <va/va_enc_av1.h>
#include "libavutil/pixdesc.h"
#include "libavutil/opt.h"
#include "libavutil/mastering_display_metadata.h"
#include "cbs_av1.h"
#include "put_bits.h"
#include "codec_internal.h"
#include "av1_levels.h"
#include "vaapi_encode.h"

Go to the source code of this file.

Data Structures
struct	VAAPIEncodeAV1Picture
struct	VAAPIEncodeAV1Context

Macros
#define	AV1_MAX_QUANT   255
#define	OFFSET(x)
#define	FLAGS   (AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_ENCODING_PARAM)
#define	PROFILE(name, value)
#define	LEVEL(name, value)

Functions
static void	vaapi_encode_av1_trace_write_log (void *ctx, PutBitContext *pbc, int length, const char *str, const int *subscripts, int64_t value)
static av_cold int	vaapi_encode_av1_get_encoder_caps (AVCodecContext *avctx)
static av_cold int	vaapi_encode_av1_configure (AVCodecContext *avctx)
static int	vaapi_encode_av1_add_obu (AVCodecContext *avctx, CodedBitstreamFragment *au, uint8_t type, void *obu_unit)
static int	vaapi_encode_av1_write_obu (AVCodecContext *avctx, char *data, size_t *data_len, CodedBitstreamFragment *bs)
static int	tile_log2 (int blkSize, int target)
static int	vaapi_encode_av1_set_tile (AVCodecContext *avctx)
static int	vaapi_encode_av1_write_sequence_header (AVCodecContext *avctx, char *data, size_t *data_len)
static int	vaapi_encode_av1_init_sequence_params (AVCodecContext *avctx)
static int	vaapi_encode_av1_init_picture_params (AVCodecContext *avctx, FFHWBaseEncodePicture *pic)
static int	vaapi_encode_av1_init_slice_params (AVCodecContext *avctx, FFHWBaseEncodePicture *base, VAAPIEncodeSlice *slice)
static int	vaapi_encode_av1_write_picture_header (AVCodecContext *avctx, FFHWBaseEncodePicture *pic, char *data, size_t *data_len)
static int	vaapi_encode_av1_write_extra_header (AVCodecContext *avctx, FFHWBaseEncodePicture *base_pic, int index, int *type, char *data, size_t *data_len)
static av_cold int	vaapi_encode_av1_init (AVCodecContext *avctx)
static av_cold int	vaapi_encode_av1_close (AVCodecContext *avctx)

Variables
static const VAAPIEncodeProfile	vaapi_encode_av1_profiles []
static const VAAPIEncodeType	vaapi_encode_type_av1
static const AVOption	vaapi_encode_av1_options []
static const FFCodecDefault	vaapi_encode_av1_defaults []
static const AVClass	vaapi_encode_av1_class
const FFCodec	ff_av1_vaapi_encoder

Macro Definition Documentation

AV1_MAX_QUANT

#define AV1_MAX_QUANT   255

Definition at line 34 of file vaapi_encode_av1.c.

Referenced by vaapi_encode_av1_configure(), and vaapi_encode_av1_init_picture_params().

OFFSET

#define OFFSET

(

x

)

Value:

offsetof(VAAPIEncodeAV1Context, x)

Definition at line 970 of file vaapi_encode_av1.c.

FLAGS

#define FLAGS   (AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_ENCODING_PARAM)

Definition at line 971 of file vaapi_encode_av1.c.

PROFILE

#define PROFILE	(		name,
			value )

Value:

    name, NULL, 0, AV_OPT_TYPE_CONST, \
    { .i64 = value }, 0, 0, FLAGS, .unit = "profile"

LEVEL

#define LEVEL	(		name,
			value )

Value:

      name, NULL, 0, AV_OPT_TYPE_CONST, \
      { .i64 = value }, 0, 0, FLAGS, .unit = "level"

Function Documentation

vaapi_encode_av1_trace_write_log()

static void vaapi_encode_av1_trace_write_log ( void * ctx, PutBitContext * pbc, int length, const char * str, const int * subscripts, int64_t value )

static

Definition at line 92 of file vaapi_encode_av1.c.

Referenced by vaapi_encode_av1_configure().

vaapi_encode_av1_get_encoder_caps()

static av_cold int vaapi_encode_av1_get_encoder_caps ( AVCodecContext * avctx )

static

Definition at line 113 of file vaapi_encode_av1.c.

vaapi_encode_av1_configure()

static av_cold int vaapi_encode_av1_configure ( AVCodecContext * avctx )

static

Arbitrary value

Definition at line 125 of file vaapi_encode_av1.c.

vaapi_encode_av1_add_obu()

static int vaapi_encode_av1_add_obu ( AVCodecContext * avctx, CodedBitstreamFragment * au, uint8_t type, void * obu_unit )

static

Definition at line 166 of file vaapi_encode_av1.c.

Referenced by vaapi_encode_av1_init_picture_params(), vaapi_encode_av1_init_sequence_params(), vaapi_encode_av1_write_extra_header(), and vaapi_encode_av1_write_picture_header().

vaapi_encode_av1_write_obu()

static int vaapi_encode_av1_write_obu ( AVCodecContext * avctx, char * data, size_t * data_len, CodedBitstreamFragment * bs )

static

Definition at line 184 of file vaapi_encode_av1.c.

Referenced by vaapi_encode_av1_init_picture_params(), vaapi_encode_av1_init_sequence_params(), vaapi_encode_av1_write_extra_header(), and vaapi_encode_av1_write_picture_header().

tile_log2()

static int tile_log2 ( int blkSize, int target )

static

Definition at line 210 of file vaapi_encode_av1.c.

Referenced by vaapi_encode_av1_set_tile().

vaapi_encode_av1_set_tile()

static int vaapi_encode_av1_set_tile ( AVCodecContext * avctx )

static

Try user setting tile rows number first.

try uniformed tile.

try non-uniformed tile.

Return invalid parameter if explicit tile rows is set.

check if tile cols/rows is supported by driver.

check if tile group numbers is valid.

Definition at line 216 of file vaapi_encode_av1.c.

Referenced by vaapi_encode_av1_init().

vaapi_encode_av1_write_sequence_header()

static int vaapi_encode_av1_write_sequence_header ( AVCodecContext * avctx, char * data, size_t * data_len )

static

Definition at line 350 of file vaapi_encode_av1.c.

vaapi_encode_av1_init_sequence_params()

static int vaapi_encode_av1_init_sequence_params ( AVCodecContext * avctx )

static

enable order hint and reserve maximum 8 bits for it by default.

Definition at line 361 of file vaapi_encode_av1.c.

vaapi_encode_av1_init_picture_params()

static int vaapi_encode_av1_init_picture_params ( AVCodecContext * avctx, FFHWBaseEncodePicture * pic )

static

set the nearest frame in L0 as all reference frame.

set the 2nd nearest frame in L0 as Golden frame.

B frame will not be referenced, disable its recon frame.

Use LAST_FRAME and BWDREF_FRAME for reference.

ignore ONLY_4x4 mode for codedlossless is not fully implemented.

let driver decide to use single or compound reference prediction mode.

set reference.

Definition at line 468 of file vaapi_encode_av1.c.

vaapi_encode_av1_init_slice_params()

static int vaapi_encode_av1_init_slice_params ( AVCodecContext * avctx, FFHWBaseEncodePicture * base, VAAPIEncodeSlice * slice )

static

Set tile group info.

Definition at line 737 of file vaapi_encode_av1.c.

vaapi_encode_av1_write_picture_header()

static int vaapi_encode_av1_write_picture_header ( AVCodecContext * avctx, FFHWBaseEncodePicture * pic, char * data, size_t * data_len )

static

Pack repeat frame header.

Definition at line 759 of file vaapi_encode_av1.c.

vaapi_encode_av1_write_extra_header()

static int vaapi_encode_av1_write_extra_header ( AVCodecContext * avctx, FFHWBaseEncodePicture * base_pic, int index, int * type, char * data, size_t * data_len )

static

Definition at line 809 of file vaapi_encode_av1.c.

vaapi_encode_av1_init()

static av_cold int vaapi_encode_av1_init ( AVCodecContext * avctx )

static

This attr provides essential indicators, return error if not support.

Definition at line 873 of file vaapi_encode_av1.c.

vaapi_encode_av1_close()

static av_cold int vaapi_encode_av1_close ( AVCodecContext * avctx )

static

Definition at line 960 of file vaapi_encode_av1.c.

Variable Documentation

vaapi_encode_av1_profiles

const VAAPIEncodeProfile vaapi_encode_av1_profiles[]

static

Initial value:

= {
    { AV_PROFILE_AV1_MAIN,  8, 3, 1, 1, VAProfileAV1Profile0 },
    { AV_PROFILE_AV1_MAIN, 10, 3, 1, 1, VAProfileAV1Profile0 },
    { AV_PROFILE_UNKNOWN }
}

Definition at line 842 of file vaapi_encode_av1.c.

vaapi_encode_type_av1

const VAAPIEncodeType vaapi_encode_type_av1

static

Initial value:

= {
    .profiles        = vaapi_encode_av1_profiles,
    .flags           = FF_HW_FLAG_B_PICTURES | FF_HW_FLAG_TIMESTAMP_NO_DELAY,
    .default_quality = 25,
 
    .get_encoder_caps = &vaapi_encode_av1_get_encoder_caps,
    .configure        = &vaapi_encode_av1_configure,
 
    .sequence_header_type  = VAEncPackedHeaderSequence,
    .sequence_params_size  = sizeof(VAEncSequenceParameterBufferAV1),
    .init_sequence_params  = &vaapi_encode_av1_init_sequence_params,
    .write_sequence_header = &vaapi_encode_av1_write_sequence_header,
 
    .picture_priv_data_size = sizeof(VAAPIEncodeAV1Picture),
    .picture_header_type    = VAEncPackedHeaderPicture,
    .picture_params_size    = sizeof(VAEncPictureParameterBufferAV1),
    .init_picture_params    = &vaapi_encode_av1_init_picture_params,
    .write_picture_header   = &vaapi_encode_av1_write_picture_header,
 
    .slice_params_size = sizeof(VAEncTileGroupBufferAV1),
    .init_slice_params = &vaapi_encode_av1_init_slice_params,
 
    .write_extra_header     = &vaapi_encode_av1_write_extra_header,
}

Definition at line 848 of file vaapi_encode_av1.c.

Referenced by vaapi_encode_av1_init().

vaapi_encode_av1_options

const AVOption vaapi_encode_av1_options[]

static

Definition at line 973 of file vaapi_encode_av1.c.

vaapi_encode_av1_defaults

const FFCodecDefault vaapi_encode_av1_defaults[]

static

Initial value:

= {
    { "b",              "0"   },
    { "bf",             "2"   },
    { "g",              "120" },
    { "qmin",           "1"   },
    { "qmax",           "255" },
    { "refs",           "0"   },
    { NULL },
}

Definition at line 1024 of file vaapi_encode_av1.c.

vaapi_encode_av1_class

const AVClass vaapi_encode_av1_class

static

Initial value:

= {
    .class_name = "av1_vaapi",
    .item_name  = av_default_item_name,
    .option     = vaapi_encode_av1_options,
    .version    = LIBAVUTIL_VERSION_INT,
}

Definition at line 1034 of file vaapi_encode_av1.c.

ff_av1_vaapi_encoder

const FFCodec ff_av1_vaapi_encoder

Initial value:

= {
    .p.name         = "av1_vaapi",
    CODEC_LONG_NAME("AV1 (VAAPI)"),
    .p.type         = AVMEDIA_TYPE_VIDEO,
    .p.id           = AV_CODEC_ID_AV1,
    .priv_data_size = sizeof(VAAPIEncodeAV1Context),
    .init           = &vaapi_encode_av1_init,
    FF_CODEC_RECEIVE_PACKET_CB(&ff_vaapi_encode_receive_packet),
    .close          = &vaapi_encode_av1_close,
    .p.priv_class   = &vaapi_encode_av1_class,
    .p.capabilities = AV_CODEC_CAP_DELAY | AV_CODEC_CAP_HARDWARE |
                      AV_CODEC_CAP_DR1 | AV_CODEC_CAP_ENCODER_REORDERED_OPAQUE,
    .caps_internal  = FF_CODEC_CAP_NOT_INIT_THREADSAFE |
                      FF_CODEC_CAP_INIT_CLEANUP,
    .defaults       = vaapi_encode_av1_defaults,
    CODEC_PIXFMTS(AV_PIX_FMT_VAAPI),
    .color_ranges   = AVCOL_RANGE_MPEG | AVCOL_RANGE_JPEG,
    .hw_configs     = ff_vaapi_encode_hw_configs,
    .p.wrapper_name = "vaapi",
}

Definition at line 1041 of file vaapi_encode_av1.c.