FFmpeg
Data Structures | Macros | Enumerations | Functions | Variables
vulkan_encode_h264.c File Reference
#include "libavutil/opt.h"
#include "libavutil/mem.h"
#include "cbs.h"
#include "cbs_h264.h"
#include "atsc_a53.h"
#include "h264_levels.h"
#include "h2645data.h"
#include "codec_internal.h"
#include "version.h"
#include "hw_base_encode_h264.h"
#include "vulkan_encode.h"

Go to the source code of this file.

Data Structures

struct  VulkanEncodeH264Picture
 
struct  VulkanEncodeH264Context
 
struct  VulkanH264Units
 

Macros

#define OFFSET(x)   offsetof(VulkanEncodeH264Context, 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_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 int vk_enc_h264_update_pic_info (AVCodecContext *avctx, FFHWBaseEncodePicture *pic)
 
static void setup_slices (AVCodecContext *avctx, FFHWBaseEncodePicture *pic)
 
static void vk_enc_h264_default_ref_pic_list (AVCodecContext *avctx, FFHWBaseEncodePicture *pic, FFHWBaseEncodePicture **rpl0, FFHWBaseEncodePicture **rpl1, int *rpl_size)
 
static void setup_refs (AVCodecContext *avctx, FFHWBaseEncodePicture *pic, VkVideoEncodeInfoKHR *encode_info)
 
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, VulkanH264Units *vk_units)
 
static int create_session_params (AVCodecContext *avctx)
 
static int parse_feedback_units (AVCodecContext *avctx, const uint8_t *data, size_t size, int sps_override, int pps_override)
 
static int init_base_units (AVCodecContext *avctx)
 
static int vulkan_encode_h264_add_nal (AVCodecContext *avctx, CodedBitstreamFragment *au, void *nal_unit)
 
static int write_access_unit (AVCodecContext *avctx, uint8_t *data, size_t *data_len, CodedBitstreamFragment *au)
 
static int write_sequence_headers (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_filler (AVCodecContext *avctx, uint32_t filler, uint8_t *data, size_t *data_len)
 
static av_cold int vulkan_encode_h264_init (AVCodecContext *avctx)
 
static av_cold int vulkan_encode_h264_close (AVCodecContext *avctx)
 

Variables

const FFVulkanEncodeDescriptor ff_vk_enc_h264_desc
 
static const uint8_t vulkan_encode_h264_sei_identifier_uuid [16]
 
static const FFVulkanCodec enc_cb
 
static const AVOption vulkan_encode_h264_options []
 
static const FFCodecDefault vulkan_encode_h264_defaults []
 
static const AVClass vulkan_encode_h264_class
 
const FFCodec ff_h264_vulkan_encoder
 

Macro Definition Documentation

◆ OFFSET

#define OFFSET (   x)    offsetof(VulkanEncodeH264Context, x)

Definition at line 1564 of file vulkan_encode_h264.c.

◆ FLAGS

#define FLAGS   (AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_ENCODING_PARAM)

Definition at line 1565 of file vulkan_encode_h264.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_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 34 of file vulkan_encode_h264.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 116 of file vulkan_encode_h264.c.

◆ vk_enc_h264_update_pic_info()

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

Definition at line 161 of file vulkan_encode_h264.c.

Referenced by init_pic_params().

◆ setup_slices()

static void setup_slices ( AVCodecContext *  avctx,
FFHWBaseEncodePicture *  pic 
)
static

Definition at line 246 of file vulkan_encode_h264.c.

Referenced by init_pic_params().

◆ vk_enc_h264_default_ref_pic_list()

static void vk_enc_h264_default_ref_pic_list ( AVCodecContext *  avctx,
FFHWBaseEncodePicture *  pic,
FFHWBaseEncodePicture **  rpl0,
FFHWBaseEncodePicture **  rpl1,
int *  rpl_size 
)
static

Definition at line 309 of file vulkan_encode_h264.c.

Referenced by setup_refs().

◆ setup_refs()

static void setup_refs ( AVCodecContext *  avctx,
FFHWBaseEncodePicture *  pic,
VkVideoEncodeInfoKHR *  encode_info 
)
static

Definition at line 406 of file vulkan_encode_h264.c.

Referenced by init_pic_params().

◆ init_pic_params()

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

Definition at line 605 of file vulkan_encode_h264.c.

◆ init_profile()

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

Definition at line 678 of file vulkan_encode_h264.c.

◆ init_enc_options()

static int init_enc_options ( AVCodecContext *  avctx)
static

Definition at line 756 of file vulkan_encode_h264.c.

Referenced by vulkan_encode_h264_init().

◆ init_sequence_headers()

static av_cold int init_sequence_headers ( AVCodecContext *  avctx)
static

Definition at line 813 of file vulkan_encode_h264.c.

Referenced by init_base_units().

◆ base_unit_to_vk()

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

Definition at line 856 of file vulkan_encode_h264.c.

Referenced by create_session_params().

◆ create_session_params()

static int create_session_params ( AVCodecContext *  avctx)
static

Definition at line 1005 of file vulkan_encode_h264.c.

Referenced by init_base_units().

◆ parse_feedback_units()

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

Definition at line 1049 of file vulkan_encode_h264.c.

Referenced by init_base_units().

◆ init_base_units()

static int init_base_units ( AVCodecContext *  avctx)
static

Definition at line 1087 of file vulkan_encode_h264.c.

Referenced by vulkan_encode_h264_init().

◆ vulkan_encode_h264_add_nal()

static int vulkan_encode_h264_add_nal ( AVCodecContext *  avctx,
CodedBitstreamFragment *  au,
void *  nal_unit 
)
static

Definition at line 1186 of file vulkan_encode_h264.c.

Referenced by write_extra_headers(), write_filler(), and write_sequence_headers().

◆ write_access_unit()

static int write_access_unit ( AVCodecContext *  avctx,
uint8_t *  data,
size_t *  data_len,
CodedBitstreamFragment *  au 
)
static

Definition at line 1201 of file vulkan_encode_h264.c.

Referenced by write_extra_headers(), write_filler(), and write_sequence_headers().

◆ write_sequence_headers()

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

Definition at line 1225 of file vulkan_encode_h264.c.

◆ write_extra_headers()

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

Definition at line 1254 of file vulkan_encode_h264.c.

◆ write_filler()

static int write_filler ( AVCodecContext *  avctx,
uint32_t  filler,
uint8_t *  data,
size_t *  data_len 
)
static

Definition at line 1321 of file vulkan_encode_h264.c.

◆ vulkan_encode_h264_init()

static av_cold int vulkan_encode_h264_init ( AVCodecContext *  avctx)
static

Definition at line 1359 of file vulkan_encode_h264.c.

◆ vulkan_encode_h264_close()

static av_cold int vulkan_encode_h264_close ( AVCodecContext *  avctx)
static

Definition at line 1557 of file vulkan_encode_h264.c.

Variable Documentation

◆ ff_vk_enc_h264_desc

const FFVulkanEncodeDescriptor ff_vk_enc_h264_desc
Initial value:
= {
.codec_id = AV_CODEC_ID_H264,
.encode_extension = FF_VK_EXT_VIDEO_ENCODE_H264,
.encode_op = VK_VIDEO_CODEC_OPERATION_ENCODE_H264_BIT_KHR,
.ext_props = {
.extensionName = VK_STD_VULKAN_VIDEO_CODEC_H264_ENCODE_EXTENSION_NAME,
.specVersion = VK_STD_VULKAN_VIDEO_CODEC_H264_ENCODE_SPEC_VERSION,
},
}

Definition at line 42 of file vulkan_encode_h264.c.

Referenced by vulkan_encode_h264_init().

◆ vulkan_encode_h264_sei_identifier_uuid

const uint8_t vulkan_encode_h264_sei_identifier_uuid[16]
static
Initial value:
= {
0x03, 0xfd, 0xf2, 0x0a, 0x5d, 0x4c, 0x05, 0x48,
0x20, 0x98, 0xca, 0x6b, 0x0c, 0x95, 0x30, 0x1c,
}

Definition at line 53 of file vulkan_encode_h264.c.

Referenced by vulkan_encode_h264_init().

◆ enc_cb

const FFVulkanCodec enc_cb
static
Initial value:
= {
.flags = FF_HW_FLAG_B_PICTURES |
FF_HW_FLAG_B_PICTURE_REFERENCES |
FF_HW_FLAG_NON_IDR_KEY_PICTURES,
.picture_priv_data_size = sizeof(VulkanEncodeH264Picture),
.filler_header_size = 6,
.init_profile = init_profile,
.init_pic_rc = init_pic_rc,
.init_pic_params = init_pic_params,
.write_sequence_headers = write_sequence_headers,
.write_extra_headers = write_extra_headers,
.write_filler = write_filler,
}

Definition at line 1345 of file vulkan_encode_h264.c.

Referenced by vulkan_encode_h264_init().

◆ vulkan_encode_h264_options

const AVOption vulkan_encode_h264_options[]
static

Definition at line 1566 of file vulkan_encode_h264.c.

◆ vulkan_encode_h264_defaults

const FFCodecDefault vulkan_encode_h264_defaults[]
static
Initial value:
= {
{ "b", "0" },
{ "bf", "2" },
{ "g", "300" },
{ "i_qfactor", "1" },
{ "i_qoffset", "0" },
{ "b_qfactor", "1" },
{ "b_qoffset", "0" },
{ "qmin", "-1" },
{ "qmax", "-1" },
{ NULL },
}

Definition at line 1623 of file vulkan_encode_h264.c.

◆ vulkan_encode_h264_class

const AVClass vulkan_encode_h264_class
static
Initial value:
= {
.class_name = "h264_vulkan",
.item_name = av_default_item_name,
.option = vulkan_encode_h264_options,
.version = LIBAVUTIL_VERSION_INT,
}

Definition at line 1636 of file vulkan_encode_h264.c.

◆ ff_h264_vulkan_encoder

const FFCodec ff_h264_vulkan_encoder
Initial value:
= {
.p.name = "h264_vulkan",
CODEC_LONG_NAME("H.264/AVC (Vulkan)"),
.p.type = AVMEDIA_TYPE_VIDEO,
.p.id = AV_CODEC_ID_H264,
.priv_data_size = sizeof(VulkanEncodeH264Context),
.init = &vulkan_encode_h264_init,
FF_CODEC_RECEIVE_PACKET_CB(&ff_vulkan_encode_receive_packet),
.close = &vulkan_encode_h264_close,
.p.priv_class = &vulkan_encode_h264_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_h264_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 1643 of file vulkan_encode_h264.c.

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:43
FF_HW_FLAG_B_PICTURE_REFERENCES
@ FF_HW_FLAG_B_PICTURE_REFERENCES
Definition: hw_base_encode.h:55
ff_vulkan_encode_hw_configs
const AVCodecHWConfigInternal *const ff_vulkan_encode_hw_configs[]
Paperwork.
Definition: vulkan_encode.c:26
AV_CODEC_CAP_HARDWARE
#define AV_CODEC_CAP_HARDWARE
Codec is backed by a hardware implementation.
Definition: codec.h:145
FF_HW_FLAG_B_PICTURES
@ FF_HW_FLAG_B_PICTURES
Definition: hw_base_encode.h:53
vulkan_encode_h264_init
static av_cold int vulkan_encode_h264_init(AVCodecContext *avctx)
Definition: vulkan_encode_h264.c:1359
vulkan_encode_h264_close
static av_cold int vulkan_encode_h264_close(AVCodecContext *avctx)
Definition: vulkan_encode_h264.c:1557
AV_PIX_FMT_VULKAN
@ AV_PIX_FMT_VULKAN
Vulkan hardware images.
Definition: pixfmt.h:379
FF_VK_EXT_VIDEO_ENCODE_H264
@ FF_VK_EXT_VIDEO_ENCODE_H264
Definition: vulkan_functions.h:55
write_extra_headers
static int write_extra_headers(AVCodecContext *avctx, FFHWBaseEncodePicture *base_pic, uint8_t *data, size_t *data_len)
Definition: vulkan_encode_h264.c:1254
AV_CODEC_CAP_ENCODER_FLUSH
#define AV_CODEC_CAP_ENCODER_FLUSH
This encoder can be flushed using avcodec_flush_buffers().
Definition: codec.h:166
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:159
AV_CODEC_ID_H264
@ AV_CODEC_ID_H264
Definition: codec_id.h:79
CODEC_LONG_NAME
#define CODEC_LONG_NAME(str)
Definition: codec_internal.h:296
LIBAVUTIL_VERSION_INT
#define LIBAVUTIL_VERSION_INT
Definition: version.h:85
NULL
#define NULL
Definition: coverity.c:32
FF_CODEC_RECEIVE_PACKET_CB
#define FF_CODEC_RECEIVE_PACKET_CB(func)
Definition: codec_internal.h:326
av_default_item_name
const char * av_default_item_name(void *ptr)
Return the context name.
Definition: log.c:237
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
VulkanEncodeH264Context
Definition: vulkan_encode_h264.c:87
VulkanEncodeH264Picture
Definition: vulkan_encode_h264.c:58
vulkan_encode_h264_options
static const AVOption vulkan_encode_h264_options[]
Definition: vulkan_encode_h264.c:1566
FLAGS
#define FLAGS
Definition: vulkan_encode_h264.c:1565
vulkan_encode_h264_defaults
static const FFCodecDefault vulkan_encode_h264_defaults[]
Definition: vulkan_encode_h264.c:1623
ff_vulkan_encode_receive_packet
int ff_vulkan_encode_receive_packet(AVCodecContext *avctx, AVPacket *pkt)
Encode.
Definition: vulkan_encode.c:542
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:1225
init_profile
static int init_profile(AVCodecContext *avctx, VkVideoProfileInfoKHR *profile, void *pnext)
Definition: vulkan_encode_h264.c:678
init_pic_rc
static int init_pic_rc(AVCodecContext *avctx, FFHWBaseEncodePicture *pic, VkVideoEncodeRateControlInfoKHR *rc_info, VkVideoEncodeRateControlLayerInfoKHR *rc_layer)
Definition: vulkan_encode_h264.c:116
FF_HW_FLAG_NON_IDR_KEY_PICTURES
@ FF_HW_FLAG_NON_IDR_KEY_PICTURES
Definition: hw_base_encode.h:58
AV_PIX_FMT_NONE
@ AV_PIX_FMT_NONE
Definition: pixfmt.h:72
vulkan_encode_h264_class
static const AVClass vulkan_encode_h264_class
Definition: vulkan_encode_h264.c:1636
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
AVMEDIA_TYPE_VIDEO
@ AVMEDIA_TYPE_VIDEO
Definition: avutil.h:201
write_filler
static int write_filler(AVCodecContext *avctx, uint32_t filler, uint8_t *data, size_t *data_len)
Definition: vulkan_encode_h264.c:1321
init_pic_params
static int init_pic_params(AVCodecContext *avctx, FFHWBaseEncodePicture *pic, VkVideoEncodeInfoKHR *encode_info)
Definition: vulkan_encode_h264.c:605
AV_OPT_TYPE_CONST
@ AV_OPT_TYPE_CONST
Special option type for declaring named constants.
Definition: opt.h:299
Generated on Fri Sep 27 2024 19:23:14 for FFmpeg by   doxygen 1.8.17