From mboxrd@z Thu Jan 1 00:00:00 1970 Return-Path: Received: from ffbox0-bg.mplayerhq.hu (ffbox0-bg.ffmpeg.org [79.124.17.100]) by master.gitmailbox.com (Postfix) with ESMTP id 18CDA446E9 for ; Mon, 24 Oct 2022 14:38:11 +0000 (UTC) Received: from [127.0.1.1] (localhost [127.0.0.1]) by ffbox0-bg.mplayerhq.hu (Postfix) with ESMTP id 5787468BBE5; Mon, 24 Oct 2022 17:38:09 +0300 (EEST) Received: from mail-qv1-f41.google.com (mail-qv1-f41.google.com [209.85.219.41]) by ffbox0-bg.mplayerhq.hu (Postfix) with ESMTPS id 5DE1268BB6B for ; Mon, 24 Oct 2022 17:38:02 +0300 (EEST) Received: by mail-qv1-f41.google.com with SMTP id i12so6629773qvs.2 for ; Mon, 24 Oct 2022 07:38:02 -0700 (PDT) DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=gmail.com; s=20210112; h=content-transfer-encoding:in-reply-to:content-language:references :to:subject:user-agent:mime-version:date:message-id:from:from:to:cc :subject:date:message-id:reply-to; bh=+FEVdDksk9XnL+RO7D1pmgaWptfkXjL5uZttogFbSy0=; b=RX9Blw5JI22VeAkczAKwMZxtF6P2UpiV+S3/a2bLFln6an5YGiwOT0HUPZBNqavPtQ SPFQZXNGcBRTZiT52e9bsJou6i45f1wF7ZJbBFoIlLzp3LShZ5O1PRsFkKZZI++IU8s2 baMSatPsXNxJ3dSXRL6R6w29eBp4GGIldf7jkBAW2/1Nn04iC1AvtZE/8ZkAQ/KEc78p CW1QUudaMph5DMzVsjetbHtyFP/mCy/tpQ41BU4U4QtylZ3GJq4/i5nV0pgFO6Ej3NeI Wpe8M+jvKh9zVAUQORqzBZX9ERsCOlV1tu/4JuRkWPuKTDNA1VUDQtrmXfzmXAgn3RzP g5SQ== X-Google-DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=1e100.net; s=20210112; h=content-transfer-encoding:in-reply-to:content-language:references :to:subject:user-agent:mime-version:date:message-id:from :x-gm-message-state:from:to:cc:subject:date:message-id:reply-to; bh=+FEVdDksk9XnL+RO7D1pmgaWptfkXjL5uZttogFbSy0=; b=wSHJytj7sNDUzgLm09SDamkYulunIZ0P2JstbjXKUZLDthlRAuEjlv0qyqEhe/ieuc /kyE5V8k+Ph7bA5KICWvWPWO4i2w7krweBW0NR+PxdtV+kwUvwu/jHhAL0YQH3nUAgwA 4OvR4TO3qUjSZs0uvx63KmHPiSgUGmJPcEDXoegX4DsptOOh4AKhaim31slg4ooNQXSE OaFQux+VaiyWP+KUoIDoLhZFbKOyYWVjMs1J71Bk6xao28RBwnzlcVWe6BwwI3U4YKnx dH+7tDkZqYUB/oM1YPEZOoJjO7mENew1ftJaZAGj74ghT6ZDPnywPvvZAEMp0pwecvN5 MFew== X-Gm-Message-State: ACrzQf2mmeqGL1Q9Ppx6uVZP8ucQSqJrSVA2YeyYYdkrbL5yhEPebkAg JECPt3u4HVogqErPD5azN0hmZvLpCIA= X-Google-Smtp-Source: AMsMyM4QOfFzTLwY0kyL4uKKS+1TYZm3hJv215Ya7t/PWjB+esm3cq1OGaRds5MvMgqYsx3m+rvBww== X-Received: by 2002:a05:6214:c6d:b0:4b1:c751:ac7 with SMTP id t13-20020a0562140c6d00b004b1c7510ac7mr28944436qvj.96.1666622280211; Mon, 24 Oct 2022 07:38:00 -0700 (PDT) Received: from [192.168.1.38] (modemcable017.92-130-66.mc.videotron.ca. [66.130.92.17]) by smtp.gmail.com with ESMTPSA id 3-20020ac85903000000b00399d5d564b7sm30631qty.56.2022.10.24.07.37.59 for (version=TLS1_3 cipher=TLS_AES_128_GCM_SHA256 bits=128/128); Mon, 24 Oct 2022 07:37:59 -0700 (PDT) From: quietvoid X-Google-Original-From: quietvoid Message-ID: <6329329f-75ac-acfe-c517-9c9be10920dc@gmail.com> Date: Mon, 24 Oct 2022 10:37:59 -0400 MIME-Version: 1.0 User-Agent: Mozilla/5.0 (X11; Linux x86_64; rv:102.0) Gecko/20100101 Thunderbird/102.4.0 To: ffmpeg-devel@ffmpeg.org References: <20221024140645.4945-1-thomas.ff@spin-digital.com> <20221024140645.4945-9-thomas.ff@spin-digital.com> Content-Language: en-US In-Reply-To: <20221024140645.4945-9-thomas.ff@spin-digital.com> Subject: Re: [FFmpeg-devel] [PATCH v2 08/10] avcodec: add external encoder libvvenc for H266/VVC X-BeenThere: ffmpeg-devel@ffmpeg.org X-Mailman-Version: 2.1.29 Precedence: list List-Id: FFmpeg development discussions and patches List-Unsubscribe: , List-Archive: List-Post: List-Help: List-Subscribe: , Reply-To: FFmpeg development discussions and patches Content-Transfer-Encoding: 7bit Content-Type: text/plain; charset="us-ascii"; Format="flowed" Errors-To: ffmpeg-devel-bounces@ffmpeg.org Sender: "ffmpeg-devel" Archived-At: List-Archive: List-Post: On 24/10/2022 10.06, Thomas Siedel wrote: > Add external encoder VVenC for H266/VVC encoding. > Register new encoder libvvencc > Add libvvenc to wrap the vvenc interface into ffmpeg > libvvenc implements encoder option: > preset,qp,period,subjopt,vvenc-params,levelidc,tier > Enable encoder by adding --enable-libvvenc in configure step > > Signed-off-by: Thomas Siedel > --- > configure | 5 + > libavcodec/Makefile | 1 + > libavcodec/allcodecs.c | 1 + > libavcodec/libvvenc.c | 442 +++++++++++++++++++++++++++++++++++++++++ > 4 files changed, 449 insertions(+) > create mode 100644 libavcodec/libvvenc.c > > diff --git a/configure b/configure > index 978f15f772..f8f7965371 100755 > --- a/configure > +++ b/configure > @@ -289,6 +289,7 @@ External library support: > native implementation exists [no] > --enable-libvpx enable VP8 and VP9 de/encoding via libvpx [no] > --enable-libvvdec enable VVC decoding via vvdec [no] > + --enable-libvvenc enable VVC encoding via vvenc [no] > --enable-libwebp enable WebP encoding via libwebp [no] > --enable-libx264 enable H.264 encoding via x264 [no] > --enable-libx265 enable HEVC encoding via x265 [no] > @@ -1877,6 +1878,7 @@ EXTERNAL_LIBRARY_LIST=" > libvorbis > libvpx > libvvdec > + libvvenc > libwebp > libxml2 > libzimg > @@ -3409,6 +3411,8 @@ libvpx_vp9_decoder_deps="libvpx" > libvpx_vp9_encoder_deps="libvpx" > libvvdec_decoder_deps="libvvdec" > libvvdec_decoder_select="vvc_mp4toannexb_bsf" > +libvvenc_encoder_deps="libvvenc" > +libvvenc_encoder_select="atsc_a53" > libwebp_encoder_deps="libwebp" > libwebp_anim_encoder_deps="libwebp" > libx262_encoder_deps="libx262" > @@ -6740,6 +6744,7 @@ enabled libvpx && { > fi > } > enabled libvvdec && require_pkg_config libvvdec "libvvdec >= 1.6.0" > "vvdec/vvdec.h" vvdec_get_version > +enabled libvvenc && require_pkg_config libvvenc "libvvenc >= 1.6.1" > "vvenc/vvenc.h" vvenc_get_version > enabled libwebp && { > enabled libwebp_encoder && require_pkg_config libwebp "libwebp >= > 0.2.0" webp/encode.h WebPGetEncoderVersion > diff --git a/libavcodec/Makefile b/libavcodec/Makefile > index f4cdbc9be1..badd266e66 100644 > --- a/libavcodec/Makefile > +++ b/libavcodec/Makefile > @@ -1105,6 +1105,7 @@ OBJS-$(CONFIG_LIBVPX_VP8_ENCODER) += libvpxenc.o > OBJS-$(CONFIG_LIBVPX_VP9_DECODER) += libvpxdec.o libvpx.o > OBJS-$(CONFIG_LIBVPX_VP9_ENCODER) += libvpxenc.o libvpx.o > OBJS-$(CONFIG_LIBVVDEC_DECODER) += libvvdec.o vvc_parse_extradata.o > vvc_paramset.o > +OBJS-$(CONFIG_LIBVVENC_ENCODER) += libvvenc.o > OBJS-$(CONFIG_LIBWEBP_ENCODER) += libwebpenc_common.o libwebpenc.o > OBJS-$(CONFIG_LIBWEBP_ANIM_ENCODER) += libwebpenc_common.o > libwebpenc_animencoder.o > OBJS-$(CONFIG_LIBX262_ENCODER) += libx264.o > diff --git a/libavcodec/allcodecs.c b/libavcodec/allcodecs.c > index 9813b291f4..73c36f3134 100644 > --- a/libavcodec/allcodecs.c > +++ b/libavcodec/allcodecs.c > @@ -794,6 +794,7 @@ extern const FFCodec ff_libvpx_vp8_decoder; > extern FFCodec ff_libvpx_vp9_encoder; > extern FFCodec ff_libvpx_vp9_decoder; > extern const FFCodec ff_libvvdec_decoder; > +extern const FFCodec ff_libvvenc_encoder; > /* preferred over libwebp */ > extern const FFCodec ff_libwebp_anim_encoder; > extern const FFCodec ff_libwebp_encoder; > diff --git a/libavcodec/libvvenc.c b/libavcodec/libvvenc.c > new file mode 100644 > index 0000000000..508fe6ceff > --- /dev/null > +++ b/libavcodec/libvvenc.c > @@ -0,0 +1,442 @@ > +/* > + * H.266 encoding using the VVenC library > + * > + * Copyright (C) 2022, Thomas Siedel > + * > + * This file is part of FFmpeg. > + * > + * FFmpeg is free software; you can redistribute it and/or > + * modify it under the terms of the GNU Lesser General Public > + * License as published by the Free Software Foundation; either > + * version 2.1 of the License, or (at your option) any later version. > + * > + * FFmpeg is distributed in the hope that it will be useful, > + * but WITHOUT ANY WARRANTY; without even the implied warranty of > + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU > + * Lesser General Public License for more details. > + * > + * You should have received a copy of the GNU Lesser General Public > + * License along with FFmpeg; if not, write to the Free Software > + * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA > 02110-1301 USA > + */ > + > +#include "config_components.h" > + > +#include > +#include > + > +#include "avcodec.h" > +#include "codec_internal.h" > +#include "encode.h" > +#include "internal.h" > +#include "packet_internal.h" > +#include "profiles.h" > + > +#include "libavutil/avutil.h" > +#include "libavutil/pixdesc.h" > +#include "libavutil/opt.h" > +#include "libavutil/common.h" > +#include "libavutil/imgutils.h" > +#include "libavutil/frame.h" > +#include "libavutil/log.h" > + > +typedef struct VVenCOptions { > + int preset; // preset 0: faster 4: slower > + int qp; // quantization parameter 0-63 > + int subjectiveOptimization; // perceptually motivated QP adaptation, > XPSNR based > + int intraRefreshSec; // intra period/refresh in seconds > + int levelIdc; // vvc level_idc > + int tier; // vvc tier > + AVDictionary *vvenc_opts; > +} VVenCOptions; > + > +typedef struct VVenCContext { > + AVClass *av_class; > + VVenCOptions options; // encoder options > + vvencEncoder *vvencEnc; > + vvencAccessUnit *pAU; > + bool encodeDone; > +}VVenCContext; > + > + > +static av_cold void ff_vvenc_log_callback(void *avctx, int level, > const char *fmt, va_list args ) > +{ > + vfprintf( level == 1 ? stderr : stdout, fmt, args ); > +} > + > +static void ff_vvenc_internalLog( void* ctx, int level, const char* > fmt, ... ) > +{ > + va_list args; > + va_start( args, fmt ); > + ff_vvenc_log_callback( ctx, level, fmt, args ); > + va_end( args ); > +} > + > +static av_cold int ff_vvenc_encode_init(AVCodecContext *avctx) > +{ > + int ret; > + int framerate, qp, parse_ret; > + VVenCContext *s; > + vvenc_config params; > + vvencPresetMode preset; > + AVDictionaryEntry *en; > + char statsfile[1024]="vvenc-rcstats.json"; > + > + s = (VVenCContext*)avctx->priv_data; > + qp = (vvencPresetMode)s->options.qp; > + preset = (vvencPresetMode)s->options.preset; > + > + if( avctx->flags & AV_CODEC_FLAG_INTERLACED_DCT ) { > + av_log(avctx, AV_LOG_ERROR, "ff_vvenc_encode_init::init() interlaced > encoding not supported yet\n"); > + return AVERROR_INVALIDDATA; > + } > + > + vvenc_config_default( ¶ms ); > + > + // set desired encoding options > + framerate = avctx->time_base.den / avctx->time_base.num; > + vvenc_init_default ( ¶ms, avctx->width, avctx->height, > framerate, avctx->bit_rate, qp, preset ); > + params.m_FrameRate = avctx->time_base.den; > + params.m_FrameScale = avctx->time_base.num; > + > + params.m_verbosity = VVENC_VERBOSE; > + if ( av_log_get_level() >= AV_LOG_DEBUG ) > + params.m_verbosity = VVENC_DETAILS; > + else if( av_log_get_level() >= AV_LOG_VERBOSE ) > + params.m_verbosity = VVENC_NOTICE; // output per picture info > + else if( av_log_get_level() >= AV_LOG_INFO ) > + params.m_verbosity = VVENC_WARNING; // ffmpeg default ffmpeg loglevel > + else > + params.m_verbosity = VVENC_SILENT; > + > + if( avctx->ticks_per_frame == 1 ) { > + params.m_TicksPerSecond = -1; // auto mode for ticks per frame = 1 > + } > + else{ > + params.m_TicksPerSecond = > ceil((avctx->time_base.den/(double)avctx->time_base.num)*(double)avctx->ticks_per_frame); > + } > + > + if( avctx->thread_count > 0 ) > + params.m_numThreads = avctx->thread_count; > + > + // GOP settings (IDR/CRA) > + if( avctx->flags & AV_CODEC_FLAG_CLOSED_GOP ) > + params.m_DecodingRefreshType = VVENC_DRT_IDR; > + > + if( avctx->gop_size == 1 ) { > + params.m_GOPSize = 1; > + params.m_IntraPeriod = 1; > + } > + else { > + params.m_IntraPeriodSec = s->options.intraRefreshSec; > + } > + > + params.m_usePerceptQPA = s->options.subjectiveOptimization; > + params.m_level = (vvencLevel)s->options.levelIdc; > + params.m_levelTier = (vvencTier)s->options.tier; > + > + params.m_AccessUnitDelimiter = true; > + > + params.m_internChromaFormat = VVENC_CHROMA_420; > + switch( avctx->pix_fmt ) { > + case AV_PIX_FMT_YUV420P : > + params.m_inputBitDepth[0] = 8; > + break; > + case AV_PIX_FMT_YUV420P10LE: > + params.m_inputBitDepth[0] = 10; > + break; > + default: { > + av_log(avctx, AV_LOG_ERROR, "unsupported pixel format %s, choose > yuv420p or yuv420p10le\n", av_get_pix_fmt_name(avctx->pix_fmt)); > + return AVERROR(EINVAL); > + break; > + } > + }; > + > + if( avctx->color_primaries != AVCOL_PRI_UNSPECIFIED ) > + params.m_colourPrimaries = (int)avctx->color_primaries; > + if( avctx->colorspace != AVCOL_SPC_UNSPECIFIED ) > + params.m_matrixCoefficients = (int)avctx->colorspace; > + if( avctx->color_trc != AVCOL_TRC_UNSPECIFIED ) { > + params.m_transferCharacteristics = (int)avctx->color_trc; > + > + if ( params.m_transferCharacteristics == 16 ) > + params.m_HdrMode = (params.m_colourPrimaries == 9) ? VVENC_HDR_PQ_BT2020 > + : VVENC_HDR_PQ; > + else if ( params.m_transferCharacteristics == 14 && > params.m_colourPrimaries == 9 ) > + params.m_HdrMode = VVENC_HDR_HLG_BT2020; Should probably still be using the AVColor enum in the conditions instead of the values. > + else if ( params.m_transferCharacteristics == 1 && > params.m_colourPrimaries == 1 && > + params.m_matrixCoefficients == 1 ) > + params.m_HdrMode = VVENC_HDR_HLG; Is this not supposed to be VVENC_HDR_OFF? HLG should be matching TRC 14, not 1. > + } > + > + params.m_RCNumPasses = 1; > + en = NULL; > + while ((en = av_dict_get(s->options.vvenc_opts, "", en, > AV_DICT_IGNORE_SUFFIX))) { > + av_log(avctx, AV_LOG_DEBUG, "vvenc_set_param: '%s:%s'\n", en->key, > en->value); > + parse_ret = vvenc_set_param(¶ms, en->key, en->value); > + switch (parse_ret) > + { > + case VVENC_PARAM_BAD_NAME: > + av_log(avctx, AV_LOG_WARNING,"Unknown vvenc option: %s.\n", en->key); > + break; > + case VVENC_PARAM_BAD_VALUE: > + av_log(avctx, AV_LOG_WARNING, > + "Invalid vvenc value for %s: %s.\n", en->key, en->value); > + break; > + default: > + break; > + } > + > + if (memcmp( en->key, "rcstatsfile", 11 ) == 0 || > + memcmp( en->key, "RCStatsFile", 11 ) == 0) { > + strncpy(statsfile, en->value, sizeof(statsfile)-1); > + statsfile[sizeof(statsfile)-1] = '\0'; > + } > + } > + > + if( params.m_RCPass != -1 && params.m_RCNumPasses == 1 ) > + params.m_RCNumPasses = 2; // enable 2pass mode > + > + s->vvencEnc = vvenc_encoder_create(); > + if( NULL == s->vvencEnc ) { > + av_log(avctx, AV_LOG_ERROR, "cannot create vvc encoder (vvenc)\n"); > + return AVERROR(ENOMEM); > + } > + > + vvenc_set_msg_callback( ¶ms, s->vvencEnc, ff_vvenc_log_callback ); > + ret = vvenc_encoder_open( s->vvencEnc, ¶ms ); > + if( 0 != ret ) { > + av_log(avctx, AV_LOG_ERROR, "cannot open vvc encoder (vvenc): %s\n", > vvenc_get_last_error(s->vvencEnc)); > + vvenc_encoder_close( s->vvencEnc ); > + return AVERROR(EINVAL); > + } > + > + vvenc_get_config( s->vvencEnc, ¶ms ); // get the adapted config > + > + if( params.m_verbosity >= VVENC_DETAILS && av_log_get_level() < > AV_LOG_DEBUG ) { > + ff_vvenc_internalLog( avctx, params.m_verbosity, "vvenc version: > %s\n", vvenc_get_version() ); > + ff_vvenc_internalLog( avctx, params.m_verbosity, "vvencinfo:\n%s\n", > vvenc_get_config_as_string( ¶ms, VVENC_DETAILS) ); > + } > + else { > + av_log(avctx, AV_LOG_DEBUG, "vvenc version: %s\n", vvenc_get_version()); > + av_log(avctx, AV_LOG_DEBUG, "vvencinfo:\n%s\n", > vvenc_get_config_as_string( ¶ms, VVENC_DETAILS)); > + } > + > + if( params.m_RCNumPasses == 2 ) { > + ret = vvenc_init_pass( s->vvencEnc, params.m_RCPass-1, &statsfile[0] ); > + if( 0 != ret ) { > + av_log(avctx, AV_LOG_ERROR, "cannot init pass %d for vvc encoder > (vvenc): %s\n", params.m_RCPass, vvenc_get_last_error(s->vvencEnc)); > + vvenc_encoder_close( s->vvencEnc ); > + return AVERROR(EINVAL); > + } > + } > + > + s->pAU = vvenc_accessUnit_alloc(); > + vvenc_accessUnit_alloc_payload( s->pAU, avctx->width * avctx->height ); > + > + if (avctx->flags & AV_CODEC_FLAG_GLOBAL_HEADER) { > + ret = vvenc_get_headers( s->vvencEnc, s->pAU ); > + if( 0 != ret ) { > + av_log(avctx, AV_LOG_ERROR, "cannot get headers (SPS,PPS) from vvc > encoder(vvenc): %s\n", vvenc_get_last_error(s->vvencEnc)); > + vvenc_encoder_close( s->vvencEnc ); > + return AVERROR(EINVAL); > + } > + > + if (s->pAU->payloadUsedSize <= 0) { > + vvenc_encoder_close( s->vvencEnc ); > + return AVERROR_INVALIDDATA; > + } > + > + avctx->extradata_size = s->pAU->payloadUsedSize; > + avctx->extradata = av_malloc(avctx->extradata_size + > AV_INPUT_BUFFER_PADDING_SIZE); > + if (!avctx->extradata) { > + av_log(avctx, AV_LOG_ERROR, > + "Cannot allocate VVC header of size %d.\n", avctx->extradata_size); > + vvenc_encoder_close( s->vvencEnc ); > + return AVERROR(ENOMEM); > + } > + > + memcpy(avctx->extradata, s->pAU->payload, avctx->extradata_size); > + memset(avctx->extradata + avctx->extradata_size, 0, > AV_INPUT_BUFFER_PADDING_SIZE); > + } > + s->encodeDone = false; > + return 0; > +} > + > +static av_cold int ff_vvenc_encode_close(AVCodecContext *avctx) > +{ > + VVenCContext *s = (VVenCContext*)avctx->priv_data; > + if( s->vvencEnc ) { > + if( av_log_get_level() >= AV_LOG_VERBOSE ) > + vvenc_print_summary(s->vvencEnc); > + > + if( 0 != vvenc_encoder_close(s->vvencEnc) ) { > + av_log(avctx, AV_LOG_ERROR, "cannot close vvenc\n" ); > + return -1; > + } > + } > + > + vvenc_accessUnit_free ( s->pAU, true ); > + > + return 0; > +} > + > +static av_cold int ff_vvenc_encode_frame( AVCodecContext *avctx, > AVPacket *pkt,const AVFrame *frame, int *got_packet ) > +{ > + VVenCContext *s = (VVenCContext*)avctx->priv_data; > + vvencYUVBuffer* pyuvbuf; > + vvencYUVBuffer yuvbuf; > + int pict_type; > + int ret; > + > + pyuvbuf = NULL; > + if (frame) { > + if( avctx->pix_fmt == AV_PIX_FMT_YUV420P10LE ) { > + vvenc_YUVBuffer_default( &yuvbuf ); > + yuvbuf.planes[0].ptr = (int16_t*)frame->data[0]; > + yuvbuf.planes[1].ptr = (int16_t*)frame->data[1]; > + yuvbuf.planes[2].ptr = (int16_t*)frame->data[2]; > + > + yuvbuf.planes[0].width = frame->width; > + yuvbuf.planes[0].height = frame->height; > + yuvbuf.planes[0].stride = frame->linesize[0] >> 1; // stride is used > in samples (16bit) in vvenc, ffmpeg uses stride in bytes > + > + yuvbuf.planes[1].width = frame->width >>1; > + yuvbuf.planes[1].height = frame->height >>1; > + yuvbuf.planes[1].stride = frame->linesize[1] >> 1; > + > + yuvbuf.planes[2].width = frame->width >>1; > + yuvbuf.planes[2].height = frame->height >>1; > + yuvbuf.planes[2].stride = frame->linesize[2] >> 1; > + > + yuvbuf.cts = frame->pts; > + yuvbuf.ctsValid = true; > + pyuvbuf = &yuvbuf; > + } > + else { > + av_log(avctx, AV_LOG_ERROR, "unsupported input colorspace! input > must be yuv420p10le"); > + return AVERROR(EINVAL); > + } > + } > + > + if( ! s->encodeDone ) { > + ret = vvenc_encode( s->vvencEnc, pyuvbuf, s->pAU, &s->encodeDone ); > + if( ret != 0 ) { > + av_log(avctx, AV_LOG_ERROR, "error in vvenc::encode - ret:%d\n", ret); > + return AVERROR(EINVAL); > + } > + } > + else { > + *got_packet = 0; > + return 0; > + } > + > + if( s->pAU->payloadUsedSize > 0 ) { > + ret = ff_get_encode_buffer(avctx, pkt, s->pAU->payloadUsedSize, 0); > + if (ret < 0) { > + av_log(avctx, AV_LOG_ERROR, "Error getting output packet.\n"); > + return ret; > + } > + > + memcpy( pkt->data, s->pAU->payload, s->pAU->payloadUsedSize ); > + > + if( s->pAU->ctsValid ) > + pkt->pts = s->pAU->cts; > + if( s->pAU->dtsValid ) > + pkt->dts = s->pAU->dts; > + pkt->flags |= AV_PKT_FLAG_KEY*s->pAU->rap; > + > + switch (s->pAU->sliceType) { > + case VVENC_I_SLICE: > + pict_type = AV_PICTURE_TYPE_I; > + break; > + case VVENC_P_SLICE: > + pict_type = AV_PICTURE_TYPE_P; > + break; > + case VVENC_B_SLICE: > + pict_type = AV_PICTURE_TYPE_B; > + break; > + default: > + av_log(avctx, AV_LOG_ERROR, "Unknown picture type encountered.\n"); > + return AVERROR_EXTERNAL; > + } > + > + ff_side_data_set_encoder_stats(pkt, 0, NULL, 0, pict_type); > + > + *got_packet = 1; > + > + return 0; > + } > + else > + { > + *got_packet = 0; > + return 0; > + } > + > + return 0; > +} > + > +static const enum AVPixelFormat pix_fmts_vvc[] = { > + //AV_PIX_FMT_YUV420P, // TODO > + AV_PIX_FMT_YUV420P10LE, > + AV_PIX_FMT_NONE > +}; > + > +#define OFFSET(x) offsetof(VVenCContext, x) > +#define VE AV_OPT_FLAG_VIDEO_PARAM | AV_OPT_FLAG_ENCODING_PARAM > +static const AVOption libvvenc_options[] = { > + {"preset", "set encoding preset(0: faster - 4: slower", OFFSET( > options.preset), AV_OPT_TYPE_INT, {.i64 = 2} , 0 , 4 , VE, "preset"}, > + { "faster", "0", 0, AV_OPT_TYPE_CONST, {.i64 = > VVENC_FASTER},INT_MIN, INT_MAX, VE, "preset" }, > + { "fast", "1", 0, AV_OPT_TYPE_CONST, {.i64 = VVENC_FAST}, INT_MIN, > INT_MAX, VE, "preset" }, > + { "medium", "2", 0, AV_OPT_TYPE_CONST, {.i64 = > VVENC_MEDIUM},INT_MIN, INT_MAX, VE, "preset" }, > + { "slow", "3", 0, AV_OPT_TYPE_CONST, {.i64 = VVENC_SLOW}, INT_MIN, > INT_MAX, VE, "preset" }, > + { "slower", "4", 0, AV_OPT_TYPE_CONST, {.i64 = > VVENC_SLOWER},INT_MIN, INT_MAX, VE, "preset" }, > + {"qp" , "set quantization" , OFFSET(options.qp) , AV_OPT_TYPE_INT, > {.i64 = 32}, 0 , 63 ,VE, "qp_mode" }, > + {"period" , "set (intra) refresh period in seconds" , > OFFSET(options.intraRefreshSec) , AV_OPT_TYPE_INT, {.i64 = 1}, 1 , > INT_MAX ,VE,"irefreshsec" }, > + {"subjopt", "set subjective (perceptually motivated) optimization" , > OFFSET(options.subjectiveOptimization) , AV_OPT_TYPE_BOOL, {.i64 = 1}, > 0 , 1, VE}, > + {"vvenc-params", "set the vvenc configuration using a :-separated > list of key=value parameters", OFFSET(options.vvenc_opts), > AV_OPT_TYPE_DICT, { 0 }, 0, 0, VE }, > + {"levelidc", "vvc level_idc", OFFSET( options.levelIdc), > AV_OPT_TYPE_INT, {.i64 = 0}, 0, 105, VE, "levelidc"}, > + {"0" , "auto" , 0, AV_OPT_TYPE_CONST, {.i64 = 0}, INT_MIN, INT_MAX, > VE, "levelidc"}, > + {"1" , "1" , 0, AV_OPT_TYPE_CONST, {.i64 = 16},INT_MIN, INT_MAX, VE, > "levelidc"}, > + {"2" , "2" , 0, AV_OPT_TYPE_CONST, {.i64 = 32},INT_MIN, INT_MAX, VE, > "levelidc"}, > + {"2.1" , "2.1" , 0, AV_OPT_TYPE_CONST, {.i64 = 35},INT_MIN, INT_MAX, > VE, "levelidc"}, > + {"3" , "3" , 0, AV_OPT_TYPE_CONST, {.i64 = 48},INT_MIN, INT_MAX, VE, > "levelidc"}, > + {"3.1" , "3.1" , 0, AV_OPT_TYPE_CONST, {.i64 = 51},INT_MIN, INT_MAX, > VE, "levelidc"}, > + {"4" , "4" , 0, AV_OPT_TYPE_CONST, {.i64 = 64},INT_MIN, INT_MAX, VE, > "levelidc"}, > + {"4.1" , "4.1" , 0, AV_OPT_TYPE_CONST, {.i64 = 67},INT_MIN, INT_MAX, > VE, "levelidc"}, > + {"5" , "5" , 0, AV_OPT_TYPE_CONST, {.i64 = 80},INT_MIN, INT_MAX, VE, > "levelidc"}, > + {"5.1" , "5.1" , 0, AV_OPT_TYPE_CONST, {.i64 = 83},INT_MIN, INT_MAX, > VE, "levelidc"}, > + {"5.2" , "5.2" , 0, AV_OPT_TYPE_CONST, {.i64 = 86},INT_MIN, INT_MAX, > VE, "levelidc"}, > + {"6" , "6" , 0, AV_OPT_TYPE_CONST, {.i64 = 96},INT_MIN, INT_MAX, VE, > "levelidc"}, > + {"6.1" , "6.1" , 0, AV_OPT_TYPE_CONST, {.i64 = 99},INT_MIN, INT_MAX, > VE, "levelidc"}, > + {"6.2" , "6.2" , 0, AV_OPT_TYPE_CONST, {.i64 = 102},INT_MIN, > INT_MAX, VE, "levelidc"}, > + {"6.3" , "6.3" , 0, AV_OPT_TYPE_CONST, {.i64 = 105},INT_MIN, > INT_MAX, VE, "levelidc"}, > + {"tier", "set vvc tier", OFFSET( options.tier), AV_OPT_TYPE_INT, > {.i64 = 0}, 0 , 1 , VE, "tier"}, > + {"main", "main" , 0, AV_OPT_TYPE_CONST, {.i64 = 0}, INT_MIN, > INT_MAX, VE, "tier"}, > + {"high", "high" , 0, AV_OPT_TYPE_CONST, {.i64 = 1}, INT_MIN, > INT_MAX, VE, "tier"}, > + {NULL} > +}; > + > +static const AVClass class_libvvenc = { > + .class_name = "libvvenc-vvc encoder", > + .item_name = av_default_item_name, > + .option = libvvenc_options, > + .version = LIBAVUTIL_VERSION_INT, > +}; > + > +FFCodec ff_libvvenc_encoder = { > + .p.name = "libvvenc", > + .p.long_name = "H.266 / VVC Encoder VVenC", > + .p.type = AVMEDIA_TYPE_VIDEO, > + .p.id = AV_CODEC_ID_VVC, > + .p.capabilities = AV_CODEC_CAP_DELAY | AV_CODEC_CAP_OTHER_THREADS, > + .p.profiles = NULL_IF_CONFIG_SMALL(ff_vvc_profiles), > + .p.priv_class = &class_libvvenc, > + .p.wrapper_name = "libvvenc", > + .priv_data_size = sizeof(VVenCContext), > + .p.pix_fmts = pix_fmts_vvc, > + .init = ff_vvenc_encode_init, > + FF_CODEC_ENCODE_CB(ff_vvenc_encode_frame), > + .close = ff_vvenc_encode_close, > + .caps_internal = FF_CODEC_CAP_AUTO_THREADS, > +}; _______________________________________________ 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".