From 1ee76c022424e3ade6fb93396be25e549f058e31 Mon Sep 17 00:00:00 2001
From: k4yt3x <k4yt3x@k4yt3x.com>
Date: Tue, 25 Jun 2019 23:34:57 -0400
Subject: [PATCH] removed redundant code block

---
 bin/upscaler.py | 110 +++++++++++++-----------------------------------
 1 file changed, 30 insertions(+), 80 deletions(-)

diff --git a/bin/upscaler.py b/bin/upscaler.py
index fe72569..c3b603d 100644
--- a/bin/upscaler.py
+++ b/bin/upscaler.py
@@ -170,7 +170,7 @@ class Upscaler:
             self.progress_bar_exit_signal = True
             progress_bar.join()
             return
-        elif self.waifu2x_driver == 'waifu2x_caffe':
+        else:
             # create a container for all upscaler threads
             upscaler_threads = []
 
@@ -211,13 +211,37 @@ class Upscaler:
             for thread_info in thread_pool:
 
                 # create a separate w2 instance for each thread
-                w2 = Waifu2xCaffe(copy.deepcopy(self.waifu2x_settings), self.method, self.model_dir)
+                if self.waifu2x_driver == 'waifu2x_caffe':
+                    w2 = Waifu2xCaffe(copy.deepcopy(self.waifu2x_settings), self.method, self.model_dir)
+                    if self.scale_ratio:
+                        thread = threading.Thread(target=w2.upscale,
+                                                  args=(thread_info[0],
+                                                  self.upscaled_frames,
+                                                  self.scale_ratio,
+                                                  False,
+                                                  False,
+                                                  self.image_format,
+                                                  self.upscaler_exceptions))
+                    else:
+                        thread = threading.Thread(target=w2.upscale, 
+                                                  args=(thread_info[0], 
+                                                  self.upscaled_frames, 
+                                                  False, 
+                                                  self.scale_width, 
+                                                  self.scale_height, 
+                                                  self.image_format, 
+                                                  self.upscaler_exceptions))
+
+                # if the driver being used is waifu2x_ncnn_vulkan 
+                elif self.waifu2x_driver == 'waifu2x_ncnn_vulkan':
+                    w2 = Waifu2xNcnnVulkan(copy.deepcopy(self.waifu2x_settings))
+                    thread = threading.Thread(target=w2.upscale,
+                                              args=(thread_info[0],
+                                                    self.upscaled_frames,
+                                                    self.scale_ratio,
+                                                    self.upscaler_exceptions))
 
                 # create thread
-                if self.scale_ratio:
-                    thread = threading.Thread(target=w2.upscale, args=(thread_info[0], self.upscaled_frames, self.scale_ratio, False, False, self.image_format, self.upscaler_exceptions))
-                else:
-                    thread = threading.Thread(target=w2.upscale, args=(thread_info[0], self.upscaled_frames, False, self.scale_width, self.scale_height, self.image_format, self.upscaler_exceptions))
                 thread.name = thread_info[1]
 
                 # add threads into the pool
@@ -248,81 +272,7 @@ class Upscaler:
 
             if len(self.upscaler_exceptions) != 0:
                 raise(self.upscaler_exceptions[0])
-        elif self.waifu2x_driver == 'waifu2x_ncnn_vulkan':
-            # create a container for all upscaler threads
-            upscaler_threads = []
 
-            # list all images in the extracted frames
-            frames = [os.path.join(self.extracted_frames, f) for f in os.listdir(self.extracted_frames) if os.path.isfile(os.path.join(self.extracted_frames, f))]
-
-            # if we have less images than threads,
-            # create only the threads necessary
-            if len(frames) < self.threads:
-                self.threads = len(frames)
-
-            # create a directory for each thread and append directory
-            # name into a list
-
-            thread_pool = []
-            thread_directories = []
-            for thread_id in range(self.threads):
-                thread_directory = os.path.join(self.extracted_frames, str(thread_id))
-                thread_directories.append(thread_directory)
-
-                # delete old directories and create new directories
-                if os.path.isdir(thread_directory):
-                    shutil.rmtree(thread_directory)
-                os.mkdir(thread_directory)
-
-                # append directory path into list
-                thread_pool.append((thread_directory, thread_id))
-
-            # evenly distribute images into each directory
-            # until there is none left in the directory
-            for image in frames:
-                # move image
-                shutil.move(image, thread_pool[0][0])
-                # rotate list
-                thread_pool = thread_pool[-1:] + thread_pool[:-1]
-
-            # create threads and start them
-            for thread_info in thread_pool:
-
-                # create a separate w2 instance for each thread
-                w2 = Waifu2xNcnnVulkan(copy.deepcopy(self.waifu2x_settings))
-
-                # create thread
-                thread = threading.Thread(target=w2.upscale, args=(thread_info[0], self.upscaled_frames, self.scale_ratio, self.upscaler_exceptions))
-                thread.name = thread_info[1]
-
-                # add threads into the pool
-                upscaler_threads.append(thread)
-
-            # start progress bar in a different thread
-            progress_bar = threading.Thread(target=self._progress_bar, args=(thread_directories,))
-            progress_bar.start()
-
-            # create the clearer and start it
-            Avalon.debug_info('Starting upscaled image cleaner')
-            image_cleaner = ImageCleaner(self.extracted_frames, self.upscaled_frames, len(upscaler_threads))
-            image_cleaner.start()
-
-            # start all threads
-            for thread in upscaler_threads:
-                thread.start()
-
-            # wait for threads to finish
-            for thread in upscaler_threads:
-                thread.join()
-
-            # upscaling done, kill the clearer
-            Avalon.debug_info('Killing upscaled image cleaner')
-            image_cleaner.stop()
-
-            self.progress_bar_exit_signal = True
-
-            if len(self.upscaler_exceptions) != 0:
-                raise(self.upscaler_exceptions[0])
 
     def run(self):
         """Main controller for Video2X