From b41f1f7e6c25c5ae884175cb99864f3bb99fcb80 Mon Sep 17 00:00:00 2001
From: K4YT3X <k4yt3x@k4yt3x.com>
Date: Thu, 7 Jan 2021 00:34:18 -0500
Subject: [PATCH] corrected ncnn Vulkan drivers' names

---
 src/video2x_gui.py                  |  8 +++++---
 src/video2x_gui.ui                  | 21 ++++++++++++++++++---
 src/wrappers/realsr_ncnn_vulkan.py  |  6 +++---
 src/wrappers/srmd_ncnn_vulkan.py    |  4 ++--
 src/wrappers/waifu2x_ncnn_vulkan.py |  2 +-
 5 files changed, 29 insertions(+), 12 deletions(-)

diff --git a/src/video2x_gui.py b/src/video2x_gui.py
index d97fa7c..b998070 100755
--- a/src/video2x_gui.py
+++ b/src/video2x_gui.py
@@ -45,10 +45,12 @@ Contact: [k4yt3x@k4yt3x.com](mailto:k4yt3x@k4yt3x.com)"""
 AVAILABLE_DRIVERS = {
     "Waifu2X Caffe": "waifu2x_caffe",
     "Waifu2X Converter CPP": "waifu2x_converter_cpp",
-    "Waifu2X NCNN Vulkan": "waifu2x_ncnn_vulkan",
-    "SRMD NCNN Vulkan": "srmd_ncnn_vulkan",
-    "RealSR NCNN Vulkan": "realsr_ncnn_vulkan",
+    "Waifu2X ncnn Vulkan": "waifu2x_ncnn_vulkan",
+    "SRMD ncnn Vulkan": "srmd_ncnn_vulkan",
+    "RealSR ncnn Vulkan": "realsr_ncnn_vulkan",
     "Anime4KCPP": "anime4kcpp",
+    "DAIN ncnn Vulkan": "dain_ncnn_vulkan",
+    "RIFE ncnn Vulkan": "rife_ncnn_vulkan",
 }
 
 # get current working directory before it is changed by drivers
diff --git a/src/video2x_gui.ui b/src/video2x_gui.ui
index 1828586..ce67e09 100644
--- a/src/video2x_gui.ui
+++ b/src/video2x_gui.ui
@@ -378,7 +378,7 @@
                      </property>
                      <item>
                       <property name="text">
-                       <string>Waifu2X NCNN Vulkan</string>
+                       <string>Waifu2X ncnn Vulkan</string>
                       </property>
                      </item>
                      <item>
@@ -393,12 +393,12 @@
                      </item>
                      <item>
                       <property name="text">
-                       <string>SRMD NCNN Vulkan</string>
+                       <string>SRMD ncnn Vulkan</string>
                       </property>
                      </item>
                      <item>
                       <property name="text">
-                       <string>RealSR NCNN Vulkan</string>
+                       <string>RealSR ncnn Vulkan</string>
                       </property>
                      </item>
                      <item>
@@ -406,6 +406,21 @@
                        <string>Anime4KCPP</string>
                       </property>
                      </item>
+                     <item>
+                      <property name="text">
+                       <string>DAIN ncnn Vulkan</string>
+                      </property>
+                     </item>
+                     <item>
+                      <property name="text">
+                       <string>CAIN ncnn Vulkan</string>
+                      </property>
+                     </item>
+                     <item>
+                      <property name="text">
+                       <string>RIFE ncnn Vulkan</string>
+                      </property>
+                     </item>
                     </widget>
                    </item>
                   </layout>
diff --git a/src/wrappers/realsr_ncnn_vulkan.py b/src/wrappers/realsr_ncnn_vulkan.py
index 6d0e174..d75b9be 100755
--- a/src/wrappers/realsr_ncnn_vulkan.py
+++ b/src/wrappers/realsr_ncnn_vulkan.py
@@ -1,7 +1,7 @@
 #!/usr/bin/env python3
 # -*- coding: utf-8 -*-
 """
-Name: RealSR NCNN Vulkan Driver
+Name: RealSR ncnn Vulkan Driver
 Creator: K4YT3X
 Date Created: May 26, 2020
 Last Modified: September 21, 2020
@@ -24,7 +24,7 @@ from avalon_framework import Avalon
 
 
 class WrapperMain:
-    """This class communicates with RealSR NCNN Vulkan engine
+    """This class communicates with RealSR ncnn Vulkan engine
 
     An object will be created for this class, containing information
     about the binary address and the processing method. When being called
@@ -66,7 +66,7 @@ class WrapperMain:
         self.driver_settings["s"] = int(scale_ratio)
 
     def upscale(self, input_directory, output_directory):
-        """This is the core function for RealSR NCNN Vulkan class
+        """This is the core function for RealSR ncnn Vulkan class
 
         Arguments:
             input_directory {string} -- source directory path
diff --git a/src/wrappers/srmd_ncnn_vulkan.py b/src/wrappers/srmd_ncnn_vulkan.py
index 5d4c152..db6dfc2 100755
--- a/src/wrappers/srmd_ncnn_vulkan.py
+++ b/src/wrappers/srmd_ncnn_vulkan.py
@@ -1,7 +1,7 @@
 #!/usr/bin/env python3
 # -*- coding: utf-8 -*-
 """
-Name: SRMD NCNN Vulkan Driver
+Name: SRMD ncnn Vulkan Driver
 Creator: K4YT3X
 Date Created: April 26, 2020
 Last Modified: September 21, 2020
@@ -24,7 +24,7 @@ from avalon_framework import Avalon
 
 
 class WrapperMain:
-    """This class communicates with SRMD NCNN Vulkan engine
+    """This class communicates with SRMD ncnn Vulkan engine
 
     An object will be created for this class, containing information
     about the binary address and the processing method. When being called
diff --git a/src/wrappers/waifu2x_ncnn_vulkan.py b/src/wrappers/waifu2x_ncnn_vulkan.py
index 9ad9834..bf087e1 100755
--- a/src/wrappers/waifu2x_ncnn_vulkan.py
+++ b/src/wrappers/waifu2x_ncnn_vulkan.py
@@ -1,7 +1,7 @@
 #!/usr/bin/env python3
 # -*- coding: utf-8 -*-
 """
-Name: Waifu2x NCNN Vulkan Driver
+Name: Waifu2x ncnn Vulkan Driver
 Creator: SAT3LL
 Date Created: June 26, 2019
 Last Modified: May 11, 2020