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 void	vaapi_encode_av1_set_cqp_filters (AV1RawFrameHeader *fh, VAEncPictureParameterBufferAV1 *vpic)
	With VA_RC_CQP the driver does not patch the packed frame header, so choose the in-loop filter parameters here from base_q_idx and pass the same values in the picture parameters, keeping the two in sync for drivers which use the application-provided header.
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 1025 of file vaapi_encode_av1.c.

FLAGS

#define FLAGS   (AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_ENCODING_PARAM)

Definition at line 1026 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_set_cqp_filters()

static void vaapi_encode_av1_set_cqp_filters ( AV1RawFrameHeader * fh, VAEncPictureParameterBufferAV1 * vpic )

static

With VA_RC_CQP the driver does not patch the packed frame header, so choose the in-loop filter parameters here from base_q_idx and pass the same values in the picture parameters, keeping the two in sync for drivers which use the application-provided header.

Drivers which generate the frame header themselves may choose their own values. In other modes the driver writes them into the header at the bit offsets we pass.

Definition at line 476 of file vaapi_encode_av1.c.

Referenced by vaapi_encode_av1_init_picture_params().

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 520 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 792 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 814 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 864 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 928 of file vaapi_encode_av1.c.

vaapi_encode_av1_close()

static av_cold int vaapi_encode_av1_close ( AVCodecContext * avctx )

static

Definition at line 1015 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 897 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 903 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 1028 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 1079 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 1089 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 1096 of file vaapi_encode_av1.c.