[FFmpeg-devel] [PATCH 6/6] avcodec: add vvdec H.266/VVC decoder
Paul B Mahol
onemda at gmail.com
Mon Dec 21 19:40:43 EET 2020
On Mon, Dec 21, 2020 at 7:08 AM Nuo Mi <nuomi2021 at gmail.com> wrote:
> you can download test clips here:
>
> https://www.itu.int/wftp3/av-arch/jvet-site/bitstream_exchange/VVC/under_test/VTM-11.0/
>
> 68.48% (163/238) clips are md5 matched with VTM 11:
>
> passed:
> 10b400_A_Bytedance_2.bit
> 10b400_B_Bytedance_2.bit
> 8b400_A_Bytedance_2.bit
> 8b400_B_Bytedance_2.bit
> 8b420_A_Bytedance_2.bit
> 8b420_B_Bytedance_2.bit
> ACTPIC_A_Huawei_3.bit
> ACTPIC_B_Huawei_3.bit
> ACTPIC_C_Huawei_3.bit
> AFF_A_HUAWEI_2.bit
> AFF_B_HUAWEI_2.bit
> ALF_A_Huawei_3.bit
> ALF_B_Huawei_3.bit
> ALF_C_KDDI_2.bit
> ALF_D_Qualcomm_2.bit
> AMVR_A_HHI_3.bit
> AMVR_B_HHI_3.bit
> APSALF_A_Qualcomm_2.bit
> APSLMCS_A_Dolby_3.bit
> APSLMCS_B_Dolby_3.bit
> APSLMCS_C_Dolby_2.bit
> APSMULT_A_MediaTek_3.bit
> APSMULT_B_MediaTek_3.bit
> AUD_A_Broadcom_3.bit
> BCW_A_MediaTek_3.bit
> BDOF_A_MediaTek_3.bit
> BDPCM_A_Orange_2.bit
> CCALF_A_Sharp_3.bit
> CCALF_B_Sharp_3.bit
> CCALF_C_Sharp_3.bit
> CCALF_D_Sharp_3.bit
> CCLM_A_KDDI_1.bit
> CIIP_A_MediaTek_3.bit
> CodingToolsSets_A_Tencent_2.bit
> CodingToolsSets_B_Tencent_2.bit
> CodingToolsSets_C_Tencent_2.bit
> CodingToolsSets_D_Tencent_2.bit
> CROP_A_Panasonic_3.bit
> CROP_B_Panasonic_4.bit
> CST_A_MediaTek_3.bit
> CTU_A_MediaTek_3.bit
> CTU_B_MediaTek_3.bit
> CTU_C_MediaTek_3.bit
> CUBEMAP_A_MediaTek_3.bit
> CUBEMAP_B_MediaTek_3.bit
> CUBEMAP_C_MediaTek_3.bit
> DEBLOCKING_A_Sharp_3.bit
> DEBLOCKING_B_Sharp_2.bit
> DEBLOCKING_C_Huawei_3.bit
> DEBLOCKING_E_Ericsson_2.bit
> DMVR_A_Huawei_3.bit
> DMVR_B_KDDI_3.bit
> DPB_A_Sharplabs_2.bit
> DPB_B_Sharplabs_2.bit
> DQ_A_HHI_3.bit
> ENT444HIGHTIER_A_Sony_3.bit
> ENT444HIGHTIER_B_Sony_3.bit
> ENT444HIGHTIER_C_Sony_3.bit
> ENT444HIGHTIER_D_Sony_3.bit
> ENT444MAINTIER_A_Sony_3.bit
> ENT444MAINTIER_B_Sony_3.bit
> ENT444MAINTIER_C_Sony_3.bit
> ENT444MAINTIER_D_Sony_3.bit
> ENTHIGHTIER_A_Sony_3.bit
> ENTHIGHTIER_B_Sony_3.bit
> ENTHIGHTIER_C_Sony_3.bit
> ENTHIGHTIER_D_Sony_3.bit
> ENTMAINTIER_A_Sony_3.bit
> ENTMAINTIER_B_Sony_3.bit
> ENTMAINTIER_C_Sony_3.bit
> ENTMAINTIER_D_Sony_3.bit
> ENTROPY_A_Chipsnmedia_2.bit
> ENTROPY_A_Qualcomm_2.bit
> ENTROPY_B_Sharp_2.bit
> ERP_A_MediaTek_3.bit
> FILLER_A_Bytedance_1.bit
> GPM_A_Alibaba_3.bit
> IBC_A_Tencent_2.bit
> IBC_B_Tencent_2.bit
> IBC_C_Tencent_2.bit
> IBC_D_Tencent_2.bit
> IP_B_Nokia_1.bit
> ISP_A_HHI_3.bit
> ISP_B_HHI_3.bit
> JCCR_A_Nokia_2.bit
> JCCR_B_Nokia_2.bit
> JCCR_C_HHI_3.bit
> JCCR_E_Nokia_1.bit
> JCCR_F_Nokia_1.bit
> LFNST_A_LGE_3.bit
> LFNST_B_LGE_3.bit
> LFNST_C_HHI_3.bit
> LMCS_A_Dolby_3.bit
> LOSSLESS_B_HHI_3.bit
> LTRP_A_ERICSSON_2.bit
> MERGE_A_Qualcomm_2.bit
> MERGE_B_Qualcomm_2.bit
> MERGE_C_Qualcomm_2.bit
> MERGE_D_Qualcomm_2.bit
> MERGE_E_Qualcomm_2.bit
> MERGE_F_Qualcomm_2.bit
> MERGE_G_Qualcomm_2.bit
> MERGE_H_Qualcomm_2.bit
> MERGE_I_Qualcomm_2.bit
> MERGE_J_Qualcomm_2.bit
> MIP_A_HHI_3.bit
> MIP_B_HHI_3.bit
> MPM_A_LGE_3.bit
> MRLP_A_HHI_2.bit
> MRLP_B_HHI_2.bit
> MTS_A_LGE_3.bit
> MTS_B_LGE_3.bit
> MTS_LFNST_A_LGE_3.bit
> MTS_LFNST_B_LGE_3.bit
> MVCOMP_A_Sharp_2.bit
> PDPC_A_Qualcomm_3.bit
> PDPC_B_Qualcomm_3.bit
> PDPC_C_Qualcomm_2.bit
> PHSH_B_Sharp_1.bit
> POC_A_Nokia_1.bit
> POUT_A_Sharplabs_2.bit
> PROF_A_Interdigital_3.bit
> PROF_B_Interdigital_3.bit
> QTBTT_A_MediaTek_3.bit
> QUANT_A_Huawei_2.bit
> QUANT_B_Huawei_2.bit
> QUANT_C_Huawei_2.bit
> RPL_A_ERICSSON_2.bit
> SAO_A_SAMSUNG_3.bit
> SAO_B_SAMSUNG_3.bit
> SAO_C_SAMSUNG_3.bit
> SbTMVP_A_Bytedance_3.bit
> SbTMVP_B_Bytedance_3.bit
> SBT_A_HUAWEI_2.bit
> SCALING_A_InterDigital_1.bit
> SCALING_B_InterDigital_1.bit
> SCALING_C_InterDigital_1.bit
> SDH_A_Dolby_2.bit
> SMVD_A_HUAWEI_2.bit
> TEMPSCAL_A_Panasonic_4.bit
> TEMPSCAL_C_Panasonic_3.bit
> TILE_A_Nokia_2.bit
> TILE_B_Nokia_2.bit
> TILE_C_Nokia_2.bit
> TILE_D_Nokia_2.bit
> TILE_E_Nokia_2.bit
> TILE_F_Nokia_2.bit
> TMVP_A_Chipsnmedia_3.bit
> TMVP_B_Chipsnmedia_3.bit
> TMVP_C_Chipsnmedia_3.bit
> TMVP_D_Chipsnmedia_3.bit
> TRANS_A_Chipsnmedia_2.bit
> TRANS_B_Chipsnmedia_2.bit
> TRANS_C_Chipsnmedia_2.bit
> TRANS_D_Chipsnmedia_2.bit
> WPP_A_Sharp_3.bit
> WPP_B_Sharp_2.bit
> WP_A_InterDigital_3.bit
> WP_B_InterDigital_3.bit
> WRAP_A_InterDigital_4.bit
> WRAP_B_InterDigital_4.bit
> WRAP_C_InterDigital_4.bit
> ---
> configure | 5 +-
> libavcodec/Makefile | 1 +
> libavcodec/allcodecs.c | 1 +
> libavcodec/libvvdec.cpp | 244 ++++++++++++++++++++++++++++++++++++++++
> 4 files changed, 250 insertions(+), 1 deletion(-)
> create mode 100644 libavcodec/libvvdec.cpp
>
> diff --git a/configure b/configure
> index 77272948e3..1e9a44db31 100755
> --- a/configure
> +++ b/configure
> @@ -285,6 +285,7 @@ External library support:
> --enable-libvorbis enable Vorbis en/decoding via libvorbis,
> native implementation exists [no]
> --enable-libvpx enable VP8 and VP9 de/encoding via libvpx [no]
> + --enable-libvvdec enable VVC video encoding via libvvdec [no]
>
I do not think this one is correct.
> --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]
> @@ -1816,6 +1817,7 @@ EXTERNAL_LIBRARY_LIST="
> libvmaf
> libvorbis
> libvpx
> + libvvdec
> libwebp
> libxml2
> libzimg
> @@ -3281,6 +3283,7 @@ libvpx_vp8_decoder_deps="libvpx"
> libvpx_vp8_encoder_deps="libvpx"
> libvpx_vp9_decoder_deps="libvpx"
> libvpx_vp9_encoder_deps="libvpx"
> +libvvdec_decoder_deps="libvvdec"
> libwebp_encoder_deps="libwebp"
> libwebp_anim_encoder_deps="libwebp"
> libx262_encoder_deps="libx262"
> @@ -6455,7 +6458,7 @@ enabled libvpx && {
> die "libvpx enabled but no supported decoders found"
> fi
> }
> -
> +enabled libvvdec && require_pkg_config libvvdec "libvvdec >=
> 0.1.2" vvdec/version.h VVDEC_VERSION_MAJOR
> enabled libwebp && {
> enabled libwebp_encoder && require_pkg_config libwebp "libwebp
> >= 0.2.0" webp/encode.h WebPGetEncoderVersion
> enabled libwebp_anim_encoder && check_pkg_config libwebp_anim_encoder
> "libwebpmux >= 0.4.0" webp/mux.h WebPAnimEncoderOptionsInit; }
> diff --git a/libavcodec/Makefile b/libavcodec/Makefile
> index 82cc9b8b93..ebd3781982 100644
> --- a/libavcodec/Makefile
> +++ b/libavcodec/Makefile
> @@ -1059,6 +1059,7 @@ OBJS-$(CONFIG_LIBVPX_VP8_DECODER) +=
> libvpxdec.o
> 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
> 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 f00d524747..545c38f541 100644
> --- a/libavcodec/allcodecs.c
> +++ b/libavcodec/allcodecs.c
> @@ -752,6 +752,7 @@ extern AVCodec ff_libvpx_vp8_encoder;
> extern AVCodec ff_libvpx_vp8_decoder;
> extern AVCodec ff_libvpx_vp9_encoder;
> extern AVCodec ff_libvpx_vp9_decoder;
> +extern AVCodec ff_libvvdec_decoder;
> /* preferred over libwebp */
> extern AVCodec ff_libwebp_anim_encoder;
> extern AVCodec ff_libwebp_encoder;
> diff --git a/libavcodec/libvvdec.cpp b/libavcodec/libvvdec.cpp
> new file mode 100644
> index 0000000000..73fe13f61e
> --- /dev/null
> +++ b/libavcodec/libvvdec.cpp
> @@ -0,0 +1,244 @@
> +/*
> + * vvdec H.266/VVC decoder
> + * Copyright (c) 2020 Nuo Mi <nuomi2021 at gmail.com>
> + *
> + * 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
> + */
> +
> +/**
> + * @file
> + * AV1 decoder support via libaom
>
Huh?
> + */
> +
> +#include "vvdec/vvdec.h"
> +
> +extern "C" {
> +
> +#include "libavutil/common.h"
> +#include "libavutil/imgutils.h"
> +#include "libavutil/internal.h"
> +
> +#include "avcodec.h"
> +#include "h2645_parse.h"
> +#include "internal.h"
> +#include "profiles.h"
> +
> +typedef struct VVdecContext {
> + AVCodecContext *avctx;
> + vvdec::VVDec* vvdec;
> +
> + H2645Packet pkt;
> + int is_nalff; ///< this flag is != 0 if bitstream is
> encapsulated
> + ///< as a format defined in 14496-15
> + int nal_length_size; ///< Number of bytes used for nal length (1,
> 2 or 4)
> +} VVdecContext;
> +
> +}
> +
> +static int map_error(int vret)
> +{
> + switch(vret)
> + {
> + case vvdec::VVDEC_OK : return 0;
> + case vvdec::VVDEC_ERR_UNSPECIFIED: return AVERROR_UNKNOWN;
> + case vvdec::VVDEC_ERR_INITIALIZE: return AVERROR_BUG;
> + case vvdec::VVDEC_ERR_ALLOCATE: return AVERROR(ENOMEM);
> + case vvdec::VVDEC_NOT_ENOUGH_MEM: return AVERROR(ENOMEM);
> + case vvdec::VVDEC_ERR_PARAMETER: return AVERROR(EINVAL);
> + case vvdec::VVDEC_ERR_NOT_SUPPORTED: return AVERROR(ENOSYS);
> + case vvdec::VVDEC_ERR_RESTART_REQUIRED: return AVERROR_BUG;
> + case vvdec::VVDEC_ERR_CPU: return AVERROR(ENOSYS);
> + case vvdec::VVDEC_TRY_AGAIN: return AVERROR(EAGAIN);
> + case vvdec::VVDEC_EOF: return AVERROR(EOF);
> + default: return AVERROR_UNKNOWN;
> + }
> +}
> +
> +static int check_vret(VVdecContext* s, int vret) {
> + vvdec::VVDec* vvdec = s->vvdec;
> + if (vret && vret != vvdec::VVDEC_EOF) {
> + av_log(s->avctx, AV_LOG_ERROR, "vvdec returns error: %s\n",
> vvdec->getErrorMsg(vret));
> + }
> + return map_error(vret);
> +}
> +
> +static const enum AVPixelFormat pix_fmts_8bit[] = {
> + AV_PIX_FMT_GRAY8, AV_PIX_FMT_YUV420P,
> + AV_PIX_FMT_YUV422P, AV_PIX_FMT_YUV444P
> +};
> +
> +static const enum AVPixelFormat pix_fmts_10bit[] = {
> + AV_PIX_FMT_GRAY10, AV_PIX_FMT_YUV420P10,
> + AV_PIX_FMT_YUV422P10, AV_PIX_FMT_YUV444P10
> +};
> +
> +static AVPixelFormat get_format(const vvdec::Frame* frame)
> +{
> + if (frame->m_eColorFormat != vvdec::VVC_CF_INVALID) {
> + switch (frame->m_uiBitDepth) {
> + case 8:
> + return pix_fmts_8bit[frame->m_eColorFormat];
> + case 10:
> + return pix_fmts_10bit[frame->m_eColorFormat];
> + }
> + }
> + return AV_PIX_FMT_NONE;
> +}
> +
> +static int set_pix_fmt(VVdecContext* s, const vvdec::Frame* frame)
> +{
> + AVPixelFormat format = get_format(frame);
> + if (format == AV_PIX_FMT_NONE) {
> + av_log(s->avctx, AV_LOG_ERROR,
> + "unsupported, depth = %d, color format = %d.\n",
> frame->m_uiBitDepth, frame->m_eColorFormat);
> + return AVERROR_INVALIDDATA;
> + }
> + s->avctx->pix_fmt = format;
> + return 0;
> +}
> +
> +static int copy_to_avframe(VVdecContext* s, const vvdec::Frame* src,
> AVFrame *dest)
> +{
> + AVCodecContext* avctx = s->avctx;
> + int i, ret, width, height;
> + uint8_t *data[4];
> + int linesize[4];
> +
> + ret = set_pix_fmt(s, src);
> + if (ret < 0)
> + return ret;
> + width = (int)src->m_uiWidth;
> + height = (int)src->m_uiHeight;
> + if (width != avctx->width || height != avctx->height) {
> + av_log(avctx, AV_LOG_INFO, "dimension change! %dx%d -> %dx%d\n",
> + avctx->width, avctx->height, width, height);
> + if ((ret = ff_set_dimensions(avctx, width, height)) < 0)
> + return ret;
> + }
> + if ((ret = ff_get_buffer(avctx, dest, 0)) < 0)
> + return ret;
> +
> + for (i = 0; i < 3; i++) {
> + const vvdec::Component& plane = src->m_cComponent[i];
> + data[i] = plane.m_pucBuffer;
> + linesize[i] = plane.m_iStride;
> + }
> + data[3] = 0; linesize[3] = 0;
> +
> + //TODO: conformance window?
> + av_image_copy(dest->data, dest->linesize, (const uint8_t **)data,
> linesize,
> + avctx->pix_fmt, width, height);
> + return 0;
> +}
> +
> +static int vvdec_decode(AVCodecContext *avctx, void *data, int *got_frame,
> + AVPacket *avpkt)
> +{
> + VVdecContext *s = (VVdecContext *)avctx->priv_data;
> + vvdec::VVDec* vvdec = s->vvdec;
> + AVFrame *picture = (AVFrame*)data;
> + int vret, ret = 0;
> + vvdec::Frame* frame = NULL;
> +
> + if (!avpkt->size) {
> + vret = vvdec->flush(&frame);
> + if ((ret = check_vret(s, vret)) < 0) {
> + goto error_out;
> + }
> + } else {
> + uint8_t *new_extradata;
> + int new_extradata_size;
> + new_extradata = av_packet_get_side_data(avpkt,
> AV_PKT_DATA_NEW_EXTRADATA,
> + &new_extradata_size);
> + if (new_extradata && new_extradata_size > 0) {
> + return AVERROR_PATCHWELCOME;
> + }
> +
> + vvdec::AccessUnit au;
> + au.m_pucBuffer = (unsigned char*)avpkt->data;
> + au.m_iUsedSize = avpkt->size;
> + au.m_uiCts = avpkt->pts;
> + au.m_bCtsValid = true;
> + vret = vvdec->decode(au, &frame);
> + if (vret && vret != vvdec::VVDEC_TRY_AGAIN) {
> + if ((ret = check_vret(s, vret)) < 0) {
> + goto error_out;
> + }
> + }
> +
> + }
> + if (frame) {
> + ret = copy_to_avframe(s, frame, picture);
> + if (ret < 0)
> + goto error_out;
> + *got_frame = 1;
> + }
> + return 0;
> +error_out:
> + if (frame) {
> + vvdec->objectUnref(frame);
> + }
> + return ret;
> +}
> +
> +static av_cold int vvdec_free(AVCodecContext *avctx)
> +{
> + VVdecContext *ctx = (VVdecContext *)avctx->priv_data;
> + vvdec::VVDec* vvdec = ctx->vvdec;
> + if (vvdec) {
> + vvdec->uninit();
> + delete vvdec;
> + }
> + return 0;
> +}
> +
> +static av_cold int vvdec_init(AVCodecContext *avctx)
> +{
> + VVdecContext *s = (VVdecContext *)avctx->priv_data;
> + vvdec::VVDecParameter param;
> + vvdec::VVDec* vvdec = new vvdec::VVDec;
> + int vret, ret = 0;
> +
> + av_log(avctx, AV_LOG_INFO, "VVDec version: %s\n",
> vvdec->getVersionNumber());
> +
> + s->avctx = avctx;
> + s->vvdec = vvdec;
> + vret = vvdec->init(param);
> + if ((ret = check_vret(s, vret)) < 0) {
> + delete vvdec;
> + s->vvdec = NULL;
> + }
> + return ret;
> +}
> +
> +extern "C" {
> +
> +AVCodec ff_libvvdec_decoder = {
> + .name = "vvdec",
>
Please name it "libvvdec"
> + .long_name = NULL_IF_CONFIG_SMALL("vvdec vvc decoder"),
>
This is simply wrong. Use full, official codec name.
> + .type = AVMEDIA_TYPE_VIDEO,
> + .id = AV_CODEC_ID_H266,
> + .capabilities = AV_CODEC_CAP_AUTO_THREADS | AV_CODEC_CAP_DR1 |
> AV_CODEC_CAP_DELAY,
> + .profiles = NULL_IF_CONFIG_SMALL(ff_h266_profiles),
> + .wrapper_name = "libvvdec",
> + .priv_data_size = sizeof(VVdecContext),
> + .init = vvdec_init,
> + .decode = vvdec_decode,
> + .close = vvdec_free,
> +};
> +
> +}
> --
> 2.25.1
>
> _______________________________________________
> ffmpeg-devel mailing list
> ffmpeg-devel at ffmpeg.org
> https://ffmpeg.org/mailman/listinfo/ffmpeg-devel
>
> To unsubscribe, visit link above, or email
> ffmpeg-devel-request at ffmpeg.org with subject "unsubscribe".
More information about the ffmpeg-devel
mailing list