#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 "internal.h"
#include <pthread.h>
#include "h264.h"
#include "h264_sei.h"
#include <dlfcn.h>

Data Structures
struct	ExtraSEI

struct	BufNode

struct	VTEncContext

Macros
#define	GET_SYM(symbol, defaultVal)

#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	VT_H264Profile { H264_PROF_AUTO, H264_PROF_BASELINE, H264_PROF_MAIN, H264_PROF_HIGH, H264_PROF_COUNT }

enum	VTH264Entropy { VT_ENTROPY_NOT_SET, VT_CAVLC, VT_CABAC }

enum	VT_HEVCProfile { HEVC_PROF_AUTO, HEVC_PROF_MAIN, HEVC_PROF_MAIN10, HEVC_PROF_COUNT }

Functions
static void	loadVTEncSymbols ()

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 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 (enum AVCodecID id)

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_color_primaries (AVCodecContext avctx, CFStringRef primaries)

static int	get_cv_transfer_function (AVCodecContext avctx, CFStringRef transfer_fnc, CFNumberRef *gamma_level)

static int	get_cv_ycbcr_matrix (AVCodecContext avctx, CFStringRef matrix)

static int	vtenc_create_encoder (AVCodecContext avctx, CMVideoCodecType codec_type, CFStringRef profile_level, CFNumberRef gamma_level, CFDictionaryRef enc_info, CFDictionaryRef pixel_buffer_info, 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 void	free_avframe (void release_ctx, const void data, size_t size, size_t plane_count, const void *plane_addresses[])

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

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_HEVC_Main_AutoLevel

CFStringRef kVTProfileLevel_HEVC_Main10_AutoLevel

CFStringRef kVTCompressionPropertyKey_RealTime

CFStringRef kVTVideoEncoderSpecification_EnableHardwareAcceleratedVideoEncoder

CFStringRef kVTVideoEncoderSpecification_RequireHardwareAcceleratedVideoEncoder

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	pix_fmts []

static const AVOption	h264_options []

static const AVClass	h264_videotoolbox_class

AVCodec	ff_h264_videotoolbox_encoder

static const AVOption	hevc_options []

static const AVClass	hevc_videotoolbox_class

AVCodec	ff_hevc_videotoolbox_encoder

Macro Definition Documentation

#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 90 of file videotoolboxenc.c.

Enumerator
H264_PROF_AUTO
H264_PROF_BASELINE
H264_PROF_MAIN
H264_PROF_HIGH
H264_PROF_COUNT

Enumerator
HEVC_PROF_AUTO
HEVC_PROF_MAIN
HEVC_PROF_MAIN10
HEVC_PROF_COUNT

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

Enumerator
VT_ENTROPY_NOT_SET
VT_CAVLC
VT_CABAC

Data Structures

Macros

Typedefs

Enumerations

Functions

Variables

Macro Definition Documentation

Typedef Documentation

Enumeration Type Documentation

Function Documentation

Variable Documentation