highperfocused/ffmpeg
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

lavc/videotoolboxenc: Get the encoder supported properties

Browse Source 
Get the encoder supported properties list, it will be used for
feature support checks.

Signed-off-by: Jun Zhao <barryjzhao@tencent.com>
This commit is contained in:
Jun Zhao
2023-09-04 10:48:01 +08:00
committed by Jun Zhao
parent 213cba9696
commit 5c635b7d8c
1 changed files with 49 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

72
libavcodec/videotoolboxenc.c
View File

@@ -232,6 +232,7 @@ typedef struct VTEncContext {
AVClass *class; AVClass *class;
enum AVCodecID codec_id; enum AVCodecID codec_id;
VTCompressionSessionRef session; VTCompressionSessionRef session;
CFDictionaryRef supported_props;
CFStringRef ycbcr_matrix; CFStringRef ycbcr_matrix;
CFStringRef color_primaries; CFStringRef color_primaries;
CFStringRef transfer_function; CFStringRef transfer_function;
@@ -321,6 +322,34 @@ static void clear_frame_queue(VTEncContext *vtctx)
set_async_error(vtctx, 0); set_async_error(vtctx, 0);
} }
static void vtenc_reset(VTEncContext *vtctx)
{
if (vtctx->session) {
CFRelease(vtctx->session);
vtctx->session = NULL;
}
if (vtctx->supported_props) {
CFRelease(vtctx->supported_props);
vtctx->supported_props = NULL;
}
if (vtctx->color_primaries) {
CFRelease(vtctx->color_primaries);
vtctx->color_primaries = NULL;
}
if (vtctx->transfer_function) {
CFRelease(vtctx->transfer_function);
vtctx->transfer_function = NULL;
}
if (vtctx->ycbcr_matrix) {
CFRelease(vtctx->ycbcr_matrix);
vtctx->ycbcr_matrix = NULL;
}
}
static int vtenc_q_pop(VTEncContext *vtctx, bool wait, CMSampleBufferRef *buf, ExtraSEI **sei) static int vtenc_q_pop(VTEncContext *vtctx, bool wait, CMSampleBufferRef *buf, ExtraSEI **sei)
{ {
BufNode *info; BufNode *info;
@@ -1110,6 +1139,22 @@ static int vtenc_create_encoder(AVCodecContext *avctx,
return AVERROR_EXTERNAL; return AVERROR_EXTERNAL;
} }
#if defined (MAC_OS_X_VERSION_10_13) && (MAC_OS_X_VERSION_MAX_ALLOWED >= MAC_OS_X_VERSION_10_13)
if (__builtin_available(macOS 10.13, *)) {
status = VTCopySupportedPropertyDictionaryForEncoder(avctx->width,
avctx->height,
codec_type,
enc_info,
NULL,
&vtctx->supported_props);
if (status != noErr) {
av_log(avctx, AV_LOG_ERROR,"Error retrieving the supported property dictionary err=%"PRId64"\n", (int64_t)status);
return AVERROR_EXTERNAL;
}
}
#endif
// Dump the init encoder // Dump the init encoder
{ {
CFStringRef encoderID = NULL; CFStringRef encoderID = NULL;
@@ -1662,7 +1707,6 @@ static av_cold int vtenc_init(AVCodecContext *avctx)
// It can happen when user set avctx->profile directly. // It can happen when user set avctx->profile directly.
if (vtctx->profile == AV_PROFILE_UNKNOWN) if (vtctx->profile == AV_PROFILE_UNKNOWN)
vtctx->profile = avctx->profile; vtctx->profile = avctx->profile;
vtctx->session = NULL;
status = vtenc_configure_encoder(avctx); status = vtenc_configure_encoder(avctx);
if (status) return status; if (status) return status;
@@ -2431,8 +2475,8 @@ static int create_cv_pixel_buffer(AVCodecContext *avctx,
vtstatus = VTCompressionSessionPrepareToEncodeFrames(vtctx->session); vtstatus = VTCompressionSessionPrepareToEncodeFrames(vtctx->session);
if (vtstatus == kVTInvalidSessionErr) { if (vtstatus == kVTInvalidSessionErr) {
CFRelease(vtctx->session); vtenc_reset(vtctx);
vtctx->session = NULL;
status = vtenc_configure_encoder(avctx); status = vtenc_configure_encoder(avctx);
if (status == 0) if (status == 0)
pix_buf_pool = VTCompressionSessionGetPixelBufferPool(vtctx->session); pix_buf_pool = VTCompressionSessionGetPixelBufferPool(vtctx->session);
@@ -2688,10 +2732,7 @@ static int vtenc_populate_extradata(AVCodecContext *avctx,
pe_cleanup: pe_cleanup:
CVPixelBufferRelease(pix_buf); CVPixelBufferRelease(pix_buf);
if(vtctx->session) vtenc_reset(vtctx);
CFRelease(vtctx->session);
vtctx->session = NULL;
vtctx->frame_ct_out = 0; vtctx->frame_ct_out = 0;
av_assert0(status != 0 || (avctx->extradata && avctx->extradata_size > 0)); av_assert0(status != 0 || (avctx->extradata && avctx->extradata_size > 0));
@@ -2714,23 +2755,8 @@ static av_cold int vtenc_close(AVCodecContext *avctx)
clear_frame_queue(vtctx); clear_frame_queue(vtctx);
pthread_cond_destroy(&vtctx->cv_sample_sent); pthread_cond_destroy(&vtctx->cv_sample_sent);
pthread_mutex_destroy(&vtctx->lock); pthread_mutex_destroy(&vtctx->lock);
CFRelease(vtctx->session);
vtctx->session = NULL;
if (vtctx->color_primaries) { vtenc_reset(vtctx);
CFRelease(vtctx->color_primaries);
vtctx->color_primaries = NULL;
}
if (vtctx->transfer_function) {
CFRelease(vtctx->transfer_function);
vtctx->transfer_function = NULL;
}
if (vtctx->ycbcr_matrix) {
CFRelease(vtctx->ycbcr_matrix);
vtctx->ycbcr_matrix = NULL;
}
return 0; return 0;
} }