#include "libavutil/opt.h"
#include "libavutil/mem.h"
#include "cbs.h"
#include "cbs_av1.h"
#include "av1_levels.h"
#include "libavutil/mastering_display_metadata.h"
#include "codec_internal.h"
#include "vulkan_encode.h"
#include "libavutil/avassert.h"

Data Structures
struct	VulkanEncodeAV1Picture

struct	VulkanEncodeAV1Context

struct	VulkanAV1Units

Macros
#define	OFFSET(x)

#define	FLAGS (AV_OPT_FLAG_VIDEO_PARAM \| AV_OPT_FLAG_ENCODING_PARAM)

#define	PROFILE(name, value)

#define	LEVEL(name, value)

Enumerations
enum	UnitElems { UNIT_MASTERING_DISPLAY = 1 << 0 , UNIT_CONTENT_LIGHT_LEVEL = 1 << 1 }

Functions
static int	init_pic_rc (AVCodecContext avctx, FFHWBaseEncodePicture pic, VkVideoEncodeRateControlInfoKHR rc_info, VkVideoEncodeRateControlLayerInfoKHR rc_layer)

static int	set_name_slot (int slot, int *slot_indices, uint32_t allowed_idx, int group)

static int	init_pic_params (AVCodecContext avctx, FFHWBaseEncodePicture pic, VkVideoEncodeInfoKHR *encode_info)

static int	init_profile (AVCodecContext avctx, VkVideoProfileInfoKHR profile, void *pnext)

static int	init_enc_options (AVCodecContext *avctx)

static av_cold int	init_sequence_headers (AVCodecContext *avctx)

static av_cold int	base_unit_to_vk (AVCodecContext avctx, VulkanAV1Units vk_units)

static int	create_session_params (AVCodecContext *avctx)

static int	parse_feedback_units (AVCodecContext avctx, const uint8_t data, size_t size)

static int	init_base_units (AVCodecContext *avctx)

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

static int	vulkan_encode_av1_write_obu (AVCodecContext avctx, uint8_t data, size_t data_len, CodedBitstreamFragment obu)

static int	write_sequence_header (AVCodecContext avctx, FFHWBaseEncodePicture base_pic, uint8_t data, size_t data_len)

static int	write_extra_headers (AVCodecContext avctx, FFHWBaseEncodePicture base_pic, uint8_t data, size_t data_len)

static int	write_padding (AVCodecContext avctx, uint32_t padding, uint8_t data, size_t *data_len)

static av_cold int	vulkan_encode_av1_init (AVCodecContext *avctx)

static av_cold int	vulkan_encode_av1_close (AVCodecContext *avctx)

Variables
const FFVulkanEncodeDescriptor	ff_vk_enc_av1_desc

static const FFVulkanCodec	enc_cb

static const AVOption	vulkan_encode_av1_options []

static const FFCodecDefault	vulkan_encode_av1_defaults []

static const AVClass	vulkan_encode_av1_class

const FFCodec	ff_av1_vulkan_encoder

Macro Definition Documentation

◆ OFFSET

#define OFFSET ( x )

Value:

offsetof(VulkanEncodeAV1Context, x)

VulkanEncodeAV1Context

Definition vulkan_encode_av1.c:73

Definition at line 1375 of file vulkan_encode_av1.c.

◆ FLAGS

#define FLAGS (AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_ENCODING_PARAM)

Definition at line 1376 of file vulkan_encode_av1.c.

◆ PROFILE

#define PROFILE	(		name,
			value )

Value:

name, NULL, 0, AV_OPT_TYPE_CONST, \

{ .i64 = value }, 0, 0, FLAGS, .unit = "profile"

FLAGS

#define FLAGS

Definition cmdutils.c:598

NULL

#define NULL

Definition coverity.c:32

value

double value

Definition eval.c:102

AV_OPT_TYPE_CONST

@ AV_OPT_TYPE_CONST

Special option type for declaring named constants.

Definition opt.h:298

name

const char * name

Definition qsvenc.c:142

◆ LEVEL

#define LEVEL	(		name,
			value )

Value:

name, NULL, 0, AV_OPT_TYPE_CONST, \

{ .i64 = value }, 0, 0, FLAGS, .unit = "level"

Enumeration Type Documentation

◆ UnitElems

enum UnitElems

Enumerator
UNIT_MASTERING_DISPLAY
UNIT_CONTENT_LIGHT_LEVEL

Definition at line 42 of file vulkan_encode_av1.c.

Function Documentation

◆ init_pic_rc()

static int init_pic_rc	(	AVCodecContext *	avctx,
		FFHWBaseEncodePicture *	pic,
		VkVideoEncodeRateControlInfoKHR *	rc_info,
		VkVideoEncodeRateControlLayerInfoKHR *	rc_layer )

static

Definition at line 108 of file vulkan_encode_av1.c.

◆ set_name_slot()

static int set_name_slot	(	int	slot,
		int *	slot_indices,
		uint32_t	allowed_idx,
		int	group )

static

Definition at line 162 of file vulkan_encode_av1.c.

Referenced by init_pic_params().

◆ init_pic_params()

static int init_pic_params	(	AVCodecContext *	avctx,
		FFHWBaseEncodePicture *	pic,
		VkVideoEncodeInfoKHR *	encode_info )

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.

hidden frame header

tail frame header

Definition at line 177 of file vulkan_encode_av1.c.

◆ init_profile()

static int init_profile	(	AVCodecContext *	avctx,
		VkVideoProfileInfoKHR *	profile,
		void *	pnext )

static

Definition at line 588 of file vulkan_encode_av1.c.

◆ init_enc_options()

static int init_enc_options ( AVCodecContext * avctx )

static

Arbitrary value

Definition at line 663 of file vulkan_encode_av1.c.

Referenced by vulkan_encode_av1_init().

◆ init_sequence_headers()

static av_cold int init_sequence_headers ( AVCodecContext * avctx )

static

Definition at line 713 of file vulkan_encode_av1.c.

Referenced by init_base_units().

◆ base_unit_to_vk()

static av_cold int base_unit_to_vk	(	AVCodecContext *	avctx,
		VulkanAV1Units *	vk_units )

static

Definition at line 804 of file vulkan_encode_av1.c.

Referenced by create_session_params().

◆ create_session_params()

static int create_session_params ( AVCodecContext * avctx )

static

Definition at line 898 of file vulkan_encode_av1.c.

Referenced by init_base_units().

◆ parse_feedback_units()

static int parse_feedback_units	(	AVCodecContext *	avctx,
		const uint8_t *	data,
		size_t	size )

static

Definition at line 935 of file vulkan_encode_av1.c.

Referenced by init_base_units().

◆ init_base_units()

static int init_base_units ( AVCodecContext * avctx )

static

Definition at line 979 of file vulkan_encode_av1.c.

Referenced by vulkan_encode_av1_init().

◆ vulkan_encode_av1_add_obu()

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

static

Definition at line 1061 of file vulkan_encode_av1.c.

Referenced by write_extra_headers(), write_padding(), and write_sequence_header().

◆ vulkan_encode_av1_write_obu()

static int vulkan_encode_av1_write_obu	(	AVCodecContext *	avctx,
		uint8_t *	data,
		size_t *	data_len,
		CodedBitstreamFragment *	obu )

static

Definition at line 1078 of file vulkan_encode_av1.c.

Referenced by write_extra_headers(), write_padding(), and write_sequence_header().

◆ write_sequence_header()

static int write_sequence_header	(	AVCodecContext *	avctx,
		FFHWBaseEncodePicture *	base_pic,
		uint8_t *	data,
		size_t *	data_len )

static

Definition at line 1097 of file vulkan_encode_av1.c.

Referenced by vulkan_encode_av1_init().

◆ write_extra_headers()

static int write_extra_headers	(	AVCodecContext *	avctx,
		FFHWBaseEncodePicture *	base_pic,
		uint8_t *	data,
		size_t *	data_len )

static

Definition at line 1117 of file vulkan_encode_av1.c.

◆ write_padding()

static int write_padding	(	AVCodecContext *	avctx,
		uint32_t	padding,
		uint8_t *	data,
		size_t *	data_len )

static

Definition at line 1179 of file vulkan_encode_av1.c.

◆ vulkan_encode_av1_init()

static av_cold int vulkan_encode_av1_init ( AVCodecContext * avctx )

static

Definition at line 1224 of file vulkan_encode_av1.c.

◆ vulkan_encode_av1_close()

static av_cold int vulkan_encode_av1_close ( AVCodecContext * avctx )

static

Definition at line 1363 of file vulkan_encode_av1.c.

Variable Documentation

◆ ff_vk_enc_av1_desc

const FFVulkanEncodeDescriptor ff_vk_enc_av1_desc

Initial value:

= {
    .codec_id         = AV_CODEC_ID_AV1,
    .encode_extension = FF_VK_EXT_VIDEO_ENCODE_AV1,
    .encode_op        = VK_VIDEO_CODEC_OPERATION_ENCODE_AV1_BIT_KHR,
    .ext_props = {
        .extensionName = VK_STD_VULKAN_VIDEO_CODEC_AV1_ENCODE_EXTENSION_NAME,
        .specVersion   = VK_STD_VULKAN_VIDEO_CODEC_AV1_ENCODE_SPEC_VERSION,
    },
}

Definition at line 32 of file vulkan_encode_av1.c.

Referenced by vulkan_encode_av1_init().

◆ enc_cb

const FFVulkanCodec enc_cb

static

Initial value:

= {
    .flags = FF_HW_FLAG_B_PICTURES |
             FF_HW_FLAG_B_PICTURE_REFERENCES |
             VK_ENC_FLAG_NO_DELAY |
             FF_HW_FLAG_SLICE_CONTROL,
    .picture_priv_data_size = sizeof(VulkanEncodeAV1Picture),
    .filler_header_size = 4,
    .init_profile = init_profile,
    .init_pic_rc = init_pic_rc,
    .init_pic_params = init_pic_params,
    .write_sequence_headers = write_sequence_header,
    .write_extra_headers = write_extra_headers,
    .write_filler = write_padding,
}

Definition at line 1209 of file vulkan_encode_av1.c.

Referenced by vulkan_encode_av1_init(), vulkan_encode_h264_init(), and vulkan_encode_h265_init().

◆ vulkan_encode_av1_options

const AVOption vulkan_encode_av1_options[]

static

Definition at line 1377 of file vulkan_encode_av1.c.

◆ vulkan_encode_av1_defaults

const FFCodecDefault vulkan_encode_av1_defaults[]

static

Initial value:

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

Definition at line 1435 of file vulkan_encode_av1.c.

◆ vulkan_encode_av1_class

const AVClass vulkan_encode_av1_class

static

Initial value:

= {
    .class_name = "av1_vulkan",
    .item_name  = av_default_item_name,
    .option     = vulkan_encode_av1_options,
    .version    = LIBAVUTIL_VERSION_INT,
}

Definition at line 1445 of file vulkan_encode_av1.c.

◆ ff_av1_vulkan_encoder

const FFCodec ff_av1_vulkan_encoder

Initial value:

= {
    .p.name         = "av1_vulkan",
    CODEC_LONG_NAME("AV1 (Vulkan)"),
    .p.type         = AVMEDIA_TYPE_VIDEO,
    .p.id           = AV_CODEC_ID_AV1,
    .priv_data_size = sizeof(VulkanEncodeAV1Context),
    .init           = &vulkan_encode_av1_init,
    FF_CODEC_RECEIVE_PACKET_CB(&ff_vulkan_encode_receive_packet),
    .close          = &vulkan_encode_av1_close,
    .p.priv_class   = &vulkan_encode_av1_class,
    .p.capabilities = AV_CODEC_CAP_DELAY |
                      AV_CODEC_CAP_HARDWARE |
                      AV_CODEC_CAP_DR1 |
                      AV_CODEC_CAP_ENCODER_FLUSH |
                      AV_CODEC_CAP_ENCODER_REORDERED_OPAQUE,
    .caps_internal  = FF_CODEC_CAP_INIT_CLEANUP,
    .defaults       = vulkan_encode_av1_defaults,
    CODEC_PIXFMTS(AV_PIX_FMT_VULKAN),
    .hw_configs     = ff_vulkan_encode_hw_configs,
    .p.wrapper_name = "vulkan",
}

Definition at line 1452 of file vulkan_encode_av1.c.

Data Structures

Macros

Enumerations

Functions

Variables

Macro Definition Documentation

◆ OFFSET

◆ FLAGS

◆ PROFILE

◆ LEVEL

Enumeration Type Documentation

◆ UnitElems

Function Documentation

◆ init_pic_rc()

◆ set_name_slot()

◆ init_pic_params()

◆ init_profile()

◆ init_enc_options()

◆ init_sequence_headers()

◆ base_unit_to_vk()

◆ create_session_params()

◆ parse_feedback_units()

◆ init_base_units()

◆ vulkan_encode_av1_add_obu()

◆ vulkan_encode_av1_write_obu()

◆ write_sequence_header()

◆ write_extra_headers()

◆ write_padding()

◆ vulkan_encode_av1_init()

◆ vulkan_encode_av1_close()

Variable Documentation

◆ ff_vk_enc_av1_desc

◆ enc_cb

◆ vulkan_encode_av1_options

◆ vulkan_encode_av1_defaults

◆ vulkan_encode_av1_class

◆ ff_av1_vulkan_encoder