#include "libavutil/opt.h"
#include "libavutil/common.h"
#include "libavutil/mem.h"
#include "libavutil/pixdesc.h"
#include "libavutil/hwcontext_d3d12va_internal.h"
#include "avcodec.h"
#include "cbs.h"
#include "cbs_h265.h"
#include "hw_base_encode_h265.h"
#include "h2645data.h"
#include "h265_profile_level.h"
#include "codec_internal.h"
#include "d3d12va_encode.h"

Data Structures
struct	D3D12VAEncodeHEVCPicture

struct	D3D12VAEncodeHEVCContext

struct	D3D12VAEncodeHEVCLevel

Macros
#define	D3D_PROFILE_DESC(name) { sizeof(D3D12_VIDEO_ENCODER_PROFILE_HEVC), { .pHEVCProfile = (D3D12_VIDEO_ENCODER_PROFILE_HEVC *)&profile_ ## name } }

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

#define	FLAGS (AV_OPT_FLAG_VIDEO_PARAM \| AV_OPT_FLAG_ENCODING_PARAM)

#define	PROFILE(name, value)

#define	LEVEL(name, value)

Functions
static uint8_t	d3d12va_encode_hevc_map_cusize (D3D12_VIDEO_ENCODER_CODEC_CONFIGURATION_HEVC_CUSIZE cusize)

static uint8_t	d3d12va_encode_hevc_map_tusize (D3D12_VIDEO_ENCODER_CODEC_CONFIGURATION_HEVC_TUSIZE tusize)

static int	d3d12va_encode_hevc_write_access_unit (AVCodecContext avctx, char data, size_t data_len, CodedBitstreamFragment au)

static int	d3d12va_encode_hevc_add_nal (AVCodecContext avctx, CodedBitstreamFragment au, void *nal_unit)

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

static int	d3d12va_encode_hevc_init_sequence_params (AVCodecContext *avctx)

static int	d3d12va_encode_hevc_get_encoder_caps (AVCodecContext *avctx)

static int	d3d12va_encode_hevc_configure (AVCodecContext *avctx)

static int	d3d12va_encode_hevc_set_level (AVCodecContext *avctx)

static void	d3d12va_encode_hevc_free_picture_params (D3D12VAEncodePicture *pic)

static int	d3d12va_encode_hevc_init_picture_params (AVCodecContext avctx, FFHWBaseEncodePicture base_pic)

static int	d3d12va_encode_hevc_init (AVCodecContext *avctx)

static int	d3d12va_encode_hevc_close (AVCodecContext *avctx)

Variables
static const D3D12_VIDEO_ENCODER_CODEC_CONFIGURATION_SUPPORT_HEVC	hevc_config_support_sets []

static const D3D12VAEncodeHEVCLevel	hevc_levels []

static const D3D12_VIDEO_ENCODER_PROFILE_HEVC	profile_main = D3D12_VIDEO_ENCODER_PROFILE_HEVC_MAIN

static const D3D12_VIDEO_ENCODER_PROFILE_HEVC	profile_main10 = D3D12_VIDEO_ENCODER_PROFILE_HEVC_MAIN10

static const D3D12VAEncodeProfile	d3d12va_encode_hevc_profiles []

static const D3D12VAEncodeType	d3d12va_encode_type_hevc

static const AVOption	d3d12va_encode_hevc_options []

static const FFCodecDefault	d3d12va_encode_hevc_defaults []

static const AVClass	d3d12va_encode_hevc_class

const FFCodec	ff_hevc_d3d12va_encoder

Macro Definition Documentation

◆ D3D_PROFILE_DESC

#define D3D_PROFILE_DESC ( name ) { sizeof(D3D12_VIDEO_ENCODER_PROFILE_HEVC), { .pHEVCProfile = (D3D12_VIDEO_ENCODER_PROFILE_HEVC *)&profile_ ## name } }

Definition at line 131 of file d3d12va_encode_hevc.c.

◆ OFFSET

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

Definition at line 705 of file d3d12va_encode_hevc.c.

◆ FLAGS

#define FLAGS (AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_ENCODING_PARAM)

Definition at line 706 of file d3d12va_encode_hevc.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_hevc_map_cusize()

static uint8_t d3d12va_encode_hevc_map_cusize ( D3D12_VIDEO_ENCODER_CODEC_CONFIGURATION_HEVC_CUSIZE cusize )

static

Definition at line 139 of file d3d12va_encode_hevc.c.

Referenced by d3d12va_encode_hevc_get_encoder_caps(), and d3d12va_encode_hevc_init_sequence_params().

◆ d3d12va_encode_hevc_map_tusize()

static uint8_t d3d12va_encode_hevc_map_tusize ( D3D12_VIDEO_ENCODER_CODEC_CONFIGURATION_HEVC_TUSIZE tusize )

static

Definition at line 151 of file d3d12va_encode_hevc.c.

Referenced by d3d12va_encode_hevc_init_sequence_params().

◆ d3d12va_encode_hevc_write_access_unit()

static int d3d12va_encode_hevc_write_access_unit	(	AVCodecContext *	avctx,
		char *	data,
		size_t *	data_len,
		CodedBitstreamFragment *	au
	)

static

Definition at line 163 of file d3d12va_encode_hevc.c.

Referenced by d3d12va_encode_hevc_write_sequence_header().

◆ d3d12va_encode_hevc_add_nal()

static int d3d12va_encode_hevc_add_nal	(	AVCodecContext *	avctx,
		CodedBitstreamFragment *	au,
		void *	nal_unit
	)

static

Definition at line 189 of file d3d12va_encode_hevc.c.

Referenced by d3d12va_encode_hevc_write_sequence_header().

◆ d3d12va_encode_hevc_write_sequence_header()

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

static

Definition at line 207 of file d3d12va_encode_hevc.c.

◆ d3d12va_encode_hevc_init_sequence_params()

static int d3d12va_encode_hevc_init_sequence_params ( AVCodecContext * avctx )

static

Definition at line 233 of file d3d12va_encode_hevc.c.

◆ d3d12va_encode_hevc_get_encoder_caps()

static int d3d12va_encode_hevc_get_encoder_caps ( AVCodecContext * avctx )

static

Definition at line 361 of file d3d12va_encode_hevc.c.

◆ d3d12va_encode_hevc_configure()

static int d3d12va_encode_hevc_configure ( AVCodecContext * avctx )

static

Definition at line 437 of file d3d12va_encode_hevc.c.

◆ d3d12va_encode_hevc_set_level()

static int d3d12va_encode_hevc_set_level ( AVCodecContext * avctx )

static

Definition at line 499 of file d3d12va_encode_hevc.c.

◆ d3d12va_encode_hevc_free_picture_params()

static void d3d12va_encode_hevc_free_picture_params ( D3D12VAEncodePicture * pic )

static

Definition at line 529 of file d3d12va_encode_hevc.c.

◆ d3d12va_encode_hevc_init_picture_params()

static int d3d12va_encode_hevc_init_picture_params	(	AVCodecContext *	avctx,
		FFHWBaseEncodePicture *	base_pic
	)

static

Definition at line 540 of file d3d12va_encode_hevc.c.

◆ d3d12va_encode_hevc_init()

static int d3d12va_encode_hevc_init ( AVCodecContext * avctx )

static

Definition at line 667 of file d3d12va_encode_hevc.c.

◆ d3d12va_encode_hevc_close()

static int d3d12va_encode_hevc_close ( AVCodecContext * avctx )

static

Definition at line 691 of file d3d12va_encode_hevc.c.

Variable Documentation

◆ hevc_config_support_sets

const D3D12_VIDEO_ENCODER_CODEC_CONFIGURATION_SUPPORT_HEVC hevc_config_support_sets[]

static

Definition at line 63 of file d3d12va_encode_hevc.c.

Referenced by d3d12va_encode_hevc_get_encoder_caps().

◆ hevc_levels

const D3D12VAEncodeHEVCLevel hevc_levels[]

static

Initial value:

= {
    { 30,  D3D12_VIDEO_ENCODER_LEVELS_HEVC_1  },
    { 60,  D3D12_VIDEO_ENCODER_LEVELS_HEVC_2  },
    { 63,  D3D12_VIDEO_ENCODER_LEVELS_HEVC_21 },
    { 90,  D3D12_VIDEO_ENCODER_LEVELS_HEVC_3  },
    { 93,  D3D12_VIDEO_ENCODER_LEVELS_HEVC_31 },
    { 120, D3D12_VIDEO_ENCODER_LEVELS_HEVC_4  },
    { 123, D3D12_VIDEO_ENCODER_LEVELS_HEVC_41 },
    { 150, D3D12_VIDEO_ENCODER_LEVELS_HEVC_5  },
    { 153, D3D12_VIDEO_ENCODER_LEVELS_HEVC_51 },
    { 156, D3D12_VIDEO_ENCODER_LEVELS_HEVC_52 },
    { 180, D3D12_VIDEO_ENCODER_LEVELS_HEVC_6  },
    { 183, D3D12_VIDEO_ENCODER_LEVELS_HEVC_61 },
    { 186, D3D12_VIDEO_ENCODER_LEVELS_HEVC_62 },
}

Definition at line 112 of file d3d12va_encode_hevc.c.

Referenced by d3d12va_encode_hevc_set_level().

◆ profile_main

const D3D12_VIDEO_ENCODER_PROFILE_HEVC profile_main = D3D12_VIDEO_ENCODER_PROFILE_HEVC_MAIN

static

Definition at line 128 of file d3d12va_encode_hevc.c.

◆ profile_main10

const D3D12_VIDEO_ENCODER_PROFILE_HEVC profile_main10 = D3D12_VIDEO_ENCODER_PROFILE_HEVC_MAIN10

static

Definition at line 129 of file d3d12va_encode_hevc.c.

◆ d3d12va_encode_hevc_profiles

const D3D12VAEncodeProfile d3d12va_encode_hevc_profiles[]

static

Initial value:

= {
    { AV_PROFILE_HEVC_MAIN,     8, 3, 1, 1, D3D_PROFILE_DESC(main)   },
    { AV_PROFILE_HEVC_MAIN_10, 10, 3, 1, 1, D3D_PROFILE_DESC(main10) },
    { AV_PROFILE_UNKNOWN },
}

Definition at line 133 of file d3d12va_encode_hevc.c.

◆ d3d12va_encode_type_hevc

const D3D12VAEncodeType d3d12va_encode_type_hevc

static

Initial value:

= {
    .profiles               = d3d12va_encode_hevc_profiles,
 
    .d3d12_codec            = D3D12_VIDEO_ENCODER_CODEC_HEVC,
 
    .flags                  = FF_HW_FLAG_B_PICTURES |
                              FF_HW_FLAG_B_PICTURE_REFERENCES |
                              FF_HW_FLAG_NON_IDR_KEY_PICTURES,
 
    .default_quality        = 25,
 
    .get_encoder_caps       = &d3d12va_encode_hevc_get_encoder_caps,
 
    .configure              = &d3d12va_encode_hevc_configure,
 
    .set_level              = &d3d12va_encode_hevc_set_level,
 
    .picture_priv_data_size = sizeof(D3D12VAEncodeHEVCPicture),
 
    .init_sequence_params   = &d3d12va_encode_hevc_init_sequence_params,
 
    .init_picture_params    = &d3d12va_encode_hevc_init_picture_params,
 
    .free_picture_params    = &d3d12va_encode_hevc_free_picture_params,
 
    .write_sequence_header  = &d3d12va_encode_hevc_write_sequence_header,
}

Definition at line 639 of file d3d12va_encode_hevc.c.

Referenced by d3d12va_encode_hevc_init().

◆ d3d12va_encode_hevc_options

const AVOption d3d12va_encode_hevc_options[]

static

Definition at line 707 of file d3d12va_encode_hevc.c.

◆ d3d12va_encode_hevc_defaults

const FFCodecDefault d3d12va_encode_hevc_defaults[]

static

Initial value:

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

Definition at line 756 of file d3d12va_encode_hevc.c.

◆ d3d12va_encode_hevc_class

const AVClass d3d12va_encode_hevc_class

static

Initial value:

= {
    .class_name = "hevc_d3d12va",
    .item_name  = av_default_item_name,
    .option     = d3d12va_encode_hevc_options,
    .version    = LIBAVUTIL_VERSION_INT,
}

Definition at line 769 of file d3d12va_encode_hevc.c.

◆ ff_hevc_d3d12va_encoder

const FFCodec ff_hevc_d3d12va_encoder

Initial value:

= {
    .p.name         = "hevc_d3d12va",
    CODEC_LONG_NAME("D3D12VA hevc encoder"),
    .p.type         = AVMEDIA_TYPE_VIDEO,
    .p.id           = AV_CODEC_ID_HEVC,
    .priv_data_size = sizeof(D3D12VAEncodeHEVCContext),
    .init           = &d3d12va_encode_hevc_init,
    FF_CODEC_RECEIVE_PACKET_CB(&ff_d3d12va_encode_receive_packet),
    .close          = &d3d12va_encode_hevc_close,
    .p.priv_class   = &d3d12va_encode_hevc_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_hevc_defaults,
    .p.pix_fmts = (const enum AVPixelFormat[]) {
        AV_PIX_FMT_D3D12,
        AV_PIX_FMT_NONE,
    },
    .hw_configs     = ff_d3d12va_encode_hw_configs,
    .p.wrapper_name = "d3d12va",
}

Definition at line 776 of file d3d12va_encode_hevc.c.

Data Structures

Macros

Functions

Variables

Macro Definition Documentation

◆ D3D_PROFILE_DESC

◆ OFFSET

◆ FLAGS

◆ PROFILE

◆ LEVEL

Function Documentation

◆ d3d12va_encode_hevc_map_cusize()

◆ d3d12va_encode_hevc_map_tusize()

◆ d3d12va_encode_hevc_write_access_unit()

◆ d3d12va_encode_hevc_add_nal()

◆ d3d12va_encode_hevc_write_sequence_header()

◆ d3d12va_encode_hevc_init_sequence_params()

◆ d3d12va_encode_hevc_get_encoder_caps()

◆ d3d12va_encode_hevc_configure()

◆ d3d12va_encode_hevc_set_level()

◆ d3d12va_encode_hevc_free_picture_params()

◆ d3d12va_encode_hevc_init_picture_params()

◆ d3d12va_encode_hevc_init()

◆ d3d12va_encode_hevc_close()

Variable Documentation

◆ hevc_config_support_sets

◆ hevc_levels

◆ profile_main

◆ profile_main10

◆ d3d12va_encode_hevc_profiles

◆ d3d12va_encode_type_hevc

◆ d3d12va_encode_hevc_options

◆ d3d12va_encode_hevc_defaults

◆ d3d12va_encode_hevc_class

◆ ff_hevc_d3d12va_encoder