From patchwork Thu Jun 17 06:10:36 2021 Content-Type: text/plain; charset="utf-8" MIME-Version: 1.0 Content-Transfer-Encoding: 7bit X-Patchwork-Submitter: Fei Wang X-Patchwork-Id: 28566 Delivered-To: ffmpegpatchwork2@gmail.com Received: by 2002:a5e:c91a:0:0:0:0:0 with SMTP id z26csp330394iol; Wed, 16 Jun 2021 23:13:22 -0700 (PDT) X-Google-Smtp-Source: ABdhPJyqJbSzVihzn4h/EAX8bSdOT1JWNOdQTG2vc168kYxfFfqzTgkzbZKGAgoU2ow6exG4jFtw X-Received: by 2002:a05:6402:34d5:: with SMTP id w21mr4278698edc.257.1623910402319; Wed, 16 Jun 2021 23:13:22 -0700 (PDT) ARC-Seal: i=1; a=rsa-sha256; t=1623910402; cv=none; d=google.com; s=arc-20160816; b=TB03TRa8Cce2zEpkn5CCeeP4c0zKa8eMqkET6PDNniHKWsu23Pe8wVMdBHLcKNqDSt OBfZ4QNn2TPFBRppLthQsBF4Zb133Tnf1cDtm276fuKQd5phQF73uLP8cvR9sni6Olem R4+8rnnIfKi+QfEfXDuLtikCdIB5z5YQ1PLk6jVPf8NS/4Pp5Kv5jilNMvn3uEW40eOF kH7eCFvbxUrDjgJw3nE5uWu+/MJJCRmw6JsKKjmqZXFibIpIivpf3nbwdmiyo/Tm1ecc vsopE7s++wqXl2llA4qQTh46za/TederE7xMWdxFYvtQWVJLrdlagc8KASx2nHdgB9Aj zm7g== ARC-Message-Signature: i=1; a=rsa-sha256; c=relaxed/relaxed; d=google.com; s=arc-20160816; h=sender:errors-to:content-transfer-encoding:mime-version:cc:reply-to :list-subscribe:list-help:list-post:list-archive:list-unsubscribe :list-id:precedence:subject:references:in-reply-to:message-id:date :to:from:ironport-sdr:ironport-sdr:delivered-to; bh=YfVIkzUvXvNk9CIV/t0Htd4miK+F2WonzgeTn6YZSUY=; b=yki9+khSrOhp6CGOQmGd39V21GPXsOpj4bNHqVsyzJigIlo/S9MQzq7oBfFJtLU+6O qq+ymFL+2s5G71ubjxwcSloVcHwDMgeWI3tnTCQv1PUg6hqflyoIhY238v6cvK9d2jlz MKUYXFFwuMm9LgVkPFMqus+sWPykTk1RksmHE0wmy+MtlFEL2xST9aZ+Z8z37gZ3ZCVm XwBmZzlcxaxWUv0x4o5ic3KtuKiSfl8JkRicVgy2wVJXqsCt84E7oUKU8Lg/3W7ECHPA VXJUPxEuEHWQcnG3aIAANP9VmHV5OxoNO74CN0jqHyd5rVTTaefyfiHLDJoTY42ASHXS vI1g== ARC-Authentication-Results: i=1; mx.google.com; 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=NONE dis=NONE) header.from=intel.com Return-Path: Received: from ffbox0-bg.mplayerhq.hu (ffbox0-bg.ffmpeg.org. [79.124.17.100]) by mx.google.com with ESMTP id x7si1209211edd.288.2021.06.16.23.13.21; Wed, 16 Jun 2021 23:13:22 -0700 (PDT) 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; 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=NONE dis=NONE) header.from=intel.com Received: from [127.0.1.1] (localhost [127.0.0.1]) by ffbox0-bg.mplayerhq.hu (Postfix) with ESMTP id 2FAC1688342; Thu, 17 Jun 2021 09:12:27 +0300 (EEST) X-Original-To: ffmpeg-devel@ffmpeg.org Delivered-To: ffmpeg-devel@ffmpeg.org Received: from mga05.intel.com (mga05.intel.com [192.55.52.43]) by ffbox0-bg.mplayerhq.hu (Postfix) with ESMTPS id 14EED68924E for ; Thu, 17 Jun 2021 09:12:22 +0300 (EEST) IronPort-SDR: ADfKQ2t0aXAAXyl7E2WOejMTvCf42x9aiSyGZA/iN+/n0SdJPxlPqkeKz/cVIy4LTMb4ycMKka BRggHB9DMQjA== X-IronPort-AV: E=McAfee;i="6200,9189,10017"; a="291942570" X-IronPort-AV: E=Sophos;i="5.83,278,1616482800"; d="scan'208";a="291942570" Received: from orsmga001.jf.intel.com ([10.7.209.18]) by fmsmga105.fm.intel.com with ESMTP/TLS/ECDHE-RSA-AES256-GCM-SHA384; 16 Jun 2021 23:12:14 -0700 IronPort-SDR: S6SC6JQ1AYrkXcE1w3BQTIsVAiGeZdmtuf9o50vYQhDIQxZPReT6fDNgwesPv7l66sGnIjWIad Lg0dssyXiZMA== X-ExtLoop1: 1 X-IronPort-AV: E=Sophos;i="5.83,278,1616482800"; d="scan'208";a="485177042" Received: from f.sh.intel.com ([10.239.159.147]) by orsmga001.jf.intel.com with ESMTP; 16 Jun 2021 23:12:14 -0700 From: Fei Wang To: ffmpeg-devel@ffmpeg.org Date: Thu, 17 Jun 2021 14:10:36 +0800 Message-Id: <20210617061037.20335-8-fei.w.wang@intel.com> X-Mailer: git-send-email 2.17.1 In-Reply-To: <20210617061037.20335-1-fei.w.wang@intel.com> References: <20210617061037.20335-1-fei.w.wang@intel.com> Subject: [FFmpeg-devel] [PATCH 8/9] avcodec/av1_vaapi: enable segmentation features 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: Fei Wang MIME-Version: 1.0 Errors-To: ffmpeg-devel-bounces@ffmpeg.org Sender: "ffmpeg-devel" X-TUID: peWthmZOKydE Signed-off-by: Fei Wang --- libavcodec/vaapi_av1.c | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/libavcodec/vaapi_av1.c b/libavcodec/vaapi_av1.c index 81b13bb1aa..6aaabed2c1 100644 --- a/libavcodec/vaapi_av1.c +++ b/libavcodec/vaapi_av1.c @@ -63,6 +63,9 @@ static int vaapi_av1_start_frame(AVCodecContext *avctx, int err = 0; int apply_grain = !(avctx->export_side_data & AV_CODEC_EXPORT_DATA_FILM_GRAIN) && film_grain->apply_grain; uint8_t remap_lr_type[4] = {AV1_RESTORE_NONE, AV1_RESTORE_SWITCHABLE, AV1_RESTORE_WIENER, AV1_RESTORE_SGRPROJ}; + uint8_t segmentation_feature_signed[AV1_SEG_LVL_MAX] = {1, 1, 1, 1, 1, 0, 0, 0}; + uint8_t segmentation_feature_max[AV1_SEG_LVL_MAX] = {255, AV1_MAX_LOOP_FILTER, + AV1_MAX_LOOP_FILTER, AV1_MAX_LOOP_FILTER, AV1_MAX_LOOP_FILTER, 7 , 0 , 0 }; pic->output_surface = vaapi_av1_surface_id(&s->cur_frame); @@ -218,6 +221,17 @@ static int vaapi_av1_start_frame(AVCodecContext *avctx, for (int j = 0; j < 6; j++) pic_param.wm[i - 1].wmmat[j] = s->cur_frame.gm_params[i][j]; } + for (int i = 0; i < AV1_MAX_SEGMENTS; i++) { + for (int j = 0; j < AV1_SEG_LVL_MAX; j++) { + pic_param.seg_info.feature_mask[i] |= (frame_header->feature_enabled[i][j] << j); + if (segmentation_feature_signed[j]) + pic_param.seg_info.feature_data[i][j] = av_clip(frame_header->feature_value[i][j], + -segmentation_feature_max[j], segmentation_feature_max[j]); + else + pic_param.seg_info.feature_data[i][j] = av_clip(frame_header->feature_value[i][j], + 0, segmentation_feature_max[j]); + } + } if (apply_grain) { for (int i = 0; i < film_grain->num_y_points; i++) { pic_param.film_grain_info.point_y_value[i] =