From patchwork Mon Jan 1 14:12:30 2024 Content-Type: text/plain; charset="utf-8" MIME-Version: 1.0 Content-Transfer-Encoding: 7bit X-Patchwork-Submitter: Nuo Mi X-Patchwork-Id: 45437 Delivered-To: ffmpegpatchwork2@gmail.com Received: by 2002:a05:6a20:6623:b0:194:e134:edd4 with SMTP id n35csp6227997pzh; Mon, 1 Jan 2024 06:15:10 -0800 (PST) X-Google-Smtp-Source: AGHT+IEDJ7rod/j1WYmGCDrKs2V0+PgmhQmE54CFViA77YbCWgYC/D5IbTLRWBRdNkHLKadhE1Ha X-Received: by 2002:a50:d650:0:b0:553:6bd5:d18e with SMTP id c16-20020a50d650000000b005536bd5d18emr2819675edj.34.1704118510043; Mon, 01 Jan 2024 06:15:10 -0800 (PST) Return-Path: Received: from ffbox0-bg.mplayerhq.hu (ffbox0-bg.ffmpeg.org. [79.124.17.100]) by mx.google.com with ESMTP id t9-20020a50d709000000b005559a32fb08si3646517edi.103.2024.01.01.06.15.09; Mon, 01 Jan 2024 06:15:10 -0800 (PST) Received-SPF: pass (google.com: domain of ffmpeg-devel-bounces@ffmpeg.org designates 79.124.17.100 as permitted sender) client-ip=79.124.17.100; Authentication-Results: mx.google.com; dkim=neutral (body hash did not verify) header.i=@outlook.com header.s=selector1 header.b=LO4OabzA; arc=fail (body hash mismatch); spf=pass (google.com: domain of ffmpeg-devel-bounces@ffmpeg.org designates 79.124.17.100 as permitted sender) smtp.mailfrom=ffmpeg-devel-bounces@ffmpeg.org; dmarc=fail (p=NONE sp=QUARANTINE dis=NONE) header.from=gmail.com Received: from [127.0.1.1] (localhost [127.0.0.1]) by ffbox0-bg.mplayerhq.hu (Postfix) with ESMTP id 6519268CEBC; Mon, 1 Jan 2024 16:13:45 +0200 (EET) X-Original-To: ffmpeg-devel@ffmpeg.org Delivered-To: ffmpeg-devel@ffmpeg.org Received: from APC01-TYZ-obe.outbound.protection.outlook.com (mail-tyzapc01olkn2103.outbound.protection.outlook.com [40.92.107.103]) by ffbox0-bg.mplayerhq.hu (Postfix) with ESMTPS id 4AE5F68CDF6 for ; Mon, 1 Jan 2024 16:13:37 +0200 (EET) ARC-Seal: i=1; a=rsa-sha256; s=arcselector9901; d=microsoft.com; cv=none; b=UvX6Q5KVJwbSXKfvPD/07ntj3xVC5l/HS2QrP11LDvXei1zjALW0+Og4sDxSBaoSJ4NA7sZAZ9qd5lCKbfXpbBqbqiK1K5QTstS9jDTTpasKq3Y7aA/X9crJdu6nQi8oJZmjYZjwcjcMhEI7zNIas3MWiekc1YjSgeNDUnbrKPkm4xZ+wLLPjuSi6j+VyyMpIdnka2lwBHfULrPSzJYxaSiBfqfYbHoRsBnIeGXkTxq87MIO385FgZk49+UdddYIA0qZnPCyYwB9AbN0Q4aCbsz/Ly8CIihdlBOnavCVHqReVkJrEC5hc9o8//rmFNMNScWPF6NosCAXdigge+AMyQ== ARC-Message-Signature: i=1; a=rsa-sha256; c=relaxed/relaxed; d=microsoft.com; s=arcselector9901; h=From:Date:Subject:Message-ID:Content-Type:MIME-Version:X-MS-Exchange-AntiSpam-MessageData-ChunkCount:X-MS-Exchange-AntiSpam-MessageData-0:X-MS-Exchange-AntiSpam-MessageData-1; bh=GcAgyoaOxwKMwJ0U1zL0JQLcpNIzQe78YBadWEe1ktE=; b=V/a91VtfOhwtU41ITXdbRHVj7NXDIoUxmldxf+jmuwYGpRkFNCS5HxZ4WDIBGkQc1tBjqBytirq/iO/eSWAKklfMD6MSBntiMy2VrW3m9YT8jokSgGC8EzO0/FQTe5E9erFHq1xZhSi0oO1FFcGCo5+XrY0GYHpi0tsV4Ngja6a8V/fNlwxSTpGLcJmLpouINTPE9a0GrjDPYl1hvFYzIsl0+HeLLcb9Gt/aACw5EMgHc1Eqe033FELeVY5kwCfRLSdVF8lomK4EpeXwj+CoM5lUeVdR89vUQOUCm288LFEJf4sqQWM65+a52n5clG5Mi2lrBmJYRlqNKvGH9mKzCA== ARC-Authentication-Results: i=1; mx.microsoft.com 1; spf=none; dmarc=none; dkim=none; arc=none DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=outlook.com; s=selector1; h=From:Date:Subject:Message-ID:Content-Type:MIME-Version:X-MS-Exchange-SenderADCheck; bh=GcAgyoaOxwKMwJ0U1zL0JQLcpNIzQe78YBadWEe1ktE=; b=LO4OabzAzhXuPSG7wBqrgdttGTZBzJBBmX1EPVY+9hn2zBIZ1NJUonISV+nX4+THjm8cydz+nvHjL7UQIbhUKwfsivgFndgEKHaNZfrXTWkzQqz9cqXT9BO+/OXyUCTQ3KZXcfU3dMq+Oyktex2ycKE6BuDtgp+4xg32Lg/4QpECBwaKCqVy70meely3rqx9AMFhHt2QCF5bqSKfEVR5Dwd/5JNZaJSg2TP0s8D1hFthuEucR2VNjAzNLlsoXSUvoL0tyg5pdlcFVjCbCnikakXY5AlU07fhcA+ysMgJDKc3AeqsFv/sciavAFmIKRnHW1KFGTurX1bB3tPZ442T+Q== Received: from TYSPR06MB6433.apcprd06.prod.outlook.com (2603:1096:400:47a::6) by TYZPR06MB3965.apcprd06.prod.outlook.com (2603:1096:400:23::8) with Microsoft SMTP Server (version=TLS1_2, cipher=TLS_ECDHE_RSA_WITH_AES_256_GCM_SHA384) id 15.20.7135.25; Mon, 1 Jan 2024 14:13:21 +0000 Received: from TYSPR06MB6433.apcprd06.prod.outlook.com ([fe80::e139:fd7a:1b66:a3f7]) by TYSPR06MB6433.apcprd06.prod.outlook.com ([fe80::e139:fd7a:1b66:a3f7%6]) with mapi id 15.20.7135.023; Mon, 1 Jan 2024 14:13:21 +0000 From: Nuo Mi To: ffmpeg-devel@ffmpeg.org Date: Mon, 1 Jan 2024 22:12:30 +0800 Message-ID: X-Mailer: git-send-email 2.25.1 In-Reply-To: <20240101141239.6623-1-nuomi2021@gmail.com> References: <20240101141239.6623-1-nuomi2021@gmail.com> X-TMN: [4bXbGDkIR0EILnUGc5MDR/ihCA59CEwk] X-ClientProxiedBy: SI1PR02CA0012.apcprd02.prod.outlook.com (2603:1096:4:1f7::8) To TYSPR06MB6433.apcprd06.prod.outlook.com (2603:1096:400:47a::6) X-Microsoft-Original-Message-ID: <20240101141239.6623-5-nuomi2021@gmail.com> MIME-Version: 1.0 X-MS-Exchange-MessageSentRepresentingType: 2 X-MS-PublicTrafficType: Email X-MS-TrafficTypeDiagnostic: TYSPR06MB6433:EE_|TYZPR06MB3965:EE_ X-MS-Office365-Filtering-Correlation-Id: feb2e7d5-9a38-4bd7-3522-08dc0ad3cf28 X-Microsoft-Antispam: BCL:0; X-Microsoft-Antispam-Message-Info: Pz3hv+kRE8BXM4iOyantR88r4p0u3jN4Stz/yKe9OSDn6ZZ4LtmuwN1y+UphwqrY/4exzvgWqgi70vTJQKAEooKGdUjvS4Fi6SJtnUhePTxmlsnGf36umQV4R38eKQh8zoNqUwizFaOJrTL8HHaqFa6871BcIQf/dBniol3gE9Jw/XxhSVnx0pOnkKAYIK7MlWz3DRWyi0SdK/uQalknMhl2vIfDbmGOZFZ1lQS4aqnoq0ZGyOy0YNZxawoLwUk1QD51ZSDzFj8AxVXdhePigU3FWwA5WC36D/JfaToa1NrjIYKBx9noArAS2npnQjb4myr2MsC5VkRjz0ORcPMuP4WzzrcCR9JSfuzWdfGa7WCmW3Xyfc+saxSvu8uWevy0mdQn8mXgY32niPpB5Mxsth3N7X8SnH7TTfYVhOg7EHwCrRKEdZcC/i0SeEDu1hd0ZsaqGaRqeNvfB+X6ME5Z6+5qz4JnyMV4BBKveHk0PiJETlj832I0Tz6GBz99Auo/H+iGzrsuj5PWSSJILJa6YF3eHn2nkKafxFKbyuuCCFb3q4RyF5Z8rayE/enwWTQkXEsRkki0/otQpWmnok9j6W9yEqaZYiyPQ7iAgmtP3kjJ9v1zzHP+n8ybJxC1up43 X-MS-Exchange-AntiSpam-MessageData-ChunkCount: 1 X-MS-Exchange-AntiSpam-MessageData-0: lYYXvQP1Xjio6xOAFIQKfl9qT9NdtYoK1UCPtey2t870rUxbja3bRWlWIbcF3p4qxzSOTye2FzKA2PFiMyoe5I8U5byCOZ+Hl3bIhhNEeJOwi2ufi+Ty0NSpyzc8PwFBGr1n5xZWmfyXxRyKvwzHy8c/UsIpHYo3fCWbZ4Ja52YlPJYOIrxlIVLN/ZhKhRx6efwl7raLchGpXb1pHtxyGWHk7gT6f97oN2FOca4Cn70okIZbwXuPZeKKx/65/AZH+DcyTiBwTD+GbJUcmvfhN3Ia5nxkCZBVnHc+qfkyglmOGgBG4Sqdu2iLV9hYLWkLviHKCx+TJCv1jCpQpRt/vbhCU7I4SuAo1CVpwCMoJb0dsLwOpHgtXGeUkBzUr+5+Pk/pg4a42Ccm6K8CNfBLI3jW75ueVjCDJoxOaCH3kiaGhN/O0q7Neht1sdHs820c72Qu5uFLtGqe72268NhUGBHbDajL9jRz25HfaVViYWS6uG5C4RX0rfpUbpYzslfwy16GGXHYDmrs3d+PEakQdEVdHClz65oWT+zVm82AtLeeOJhyHMTKaHVy/g8JMW4cBxG9eRLqezvNDkFE/3MQpgUCYtnbo1F1eWgnFMIM2nGW1Hqn1uLynl2MQ0cqQCv2DyRcQobpJu0/tZ0BUze6aa0DC+6y0pABHeKvDGot/5aK7R1scTKf9rvQSgX47z3paxjrrtLnyY3D9ley1aRECaIvKI4/N6L1mYUR1jlhkKaqSj854/ZahJsQlKKTcuMwyuc+1hKtQHTWgxlgmghMpNWSoxaDt1q8P+1C67YaccpadXn6ocUqqIfGhD1EuFiXVsf7PjCd+1nJIPjMxIleTI7baO7T6P2keqkfkaVMyDSCpULGvs3gjnhcE9h/NGN94L4R67HAErJSxlMzYYcevVAFPEwCUIH48ohbp3wTiq61k9FA7MDNX+2wZo5lg9FWclm/x62GUV7t+XiIHL7fCiGgq0EUenCsoA6JtpuOup+7IGvP4Q88M2SNyWCDYfcj3dZYekx641uuPXr3EgEmw1GYSEC/Kp83ezoUAZmDlpBO/u1q7BGfuWzzoq/tYyK4CyPBJJK++W9XNWHAOtmoM2AqDv8DEsOUjrpcPl0XsmCuf0Yu6XqQB7PHUNCjOTKOlpukTSnXWQqbijipIG3tD74fxtOIoMge/9DtARLFF8S65VhDvBs4FCDFiyE8LqqZOPw40KoN2pWH9FsSYbcGQxq7X67et1CbKmzlWkyCGgiBGrxQbpBZcQ3D/eSuwN0x X-OriginatorOrg: outlook.com X-MS-Exchange-CrossTenant-Network-Message-Id: feb2e7d5-9a38-4bd7-3522-08dc0ad3cf28 X-MS-Exchange-CrossTenant-AuthSource: TYSPR06MB6433.apcprd06.prod.outlook.com X-MS-Exchange-CrossTenant-AuthAs: Internal X-MS-Exchange-CrossTenant-OriginalArrivalTime: 01 Jan 2024 14:13:21.1741 (UTC) X-MS-Exchange-CrossTenant-FromEntityHeader: Hosted X-MS-Exchange-CrossTenant-Id: 84df9e7f-e9f6-40af-b435-aaaaaaaaaaaa X-MS-Exchange-CrossTenant-RMS-PersistedConsumerOrg: 00000000-0000-0000-0000-000000000000 X-MS-Exchange-Transport-CrossTenantHeadersStamped: TYZPR06MB3965 Subject: [FFmpeg-devel] [PATCH v9 04/13] vvcdec: add reference management 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 Cc: Frank Plowman , Nuo Mi , Xu Mu , Shaun Loo , Wu Jianhua Errors-To: ffmpeg-devel-bounces@ffmpeg.org Sender: "ffmpeg-devel" X-TUID: AuqeodyTA7au Co-authored-by: Xu Mu Co-authored-by: Frank Plowman Co-authored-by: Shaun Loo Co-authored-by: Wu Jianhua --- libavcodec/vvc/Makefile | 1 + libavcodec/vvc/vvc_refs.c | 559 ++++++++++++++++++++++++++++++++++++++ libavcodec/vvc/vvc_refs.h | 57 ++++ 3 files changed, 617 insertions(+) create mode 100644 libavcodec/vvc/vvc_refs.c create mode 100644 libavcodec/vvc/vvc_refs.h diff --git a/libavcodec/vvc/Makefile b/libavcodec/vvc/Makefile index a2f3cc1883..0c8e942ba4 100644 --- a/libavcodec/vvc/Makefile +++ b/libavcodec/vvc/Makefile @@ -5,3 +5,4 @@ OBJS-$(CONFIG_VVC_DECODER) += vvc/vvc_cabac.o \ vvc/vvc_ctu.o \ vvc/vvc_data.o \ vvc/vvc_ps.o \ + vvc/vvc_refs.o \ diff --git a/libavcodec/vvc/vvc_refs.c b/libavcodec/vvc/vvc_refs.c new file mode 100644 index 0000000000..bf503e429e --- /dev/null +++ b/libavcodec/vvc/vvc_refs.c @@ -0,0 +1,559 @@ +/* + * VVC reference management + * + * Copyright (C) 2023 Nuo Mi + * + * 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 + +#include "libavutil/thread.h" +#include "libavcodec/refstruct.h" +#include "libavcodec/thread.h" + +#include "vvc_refs.h" + +#define VVC_FRAME_FLAG_OUTPUT (1 << 0) +#define VVC_FRAME_FLAG_SHORT_REF (1 << 1) +#define VVC_FRAME_FLAG_LONG_REF (1 << 2) +#define VVC_FRAME_FLAG_BUMPING (1 << 3) + +typedef struct FrameProgress { + atomic_int progress[VVC_PROGRESS_LAST]; + VVCProgressListener *listener[VVC_PROGRESS_LAST]; + AVMutex lock; + AVCond cond; + uint8_t has_lock; + uint8_t has_cond; +} FrameProgress; + +void ff_vvc_unref_frame(VVCFrameContext *fc, VVCFrame *frame, int flags) +{ + /* frame->frame can be NULL if context init failed */ + if (!frame->frame || !frame->frame->buf[0]) + return; + + frame->flags &= ~flags; + if (!frame->flags) { + av_frame_unref(frame->frame); + ff_refstruct_unref(&frame->progress); + + ff_refstruct_unref(&frame->tab_dmvr_mvf); + + ff_refstruct_unref(&frame->rpl); + frame->nb_rpl_elems = 0; + ff_refstruct_unref(&frame->rpl_tab); + + frame->collocated_ref = NULL; + } +} + +const RefPicList *ff_vvc_get_ref_list(const VVCFrameContext *fc, const VVCFrame *ref, int x0, int y0) +{ + const int x_cb = x0 >> fc->ps.sps->ctb_log2_size_y; + const int y_cb = y0 >> fc->ps.sps->ctb_log2_size_y; + const int pic_width_cb = fc->ps.pps->ctb_width; + const int ctb_addr_rs = y_cb * pic_width_cb + x_cb; + + return (const RefPicList *)ref->rpl_tab[ctb_addr_rs]; +} + +void ff_vvc_clear_refs(VVCFrameContext *fc) +{ + for (int i = 0; i < FF_ARRAY_ELEMS(fc->DPB); i++) + ff_vvc_unref_frame(fc, &fc->DPB[i], + VVC_FRAME_FLAG_SHORT_REF | VVC_FRAME_FLAG_LONG_REF); +} + +static void free_progress(FFRefStructOpaque unused, void *obj) +{ + FrameProgress *p = (FrameProgress *)obj; + + if (p->has_cond) + ff_cond_destroy(&p->cond); + if (p->has_lock) + ff_mutex_destroy(&p->lock); +} + +static FrameProgress *alloc_progress(void) +{ + FrameProgress *p = ff_refstruct_alloc_ext(sizeof(*p), 0, NULL, free_progress); + + if (p) { + p->has_lock = !ff_mutex_init(&p->lock, NULL); + p->has_cond = !ff_cond_init(&p->cond, NULL); + if (!p->has_lock || !p->has_cond) + ff_refstruct_unref(&p); + } + return p; +} + +static VVCFrame *alloc_frame(VVCContext *s, VVCFrameContext *fc) +{ + const VVCPPS *pps = fc->ps.pps; + for (int i = 0; i < FF_ARRAY_ELEMS(fc->DPB); i++) { + int ret; + VVCFrame *frame = &fc->DPB[i]; + if (frame->frame->buf[0]) + continue; + + ret = ff_thread_get_buffer(s->avctx, frame->frame, AV_GET_BUFFER_FLAG_REF); + if (ret < 0) + return NULL; + + frame->rpl = ff_refstruct_allocz(s->current_frame.nb_units * sizeof(RefPicListTab)); + if (!frame->rpl) + goto fail; + frame->nb_rpl_elems = s->current_frame.nb_units; + + frame->tab_dmvr_mvf = ff_refstruct_pool_get(fc->tab_dmvr_mvf_pool); + if (!frame->tab_dmvr_mvf) + goto fail; + + frame->rpl_tab = ff_refstruct_pool_get(fc->rpl_tab_pool); + if (!frame->rpl_tab) + goto fail; + frame->ctb_count = pps->ctb_width * pps->ctb_height; + for (int j = 0; j < frame->ctb_count; j++) + frame->rpl_tab[j] = frame->rpl; + + frame->progress = alloc_progress(); + if (!frame->progress) + goto fail; + + return frame; +fail: + ff_vvc_unref_frame(fc, frame, ~0); + return NULL; + } + av_log(s->avctx, AV_LOG_ERROR, "Error allocating frame, DPB full.\n"); + return NULL; +} + +int ff_vvc_set_new_ref(VVCContext *s, VVCFrameContext *fc, AVFrame **frame) +{ + const VVCPH *ph= &fc->ps.ph; + const int poc = ph->poc; + VVCFrame *ref; + + /* check that this POC doesn't already exist */ + for (int i = 0; i < FF_ARRAY_ELEMS(fc->DPB); i++) { + VVCFrame *frame = &fc->DPB[i]; + + if (frame->frame->buf[0] && frame->sequence == s->seq_decode && + frame->poc == poc) { + av_log(s->avctx, AV_LOG_ERROR, "Duplicate POC in a sequence: %d.\n", poc); + return AVERROR_INVALIDDATA; + } + } + + ref = alloc_frame(s, fc); + if (!ref) + return AVERROR(ENOMEM); + + *frame = ref->frame; + fc->ref = ref; + + if (s->no_output_before_recovery_flag && (IS_RASL(s) || !GDR_IS_RECOVERED(s))) + ref->flags = 0; + else if (ph->r->ph_pic_output_flag) + ref->flags = VVC_FRAME_FLAG_OUTPUT; + + if (!ph->r->ph_non_ref_pic_flag) + ref->flags |= VVC_FRAME_FLAG_SHORT_REF; + + ref->poc = poc; + ref->sequence = s->seq_decode; + ref->frame->crop_left = fc->ps.pps->r->pps_conf_win_left_offset; + ref->frame->crop_right = fc->ps.pps->r->pps_conf_win_right_offset; + ref->frame->crop_top = fc->ps.pps->r->pps_conf_win_top_offset; + ref->frame->crop_bottom = fc->ps.pps->r->pps_conf_win_bottom_offset; + + return 0; +} + +int ff_vvc_output_frame(VVCContext *s, VVCFrameContext *fc, AVFrame *out, const int no_output_of_prior_pics_flag, int flush) +{ + const VVCSPS *sps = fc->ps.sps; + do { + int nb_output = 0; + int min_poc = INT_MAX; + int min_idx, ret; + + if (no_output_of_prior_pics_flag) { + for (int i = 0; i < FF_ARRAY_ELEMS(fc->DPB); i++) { + VVCFrame *frame = &fc->DPB[i]; + if (!(frame->flags & VVC_FRAME_FLAG_BUMPING) && frame->poc != fc->ps.ph.poc && + frame->sequence == s->seq_output) { + ff_vvc_unref_frame(fc, frame, VVC_FRAME_FLAG_OUTPUT); + } + } + } + + for (int i = 0; i < FF_ARRAY_ELEMS(fc->DPB); i++) { + VVCFrame *frame = &fc->DPB[i]; + if ((frame->flags & VVC_FRAME_FLAG_OUTPUT) && + frame->sequence == s->seq_output) { + nb_output++; + if (frame->poc < min_poc || nb_output == 1) { + min_poc = frame->poc; + min_idx = i; + } + } + } + + /* wait for more frames before output */ + if (!flush && s->seq_output == s->seq_decode && sps && + nb_output <= sps->r->sps_dpb_params.dpb_max_dec_pic_buffering_minus1[sps->r->sps_max_sublayers_minus1] + 1) + return 0; + + if (nb_output) { + VVCFrame *frame = &fc->DPB[min_idx]; + + ret = av_frame_ref(out, frame->frame); + if (frame->flags & VVC_FRAME_FLAG_BUMPING) + ff_vvc_unref_frame(fc, frame, VVC_FRAME_FLAG_OUTPUT | VVC_FRAME_FLAG_BUMPING); + else + ff_vvc_unref_frame(fc, frame, VVC_FRAME_FLAG_OUTPUT); + if (ret < 0) + return ret; + + av_log(s->avctx, AV_LOG_DEBUG, + "Output frame with POC %d.\n", frame->poc); + return 1; + } + + if (s->seq_output != s->seq_decode) + s->seq_output = (s->seq_output + 1) & 0xff; + else + break; + } while (1); + return 0; +} + +void ff_vvc_bump_frame(VVCContext *s, VVCFrameContext *fc) +{ + const VVCSPS *sps = fc->ps.sps; + const int poc = fc->ps.ph.poc; + int dpb = 0; + int min_poc = INT_MAX; + + for (int i = 0; i < FF_ARRAY_ELEMS(fc->DPB); i++) { + VVCFrame *frame = &fc->DPB[i]; + if ((frame->flags) && + frame->sequence == s->seq_output && + frame->poc != poc) { + dpb++; + } + } + + if (sps && dpb >= sps->r->sps_dpb_params.dpb_max_dec_pic_buffering_minus1[sps->r->sps_max_sublayers_minus1] + 1) { + for (int i = 0; i < FF_ARRAY_ELEMS(fc->DPB); i++) { + VVCFrame *frame = &fc->DPB[i]; + if ((frame->flags) && + frame->sequence == s->seq_output && + frame->poc != poc) { + if (frame->flags == VVC_FRAME_FLAG_OUTPUT && frame->poc < min_poc) { + min_poc = frame->poc; + } + } + } + + for (int i = 0; i < FF_ARRAY_ELEMS(fc->DPB); i++) { + VVCFrame *frame = &fc->DPB[i]; + if (frame->flags & VVC_FRAME_FLAG_OUTPUT && + frame->sequence == s->seq_output && + frame->poc <= min_poc) { + frame->flags |= VVC_FRAME_FLAG_BUMPING; + } + } + + dpb--; + } +} + +static VVCFrame *find_ref_idx(VVCContext *s, VVCFrameContext *fc, int poc, uint8_t use_msb) +{ + const int mask = use_msb ? ~0 : fc->ps.sps->max_pic_order_cnt_lsb - 1; + + for (int i = 0; i < FF_ARRAY_ELEMS(fc->DPB); i++) { + VVCFrame *ref = &fc->DPB[i]; + if (ref->frame->buf[0] && ref->sequence == s->seq_decode) { + if ((ref->poc & mask) == poc) + return ref; + } + } + return NULL; +} + +static void mark_ref(VVCFrame *frame, int flag) +{ + frame->flags &= ~(VVC_FRAME_FLAG_LONG_REF | VVC_FRAME_FLAG_SHORT_REF); + frame->flags |= flag; +} + +static VVCFrame *generate_missing_ref(VVCContext *s, VVCFrameContext *fc, int poc) +{ + const VVCSPS *sps = fc->ps.sps; + VVCFrame *frame; + + frame = alloc_frame(s, fc); + if (!frame) + return NULL; + + if (!s->avctx->hwaccel) { + if (!sps->pixel_shift) { + for (int i = 0; frame->frame->buf[i]; i++) + memset(frame->frame->buf[i]->data, 1 << (sps->bit_depth - 1), + frame->frame->buf[i]->size); + } else { + for (int i = 0; frame->frame->data[i]; i++) + for (int y = 0; y < (sps->height >> sps->vshift[i]); y++) { + uint8_t *dst = frame->frame->data[i] + y * frame->frame->linesize[i]; + AV_WN16(dst, 1 << (sps->bit_depth - 1)); + av_memcpy_backptr(dst + 2, 2, 2*(sps->width >> sps->hshift[i]) - 2); + } + } + } + + frame->poc = poc; + frame->sequence = s->seq_decode; + frame->flags = 0; + + ff_vvc_report_frame_finished(frame); + + return frame; +} + +/* add a reference with the given poc to the list and mark it as used in DPB */ +static int add_candidate_ref(VVCContext *s, VVCFrameContext *fc, RefPicList *list, + int poc, int ref_flag, uint8_t use_msb) +{ + VVCFrame *ref = find_ref_idx(s, fc, poc, use_msb); + + if (ref == fc->ref || list->nb_refs >= VVC_MAX_REF_ENTRIES) + return AVERROR_INVALIDDATA; + + if (!ref) { + ref = generate_missing_ref(s, fc, poc); + if (!ref) + return AVERROR(ENOMEM); + } + + list->list[list->nb_refs] = poc; + list->ref[list->nb_refs] = ref; + list->isLongTerm[list->nb_refs] = ref_flag & VVC_FRAME_FLAG_LONG_REF; + list->nb_refs++; + + mark_ref(ref, ref_flag); + return 0; +} + +static int init_slice_rpl(const VVCFrameContext *fc, SliceContext *sc) +{ + VVCFrame *frame = fc->ref; + const VVCSH *sh = &sc->sh; + + if (sc->slice_idx >= frame->nb_rpl_elems) + return AVERROR_INVALIDDATA; + + for (int i = 0; i < sh->num_ctus_in_curr_slice; i++) { + const int rs = sh->ctb_addr_in_curr_slice[i]; + frame->rpl_tab[rs] = frame->rpl + sc->slice_idx; + } + + sc->rpl = frame->rpl_tab[sh->ctb_addr_in_curr_slice[0]]->refPicList; + + return 0; +} + +static int delta_poc_st(const H266RefPicListStruct *rpls, + const int lx, const int i, const VVCSPS *sps) +{ + int abs_delta_poc_st = rpls->abs_delta_poc_st[i]; + if (!((sps->r->sps_weighted_pred_flag || + sps->r->sps_weighted_bipred_flag) && i != 0)) + abs_delta_poc_st++; + return (1 - 2 * rpls->strp_entry_sign_flag[i]) * abs_delta_poc_st; +} + +static int poc_lt(int *prev_delta_poc_msb, const int poc, const H266RefPicLists *ref_lists, + const int lx, const int j, const int max_poc_lsb) +{ + const H266RefPicListStruct *rpls = ref_lists->rpl_ref_list + lx; + int lt_poc = rpls->ltrp_in_header_flag ? ref_lists->poc_lsb_lt[lx][j] : rpls->rpls_poc_lsb_lt[j]; + + if (ref_lists->delta_poc_msb_cycle_present_flag[lx][j]) { + const uint32_t delta = ref_lists->delta_poc_msb_cycle_lt[lx][j] + *prev_delta_poc_msb; + lt_poc += poc - delta * max_poc_lsb - (poc & (max_poc_lsb - 1)); + *prev_delta_poc_msb = delta; + } + return lt_poc; +} + +int ff_vvc_slice_rpl(VVCContext *s, VVCFrameContext *fc, SliceContext *sc) +{ + const VVCSPS *sps = fc->ps.sps; + const H266RawPPS *pps = fc->ps.pps->r; + const VVCPH *ph = &fc->ps.ph; + const H266RawSliceHeader *rsh = sc->sh.r; + const int max_poc_lsb = sps->max_pic_order_cnt_lsb; + const H266RefPicLists *ref_lists = + pps->pps_rpl_info_in_ph_flag ? &ph->r->ph_ref_pic_lists : &rsh->sh_ref_pic_lists; + int ret = 0; + + ret = init_slice_rpl(fc, sc); + if (ret < 0) + return ret; + + for (int lx = L0; lx <= L1; lx++) { + const H266RefPicListStruct *rpls = ref_lists->rpl_ref_list + lx; + RefPicList *rpl = sc->rpl + lx; + int poc_base = ph->poc; + int prev_delta_poc_msb = 0; + + rpl->nb_refs = 0; + for (int i = 0, j = 0; i < rpls->num_ref_entries; i++) { + int poc; + if (!rpls->inter_layer_ref_pic_flag[i]) { + int use_msb = 1; + int ref_flag; + if (rpls->st_ref_pic_flag[i]) { + poc = poc_base + delta_poc_st(rpls, lx, i, sps); + poc_base = poc; + ref_flag = VVC_FRAME_FLAG_SHORT_REF; + } else { + use_msb = ref_lists->delta_poc_msb_cycle_present_flag[lx][j]; + poc = poc_lt(&prev_delta_poc_msb, ph->poc, ref_lists, lx, j, max_poc_lsb); + ref_flag = VVC_FRAME_FLAG_LONG_REF; + j++; + } + ret = add_candidate_ref(s, fc, rpl, poc, ref_flag, use_msb); + if (ret < 0) + return ret; + } else { + avpriv_request_sample(fc->log_ctx, "Inter layer ref"); + ret = AVERROR_PATCHWELCOME; + return ret; + } + } + if ((!rsh->sh_collocated_from_l0_flag) == lx && + rsh->sh_collocated_ref_idx < rpl->nb_refs) + fc->ref->collocated_ref = rpl->ref[rsh->sh_collocated_ref_idx]; + } + return 0; +} + +int ff_vvc_frame_rpl(VVCContext *s, VVCFrameContext *fc, SliceContext *sc) +{ + int ret = 0; + + /* clear the reference flags on all frames except the current one */ + for (int i = 0; i < FF_ARRAY_ELEMS(fc->DPB); i++) { + VVCFrame *frame = &fc->DPB[i]; + + if (frame == fc->ref) + continue; + + mark_ref(frame, 0); + } + + if ((ret = ff_vvc_slice_rpl(s, fc, sc)) < 0) + goto fail; + +fail: + /* release any frames that are now unused */ + for (int i = 0; i < FF_ARRAY_ELEMS(fc->DPB); i++) + ff_vvc_unref_frame(fc, &fc->DPB[i], 0); + return ret; +} + +void ff_vvc_report_frame_finished(VVCFrame *frame) +{ + ff_vvc_report_progress(frame, VVC_PROGRESS_MV, INT_MAX); + ff_vvc_report_progress(frame, VVC_PROGRESS_PIXEL, INT_MAX); +} + +static int is_progress_done(const FrameProgress *p, const VVCProgressListener *l) +{ + return p->progress[l->vp] > l->y; +} + +static void add_listener(VVCProgressListener **prev, VVCProgressListener *l) +{ + l->next = *prev; + *prev = l; +} + +static VVCProgressListener* remove_listener(VVCProgressListener **prev, VVCProgressListener *l) +{ + *prev = l->next; + l->next = NULL; + return l; +} + +static VVCProgressListener* get_done_listener(FrameProgress *p, const VVCProgress vp) +{ + VVCProgressListener *list = NULL; + VVCProgressListener **prev = &p->listener[vp]; + + while (*prev) { + if (is_progress_done(p, *prev)) { + VVCProgressListener *l = remove_listener(prev, *prev); + add_listener(&list, l); + } else { + prev = &(*prev)->next; + } + } + return list; +} + +void ff_vvc_report_progress(VVCFrame *frame, const VVCProgress vp, const int y) +{ + FrameProgress *p = frame->progress; + VVCProgressListener *l = NULL; + + ff_mutex_lock(&p->lock); + + av_assert0(p->progress[vp] < y || p->progress[vp] == INT_MAX); + p->progress[vp] = y; + l = get_done_listener(p, vp); + ff_cond_signal(&p->cond); + + ff_mutex_unlock(&p->lock); + + while (l) { + l->progress_done(l); + l = l->next; + } +} + +void ff_vvc_add_progress_listener(VVCFrame *frame, VVCProgressListener *l) +{ + FrameProgress *p = frame->progress; + + ff_mutex_lock(&p->lock); + + if (is_progress_done(p, l)) { + ff_mutex_unlock(&p->lock); + l->progress_done(l); + } else { + add_listener(p->listener + l->vp, l); + ff_mutex_unlock(&p->lock); + } +} diff --git a/libavcodec/vvc/vvc_refs.h b/libavcodec/vvc/vvc_refs.h new file mode 100644 index 0000000000..cd3b5f5632 --- /dev/null +++ b/libavcodec/vvc/vvc_refs.h @@ -0,0 +1,57 @@ +/* + * VVC reference management + * + * Copyright (C) 2023 Nuo Mi + * + * 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_VVC_REFS_H +#define AVCODEC_VVC_VVC_REFS_H + +#include "vvcdec.h" + +int ff_vvc_output_frame(VVCContext *s, VVCFrameContext *fc, struct AVFrame *out, int no_output_of_prior_pics_flag, int flush); +void ff_vvc_bump_frame(VVCContext *s, VVCFrameContext *fc); +int ff_vvc_set_new_ref(VVCContext *s, VVCFrameContext *fc, struct AVFrame **frame); +const RefPicList *ff_vvc_get_ref_list(const VVCFrameContext *fc, const VVCFrame *ref, int x0, int y0); +int ff_vvc_frame_rpl(VVCContext *s, VVCFrameContext *fc, SliceContext *sc); +int ff_vvc_slice_rpl(VVCContext *s, VVCFrameContext *fc, SliceContext *sc); +void ff_vvc_unref_frame(VVCFrameContext *fc, VVCFrame *frame, int flags); +void ff_vvc_clear_refs(VVCFrameContext *fc); + +typedef enum VVCProgress { + VVC_PROGRESS_MV, + VVC_PROGRESS_PIXEL, + VVC_PROGRESS_LAST, +} VVCProgress; + +typedef struct VVCProgressListener VVCProgressListener; +typedef void (*progress_done_fn)(VVCProgressListener *l); + +struct VVCProgressListener { + VVCProgress vp; + int y; + progress_done_fn progress_done; + VVCProgressListener *next; //used by ff_vvc_add_progress_listener only +}; + +void ff_vvc_report_frame_finished(VVCFrame *frame); +void ff_vvc_report_progress(VVCFrame *frame, VVCProgress vp, int y); +void ff_vvc_add_progress_listener(VVCFrame *frame, VVCProgressListener *l); + +#endif // AVCODEC_VVC_VVC_REFS_H