From patchwork Mon Nov 6 00:56:06 2023 Content-Type: text/plain; charset="utf-8" MIME-Version: 1.0 Content-Transfer-Encoding: 7bit X-Patchwork-Submitter: "Dai, Jianhui J" X-Patchwork-Id: 44529 Delivered-To: ffmpegpatchwork2@gmail.com Received: by 2002:a05:6a20:3aa6:b0:181:818d:5e7f with SMTP id d38csp1173232pzh; Sun, 5 Nov 2023 16:56:55 -0800 (PST) X-Google-Smtp-Source: AGHT+IFIaLYOCtCaJkYx61qycTep1Ck0GlyVJ57HrbJi7lnl1F5o7DCOgQmzR2eBekhikh8FLB56 X-Received: by 2002:a05:6402:8ce:b0:53e:72be:2b31 with SMTP id d14-20020a05640208ce00b0053e72be2b31mr17510520edz.42.1699232215253; Sun, 05 Nov 2023 16:56:55 -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 y6-20020a056402440600b00538767844f0si3754964eda.346.2023.11.05.16.56.54; Sun, 05 Nov 2023 16:56:55 -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=@intel.com header.s=Intel header.b=eWt9ZWut; 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 Received: from [127.0.1.1] (localhost [127.0.0.1]) by ffbox0-bg.mplayerhq.hu (Postfix) with ESMTP id 6AD8668CD7F; Mon, 6 Nov 2023 02:56:50 +0200 (EET) X-Original-To: ffmpeg-devel@ffmpeg.org Delivered-To: ffmpeg-devel@ffmpeg.org Received: from mgamail.intel.com (mgamail.intel.com [192.55.52.115]) by ffbox0-bg.mplayerhq.hu (Postfix) with ESMTPS id DF50068CD8B for ; Mon, 6 Nov 2023 02:56:41 +0200 (EET) DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/simple; d=intel.com; i=@intel.com; q=dns/txt; s=Intel; t=1699232207; x=1730768207; h=from:to:subject:date:message-id: content-transfer-encoding:mime-version; bh=MysjAS+r0dTT6syGqpyAQLVV8hn88KWM9jsBFmtczzs=; b=eWt9ZWutVNtkXGM6q4cCjHRDJ3SNzlumC8pdZZ1RwHUzOl5L7CVmIPUd Ud1J+OMIfspWPfKrNKPCPmLj8VKxSyt/wQ+7bNrS9npiqwgdf2ZVoCTNJ YrV95aJDsodT+S+IYuDDs97PcQYp9klJbtsRg3KtPGZkA2ACPPAVGcbZS jLOQI47ohyH3YNu6V+eTXZm5M+uGEkrNCtxVtR3GEl2j5Sl1VUHjVrltf 6K1rrIc1KjIAzl388PpKLZEKhoAPNKGsq9XSVRyvGpok77rWc4+gNm+Eg 8VjJN2q/Jaul+fMs3Gow0+mOiZTzr53z70kJhf05bITqApinYUgSntLMo Q==; X-IronPort-AV: E=McAfee;i="6600,9927,10885"; a="389004773" X-IronPort-AV: E=Sophos;i="6.03,279,1694761200"; d="scan'208";a="389004773" Received: from fmsmga008.fm.intel.com ([10.253.24.58]) by fmsmga103.fm.intel.com with ESMTP/TLS/ECDHE-RSA-AES256-GCM-SHA384; 05 Nov 2023 16:56:38 -0800 X-ExtLoop1: 1 X-IronPort-AV: E=McAfee;i="6600,9927,10885"; a="828049978" X-IronPort-AV: E=Sophos;i="6.03,279,1694761200"; d="scan'208";a="828049978" Received: from orsmsx603.amr.corp.intel.com ([10.22.229.16]) by fmsmga008.fm.intel.com with ESMTP/TLS/AES256-GCM-SHA384; 05 Nov 2023 16:56:38 -0800 Received: from orsmsx610.amr.corp.intel.com (10.22.229.23) by ORSMSX603.amr.corp.intel.com (10.22.229.16) with Microsoft SMTP Server (version=TLS1_2, cipher=TLS_ECDHE_RSA_WITH_AES_128_GCM_SHA256) id 15.1.2507.34; Sun, 5 Nov 2023 16:56:37 -0800 Received: from ORSEDG602.ED.cps.intel.com (10.7.248.7) by orsmsx610.amr.corp.intel.com (10.22.229.23) with Microsoft SMTP Server (version=TLS1_2, cipher=TLS_ECDHE_RSA_WITH_AES_128_GCM_SHA256) id 15.1.2507.34 via Frontend Transport; Sun, 5 Nov 2023 16:56:37 -0800 Received: from NAM02-SN1-obe.outbound.protection.outlook.com (104.47.57.41) by edgegateway.intel.com (134.134.137.103) with Microsoft SMTP Server (version=TLS1_2, cipher=TLS_ECDHE_RSA_WITH_AES_256_GCM_SHA384) id 15.1.2507.34; Sun, 5 Nov 2023 16:56:36 -0800 ARC-Seal: i=1; a=rsa-sha256; s=arcselector9901; d=microsoft.com; cv=none; b=i46JEE014ztCFfLrBIHhjk/sCClRj7GhAWaYeN0++MUxZClRuuCR3vGlPPi6j8rC+y7DRN4QWlcJNwC43/c9hat8qbVRQgUpqro9HAqyf2nzg3wVX+RNAFnC0UTvNjrR5peIIn5lvQlJKRXR121ac/4Olz71I9aPOf6tcGeJVzKGRlLXStL1w9C1gSOaE5KH83Mg6axxtQqDPGuvXZ9akuojcrU2Skm0Xzi7CP06h34Y4NDhfx2tAILp0AZJ/KQSZAV/RMBCbSftpDIRpIBnjHuElcVqQ70qLm0Pxob7qTB2rBIx181LRc/bKgwCtxdL3XxYKEMgSXL6GVS4oQM3rQ== 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=r2I+qY10OtZP15dmi17IX3TYq8Vf9twgMz++46lYlQQ=; b=IlBKf+IX8VW4kSYuMvuicQsD+FuAKvOUPJO0Psu+c6K3bmJJe2sUIlcPmmbvzZAjnTX/rfXz7dKoKvdy6aBLX3uUw7c/vbwA029VWfuD48Nok/4uPWmRTItZoIijsv6KuOobYEnFXRVYRlGxsjUTR3XPvzn02ZJgzcHVXgEeWrj9mqpjBGmQZOrplSj+eUJpf0VmOFMVRinPQgrQVmFjxS0ZrBPS8Ae0nal77xPWvC5VwIPMLzqpXmpPh0K+/ejb+x/pxGuHZmw29uKjKsHg0o2ROkrA7xQxbw8Uid81VYmGehZSOCU/BpiE3C6iSFSRkx2H5AJZd2AJPn8bKK1wpQ== ARC-Authentication-Results: i=1; mx.microsoft.com 1; spf=pass smtp.mailfrom=intel.com; dmarc=pass action=none header.from=intel.com; dkim=pass header.d=intel.com; arc=none Received: from DM6PR11MB2681.namprd11.prod.outlook.com (2603:10b6:5:bd::33) by CH0PR11MB5252.namprd11.prod.outlook.com (2603:10b6:610:e3::15) with Microsoft SMTP Server (version=TLS1_2, cipher=TLS_ECDHE_RSA_WITH_AES_256_GCM_SHA384) id 15.20.6954.27; Mon, 6 Nov 2023 00:56:34 +0000 Received: from DM6PR11MB2681.namprd11.prod.outlook.com ([fe80::c129:6547:c4b5:f256]) by DM6PR11MB2681.namprd11.prod.outlook.com ([fe80::c129:6547:c4b5:f256%7]) with mapi id 15.20.6954.024; Mon, 6 Nov 2023 00:56:32 +0000 From: "Dai, Jianhui J" To: "ffmpeg-devel@ffmpeg.org" Thread-Topic: [PATCH v5] avcodec/cbs_vp8: Add support for VP8 codec bitstream Thread-Index: AdmG0tDkaPPFmNsRRsy2yJkNGMWOQw== Date: Mon, 6 Nov 2023 00:56:06 +0000 Message-ID: Accept-Language: en-US Content-Language: en-US X-MS-Has-Attach: X-MS-TNEF-Correlator: authentication-results: dkim=none (message not signed) header.d=none;dmarc=none action=none header.from=intel.com; x-ms-exchange-messagesentrepresentingtype: 1 x-ms-publictraffictype: Email x-ms-traffictypediagnostic: DM6PR11MB2681:EE_|CH0PR11MB5252:EE_ x-ms-office365-filtering-correlation-id: 09af237f-8be2-42e0-8044-08dbde6328a9 x-ms-exchange-senderadcheck: 1 x-ms-exchange-antispam-relay: 0 x-microsoft-antispam: BCL:0; x-microsoft-antispam-message-info: aszwbsrCQUlxdsZNfYe3UmosVmvAU6coCuNLYuFPTnUkNHIxoaU8+ZK4jl79h2ttWIaIh3Mu74zXykLlanZu5jiV7nZGDpegVU/WrF4n/FFTcPhm4vygnaZxTGlCx0ltfB4ZO0ICLZA4wnb8bGTLKwkIxsppzA8s526/TKtQjmthSDTgrRoOUsJWuWflF460OF3uV/JRuNtdMbJHD7WrXWYjBPrp2TISL6dWn7DO6ooP5nGh3J9p95FSwifitTsU+L/ru4eW5GvrD5Xy5qIXH3zks03iG+0SkVjLMpuRb8mWQbx+3hY/BbBv7nkh1584jeDcImusLtw9jaWaIYP+itR2ktbD1TKdAQvvyqgv0ys06b30rAiOaNQa5IuOkIe6h2XK3rgANcQyJ61Y6uewelG4ODC+IXoYyZLDS2WH8jgJJ8IYHE+e1NkLavbbQYnKeaWQBCwByFBbobw0CKGl4djttcy+/QF7vbBl0PiFTLnZMgLtVZqhjemr3o4oRCsP9s+WZ+JwHHqgqKrDUTGW7ystOPT6fdtOiaSkkAz+wBez2ZxgrMBfSbJ6T7Pp7zlGRFQVRXAprIOOWG2mfj4UYDXkCoYofd+RZyrnSoprjKw6qmqgGRQyvWjdlSx/3uSv2EJoHuhhwpMIabmf4wi3iA== x-forefront-antispam-report: CIP:255.255.255.255; CTRY:; LANG:en; SCL:1; SRV:; IPV:NLI; SFV:NSPM; H:DM6PR11MB2681.namprd11.prod.outlook.com; PTR:; CAT:NONE; SFS:(13230031)(396003)(366004)(346002)(376002)(136003)(39860400002)(230922051799003)(186009)(451199024)(1800799009)(64100799003)(66899024)(83380400001)(71200400001)(38100700002)(26005)(52536014)(8936002)(6916009)(316002)(8676002)(30864003)(2906002)(5660300002)(41300700001)(478600001)(6506007)(7696005)(64756008)(9686003)(6666004)(66476007)(66556008)(66946007)(66446008)(76116006)(86362001)(33656002)(19627235002)(82960400001)(122000001)(38070700009)(55016003)(2004002)(559001)(579004); DIR:OUT; SFP:1102; x-ms-exchange-antispam-messagedata-chunkcount: 1 x-ms-exchange-antispam-messagedata-0: nCCDGITiu3tipBt5mzbhvKAw2rtu+qWrqsqzkZawcelW2EKgePfPBz3ibuABqBCUU4a0Felum5rKM9H/cRctH+z4F6IXW6/auyS1wt4rRNvPTsR+OPLQqF3Ynfg69YHSZQ3sTO72EIO1gvnmmxS/A2gABJUcjcuG7c5RnH84nPCEvBw1Y2iTDhIBUhadnRjhNbL15hE7rB68npRhz64qjivszH8mm8JfCi/Y+pHnPpdnaH1kV8tqQsE1embtXV5FVGu9H4feBegNSffPgGVDzyRFqBSFV8ILmCE6DmykV52kAsL5FZx/3ayGvnZ3Or0nHU7KtjfcMzv2b/9yCq+j7Hg/AI3diPoNYjWQNBlYQUyUuBvUHJfXgrzIOBs4r4r4EU8vmQAIj1tpcx1MRsNxCUrRN2wAD9Dj/0u2C3ruYJoH73OD751GIRK+9xNP1WkvjDbtIQzJUcpAQ+GB39LR+8yZPyTK6mpm6NpFKKQasl1qFSihQEBCXTkPRMaN5W5C5XVkQ1FrW9YkXmc6B1wdVEzCzyLhF6OuQBlVUIZBIY9ZjuTNA8qLLhHuzSPqoRAZYAf+FDAAS70P7Eq4wX5g31Uh1fbjeB1xtwRg2FRRJKb//NIy3TcPEgBRIxqn3uk/RdCHufJj06N7LHwepA/5ocv1e2atvam7JTSav7ggqVQ6NgQRgSC7Toq+VIr4S3+QwVAhFi/+jNY3lOSUbc/gqRH2UgbQqzrDWupGN+4YDJcx4ByxW45LOM2S4mnDJ9191QKAD5+3IrEsWUeYVQ7VxV5a4iGXde5IQc26iKTG6AQnqdHzlFGIj8tf+uLtfH+1y9VAIcN0PFslQCL88LAzsRDUDbzxa2TQOG2tZVslYfRqLm/A2/H1Fobx2vQXJlKJexHKLXTozc0E6Zef+5f8MLn/i4oz0GA5xo8KRFqKJYn0zZRK/wHAloeP/tzdFu81FmL1X3/Zzc3vImZm0OHyR2lOLcIsiHlNYkPNgqpcaXXS64MGV5idJVPjq6u+t1eFmYggcQerQmzux9fq7QxQcdQxN4yIhWuSl4Zvdfi1pyxGW3dRGi7PBTUyt7vikJ+CevrKzaiXN3Rn1ssqbiBWI9AmJKT8E4ZFD4jZnrpeEBjzrByLlsnsvH2HqtVYtZ1NQI1zIv13soKl3Xda4d4gOpNTjJXHqEMqKzPcRvZZLyAruH3gQ3hmHo8SoyP253J1rKLUhccU3ZyiwUqA3uIRGtAK39D8coG4Sp4YL42O8JmVW7oyK7yKCduzHl4sEc/Lz/2epU63Kjp2b94sMZh1VTHdglAQcFf5Gs1ug3CqL32U8Yw4/fgEyydw3RtrR8sR+9wLmS9uV5kocwH51F44/JILfy7rckIwoBG01ciqJDxn1kvBeFuwRaAikS0YKIoBnjvkIlnja52KSyL/dWz0Si4WAY3d3na3of1c9yHtwI6hOK/IV79zerBWWM15lzvIHOs60tJaan4ovBccjqBHQlc32qbG5kS5HdIj1nz51LFmTFkp6zRA1rKBuhJvWFgIczC8IpzUhcGjNwcflnA/MET1oKyLgpOYNEBh/QqWaKTnzbzKMm6fSfXSazN+4nK9 MIME-Version: 1.0 X-MS-Exchange-CrossTenant-AuthAs: Internal X-MS-Exchange-CrossTenant-AuthSource: DM6PR11MB2681.namprd11.prod.outlook.com X-MS-Exchange-CrossTenant-Network-Message-Id: 09af237f-8be2-42e0-8044-08dbde6328a9 X-MS-Exchange-CrossTenant-originalarrivaltime: 06 Nov 2023 00:56:06.8204 (UTC) X-MS-Exchange-CrossTenant-fromentityheader: Hosted X-MS-Exchange-CrossTenant-id: 46c98d88-e344-4ed4-8496-4ed7712e255d X-MS-Exchange-CrossTenant-mailboxtype: HOSTED X-MS-Exchange-CrossTenant-userprincipalname: jLrOj32XsA7KRIjM9HyamiFAbuKk9ORdy00Qde6Qn06OdUqt/BXVJCLtpx86ZtCFgPtAryoRiXRlTdzCAoKMmw== X-MS-Exchange-Transport-CrossTenantHeadersStamped: CH0PR11MB5252 X-OriginatorOrg: intel.com Subject: [FFmpeg-devel] [PATCH v5] avcodec/cbs_vp8: Add support for VP8 codec bitstream 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 Errors-To: ffmpeg-devel-bounces@ffmpeg.org Sender: "ffmpeg-devel" X-TUID: qoY5vcpehq8C This commit adds support for VP8 bitstream read methods to the cbs codec. This enables the trace_headers bitstream filter to support VP8, in addition to AV1, H.264, H.265, and VP9. This can be useful for debugging VP8 stream issues. The CBS VP8 implements a simple VP8 boolean decoder using GetBitContext to read the bitstream. Only the read methods `read_unit` and `split_fragment` are implemented. The write methods `write_unit` and `assemble_fragment` return the error code AVERROR_PATCHWELCOME. This is because CBS VP8 write is unlikely to be used by any applications at the moment. The write methods can be added later if there is a real need for them. TESTS: ffmpeg -i fate-suite/vp8/frame_size_change.webm -vcodec copy -bsf:v trace_headers -f null - Signed-off-by: Jianhui Dai --- configure | 4 +- libavcodec/Makefile | 1 + libavcodec/cbs.c | 6 + libavcodec/cbs_internal.h | 1 + libavcodec/cbs_vp8.c | 548 +++++++++++++++++++++++++++ libavcodec/cbs_vp8.h | 133 +++++++ libavcodec/cbs_vp8_syntax_template.c | 248 ++++++++++++ 7 files changed, 940 insertions(+), 1 deletion(-) create mode 100644 libavcodec/cbs_vp8.c create mode 100644 libavcodec/cbs_vp8.h create mode 100644 libavcodec/cbs_vp8_syntax_template.c diff --git a/configure b/configure index 7afeebebcd..8a4ea125ad 100755 --- a/configure +++ b/configure @@ -2471,6 +2471,7 @@ CONFIG_EXTRA=" cbs_h266 cbs_jpeg cbs_mpeg2 + cbs_vp8 cbs_vp9 deflate_wrapper dirac_parse @@ -2756,6 +2757,7 @@ cbs_h265_select="cbs" cbs_h266_select="cbs" cbs_jpeg_select="cbs" cbs_mpeg2_select="cbs" +cbs_vp8_select="cbs" cbs_vp9_select="cbs" deflate_wrapper_deps="zlib" dirac_parse_select="golomb" @@ -3339,7 +3341,7 @@ h264_redundant_pps_bsf_select="cbs_h264" hevc_metadata_bsf_select="cbs_h265" mjpeg2jpeg_bsf_select="jpegtables" mpeg2_metadata_bsf_select="cbs_mpeg2" -trace_headers_bsf_select="cbs" +trace_headers_bsf_select="cbs cbs_vp8" vp9_metadata_bsf_select="cbs_vp9" vvc_metadata_bsf_select="cbs_h266" diff --git a/libavcodec/Makefile b/libavcodec/Makefile index ab7fba394a..7b0f9ab550 100644 --- a/libavcodec/Makefile +++ b/libavcodec/Makefile @@ -81,6 +81,7 @@ OBJS-$(CONFIG_CBS_H265) += cbs_h2645.o cbs_sei.o h2645_parse.o OBJS-$(CONFIG_CBS_H266) += cbs_h2645.o cbs_sei.o h2645_parse.o OBJS-$(CONFIG_CBS_JPEG) += cbs_jpeg.o OBJS-$(CONFIG_CBS_MPEG2) += cbs_mpeg2.o +OBJS-$(CONFIG_CBS_VP8) += cbs_vp8.o OBJS-$(CONFIG_CBS_VP9) += cbs_vp9.o OBJS-$(CONFIG_CRYSTALHD) += crystalhd.o OBJS-$(CONFIG_DEFLATE_WRAPPER) += zlib_wrapper.o diff --git a/libavcodec/cbs.c b/libavcodec/cbs.c index cdd7adebeb..de7b1361aa 100644 --- a/libavcodec/cbs.c +++ b/libavcodec/cbs.c @@ -50,6 +50,9 @@ static const CodedBitstreamType *const cbs_type_table[] = { #if CONFIG_CBS_MPEG2 &ff_cbs_type_mpeg2, #endif +#if CONFIG_CBS_VP8 + &ff_cbs_type_vp8, +#endif #if CONFIG_CBS_VP9 &ff_cbs_type_vp9, #endif @@ -74,6 +77,9 @@ const enum AVCodecID ff_cbs_all_codec_ids[] = { #if CONFIG_CBS_MPEG2 AV_CODEC_ID_MPEG2VIDEO, #endif +#if CONFIG_CBS_VP8 + AV_CODEC_ID_VP8, +#endif #if CONFIG_CBS_VP9 AV_CODEC_ID_VP9, #endif diff --git a/libavcodec/cbs_internal.h b/libavcodec/cbs_internal.h index 07220f1f3e..d982262bd9 100644 --- a/libavcodec/cbs_internal.h +++ b/libavcodec/cbs_internal.h @@ -341,6 +341,7 @@ extern const CodedBitstreamType ff_cbs_type_h265; extern const CodedBitstreamType ff_cbs_type_h266; extern const CodedBitstreamType ff_cbs_type_jpeg; extern const CodedBitstreamType ff_cbs_type_mpeg2; +extern const CodedBitstreamType ff_cbs_type_vp8; extern const CodedBitstreamType ff_cbs_type_vp9; diff --git a/libavcodec/cbs_vp8.c b/libavcodec/cbs_vp8.c new file mode 100644 index 0000000000..79a2a9bba6 --- /dev/null +++ b/libavcodec/cbs_vp8.c @@ -0,0 +1,548 @@ +/* + * 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 "libavutil/avassert.h" + +#include "cbs.h" +#include "cbs_internal.h" +#include "cbs_vp8.h" + +#include + +#define DEFAULT_PROB 0x80 + +static const uint8_t vp8_token_update_probs[4][8][3][11] = { + { + { + {255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255}, + {255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255}, + {255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255}, + }, + { + {176, 246, 255, 255, 255, 255, 255, 255, 255, 255, 255}, + {223, 241, 252, 255, 255, 255, 255, 255, 255, 255, 255}, + {249, 253, 253, 255, 255, 255, 255, 255, 255, 255, 255}, + }, + { + {255, 244, 252, 255, 255, 255, 255, 255, 255, 255, 255}, + {234, 254, 254, 255, 255, 255, 255, 255, 255, 255, 255}, + {253, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255}, + }, + { + {255, 246, 254, 255, 255, 255, 255, 255, 255, 255, 255}, + {239, 253, 254, 255, 255, 255, 255, 255, 255, 255, 255}, + {254, 255, 254, 255, 255, 255, 255, 255, 255, 255, 255}, + }, + { + {255, 248, 254, 255, 255, 255, 255, 255, 255, 255, 255}, + {251, 255, 254, 255, 255, 255, 255, 255, 255, 255, 255}, + {255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255}, + }, + { + {255, 253, 254, 255, 255, 255, 255, 255, 255, 255, 255}, + {251, 254, 254, 255, 255, 255, 255, 255, 255, 255, 255}, + {254, 255, 254, 255, 255, 255, 255, 255, 255, 255, 255}, + }, + { + {255, 254, 253, 255, 254, 255, 255, 255, 255, 255, 255}, + {250, 255, 254, 255, 254, 255, 255, 255, 255, 255, 255}, + {254, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255}, + }, + { + {255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255}, + {255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255}, + {255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255}, + }, + }, + { + { + {217, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255}, + {225, 252, 241, 253, 255, 255, 254, 255, 255, 255, 255}, + {234, 250, 241, 250, 253, 255, 253, 254, 255, 255, 255}, + }, + { + {255, 254, 255, 255, 255, 255, 255, 255, 255, 255, 255}, + {223, 254, 254, 255, 255, 255, 255, 255, 255, 255, 255}, + {238, 253, 254, 254, 255, 255, 255, 255, 255, 255, 255}, + }, + { + {255, 248, 254, 255, 255, 255, 255, 255, 255, 255, 255}, + {249, 254, 255, 255, 255, 255, 255, 255, 255, 255, 255}, + {255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255}, + }, + { + {255, 253, 255, 255, 255, 255, 255, 255, 255, 255, 255}, + {247, 254, 255, 255, 255, 255, 255, 255, 255, 255, 255}, + {255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255}, + }, + { + {255, 253, 254, 255, 255, 255, 255, 255, 255, 255, 255}, + {252, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255}, + {255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255}, + }, + { + {255, 254, 254, 255, 255, 255, 255, 255, 255, 255, 255}, + {253, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255}, + {255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255}, + }, + { + {255, 254, 253, 255, 255, 255, 255, 255, 255, 255, 255}, + {250, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255}, + {254, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255}, + }, + { + {255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255}, + {255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255}, + {255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255}, + }, + }, + { + { + {186, 251, 250, 255, 255, 255, 255, 255, 255, 255, 255}, + {234, 251, 244, 254, 255, 255, 255, 255, 255, 255, 255}, + {251, 251, 243, 253, 254, 255, 254, 255, 255, 255, 255}, + }, + { + {255, 253, 254, 255, 255, 255, 255, 255, 255, 255, 255}, + {236, 253, 254, 255, 255, 255, 255, 255, 255, 255, 255}, + {251, 253, 253, 254, 254, 255, 255, 255, 255, 255, 255}, + }, + { + {255, 254, 254, 255, 255, 255, 255, 255, 255, 255, 255}, + {254, 254, 254, 255, 255, 255, 255, 255, 255, 255, 255}, + {255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255}, + }, + { + {255, 254, 255, 255, 255, 255, 255, 255, 255, 255, 255}, + {254, 254, 255, 255, 255, 255, 255, 255, 255, 255, 255}, + {254, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255}, + }, + { + {255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255}, + {254, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255}, + {255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255}, + }, + { + {255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255}, + {255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255}, + {255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255}, + }, + { + {255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255}, + {255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255}, + {255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255}, + }, + { + {255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255}, + {255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255}, + {255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255}, + }, + }, + { + { + {248, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255}, + {250, 254, 252, 254, 255, 255, 255, 255, 255, 255, 255}, + {248, 254, 249, 253, 255, 255, 255, 255, 255, 255, 255}, + }, + { + {255, 253, 253, 255, 255, 255, 255, 255, 255, 255, 255}, + {246, 253, 253, 255, 255, 255, 255, 255, 255, 255, 255}, + {252, 254, 251, 254, 254, 255, 255, 255, 255, 255, 255}, + }, + { + {255, 254, 252, 255, 255, 255, 255, 255, 255, 255, 255}, + {248, 254, 253, 255, 255, 255, 255, 255, 255, 255, 255}, + {253, 255, 254, 254, 255, 255, 255, 255, 255, 255, 255}, + }, + { + {255, 251, 254, 255, 255, 255, 255, 255, 255, 255, 255}, + {245, 251, 254, 255, 255, 255, 255, 255, 255, 255, 255}, + {253, 253, 254, 255, 255, 255, 255, 255, 255, 255, 255}, + }, + { + {255, 251, 253, 255, 255, 255, 255, 255, 255, 255, 255}, + {252, 253, 254, 255, 255, 255, 255, 255, 255, 255, 255}, + {255, 254, 255, 255, 255, 255, 255, 255, 255, 255, 255}, + }, + { + {255, 252, 255, 255, 255, 255, 255, 255, 255, 255, 255}, + {249, 255, 254, 255, 255, 255, 255, 255, 255, 255, 255}, + {255, 255, 254, 255, 255, 255, 255, 255, 255, 255, 255}, + }, + { + {255, 255, 253, 255, 255, 255, 255, 255, 255, 255, 255}, + {250, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255}, + {255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255}, + }, + { + {255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255}, + {254, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255}, + {255, 255, 255, 255, 255, 255, 255, 255, 255, 255, 255}, + }, + }, +}; + +// Implements VP8 boolean decoder using GetBitContext to read the bitstream. +typedef struct CBSVP8BoolDecoder { + GetBitContext *gbc; + + uint8_t value; + uint8_t range; + + uint8_t count; // Store the number of bits in the `value` buffer. + +} CBSVP8BoolDecoder; + +static int cbs_vp8_bool_decoder_init(CBSVP8BoolDecoder *decoder, GetBitContext *gbc) +{ + av_assert0(decoder); + av_assert0(gbc); + + decoder->gbc = gbc; + decoder->value = 0; + decoder->range = 255; + + decoder->count = 0; + + return 0; +} + +static bool cbs_vp8_bool_decoder_fill_value(CBSVP8BoolDecoder *decoder) +{ + int bits = 8 - decoder->count; + + av_assert0(decoder->count <= 8); + if (decoder->count == 8) { + return true; + } + + if (get_bits_left(decoder->gbc) >= bits) { + decoder->value |= get_bits(decoder->gbc, bits); + decoder->count += bits; + } + + return (decoder->count == 8); +} + +static int cbs_vp8_bool_decoder_read_bool(CBSVP8BoolDecoder *decoder, + const uint8_t prob, uint8_t *output) +{ + uint8_t split = 1 + (((decoder->range - 1) * prob) >> 8); + + if (!cbs_vp8_bool_decoder_fill_value(decoder)) { + return AVERROR_INVALIDDATA; + } + + av_assert0(decoder->count == 8); + if (decoder->value >= split) { + *output = 1; + decoder->range -= split; + decoder->value -= split; + } else { + *output = 0; + decoder->range = split; + } + + while (decoder->range < 128) { + decoder->value <<= 1; + decoder->range <<= 1; + --decoder->count; + } + + return 0; +} + +static int cbs_vp8_bool_decoder_read_literal(CBSVP8BoolDecoder *decoder, + const uint8_t prob, + uint32_t num_bits, + uint32_t *output) +{ + int ret = 0; + + av_assert0(num_bits <= 32); + + *output = 0; + for (; num_bits > 0; --num_bits) { + uint8_t bit_output = 0; + if ((ret = cbs_vp8_bool_decoder_read_bool(decoder, prob, + &bit_output)) != 0) { + return ret; + } + + *output = (*output << 1) | bit_output; + } + + return 0; +} + +static int cbs_vp8_bool_decoder_read_unsigned( + CodedBitstreamContext *ctx, CBSVP8BoolDecoder *bool_decoder, int width, + uint8_t prob, const char *name, const int *subscripts, uint32_t *write_to, + bool trace_enable) +{ + int ret = 0; + GetBitContext *gbc = bool_decoder->gbc; + uint32_t value; + + CBS_TRACE_READ_START(); + + av_assert0(width >= 0 && width <= 8); + + ret = cbs_vp8_bool_decoder_read_literal(bool_decoder, prob, width, &value); + if (ret != 0) { + return ret; + } + + if (trace_enable) { + CBS_TRACE_READ_END(); + } + + *write_to = value; + return 0; +} + +static int cbs_vp8_bool_decoder_read_signed( + CodedBitstreamContext *ctx, CBSVP8BoolDecoder *bool_decoder, int width, + uint8_t prob, const char *name, const int *subscripts, int32_t *write_to) +{ + int ret = 0; + GetBitContext *gbc = bool_decoder->gbc; + int32_t value; + uint8_t sign = 0; + + CBS_TRACE_READ_START(); + + av_assert0(width >= 0 && width <= 8); + + ret = cbs_vp8_bool_decoder_read_literal(bool_decoder, prob, width, &value); + if (ret != 0) { + return ret; + } + + ret = cbs_vp8_bool_decoder_read_bool(bool_decoder, prob, &sign); + if (ret != 0) { + return ret; + } + + if (sign) { + value = -value; + } + + CBS_TRACE_READ_END(); + + *write_to = value; + return 0; +} + +static int cbs_vp8_read_unsigned_le(CodedBitstreamContext *ctx, GetBitContext *gbc, + int width, const char *name, + const int *subscripts, uint32_t *write_to) +{ + int32_t value; + + CBS_TRACE_READ_START(); + + av_assert0(width > 0 && width <= 24); + + if (get_bits_left(gbc) < width) { + av_log(ctx->log_ctx, AV_LOG_ERROR, "Invalid value: bitstream ended.\n"); + return AVERROR_INVALIDDATA; + } + + value = get_bits_le(gbc, width); + + CBS_TRACE_READ_END(); + + *write_to = value; + return 0; +} + +#define HEADER(name) \ + do { \ + ff_cbs_trace_header(ctx, name); \ + } while (0) + +#define CHECK(call) \ + do { \ + int err = (call); \ + if (err < 0) \ + return err; \ + } while (0) + +#define FUNC_NAME(rw, codec, name) cbs_##codec##_##rw##_##name +#define FUNC_VP8(rw, name) FUNC_NAME(rw, vp8, name) +#define FUNC(name) FUNC_VP8(READWRITE, name) + +#define SUBSCRIPTS(subs, ...) \ + (subs > 0 ? ((int[subs + 1]){subs, __VA_ARGS__}) : NULL) + +#define f(width, name) xf(width, name, 0) + +// bool [de|en]coder methods. +#define bc_f(width, name) bc_unsigned_subs(width, DEFAULT_PROB, true, name, 0) +#define bc_s(width, name) bc_signed_subs(width, DEFAULT_PROB, name, 0) +#define bc_fs(width, name, subs, ...) \ + bc_unsigned_subs(width, DEFAULT_PROB, true, name, subs, __VA_ARGS__) +#define bc_ss(width, name, subs, ...) \ + bc_signed_subs(width, DEFAULT_PROB, name, subs, __VA_ARGS__) + +// bool [de|en]coder methods for boolean value and disable tracing. +#define bc_b(name) bc_unsigned_subs(1, DEFAULT_PROB, false, name, 0) +#define bc_b_prob(prob, name) bc_unsigned_subs(1, prob, false, name, 0) + +#define READ +#define READWRITE read +#define RWContext GetBitContext +#define CBSVP8BoolCodingRW CBSVP8BoolDecoder + +#define xf(width, name, subs, ...) \ + do { \ + uint32_t value; \ + CHECK(cbs_vp8_read_unsigned_le(ctx, rw, width, #name, \ + SUBSCRIPTS(subs, __VA_ARGS__), &value)); \ + current->name = value; \ + } while (0) + +#define fixed(width, name, value) \ + do { \ + uint32_t fixed_value; \ + CHECK(ff_cbs_read_unsigned(ctx, rw, width, #name, 0, &fixed_value, \ + value, value)); \ + } while (0) + +#define bc_unsigned_subs(width, prob, enable_trace, name, subs, ...) \ + do { \ + uint32_t value; \ + CHECK(cbs_vp8_bool_decoder_read_unsigned( \ + ctx, bool_coding_rw, width, prob, #name, \ + SUBSCRIPTS(subs, __VA_ARGS__), &value, enable_trace)); \ + current->name = value; \ + } while (0) + +#define bc_signed_subs(width, prob, name, subs, ...) \ + do { \ + int32_t value; \ + CHECK(cbs_vp8_bool_decoder_read_signed( \ + ctx, bool_coding_rw, width, prob, #name, \ + SUBSCRIPTS(subs, __VA_ARGS__), &value)); \ + current->name = value; \ + } while (0) + +#include "cbs_vp8_syntax_template.c" + +static int cbs_vp8_split_fragment(CodedBitstreamContext *ctx, + CodedBitstreamFragment *frag, int header) +{ + int err; + + if (frag->data_size == 0) + return AVERROR_INVALIDDATA; + + err = ff_cbs_append_unit_data(frag, 0, frag->data, frag->data_size, + frag->data_ref); + if (err < 0) + return err; + + return 0; +} + +static int cbs_vp8_read_unit(CodedBitstreamContext *ctx, + CodedBitstreamUnit *unit) +{ + VP8RawFrame *frame; + GetBitContext gbc; + CBSVP8BoolDecoder bool_decoder; + int err, pos; + + err = ff_cbs_alloc_unit_content(ctx, unit); + if (err < 0) + return err; + frame = unit->content; + + // Create GetBitContext for uncompressed header. + err = init_get_bits8_le(&gbc, unit->data, 8 * unit->data_size); + if (err < 0) + return err; + + err = cbs_vp8_read_uncompressed_header(ctx, &gbc, frame); + if (err < 0) + return err; + + pos = get_bits_count(&gbc); + av_assert0(pos % 8 == 0); + + // Create boolean decoder for compressed header. + err = cbs_vp8_bool_decoder_init(&bool_decoder, &gbc); + if (err < 0) + return err; + + err = cbs_vp8_read_compressed_header(ctx, &bool_decoder, frame); + if (err < 0) + return err; + + pos = get_bits_count(&gbc); + pos /= 8; + av_assert0(pos <= unit->data_size); + + frame->data_ref = av_buffer_ref(unit->data_ref); + if (!frame->data_ref) + return AVERROR(ENOMEM); + + frame->data = unit->data + pos; + frame->data_size = unit->data_size - pos; + + return 0; +} + +static int cbs_vp8_write_unit(CodedBitstreamContext *ctx, + CodedBitstreamUnit *unit, PutBitContext *pbc) +{ + return AVERROR_PATCHWELCOME; +} + +static int cbs_vp8_assemble_fragment(CodedBitstreamContext *ctx, + CodedBitstreamFragment *frag) +{ + return AVERROR_PATCHWELCOME; +} + +static void cbs_vp8_flush(CodedBitstreamContext *ctx) +{ + // Do nothing. +} + +static const CodedBitstreamUnitTypeDescriptor cbs_vp8_unit_types[] = { + CBS_UNIT_TYPE_INTERNAL_REF(0, VP8RawFrame, data), + CBS_UNIT_TYPE_END_OF_LIST, +}; + +const CodedBitstreamType ff_cbs_type_vp8 = { + .codec_id = AV_CODEC_ID_VP8, + + .priv_data_size = 0, + + .unit_types = cbs_vp8_unit_types, + + .split_fragment = &cbs_vp8_split_fragment, + .read_unit = &cbs_vp8_read_unit, + .write_unit = &cbs_vp8_write_unit, + + .flush = &cbs_vp8_flush, + + .assemble_fragment = &cbs_vp8_assemble_fragment, +}; diff --git a/libavcodec/cbs_vp8.h b/libavcodec/cbs_vp8.h new file mode 100644 index 0000000000..9c09e21a1c --- /dev/null +++ b/libavcodec/cbs_vp8.h @@ -0,0 +1,133 @@ +/* + * 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_CBS_VP8_H +#define AVCODEC_CBS_VP8_H + +#include +#include + +#include "cbs.h" + +enum { + VP8_START_CODE_0 = 0x9D, + VP8_START_CODE_1 = 0x01, + VP8_START_CODE_2 = 0x2A, +}; + +enum { + VP8_KEY_FRAME = 0, + VP8_NON_KEY_FRAME = 1, +}; + +typedef struct VP8RawFrameHeader { + // frame tag + uint8_t frame_type; + uint8_t profile; + uint8_t show_frame; + uint32_t first_partition_length_in_bytes; + + uint16_t width; + uint8_t horizontal_scale; + uint16_t height; + uint8_t vertical_scale; + + // frame header + uint8_t color_space; + uint8_t clamping_type; + + // segmentation + uint8_t segmentation_enable; + uint8_t update_segment_map; + uint8_t update_segment_feature_data; + uint8_t segment_feature_mode; + uint8_t segment_qp_update[4]; + int8_t segment_qp[4]; + uint8_t segment_loop_filter_level_update[4]; + int8_t segment_loop_filter_level[4]; + uint8_t segment_probs_update[3]; + uint8_t segment_probs[3]; + + // loop filter + uint8_t loop_filter_type; + uint8_t loop_filter_level; + uint8_t loop_filter_sharpness; + uint8_t mode_ref_lf_delta_enable; + uint8_t mode_ref_lf_delta_update; + uint8_t ref_lf_deltas_update[4]; + int8_t ref_lf_deltas[4]; + uint8_t mode_lf_deltas_update[4]; + int8_t mode_lf_deltas[4]; + + uint8_t log2_token_partitions; + + // qp + uint8_t base_qindex; + uint8_t y1dc_delta_q_present; + int8_t y1dc_delta_q; + uint8_t y2dc_delta_q_present; + int8_t y2dc_delta_q; + uint8_t y2ac_delta_q_present; + int8_t y2ac_delta_q; + uint8_t uvdc_delta_q_present; + int8_t uvdc_delta_q; + uint8_t uvac_delta_q_present; + int8_t uvac_delta_q; + + // ref + uint8_t refresh_golden_frame; + uint8_t refresh_alternate_frame; + uint8_t copy_buffer_to_golden; + uint8_t copy_buffer_to_alternate; + uint8_t ref_frame_sign_bias_golden; + uint8_t ref_frame_sign_bias_alternate; + uint8_t refresh_last_frame; + + uint8_t refresh_entropy_probs; + + // token probs + uint8_t coeff_prob_update[4][8][3][11]; + uint8_t coeff_prob[4][8][3][11]; + + uint8_t mb_no_skip_coeff; + uint8_t prob_skip_false; + + uint8_t prob_intra; + uint8_t prob_last; + uint8_t prob_golden; + + uint8_t intra_16x16_prob_update; + uint8_t intra_16x16_prob[4]; + + uint8_t intra_chrome_prob_update; + uint8_t intra_chrome_prob[3]; + + // mv probs + uint8_t mv_prob_update[2][19]; + uint8_t mv_prob[2][19]; +} VP8RawFrameHeader; + +typedef struct VP8RawFrame { + VP8RawFrameHeader header; + + uint8_t *data; + AVBufferRef *data_ref; + size_t data_size; +} VP8RawFrame; + +#endif /* AVCODEC_CBS_VP8_H */ diff --git a/libavcodec/cbs_vp8_syntax_template.c b/libavcodec/cbs_vp8_syntax_template.c new file mode 100644 index 0000000000..0bdf699f39 --- /dev/null +++ b/libavcodec/cbs_vp8_syntax_template.c @@ -0,0 +1,248 @@ +/* + * 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 + */ + +static int FUNC(update_segmentation)(CodedBitstreamContext *ctx, + CBSVP8BoolCodingRW *bool_coding_rw, + VP8RawFrameHeader *current) +{ + bc_f(1, update_segment_map); + bc_f(1, update_segment_feature_data); + + if (current->update_segment_feature_data) { + bc_f(1, segment_feature_mode); + // quantizer + for (int i = 0; i < 4; i++) { + bc_b(segment_qp_update[i]); + if (current->segment_qp_update[i]) + bc_ss(7, segment_qp[i], 1, i); + } + // loop filter + for (int i = 0; i < 4; i++) { + bc_b(segment_loop_filter_level_update[i]); + if (current->segment_loop_filter_level_update[i]) + bc_ss(6, segment_loop_filter_level[i], 1, i); + } + } + + if (current->update_segment_map) { + for (int i = 0; i < 3; i++) { + bc_b(segment_probs_update[i]); + if (current->segment_probs_update[i]) + bc_fs(8, segment_probs[i], 1, i); + } + } + + return 0; +} + +static int FUNC(mode_ref_lf_deltas)(CodedBitstreamContext *ctx, + CBSVP8BoolCodingRW *bool_coding_rw, + VP8RawFrameHeader *current) +{ + bc_f(1, mode_ref_lf_delta_enable); + if (current->mode_ref_lf_delta_enable) { + bc_b(mode_ref_lf_delta_update); + if (current->mode_ref_lf_delta_update) { + // ref_lf_deltas + for (int i = 0; i < 4; i++) { + bc_b(ref_lf_deltas_update[i]); + if (current->ref_lf_deltas_update[i]) + bc_ss(6, ref_lf_deltas[i], 1, i); + } + // mode_lf_deltas + for (int i = 0; i < 4; i++) { + bc_b(mode_lf_deltas_update[i]); + if (current->mode_lf_deltas_update[i]) + bc_ss(6, mode_lf_deltas[i], 1, i); + } + } + } + + return 0; +} + +static int FUNC(quantization_params)(CodedBitstreamContext *ctx, + CBSVP8BoolCodingRW *bool_coding_rw, + VP8RawFrameHeader *current) +{ + bc_f(7, base_qindex); + + bc_b(y1dc_delta_q_present); + if (current->y1dc_delta_q_present) + bc_s(4, y1dc_delta_q); + + bc_b(y2dc_delta_q_present); + if (current->y2dc_delta_q_present) + bc_s(4, y2dc_delta_q); + + bc_b(y2ac_delta_q_present); + if (current->y2ac_delta_q_present) + bc_s(4, y2ac_delta_q); + + bc_b(uvdc_delta_q_present); + if (current->uvdc_delta_q_present) + bc_s(4, uvdc_delta_q); + + bc_b(uvac_delta_q_present); + if (current->uvac_delta_q_present) + bc_s(4, uvac_delta_q); + + return 0; +} + +static int FUNC(update_token_probs)(CodedBitstreamContext *ctx, + CBSVP8BoolCodingRW *bool_coding_rw, + VP8RawFrameHeader *current) +{ + for (int i = 0; i < 4; ++i) { + for (int j = 0; j < 8; ++j) { + for (int k = 0; k < 3; ++k) { + for (int l = 0; l < 11; ++l) { + bc_b_prob(vp8_token_update_probs[i][j][k][l], + coeff_prob_update[i][j][k][l]); + if (current->coeff_prob_update[i][j][k][l]) + bc_fs(8, coeff_prob[i][j][k][l], 4, i, j, k, l); + } + } + } + } + + return 0; +} + +static int FUNC(update_mv_probs)(CodedBitstreamContext *ctx, + CBSVP8BoolCodingRW *bool_coding_rw, + VP8RawFrameHeader *current) +{ + for (int i = 0; i < 2; ++i) { + for (int j = 0; j < 19; ++j) { + bc_b(mv_prob_update[i][j]); + if (current->mv_prob_update[i][j]) + bc_fs(7, mv_prob[i][j], 2, i, j); + } + } + + return 0; +} + +static int FUNC(frame_tag)(CodedBitstreamContext *ctx, RWContext *rw, + VP8RawFrameHeader *current) +{ + f(1, frame_type); + f(3, profile); + f(1, show_frame); + f(19, first_partition_length_in_bytes); + + if (current->frame_type == VP8_KEY_FRAME) { + fixed(8, start_code_0, VP8_START_CODE_0); + fixed(8, start_code_1, VP8_START_CODE_1); + fixed(8, start_code_2, VP8_START_CODE_2); + + f(14, width); + f(2, horizontal_scale); + f(14, height); + f(2, vertical_scale); + } + + return 0; +} + +static int FUNC(frame_header)(CodedBitstreamContext *ctx, + CBSVP8BoolCodingRW *bool_coding_rw, + VP8RawFrameHeader *current) +{ + if (current->frame_type == VP8_KEY_FRAME) { + bc_f(1, color_space); + bc_f(1, clamping_type); + } + + bc_f(1, segmentation_enable); + if (current->segmentation_enable) + CHECK(FUNC(update_segmentation)(ctx, bool_coding_rw, current)); + + bc_f(1, loop_filter_type); + bc_f(6, loop_filter_level); + bc_f(3, loop_filter_sharpness); + + CHECK(FUNC(mode_ref_lf_deltas)(ctx, bool_coding_rw, current)); + + bc_f(2, log2_token_partitions); + + CHECK(FUNC(quantization_params)(ctx, bool_coding_rw, current)); + + if (current->frame_type != VP8_KEY_FRAME) { + bc_f(1, refresh_golden_frame); + bc_f(1, refresh_alternate_frame); + if (!current->refresh_golden_frame) + bc_f(2, copy_buffer_to_golden); + if (!current->refresh_alternate_frame) + bc_f(2, copy_buffer_to_alternate); + bc_f(1, ref_frame_sign_bias_golden); + bc_f(1, ref_frame_sign_bias_alternate); + } + bc_f(1, refresh_entropy_probs); + if (current->frame_type != VP8_KEY_FRAME) + bc_f(1, refresh_last_frame); + + CHECK(FUNC(update_token_probs)(ctx, bool_coding_rw, current)); + + bc_f(1, mb_no_skip_coeff); + if (current->mb_no_skip_coeff) + bc_f(8, prob_skip_false); + + if (current->frame_type != VP8_KEY_FRAME) { + bc_f(8, prob_intra); + bc_f(8, prob_last); + bc_f(8, prob_golden); + + // intra_16x16_prob + bc_b(intra_16x16_prob_update); + if (current->intra_16x16_prob_update) + for (int i = 0; i < 4; i++) + bc_fs(8, intra_16x16_prob[i], 1, i); + + // intra_chroma_prob + bc_b(intra_chrome_prob_update); + if (current->intra_chrome_prob_update) + for (int i = 0; i < 3; i++) + bc_fs(8, intra_chrome_prob[i], 1, i); + + CHECK(FUNC(update_mv_probs)(ctx, bool_coding_rw, current)); + } + + return 0; +} + +static int FUNC(uncompressed_header)(CodedBitstreamContext *ctx, RWContext *rw, + VP8RawFrame *current) +{ + HEADER("Frame"); + + CHECK(FUNC(frame_tag)(ctx, rw, ¤t->header)); + + return 0; +} + +static int FUNC(compressed_header)(CodedBitstreamContext *ctx, + CBSVP8BoolCodingRW *bool_coding_rw, + VP8RawFrame *current) +{ + CHECK(FUNC(frame_header)(ctx, bool_coding_rw, ¤t->header)); + + return 0; +}