Now converting bmps to pngs.

Author: MyNameIsTrez

Media/palette.bmp

@@ -7,6 +7,7 @@
 from Python import regex_rules
 from Python import zips as zips_py
 from Python import update_progress
+from Python import palette
 
 
 warnings_file_name = "Warnings.json"
@@ -26,7 +27,7 @@ def resource_path(relative_path):
 
 
 # If an exe executes this program then sys is frozen.
-output_folder = ".." if getattr(sys, 'frozen', False) else "Output"
+output_folder_path = ".." if getattr(sys, 'frozen', False) else "Output"
 
 
 def load_conversion_and_warning_rules():
@@ -57,7 +58,7 @@ def check_github_button_clicked_and_exit(clicked_github_button):
 
 
 def convert():
-	global output_folder, warnings
+	global output_folder_path, warnings
 
 	print("") # Only prints a newline.
 
@@ -76,12 +77,12 @@ def convert():
 
 		if len(mod_subfolder_parts) > 0 and mod_subfolder_parts[0].endswith(".rte"):
 			try_print_mod_name(mod_subfolder_parts, mod_subfolder)
-			output_subfolder = os.path.join(output_folder, mod_subfolder)
+			output_subfolder = os.path.join(output_folder_path, mod_subfolder)
 			create_folder(input_subfolder_path, output_subfolder)
-			process_files(input_subfiles, input_subfolder_path, output_subfolder, input_folder_path)
+			process_files(input_subfiles, input_subfolder_path, output_subfolder, input_folder_path, output_folder_path)
 
 	if cfg.sg.user_settings_get_entry("output_zips"):
-		zips_py.create_zips(input_folder_path, output_folder)
+		zips_py.create_zips(input_folder_path, output_folder_path)
 
 	if len(warnings) > 0:
 		warnings_popup()
@@ -115,20 +116,25 @@ def create_folder(input_subfolder_path, output_subfolder):
 		pass
 
 
-def process_files(input_subfiles, input_subfolder_path, output_subfolder, input_folder_path):
+def process_files(input_subfiles, input_subfolder_path, output_subfolder, input_folder_path, output_folder_path):
 	for full_filename in input_subfiles:
 		filename, file_extension = os.path.splitext(full_filename)
 
-		input_file_path  = os.path.join(input_subfolder_path, full_filename)
+		input_file_path = os.path.join(input_subfolder_path, full_filename)
+
 		output_file_path = os.path.join(output_subfolder, full_filename)
 
+		if palette.is_input_image(full_filename):
+			palette.process_image(full_filename, input_file_path, output_file_path)
+
 		if full_filename == "desktop.ini":
 			continue
 
 		if file_extension in (".ini", ".lua"):
 			create_converted_file(input_file_path, output_file_path, input_folder_path)
 		else:
-			shutil.copyfile(input_file_path, output_file_path)
+			if not palette.is_input_image(full_filename):
+				shutil.copyfile(input_file_path, output_file_path)
 
 		update_progress.increment_progress()
 

Python/convert.py

@@ -0,0 +1,35 @@
+import os
+
+from PIL import Image
+import cv2
+
+
+palette = Image.open(os.path.join("Media", "palette.bmp")).getpalette()
+
+
+def is_input_image(full_filename):
+	return full_filename.lower().endswith((".bmp", ".png", ".jpg", ".jpeg")) and full_filename.lower() != "palette.bmp"
+
+
+def process_image(full_filename, input_image_path, output_image_path):
+	global palette
+
+	# A bmp threw an "Unsupported BMP compression (1)" with Image.open(), so we open it in cv2 first.
+	# cv2 solution: https://stackoverflow.com/a/52416250/13279557
+	# cv2 to PIL:   https://stackoverflow.com/a/43234001/13279557
+	# old_img = Image.fromarray(cv2.imread(input_image_path)) # TODO: This may work just as well as the line below for all cases. 
+	old_img = Image.fromarray(cv2.cvtColor(cv2.imread(input_image_path), cv2.COLOR_BGR2RGB))
+
+	scale = 1
+	old_img = old_img.resize((int(old_img.width * scale), int(old_img.height * scale)))
+
+	# putpalette() always expects 256 * 3 ints.
+	for k in range(256 - int(len(palette) / 3)):
+		for j in range(3):
+			palette.append(palette[j])
+
+	palette_img = Image.new('P', (1, 1))
+	palette_img.putpalette(palette)
+	new_img = old_img.convert(mode="RGB").quantize(palette=palette_img, dither=False)
+
+	new_img.save(os.path.splitext(output_image_path)[0] + ".png")

Python/palette.py

@@ -2,4 +2,6 @@ PySimpleGUI
 playsound
 pyinstaller
 jsoncomment
+Pillow
+opencv-python
 # python3 needs sudo apt-get install python3-tk