From 3af64c8a66ae2ecdc310fe91a2e1c56cc78242b4 Mon Sep 17 00:00:00 2001
From: gfxVPLsdm <gfxvplsdm@intel.com>
Date: Fri, 23 Dec 2022 07:06:22 +0800
Subject: [PATCH] Use EncTool + Noisy Content Check to enable ACQM + dblk
 (#4244)

Co-authored-by: Chandler Brown <david.c.brown@intel.com>
---
 .../agnostic/base/hevcehw_base_enctools.cpp    | 16 +++++++---------
 .../hevc/agnostic/base/hevcehw_base_enctools.h | 18 ++++++++++++++++++
 .../base/hevcehw_base_legacy_defaults.cpp      | 17 +++++++++++++++++
 3 files changed, 42 insertions(+), 9 deletions(-)

diff --git a/_studio/mfx_lib/encode_hw/hevc/agnostic/base/hevcehw_base_enctools.cpp b/_studio/mfx_lib/encode_hw/hevc/agnostic/base/hevcehw_base_enctools.cpp
index 254bc867e8..4019ab51db 100644
--- a/_studio/mfx_lib/encode_hw/hevc/agnostic/base/hevcehw_base_enctools.cpp
+++ b/_studio/mfx_lib/encode_hw/hevc/agnostic/base/hevcehw_base_enctools.cpp
@@ -139,15 +139,6 @@ inline bool IsBRC(const mfxExtEncToolsConfig &config)
         IsOn(config.BRC);
 }
 
-
-inline bool IsEncToolsOptSet(const mfxExtEncToolsConfig& config)
-{
-    return
-        (config.AdaptiveB | config.AdaptiveI | config.AdaptiveLTR | config.AdaptivePyramidQuantB
-            | config.AdaptivePyramidQuantP | config.AdaptiveQuantMatrices | config.AdaptiveRefB
-            | config.AdaptiveRefP | config.BRC | config.BRCBufferHints | config.SceneChange);
-}
-
 bool HEVCEHW::Base::IsEncToolsOptOn(const mfxExtEncToolsConfig &config, bool bGameStreaming)
 {
     return
@@ -312,7 +303,14 @@ static void SetDefaultConfig(const mfxVideoParam &video, mfxExtEncToolsConfig &c
         bool lplaAssistedBRC = IsOn(config.BRC) && isSWLACondition(video);
         SetDefaultOpt(config.BRCBufferHints, lplaAssistedBRC);
         SetDefaultOpt(config.AdaptiveMBQP,  bMBQPSupport && lplaAssistedBRC && IsOn(pExtOpt2->MBBRC));
+#ifdef ONEVPL_EXPERIMENTAL
+        if (IsEncToolsOptSet(config) && pExtOpt3->ContentInfo == MFX_CONTENT_NOISY_VIDEO)
+            SetDefaultOpt(config.AdaptiveQuantMatrices, true);
+        else
+            SetDefaultOpt(config.AdaptiveQuantMatrices, false);
+#else
         SetDefaultOpt(config.AdaptiveQuantMatrices, false);
+#endif
     }
 #ifdef MFX_ENABLE_ENCTOOLS_LPLA
     else
diff --git a/_studio/mfx_lib/encode_hw/hevc/agnostic/base/hevcehw_base_enctools.h b/_studio/mfx_lib/encode_hw/hevc/agnostic/base/hevcehw_base_enctools.h
index d0d3bbe2b7..b7cfb23392 100644
--- a/_studio/mfx_lib/encode_hw/hevc/agnostic/base/hevcehw_base_enctools.h
+++ b/_studio/mfx_lib/encode_hw/hevc/agnostic/base/hevcehw_base_enctools.h
@@ -100,6 +100,24 @@ namespace Base
     bool IsEncToolsOptOn(const mfxExtEncToolsConfig &config, bool bGameStreaming);
     bool IsLPLAEncToolsOn(const mfxExtEncToolsConfig &config, bool bGameStreaming);
 
+    inline bool IsEncToolsOptSet(const mfxExtEncToolsConfig& config)
+    {
+        return
+            (config.AdaptiveB | config.AdaptiveI | config.AdaptiveLTR | config.AdaptivePyramidQuantB
+                | config.AdaptivePyramidQuantP | config.AdaptiveQuantMatrices | config.AdaptiveRefB
+                | config.AdaptiveRefP | config.BRC | config.BRCBufferHints | config.SceneChange);
+    }
+
+    inline int EncToolsDeblockingBetaOffset()
+    {
+        return 4;  // Currently hard coded to best tested value when using enctools
+    }
+
+    inline int EncToolsDeblockingAlphaTcOffset()
+    {
+        return 2;  // Currently hard coded to best tested value when using enctools
+    }
+
 } //Base
 } //namespace HEVCEHW
 
diff --git a/_studio/mfx_lib/encode_hw/hevc/agnostic/base/hevcehw_base_legacy_defaults.cpp b/_studio/mfx_lib/encode_hw/hevc/agnostic/base/hevcehw_base_legacy_defaults.cpp
index 47fb7b58cf..d5dae0afcf 100644
--- a/_studio/mfx_lib/encode_hw/hevc/agnostic/base/hevcehw_base_legacy_defaults.cpp
+++ b/_studio/mfx_lib/encode_hw/hevc/agnostic/base/hevcehw_base_legacy_defaults.cpp
@@ -27,6 +27,10 @@
 #include <numeric>
 #include <set>
 
+#ifdef MFX_ENABLE_ENCTOOLS
+#include "hevcehw_base_enctools.h"
+#endif
+
 using namespace HEVCEHW;
 using namespace HEVCEHW::Base;
 
@@ -2020,6 +2024,19 @@ class GetDefault
         pps.deblocking_filter_control_present_flag  = 1;
         pps.deblocking_filter_disabled_flag         = !!CO2.DisableDeblockingIdc;
         pps.deblocking_filter_override_enabled_flag = 1; // to disable deblocking per frame
+
+#if defined(MFX_ENABLE_ENCTOOLS)
+#if defined(ONEVPL_EXPERIMENTAL)
+        const mfxExtEncToolsConfig* pConfig = ExtBuffer::Get(par);
+        bool bEncTools = IsEncToolsOptSet(*pConfig);
+
+        if (bEncTools && CO3.ScenarioInfo != MFX_SCENARIO_GAME_STREAMING && CO3.ContentInfo == MFX_CONTENT_NOISY_VIDEO) {
+            pps.beta_offset_div2                        = mfxI8(EncToolsDeblockingBetaOffset() * 0.5 * !pps.deblocking_filter_disabled_flag);
+            pps.tc_offset_div2                          = mfxI8(EncToolsDeblockingAlphaTcOffset() * 0.5 * !pps.deblocking_filter_disabled_flag);
+        }
+#endif
+#endif
+
         pps.scaling_list_data_present_flag              = 0;
         pps.lists_modification_present_flag             = 1;
         pps.log2_parallel_merge_level_minus2            = 0;