#include "avassert.h"
#include "mem.h"
#include "vulkan.h"
#include "libavutil/vulkan_loader.h"

Data Structures
struct	TempSyncCtx

struct	ImageViewCtx

struct	descriptor_props

Macros
#define	CASE(VAL) case VAL: return #VAL

#define	ARR_REALLOC(str, arr, alloc_s, cnt)

#define	REPS_FMT(fmt)

#define	REPS_FMT_PACK(fmt, num)

Functions
const char *	ff_vk_ret2str (VkResult res)
	Converts Vulkan return values to strings. More...

static void	load_enabled_qfs (FFVulkanContext *s)

int	ff_vk_load_props (FFVulkanContext *s)
	Loads props/mprops/driver_props. More...

static int	vk_qf_get_index (FFVulkanContext s, VkQueueFlagBits dev_family, int nb)

int	ff_vk_qf_init (FFVulkanContext s, FFVkQueueFamilyCtx qf, VkQueueFlagBits dev_family)
	Chooses a QF and loads it into a context. More...

void	ff_vk_exec_pool_free (FFVulkanContext s, FFVkExecPool pool)

int	ff_vk_exec_pool_init (FFVulkanContext s, FFVkQueueFamilyCtx qf, FFVkExecPool pool, int nb_contexts, int nb_queries, VkQueryType query_type, int query_64bit, const void query_create_pnext)
	Allocates/frees an execution pool. More...

VkResult	ff_vk_exec_get_query (FFVulkanContext s, FFVkExecContext e, void **data, VkQueryResultFlagBits flags)
	Performs nb_queries queries and returns their results and statuses. More...

FFVkExecContext *	ff_vk_exec_get (FFVulkanContext s, FFVkExecPool pool)
	Retrieve an execution pool. More...

void	ff_vk_exec_wait (FFVulkanContext s, FFVkExecContext e)

int	ff_vk_exec_start (FFVulkanContext s, FFVkExecContext e)
	Start/submit/wait an execution. More...

void	ff_vk_exec_discard_deps (FFVulkanContext s, FFVkExecContext e)

int	ff_vk_exec_add_dep_buf (FFVulkanContext s, FFVkExecContext e, AVBufferRef **deps, int nb_deps, int ref)
	Execution dependency management. More...

static void	destroy_tmp_semaphores (void opaque, uint8_t data)

int	ff_vk_exec_add_dep_bool_sem (FFVulkanContext s, FFVkExecContext e, VkSemaphore *sem, int nb, VkPipelineStageFlagBits2 stage, int wait)

int	ff_vk_exec_add_dep_frame (FFVulkanContext s, FFVkExecContext e, AVFrame *f, VkPipelineStageFlagBits2 wait_stage, VkPipelineStageFlagBits2 signal_stage)

void	ff_vk_exec_update_frame (FFVulkanContext s, FFVkExecContext e, AVFrame f, VkImageMemoryBarrier2 bar, uint32_t *nb_img_bar)

int	ff_vk_exec_mirror_sem_value (FFVulkanContext s, FFVkExecContext e, VkSemaphore dst, uint64_t dst_val, AVFrame *f)

int	ff_vk_exec_submit (FFVulkanContext s, FFVkExecContext e)

int	ff_vk_alloc_mem (FFVulkanContext s, VkMemoryRequirements req, VkMemoryPropertyFlagBits req_flags, void alloc_extension, VkMemoryPropertyFlagBits mem_flags, VkDeviceMemory *mem)
	Memory/buffer/image allocation helpers. More...

int	ff_vk_create_buf (FFVulkanContext s, FFVkBuffer buf, size_t size, void pNext, void alloc_pNext, VkBufferUsageFlags usage, VkMemoryPropertyFlagBits flags)

static void	destroy_avvkbuf (void opaque, uint8_t data)

int	ff_vk_create_avbuf (FFVulkanContext s, AVBufferRef ref, size_t size, void pNext, void *alloc_pNext, VkBufferUsageFlags usage, VkMemoryPropertyFlagBits flags)

int	ff_vk_map_buffers (FFVulkanContext s, FFVkBuffer buf, uint8_t mem[], int nb_buffers, int invalidate)
	Buffer management code. More...

int	ff_vk_unmap_buffers (FFVulkanContext s, FFVkBuffer *buf, int nb_buffers, int flush)

void	ff_vk_free_buf (FFVulkanContext s, FFVkBuffer buf)

static void	free_data_buf (void opaque, uint8_t data)

static AVBufferRef *	alloc_data_buf (void *opaque, size_t size)

int	ff_vk_get_pooled_buffer (FFVulkanContext ctx, AVBufferPool buf_pool, AVBufferRef buf, VkBufferUsageFlags usage, void create_pNext, size_t size, VkMemoryPropertyFlagBits mem_props)
	Initialize a pool and create AVBufferRefs containing FFVkBuffer. More...

int	ff_vk_shader_add_push_const (FFVulkanShader *shd, int offset, int size, VkShaderStageFlagBits stage)
	Add/update push constants for execution. More...

int	ff_vk_init_sampler (FFVulkanContext s, VkSampler sampler, int unnorm_coords, VkFilter filt)
	Create a sampler. More...

int	ff_vk_mt_is_np_rgb (enum AVPixelFormat pix_fmt)
	Returns 1 if pixfmt is a usable RGB format. More...

const char *	ff_vk_shader_rep_fmt (enum AVPixelFormat pix_fmt, enum FFVkShaderRepFormat rep_fmt)

static void	destroy_imageviews (void opaque, uint8_t data)

static VkFormat	map_fmt_to_rep (VkFormat fmt, enum FFVkShaderRepFormat rep_fmt)

int	ff_vk_create_imageviews (FFVulkanContext s, FFVkExecContext e, VkImageView views[AV_NUM_DATA_POINTERS], AVFrame *f, enum FFVkShaderRepFormat rep_fmt)
	Create an imageview and add it as a dependency to an execution. More...

void	ff_vk_frame_barrier (FFVulkanContext s, FFVkExecContext e, AVFrame pic, VkImageMemoryBarrier2 bar, int *nb_bar, VkPipelineStageFlags src_stage, VkPipelineStageFlags dst_stage, VkAccessFlagBits new_access, VkImageLayout new_layout, uint32_t new_qf)

int	ff_vk_shader_init (FFVulkanContext s, FFVulkanShader shd, const char name, VkPipelineStageFlags stage, const char extensions[], int nb_extensions, int lg_x, int lg_y, int lg_z, uint32_t required_subgroup_size)
	Initialize a shader object, with a specific set of extensions, type+bind, local group size, and subgroup requirements. More...

void	ff_vk_shader_print (void ctx, FFVulkanShader shd, int prio)
	Output the shader code as logging data, with a specific priority. More...

static int	init_pipeline_layout (FFVulkanContext s, FFVulkanShader shd)

static int	create_shader_module (FFVulkanContext s, FFVulkanShader shd, VkShaderModule mod, uint8_t spirv, size_t spirv_len)

static int	init_compute_pipeline (FFVulkanContext s, FFVulkanShader shd, VkShaderModule mod, const char *entrypoint)

static int	create_shader_object (FFVulkanContext s, FFVulkanShader shd, uint8_t spirv, size_t spirv_len, const char entrypoint)

static int	init_descriptors (FFVulkanContext s, FFVulkanShader shd)

int	ff_vk_shader_link (FFVulkanContext s, FFVulkanShader shd, uint8_t spirv, size_t spirv_len, const char entrypoint)
	Link a shader into an executable. More...

int	ff_vk_shader_add_descriptor_set (FFVulkanContext s, FFVulkanShader shd, FFVulkanDescriptorSetBinding *desc, int nb, int singular, int print_to_shader_only)
	Add descriptor to a shader. More...

int	ff_vk_shader_register_exec (FFVulkanContext s, FFVkExecPool pool, FFVulkanShader *shd)
	Register a shader with an exec pool. More...

static FFVulkanShaderData *	get_shd_data (FFVkExecContext e, FFVulkanShader shd)

static void	update_set_descriptor (FFVulkanContext s, FFVkExecContext e, FFVulkanShader shd, int set, int bind_idx, int array_idx, VkDescriptorGetInfoEXT desc_get_info, size_t desc_size)

static void	update_set_pool_write (FFVulkanContext s, FFVkExecContext e, FFVulkanShader shd, int set, VkWriteDescriptorSet write_info)

static int	vk_set_descriptor_image (FFVulkanContext s, FFVulkanShader shd, FFVkExecContext *e, int set, int bind, int offs, VkImageView view, VkImageLayout layout, VkSampler sampler)

int	ff_vk_shader_update_desc_buffer (FFVulkanContext s, FFVkExecContext e, FFVulkanShader shd, int set, int bind, int elem, FFVkBuffer buf, VkDeviceSize offset, VkDeviceSize len, VkFormat fmt)
	Update a descriptor in a buffer with a buffer. More...

void	ff_vk_shader_update_img_array (FFVulkanContext s, FFVkExecContext e, FFVulkanShader shd, AVFrame f, VkImageView *views, int set, int binding, VkImageLayout layout, VkSampler sampler)
	Update a descriptor in a buffer with an image array. More...

void	ff_vk_shader_update_push_const (FFVulkanContext s, FFVkExecContext e, FFVulkanShader shd, VkShaderStageFlagBits stage, int offset, size_t size, void src)
	Update push constant in a shader. More...

void	ff_vk_exec_bind_shader (FFVulkanContext s, FFVkExecContext e, FFVulkanShader *shd)
	Bind a shader. More...

void	ff_vk_shader_free (FFVulkanContext s, FFVulkanShader shd)
	Free a shader. More...

void	ff_vk_uninit (FFVulkanContext *s)
	Frees main context. More...

int	ff_vk_init (FFVulkanContext s, void log_parent, AVBufferRef device_ref, AVBufferRef frames_ref)
	Initializes the AVClass, in case this context is not used as the main user's context. More...

Variables
const VkComponentMapping	ff_comp_identity_map

static const struct descriptor_props	descriptor_props []

Macro Definition Documentation

◆ CASE

#define CASE ( VAL ) case VAL: return #VAL

◆ ARR_REALLOC

#define ARR_REALLOC	(	str,
		arr,
		alloc_s,
		cnt
	)

Value:

    do {                                                                  \
        arr = av_fast_realloc(str->arr, alloc_s, (cnt + 1)*sizeof(*arr)); \
        if (!arr) {                                                       \
            ff_vk_exec_discard_deps(s, e);                                \
            return AVERROR(ENOMEM);                                       \
        }                                                                 \
        str->arr = arr;                                                   \
    } while (0)

Definition at line 594 of file vulkan.c.

◆ REPS_FMT

#define REPS_FMT ( fmt )

Value:

    [FF_VK_REP_NATIVE] = fmt ## _UINT, \
    [FF_VK_REP_FLOAT]  = fmt ## _UNORM, \
    [FF_VK_REP_INT]    = fmt ## _SINT, \
    [FF_VK_REP_UINT]   = fmt ## _UINT,

◆ REPS_FMT_PACK

#define REPS_FMT_PACK	(	fmt,
		num
	)

Value:

    [FF_VK_REP_NATIVE] = fmt ## _UINT_PACK ## num, \
    [FF_VK_REP_FLOAT]  = fmt ## _UNORM_PACK ## num, \
    [FF_VK_REP_INT]    = fmt ## _SINT_PACK ## num, \
    [FF_VK_REP_UINT]   = fmt ## _UINT_PACK ## num,

Function Documentation

◆ ff_vk_ret2str()

const char* ff_vk_ret2str ( VkResult res )

◆ load_enabled_qfs()

static void load_enabled_qfs ( FFVulkanContext * s )

static

Definition at line 87 of file vulkan.c.

Referenced by ff_vk_load_props(), and ff_vk_qf_init().

◆ ff_vk_load_props()

int ff_vk_load_props ( FFVulkanContext * s )

Loads props/mprops/driver_props.

Definition at line 106 of file vulkan.c.

Referenced by ff_vk_filter_init_context(), ff_vk_init(), ff_vulkan_encode_init(), and vulkan_device_init().

◆ vk_qf_get_index()

static int vk_qf_get_index	(	FFVulkanContext *	s,
		VkQueueFlagBits	dev_family,
		int *	nb
	)

static

Definition at line 216 of file vulkan.c.

Referenced by ff_vk_qf_init().

◆ ff_vk_qf_init()

int ff_vk_qf_init	(	FFVulkanContext *	s,
		FFVkQueueFamilyCtx *	qf,
		VkQueueFlagBits	dev_family
	)

Chooses a QF and loads it into a context.

Definition at line 228 of file vulkan.c.

Referenced by init_filter(), init_vulkan(), and vulkan_device_init().

◆ ff_vk_exec_pool_free()

void ff_vk_exec_pool_free	(	FFVulkanContext *	s,
		FFVkExecPool *	pool
	)

Definition at line 238 of file vulkan.c.

Referenced by avgblur_vulkan_uninit(), bwdif_vulkan_uninit(), chromaber_vulkan_uninit(), ff_vk_decode_uninit(), ff_vk_exec_pool_init(), ff_vulkan_encode_uninit(), flip_vulkan_uninit(), gblur_vulkan_uninit(), overlay_vulkan_uninit(), scale_vulkan_uninit(), testsrc_vulkan_uninit(), transpose_vulkan_uninit(), uninit(), and vulkan_frames_uninit().

◆ ff_vk_exec_pool_init()

int ff_vk_exec_pool_init	(	FFVulkanContext *	s,
		FFVkQueueFamilyCtx *	qf,
		FFVkExecPool *	pool,
		int	nb_contexts,
		int	nb_queries,
		VkQueryType	query_type,
		int	query_64bit,
		const void *	query_create_pnext
	)

Allocates/frees an execution pool.

ff_vk_exec_pool_init_desc() MUST be called if ff_vk_exec_descriptor_set_add() has been called.

Definition at line 300 of file vulkan.c.

Referenced by ff_vk_decode_init(), ff_vk_update_thread_context(), ff_vulkan_encode_init(), init_filter(), init_vulkan(), and vulkan_frames_init().

◆ ff_vk_exec_get_query()

VkResult ff_vk_exec_get_query	(	FFVulkanContext *	s,
		FFVkExecContext *	e,
		void **	data,
		VkQueryResultFlagBits	flags
	)

Performs nb_queries queries and returns their results and statuses.

64_BIT and WITH_STATUS flags are ignored as 64_BIT must be specified via query_64bit in ff_vk_exec_pool_init() and WITH_STATUS is always enabled.

Definition at line 457 of file vulkan.c.

Referenced by ff_vk_decode_frame(), and vulkan_encode_output().

◆ ff_vk_exec_get()

FFVkExecContext* ff_vk_exec_get	(	FFVulkanContext *	s,
		FFVkExecPool *	pool
	)

Retrieve an execution pool.

Threadsafe.

Definition at line 485 of file vulkan.c.

Referenced by ff_vk_decode_flush(), ff_vk_decode_frame(), ff_vk_filter_process_2pass(), ff_vk_filter_process_Nin(), ff_vk_filter_process_simple(), prepare_frame(), vulkan_encode_issue(), and vulkan_transfer_frame().

◆ ff_vk_exec_wait()

void ff_vk_exec_wait	(	FFVulkanContext *	s,
		FFVkExecContext *	e
	)

Definition at line 499 of file vulkan.c.

Referenced by vulkan_encode_wait(), and vulkan_transfer_frame().

◆ ff_vk_exec_start()

int ff_vk_exec_start	(	FFVulkanContext *	s,
		FFVkExecContext *	e
	)

Start/submit/wait an execution.

ff_vk_exec_start() always waits on a submission, so using ff_vk_exec_wait() is not necessary (unless using it is just better).

Definition at line 508 of file vulkan.c.

Referenced by ff_vk_decode_flush(), ff_vk_decode_frame(), ff_vk_filter_process_2pass(), ff_vk_filter_process_Nin(), ff_vk_filter_process_simple(), prepare_frame(), vulkan_encode_issue(), and vulkan_transfer_frame().

◆ ff_vk_exec_discard_deps()

void ff_vk_exec_discard_deps	(	FFVulkanContext *	s,
		FFVkExecContext *	e
	)

Definition at line 544 of file vulkan.c.

Referenced by ff_vk_exec_add_dep_buf(), ff_vk_exec_add_dep_frame(), ff_vk_exec_pool_free(), ff_vk_exec_start(), ff_vk_exec_submit(), ff_vk_exec_wait(), ff_vk_filter_process_2pass(), ff_vk_filter_process_Nin(), ff_vk_filter_process_simple(), prepare_frame(), and vulkan_transfer_frame().

◆ ff_vk_exec_add_dep_buf()

int ff_vk_exec_add_dep_buf	(	FFVulkanContext *	s,
		FFVkExecContext *	e,
		AVBufferRef **	deps,
		int	nb_deps,
		int	ref
	)

Execution dependency management.

Can attach buffers to executions that will only be unref'd once the buffer has finished executing. Adding a frame dep will lock the frame, until either the dependencies are discarded, the execution is submitted, or a failure happens. update_frame will update the frame's properties before it is unlocked, only if submission was successful.

Definition at line 570 of file vulkan.c.

Referenced by ff_vk_create_imageviews(), ff_vk_decode_frame(), ff_vk_exec_add_dep_bool_sem(), vulkan_encode_issue(), and vulkan_transfer_frame().

◆ destroy_tmp_semaphores()

static void destroy_tmp_semaphores	(	void *	opaque,
		uint8_t *	data
	)

static

Definition at line 609 of file vulkan.c.

Referenced by ff_vk_exec_add_dep_bool_sem().

◆ ff_vk_exec_add_dep_bool_sem()

int ff_vk_exec_add_dep_bool_sem	(	FFVulkanContext *	s,
		FFVkExecContext *	e,
		VkSemaphore *	sem,
		int	nb,
		VkPipelineStageFlagBits2	stage,
		int	wait
	)

Definition at line 621 of file vulkan.c.

◆ ff_vk_exec_add_dep_frame()

int ff_vk_exec_add_dep_frame	(	FFVulkanContext *	s,
		FFVkExecContext *	e,
		AVFrame *	f,
		VkPipelineStageFlagBits2	wait_stage,
		VkPipelineStageFlagBits2	signal_stage
	)

Definition at line 692 of file vulkan.c.

Referenced by ff_vk_decode_frame(), ff_vk_filter_process_2pass(), ff_vk_filter_process_Nin(), ff_vk_filter_process_simple(), prepare_frame(), vulkan_encode_issue(), and vulkan_transfer_frame().

◆ ff_vk_exec_update_frame()

void ff_vk_exec_update_frame	(	FFVulkanContext *	s,
		FFVkExecContext *	e,
		AVFrame *	f,
		VkImageMemoryBarrier2 *	bar,
		uint32_t *	nb_img_bar
	)

Definition at line 762 of file vulkan.c.

Referenced by ff_vk_decode_frame(), ff_vk_frame_barrier(), and vulkan_encode_issue().

◆ ff_vk_exec_mirror_sem_value()

int ff_vk_exec_mirror_sem_value	(	FFVulkanContext *	s,
		FFVkExecContext *	e,
		VkSemaphore *	dst,
		uint64_t *	dst_val,
		AVFrame *	f
	)

Definition at line 781 of file vulkan.c.

Referenced by ff_vk_decode_frame().

◆ ff_vk_exec_submit()

int ff_vk_exec_submit	(	FFVulkanContext *	s,
		FFVkExecContext *	e
	)

Definition at line 807 of file vulkan.c.

Referenced by ff_vk_decode_flush(), ff_vk_decode_frame(), ff_vk_filter_process_2pass(), ff_vk_filter_process_Nin(), ff_vk_filter_process_simple(), prepare_frame(), vulkan_encode_issue(), and vulkan_transfer_frame().

◆ ff_vk_alloc_mem()

int ff_vk_alloc_mem	(	FFVulkanContext *	s,
		VkMemoryRequirements *	req,
		VkMemoryPropertyFlagBits	req_flags,
		void *	alloc_extension,
		VkMemoryPropertyFlagBits *	mem_flags,
		VkDeviceMemory *	mem
	)

Memory/buffer/image allocation helpers.

Definition at line 873 of file vulkan.c.

Referenced by create_mapped_buffer(), ff_vk_create_buf(), and ff_vk_video_common_init().

◆ ff_vk_create_buf()

int ff_vk_create_buf	(	FFVulkanContext *	s,
		FFVkBuffer *	buf,
		size_t	size,
		void *	pNext,
		void *	alloc_pNext,
		VkBufferUsageFlags	usage,
		VkMemoryPropertyFlagBits	flags
	)

Definition at line 924 of file vulkan.c.

Referenced by ff_vk_create_avbuf(), ff_vk_get_pooled_buffer(), ff_vk_shader_register_exec(), and init_gblur_pipeline().

◆ destroy_avvkbuf()

static void destroy_avvkbuf	(	void *	opaque,
		uint8_t *	data
	)

static

Definition at line 1016 of file vulkan.c.

Referenced by ff_vk_create_avbuf().

◆ ff_vk_create_avbuf()

int ff_vk_create_avbuf	(	FFVulkanContext *	s,
		AVBufferRef **	ref,
		size_t	size,
		void *	pNext,
		void *	alloc_pNext,
		VkBufferUsageFlags	usage,
		VkMemoryPropertyFlagBits	flags
	)

Definition at line 1024 of file vulkan.c.

◆ ff_vk_map_buffers()

int ff_vk_map_buffers	(	FFVulkanContext *	s,
		FFVkBuffer **	buf,
		uint8_t *	mem[],
		int	nb_buffers,
		int	invalidate
	)

Buffer management code.

Definition at line 1051 of file vulkan.c.

Referenced by ff_vk_map_buffer().

◆ ff_vk_unmap_buffers()

int ff_vk_unmap_buffers	(	FFVulkanContext *	s,
		FFVkBuffer **	buf,
		int	nb_buffers,
		int	flush
	)

Definition at line 1098 of file vulkan.c.

Referenced by ff_vk_unmap_buffer().

◆ ff_vk_free_buf()

void ff_vk_free_buf	(	FFVulkanContext *	s,
		FFVkBuffer *	buf
	)

Definition at line 1138 of file vulkan.c.

Referenced by destroy_avvkbuf(), ff_vk_exec_pool_free(), ff_vk_get_pooled_buffer(), free_data_buf(), and gblur_vulkan_uninit().

◆ free_data_buf()

static void free_data_buf	(	void *	opaque,
		uint8_t *	data
	)

static

Definition at line 1153 of file vulkan.c.

Referenced by alloc_data_buf().

◆ alloc_data_buf()

static AVBufferRef* alloc_data_buf	(	void *	opaque,
		size_t	size
	)

static

Definition at line 1161 of file vulkan.c.

Referenced by ff_vk_get_pooled_buffer().

◆ ff_vk_get_pooled_buffer()

int ff_vk_get_pooled_buffer	(	FFVulkanContext *	ctx,
		AVBufferPool **	buf_pool,
		AVBufferRef **	buf,
		VkBufferUsageFlags	usage,
		void *	create_pNext,
		size_t	size,
		VkMemoryPropertyFlagBits	mem_props
	)

Initialize a pool and create AVBufferRefs containing FFVkBuffer.

Threadsafe to use. Buffers are automatically mapped on creation if VK_MEMORY_PROPERTY_HOST_VISIBLE_BIT is set in mem_props. Users should synchronize access themselvesd. Mainly meant for device-local buffers.

Definition at line 1174 of file vulkan.c.

Referenced by ff_vk_decode_add_slice(), get_plane_buf(), and vulkan_encode_issue().

◆ ff_vk_shader_add_push_const()

int ff_vk_shader_add_push_const	(	FFVulkanShader *	shd,
		int	offset,
		int	size,
		VkShaderStageFlagBits	stage
	)

Add/update push constants for execution.

Definition at line 1223 of file vulkan.c.

Referenced by init_filter(), and init_vulkan().

◆ ff_vk_init_sampler()

int ff_vk_init_sampler	(	FFVulkanContext *	s,
		VkSampler *	sampler,
		int	unnorm_coords,
		VkFilter	filt
	)

Create a sampler.

Definition at line 1244 of file vulkan.c.

Referenced by init_filter(), and init_vulkan().

◆ ff_vk_mt_is_np_rgb()

int ff_vk_mt_is_np_rgb ( enum AVPixelFormat pix_fmt )

Returns 1 if pixfmt is a usable RGB format.

Definition at line 1276 of file vulkan.c.

Referenced by scale_vulkan_config_output().

◆ ff_vk_shader_rep_fmt()

const char* ff_vk_shader_rep_fmt	(	enum AVPixelFormat	pix_fmt,
		enum FFVkShaderRepFormat	rep_fmt
	)

Definition at line 1291 of file vulkan.c.

Referenced by init_filter(), and init_vulkan().

◆ destroy_imageviews()

static void destroy_imageviews	(	void *	opaque,
		uint8_t *	data
	)

static

Definition at line 1449 of file vulkan.c.

Referenced by ff_vk_create_imageviews().

◆ map_fmt_to_rep()

static VkFormat map_fmt_to_rep	(	VkFormat	fmt,
		enum FFVkShaderRepFormat	rep_fmt
	)

static

Definition at line 1461 of file vulkan.c.

Referenced by ff_vk_create_imageviews().

◆ ff_vk_create_imageviews()

int ff_vk_create_imageviews	(	FFVulkanContext *	s,
		FFVkExecContext *	e,
		VkImageView	views[AV_NUM_DATA_POINTERS],
		AVFrame *	f,
		enum FFVkShaderRepFormat	rep_fmt
	)

Create an imageview and add it as a dependency to an execution.

Definition at line 1548 of file vulkan.c.

Referenced by ff_vk_filter_process_2pass(), ff_vk_filter_process_Nin(), and ff_vk_filter_process_simple().

◆ ff_vk_frame_barrier()

void ff_vk_frame_barrier	(	FFVulkanContext *	s,
		FFVkExecContext *	e,
		AVFrame *	pic,
		VkImageMemoryBarrier2 *	bar,
		int *	nb_bar,
		VkPipelineStageFlags	src_stage,
		VkPipelineStageFlags	dst_stage,
		VkAccessFlagBits	new_access,
		VkImageLayout	new_layout,
		uint32_t	new_qf
	)

Definition at line 1637 of file vulkan.c.

Referenced by ff_vk_filter_process_2pass(), ff_vk_filter_process_Nin(), ff_vk_filter_process_simple(), prepare_frame(), and vulkan_transfer_frame().

◆ ff_vk_shader_init()

int ff_vk_shader_init	(	FFVulkanContext *	s,
		FFVulkanShader *	shd,
		const char *	name,
		VkPipelineStageFlags	stage,
		const char *	extensions[],
		int	nb_extensions,
		int	lg_x,
		int	lg_y,
		int	lg_z,
		uint32_t	required_subgroup_size
	)

Initialize a shader object, with a specific set of extensions, type+bind, local group size, and subgroup requirements.

Definition at line 1680 of file vulkan.c.

Referenced by init_filter(), and init_vulkan().

◆ ff_vk_shader_print()

void ff_vk_shader_print	(	void *	ctx,
		FFVulkanShader *	shd,
		int	prio
	)

Output the shader code as logging data, with a specific priority.

Definition at line 1755 of file vulkan.c.

Referenced by glslc_shader_compile(), and shdc_shader_compile().

◆ init_pipeline_layout()

static int init_pipeline_layout	(	FFVulkanContext *	s,
		FFVulkanShader *	shd
	)

static

Definition at line 1777 of file vulkan.c.

Referenced by ff_vk_shader_link().

◆ create_shader_module()

static int create_shader_module	(	FFVulkanContext *	s,
		FFVulkanShader *	shd,
		VkShaderModule *	mod,
		uint8_t *	spirv,
		size_t	spirv_len
	)

static

Definition at line 1803 of file vulkan.c.

Referenced by ff_vk_shader_link().

◆ init_compute_pipeline()

static int init_compute_pipeline	(	FFVulkanContext *	s,
		FFVulkanShader *	shd,
		VkShaderModule	mod,
		const char *	entrypoint
	)

static

Definition at line 1829 of file vulkan.c.

Referenced by ff_vk_shader_link().

◆ create_shader_object()

static int create_shader_object	(	FFVulkanContext *	s,
		FFVulkanShader *	shd,
		uint8_t *	spirv,
		size_t	spirv_len,
		const char *	entrypoint
	)

static

Definition at line 1864 of file vulkan.c.

Referenced by ff_vk_shader_link().

◆ init_descriptors()

static int init_descriptors	(	FFVulkanContext *	s,
		FFVulkanShader *	shd
	)

static

Definition at line 1905 of file vulkan.c.

Referenced by ff_vk_shader_link().

◆ ff_vk_shader_link()

int ff_vk_shader_link	(	FFVulkanContext *	s,
		FFVulkanShader *	shd,
		uint8_t *	spirv,
		size_t	spirv_len,
		const char *	entrypoint
	)

Link a shader into an executable.

Definition at line 1969 of file vulkan.c.

Referenced by init_filter(), init_gblur_pipeline(), and init_vulkan().

◆ ff_vk_shader_add_descriptor_set()

int ff_vk_shader_add_descriptor_set	(	FFVulkanContext *	s,
		FFVulkanShader *	shd,
		FFVulkanDescriptorSetBinding *	desc,
		int	nb,
		int	singular,
		int	print_to_shader_only
	)

Add descriptor to a shader.

Must be called before shader init.

Definition at line 2044 of file vulkan.c.

Referenced by init_filter(), init_gblur_pipeline(), and init_vulkan().

◆ ff_vk_shader_register_exec()

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

Register a shader with an exec pool.

Pool may be NULL if all descriptor sets are read-only.

Definition at line 2169 of file vulkan.c.

Referenced by init_filter(), init_gblur_pipeline(), and init_vulkan().

◆ get_shd_data()

static FFVulkanShaderData* get_shd_data	(	FFVkExecContext *	e,
		FFVulkanShader *	shd
	)

inlinestatic

Definition at line 2286 of file vulkan.c.

Referenced by ff_vk_exec_bind_shader(), update_set_descriptor(), and update_set_pool_write().

◆ update_set_descriptor()

static void update_set_descriptor	(	FFVulkanContext *	s,
		FFVkExecContext *	e,
		FFVulkanShader *	shd,
		int	set,
		int	bind_idx,
		int	array_idx,
		VkDescriptorGetInfoEXT *	desc_get_info,
		size_t	desc_size
	)

inlinestatic

Definition at line 2295 of file vulkan.c.

Referenced by ff_vk_shader_update_desc_buffer(), and vk_set_descriptor_image().

◆ update_set_pool_write()

static void update_set_pool_write	(	FFVulkanContext *	s,
		FFVkExecContext *	e,
		FFVulkanShader *	shd,
		int	set,
		VkWriteDescriptorSet *	write_info
	)

inlinestatic

Definition at line 2314 of file vulkan.c.

Referenced by ff_vk_shader_update_desc_buffer(), and vk_set_descriptor_image().

◆ vk_set_descriptor_image()

static int vk_set_descriptor_image	(	FFVulkanContext *	s,
		FFVulkanShader *	shd,
		FFVkExecContext *	e,
		int	set,
		int	bind,
		int	offs,
		VkImageView	view,
		VkImageLayout	layout,
		VkSampler	sampler
	)

static

Definition at line 2341 of file vulkan.c.

Referenced by ff_vk_shader_update_img_array().

◆ ff_vk_shader_update_desc_buffer()

int ff_vk_shader_update_desc_buffer	(	FFVulkanContext *	s,
		FFVkExecContext *	e,
		FFVulkanShader *	shd,
		int	set,
		int	bind,
		int	elem,
		FFVkBuffer *	buf,
		VkDeviceSize	offset,
		VkDeviceSize	len,
		VkFormat	fmt
	)

Update a descriptor in a buffer with a buffer.

Must be called before binding the shader.

Definition at line 2406 of file vulkan.c.

Referenced by init_gblur_pipeline().

◆ ff_vk_shader_update_img_array()

void ff_vk_shader_update_img_array	(	FFVulkanContext *	s,
		FFVkExecContext *	e,
		FFVulkanShader *	shd,
		AVFrame *	f,
		VkImageView *	views,
		int	set,
		int	binding,
		VkImageLayout	layout,
		VkSampler	sampler
	)

Update a descriptor in a buffer with an image array.

Must be called before binding the shader.

Definition at line 2472 of file vulkan.c.

Referenced by ff_vk_filter_process_2pass(), ff_vk_filter_process_Nin(), and ff_vk_filter_process_simple().

◆ ff_vk_shader_update_push_const()

void ff_vk_shader_update_push_const	(	FFVulkanContext *	s,
		FFVkExecContext *	e,
		FFVulkanShader *	shd,
		VkShaderStageFlagBits	stage,
		int	offset,
		size_t	size,
		void *	src
	)

Update push constant in a shader.

Must be called before binding the shader.

Definition at line 2485 of file vulkan.c.

Referenced by ff_vk_filter_process_2pass(), ff_vk_filter_process_Nin(), and ff_vk_filter_process_simple().

◆ ff_vk_exec_bind_shader()

void ff_vk_exec_bind_shader	(	FFVulkanContext *	s,
		FFVkExecContext *	e,
		FFVulkanShader *	shd
	)

Bind a shader.

Definition at line 2495 of file vulkan.c.

Referenced by ff_vk_filter_process_2pass(), ff_vk_filter_process_Nin(), and ff_vk_filter_process_simple().

◆ ff_vk_shader_free()

void ff_vk_shader_free	(	FFVulkanContext *	s,
		FFVulkanShader *	shd
	)

Free a shader.

Definition at line 2529 of file vulkan.c.

Referenced by avgblur_vulkan_uninit(), bwdif_vulkan_uninit(), chromaber_vulkan_uninit(), flip_vulkan_uninit(), gblur_vulkan_uninit(), overlay_vulkan_uninit(), scale_vulkan_uninit(), testsrc_vulkan_uninit(), transpose_vulkan_uninit(), and uninit().

◆ ff_vk_uninit()

void ff_vk_uninit ( FFVulkanContext * s )

Frees main context.

Definition at line 2568 of file vulkan.c.

Referenced by avgblur_vulkan_uninit(), bwdif_vulkan_uninit(), chromaber_vulkan_uninit(), ff_vk_init(), ff_vulkan_encode_uninit(), flip_vulkan_uninit(), free_common(), gblur_vulkan_uninit(), libplacebo_uninit(), overlay_vulkan_uninit(), scale_vulkan_uninit(), testsrc_vulkan_uninit(), transpose_vulkan_uninit(), uninit(), and vulkan_device_uninit().

◆ ff_vk_init()

int ff_vk_init	(	FFVulkanContext *	s,
		void *	log_parent,
		AVBufferRef *	device_ref,
		AVBufferRef *	frames_ref
	)

Initializes the AVClass, in case this context is not used as the main user's context.

May use either a frames context reference, or a device context reference.

Definition at line 2579 of file vulkan.c.

Referenced by ff_vk_decode_init().

Variable Documentation

◆ ff_comp_identity_map

const VkComponentMapping ff_comp_identity_map

Initial value:

= {
    .r = VK_COMPONENT_SWIZZLE_IDENTITY,
    .g = VK_COMPONENT_SWIZZLE_IDENTITY,
    .b = VK_COMPONENT_SWIZZLE_IDENTITY,
    .a = VK_COMPONENT_SWIZZLE_IDENTITY,
}

Definition at line 27 of file vulkan.c.

Referenced by ff_vk_create_imageviews(), and ff_vk_video_common_init().

◆ descriptor_props

const struct descriptor_props descriptor_props[]

static

Initial value:

= {
    [VK_DESCRIPTOR_TYPE_SAMPLER]                = { sizeof(VkDescriptorImageInfo),  "sampler",       1, 0, 0, 0, },
    [VK_DESCRIPTOR_TYPE_SAMPLED_IMAGE]          = { sizeof(VkDescriptorImageInfo),  "texture",       1, 0, 1, 0, },
    [VK_DESCRIPTOR_TYPE_STORAGE_IMAGE]          = { sizeof(VkDescriptorImageInfo),  "image",         1, 1, 1, 0, },
    [VK_DESCRIPTOR_TYPE_INPUT_ATTACHMENT]       = { sizeof(VkDescriptorImageInfo),  "subpassInput",  1, 0, 0, 0, },
    [VK_DESCRIPTOR_TYPE_COMBINED_IMAGE_SAMPLER] = { sizeof(VkDescriptorImageInfo),  "sampler",       1, 0, 1, 0, },
    [VK_DESCRIPTOR_TYPE_UNIFORM_BUFFER]         = { sizeof(VkDescriptorBufferInfo),  NULL,           1, 0, 0, 1, },
    [VK_DESCRIPTOR_TYPE_STORAGE_BUFFER]         = { sizeof(VkDescriptorBufferInfo), "buffer",        0, 1, 0, 1, },
    [VK_DESCRIPTOR_TYPE_UNIFORM_BUFFER_DYNAMIC] = { sizeof(VkDescriptorBufferInfo),  NULL,           1, 0, 0, 1, },
    [VK_DESCRIPTOR_TYPE_STORAGE_BUFFER_DYNAMIC] = { sizeof(VkDescriptorBufferInfo), "buffer",        0, 1, 0, 1, },
    [VK_DESCRIPTOR_TYPE_UNIFORM_TEXEL_BUFFER]   = { sizeof(VkBufferView),           "samplerBuffer", 1, 0, 0, 0, },
    [VK_DESCRIPTOR_TYPE_STORAGE_TEXEL_BUFFER]   = { sizeof(VkBufferView),           "imageBuffer",   1, 0, 0, 0, },
}

Data Structures

Macros

Functions

Variables

Macro Definition Documentation

◆ CASE

◆ ARR_REALLOC

◆ REPS_FMT

◆ REPS_FMT_PACK

Function Documentation

◆ ff_vk_ret2str()

◆ load_enabled_qfs()

◆ ff_vk_load_props()

◆ vk_qf_get_index()

◆ ff_vk_qf_init()

◆ ff_vk_exec_pool_free()

◆ ff_vk_exec_pool_init()

◆ ff_vk_exec_get_query()

◆ ff_vk_exec_get()

◆ ff_vk_exec_wait()

◆ ff_vk_exec_start()

◆ ff_vk_exec_discard_deps()

◆ ff_vk_exec_add_dep_buf()

◆ destroy_tmp_semaphores()

◆ ff_vk_exec_add_dep_bool_sem()

◆ ff_vk_exec_add_dep_frame()

◆ ff_vk_exec_update_frame()

◆ ff_vk_exec_mirror_sem_value()

◆ ff_vk_exec_submit()

◆ ff_vk_alloc_mem()

◆ ff_vk_create_buf()

◆ destroy_avvkbuf()

◆ ff_vk_create_avbuf()

◆ ff_vk_map_buffers()

◆ ff_vk_unmap_buffers()

◆ ff_vk_free_buf()

◆ free_data_buf()

◆ alloc_data_buf()

◆ ff_vk_get_pooled_buffer()

◆ ff_vk_shader_add_push_const()

◆ ff_vk_init_sampler()

◆ ff_vk_mt_is_np_rgb()

◆ ff_vk_shader_rep_fmt()

◆ destroy_imageviews()

◆ map_fmt_to_rep()

◆ ff_vk_create_imageviews()

◆ ff_vk_frame_barrier()

◆ ff_vk_shader_init()

◆ ff_vk_shader_print()

◆ init_pipeline_layout()

◆ create_shader_module()

◆ init_compute_pipeline()

◆ create_shader_object()

◆ init_descriptors()

◆ ff_vk_shader_link()

◆ ff_vk_shader_add_descriptor_set()

◆ ff_vk_shader_register_exec()

◆ get_shd_data()

◆ update_set_descriptor()

◆ update_set_pool_write()

◆ vk_set_descriptor_image()

◆ ff_vk_shader_update_desc_buffer()

◆ ff_vk_shader_update_img_array()

◆ ff_vk_shader_update_push_const()

◆ ff_vk_exec_bind_shader()

◆ ff_vk_shader_free()

◆ ff_vk_uninit()

◆ ff_vk_init()

Variable Documentation

◆ ff_comp_identity_map

◆ descriptor_props