From 20b740df4e8fc22dd727f5d1c9fcb42b24e809f4 Mon Sep 17 00:00:00 2001
From: Nikolas <nikolas@linusmediagroup.com>
Date: Thu, 28 Nov 2024 17:14:04 -0800
Subject: [PATCH] please

---
 handbrake/handbrake.py  | 19 +++++++++++++++----
 handbrake/manifest.yaml |  9 ++++++---
 2 files changed, 21 insertions(+), 7 deletions(-)

diff --git a/handbrake/handbrake.py b/handbrake/handbrake.py
index 6c86428..998a2f7 100644
--- a/handbrake/handbrake.py
+++ b/handbrake/handbrake.py
@@ -17,7 +17,6 @@ from harness_utils.output import (
 )
 
 
-
 SCRIPT_DIR = Path(__file__).resolve().parent
 LOG_DIR = SCRIPT_DIR.joinpath("run")
 LOG_DIR.mkdir(exist_ok=True)
@@ -31,17 +30,29 @@ logging.basicConfig(
 )
 
 ENCODER_TO_PRESET = {
-    "h264": {
+    "h264 (cpu)": {
         "file": f"{SCRIPT_DIR}\\presets\\h264_bigbuckbunny_1080p_cpu_test.json",
         "name": "\"CPU 1080p BBB H264\""
     },
-    "h265": {
+    "h265 (cpu)": {
         "file": f"{SCRIPT_DIR}\\presets\\h265_bigbuckbunny_1080p_cpu_test.json",
         "name": "\"CPU 1080p BBB H265\""
     },
-    "av1": {
+    "av1 (cpu)": {
         "file": f"{SCRIPT_DIR}\\presets\\av1-svt_bigbuckbunny_1080p_cpu_test.json",
         "name": "\"CPU 1080p BBB AV1\""
+    },
+    "h264 (nvenc)": {
+        "file": f"{SCRIPT_DIR}\\presets\\h264_nvenc_bigbuckbunny_1080p_gpu_test.json",
+        "name": "\"NVENC 1080p BBB H264\""
+    },
+    "h265 (nvenc)": {
+        "file": f"{SCRIPT_DIR}\\presets\\h265_nvenc_bigbuckbunny_1080p_gpu_test.json",
+        "name": "\"NVENC 1080p BBB H265\""
+    },
+    "av1 (nvenc)": {
+        "file": f"{SCRIPT_DIR}\\presets\\av1-nvenc_bigbuckbunny_1080p_gpu_test.json",
+        "name": "\"NVENC 1080p BBB AV1\""
     }
 }
 
diff --git a/handbrake/manifest.yaml b/handbrake/manifest.yaml
index 50207ad..d9801f9 100644
--- a/handbrake/manifest.yaml
+++ b/handbrake/manifest.yaml
@@ -7,7 +7,10 @@ options:
   - name: encoder
     type: select
     values:
-      - h264
-      - h265
-      - av1
+      - h264 (cpu)
+      - h265 (cpu)
+      - av1 (cpu)
+      - h264 (nvenc)
+      - h265 (nvenc)
+      - av1 (nvenc)
     tooltip: Select which preset to use