GPUの自動選択時にnvvfx使用の有無を考慮。

rigaya · Oct 7, 2023 · d327bf7 · d327bf7
1 parent a7f7bfa
commit d327bf7
Show file tree

Hide file tree

Showing 3 changed files with 32 additions and 0 deletions.
diff --git a/NVEnc/NVEnc_readme.txt b/NVEnc/NVEnc_readme.txt
@@ -214,8 +214,14 @@ NVIDIA グラフィックドライバ 536.23
 
 【メモ】
 2023.10.05 (7.32)
+[NVEncC]
 - エンコードせずにy4m出力するオプションを追加。 (-c raw)
 - 出力色空間を設定するオプションを追加。 (--output-csp)
+- GPUの自動選択時にnvvfx使用の有無を考慮。
+
+[NVEnc.auo]
+- ログ出力に映像と音声の長さを表示するように。
+- 一時ファイルの削除・リネーム失敗に関するエラーメッセージを拡充。
 
 2023.08.28 (7.31)
 - --audio-streamがavs読み込み時に反映されないのを修正。

diff --git a/NVEncCore/NVEncCore.cpp b/NVEncCore/NVEncCore.cpp
@@ -692,6 +692,18 @@ NVENCSTATUS NVEncCore::InitOutput(InEncodeVideoParam *inputParams, NV_ENC_BUFFER
     return NV_ENC_SUCCESS;
 }
 
+bool NVEncCore::useNVVFX(const InEncodeVideoParam *inputParam) {
+    auto& vppnv = inputParam->vppnv;
+    if (   vppnv.nvvfxArtifactReduction.enable
+        || vppnv.nvvfxDenoise.enable
+        || vppnv.nvvfxSuperRes.enable
+        || vppnv.nvvfxUpScaler.enable
+        || inputParam->vpp.resize_algo == RGY_VPP_RESIZE_NVVFX_SUPER_RES) {
+        return true;
+    }
+    return false;
+}
+
 NVENCSTATUS NVEncCore::CheckGPUListByEncoder(std::vector<std::unique_ptr<NVGPUInfo>> &gpuList, const InEncodeVideoParam *inputParam) {
     if (m_nDeviceId >= 0) {
         //手動で設定されている
@@ -802,6 +814,17 @@ NVENCSTATUS NVEncCore::CheckGPUListByEncoder(std::vector<std::unique_ptr<NVGPUIn
                continue;
            }
         }
+        //フィルタのチェック
+        if (useNVVFX(inputParam)) {
+            //nvvfxにはturing以降(CC7.0)が必要
+            const int nvvfxRequiredCCMajor = 7;
+            if ((*gpu)->cc().first < nvvfxRequiredCCMajor) {
+                message += strsprintf(_T("GPU #%d (%s) does not support nvvfx, CC 7.0 is required but GPU is CC %d.%d.\n"), (*gpu)->id(), (*gpu)->name().c_str(), (*gpu)->cc().first, (*gpu)->cc().second);
+                gpu = gpuList.erase(gpu);
+                continue;
+            }
+        }
+
         PrintMes(RGY_LOG_DEBUG, _T("GPU #%d (%s) available for encode.\n"), (*gpu)->id(), (*gpu)->name().c_str());
         gpu++;
     }

diff --git a/NVEncCore/NVEncCore.h b/NVEncCore/NVEncCore.h
@@ -115,6 +115,9 @@ class NVEncCore : public NVEncCtrl {
     //perfMonitorの初期化
     virtual NVENCSTATUS InitPerfMonitor(const InEncodeVideoParam *inputParam);
 
+    //nvvfxを使用するかチェック
+    bool useNVVFX(const InEncodeVideoParam *inputParam);
+
     //GPUListのGPUが必要なエンコードを行えるかチェック
     NVENCSTATUS CheckGPUListByEncoder(std::vector<std::unique_ptr<NVGPUInfo>> &gpuList, const InEncodeVideoParam *inputParam);