#include <math.h>
#include <stdlib.h>
#include "libavutil/mem.h"
#include "libavutil/opt.h"
#include "libavutil/pixdesc.h"
#include "libavutil/vulkan.h"
#include "libavutil/vulkan_spirv.h"
#include "avcodec.h"
#include "codec_internal.h"
#include "encode.h"
#include "hwconfig.h"
#include "internal.h"
#include "apv.h"
#include "cbs.h"
#include "cbs_apv.h"

Data Structures
struct	DCTPushData

struct	EntropyPushData

struct	CompactPushData

struct	VulkanEncodeAPVFrameData

struct	VulkanEncodeAPVContext

Macros
#define	APV_DEFAULT_QMAT 16

#define	APV_MAX_NUM_COMP 4

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

#define	VE AV_OPT_FLAG_VIDEO_PARAM \| AV_OPT_FLAG_ENCODING_PARAM

Enumerations
enum	{ APV_QMATRIX_FLAT = 0, APV_QMATRIX_HEVC = 1 }

Functions
static int	apv_qmatrix_value (int qmatrix, int i)

static int	chroma_format_from_pix_fmt (enum AVPixelFormat sw_fmt)

static int	profile_idc_from_pix_fmt (enum AVPixelFormat sw_fmt)

static int	init_dct_shader (AVCodecContext *avctx)

static int	init_entropy_shader (AVCodecContext avctx, int blocks_per_mb, FFVulkanShader shd)

static int	init_compact_shader (AVCodecContext *avctx)

static void	build_dct_push_const (AVCodecContext *avctx)

static int	submit_frame (AVCodecContext avctx, FFVkExecContext exec, AVFrame *frame)

static int	build_packet (AVCodecContext avctx, FFVkExecContext exec, AVPacket *pkt)

static int	vulkan_encode_apv_receive_packet (AVCodecContext avctx, AVPacket pkt)

static av_cold int	vulkan_encode_apv_close (AVCodecContext *avctx)

static av_cold int	vulkan_encode_apv_init (AVCodecContext *avctx)

Variables
const unsigned char	ff_apv_encode_dct_comp_spv_data []

const unsigned int	ff_apv_encode_dct_comp_spv_len

const unsigned char	ff_apv_encode_tiles_comp_spv_data []

const unsigned int	ff_apv_encode_tiles_comp_spv_len

const unsigned char	ff_seg_gather_comp_spv_data []

const unsigned int	ff_seg_gather_comp_spv_len

static const uint8_t	apv_qmat_hevc_intra [64]

static const uint8_t	apv_level_scale [6] = { 40, 45, 51, 57, 64, 71 }

static const AVOption	vulkan_encode_apv_options []

static const FFCodecDefault	vulkan_encode_apv_defaults []

static const AVClass	vulkan_encode_apv_class

static const AVCodecHWConfigInternal *const	vulkan_encode_apv_hw_configs []

const FFCodec	ff_apv_vulkan_encoder

Macro Definition Documentation

◆ APV_DEFAULT_QMAT

#define APV_DEFAULT_QMAT 16

Definition at line 49 of file apv_encode_vulkan.c.

◆ APV_MAX_NUM_COMP

#define APV_MAX_NUM_COMP 4

Definition at line 50 of file apv_encode_vulkan.c.

◆ OFFSET

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

Definition at line 1041 of file apv_encode_vulkan.c.

◆ VE

#define VE AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_ENCODING_PARAM

Definition at line 1042 of file apv_encode_vulkan.c.

Enumeration Type Documentation

◆ anonymous enum

anonymous enum

Enumerator
APV_QMATRIX_FLAT
APV_QMATRIX_HEVC

Definition at line 180 of file apv_encode_vulkan.c.

Function Documentation

◆ apv_qmatrix_value()

static int apv_qmatrix_value	(	int	qmatrix,
		int	i
	)

static

Definition at line 191 of file apv_encode_vulkan.c.

Referenced by build_dct_push_const(), and build_packet().

◆ chroma_format_from_pix_fmt()

static int chroma_format_from_pix_fmt ( enum AVPixelFormat sw_fmt )

static

Definition at line 199 of file apv_encode_vulkan.c.

Referenced by vulkan_encode_apv_init().

◆ profile_idc_from_pix_fmt()

static int profile_idc_from_pix_fmt ( enum AVPixelFormat sw_fmt )

static

Definition at line 219 of file apv_encode_vulkan.c.

Referenced by vulkan_encode_apv_init().

◆ init_dct_shader()

static int init_dct_shader ( AVCodecContext * avctx )

static

Definition at line 233 of file apv_encode_vulkan.c.

Referenced by vulkan_encode_apv_init().

◆ init_entropy_shader()

static int init_entropy_shader	(	AVCodecContext *	avctx,
		int	blocks_per_mb,
		FFVulkanShader *	shd
	)

static

Definition at line 273 of file apv_encode_vulkan.c.

Referenced by vulkan_encode_apv_init().

◆ init_compact_shader()

static int init_compact_shader ( AVCodecContext * avctx )

static

Definition at line 314 of file apv_encode_vulkan.c.

Referenced by vulkan_encode_apv_init().

◆ build_dct_push_const()

static void build_dct_push_const ( AVCodecContext * avctx )

static

Definition at line 350 of file apv_encode_vulkan.c.

Referenced by vulkan_encode_apv_init().

◆ submit_frame()

static int submit_frame	(	AVCodecContext *	avctx,
		FFVkExecContext *	exec,
		AVFrame *	frame
	)

static

Definition at line 391 of file apv_encode_vulkan.c.

Referenced by vulkan_encode_apv_receive_packet().

◆ build_packet()

static int build_packet	(	AVCodecContext *	avctx,
		FFVkExecContext *	exec,
		AVPacket *	pkt
	)

static

Definition at line 601 of file apv_encode_vulkan.c.

Referenced by vulkan_encode_apv_receive_packet().

◆ vulkan_encode_apv_receive_packet()

static int vulkan_encode_apv_receive_packet	(	AVCodecContext *	avctx,
		AVPacket *	pkt
	)

static

Definition at line 757 of file apv_encode_vulkan.c.

◆ vulkan_encode_apv_close()

static av_cold int vulkan_encode_apv_close ( AVCodecContext * avctx )

static

Definition at line 807 of file apv_encode_vulkan.c.

◆ vulkan_encode_apv_init()

static av_cold int vulkan_encode_apv_init ( AVCodecContext * avctx )

static

Definition at line 844 of file apv_encode_vulkan.c.

Variable Documentation

◆ ff_apv_encode_dct_comp_spv_data

const unsigned char ff_apv_encode_dct_comp_spv_data[]

Referenced by init_dct_shader().

◆ ff_apv_encode_dct_comp_spv_len

const unsigned int ff_apv_encode_dct_comp_spv_len

Referenced by init_dct_shader().

◆ ff_apv_encode_tiles_comp_spv_data

const unsigned char ff_apv_encode_tiles_comp_spv_data[]

Referenced by init_entropy_shader().

◆ ff_apv_encode_tiles_comp_spv_len

const unsigned int ff_apv_encode_tiles_comp_spv_len

Referenced by init_entropy_shader().

◆ ff_seg_gather_comp_spv_data

const unsigned char ff_seg_gather_comp_spv_data[]

Referenced by init_compact_shader().

◆ ff_seg_gather_comp_spv_len

const unsigned int ff_seg_gather_comp_spv_len

Referenced by init_compact_shader().

◆ apv_qmat_hevc_intra

const uint8_t apv_qmat_hevc_intra[64]

static

Initial value:

= {
    16, 16, 16, 16, 17, 18, 21, 24,
    16, 16, 16, 16, 17, 19, 22, 25,
    16, 16, 17, 18, 20, 22, 25, 29,
    16, 16, 18, 21, 24, 27, 31, 36,
    17, 17, 20, 24, 30, 35, 41, 47,
    18, 19, 22, 27, 35, 44, 54, 65,
    21, 22, 25, 31, 41, 54, 70, 88,
    24, 25, 29, 36, 47, 65, 88, 115,
}

Definition at line 169 of file apv_encode_vulkan.c.

Referenced by apv_qmatrix_value().

◆ apv_level_scale

const uint8_t apv_level_scale[6] = { 40, 45, 51, 57, 64, 71 }

static

Definition at line 197 of file apv_encode_vulkan.c.

Referenced by apv_decode_tile_component(), and build_dct_push_const().

◆ vulkan_encode_apv_options

const AVOption vulkan_encode_apv_options[]

static

Initial value:

= {
    { "qp",          "Quantization parameter (luma)", OFFSET(qp_y),
        AV_OPT_TYPE_INT, { .i64 = 22 }, 0, 255, VE },
    { "qp_chroma",   "Chroma quantization parameter (0 = same as luma qp)", OFFSET(qp_c),
        AV_OPT_TYPE_INT, { .i64 = 0 }, 0, 255, VE },
    { "qmatrix",     "Quantization matrix", OFFSET(qmatrix),
        AV_OPT_TYPE_INT, { .i64 = APV_QMATRIX_HEVC }, 0, 1, VE, "qmatrix" },
        { "flat",    "Uniform matrix, all 16 (APV spec default)", 0,
            AV_OPT_TYPE_CONST, { .i64 = APV_QMATRIX_FLAT }, 0, 0, VE, "qmatrix" },
        { "hevc",    "HEVC default intra scaling list (mild perceptual shaping)", 0,
            AV_OPT_TYPE_CONST, { .i64 = APV_QMATRIX_HEVC }, 0, 0, VE, "qmatrix" },
    
    { "tile_width",  "Tile width in macroblocks (0 = adaptive, auto-sized per frame)", OFFSET(tile_w_mbs_opt),
        AV_OPT_TYPE_INT, { .i64 = 0 }, 0, 32, VE },
    { "tile_height", "Tile height in macroblocks (0 = adaptive, auto-sized per frame)", OFFSET(tile_h_mbs_opt),
        AV_OPT_TYPE_INT, { .i64 = 0 }, 0, 16, VE },
    { "async_depth", "Internal parallelization depth", OFFSET(async_depth),
        AV_OPT_TYPE_INT, { .i64 = 1 }, 1, INT_MAX, VE },
    { NULL }
}

Definition at line 1043 of file apv_encode_vulkan.c.

◆ vulkan_encode_apv_defaults

const FFCodecDefault vulkan_encode_apv_defaults[]

static

Initial value:

= {
    { "g", "1" },
    { NULL },
}

Definition at line 1067 of file apv_encode_vulkan.c.

◆ vulkan_encode_apv_class

const AVClass vulkan_encode_apv_class

static

Initial value:

= {
    .class_name = "apv_vulkan",
    .item_name  = av_default_item_name,
    .option     = vulkan_encode_apv_options,
    .version    = LIBAVUTIL_VERSION_INT,
}

Definition at line 1072 of file apv_encode_vulkan.c.

◆ vulkan_encode_apv_hw_configs

const AVCodecHWConfigInternal* const vulkan_encode_apv_hw_configs[]

static

Initial value:

= {
    HW_CONFIG_ENCODER_FRAMES(VULKAN, VULKAN),
    NULL,
}

Definition at line 1079 of file apv_encode_vulkan.c.

◆ ff_apv_vulkan_encoder

const FFCodec ff_apv_vulkan_encoder

Initial value:

= {
    .p.name         = "apv_vulkan",
    CODEC_LONG_NAME("Advanced Professional Video (Vulkan)"),
    .p.type         = AVMEDIA_TYPE_VIDEO,
    .p.id           = AV_CODEC_ID_APV,
    .priv_data_size = sizeof(VulkanEncodeAPVContext),
    .init           = &vulkan_encode_apv_init,
    FF_CODEC_RECEIVE_PACKET_CB(&vulkan_encode_apv_receive_packet),
    .close          = &vulkan_encode_apv_close,
    .p.priv_class   = &vulkan_encode_apv_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 | FF_CODEC_CAP_EOF_FLUSH,
    .defaults       = vulkan_encode_apv_defaults,
    CODEC_PIXFMTS(AV_PIX_FMT_VULKAN),
    .hw_configs     = vulkan_encode_apv_hw_configs,
    .p.wrapper_name = "vulkan",
}

Definition at line 1084 of file apv_encode_vulkan.c.

Data Structures

Macros

Enumerations

Functions

Variables

Macro Definition Documentation

◆ APV_DEFAULT_QMAT

◆ APV_MAX_NUM_COMP

◆ OFFSET

◆ VE

Enumeration Type Documentation

◆ anonymous enum

Function Documentation

◆ apv_qmatrix_value()

◆ chroma_format_from_pix_fmt()

◆ profile_idc_from_pix_fmt()

◆ init_dct_shader()

◆ init_entropy_shader()

◆ init_compact_shader()

◆ build_dct_push_const()

◆ submit_frame()

◆ build_packet()

◆ vulkan_encode_apv_receive_packet()

◆ vulkan_encode_apv_close()

◆ vulkan_encode_apv_init()

Variable Documentation

◆ ff_apv_encode_dct_comp_spv_data

◆ ff_apv_encode_dct_comp_spv_len

◆ ff_apv_encode_tiles_comp_spv_data

◆ ff_apv_encode_tiles_comp_spv_len

◆ ff_seg_gather_comp_spv_data

◆ ff_seg_gather_comp_spv_len

◆ apv_qmat_hevc_intra

◆ apv_level_scale

◆ vulkan_encode_apv_options

◆ vulkan_encode_apv_defaults

◆ vulkan_encode_apv_class

◆ vulkan_encode_apv_hw_configs

◆ ff_apv_vulkan_encoder