Message ID | 20230905160221.2347008-2-mypopydev@gmail.com |
---|---|
State | New |
Headers | show |
Series | [FFmpeg-devel,v2,1/2] lavc/videotoolboxenc: Dump the encoder | expand |
On 5 Sep 2023, at 18:02, Jun Zhao wrote: > Get the encoder supported properties list, it will be used for > feature support checks. > > Signed-off-by: Jun Zhao <barryjzhao@tencent.com> > --- > libavcodec/videotoolboxenc.c | 68 ++++++++++++++++++++++++------------ > 1 file changed, 45 insertions(+), 23 deletions(-) > > diff --git a/libavcodec/videotoolboxenc.c b/libavcodec/videotoolboxenc.c > index 3512d35b73..0142342c58 100644 > --- a/libavcodec/videotoolboxenc.c > +++ b/libavcodec/videotoolboxenc.c > @@ -232,6 +232,7 @@ typedef struct VTEncContext { > AVClass *class; > enum AVCodecID codec_id; > VTCompressionSessionRef session; > + CFDictionaryRef supported_props; > CFStringRef ycbcr_matrix; > CFStringRef color_primaries; > CFStringRef transfer_function; > @@ -321,6 +322,34 @@ static void clear_frame_queue(VTEncContext *vtctx) > 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) > { > BufNode *info; > @@ -1110,6 +1139,18 @@ static int vtenc_create_encoder(AVCodecContext *avctx, > return AVERROR_EXTERNAL; > } > > + status = VTCopySupportedPropertyDictionaryForEncoder(avctx->width, > + avctx->height, > + codec_type, > + enc_info, > + NULL, > + &vtctx->supported_props); Only available starting macOS 10.13+, needs to be handled accordingly, I think. > + > + if (status != noErr) { > + av_log(avctx, AV_LOG_ERROR,"Error retrieving the supported property dictionary err=%"PRId64"\n", (int64_t)status); > + return AVERROR_EXTERNAL; > + } > + > // Dump the init encoder > { > CFStringRef encoderID = NULL; > @@ -1660,7 +1701,6 @@ static av_cold int vtenc_init(AVCodecContext *avctx) > // It can happen when user set avctx->profile directly. > if (vtctx->profile == FF_PROFILE_UNKNOWN) > vtctx->profile = avctx->profile; > - vtctx->session = NULL; > status = vtenc_configure_encoder(avctx); > if (status) return status; > > @@ -2429,8 +2469,8 @@ static int create_cv_pixel_buffer(AVCodecContext *avctx, > > vtstatus = VTCompressionSessionPrepareToEncodeFrames(vtctx->session); > if (vtstatus == kVTInvalidSessionErr) { > - CFRelease(vtctx->session); > - vtctx->session = NULL; > + vtenc_reset(vtctx); > + > status = vtenc_configure_encoder(avctx); > if (status == 0) > pix_buf_pool = VTCompressionSessionGetPixelBufferPool(vtctx->session); > @@ -2686,10 +2726,7 @@ static int vtenc_populate_extradata(AVCodecContext *avctx, > > pe_cleanup: > CVPixelBufferRelease(pix_buf); > - if(vtctx->session) > - CFRelease(vtctx->session); > - > - vtctx->session = NULL; > + vtenc_reset(vtctx); > vtctx->frame_ct_out = 0; > > av_assert0(status != 0 || (avctx->extradata && avctx->extradata_size > 0)); > @@ -2712,23 +2749,8 @@ static av_cold int vtenc_close(AVCodecContext *avctx) > clear_frame_queue(vtctx); > pthread_cond_destroy(&vtctx->cv_sample_sent); > pthread_mutex_destroy(&vtctx->lock); > - CFRelease(vtctx->session); > - vtctx->session = 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; > - } > + vtenc_reset(vtctx); > > return 0; > } > -- > 2.25.1 > > _______________________________________________ > ffmpeg-devel mailing list > ffmpeg-devel@ffmpeg.org > https://ffmpeg.org/mailman/listinfo/ffmpeg-devel > > To unsubscribe, visit link above, or email > ffmpeg-devel-request@ffmpeg.org with subject "unsubscribe".
On Wed, Sep 6, 2023 at 12:13 AM Marvin Scholz <epirat07@gmail.com> wrote: > > On 5 Sep 2023, at 18:02, Jun Zhao wrote: > > > Get the encoder supported properties list, it will be used for > > feature support checks. > > > > Signed-off-by: Jun Zhao <barryjzhao@tencent.com> > > --- > > libavcodec/videotoolboxenc.c | 68 ++++++++++++++++++++++++------------ > > 1 file changed, 45 insertions(+), 23 deletions(-) > > > > diff --git a/libavcodec/videotoolboxenc.c b/libavcodec/videotoolboxenc.c > > index 3512d35b73..0142342c58 100644 > > --- a/libavcodec/videotoolboxenc.c > > +++ b/libavcodec/videotoolboxenc.c > > @@ -232,6 +232,7 @@ typedef struct VTEncContext { > > AVClass *class; > > enum AVCodecID codec_id; > > VTCompressionSessionRef session; > > + CFDictionaryRef supported_props; > > CFStringRef ycbcr_matrix; > > CFStringRef color_primaries; > > CFStringRef transfer_function; > > @@ -321,6 +322,34 @@ static void clear_frame_queue(VTEncContext *vtctx) > > 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) > > { > > BufNode *info; > > @@ -1110,6 +1139,18 @@ static int vtenc_create_encoder(AVCodecContext *avctx, > > return AVERROR_EXTERNAL; > > } > > > > + status = VTCopySupportedPropertyDictionaryForEncoder(avctx->width, > > + avctx->height, > > + codec_type, > > + enc_info, > > + NULL, > > + &vtctx->supported_props); > > Only available starting macOS 10.13+, needs to be handled accordingly, I think. > Add the version check in local, thx, any other comments? > > + > > + if (status != noErr) { > > + av_log(avctx, AV_LOG_ERROR,"Error retrieving the supported property dictionary err=%"PRId64"\n", (int64_t)status); > > + return AVERROR_EXTERNAL; > > + } > > + > > // Dump the init encoder > > { > > CFStringRef encoderID = NULL; > > @@ -1660,7 +1701,6 @@ static av_cold int vtenc_init(AVCodecContext *avctx) > > // It can happen when user set avctx->profile directly. > > if (vtctx->profile == FF_PROFILE_UNKNOWN) > > vtctx->profile = avctx->profile; > > - vtctx->session = NULL; > > status = vtenc_configure_encoder(avctx); > > if (status) return status; > > > > @@ -2429,8 +2469,8 @@ static int create_cv_pixel_buffer(AVCodecContext *avctx, > > > > vtstatus = VTCompressionSessionPrepareToEncodeFrames(vtctx->session); > > if (vtstatus == kVTInvalidSessionErr) { > > - CFRelease(vtctx->session); > > - vtctx->session = NULL; > > + vtenc_reset(vtctx); > > + > > status = vtenc_configure_encoder(avctx); > > if (status == 0) > > pix_buf_pool = VTCompressionSessionGetPixelBufferPool(vtctx->session); > > @@ -2686,10 +2726,7 @@ static int vtenc_populate_extradata(AVCodecContext *avctx, > > > > pe_cleanup: > > CVPixelBufferRelease(pix_buf); > > - if(vtctx->session) > > - CFRelease(vtctx->session); > > - > > - vtctx->session = NULL; > > + vtenc_reset(vtctx); > > vtctx->frame_ct_out = 0; > > > > av_assert0(status != 0 || (avctx->extradata && avctx->extradata_size > 0)); > > @@ -2712,23 +2749,8 @@ static av_cold int vtenc_close(AVCodecContext *avctx) > > clear_frame_queue(vtctx); > > pthread_cond_destroy(&vtctx->cv_sample_sent); > > pthread_mutex_destroy(&vtctx->lock); > > - CFRelease(vtctx->session); > > - vtctx->session = 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; > > - } > > + vtenc_reset(vtctx); > > > > return 0; > > } > > -- > > 2.25.1
diff --git a/libavcodec/videotoolboxenc.c b/libavcodec/videotoolboxenc.c index 3512d35b73..0142342c58 100644 --- a/libavcodec/videotoolboxenc.c +++ b/libavcodec/videotoolboxenc.c @@ -232,6 +232,7 @@ typedef struct VTEncContext { AVClass *class; enum AVCodecID codec_id; VTCompressionSessionRef session; + CFDictionaryRef supported_props; CFStringRef ycbcr_matrix; CFStringRef color_primaries; CFStringRef transfer_function; @@ -321,6 +322,34 @@ static void clear_frame_queue(VTEncContext *vtctx) 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) { BufNode *info; @@ -1110,6 +1139,18 @@ static int vtenc_create_encoder(AVCodecContext *avctx, return AVERROR_EXTERNAL; } + 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; + } + // Dump the init encoder { CFStringRef encoderID = NULL; @@ -1660,7 +1701,6 @@ static av_cold int vtenc_init(AVCodecContext *avctx) // It can happen when user set avctx->profile directly. if (vtctx->profile == FF_PROFILE_UNKNOWN) vtctx->profile = avctx->profile; - vtctx->session = NULL; status = vtenc_configure_encoder(avctx); if (status) return status; @@ -2429,8 +2469,8 @@ static int create_cv_pixel_buffer(AVCodecContext *avctx, vtstatus = VTCompressionSessionPrepareToEncodeFrames(vtctx->session); if (vtstatus == kVTInvalidSessionErr) { - CFRelease(vtctx->session); - vtctx->session = NULL; + vtenc_reset(vtctx); + status = vtenc_configure_encoder(avctx); if (status == 0) pix_buf_pool = VTCompressionSessionGetPixelBufferPool(vtctx->session); @@ -2686,10 +2726,7 @@ static int vtenc_populate_extradata(AVCodecContext *avctx, pe_cleanup: CVPixelBufferRelease(pix_buf); - if(vtctx->session) - CFRelease(vtctx->session); - - vtctx->session = NULL; + vtenc_reset(vtctx); vtctx->frame_ct_out = 0; av_assert0(status != 0 || (avctx->extradata && avctx->extradata_size > 0)); @@ -2712,23 +2749,8 @@ static av_cold int vtenc_close(AVCodecContext *avctx) clear_frame_queue(vtctx); pthread_cond_destroy(&vtctx->cv_sample_sent); pthread_mutex_destroy(&vtctx->lock); - CFRelease(vtctx->session); - vtctx->session = 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; - } + vtenc_reset(vtctx); return 0; }
Get the encoder supported properties list, it will be used for feature support checks. Signed-off-by: Jun Zhao <barryjzhao@tencent.com> --- libavcodec/videotoolboxenc.c | 68 ++++++++++++++++++++++++------------ 1 file changed, 45 insertions(+), 23 deletions(-)