#include "libavutil/opt.h"
#include "libavutil/common.h"
#include "libavutil/mem.h"
#include "libavutil/pixdesc.h"
#include "libavutil/hwcontext_d3d12va_internal.h"
#include "config_components.h"
#include "avcodec.h"
#include "cbs.h"
#include "cbs_av1.h"
#include "av1_levels.h"
#include "codec_internal.h"
#include "d3d12va_encode.h"
#include "encode.h"
#include "hw_base_encode.h"
#include <d3d12.h>
#include <d3d12video.h>

Data Structures
struct	D3D12VAHWBaseEncodeAV1

struct	D3D12VAHWBaseEncodeAV1Opts

struct	D3D12VAEncodeAV1Picture

struct	D3D12VAEncodeAV1Context

struct	D3D12VAEncodeAV1Level

Macros
#define	D3D12_VIDEO_ENCODER_AV1_INVALID_DPB_RESOURCE_INDEX ( 0xff )

#define	D3D_PROFILE_DESC(name) { sizeof(D3D12_VIDEO_ENCODER_AV1_PROFILE), { .pAV1Profile = (D3D12_VIDEO_ENCODER_AV1_PROFILE *)&profile_ ## name } }

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

#define	FLAGS (AV_OPT_FLAG_VIDEO_PARAM \| AV_OPT_FLAG_ENCODING_PARAM)

#define	PROFILE(name, value)

#define	LEVEL(name, value)

Functions
static int	d3d12va_encode_av1_write_obu (AVCodecContext avctx, char data, size_t data_len, CodedBitstreamFragment obu)

static int	d3d12va_encode_av1_add_obu (AVCodecContext avctx, CodedBitstreamFragment au, CodedBitstreamUnitType obu_type, void *obu_unit)

static int	d3d12va_encode_av1_write_sequence_header (AVCodecContext avctx, char data, size_t *data_len)

static int	d3d12va_encode_av1_update_current_frame_picture_header (AVCodecContext avctx, D3D12VAEncodePicture pic, AV1RawOBU *frameheader_obu)

static int	d3d12va_encode_av1_write_picture_header (AVCodecContext avctx, D3D12VAEncodePicture pic, char data, size_t data_len)

static int	d3d12va_encode_av1_write_tile_group (AVCodecContext avctx, uint8_t tile_group, uint32_t tile_group_size, char data, size_t data_len)

static int	d3d12va_encode_av1_get_buffer_size (AVCodecContext avctx, D3D12VAEncodePicture pic, size_t *size)

static int	d3d12va_encode_av1_get_coded_data (AVCodecContext avctx, D3D12VAEncodePicture pic, AVPacket *pkt)

static int	d3d12va_hw_base_encode_init_params_av1 (FFHWBaseEncodeContext base_ctx, AVCodecContext avctx, D3D12VAHWBaseEncodeAV1 common, D3D12VAHWBaseEncodeAV1Opts opts)

static int	d3d12va_encode_av1_init_sequence_params (AVCodecContext *avctx)

static int	d3d12va_encode_av1_get_encoder_caps (AVCodecContext *avctx)

static int	d3d12va_encode_av1_configure (AVCodecContext *avctx)

static int	d3d12va_encode_av1_set_level (AVCodecContext *avctx)

static int	d3d12va_encode_av1_set_tile (AVCodecContext *avctx)

static void	d3d12va_encode_av1_free_picture_params (D3D12VAEncodePicture *pic)

static int	d3d12va_encode_av1_init_picture_params (AVCodecContext avctx, FFHWBaseEncodePicture pic)

static int	d3d12va_encode_av1_init (AVCodecContext *avctx)

static int	d3d12va_encode_av1_close (AVCodecContext *avctx)

Variables
static const D3D12VAEncodeAV1Level	av1_levels []

static const D3D12_VIDEO_ENCODER_AV1_PROFILE	profile_main = D3D12_VIDEO_ENCODER_AV1_PROFILE_MAIN

static const D3D12_VIDEO_ENCODER_AV1_PROFILE	profile_high = D3D12_VIDEO_ENCODER_AV1_PROFILE_HIGH

static const D3D12_VIDEO_ENCODER_AV1_PROFILE	profile_professional = D3D12_VIDEO_ENCODER_AV1_PROFILE_PROFESSIONAL

static const D3D12VAEncodeProfile	d3d12va_encode_av1_profiles []

static const D3D12VAEncodeType	d3d12va_encode_type_av1

static const AVOption	d3d12va_encode_av1_options []

static const FFCodecDefault	d3d12va_encode_av1_defaults []

static const AVClass	d3d12va_encode_av1_class

const FFCodec	ff_av1_d3d12va_encoder

Macro Definition Documentation

◆ D3D12_VIDEO_ENCODER_AV1_INVALID_DPB_RESOURCE_INDEX

#define D3D12_VIDEO_ENCODER_AV1_INVALID_DPB_RESOURCE_INDEX ( 0xff )

Definition at line 43 of file d3d12va_encode_av1.c.

◆ D3D_PROFILE_DESC

#define D3D_PROFILE_DESC ( name ) { sizeof(D3D12_VIDEO_ENCODER_AV1_PROFILE), { .pAV1Profile = (D3D12_VIDEO_ENCODER_AV1_PROFILE *)&profile_ ## name } }

Definition at line 145 of file d3d12va_encode_av1.c.

◆ OFFSET

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

Definition at line 1081 of file d3d12va_encode_av1.c.

◆ FLAGS

#define FLAGS (AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_ENCODING_PARAM)

Definition at line 1082 of file d3d12va_encode_av1.c.

◆ PROFILE

#define PROFILE	(	name,
		value
	)

Value:

name, NULL, 0, AV_OPT_TYPE_CONST, \

{ .i64 = value }, 0, 0, FLAGS, "profile"

◆ LEVEL

#define LEVEL	(	name,
		value
	)

Value:

name, NULL, 0, AV_OPT_TYPE_CONST, \

{ .i64 = value }, 0, 0, FLAGS, "level"

Function Documentation

◆ d3d12va_encode_av1_write_obu()

static int d3d12va_encode_av1_write_obu	(	AVCodecContext *	avctx,
		char *	data,
		size_t *	data_len,
		CodedBitstreamFragment *	obu
	)

static

Definition at line 157 of file d3d12va_encode_av1.c.

Referenced by d3d12va_encode_av1_write_picture_header(), d3d12va_encode_av1_write_sequence_header(), and d3d12va_encode_av1_write_tile_group().

◆ d3d12va_encode_av1_add_obu()

static int d3d12va_encode_av1_add_obu	(	AVCodecContext *	avctx,
		CodedBitstreamFragment *	au,
		CodedBitstreamUnitType	obu_type,
		void *	obu_unit
	)

static

Definition at line 176 of file d3d12va_encode_av1.c.

Referenced by d3d12va_encode_av1_write_picture_header(), d3d12va_encode_av1_write_sequence_header(), and d3d12va_encode_av1_write_tile_group().

◆ d3d12va_encode_av1_write_sequence_header()

static int d3d12va_encode_av1_write_sequence_header	(	AVCodecContext *	avctx,
		char *	data,
		size_t *	data_len
	)

static

Definition at line 192 of file d3d12va_encode_av1.c.

◆ d3d12va_encode_av1_update_current_frame_picture_header()

static int d3d12va_encode_av1_update_current_frame_picture_header	(	AVCodecContext *	avctx,
		D3D12VAEncodePicture *	pic,
		AV1RawOBU *	frameheader_obu
	)

static

Definition at line 211 of file d3d12va_encode_av1.c.

Referenced by d3d12va_encode_av1_write_picture_header().

◆ d3d12va_encode_av1_write_picture_header()

static int d3d12va_encode_av1_write_picture_header	(	AVCodecContext *	avctx,
		D3D12VAEncodePicture *	pic,
		char *	data,
		size_t *	data_len
	)

static

Definition at line 289 of file d3d12va_encode_av1.c.

Referenced by d3d12va_encode_av1_get_coded_data().

◆ d3d12va_encode_av1_write_tile_group()

static int d3d12va_encode_av1_write_tile_group	(	AVCodecContext *	avctx,
		uint8_t *	tile_group,
		uint32_t	tile_group_size,
		char *	data,
		size_t *	data_len
	)

static

Definition at line 319 of file d3d12va_encode_av1.c.

Referenced by d3d12va_encode_av1_get_coded_data().

◆ d3d12va_encode_av1_get_buffer_size()

static int d3d12va_encode_av1_get_buffer_size	(	AVCodecContext *	avctx,
		D3D12VAEncodePicture *	pic,
		size_t *	size
	)

static

Definition at line 346 of file d3d12va_encode_av1.c.

Referenced by d3d12va_encode_av1_get_coded_data().

◆ d3d12va_encode_av1_get_coded_data()

static int d3d12va_encode_av1_get_coded_data	(	AVCodecContext *	avctx,
		D3D12VAEncodePicture *	pic,
		AVPacket *	pkt
	)

static

Definition at line 375 of file d3d12va_encode_av1.c.

◆ d3d12va_hw_base_encode_init_params_av1()

static int d3d12va_hw_base_encode_init_params_av1	(	FFHWBaseEncodeContext *	base_ctx,
		AVCodecContext *	avctx,
		D3D12VAHWBaseEncodeAV1 *	common,
		D3D12VAHWBaseEncodeAV1Opts *	opts
	)

static

Definition at line 449 of file d3d12va_encode_av1.c.

Referenced by d3d12va_encode_av1_init_sequence_params().

◆ d3d12va_encode_av1_init_sequence_params()

static int d3d12va_encode_av1_init_sequence_params ( AVCodecContext * avctx )

static

Definition at line 543 of file d3d12va_encode_av1.c.

◆ d3d12va_encode_av1_get_encoder_caps()

static int d3d12va_encode_av1_get_encoder_caps ( AVCodecContext * avctx )

static

Definition at line 619 of file d3d12va_encode_av1.c.

◆ d3d12va_encode_av1_configure()

static int d3d12va_encode_av1_configure ( AVCodecContext * avctx )

static

Definition at line 726 of file d3d12va_encode_av1.c.

◆ d3d12va_encode_av1_set_level()

static int d3d12va_encode_av1_set_level ( AVCodecContext * avctx )

static

Definition at line 780 of file d3d12va_encode_av1.c.

◆ d3d12va_encode_av1_set_tile()

static int d3d12va_encode_av1_set_tile ( AVCodecContext * avctx )

static

Definition at line 830 of file d3d12va_encode_av1.c.

◆ d3d12va_encode_av1_free_picture_params()

static void d3d12va_encode_av1_free_picture_params ( D3D12VAEncodePicture * pic )

static

Definition at line 845 of file d3d12va_encode_av1.c.

◆ d3d12va_encode_av1_init_picture_params()

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

static

The encoder uses a simple alternating reference frame strategy:

For P-frames, it uses the last reconstructed frame as a reference.
To simplify the reference model of the encoder, the encoder alternates between two reference frame slots (typically slot 0 and slot 1) for storing reconstructed images and providing prediction references for the next frame.

Definition at line 853 of file d3d12va_encode_av1.c.

◆ d3d12va_encode_av1_init()

static int d3d12va_encode_av1_init ( AVCodecContext * avctx )

static

Definition at line 1038 of file d3d12va_encode_av1.c.

◆ d3d12va_encode_av1_close()

static int d3d12va_encode_av1_close ( AVCodecContext * avctx )

static

Definition at line 1064 of file d3d12va_encode_av1.c.

Variable Documentation

◆ av1_levels

const D3D12VAEncodeAV1Level av1_levels[]

static

Initial value:

= {
    { 0,  D3D12_VIDEO_ENCODER_AV1_LEVELS_2_0 },
    { 1,  D3D12_VIDEO_ENCODER_AV1_LEVELS_2_1 },
    { 2,  D3D12_VIDEO_ENCODER_AV1_LEVELS_2_2 },
    { 3,  D3D12_VIDEO_ENCODER_AV1_LEVELS_2_3 },
    { 4,  D3D12_VIDEO_ENCODER_AV1_LEVELS_3_0 },
    { 5,  D3D12_VIDEO_ENCODER_AV1_LEVELS_3_1 },
    { 6,  D3D12_VIDEO_ENCODER_AV1_LEVELS_3_2 },
    { 7,  D3D12_VIDEO_ENCODER_AV1_LEVELS_3_3 },
    { 8,  D3D12_VIDEO_ENCODER_AV1_LEVELS_4_0 },
    { 9,  D3D12_VIDEO_ENCODER_AV1_LEVELS_4_1 },
    { 10, D3D12_VIDEO_ENCODER_AV1_LEVELS_4_2 },
    { 11, D3D12_VIDEO_ENCODER_AV1_LEVELS_4_3 },
    { 12, D3D12_VIDEO_ENCODER_AV1_LEVELS_5_0 },
    { 13, D3D12_VIDEO_ENCODER_AV1_LEVELS_5_1 },
    { 14, D3D12_VIDEO_ENCODER_AV1_LEVELS_5_2 },
    { 15, D3D12_VIDEO_ENCODER_AV1_LEVELS_5_3 },
    { 16, D3D12_VIDEO_ENCODER_AV1_LEVELS_6_0 },
    { 17, D3D12_VIDEO_ENCODER_AV1_LEVELS_6_1 },
    { 18, D3D12_VIDEO_ENCODER_AV1_LEVELS_6_2 },
    { 19, D3D12_VIDEO_ENCODER_AV1_LEVELS_6_3 },
    { 20, D3D12_VIDEO_ENCODER_AV1_LEVELS_7_0 },
    { 21, D3D12_VIDEO_ENCODER_AV1_LEVELS_7_1 },
    { 22, D3D12_VIDEO_ENCODER_AV1_LEVELS_7_2 },
    { 23, D3D12_VIDEO_ENCODER_AV1_LEVELS_7_3 },
}

Definition at line 114 of file d3d12va_encode_av1.c.

Referenced by d3d12va_encode_av1_set_level().

◆ profile_main

const D3D12_VIDEO_ENCODER_AV1_PROFILE profile_main = D3D12_VIDEO_ENCODER_AV1_PROFILE_MAIN

static

Definition at line 141 of file d3d12va_encode_av1.c.

◆ profile_high

const D3D12_VIDEO_ENCODER_AV1_PROFILE profile_high = D3D12_VIDEO_ENCODER_AV1_PROFILE_HIGH

static

Definition at line 142 of file d3d12va_encode_av1.c.

◆ profile_professional

const D3D12_VIDEO_ENCODER_AV1_PROFILE profile_professional = D3D12_VIDEO_ENCODER_AV1_PROFILE_PROFESSIONAL

static

Definition at line 143 of file d3d12va_encode_av1.c.

◆ d3d12va_encode_av1_profiles

const D3D12VAEncodeProfile d3d12va_encode_av1_profiles[]

static

Initial value:

= {
    { AV_PROFILE_AV1_MAIN,          8, 3, 1, 1, D3D_PROFILE_DESC(main)         },
    { AV_PROFILE_AV1_MAIN,         10, 3, 1, 1, D3D_PROFILE_DESC(main)         },
    { AV_PROFILE_AV1_HIGH,         10, 3, 1, 1, D3D_PROFILE_DESC(high)         },
    { AV_PROFILE_AV1_PROFESSIONAL,  8, 3, 1, 1, D3D_PROFILE_DESC(professional) },
    { AV_PROFILE_AV1_PROFESSIONAL, 10, 3, 1, 1, D3D_PROFILE_DESC(professional) },
    { AV_PROFILE_AV1_PROFESSIONAL, 12, 3, 1, 1, D3D_PROFILE_DESC(professional) },
    { AV_PROFILE_UNKNOWN },
}

Definition at line 147 of file d3d12va_encode_av1.c.

◆ d3d12va_encode_type_av1

const D3D12VAEncodeType d3d12va_encode_type_av1

static

Definition at line 1004 of file d3d12va_encode_av1.c.

Referenced by d3d12va_encode_av1_init().

◆ d3d12va_encode_av1_options

const AVOption d3d12va_encode_av1_options[]

static

Definition at line 1083 of file d3d12va_encode_av1.c.

◆ d3d12va_encode_av1_defaults

const FFCodecDefault d3d12va_encode_av1_defaults[]

static

Initial value:

= {
    { "b",              "0"   },
    { "bf",             "0"   },
    { "g",              "120" },
    { "i_qfactor",      "1"   },
    { "i_qoffset",      "0"   },
    { "b_qfactor",      "1"   },
    { "b_qoffset",      "0"   },
    { "qmin",           "-1"  },
    { "qmax",           "-1"  },
    { "refs",           "0"   },
    { NULL },
}

Definition at line 1143 of file d3d12va_encode_av1.c.

◆ d3d12va_encode_av1_class

const AVClass d3d12va_encode_av1_class

static

Initial value:

= {
    .class_name = "av1_d3d12va",
    .item_name  = av_default_item_name,
    .option     = d3d12va_encode_av1_options,
    .version    = LIBAVUTIL_VERSION_INT,
}

Definition at line 1157 of file d3d12va_encode_av1.c.

◆ ff_av1_d3d12va_encoder

const FFCodec ff_av1_d3d12va_encoder

Initial value:

= {
    .p.name         = "av1_d3d12va",
    CODEC_LONG_NAME("D3D12VA av1 encoder"),
    .p.type         = AVMEDIA_TYPE_VIDEO,
    .p.id           = AV_CODEC_ID_AV1,
    .priv_data_size = sizeof(D3D12VAEncodeAV1Context),
    .init           = &d3d12va_encode_av1_init,
    FF_CODEC_RECEIVE_PACKET_CB(&ff_d3d12va_encode_receive_packet),
    .close          = &d3d12va_encode_av1_close,
    .p.priv_class   = &d3d12va_encode_av1_class,
    .p.capabilities = AV_CODEC_CAP_DELAY | AV_CODEC_CAP_HARDWARE |
                      AV_CODEC_CAP_DR1 | AV_CODEC_CAP_ENCODER_REORDERED_OPAQUE,
    .caps_internal  = FF_CODEC_CAP_NOT_INIT_THREADSAFE |
                      FF_CODEC_CAP_INIT_CLEANUP,
    .defaults       = d3d12va_encode_av1_defaults,
    CODEC_PIXFMTS(AV_PIX_FMT_D3D12),
    .hw_configs     = ff_d3d12va_encode_hw_configs,
    .p.wrapper_name = "d3d12va",
}

Definition at line 1164 of file d3d12va_encode_av1.c.

Data Structures

Macros

Functions

Variables

Macro Definition Documentation

◆ D3D12_VIDEO_ENCODER_AV1_INVALID_DPB_RESOURCE_INDEX

◆ D3D_PROFILE_DESC

◆ OFFSET

◆ FLAGS

◆ PROFILE

◆ LEVEL

Function Documentation

◆ d3d12va_encode_av1_write_obu()

◆ d3d12va_encode_av1_add_obu()

◆ d3d12va_encode_av1_write_sequence_header()

◆ d3d12va_encode_av1_update_current_frame_picture_header()

◆ d3d12va_encode_av1_write_picture_header()

◆ d3d12va_encode_av1_write_tile_group()

◆ d3d12va_encode_av1_get_buffer_size()

◆ d3d12va_encode_av1_get_coded_data()

◆ d3d12va_hw_base_encode_init_params_av1()

◆ d3d12va_encode_av1_init_sequence_params()

◆ d3d12va_encode_av1_get_encoder_caps()

◆ d3d12va_encode_av1_configure()

◆ d3d12va_encode_av1_set_level()

◆ d3d12va_encode_av1_set_tile()

◆ d3d12va_encode_av1_free_picture_params()

◆ d3d12va_encode_av1_init_picture_params()

◆ d3d12va_encode_av1_init()

◆ d3d12va_encode_av1_close()

Variable Documentation

◆ av1_levels

◆ profile_main

◆ profile_high

◆ profile_professional

◆ d3d12va_encode_av1_profiles

◆ d3d12va_encode_type_av1

◆ d3d12va_encode_av1_options

◆ d3d12va_encode_av1_defaults

◆ d3d12va_encode_av1_class

◆ ff_av1_d3d12va_encoder