from tkinter import messagebox from PIL import Image import shutil import glob import json import os from .CT_Config import CT_Config from .definition import * from . import wszst class RomAlreadyPatched(Exception): def __init__(self): super().__init__("ROM Already patched !") class InvalidGamePath(Exception): def __init__(self): super().__init__("This path is not valid !") class InvalidFormat(Exception): def __init__(self): super().__init__("This game format is not supported !") class TooMuchDownloadFailed(Exception): def __init__(self): super().__init__("Too much download failed !") class TooMuchSha1CheckFailed(Exception): def __init__(self): super().__init__("Too much sha1 check failed !") class CantConvertTrack(Exception): def __init__(self): super().__init__("Can't convert track, check if download are enabled.") class Game: def __init__(self, path: str = "", region_ID: str = "P", game_ID: str = "RMCP01", gui=None): if not os.path.exists(path) and path: raise InvalidGamePath() self.extension = None self.path = path self.set_path(path) self.region = region_id_to_name[region_ID] self.region_ID = region_ID self.game_ID = game_ID self.gui = gui self.ctconfig = CT_Config() def set_path(self, path): self.extension = get_extension(path).upper() self.path = path def convert_to(self, format: str = "FST"): """ :param format: game format (ISO, WBFS, ...) :return: converted game path """ if format in ["ISO", "WBFS", "CISO"]: path_game_format: str = os.path.realpath(self.path + "/../MKWFaraphel." + format.lower()) wszst.wit_copy(self.path, path_game_format, format) shutil.rmtree(self.path) self.path = path_game_format self.gui.progress(statut=self.gui.translate("Changing game's ID"), add=1) wszst.edit(self.path, region_ID=self.region_ID, name=f"Mario Kart Wii Faraphel {self.ctconfig.version}") def extract(self): if self.extension == "DOL": self.path = os.path.realpath(self.path + "/../../") # main.dol is in PATH/sys/, so go back 2 dir upper elif self.extension in ["ISO", "WBFS", "CSIO"]: # Fiding a directory name that doesn't already exist directory_name, i = "MKWiiFaraphel", 1 while True: path_dir = os.path.realpath(self.path + f"/../{directory_name}") if not (os.path.exists(path_dir)): break directory_name, i = f"MKWiiFaraphel ({i})", i + 1 wszst.wit_extract(self.path, path_dir) self.path = path_dir if os.path.exists(self.path + "/DATA"): self.path += "/DATA" self.extension = "DOL" else: raise InvalidFormat() if glob.glob(self.path + "/files/rel/lecode-???.bin"): # if a LECODE file is already here raise RomAlreadyPatched() # warning already patched with open(self.path + "/setup.txt") as f: setup = f.read() setup = setup[setup.find("!part-id = ") + len("!part-id = "):] self.game_ID = setup[:setup.find("\n")] self.region_ID = self.game_ID[3] self.region = region_id_to_name[self.region_ID] if self.region_ID in region_id_to_name else self.region @in_thread def install_mod(self): try: with open("./fs.json") as f: fs = json.load(f) # This part is used to estimate the max_step extracted_file = [] max_step, step = 1, 0 def count_rf(path): nonlocal max_step max_step += 1 if get_extension(path) == "szs": if not (os.path.realpath(path) in extracted_file): extracted_file.append(os.path.realpath(path)) max_step += 1 for fp in fs: for f in glob.glob(self.path + "/files/" + fp, recursive=True): if type(fs[fp]) == str: count_rf(path=f) elif type(fs[fp]) == dict: for nf in fs[fp]: if type(fs[fp][nf]) == str: count_rf(path=f) elif type(fs[fp][nf]) == list: for ffp in fs[fp][nf]: count_rf(path=f) ### extracted_file = [] max_step += 4 # PATCH main.dol and PATCH lecode.bin, converting, changing ID self.gui.progress(show=True, indeter=False, statut=self.gui.translate("Installing mod"), max=max_step, step=0) def replace_file(path, file, subpath="/"): self.gui.progress(statut=self.gui.translate("Editing", "\n", get_nodir(path)), add=1) extension = get_extension(path) if extension == "szs": if not (os.path.realpath(path) in extracted_file): wszst.szs_extract(path, get_nodir(path)) extracted_file.append(os.path.realpath(path)) szs_extract_path = path + ".d" if os.path.exists(szs_extract_path + subpath): if subpath[-1] == "/": filecopy(f"./file/{file}", szs_extract_path + subpath + file) else: filecopy(f"./file/{file}", szs_extract_path + subpath) elif path[-1] == "/": filecopy(f"./file/{file}", path + file) else: filecopy(f"./file/{file}", path) for fp in fs: for f in glob.glob(self.path + "/files/" + fp, recursive=True): if type(fs[fp]) == str: replace_file(path=f, file=fs[fp]) elif type(fs[fp]) == dict: for nf in fs[fp]: if type(fs[fp][nf]) == str: replace_file(path=f, subpath=nf, file=fs[fp][nf]) elif type(fs[fp][nf]) == list: for ffp in fs[fp][nf]: replace_file(path=f, subpath=nf, file=ffp) for file in extracted_file: self.gui.progress(statut=self.gui.translate("Recompilating", "\n", get_nodir(file)), add=1) wszst.create(file) if os.path.exists(file + ".d"): shutil.rmtree(file + ".d") self.gui.progress(statut=self.gui.translate("Patch main.dol"), add=1) wszst.str_patch(self.path) self.gui.progress(statut=self.gui.translate("Patch lecode.bin"), add=1) shutil.copytree("./file/Track/", self.path + "/files/Race/Course/", dirs_exist_ok=True) if not (os.path.exists(self.path + "/tmp/")): os.makedirs(self.path + "/tmp/") filecopy("./file/CTFILE.txt", self.path + "/tmp/CTFILE.txt") filecopy("./file/lpar-default.txt", self.path + "/tmp/lpar-default.txt") filecopy(f"./file/lecode-{self.region}.bin", self.path + f"/tmp/lecode-{self.region}.bin") wszst.lec_patch( self.path, lecode_file=f"./tmp/lecode-{self.region}.bin", dest_lecode_file=f"./files/rel/lecode-{self.region}.bin", ) shutil.rmtree(self.path + "/tmp/") output_format = self.gui.stringvar_game_format.get() self.gui.progress(statut=self.gui.translate("Converting to", " ", output_format), add=1) self.convert_to(output_format) messagebox.showinfo(self.gui.translate("End"), self.gui.translate("The mod has been installed !")) except: self.gui.log_error() finally: self.gui.progress(show=False) def patch_autoadd(self, auto_add_dir: str = "./file/auto-add"): if os.path.exists(auto_add_dir): shutil.rmtree(auto_add_dir) if not os.path.exists(self.path + "/tmp/"): os.makedirs(self.path + "/tmp/") wszst.autoadd(self.path, get_nodir(self.path) + "/tmp/auto-add/") shutil.move(self.path + "/tmp/auto-add/", auto_add_dir) shutil.rmtree(self.path + "/tmp/") def patch_bmg(self, gamefile: str): # gamefile est le fichier .szs trouvé dans le /files/Scene/UI/ du jeu NINTENDO_CWF_REPLACE = "Wiimmfi" MAINMENU_REPLACE = f"MKWFaraphel {self.ctconfig.version}" menu_replacement = { "CWF de Nintendo": NINTENDO_CWF_REPLACE, "Wi-Fi Nintendo": NINTENDO_CWF_REPLACE, "CWF Nintendo": NINTENDO_CWF_REPLACE, "Nintendo WFC": NINTENDO_CWF_REPLACE, "Wi-Fi": NINTENDO_CWF_REPLACE, "インターネット": NINTENDO_CWF_REPLACE, "Menu principal": MAINMENU_REPLACE, "Menú principal": MAINMENU_REPLACE, "Main Menu": MAINMENU_REPLACE, "トップメニュー": MAINMENU_REPLACE, "Mario Kart Wii": MAINMENU_REPLACE, } bmglang = gamefile[-len("E.txt"):-len(".txt")] # Langue du fichier self.gui.progress(statut=self.gui.translate("Patching text", " ", bmglang), add=1) wszst.szs_extract(gamefile, get_nodir(gamefile)) bmgmenu = wszst.bmg_cat(gamefile, ".d/message/Menu.bmg") # Menu.bmg bmgtracks = wszst.bmg_cat(gamefile, ".d/message/Common.bmg") # Common.bmg trackheader = "#--- standard track names" trackend = "2328" bmgtracks = bmgtracks[bmgtracks.find(trackheader) + len(trackheader):bmgtracks.find(trackend)] with open("./file/ExtraCommon.txt", "w", encoding="utf8") as f: f.write("#BMG\n\n" f" 703e\t= \\\\c{{white}}{self.gui.translate('Random: All tracks', lang=bmglang)}\n" f" 703f\t= \\\\c{{white}}{self.gui.translate('Random: Original tracks', lang=bmglang)}\n" f" 7040\t= \\\\c{{white}}{self.gui.translate('Random: Custom Tracks', lang=bmglang)}\n" f" 7041\t= \\\\c{{white}}{self.gui.translate('Random: New tracks', lang=bmglang)}\n") for bmgtrack in bmgtracks.split("\n"): if "=" in bmgtrack: prefix = "" if "T" in bmgtrack[:bmgtrack.find("=")]: sTid = bmgtrack.find("T") Tid = bmgtrack[sTid:sTid + 3] if Tid[1] in "1234": prefix = trackname_color["Wii"] + " " # Si la course est original à la wii Tid = hex(bmgID_track_move[Tid])[2:] else: # Arena sTid = bmgtrack.find("U") + 1 Tid = bmgtrack[sTid:sTid + 2] Tid = hex((int(Tid[0]) - 1) * 5 + (int(Tid[1]) - 1) + 0x7020)[2:] Tname = bmgtrack[bmgtrack.find("= ") + 2:] f.write(f" {Tid}\t= {prefix}{Tname}\n") if not (os.path.exists("./file/tmp/")): os.makedirs("./file/tmp/") filecopy(gamefile + ".d/message/Common.bmg", "./file/tmp/Common.bmg") bmgcommon = wszst.ctc_patch_bmg(ctfile="./file/CTFILE.txt", bmgs=["./file/tmp/Common.bmg", "./file/ExtraCommon.txt"]) rbmgcommon = wszst.ctc_patch_bmg(ctfile="./file/RCTFILE.txt", bmgs=["./file/tmp/Common.bmg", "./file/ExtraCommon.txt"]) shutil.rmtree(gamefile + ".d") os.remove("./file/tmp/Common.bmg") os.remove("./file/ExtraCommon.txt") def finalise(file, bmgtext, replacement_list=None): if replacement_list: for text, colored_text in replacement_list.items(): bmgtext = bmgtext.replace(text, colored_text) with open(file, "w", encoding="utf-8") as f: f.write(bmgtext) wszst.bmg_encode(file) os.remove(file) finalise(f"./file/Menu_{bmglang}.txt", bmgmenu, menu_replacement) finalise(f"./file/Common_{bmglang}.txt", bmgcommon) finalise(f"./file/Common_R{bmglang}.txt", rbmgcommon) @in_thread def patch_file(self): try: if not (os.path.exists("./file/Track-WU8/")): os.makedirs("./file/Track-WU8/") with open("./convert_file.json") as f: fc = json.load(f) max_step = len(fc["img"]) + len(self.ctconfig.all_tracks) + 3 + len("EGFIS") self.gui.progress(show=True, indeter=False, statut=self.gui.translate("Converting files"), max=max_step, step=0) self.gui.progress(statut=self.gui.translate("Configurating LE-CODE"), add=1) self.ctconfig.create_ctfile() self.gui.progress(statut=self.gui.translate("Creating ct_icon.png"), add=1) ct_icon = self.ctconfig.get_cticon() ct_icon.save("./file/ct_icons.tpl.png") self.gui.progress(statut=self.gui.translate("Creating descriptive images"), add=1) self.patch_img_desc() self.patch_image(fc) for file in glob.glob(self.path + "/files/Scene/UI/MenuSingle_?.szs"): self.patch_bmg(file) # MenuSingle could be any other file, Common and Menu are all the same in all other files. self.patch_autoadd() if self.patch_tracks() != 0: return self.gui.button_install_mod.grid(row=2, column=1, columnspan=2, sticky="NEWS") self.gui.button_install_mod.config( text=self.gui.translate("Install mod", " (v", self.ctconfig.version, ")")) except: self.gui.log_error() finally: self.gui.progress(show=False) def patch_image(self, fc): for i, file in enumerate(fc["img"]): self.gui.progress(statut=self.gui.translate("Converting images") + f"\n({i + 1}/{len(fc['img'])}) {file}", add=1) wszst.img_encode("./file/" + file, fc["img"][file]) def patch_img_desc(self, img_desc_path: str = "./file/img_desc", dest_dir: str = "./file"): il = Image.open(img_desc_path + "/illustration.png") il_16_9 = il.resize((832, 456)) il_4_3 = il.resize((608, 456)) for file_lang in glob.glob(img_desc_path + "??.png"): img_lang = Image.open(file_lang) img_lang_16_9 = img_lang.resize((832, 456)) img_lang_4_3 = img_lang.resize((608, 456)) new_16_9 = Image.new("RGBA", (832, 456), (0, 0, 0, 255)) new_16_9.paste(il_16_9, (0, 0), il_16_9) new_16_9.paste(img_lang_16_9, (0, 0), img_lang_16_9) new_16_9.save(dest_dir + f"/strapA_16_9_832x456{get_filename(get_nodir(file_lang))}.png") new_4_3 = Image.new("RGBA", (608, 456), (0, 0, 0, 255)) new_4_3.paste(il_4_3, (0, 0), il_4_3) new_4_3.paste(img_lang_4_3, (0, 0), img_lang_4_3) new_4_3.save(dest_dir + f"/strapA_608x456{get_filename(get_nodir(file_lang))}.png") def patch_tracks(self): max_process = self.gui.intvar_process_track.get() process_list = {} error_count, error_max = 0, 3 def add_process(track): nonlocal error_count, error_max, process_list track_file = track.get_track_name() total_track = len(self.ctconfig.all_tracks) process_list[track_file] = None # Used for showing track in progress even if there's no process self.gui.progress(statut=self.gui.translate("Converting tracks", f"\n({i + 1}/{total_track})\n", "\n".join(process_list.keys())), add=1) for _track in [track.file_szs, track.file_wu8]: if os.path.exists(_track): if os.path.getsize(_track) < 1000: # File under this size are corrupted os.remove(_track) if not self.gui.boolvar_disable_download.get(): while True: download_returncode = track.download_wu8() if download_returncode == -1: # can't download error_count += 1 if error_count > error_max: # Too much track wasn't correctly converted messagebox.showerror( self.gui.translate("Error"), self.gui.translate("Too much tracks had a download issue.")) raise TooMuchDownloadFailed() else: messagebox.showwarning(self.gui.translate("Warning"), self.gui.translate("Can't download this track !", f" ({error_count} / {error_max})")) elif download_returncode == 2: break # if download is disabled, do not check sha1 if track.sha1: if not self.gui.boolvar_dont_check_track_sha1.get(): if not track.check_sha1(): # Check si le sha1 du fichier est le bon error_count += 1 if error_count > error_max: # Too much track wasn't correctly converted messagebox.showerror( self.gui.translate("Error"), self.gui.translate("Too much tracks had an issue during sha1 check.")) raise TooMuchSha1CheckFailed() continue break if not (os.path.exists(track.file_szs)) or download_returncode == 3: # returncode 3 is track has been updated if os.path.exists(track.file_wu8): process_list[track_file] = track.convert_wu8_to_szs() else: messagebox.showerror(self.gui.translate("Error"), self.gui.translate("Can't convert track.\nEnable track download and retry.")) raise CantConvertTrack() elif self.gui.boolvar_del_track_after_conv.get(): os.remove(track.file_wu8) return 0 def clean_process(): nonlocal error_count, error_max, process_list for track_file, process in process_list.copy().items(): if process is not None: if process.poll() is None: pass # if the process is still running else: # process ended process_list.pop(track_file) stderr = process.stderr.read() if b"wszst: ERROR" in stderr: # Error occured os.remove(track.file_szs) error_count += 1 if error_count > error_max: # Too much track wasn't correctly converted messagebox.showerror( self.gui.translate("Error"), self.gui.translate("Too much track had a conversion issue.")) raise CantConvertTrack() else: # if the error max hasn't been reach messagebox.showwarning( self.gui.translate("Warning"), self.gui.translate("The track", " ", track.file_wu8, "do not have been properly converted.", f" ({error_count} / {error_max})")) else: if self.gui.boolvar_del_track_after_conv.get(): os.remove(track.file_wu8) else: process_list.pop(track_file) if not (any(process_list.values())): return 1 # if there is no more process if len(process_list): return 1 else: return 0 for i, track in enumerate(self.ctconfig.all_tracks): while True: if len(process_list) < max_process: returncode = add_process(track) if returncode == 0: break elif returncode == -1: return -1 # if error occur, stop function elif clean_process() == -1: return -1 while True: returncode = clean_process() if returncode == 1: break # End the process if all process ended elif returncode == 0: pass else: return -1 return 0