From mboxrd@z Thu Jan 1 00:00:00 1970 Return-Path: Received: from ffbox0-bg.mplayerhq.hu (ffbox0-bg.ffmpeg.org [79.124.17.100]) by master.gitmailbox.com (Postfix) with ESMTP id 1FD434990A for ; Fri, 23 Feb 2024 07:13:27 +0000 (UTC) Received: from [127.0.1.1] (localhost [127.0.0.1]) by ffbox0-bg.mplayerhq.hu (Postfix) with ESMTP id B289568C5DA; Fri, 23 Feb 2024 09:13:24 +0200 (EET) Received: from mgamail.intel.com (mgamail.intel.com [192.198.163.8]) by ffbox0-bg.mplayerhq.hu (Postfix) with ESMTPS id E996D680BE7 for ; Fri, 23 Feb 2024 09:13:17 +0200 (EET) DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/simple; d=intel.com; i=@intel.com; q=dns/txt; s=Intel; t=1708672403; x=1740208403; h=from:to:cc:subject:date:message-id:mime-version: content-transfer-encoding; bh=PXjrc0HkoPsPU2TBvlvjEuufPLhwkkOaz/SQ7oESm1Q=; b=GmgYQPAs+HA8tcjLv2xXKjiA/X+bVW/04Bvn7DYmh/NcXGixH8fQxqgG /2y/OvAziW6EdmPV1WqXPTzkLkuguYlwI2KQEL/1IjDx93JH1bhF/t3FT vOgu9nWShdzYVoO+XC3XVjF6+1EZePKiJ8onKKUrecN0Ge8FCi8xYMOkZ Auf/pxliD64lThk1Dk/NS60QAjjgAaxj1roHWhBXLkHpSUEGw05Hwy57k hgtXcU3vWCZ9i3nzwtQEl/JvORYToOC1dWkEkcYE8Lsi4I2bUnLKcCtJn dDOGxn9fovZ1RhkU5XA069PR6kRakE5ossRfBgTz6gmEVKJCzayN2ptu5 A==; X-IronPort-AV: E=McAfee;i="6600,9927,10992"; a="20510852" X-IronPort-AV: E=Sophos;i="6.06,179,1705392000"; d="scan'208";a="20510852" Received: from orviesa002.jf.intel.com ([10.64.159.142]) by fmvoesa102.fm.intel.com with ESMTP/TLS/ECDHE-RSA-AES256-GCM-SHA384; 22 Feb 2024 23:13:15 -0800 X-ExtLoop1: 1 X-IronPort-AV: E=Sophos;i="6.06,179,1705392000"; d="scan'208";a="36603323" Received: from t-dg2.sh.intel.com ([10.239.159.57]) by orviesa002.jf.intel.com with ESMTP; 22 Feb 2024 23:13:14 -0800 From: fei.w.wang-at-intel.com@ffmpeg.org To: ffmpeg-devel@ffmpeg.org Date: Fri, 23 Feb 2024 15:14:42 +0800 Message-Id: <20240223071442.170041-1-fei.w.wang@intel.com> X-Mailer: git-send-email 2.25.1 MIME-Version: 1.0 Subject: [FFmpeg-devel] [PATCH v2] lavc/vaapi_encode: Enable Macroblock based bitrate control 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.w.wang@intel.com Content-Type: text/plain; charset="us-ascii" Content-Transfer-Encoding: 7bit Errors-To: ffmpeg-devel-bounces@ffmpeg.org Sender: "ffmpeg-devel" Archived-At: List-Archive: List-Post: From: Fei Wang Signed-off-by: Fei Wang --- update: Print mbbrc status with "ON/OFF" instead of "0/1". doc/encoders.texi | 3 +++ libavcodec/vaapi_encode.c | 11 ++++++++++- libavcodec/vaapi_encode.h | 9 ++++++++- 3 files changed, 21 insertions(+), 2 deletions(-) diff --git a/doc/encoders.texi b/doc/encoders.texi index 9f477d7c53..3f11a80039 100644 --- a/doc/encoders.texi +++ b/doc/encoders.texi @@ -4086,6 +4086,9 @@ Quality-defined variable-bitrate. Average variable bitrate. @end table +@item mbbrc +Enable Macroblock based rate control. Invalid for CQP mode. + @end table Each encoder also has its own specific options: diff --git a/libavcodec/vaapi_encode.c b/libavcodec/vaapi_encode.c index 86f4110cd2..e4bf15fea0 100644 --- a/libavcodec/vaapi_encode.c +++ b/libavcodec/vaapi_encode.c @@ -1806,6 +1806,9 @@ static av_cold int vaapi_encode_init_rate_control(AVCodecContext *avctx) int i, first = 1, res; supported_va_rc_modes = rc_attr.value; + if (ctx->mbbrc && !(supported_va_rc_modes & VA_RC_MB)) + ctx->mbbrc = 0; + for (i = 0; i < FF_ARRAY_ELEMS(vaapi_encode_rc_modes); i++) { rc_mode = &vaapi_encode_rc_modes[i]; if (supported_va_rc_modes & rc_mode->va_mode) { @@ -2017,13 +2020,18 @@ rc_mode_found: ctx->va_bit_rate = rc_bits_per_second; av_log(avctx, AV_LOG_VERBOSE, "RC mode: %s.\n", rc_mode->name); + + if (ctx->mbbrc && ctx->va_rc_mode == VA_RC_CQP) + ctx->mbbrc = 0; + av_log(avctx, AV_LOG_VERBOSE, "MB bitrate control: %s.\n", ctx->mbbrc ? "ON" : "OFF"); + if (rc_attr.value == VA_ATTRIB_NOT_SUPPORTED) { // This driver does not want the RC mode attribute to be set. } else { ctx->config_attributes[ctx->nb_config_attributes++] = (VAConfigAttrib) { .type = VAConfigAttribRateControl, - .value = ctx->va_rc_mode, + .value = ctx->mbbrc ? ctx->va_rc_mode | VA_RC_MB : ctx->va_rc_mode, }; } @@ -2052,6 +2060,7 @@ rc_mode_found: #if VA_CHECK_VERSION(1, 1, 0) .ICQ_quality_factor = av_clip(rc_quality, 1, 51), .max_qp = (avctx->qmax > 0 ? avctx->qmax : 0), + .rc_flags.bits.mb_rate_control = ctx->mbbrc ? 1 : 2, #endif #if VA_CHECK_VERSION(1, 3, 0) .quality_factor = rc_quality, diff --git a/libavcodec/vaapi_encode.h b/libavcodec/vaapi_encode.h index 6964055b93..9774cc86b1 100644 --- a/libavcodec/vaapi_encode.h +++ b/libavcodec/vaapi_encode.h @@ -216,6 +216,9 @@ typedef struct VAAPIEncodeContext { // available modes). int explicit_rc_mode; + // Macroblock based rate control. + int mbbrc; + // Explicitly-set QP, for use with the "qp" options. // (Forces CQP mode when set, overriding everything else.) int explicit_qp; @@ -538,7 +541,11 @@ int ff_vaapi_encode_close(AVCodecContext *avctx); VAAPI_ENCODE_RC_MODE(VBR, "Variable-bitrate"), \ VAAPI_ENCODE_RC_MODE(ICQ, "Intelligent constant-quality"), \ VAAPI_ENCODE_RC_MODE(QVBR, "Quality-defined variable-bitrate"), \ - VAAPI_ENCODE_RC_MODE(AVBR, "Average variable-bitrate") + VAAPI_ENCODE_RC_MODE(AVBR, "Average variable-bitrate"), \ + { "mbbrc", \ + "Macroblock based bitrate control",\ + OFFSET(common.mbbrc), AV_OPT_TYPE_BOOL, \ + { .i64 = 0 }, 0, 1, FLAGS } #endif /* AVCODEC_VAAPI_ENCODE_H */ -- 2.25.1 _______________________________________________ ffmpeg-devel mailing list ffmpeg-devel@ffmpeg.org https://ffmpeg.org/mailman/listinfo/ffmpeg-devel To unsubscribe, visit link above, or email ffmpeg-devel-request@ffmpeg.org with subject "unsubscribe".