sharpening fixed

bin/handler.py

@@ -32,7 +32,7 @@ class Handler:
         self.tmppath = ""
         self.videometa = {}
 
-    def handler(self, fsrpath, filepath, quality_mode, quality_setting, output_path, sharpening, scaling, threads=4 ):
+    def handler(self, fsrpath, filepath, quality_mode, quality_setting, output_path, sharpening, scaling, filetype, threads=4 ):
         # Function to be called when using this class as this function automatically determines if file is video or image
         print( '\n\nFSRImageVideoUpscalerFrontend - V1.1.0\n\nCopyright 2023 FSRImageVideoUpscalerFrontend contributors\n\n\n\n' );
 
@@ -61,7 +61,7 @@ class Handler:
         # Determining filetype
         if str(filepath)[len(filepath) - 4:] == ".mp4" or str(filepath)[len(filepath) - 4:] == ".mkv" or str(filepath)[len(filepath) - 4:] == ".MP4":
             print( '\n\n==> Upscaling video' )
-            self.video_scaling(fsrpath, filepath, quality_mode, quality_setting, output_path, threads, sharpening, scaling)
+            self.video_scaling( fsrpath, filepath, quality_mode, quality_setting, output_path, threads, sharpening, scaling, filetype )
         elif str(filepath)[len(filepath) - 4:] == ".JPG" or str(filepath)[len(filepath) - 4:] == ".png" or str(filepath)[len(filepath) - 4:] == ".jpg" or str(filepath)[len(filepath) - 5:] == ".jpeg":
             print( '\n==>upscaling image' )
             self.photo_scaling(fsrpath, filepath, quality_mode, quality_setting, output_path)
@@ -91,7 +91,7 @@ class Handler:
         os.system(self.command)
         print( '\n\n==>Photo upscaled' );
 
-    def video_scaling( self, fsrpath, filepath, quality_mode, quality_setting, output_path, threads, sharpening, scaling ):
+    def video_scaling( self, fsrpath, filepath, quality_mode, quality_setting, output_path, threads, sharpening, scaling, filetype ):
         # DO NOT CALL THIS! Use Handler().handler() instead!
         
         # Splitting video into frames
@@ -107,9 +107,9 @@ class Handler:
         print( '\n==> Created directory' )
                 
         if self.os_type == "linux":
-            self.command = f"ffmpeg -i {str(self.filepath)} {self.tmppath}ex%08d.bmp"
+            self.command = f"ffmpeg -i {str(self.filepath)} {self.tmppath}ex%08d.{ filetype }"
         elif self.os_type == "win32":
-            self.command = f"ffmpeg -i {str(self.filepath)} \"{self.tmppath}ex%08d.bmp\""
+            self.command = f"ffmpeg -i {str(self.filepath)} \"{self.tmppath}ex%08d.{ filetype }\""
         else:
             print("OS CURRENTLY UNSUPPORTED!")
             return False
@@ -126,9 +126,9 @@ class Handler:
         for self.file in self.filelist:
             self.number += 1
             if ( self.os_type == 'win32' ):
-                self.file_list.append( f"{self.tmppath}{self.file} {self.tmppath}sc\\ig{str(self.number).zfill(8)}.bmp " );
+                self.file_list.append( f"{self.tmppath}{self.file} {self.tmppath}us\\ig{str(self.number).zfill(8)}.{ filetype } " );
             else:
-                self.file_list.append( f"{self.tmppath}{self.file} {self.tmppath}sc/ig{str(self.number).zfill(8)}.bmp " );
+                self.file_list.append( f"{self.tmppath}{self.file} {self.tmppath}us/ig{str(self.number).zfill(8)}.{ filetype } " );
         
         if ( self.os_type == 'win32' ):
             self.maxlength = 8000
@@ -157,7 +157,7 @@ class Handler:
         time.sleep( 2 );
 
         try:
-            os.mkdir(f"{self.tmppath}sc")
+            os.mkdir( f'{self.tmppath}us' )
         except FileExistsError:
             pass
 
@@ -166,31 +166,82 @@ class Handler:
         # Thread optimisation: Divide workload up into different threads & upscale using helper function
         #
         ############################################
-
         self.threads = threads
         if ( threads > multiprocessing.cpu_count() ):
             self.threads = multiprocessing.cpu_count();
 
-        print( f'\n\n==>Using { self.threads } threads <==\n\n' );
+        if ( not scaling ):
+            print( f'\n\n==> Upscaling using { self.threads } threads <==\n\n' );
 
-        time.sleep( 2 );
+            time.sleep( 2 );
 
-        self.command_list = [];
-        self.file_list_length = len( self.file_list );
-        for i in range( self.threads ):
-            self.files = '';
-            for _ in range( int( self.file_list_length // self.threads ) ):
-                self.files += self.file_list.pop( 0 );
+            self.command_list = [];
+            self.file_list_length = len( self.file_list );
+            for i in range( self.threads ):
+                self.files = '';
+                for _ in range( int( self.file_list_length // self.threads ) ):
+                    self.files += self.file_list.pop( 0 );
+                
+                if ( i == self.threads - 1 ):
+                    for element in self.file_list:
+                        self.files += element;
+                self.command_list.append( ( quality_mode, self.files, fsrpath, quality_setting, i, self.maxlength, self.os_type ) )
+
+            self.pool = multiprocessing.Pool( self.threads )
+            self.pool.starmap( upscalerEngine, self.command_list );
+            self.pool.close();
+            self.pool.join();
+
+        if sharpening != '':
+            print( f'\n\n\n==> Sharpening using { self.threads } threads <==\n\n' );
+            time.sleep( 2 );
+
+            self.pathSharpening = self.tmppath
+
+            if ( not scaling ):
+                self.pathSharpening += 'us'
+
+            time.sleep( 2 );
+            try:
+                os.mkdir( f'{self.tmppath}sh' )
+            except FileExistsError:
+                pass
+            # Locate Images and assemble FSR-Command
+            self.file_list = []
+            self.filelist = os.listdir( self.pathSharpening )
+            self.filelist.pop(0)
+            self.filelist.sort()
+            self.number = 0
+            for self.file in self.filelist:
+                self.number += 1
+                if ( self.os_type == 'win32' ):
+                    self.file_list.append( f"{self.pathSharpening}\\{self.file} {self.tmppath}sh\\ig{str(self.number).zfill(8)}.{ filetype } " );
+                else:
+                    self.file_list.append( f"{self.pathSharpening}/{self.file} {self.tmppath}sh/ig{str(self.number).zfill(8)}.{ filetype } " );
             
-            if ( i == self.threads - 1 ):
-                for element in self.file_list:
-                    self.files += element;
-            self.command_list.append( ( quality_mode, self.files, fsrpath, quality_setting, i, self.maxlength, self.os_type, sharpening, scaling ) )
+            if ( self.os_type == 'win32' ):
+                self.maxlength = 8000
+            else:
+                self.maxlength = 31900
+            self.pos = 1
 
-        self.pool = multiprocessing.Pool( self.threads )
-        self.pool.starmap( upscalerEngine, self.command_list );
-        self.pool.close();
-        self.pool.join();
+            # assemble command list
+            self.command_list = [];
+            self.file_list_length = len( self.file_list );
+            for i in range( self.threads ):
+                self.files = '';
+                for _ in range( int( self.file_list_length // self.threads ) ):
+                    self.files += self.file_list.pop( 0 );
+                
+                if ( i == self.threads - 1 ):
+                    for element in self.file_list:
+                        self.files += element;
+                self.command_list.append( ( self.files, fsrpath, i, self.maxlength, self.os_type, sharpening ) )
+
+            self.pool = multiprocessing.Pool( self.threads )
+            self.pool.starmap( sharpeningEngine, self.command_list );
+            self.pool.close();
+            self.pool.join();
 
         
         # get Video's audio
@@ -211,11 +262,19 @@ class Handler:
         os.system( self.command )
 
         # reassemble Video
-        print( '\n\n==>Reassembling Video... with framerate @', self.framerate, '\n\n' )
+
+        self.outputPath = self.tmppath
+        if ( sharpening != '' ):
+            self.outputPath += 'sh'
+        else:
+            self.outputPath += 'us'
+
+
+        print( '\n\n==> Reassembling Video... with framerate @', self.framerate, ' <==\n\n' )
         if self.os_type == 'linux':
-            self.command = f'ffmpeg -framerate {self.framerate} -i {self.tmppath}sc/ig%08d.bmp {output_path} -i {self.tmppath}audio.aac'
+            self.command = f'ffmpeg -framerate {self.framerate} -i {self.outputPath}/ig%08d.{ filetype } { output_path } -i {self.tmppath}audio.aac'
         elif self.os_type == 'win32':
-            self.command = f'ffmpeg -framerate {self.framerate} -i \"{self.tmppath}sc\\ig%08d.bmp\" {output_path} -i {self.tmppath}audio.aac'
+            self.command = f'ffmpeg -framerate {self.framerate} -i \"{self.outputPath}\\ig%08d.{ filetype }\" { output_path } -i {self.tmppath}audio.aac'
         else:
             print( 'OS CURRENTLY UNSUPPORTED!' );
             return False
@@ -223,7 +282,7 @@ class Handler:
 
 
 
-def upscalerEngine (  quality_mode, files, fsrpath, quality_setting, number, maxlength, os_type, sharpening, scaling ):
+def upscalerEngine (  quality_mode, files, fsrpath, quality_setting, number, maxlength, os_type ):
     files = files;
     # Refactoring of commands that are longer than 32K characters
     fileout = [];
@@ -273,36 +332,94 @@ def upscalerEngine (  quality_mode, files, fsrpath, quality_setting, number, max
 
     while len( fileout ) > 0:
         files_handle = fileout.pop(0)
-        if ( not scaling ):
-            if quality_mode == 'default':
-                if os_type == 'linux':
-                    command_us = f'wine {fsrpath} -QualityMode {quality_setting} {files_handle}'
-                elif os_type == 'win32':
-                    command_us = f'FidelityFX_CLI -QualityMode {quality_setting} {files_handle}'
-                else:
-                    print( 'OS CURRENTLY UNSUPPORTED!' )
-                    return False
-            else:
-                if os_type == 'linux':
-                    command_us = f'wine {fsrpath} -Scale {quality_setting} {quality_setting} {files_handle}'
-                elif os_type == 'win32':
-                    command_us = f'FidelityFX_CLI -Scale {quality_setting} {quality_setting} {files_handle}'
-                else:
-                    print( 'OS CURRENTLY UNSUPPORTED!' )
-                    return False
-            sub = subprocess.Popen( command_us, shell=True );
-            sub.wait();
-        if sharpening != '':
-            print( '\n\n\n PROCESS: ', number, '\nRunning sharpening filter\n\n\n' );
+        if quality_mode == 'default':
             if os_type == 'linux':
-                command_sharpening = f'wine {fsrpath} -Mode CAS -Sharpness {sharpening} {files_handle}'
+                command_us = f'wine {fsrpath} -QualityMode {quality_setting} {files_handle}'
             elif os_type == 'win32':
-                command_sharpening = f'FidelityFX_CLI -Mode CAS -Sharpness {sharpening} {files_handle}'
+                command_us = f'FidelityFX_CLI -QualityMode {quality_setting} {files_handle}'
             else:
                 print( 'OS CURRENTLY UNSUPPORTED!' )
                 return False
-            sub2 = subprocess.Popen( command_sharpening, shell=True );
-            sub2.wait()
-        
+        else:
+            if os_type == 'linux':
+                command_us = f'wine {fsrpath} -Scale {quality_setting} {quality_setting} {files_handle}'
+            elif os_type == 'win32':
+                command_us = f'FidelityFX_CLI -Scale {quality_setting} {quality_setting} {files_handle}'
+            else:
+                print( 'OS CURRENTLY UNSUPPORTED!' )
+                return False
+        sub = subprocess.Popen( command_us, shell=True );
+        sub.wait();        
         time.sleep(3)
     print( '\n\nCompleted executing Job\n\n\n PROCESS: ', number, '\n\n\n' );
+
+
+########################
+# 
+#   Sharpening
+#
+#######################
+
+def sharpeningEngine ( files, fsrpath, number, maxlength, os_type, sharpening ):
+    files = files;
+    # Refactoring of commands that are longer than 32K characters
+    fileout = [];
+    pos = 0;
+    if len( files ) > maxlength:
+        while files[maxlength - pos:maxlength - pos + 1] != ' ':
+            pos += 1
+        file_processing = files[:maxlength - pos]
+        if file_processing[len(file_processing) - 14:len(file_processing) - 12] == 'ex':
+            pos += 5
+        else:
+            pass
+        while files[maxlength - pos:maxlength - pos + 1] != ' ':
+            pos += 1
+        fileout.append(files[:maxlength - pos])
+        filesopt = files[maxlength - pos:]
+        posx = 0
+        posy = maxlength
+
+        # Command refactoring for commands that are longer than 64K characters
+        if len(filesopt) > maxlength:
+            while len(filesopt) > maxlength:
+                posx += maxlength - pos
+                posy += maxlength - pos
+                pos = 1
+                while files[posy - pos:posy - pos + 1] != ' ':
+                    pos += 1
+                file_processing = files[posx:posy - pos]
+                if file_processing[len(file_processing) - 14:len(file_processing) - 12] == 'ex':
+                    pos += 5
+                else:
+                    pass
+                while files[posy - pos:posy - pos + 1] != ' ':
+                    pos += 1
+
+                file_processing = files[posx:posy - pos]
+                fileout.append(file_processing)
+                filesopt = files[posy - pos:]
+            fileout.append(filesopt)
+        else:
+            fileout.append(files[maxlength - pos:])
+    else:
+        fileout.append(files)
+
+    # Upscaling images
+    print( '\n\n\nSharpening images... \n\n\n\n\n\n PROCESS: ', number, '\n\n\n' )
+
+    while len( fileout ) > 0:
+        files_handle = fileout.pop(0)
+        print( '\n\n\n PROCESS: ', number, '\nRunning sharpening filter\n\n\n' );
+        if os_type == 'linux':
+            command_sharpening = f'wine {fsrpath} -Mode CAS -Sharpness {sharpening} {files_handle}'
+        elif os_type == 'win32':
+            command_sharpening = f'FidelityFX_CLI -Mode CAS -Sharpness {sharpening} {files_handle}'
+        else:
+            print( 'OS CURRENTLY UNSUPPORTED!' )
+            return False
+        sub2 = subprocess.Popen( command_sharpening, shell=True );
+        sub2.wait()
+        time.sleep(3)
+    print( '\n\nCompleted executing Job\n\n\n PROCESS: ', number, '\n\n\n' );
+