#include <VideoToolbox/VideoToolbox.h>
#include <CoreVideo/CoreVideo.h>
#include <CoreMedia/CoreMedia.h>
#include <TargetConditionals.h>
#include <Availability.h>
#include "avcodec.h"
#include "libavutil/opt.h"
#include "libavutil/avassert.h"
#include "libavutil/avstring.h"
#include "libavutil/pixdesc.h"
#include "libavutil/hwcontext_videotoolbox.h"
#include "codec_internal.h"
#include "internal.h"
#include <pthread.h>
#include "atsc_a53.h"
#include "encode.h"
#include "h264.h"
#include "h264_sei.h"
#include "hwconfig.h"
#include <dlfcn.h>

Data Structures
struct	ExtraSEI

struct	BufNode

struct	VTEncContext

Macros
#define	TARGET_CPU_ARM64 0

#define	GET_SYM(symbol, defaultVal)

#define	H264_PROFILE_CONSTRAINED_HIGH (AV_PROFILE_H264_HIGH \| AV_PROFILE_H264_CONSTRAINED)

#define	VE AV_OPT_FLAG_VIDEO_PARAM \| AV_OPT_FLAG_ENCODING_PARAM

#define	COMMON_OPTIONS

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

Typedefs
typedef OSStatus(*	getParameterSetAtIndex) (CMFormatDescriptionRef videoDesc, size_t parameterSetIndex, const uint8_t *parameterSetPointerOut, size_t parameterSetSizeOut, size_t parameterSetCountOut, int NALUnitHeaderLengthOut)

Enumerations
enum	{ kCMVideoCodecType_HEVC = 'hvc1' }

enum	{ kCMVideoCodecType_HEVCWithAlpha = 'muxa' }

enum	{ kCVPixelFormatType_420YpCbCr10BiPlanarFullRange = 'xf20' }

enum	{ kCVPixelFormatType_420YpCbCr10BiPlanarVideoRange = 'x420' }

enum	VTH264Entropy { VT_ENTROPY_NOT_SET, VT_CAVLC, VT_CABAC }

Functions
static void	loadVTEncSymbols (void)

static int	vt_dump_encoder (AVCodecContext *avctx)

static int	vtenc_populate_extradata (AVCodecContext *avctx, CMVideoCodecType codec_type, CFStringRef profile_level, CFNumberRef gamma_level, CFDictionaryRef enc_info, CFDictionaryRef pixel_buffer_info)

static void	vt_release_num (CFNumberRef *refPtr)
	NULL-safe release of *refPtr, and sets value to NULL. More...

static void	set_async_error (VTEncContext *vtctx, int err)

static void	clear_frame_queue (VTEncContext *vtctx)

static void	vtenc_reset (VTEncContext *vtctx)

static int	vtenc_q_pop (VTEncContext vtctx, bool wait, CMSampleBufferRef buf, ExtraSEI **sei)

static void	vtenc_q_push (VTEncContext vtctx, CMSampleBufferRef buffer, ExtraSEI sei)

static int	count_nalus (size_t length_code_size, CMSampleBufferRef sample_buffer, int *count)

static CMVideoCodecType	get_cm_codec_type (AVCodecContext *avctx, int profile, double alpha_quality)

static int	get_params_size (AVCodecContext avctx, CMVideoFormatDescriptionRef vid_fmt, size_t size)
	Get the parameter sets from a CMSampleBufferRef. More...

static int	copy_param_sets (AVCodecContext avctx, CMVideoFormatDescriptionRef vid_fmt, uint8_t dst, size_t dst_size)

static int	set_extradata (AVCodecContext *avctx, CMSampleBufferRef sample_buffer)

static void	vtenc_output_callback (void ctx, void sourceFrameCtx, OSStatus status, VTEncodeInfoFlags flags, CMSampleBufferRef sample_buffer)

static int	get_length_code_size (AVCodecContext avctx, CMSampleBufferRef sample_buffer, size_t size)

static bool	get_vt_h264_profile_level (AVCodecContext avctx, CFStringRef profile_level_val)

static bool	get_vt_hevc_profile_level (AVCodecContext avctx, CFStringRef profile_level_val)

static int	get_cv_pixel_format (AVCodecContext avctx, enum AVPixelFormat fmt, enum AVColorRange range, int av_pixel_format, int *range_guessed)

static void	add_color_attr (AVCodecContext *avctx, CFMutableDictionaryRef dict)

static int	create_cv_pixel_buffer_info (AVCodecContext avctx, CFMutableDictionaryRef dict)

static int	get_cv_gamma (AVCodecContext avctx, CFNumberRef gamma_level)

static bool	vtenc_qscale_enabled (void)

static void	set_encoder_property_or_log (AVCodecContext avctx, CFStringRef key, const char print_option_name, CFTypeRef value)

static int	set_encoder_int_property_or_log (AVCodecContext avctx, CFStringRef key, const char print_option_name, int value)

static int	vtenc_create_encoder (AVCodecContext avctx, CMVideoCodecType codec_type, CFStringRef profile_level, CFNumberRef gamma_level, CFDictionaryRef enc_info, CFDictionaryRef pixel_buffer_info, bool constant_bit_rate, VTCompressionSessionRef session)

static int	vtenc_configure_encoder (AVCodecContext *avctx)

static av_cold int	vtenc_init (AVCodecContext *avctx)

static void	vtenc_get_frame_info (CMSampleBufferRef buffer, bool *is_key_frame)

static int	is_post_sei_nal_type (int nal_type)

static int	find_sei_end (AVCodecContext avctx, uint8_t nal_data, size_t nal_size, uint8_t **sei_end)

static int	copy_emulation_prev (const uint8_t src, size_t src_size, uint8_t dst, ssize_t dst_offset, size_t dst_size)
	Copies the data inserting emulation prevention bytes as needed. More...

static int	write_sei (const ExtraSEI sei, int sei_type, uint8_t dst, size_t dst_size)

static int	copy_replace_length_codes (AVCodecContext avctx, size_t length_code_size, CMSampleBufferRef sample_buffer, ExtraSEI sei, uint8_t *dst_data, size_t dst_size)
	Copies NAL units and replaces length codes with H.264 Annex B start codes. More...

static int	get_sei_msg_bytes (const ExtraSEI *sei, int type)
	Returns a sufficient number of bytes to contain the sei data. More...

static int	vtenc_cm_to_avpacket (AVCodecContext avctx, CMSampleBufferRef sample_buffer, AVPacket pkt, ExtraSEI *sei)

static int	get_cv_pixel_info (AVCodecContext avctx, const AVFrame frame, int color, int plane_count, size_t widths, size_t heights, size_t strides, size_t contiguous_buf_size)

static int	copy_avframe_to_pixel_buffer (AVCodecContext avctx, const AVFrame frame, CVPixelBufferRef cv_img, const size_t plane_strides, const size_t plane_rows)

static int	create_cv_pixel_buffer (AVCodecContext avctx, const AVFrame frame, CVPixelBufferRef *cv_img)

static int	create_encoder_dict_h264 (const AVFrame frame, CFDictionaryRef dict_out)

static int	vtenc_send_frame (AVCodecContext avctx, VTEncContext vtctx, const AVFrame *frame)

static av_cold int	vtenc_frame (AVCodecContext avctx, AVPacket pkt, const AVFrame frame, int got_packet)

static av_cold int	vtenc_close (AVCodecContext *avctx)

Variables
struct {
CFStringRef kCVImageBufferColorPrimaries_ITU_R_2020

CFStringRef kCVImageBufferTransferFunction_ITU_R_2020

CFStringRef kCVImageBufferYCbCrMatrix_ITU_R_2020

CFStringRef kVTCompressionPropertyKey_H264EntropyMode

CFStringRef kVTH264EntropyMode_CAVLC

CFStringRef kVTH264EntropyMode_CABAC

CFStringRef kVTProfileLevel_H264_Baseline_4_0

CFStringRef kVTProfileLevel_H264_Baseline_4_2

CFStringRef kVTProfileLevel_H264_Baseline_5_0

CFStringRef kVTProfileLevel_H264_Baseline_5_1

CFStringRef kVTProfileLevel_H264_Baseline_5_2

CFStringRef kVTProfileLevel_H264_Baseline_AutoLevel

CFStringRef kVTProfileLevel_H264_Main_4_2

CFStringRef kVTProfileLevel_H264_Main_5_1

CFStringRef kVTProfileLevel_H264_Main_5_2

CFStringRef kVTProfileLevel_H264_Main_AutoLevel

CFStringRef kVTProfileLevel_H264_High_3_0

CFStringRef kVTProfileLevel_H264_High_3_1

CFStringRef kVTProfileLevel_H264_High_3_2

CFStringRef kVTProfileLevel_H264_High_4_0

CFStringRef kVTProfileLevel_H264_High_4_1

CFStringRef kVTProfileLevel_H264_High_4_2

CFStringRef kVTProfileLevel_H264_High_5_1

CFStringRef kVTProfileLevel_H264_High_5_2

CFStringRef kVTProfileLevel_H264_High_AutoLevel

CFStringRef kVTProfileLevel_H264_Extended_5_0

CFStringRef kVTProfileLevel_H264_Extended_AutoLevel

CFStringRef kVTProfileLevel_H264_ConstrainedBaseline_AutoLevel

CFStringRef kVTProfileLevel_H264_ConstrainedHigh_AutoLevel

CFStringRef kVTProfileLevel_HEVC_Main_AutoLevel

CFStringRef kVTProfileLevel_HEVC_Main10_AutoLevel

CFStringRef kVTCompressionPropertyKey_RealTime

CFStringRef kVTCompressionPropertyKey_TargetQualityForAlpha

CFStringRef kVTCompressionPropertyKey_PrioritizeEncodingSpeedOverQuality

CFStringRef kVTCompressionPropertyKey_ConstantBitRate

CFStringRef kVTCompressionPropertyKey_EncoderID

CFStringRef kVTVideoEncoderSpecification_EnableHardwareAcceleratedVideoEncoder

CFStringRef kVTVideoEncoderSpecification_RequireHardwareAcceleratedVideoEncoder

CFStringRef kVTVideoEncoderSpecification_EnableLowLatencyRateControl

CFStringRef kVTCompressionPropertyKey_AllowOpenGOP

CFStringRef kVTCompressionPropertyKey_MaximizePowerEfficiency

CFStringRef kVTCompressionPropertyKey_ReferenceBufferCount

CFStringRef kVTCompressionPropertyKey_MaxAllowedFrameQP

CFStringRef kVTCompressionPropertyKey_MinAllowedFrameQP

getParameterSetAtIndex CMVideoFormatDescriptionGetHEVCParameterSetAtIndex

}	compat_keys

static pthread_once_t	once_ctrl = PTHREAD_ONCE_INIT

static const uint8_t	start_code [] = { 0, 0, 0, 1 }

static enum AVPixelFormat	avc_pix_fmts []

static enum AVPixelFormat	hevc_pix_fmts []

static enum AVPixelFormat	prores_pix_fmts []

static const AVCodecHWConfigInternal *const	vt_encode_hw_configs []

static const AVOption	h264_options []

static const FFCodecDefault	vt_defaults []

static const AVClass	h264_videotoolbox_class

const FFCodec	ff_h264_videotoolbox_encoder

static const AVOption	hevc_options []

static const AVClass	hevc_videotoolbox_class

const FFCodec	ff_hevc_videotoolbox_encoder

static const AVOption	prores_options []

static const AVClass	prores_videotoolbox_class

const FFCodec	ff_prores_videotoolbox_encoder

Macro Definition Documentation

◆ TARGET_CPU_ARM64

#define TARGET_CPU_ARM64 0

Definition at line 57 of file videotoolboxenc.c.

◆ GET_SYM

#define GET_SYM	(	symbol,
		defaultVal
	)

Value:

do{                                                                     \
    CFStringRef* handle = (CFStringRef*)dlsym(RTLD_DEFAULT, #symbol);   \
    if(!handle)                                                         \
        compat_keys.symbol = CFSTR(defaultVal);                         \
    else                                                                \
        compat_keys.symbol = *handle;                                   \
}while(0)

Definition at line 136 of file videotoolboxenc.c.

◆ H264_PROFILE_CONSTRAINED_HIGH

#define H264_PROFILE_CONSTRAINED_HIGH (AV_PROFILE_H264_HIGH | AV_PROFILE_H264_CONSTRAINED)

Definition at line 212 of file videotoolboxenc.c.

◆ VE

#define VE AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_ENCODING_PARAM

Definition at line 2847 of file videotoolboxenc.c.

◆ COMMON_OPTIONS

#define COMMON_OPTIONS

Value:

    { "allow_sw", "Allow software encoding", OFFSET(allow_sw), AV_OPT_TYPE_BOOL, \
        { .i64 = 0 }, 0, 1, VE }, \
    { "require_sw", "Require software encoding", OFFSET(require_sw), AV_OPT_TYPE_BOOL, \
        { .i64 = 0 }, 0, 1, VE }, \
    { "realtime", "Hint that encoding should happen in real-time if not faster (e.g. capturing from camera).", \
        OFFSET(realtime), AV_OPT_TYPE_BOOL, { .i64 = 0 }, -1, 1, VE }, \
    { "frames_before", "Other frames will come before the frames in this session. This helps smooth concatenation issues.", \
        OFFSET(frames_before), AV_OPT_TYPE_BOOL, { .i64 = 0 }, 0, 1, VE }, \
    { "frames_after", "Other frames will come after the frames in this session. This helps smooth concatenation issues.", \
        OFFSET(frames_after), AV_OPT_TYPE_BOOL, { .i64 = 0 }, 0, 1, VE }, \
    { "prio_speed", "prioritize encoding speed", OFFSET(prio_speed), AV_OPT_TYPE_BOOL, \
        { .i64 = -1 }, -1, 1, VE }, \
    { "power_efficient", "Set to 1 to enable more power-efficient encoding if supported.", \
        OFFSET(power_efficient), AV_OPT_TYPE_INT, { .i64 = -1 }, -1, 1, VE }, \
    { "max_ref_frames", \
        "Sets the maximum number of reference frames. This only has an effect when the value is less than the maximum allowed by the profile/level.", \
        OFFSET(max_ref_frames), AV_OPT_TYPE_INT, { .i64 = 0 }, 0, INT_MAX, VE },

Definition at line 2848 of file videotoolboxenc.c.

◆ OFFSET

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

Definition at line 2872 of file videotoolboxenc.c.

Typedef Documentation

◆ getParameterSetAtIndex

typedef OSStatus(* getParameterSetAtIndex) (CMFormatDescriptionRef videoDesc, size_t parameterSetIndex, const uint8_t **parameterSetPointerOut, size_t *parameterSetSizeOut, size_t *parameterSetCountOut, int *NALUnitHeaderLengthOut)

Definition at line 60 of file videotoolboxenc.c.

Enumeration Type Documentation

◆ anonymous enum

anonymous enum

Enumerator
kCMVideoCodecType_HEVC

Definition at line 44 of file videotoolboxenc.c.

◆ anonymous enum

anonymous enum

Enumerator
kCMVideoCodecType_HEVCWithAlpha

Definition at line 48 of file videotoolboxenc.c.

◆ anonymous enum

anonymous enum

Enumerator
kCVPixelFormatType_420YpCbCr10BiPlanarFullRange

Definition at line 52 of file videotoolboxenc.c.

◆ anonymous enum

anonymous enum

Enumerator
kCVPixelFormatType_420YpCbCr10BiPlanarVideoRange

Definition at line 53 of file videotoolboxenc.c.

◆ VTH264Entropy

enum VTH264Entropy

Enumerator
VT_ENTROPY_NOT_SET
VT_CAVLC
VT_CABAC

Definition at line 214 of file videotoolboxenc.c.

Function Documentation

◆ loadVTEncSymbols()

static void loadVTEncSymbols ( void )

static

Definition at line 147 of file videotoolboxenc.c.

Referenced by vtenc_init().

◆ vt_dump_encoder()

static int vt_dump_encoder ( AVCodecContext * avctx )

static

Definition at line 283 of file videotoolboxenc.c.

Referenced by vtenc_create_encoder().

◆ vtenc_populate_extradata()

static int vtenc_populate_extradata	(	AVCodecContext *	avctx,
		CMVideoCodecType	codec_type,
		CFStringRef	profile_level,
		CFNumberRef	gamma_level,
		CFDictionaryRef	enc_info,
		CFDictionaryRef	pixel_buffer_info
	)

static

Definition at line 2688 of file videotoolboxenc.c.

Referenced by vtenc_configure_encoder().

◆ vt_release_num()

static void vt_release_num ( CFNumberRef * refPtr )

static

NULL-safe release of *refPtr, and sets value to NULL.

Definition at line 328 of file videotoolboxenc.c.

Referenced by create_cv_pixel_buffer_info().

◆ set_async_error()

static void set_async_error	(	VTEncContext *	vtctx,
		int	err
	)

static

Definition at line 337 of file videotoolboxenc.c.

Referenced by clear_frame_queue(), vtenc_output_callback(), and vtenc_q_push().

◆ clear_frame_queue()

static void clear_frame_queue ( VTEncContext * vtctx )

static

Definition at line 358 of file videotoolboxenc.c.

Referenced by vtenc_close().

◆ vtenc_reset()

static void vtenc_reset ( VTEncContext * vtctx )

static

Definition at line 363 of file videotoolboxenc.c.

Referenced by create_cv_pixel_buffer(), vtenc_close(), and vtenc_populate_extradata().

◆ vtenc_q_pop()

static int vtenc_q_pop	(	VTEncContext *	vtctx,
		bool	wait,
		CMSampleBufferRef *	buf,
		ExtraSEI **	sei
	)

static

Definition at line 391 of file videotoolboxenc.c.

Referenced by vtenc_frame(), and vtenc_populate_extradata().

◆ vtenc_q_push()

static void vtenc_q_push	(	VTEncContext *	vtctx,
		CMSampleBufferRef	buffer,
		ExtraSEI *	sei
	)

static

Definition at line 441 of file videotoolboxenc.c.

Referenced by vtenc_output_callback().

◆ count_nalus()

static int count_nalus	(	size_t	length_code_size,
		CMSampleBufferRef	sample_buffer,
		int *	count
	)

static

Definition at line 468 of file videotoolboxenc.c.

Referenced by vtenc_cm_to_avpacket().

◆ get_cm_codec_type()

static CMVideoCodecType get_cm_codec_type	(	AVCodecContext *	avctx,
		int	profile,
		double	alpha_quality
	)

static

Definition at line 511 of file videotoolboxenc.c.

Referenced by vtenc_configure_encoder().

◆ get_params_size()

static int get_params_size	(	AVCodecContext *	avctx,
		CMVideoFormatDescriptionRef	vid_fmt,
		size_t *	size
	)

static

Get the parameter sets from a CMSampleBufferRef.

Parameters

dst	If dst isn't NULL, the parameters are copied into existing memory. dst_size must be set accordingly when dst != NULL. If dst is NULL, it will be allocated. In all cases, dst_size is set to the number of bytes used starting at dst.

Definition at line 562 of file videotoolboxenc.c.

Referenced by set_extradata(), and vtenc_cm_to_avpacket().

◆ copy_param_sets()

static int copy_param_sets	(	AVCodecContext *	avctx,
		CMVideoFormatDescriptionRef	vid_fmt,
		uint8_t *	dst,
		size_t	dst_size
	)

static

Definition at line 616 of file videotoolboxenc.c.

Referenced by set_extradata(), and vtenc_cm_to_avpacket().

◆ set_extradata()

static int set_extradata	(	AVCodecContext *	avctx,
		CMSampleBufferRef	sample_buffer
	)

static

Definition at line 680 of file videotoolboxenc.c.

Referenced by vtenc_output_callback().

◆ vtenc_output_callback()

static void vtenc_output_callback	(	void *	ctx,
		void *	sourceFrameCtx,
		OSStatus	status,
		VTEncodeInfoFlags	flags,
		CMSampleBufferRef	sample_buffer
	)

static

Definition at line 729 of file videotoolboxenc.c.

Referenced by vtenc_create_encoder().

◆ get_length_code_size()

static int get_length_code_size	(	AVCodecContext *	avctx,
		CMSampleBufferRef	sample_buffer,
		size_t *	size
	)

static

Definition at line 765 of file videotoolboxenc.c.

Referenced by vtenc_cm_to_avpacket().

◆ get_vt_h264_profile_level()

static bool get_vt_h264_profile_level	(	AVCodecContext *	avctx,
		CFStringRef *	profile_level_val
	)

static

Definition at line 802 of file videotoolboxenc.c.

Referenced by vtenc_configure_encoder().

◆ get_vt_hevc_profile_level()

static bool get_vt_hevc_profile_level	(	AVCodecContext *	avctx,
		CFStringRef *	profile_level_val
	)

static

Definition at line 931 of file videotoolboxenc.c.

Referenced by vtenc_configure_encoder().

◆ get_cv_pixel_format()

static int get_cv_pixel_format	(	AVCodecContext *	avctx,
		enum AVPixelFormat	fmt,
		enum AVColorRange	range,
		int *	av_pixel_format,
		int *	range_guessed
	)

static

Definition at line 978 of file videotoolboxenc.c.

Referenced by create_cv_pixel_buffer_info(), and get_cv_pixel_info().

◆ add_color_attr()

static void add_color_attr	(	AVCodecContext *	avctx,
		CFMutableDictionaryRef	dict
	)

static

Definition at line 1002 of file videotoolboxenc.c.

Referenced by create_cv_pixel_buffer_info().

◆ create_cv_pixel_buffer_info()

static int create_cv_pixel_buffer_info	(	AVCodecContext *	avctx,
		CFMutableDictionaryRef *	dict
	)

static

Definition at line 1024 of file videotoolboxenc.c.

Referenced by vtenc_configure_encoder().

◆ get_cv_gamma()

static int get_cv_gamma	(	AVCodecContext *	avctx,
		CFNumberRef *	gamma_level
	)

static

Definition at line 1091 of file videotoolboxenc.c.

Referenced by vtenc_configure_encoder().

◆ vtenc_qscale_enabled()

static bool vtenc_qscale_enabled ( void )

static

Definition at line 1109 of file videotoolboxenc.c.

Referenced by vtenc_create_encoder().

◆ set_encoder_property_or_log()

static void set_encoder_property_or_log	(	AVCodecContext *	avctx,
		CFStringRef	key,
		const char *	print_option_name,
		CFTypeRef	value
	)

static

Definition at line 1114 of file videotoolboxenc.c.

Referenced by set_encoder_int_property_or_log(), and vtenc_create_encoder().

◆ set_encoder_int_property_or_log()

static int set_encoder_int_property_or_log	(	AVCodecContext *	avctx,
		CFStringRef	key,
		const char *	print_option_name,
		int	value
	)

static

Definition at line 1136 of file videotoolboxenc.c.

Referenced by vtenc_create_encoder().

◆ vtenc_create_encoder()

static int vtenc_create_encoder	(	AVCodecContext *	avctx,
		CMVideoCodecType	codec_type,
		CFStringRef	profile_level,
		CFNumberRef	gamma_level,
		CFDictionaryRef	enc_info,
		CFDictionaryRef	pixel_buffer_info,
		bool	constant_bit_rate,
		VTCompressionSessionRef *	session
	)

static

Definition at line 1155 of file videotoolboxenc.c.

Referenced by vtenc_configure_encoder(), and vtenc_populate_extradata().

◆ vtenc_configure_encoder()

static int vtenc_configure_encoder ( AVCodecContext * avctx )

static

Definition at line 1601 of file videotoolboxenc.c.

Referenced by create_cv_pixel_buffer(), and vtenc_init().

◆ vtenc_init()

static av_cold int vtenc_init ( AVCodecContext * avctx )

static

Definition at line 1730 of file videotoolboxenc.c.

◆ vtenc_get_frame_info()

static void vtenc_get_frame_info	(	CMSampleBufferRef	buffer,
		bool *	is_key_frame
	)

static

Definition at line 1766 of file videotoolboxenc.c.

Referenced by vtenc_cm_to_avpacket().

◆ is_post_sei_nal_type()

static int is_post_sei_nal_type ( int nal_type )

static

Definition at line 1793 of file videotoolboxenc.c.

Referenced by copy_replace_length_codes().

◆ find_sei_end()

static int find_sei_end	(	AVCodecContext *	avctx,
		uint8_t *	nal_data,
		size_t	nal_size,
		uint8_t **	sei_end
	)

static

Definition at line 1804 of file videotoolboxenc.c.

Referenced by copy_replace_length_codes().

◆ copy_emulation_prev()

static int copy_emulation_prev	(	const uint8_t *	src,
		size_t	src_size,
		uint8_t *	dst,
		ssize_t	dst_offset,
		size_t	dst_size
	)

static

Copies the data inserting emulation prevention bytes as needed.

Existing data in the destination can be taken into account by providing dst with a dst_offset > 0.

Returns: The number of bytes copied on success. On failure, the negative of the number of bytes needed to copy src is returned.

Definition at line 1866 of file videotoolboxenc.c.

Referenced by get_sei_msg_bytes(), and write_sei().

◆ write_sei()

static int write_sei	(	const ExtraSEI *	sei,
		int	sei_type,
		uint8_t *	dst,
		size_t	dst_size
	)

static

Definition at line 1917 of file videotoolboxenc.c.

Referenced by copy_replace_length_codes().

◆ copy_replace_length_codes()

static int copy_replace_length_codes	(	AVCodecContext *	avctx,
		size_t	length_code_size,
		CMSampleBufferRef	sample_buffer,
		ExtraSEI *	sei,
		uint8_t *	dst_data,
		size_t	dst_size
	)

static

Copies NAL units and replaces length codes with H.264 Annex B start codes.

On failure, the contents of dst_data may have been modified.

Parameters

length_code_size	Byte length of each length code
sample_buffer	NAL units prefixed with length codes.
sei	Optional A53 closed captions SEI data.
dst_data	Must be zeroed before calling this function. Contains the copied NAL units prefixed with start codes when the function returns successfully.
dst_size	Length of dst_data

Returns: 0 on success AVERROR_INVALIDDATA if length_code_size is invalid AVERROR_BUFFER_TOO_SMALL if dst_data is too small or if a length_code in src_data specifies data beyond the end of its buffer.

Definition at line 1990 of file videotoolboxenc.c.

Referenced by vtenc_cm_to_avpacket().

◆ get_sei_msg_bytes()

static int get_sei_msg_bytes	(	const ExtraSEI *	sei,
		int	type
	)

static

Returns a sufficient number of bytes to contain the sei data.

It may be greater than the minimum required.

Definition at line 2147 of file videotoolboxenc.c.

Referenced by vtenc_cm_to_avpacket().

◆ vtenc_cm_to_avpacket()

static int vtenc_cm_to_avpacket	(	AVCodecContext *	avctx,
		CMSampleBufferRef	sample_buffer,
		AVPacket *	pkt,
		ExtraSEI *	sei
	)

static

Definition at line 2164 of file videotoolboxenc.c.

Referenced by vtenc_frame().

◆ get_cv_pixel_info()

static int get_cv_pixel_info	(	AVCodecContext *	avctx,
		const AVFrame *	frame,
		int *	color,
		int *	plane_count,
		size_t *	widths,
		size_t *	heights,
		size_t *	strides,
		size_t *	contiguous_buf_size
	)

static

Definition at line 2294 of file videotoolboxenc.c.

Referenced by create_cv_pixel_buffer().

◆ copy_avframe_to_pixel_buffer()

static int copy_avframe_to_pixel_buffer	(	AVCodecContext *	avctx,
		const AVFrame *	frame,
		CVPixelBufferRef	cv_img,
		const size_t *	plane_strides,
		const size_t *	plane_rows
	)

static

Definition at line 2358 of file videotoolboxenc.c.

Referenced by create_cv_pixel_buffer().

◆ create_cv_pixel_buffer()

static int create_cv_pixel_buffer	(	AVCodecContext *	avctx,
		const AVFrame *	frame,
		CVPixelBufferRef *	cv_img
	)

static

Definition at line 2450 of file videotoolboxenc.c.

Referenced by vtenc_send_frame().

◆ create_encoder_dict_h264()

static int create_encoder_dict_h264	(	const AVFrame *	frame,
		CFDictionaryRef *	dict_out
	)

static

Definition at line 2547 of file videotoolboxenc.c.

Referenced by vtenc_send_frame().

◆ vtenc_send_frame()

static int vtenc_send_frame	(	AVCodecContext *	avctx,
		VTEncContext *	vtctx,
		const AVFrame *	frame
	)

static

Definition at line 2563 of file videotoolboxenc.c.

Referenced by vtenc_frame().

◆ vtenc_frame()

static av_cold int vtenc_frame	(	AVCodecContext *	avctx,
		AVPacket *	pkt,
		const AVFrame *	frame,
		int *	got_packet
	)

static

Definition at line 2621 of file videotoolboxenc.c.

◆ vtenc_close()

static av_cold int vtenc_close ( AVCodecContext * avctx )

static

Definition at line 2777 of file videotoolboxenc.c.

Variable Documentation

◆ kCVImageBufferColorPrimaries_ITU_R_2020

CFStringRef kCVImageBufferColorPrimaries_ITU_R_2020

Definition at line 83 of file videotoolboxenc.c.

Referenced by av_map_videotoolbox_color_primaries_from_av(), and loadVTEncSymbols().

◆ kCVImageBufferTransferFunction_ITU_R_2020

CFStringRef kCVImageBufferTransferFunction_ITU_R_2020

Definition at line 84 of file videotoolboxenc.c.

Referenced by av_map_videotoolbox_color_trc_from_av(), and loadVTEncSymbols().

◆ kCVImageBufferYCbCrMatrix_ITU_R_2020

CFStringRef kCVImageBufferYCbCrMatrix_ITU_R_2020

Definition at line 85 of file videotoolboxenc.c.

Referenced by av_map_videotoolbox_color_matrix_from_av(), and loadVTEncSymbols().

◆ kVTCompressionPropertyKey_H264EntropyMode

CFStringRef kVTCompressionPropertyKey_H264EntropyMode

Definition at line 87 of file videotoolboxenc.c.

Data Structures

Macros

Typedefs

Enumerations

Functions

Variables

Macro Definition Documentation

◆ TARGET_CPU_ARM64

◆ GET_SYM

◆ H264_PROFILE_CONSTRAINED_HIGH

◆ VE

◆ COMMON_OPTIONS

◆ OFFSET

Typedef Documentation

◆ getParameterSetAtIndex

Enumeration Type Documentation

◆ anonymous enum

◆ anonymous enum

◆ anonymous enum

◆ anonymous enum

◆ VTH264Entropy

Function Documentation

◆ loadVTEncSymbols()

◆ vt_dump_encoder()

◆ vtenc_populate_extradata()

◆ vt_release_num()

◆ set_async_error()

◆ clear_frame_queue()

◆ vtenc_reset()

◆ vtenc_q_pop()

◆ vtenc_q_push()

◆ count_nalus()

◆ get_cm_codec_type()

◆ get_params_size()

◆ copy_param_sets()

◆ set_extradata()

◆ vtenc_output_callback()

◆ get_length_code_size()

◆ get_vt_h264_profile_level()

◆ get_vt_hevc_profile_level()

◆ get_cv_pixel_format()

◆ add_color_attr()

◆ create_cv_pixel_buffer_info()

◆ get_cv_gamma()

◆ vtenc_qscale_enabled()

◆ set_encoder_property_or_log()

◆ set_encoder_int_property_or_log()

◆ vtenc_create_encoder()

◆ vtenc_configure_encoder()

◆ vtenc_init()

◆ vtenc_get_frame_info()

◆ is_post_sei_nal_type()

◆ find_sei_end()

◆ copy_emulation_prev()

◆ write_sei()

◆ copy_replace_length_codes()

◆ get_sei_msg_bytes()

◆ vtenc_cm_to_avpacket()

◆ get_cv_pixel_info()

◆ copy_avframe_to_pixel_buffer()

◆ create_cv_pixel_buffer()

◆ create_encoder_dict_h264()

◆ vtenc_send_frame()

◆ vtenc_frame()

◆ vtenc_close()

Variable Documentation

◆ kCVImageBufferColorPrimaries_ITU_R_2020

◆ kCVImageBufferTransferFunction_ITU_R_2020

◆ kCVImageBufferYCbCrMatrix_ITU_R_2020

◆ kVTCompressionPropertyKey_H264EntropyMode

◆ kVTH264EntropyMode_CAVLC

◆ kVTH264EntropyMode_CABAC

◆ kVTProfileLevel_H264_Baseline_4_0

◆ kVTProfileLevel_H264_Baseline_4_2

◆ kVTProfileLevel_H264_Baseline_5_0

◆ kVTProfileLevel_H264_Baseline_5_1

◆ kVTProfileLevel_H264_Baseline_5_2

◆ kVTProfileLevel_H264_Baseline_AutoLevel

◆ kVTProfileLevel_H264_Main_4_2

◆ kVTProfileLevel_H264_Main_5_1