FFmpeg
Data Structures | Macros | Enumerations | Functions | Variables
vulkan_encode_av1.c File Reference
#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"

Go to the source code of this file.

Data Structures

struct  VulkanEncodeAV1Picture
 
struct  VulkanEncodeAV1Context
 
struct  VulkanAV1Units
 

Macros

#define OFFSET(x)   offsetof(VulkanEncodeAV1Context, 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, UNIT_AUD = 1 << 0, UNIT_SEI_TIMING = 1 << 1,
  UNIT_SEI_IDENTIFIER = 1 << 2, UNIT_SEI_RECOVERY = 1 << 3, UNIT_SEI_A53_CC = 1 << 4, UNIT_AUD = 1 << 0,
  UNIT_SEI_MASTERING_DISPLAY = 1 << 1, UNIT_SEI_CONTENT_LIGHT_LEVEL = 1 << 2, UNIT_SEI_A53_CC = 1 << 3
}
 

Functions

static int init_pic_rc (AVCodecContext *avctx, FFHWBaseEncodePicture *pic, VkVideoEncodeRateControlInfoKHR *rc_info, VkVideoEncodeRateControlLayerInfoKHR *rc_layer)
 
static void 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)    offsetof(VulkanEncodeAV1Context, x)

Definition at line 1304 of file vulkan_encode_av1.c.

◆ FLAGS

#define FLAGS   (AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_ENCODING_PARAM)

Definition at line 1305 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"

◆ 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 
UNIT_AUD 
UNIT_SEI_TIMING 
UNIT_SEI_IDENTIFIER 
UNIT_SEI_RECOVERY 
UNIT_SEI_A53_CC 
UNIT_AUD 
UNIT_SEI_MASTERING_DISPLAY 
UNIT_SEI_CONTENT_LIGHT_LEVEL 
UNIT_SEI_A53_CC 

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 106 of file vulkan_encode_av1.c.

◆ set_name_slot()

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

Definition at line 160 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.

Definition at line 176 of file vulkan_encode_av1.c.

◆ init_profile()

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

Definition at line 548 of file vulkan_encode_av1.c.

◆ init_enc_options()

static int init_enc_options ( AVCodecContext *  avctx)
static

Arbitrary value

Definition at line 649 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 699 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 764 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 858 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 895 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 939 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 1019 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 1036 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 1055 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 1075 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 1114 of file vulkan_encode_av1.c.

◆ vulkan_encode_av1_init()

static av_cold int vulkan_encode_av1_init ( AVCodecContext *  avctx)
static

Definition at line 1159 of file vulkan_encode_av1.c.

◆ vulkan_encode_av1_close()

static av_cold int vulkan_encode_av1_close ( AVCodecContext *  avctx)
static

Definition at line 1296 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 1144 of file vulkan_encode_av1.c.

Referenced by vulkan_encode_av1_init().

◆ vulkan_encode_av1_options

const AVOption vulkan_encode_av1_options[]
static

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

Definition at line 1364 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 1373 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,
.p.pix_fmts = (const enum AVPixelFormat[]) {
AV_PIX_FMT_VULKAN,
AV_PIX_FMT_NONE,
},
.hw_configs = ff_vulkan_encode_hw_configs,
.p.wrapper_name = "vulkan",
}

Definition at line 1380 of file vulkan_encode_av1.c.

FF_HW_FLAG_B_PICTURE_REFERENCES
@ FF_HW_FLAG_B_PICTURE_REFERENCES
Definition: hw_base_encode.h:55
FLAGS
#define FLAGS
Definition: vulkan_encode_av1.c:1305
AVPixelFormat
AVPixelFormat
Pixel format.
Definition: pixfmt.h:71
name
it s the only field you need to keep assuming you have a context There is some magic you don t need to care about around this just let it vf default minimum maximum flags name is the option name
Definition: writing_filters.txt:88
FF_CODEC_CAP_INIT_CLEANUP
#define FF_CODEC_CAP_INIT_CLEANUP
The codec allows calling the close function for deallocation even if the init function returned a fai...
Definition: codec_internal.h:42
ff_vulkan_encode_hw_configs
const AVCodecHWConfigInternal *const ff_vulkan_encode_hw_configs[]
Paperwork.
Definition: vulkan_encode.c:26
init_pic_rc
static int init_pic_rc(AVCodecContext *avctx, FFHWBaseEncodePicture *pic, VkVideoEncodeRateControlInfoKHR *rc_info, VkVideoEncodeRateControlLayerInfoKHR *rc_layer)
Definition: vulkan_encode_av1.c:106
AV_CODEC_CAP_HARDWARE
#define AV_CODEC_CAP_HARDWARE
Codec is backed by a hardware implementation.
Definition: codec.h:130
AV_PIX_FMT_VULKAN
@ AV_PIX_FMT_VULKAN
Vulkan hardware images.
Definition: pixfmt.h:379
VulkanEncodeAV1Context
Definition: vulkan_encode_av1.c:73
vulkan_encode_av1_defaults
static const FFCodecDefault vulkan_encode_av1_defaults[]
Definition: vulkan_encode_av1.c:1364
write_extra_headers
static int write_extra_headers(AVCodecContext *avctx, FFHWBaseEncodePicture *base_pic, uint8_t *data, size_t *data_len)
Definition: vulkan_encode_av1.c:1075
AV_CODEC_CAP_ENCODER_FLUSH
#define AV_CODEC_CAP_ENCODER_FLUSH
This encoder can be flushed using avcodec_flush_buffers().
Definition: codec.h:151
vulkan_encode_av1_class
static const AVClass vulkan_encode_av1_class
Definition: vulkan_encode_av1.c:1373
AV_CODEC_CAP_ENCODER_REORDERED_OPAQUE
#define AV_CODEC_CAP_ENCODER_REORDERED_OPAQUE
This encoder can reorder user opaque values from input AVFrames and return them with corresponding ou...
Definition: codec.h:144
FF_HW_FLAG_SLICE_CONTROL
@ FF_HW_FLAG_SLICE_CONTROL
Definition: hw_base_encode.h:47
CODEC_LONG_NAME
#define CODEC_LONG_NAME(str)
Definition: codec_internal.h:326
LIBAVUTIL_VERSION_INT
#define LIBAVUTIL_VERSION_INT
Definition: version.h:85
NULL
#define NULL
Definition: coverity.c:32
AV_CODEC_ID_AV1
@ AV_CODEC_ID_AV1
Definition: codec_id.h:284
FF_CODEC_RECEIVE_PACKET_CB
#define FF_CODEC_RECEIVE_PACKET_CB(func)
Definition: codec_internal.h:361
av_default_item_name
const char * av_default_item_name(void *ptr)
Return the context name.
Definition: log.c:241
VK_ENC_FLAG_NO_DELAY
#define VK_ENC_FLAG_NO_DELAY
Definition: vulkan_encode.h:97
VulkanEncodeAV1Picture
Definition: vulkan_encode_av1.c:47
vulkan_encode_av1_close
static av_cold int vulkan_encode_av1_close(AVCodecContext *avctx)
Definition: vulkan_encode_av1.c:1296
init
int(* init)(AVBSFContext *ctx)
Definition: dts2pts.c:368
AV_CODEC_CAP_DR1
#define AV_CODEC_CAP_DR1
Codec uses get_buffer() or get_encode_buffer() for allocating buffers and supports custom allocators.
Definition: codec.h:52
write_padding
static int write_padding(AVCodecContext *avctx, uint32_t padding, uint8_t *data, size_t *data_len)
Definition: vulkan_encode_av1.c:1114
init_profile
static int init_profile(AVCodecContext *avctx, VkVideoProfileInfoKHR *profile, void *pnext)
Definition: vulkan_encode_av1.c:548
ff_vulkan_encode_receive_packet
int ff_vulkan_encode_receive_packet(AVCodecContext *avctx, AVPacket *pkt)
Encode.
Definition: vulkan_encode.c:541
value
it s the only field you need to keep assuming you have a context There is some magic you don t need to care about around this just let it vf default value
Definition: writing_filters.txt:86
write_sequence_headers
static int write_sequence_headers(AVCodecContext *avctx, FFHWBaseEncodePicture *base_pic, uint8_t *data, size_t *data_len)
Definition: vulkan_encode_h264.c:1228
write_sequence_header
static int write_sequence_header(AVCodecContext *avctx, FFHWBaseEncodePicture *base_pic, uint8_t *data, size_t *data_len)
Definition: vulkan_encode_av1.c:1055
AV_PIX_FMT_NONE
@ AV_PIX_FMT_NONE
Definition: pixfmt.h:72
AV_CODEC_CAP_DELAY
#define AV_CODEC_CAP_DELAY
Encoder or decoder requires flushing with NULL input at the end in order to give the complete and cor...
Definition: codec.h:76
FF_HW_FLAG_B_PICTURES
@ FF_HW_FLAG_B_PICTURES
Definition: hw_base_encode.h:53
AVMEDIA_TYPE_VIDEO
@ AVMEDIA_TYPE_VIDEO
Definition: avutil.h:200
vulkan_encode_av1_options
static const AVOption vulkan_encode_av1_options[]
Definition: vulkan_encode_av1.c:1306
vulkan_encode_av1_init
static av_cold int vulkan_encode_av1_init(AVCodecContext *avctx)
Definition: vulkan_encode_av1.c:1159
FF_VK_EXT_VIDEO_ENCODE_AV1
#define FF_VK_EXT_VIDEO_ENCODE_AV1
Definition: vulkan_functions.h:68
write_filler
static int write_filler(AVCodecContext *avctx, uint32_t filler, uint8_t *data, size_t *data_len)
Definition: vulkan_encode_h264.c:1325
AV_OPT_TYPE_CONST
@ AV_OPT_TYPE_CONST
Special option type for declaring named constants.
Definition: opt.h:299
init_pic_params
static int init_pic_params(AVCodecContext *avctx, FFHWBaseEncodePicture *pic, VkVideoEncodeInfoKHR *encode_info)
Definition: vulkan_encode_av1.c:176
Generated on Sun Aug 10 2025 19:23:45 for FFmpeg by   doxygen 1.8.17