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 "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)   offsetof(VAAPIEncodeAV1Context, 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, VAAPIEncodePicture *pic)
static int	vaapi_encode_av1_init_slice_params (AVCodecContext *avctx, VAAPIEncodePicture *pic, VAAPIEncodeSlice *slice)
static int	vaapi_encode_av1_write_picture_header (AVCodecContext *avctx, VAAPIEncodePicture *pic, 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 33 of file vaapi_encode_av1.c.

OFFSET

#define OFFSET

(

x

)

offsetof(VAAPIEncodeAV1Context, x)

Definition at line 861 of file vaapi_encode_av1.c.

FLAGS

#define FLAGS   (AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_ENCODING_PARAM)

Definition at line 862 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 89 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 110 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 122 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 161 of file vaapi_encode_av1.c.

Referenced by vaapi_encode_av1_init_picture_params(), vaapi_encode_av1_init_sequence_params(), 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 179 of file vaapi_encode_av1.c.

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

tile_log2()

static int tile_log2 ( int  blkSize, int  target  )

static

Definition at line 205 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 211 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 345 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 356 of file vaapi_encode_av1.c.

vaapi_encode_av1_init_picture_params()

static int vaapi_encode_av1_init_picture_params ( AVCodecContext *  avctx, VAAPIEncodePicture *  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 462 of file vaapi_encode_av1.c.

vaapi_encode_av1_init_slice_params()

static int vaapi_encode_av1_init_slice_params ( AVCodecContext *  avctx, VAAPIEncodePicture *  pic, VAAPIEncodeSlice *  slice  )

static

Set tile group info.

Definition at line 665 of file vaapi_encode_av1.c.

vaapi_encode_av1_write_picture_header()

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

static

Pack repeat frame header.

Definition at line 687 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 765 of file vaapi_encode_av1.c.

vaapi_encode_av1_close()

static av_cold int vaapi_encode_av1_close ( AVCodecContext *  avctx )

static

Definition at line 851 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 736 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           = FLAG_B_PICTURES | 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,
}

Definition at line 742 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 864 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" },
    { NULL },
}

Definition at line 914 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 923 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,
    .p.pix_fmts = (const enum AVPixelFormat[]) {
        AV_PIX_FMT_VAAPI,
        AV_PIX_FMT_NONE,
    },
    .hw_configs     = ff_vaapi_encode_hw_configs,
    .p.wrapper_name = "vaapi",
}

Definition at line 930 of file vaapi_encode_av1.c.