#include "avcodec.h"
#include "libavutil/refstruct.h"
#include "libavutil/vulkan.h"
#include <vk_video/vulkan_video_codecs_common.h>

Data Structures
struct	FFVkVideoDPBImage

struct	FFVkVideoDPB

struct	FFVkVideoCommon

struct	FFVkEncodeLoop
	Frame loop for compute encoders. More...

Macros
#define	CODEC_VER_MAJ(ver)

#define	CODEC_VER_MIN(ver)

#define	CODEC_VER_PAT(ver)

#define	CODEC_VER(ver)

Functions
enum AVPixelFormat	ff_vk_pix_fmt_from_vkfmt (VkFormat vkf)
	Get pixfmt from a Vulkan format.

VkImageAspectFlags	ff_vk_aspect_bits_from_vkfmt (VkFormat vkf)
	Get aspect bits which include all planes from a VkFormat.

VkVideoChromaSubsamplingFlagBitsKHR	ff_vk_subsampling_from_av_desc (const AVPixFmtDescriptor *desc)
	Get Vulkan's chroma subsampling from a pixfmt descriptor.

VkVideoComponentBitDepthFlagBitsKHR	ff_vk_depth_from_av_depth (int depth)
	Get Vulkan's bit depth from an [8:12] integer.

int	ff_vk_h264_level_to_av (StdVideoH264LevelIdc level)
	Convert level from Vulkan to AV.

int	ff_vk_h265_level_to_av (StdVideoH265LevelIdc level)

StdVideoH264LevelIdc	ff_vk_h264_level_to_vk (int level_idc)

StdVideoH265LevelIdc	ff_vk_h265_level_to_vk (int level_idc)

StdVideoAV1Level	ff_vk_av1_level_to_vk (int level)

StdVideoH264ProfileIdc	ff_vk_h264_profile_to_vk (int profile)
	Convert profile from/to AV to Vulkan.

StdVideoH265ProfileIdc	ff_vk_h265_profile_to_vk (int profile)

StdVideoAV1Profile	ff_vk_av1_profile_to_vk (int profile)

int	ff_vk_create_view (FFVulkanContext s, VkImageView view, VkImageAspectFlags *aspect, VkImage img, VkFormat vkf, VkImageUsageFlags usage, int layered)
	Creates image views for video frames.

int	ff_vk_video_dpb_init (FFVulkanContext s, FFVkVideoCommon common, VkFormat format, VkImageUsageFlags usage, VkImageTiling tiling, void *create_pnext, int width, int height, int nb_layers)
	Initialize the internal DPB image pool.

int	ff_vk_video_common_init (AVCodecContext avctx, FFVulkanContext s, FFVkVideoCommon common, VkVideoSessionCreateInfoKHR session_create)
	Initialize video session, allocating and binding necessary memory.

void	ff_vk_video_common_uninit (FFVulkanContext s, FFVkVideoCommon common)
	Free video session and required resources.

int	ff_vk_seg_gather_init (FFVulkanContext s, FFVkExecPool pool, FFVulkanShader *shd)
	Packs fixed-stride segment slots back to back into a contiguous buffer.

int	ff_vk_seg_gather (FFVulkanContext s, FFVkExecContext exec, FFVulkanShader shd, FFVkBuffer sizes, size_t sizes_offset, uint32_t nb_segs, FFVkBuffer sparse, uint32_t slot_size, FFVkBuffer compacted, size_t compacted_offset, VkDeviceAddress offset_addr)
	Gathers nb_segs slots of slot_size bytes from sparse into compacted, with the segment sizes given as nb_segs uint32_t in sizes at sizes_offset, followed by one more that receives the packed size.

int	ff_vk_encode_loop_init (FFVulkanContext s, FFVkExecPool pool, FFVkEncodeLoop l, int(submit_frame)(AVCodecContext avctx, FFVkExecContext exec, AVFrame frame), int(get_packet)(AVCodecContext avctx, FFVkExecContext exec, AVPacket *pkt))

int	ff_vk_encode_loop_receive_packet (AVCodecContext avctx, FFVkEncodeLoop l, AVPacket *pkt)
	Call from FFCodec.cb.receive_packet; the packet metadata is carried from the submitted frame to its packet.

void	ff_vk_encode_loop_flush (AVCodecContext avctx, FFVkEncodeLoop l)
	Waits for and discards every frame in flight.

void	ff_vk_encode_loop_uninit (FFVkEncodeLoop *l)

Macro Definition Documentation

◆ CODEC_VER_MAJ

#define CODEC_VER_MAJ ( ver )

Value:

(ver >> 22)

Definition at line 28 of file vulkan_video.h.

◆ CODEC_VER_MIN

#define CODEC_VER_MIN ( ver )

Value:

((ver >> 12) & ((1 << 10) - 1))

Definition at line 29 of file vulkan_video.h.

◆ CODEC_VER_PAT

#define CODEC_VER_PAT ( ver )

Value:

(ver & ((1 << 12) - 1))

Definition at line 30 of file vulkan_video.h.

◆ CODEC_VER

#define CODEC_VER ( ver )

Value:

CODEC_VER_MAJ(ver), CODEC_VER_MIN(ver), CODEC_VER_PAT(ver)

CODEC_VER_MIN

#define CODEC_VER_MIN(ver)

Definition vulkan_video.h:29

CODEC_VER_PAT

#define CODEC_VER_PAT(ver)

Definition vulkan_video.h:30

CODEC_VER_MAJ

#define CODEC_VER_MAJ(ver)

Definition vulkan_video.h:28

Definition at line 31 of file vulkan_video.h.

Referenced by ff_vulkan_encode_init(), vulkan_decode_get_profile(), and vulkan_encode_h264_init().

Function Documentation

◆ ff_vk_pix_fmt_from_vkfmt()

enum AVPixelFormat ff_vk_pix_fmt_from_vkfmt ( VkFormat vkf )

Get pixfmt from a Vulkan format.

Definition at line 101 of file vulkan_video.c.

Referenced by ff_vulkan_encode_init(), and vulkan_decode_get_profile().

◆ ff_vk_aspect_bits_from_vkfmt()

VkImageAspectFlags ff_vk_aspect_bits_from_vkfmt ( VkFormat vkf )

Get aspect bits which include all planes from a VkFormat.

Definition at line 109 of file vulkan_video.c.

Referenced by ff_vk_create_view().

◆ ff_vk_subsampling_from_av_desc()

VkVideoChromaSubsamplingFlagBitsKHR ff_vk_subsampling_from_av_desc ( const AVPixFmtDescriptor * desc )

Get Vulkan's chroma subsampling from a pixfmt descriptor.

Definition at line 117 of file vulkan_video.c.

Referenced by ff_vulkan_encode_init(), and vulkan_setup_profile().

◆ ff_vk_depth_from_av_depth()

VkVideoComponentBitDepthFlagBitsKHR ff_vk_depth_from_av_depth ( int depth )

Get Vulkan's bit depth from an [8:12] integer.

Definition at line 130 of file vulkan_video.c.

Referenced by ff_vulkan_encode_init(), and vulkan_setup_profile().

◆ ff_vk_h264_level_to_av()

int ff_vk_h264_level_to_av ( StdVideoH264LevelIdc level )

Convert level from Vulkan to AV.

Definition at line 141 of file vulkan_video.c.

Referenced by vulkan_decode_get_profile().

◆ ff_vk_h265_level_to_av()

int ff_vk_h265_level_to_av ( StdVideoH265LevelIdc level )

Definition at line 193 of file vulkan_video.c.

Referenced by vulkan_decode_get_profile().

◆ ff_vk_h264_level_to_vk()

StdVideoH264LevelIdc ff_vk_h264_level_to_vk ( int level_idc )

Definition at line 167 of file vulkan_video.c.

Referenced by base_unit_to_vk().

◆ ff_vk_h265_level_to_vk()

StdVideoH265LevelIdc ff_vk_h265_level_to_vk ( int level_idc )

Definition at line 212 of file vulkan_video.c.

Referenced by base_unit_to_vk(), set_sps(), and set_vps().

◆ ff_vk_av1_level_to_vk()

StdVideoAV1Level ff_vk_av1_level_to_vk ( int level )

Definition at line 231 of file vulkan_video.c.

◆ ff_vk_h264_profile_to_vk()

StdVideoH264ProfileIdc ff_vk_h264_profile_to_vk ( int profile )

Convert profile from/to AV to Vulkan.

Definition at line 262 of file vulkan_video.c.

Referenced by base_unit_to_vk(), and init_profile().

◆ ff_vk_h265_profile_to_vk()

StdVideoH265ProfileIdc ff_vk_h265_profile_to_vk ( int profile )

Definition at line 274 of file vulkan_video.c.

Referenced by base_unit_to_vk(), init_profile(), and set_vps().

◆ ff_vk_av1_profile_to_vk()

StdVideoAV1Profile ff_vk_av1_profile_to_vk ( int profile )

Definition at line 284 of file vulkan_video.c.

Referenced by init_profile().

◆ ff_vk_create_view()

int ff_vk_create_view	(	FFVulkanContext *	s,
		VkImageView *	view,
		VkImageAspectFlags *	aspect,
		VkImage	img,
		VkFormat	vkf,
		VkImageUsageFlags	usage,
		int	layered )

Creates image views for video frames.

Definition at line 294 of file vulkan_video.c.

Referenced by dpb_image_init(), ff_vk_decode_prepare_frame(), and vulkan_encode_init().

◆ ff_vk_video_dpb_init()

int ff_vk_video_dpb_init	(	FFVulkanContext *	s,
		FFVkVideoCommon *	common,
		VkFormat	format,
		VkImageUsageFlags	usage,
		VkImageTiling	tiling,
		void *	create_pnext,
		int	width,
		int	height,
		int	nb_layers )

Initialize the internal DPB image pool.

Definition at line 380 of file vulkan_video.c.

Referenced by ff_vk_decode_init(), and vulkan_encode_create_dpb().

◆ ff_vk_video_common_init()

int ff_vk_video_common_init	(	AVCodecContext *	avctx,
		FFVulkanContext *	s,
		FFVkVideoCommon *	common,
		VkVideoSessionCreateInfoKHR *	session_create )

Initialize video session, allocating and binding necessary memory.

Definition at line 447 of file vulkan_video.c.

Referenced by ff_vk_decode_init(), and ff_vulkan_encode_init().

◆ ff_vk_video_common_uninit()

void ff_vk_video_common_uninit	(	FFVulkanContext *	s,
		FFVkVideoCommon *	common )

Free video session and required resources.

Definition at line 419 of file vulkan_video.c.

Referenced by ff_vk_video_common_init(), ff_vulkan_encode_uninit(), and free_common().

◆ ff_vk_seg_gather_init()

int ff_vk_seg_gather_init	(	FFVulkanContext *	s,
		FFVkExecPool *	pool,
		FFVulkanShader *	shd )

Packs fixed-stride segment slots back to back into a contiguous buffer.

Referenced by encode_init(), vulkan_encode_apv_init(), and vulkan_encode_ffv1_init().

◆ ff_vk_seg_gather()

int ff_vk_seg_gather	(	FFVulkanContext *	s,
		FFVkExecContext *	exec,
		FFVulkanShader *	shd,
		FFVkBuffer *	sizes,
		size_t	sizes_offset,
		uint32_t	nb_segs,
		FFVkBuffer *	sparse,
		uint32_t	slot_size,
		FFVkBuffer *	compacted,
		size_t	compacted_offset,
		VkDeviceAddress	offset_addr )

Gathers nb_segs slots of slot_size bytes from sparse into compacted, with the segment sizes given as nb_segs uint32_t in sizes at sizes_offset, followed by one more that receives the packed size.

Both are made visible to the host. If offset_addr is non-zero, the uint32_t it points to is added to compacted_offset.

Referenced by vulkan_encode_ffv1_submit_frame().

◆ ff_vk_encode_loop_init()

int ff_vk_encode_loop_init	(	FFVulkanContext *	s,
		FFVkExecPool *	pool,
		FFVkEncodeLoop *	l,
		int(*	submit_frame )(AVCodecContext avctx, FFVkExecContext exec, AVFrame *frame),
		int(*	get_packet )(AVCodecContext avctx, FFVkExecContext exec, AVPacket *pkt) )

Definition at line 640 of file vulkan_video.c.

Referenced by encode_init(), vulkan_encode_apv_init(), and vulkan_encode_ffv1_init().

◆ ff_vk_encode_loop_receive_packet()

int ff_vk_encode_loop_receive_packet	(	AVCodecContext *	avctx,
		FFVkEncodeLoop *	l,
		AVPacket *	pkt )

Call from FFCodec.cb.receive_packet; the packet metadata is carried from the submitted frame to its packet.

Definition at line 696 of file vulkan_video.c.

Referenced by vulkan_encode_apv_receive_packet(), vulkan_encode_ffv1_receive_packet(), and vulkan_encode_prores_receive_packet().

◆ ff_vk_encode_loop_flush()

void ff_vk_encode_loop_flush	(	AVCodecContext *	avctx,
		FFVkEncodeLoop *	l )

Waits for and discards every frame in flight.

Definition at line 735 of file vulkan_video.c.

Referenced by vulkan_encode_apv_flush(), vulkan_encode_ffv1_flush(), and vulkan_encode_prores_flush().

◆ ff_vk_encode_loop_uninit()

void ff_vk_encode_loop_uninit ( FFVkEncodeLoop * l )

Definition at line 744 of file vulkan_video.c.

Referenced by encode_close(), vulkan_encode_apv_close(), and vulkan_encode_ffv1_close().

Data Structures

Macros

Functions

Macro Definition Documentation

◆ CODEC_VER_MAJ

◆ CODEC_VER_MIN

◆ CODEC_VER_PAT

◆ CODEC_VER

Function Documentation

◆ ff_vk_pix_fmt_from_vkfmt()

◆ ff_vk_aspect_bits_from_vkfmt()

◆ ff_vk_subsampling_from_av_desc()

◆ ff_vk_depth_from_av_depth()

◆ ff_vk_h264_level_to_av()

◆ ff_vk_h265_level_to_av()

◆ ff_vk_h264_level_to_vk()

◆ ff_vk_h265_level_to_vk()

◆ ff_vk_av1_level_to_vk()

◆ ff_vk_h264_profile_to_vk()

◆ ff_vk_h265_profile_to_vk()

◆ ff_vk_av1_profile_to_vk()

◆ ff_vk_create_view()

◆ ff_vk_video_dpb_init()

◆ ff_vk_video_common_init()

◆ ff_vk_video_common_uninit()

◆ ff_vk_seg_gather_init()

◆ ff_vk_seg_gather()

◆ ff_vk_encode_loop_init()

◆ ff_vk_encode_loop_receive_packet()

◆ ff_vk_encode_loop_flush()

◆ ff_vk_encode_loop_uninit()