proresenc_kostya_vulkan.c File Reference

#include "libavutil/avassert.h"
#include "libavutil/buffer.h"
#include "libavutil/macros.h"
#include "libavutil/mem.h"
#include "libavutil/mem_internal.h"
#include "libavutil/opt.h"
#include "libavutil/pixdesc.h"
#include "libavutil/hwcontext_vulkan.h"
#include "libavutil/vulkan_loader.h"
#include "libavutil/vulkan.h"
#include "avcodec.h"
#include "codec.h"
#include "codec_internal.h"
#include "encode.h"
#include "packet.h"
#include "put_bits.h"
#include "profiles.h"
#include "bytestream.h"
#include "proresdata.h"
#include "proresenc_kostya_common.h"
#include "hwconfig.h"
#include "vulkan_video.h"

Go to the source code of this file.

Data Structures
struct	ProresDataTables
struct	SliceDataInfo
struct	EncodeSliceInfo
struct	SliceData
struct	SliceScore
struct	VulkanEncodeProresFrameData
struct	ProresVulkanContext

Macros
#define	DCTSIZE   8
#define	OFFSET(x)
#define	VE   AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_ENCODING_PARAM

Functions
static int	init_slice_data_pipeline (ProresVulkanContext *pv, FFVulkanShader *shd, int blocks_per_mb)
static int	init_alpha_data_pipeline (ProresVulkanContext *pv, FFVulkanShader *shd)
static int	init_estimate_slice_pipeline (ProresVulkanContext *pv, FFVulkanShader *shd)
static int	init_trellis_node_pipeline (ProresVulkanContext *pv, FFVulkanShader *shd)
static int	init_encode_slice_pipeline (ProresVulkanContext *pv, FFVulkanShader *shd)
static int	encode_picture (AVCodecContext *avctx, FFVkExecContext *exec, AVFrame *frame, int picture_idx)
static uint8_t *	write_frame_header (AVCodecContext *avctx, ProresContext *ctx, uint8_t **orig_buf, int flags, enum AVColorPrimaries color_primaries, enum AVColorTransferCharacteristic color_trc, enum AVColorSpace colorspace)
static void	prores_vk_packet_free (void *opaque, uint8_t *data)
static int	get_packet (AVCodecContext *avctx, FFVkExecContext *exec, AVPacket *pkt)
static int	vulkan_encode_prores_submit_frame (AVCodecContext *avctx, FFVkExecContext *exec, AVFrame *frame)
static int	vulkan_encode_prores_receive_packet (AVCodecContext *avctx, AVPacket *pkt)
static av_cold void	vulkan_encode_prores_flush (AVCodecContext *avctx)
static av_cold int	encode_close (AVCodecContext *avctx)
static av_cold int	encode_init (AVCodecContext *avctx)

Variables
const unsigned char	ff_prores_ks_alpha_data_comp_spv_data []
const unsigned int	ff_prores_ks_alpha_data_comp_spv_len
const unsigned char	ff_prores_ks_slice_data_comp_spv_data []
const unsigned int	ff_prores_ks_slice_data_comp_spv_len
const unsigned char	ff_prores_ks_estimate_slice_comp_spv_data []
const unsigned int	ff_prores_ks_estimate_slice_comp_spv_len
const unsigned char	ff_prores_ks_trellis_node_comp_spv_data []
const unsigned int	ff_prores_ks_trellis_node_comp_spv_len
const unsigned char	ff_prores_ks_encode_slice_comp_spv_data []
const unsigned int	ff_prores_ks_encode_slice_comp_spv_len
static const AVOption	options []
static const AVClass	proresenc_class
static const AVCodecHWConfigInternal *const	prores_ks_hw_configs []
const FFCodec	ff_prores_ks_vulkan_encoder

Macro Definition Documentation

DCTSIZE

#define DCTSIZE   8

Definition at line 47 of file proresenc_kostya_vulkan.c.

OFFSET

#define OFFSET

(

x

)

Value:

offsetof(ProresVulkanContext, x)

Definition at line 913 of file proresenc_kostya_vulkan.c.

VE

#define VE   AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_ENCODING_PARAM

Definition at line 914 of file proresenc_kostya_vulkan.c.

Function Documentation

init_slice_data_pipeline()

static int init_slice_data_pipeline ( ProresVulkanContext * pv, FFVulkanShader * shd, int blocks_per_mb )

static

Definition at line 142 of file proresenc_kostya_vulkan.c.

Referenced by encode_init().

init_alpha_data_pipeline()

static int init_alpha_data_pipeline ( ProresVulkanContext * pv, FFVulkanShader * shd )

static

Definition at line 185 of file proresenc_kostya_vulkan.c.

Referenced by encode_init().

init_estimate_slice_pipeline()

static int init_estimate_slice_pipeline ( ProresVulkanContext * pv, FFVulkanShader * shd )

static

Definition at line 224 of file proresenc_kostya_vulkan.c.

Referenced by encode_init().

init_trellis_node_pipeline()

static int init_trellis_node_pipeline ( ProresVulkanContext * pv, FFVulkanShader * shd )

static

Definition at line 279 of file proresenc_kostya_vulkan.c.

Referenced by encode_init().

init_encode_slice_pipeline()

static int init_encode_slice_pipeline ( ProresVulkanContext * pv, FFVulkanShader * shd )

static

Definition at line 316 of file proresenc_kostya_vulkan.c.

Referenced by encode_init().

encode_picture()

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

static

Definition at line 365 of file proresenc_kostya_vulkan.c.

Referenced by vulkan_encode_prores_submit_frame().

write_frame_header()

static uint8_t * write_frame_header ( AVCodecContext * avctx, ProresContext * ctx, uint8_t ** orig_buf, int flags, enum AVColorPrimaries color_primaries, enum AVColorTransferCharacteristic color_trc, enum AVColorSpace colorspace )

static

Definition at line 596 of file proresenc_kostya_vulkan.c.

Referenced by encode_init(), and get_packet().

prores_vk_packet_free()

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

static

Definition at line 641 of file proresenc_kostya_vulkan.c.

Referenced by get_packet().

get_packet()

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

static

Definition at line 646 of file proresenc_kostya_vulkan.c.

vulkan_encode_prores_submit_frame()

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

static

Definition at line 742 of file proresenc_kostya_vulkan.c.

Referenced by encode_init().

vulkan_encode_prores_receive_packet()

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

static

Definition at line 769 of file proresenc_kostya_vulkan.c.

vulkan_encode_prores_flush()

static av_cold void vulkan_encode_prores_flush ( AVCodecContext * avctx )

static

Definition at line 775 of file proresenc_kostya_vulkan.c.

encode_close()

static av_cold int encode_close ( AVCodecContext * avctx )

static

Definition at line 781 of file proresenc_kostya_vulkan.c.

encode_init()

static av_cold int encode_init ( AVCodecContext * avctx )

static

Definition at line 827 of file proresenc_kostya_vulkan.c.

Variable Documentation

ff_prores_ks_alpha_data_comp_spv_data

const unsigned char ff_prores_ks_alpha_data_comp_spv_data[]

extern

Referenced by init_alpha_data_pipeline().

ff_prores_ks_alpha_data_comp_spv_len

const unsigned int ff_prores_ks_alpha_data_comp_spv_len

extern

Referenced by init_alpha_data_pipeline().

ff_prores_ks_slice_data_comp_spv_data

const unsigned char ff_prores_ks_slice_data_comp_spv_data[]

extern

Referenced by init_slice_data_pipeline().

ff_prores_ks_slice_data_comp_spv_len

const unsigned int ff_prores_ks_slice_data_comp_spv_len

extern

Referenced by init_slice_data_pipeline().

ff_prores_ks_estimate_slice_comp_spv_data

const unsigned char ff_prores_ks_estimate_slice_comp_spv_data[]

extern

Referenced by init_estimate_slice_pipeline().

ff_prores_ks_estimate_slice_comp_spv_len

const unsigned int ff_prores_ks_estimate_slice_comp_spv_len

extern

Referenced by init_estimate_slice_pipeline().

ff_prores_ks_trellis_node_comp_spv_data

const unsigned char ff_prores_ks_trellis_node_comp_spv_data[]

extern

Referenced by init_trellis_node_pipeline().

ff_prores_ks_trellis_node_comp_spv_len

const unsigned int ff_prores_ks_trellis_node_comp_spv_len

extern

Referenced by init_trellis_node_pipeline().

ff_prores_ks_encode_slice_comp_spv_data

const unsigned char ff_prores_ks_encode_slice_comp_spv_data[]

extern

Referenced by init_encode_slice_pipeline().

ff_prores_ks_encode_slice_comp_spv_len

const unsigned int ff_prores_ks_encode_slice_comp_spv_len

extern

Referenced by init_encode_slice_pipeline().

options

const AVOption options[]

static

Definition at line 916 of file proresenc_kostya_vulkan.c.

proresenc_class

const AVClass proresenc_class

static

Initial value:

= {
    .class_name = "ProRes vulkan encoder",
    .item_name  = av_default_item_name,
    .option     = options,
    .version    = LIBAVUTIL_VERSION_INT,
}

Definition at line 961 of file proresenc_kostya_vulkan.c.

prores_ks_hw_configs

const AVCodecHWConfigInternal* const prores_ks_hw_configs[]

static

Initial value:

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

Definition at line 968 of file proresenc_kostya_vulkan.c.

ff_prores_ks_vulkan_encoder

const FFCodec ff_prores_ks_vulkan_encoder

Initial value:

= {
    .p.name         = "prores_ks_vulkan",
    CODEC_LONG_NAME("Apple ProRes (iCodec Pro)"),
    .p.type         = AVMEDIA_TYPE_VIDEO,
    .p.id           = AV_CODEC_ID_PRORES,
    .priv_data_size = sizeof(ProresVulkanContext),
    .init           = encode_init,
    .close          = encode_close,
    FF_CODEC_RECEIVE_PACKET_CB(&vulkan_encode_prores_receive_packet),
    .flush          = &vulkan_encode_prores_flush,
    .p.capabilities = AV_CODEC_CAP_DELAY |
                      AV_CODEC_CAP_HARDWARE |
                      AV_CODEC_CAP_ENCODER_FLUSH |
                      AV_CODEC_CAP_ENCODER_REORDERED_OPAQUE,
    CODEC_PIXFMTS(AV_PIX_FMT_VULKAN),
    .hw_configs     = prores_ks_hw_configs,
    .color_ranges   = AVCOL_RANGE_MPEG,
    .p.priv_class   = &proresenc_class,
    .p.profiles     = NULL_IF_CONFIG_SMALL(ff_prores_profiles),
    .caps_internal  = FF_CODEC_CAP_INIT_CLEANUP,
}

Definition at line 974 of file proresenc_kostya_vulkan.c.