/* * VVC video decoder * * Copyright (C) 2021 Nuo Mi * Copyright (C) 2022 Xu Mu * * 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 */ #ifndef AVCODEC_VVC_DEC_H #define AVCODEC_VVC_DEC_H #include "libavcodec/videodsp.h" #include "libavcodec/vvc.h" #include "ps.h" #include "dsp.h" #define LUMA 0 #define CHROMA 1 #define CB 1 #define CR 2 #define JCBCR 3 #define MIN_TU_LOG2 2 ///< MinTbLog2SizeY #define MIN_PU_LOG2 2 #define L0 0 #define L1 1 typedef struct RefPicList { struct VVCFrame *ref[VVC_MAX_REF_ENTRIES]; int list[VVC_MAX_REF_ENTRIES]; int isLongTerm[VVC_MAX_REF_ENTRIES]; int nb_refs; } RefPicList; typedef struct RefPicListTab { RefPicList refPicList[2]; } RefPicListTab; typedef struct VVCFrame { struct AVFrame *frame; struct MvField *tab_dmvr_mvf; ///< RefStruct reference RefPicListTab **rpl_tab; ///< RefStruct reference RefPicListTab *rpl; ///< RefStruct reference int nb_rpl_elems; int ctb_count; int poc; struct VVCFrame *collocated_ref; struct FrameProgress *progress; ///< RefStruct reference /** * A sequence counter, so that old frames are output first * after a POC reset */ uint16_t sequence; /** * A combination of VVC_FRAME_FLAG_* */ uint8_t flags; } VVCFrame; typedef struct SliceContext { int slice_idx; VVCSH sh; struct EntryPoint *eps; int nb_eps; RefPicList *rpl; void *ref; ///< RefStruct reference, backing slice data } SliceContext; typedef struct VVCFrameContext { void *log_ctx; // +1 for the current frame VVCFrame DPB[VVC_MAX_DPB_SIZE + 1]; struct AVFrame *frame; struct AVFrame *output_frame; VVCFrameParamSets ps; SliceContext **slices; int nb_slices; int nb_slices_allocated; VVCFrame *ref; VVCDSPContext vvcdsp; VideoDSPContext vdsp; struct VVCFrameThread *ft; uint64_t decode_order; struct FFRefStructPool *tab_dmvr_mvf_pool; struct FFRefStructPool *rpl_tab_pool; struct FFRefStructPool *cu_pool; struct FFRefStructPool *tu_pool; struct { int16_t *slice_idx; DBParams *deblock; struct SAOParams *sao; struct ALFParams *alf; int *cb_pos_x[2]; ///< CbPosX[][][] int *cb_pos_y[2]; ///< CbPosY[][][] uint8_t *cb_width[2]; ///< CbWidth[][][] uint8_t *cb_height[2]; ///< CbHeight[][][] uint8_t *cqt_depth[2]; ///< CqtDepth[][][] int8_t *qp[VVC_MAX_SAMPLE_ARRAYS]; uint8_t *skip; ///< CuSkipFlag[][] uint8_t *ispmf; ///< intra_sub_partitions_mode_flag uint8_t *msm[2]; ///< MttSplitMode[][][] in 32 pixels uint8_t *imf; ///< IntraMipFlag[][] uint8_t *imtf; ///< intra_mip_transposed_flag[][] uint8_t *imm; ///< intra_mip_mode[][] uint8_t *ipm; ///< IntraPredModeY[][] uint8_t *cpm[2]; ///< CuPredMode[][][] uint8_t *msf; ///< MergeSubblockFlag[][] uint8_t *iaf; ///< InterAffineFlag[][] uint8_t *mmi; ///< MotionModelIdc[][] struct Mv *cp_mv[2]; ///< CpMvLX[][][][MAX_CONTROL_POINTS]; struct MvField *mvf; ///< MvDmvrL0, MvDmvrL1 uint8_t *tu_coded_flag[VVC_MAX_SAMPLE_ARRAYS]; ///< tu_y_coded_flag[][], tu_cb_coded_flag[][], tu_cr_coded_flag[][] uint8_t *tu_joint_cbcr_residual_flag; ///< tu_joint_cbcr_residual_flag[][] int *tb_pos_x0[2]; int *tb_pos_y0[2]; uint8_t *tb_width[2]; uint8_t *tb_height[2]; uint8_t *pcmf[2]; uint8_t *horizontal_bs[VVC_MAX_SAMPLE_ARRAYS]; uint8_t *vertical_bs[VVC_MAX_SAMPLE_ARRAYS]; uint8_t *horizontal_p; ///< horizontal maxFilterLengthPs for luma uint8_t *horizontal_q; ///< horizontal maxFilterLengthQs for luma uint8_t *vertical_p; ///< vertical maxFilterLengthPs for luma uint8_t *vertical_q; ///< vertical maxFilterLengthQs for luma uint8_t *sao_pixel_buffer_h[VVC_MAX_SAMPLE_ARRAYS]; uint8_t *sao_pixel_buffer_v[VVC_MAX_SAMPLE_ARRAYS]; uint8_t *alf_pixel_buffer_h[VVC_MAX_SAMPLE_ARRAYS][2]; uint8_t *alf_pixel_buffer_v[VVC_MAX_SAMPLE_ARRAYS][2]; int *coeffs; struct CTU *ctus; uint8_t *ibc_vir_buf[VVC_MAX_SAMPLE_ARRAYS]; ///< IbcVirBuf[] //used in arrays_init only struct { int ctu_count; int ctu_size; int pic_size_in_min_cb; int pic_size_in_min_pu; int pic_size_in_min_tu; int ctu_width; int ctu_height; int width; int height; int chroma_format_idc; int pixel_shift; int bs_width; int bs_height; int ibc_buffer_width; ///< IbcBufWidth } sz; } tab; } VVCFrameContext; typedef struct VVCContext { struct AVCodecContext *avctx; CodedBitstreamContext *cbc; CodedBitstreamFragment current_frame; VVCParamSets ps; int temporal_id; ///< temporal_id_plus1 - 1 int poc_tid0; int eos; ///< current packet contains an EOS/EOB NAL int last_eos; ///< last packet contains an EOS/EOB NAL enum VVCNALUnitType vcl_unit_type; int no_output_before_recovery_flag; ///< NoOutputBeforeRecoveryFlag int gdr_recovery_point_poc; ///< recoveryPointPocVal /** * Sequence counters for decoded and output frames, so that old * frames are output first after a POC reset */ uint16_t seq_decode; uint16_t seq_output; struct AVExecutor *executor; VVCFrameContext *fcs; int nb_fcs; uint64_t nb_frames; ///< processed frames int nb_delayed; ///< delayed frames } VVCContext ; #endif /* AVCODEC_VVC_DEC_H */