Atlas-Install/source/patch_bmg.py

import subprocess
import shutil
import os

from .definition import *

bmgID_track_move = {
    "T11": 0x7008, "T12": 0x7001, "T13": 0x7002, "T14": 0x7004,
    "T21": 0x7000, "T22": 0x7005, "T23": 0x7006, "T24": 0x7007,
    "T31": 0x7009, "T32": 0x700f, "T33": 0x700b, "T34": 0x7003,
    "T41": 0x700e, "T42": 0x700a, "T43": 0x700c, "T44": 0x700d,

    "T51": 0x7010, "T52": 0x7014, "T53": 0x7019, "T54": 0x701a,
    "T61": 0x701b, "T62": 0x701f, "T63": 0x7017, "T64": 0x7012,
    "T71": 0x7015, "T72": 0x701e, "T73": 0x701d, "T74": 0x7011,
    "T81": 0x7018, "T82": 0x7016, "T83": 0x7013, "T84": 0x701c,
}
trackname_color = {
    "MSRDS ": "\c{green}MSRDS\c{off} ",
    "CTR ": "\c{YOR4}CTR\c{off} ",
    "CTTR ": "\c{YOR5}CTTR\c{off} ",
    "CNR ": "\c{YOR5}CNR\c{off} ",
    "DKR ": "\c{YOR6}DKR\c{off} ",
    "LCP ": "\c{green}LCP\c{off} ",
    "LEGO-R ": "\c{red2}LEGO-R\c{off} ",
    "MP9 ": "\c{YOR0}MP9\c{off} ",
    "MSUSA ": "\c{green}MSUSA\c{off} ",
    "FZMV ": "\c{YOR2}FZMV\c{off} ",
    "KAR ": "\c{green}KAR\c{off} ",
    "KO ": "\c{YOR5}KO\c{off} ",
    "FZ ": "\c{YOR2}FZ\c{off} ",
    "RV ": "\c{white}RV\c{off} ",
    "SADX ": "\c{blue2}SADX\c{off} ",
    "SCR ": "\c{YOR2}SCR\c{off} ",
    "SH ": "\c{red}SH\c{off} ",
    "SM64 ": "\c{red1}SM64\c{off} ",
    "SMB1 ": "\c{red2}SMB1\c{off} ",
    "SMB2 ": "\c{red3}SMB2\c{off} ",
    "SSBB ": "\c{red4}SSBB\c{off} ",
    "SMS ": "\c{YOR6}SMS\c{off} ",
    "SMO ": "\c{YOR7}SMO\c{off} ",
    "VVVVVV ": "\c{blue}VVVVVV\c{off} ",
    "WF ": "\c{green}WF\c{off} ",
    "WP ": "\c{yellow}WP\c{off} ",
    "Zelda OoT ": "\c{green}Zelda OoT\c{off} ",
    "Zelda TP ": "\c{green}Zelda TP\c{off} ",
    "Zelda WW ": "\c{green}Zelda WW\c{off} ",
    "PMWR ": "\c{yellow}PMWR\c{off} ",
    "SHR ": "\c{green}SHR\c{off} ",
    "SK64 ": "\c{green}SK64\c{off} ",
    "SMG ": "\c{red2}SMG\c{off} ",
    "Spyro 1 ": "\c{blue}Spyro 1\c{off} ",

    "Aléatoire: ": "\c{white}Aléatoire: ",
    "Random: ": "\c{white}Random: ",
    "Zufällig: ": "\c{white}Zufällig: ",
    "Casuale: ": "\c{white}Casuale: ",
    "Aleatorio: ": "\c{white}Aleatorio: ",

    "Wii U ": "WiiU ",
    "Wii ": "\c{blue}Wii\c{off} ",
    "WiiU ": "\c{red4}Wii U\c{off} ", # Permet d'éviter que Wii et Wii U se mélange

    "3DS ": "\c{YOR3}3DS\c{off} ",
    "DS ": "\c{white}DS\c{off} ",
    "GCN ": "\c{blue2}GCN\c{off} ",
    "GBA ": "\c{blue1}GBA\c{off} ",
    "N64 ": "\c{red}N64\c{off} ",
    "SNES ": "\c{green}SNES\c{off} ",
    "RMX ": "\c{YOR4}RMX\c{off} ",
    "MKT ": "\c{YOR5}MKT\c{off} ",
    "GP ": "\c{YOR6}GP\c{off} ",

    "(Boost)": "\c{YOR3}(Boost)\c{off}",

    "★★★ ": "\c{YOR2}★★★\c{off} ",
    "★★☆ ": "\c{YOR2}★★☆\c{off} ",
    "★☆☆ ": "\c{YOR2}★☆☆\c{off} ",
    "★★★! ": "\c{YOR4}★★★\c{off} ",
    "★★☆! ": "\c{YOR4}★★☆\c{off} ",
    "★☆☆! ": "\c{YOR4}★☆☆\c{off} ",
    "★★★!! ": "\c{YOR6}★★★\c{off} ",
    "★★☆!! ": "\c{YOR6}★★☆\c{off} ",
    "★☆☆!! ": "\c{YOR6}★☆☆\c{off} ",
}


def patch_bmg(self, gamefile):  # gamefile est le fichier .szs trouvé dans le /files/Scene/UI/ du jeu
    try:
        bmglang = gamefile[-len("E.txt"):-len(".txt")]  # Langue du fichier
        self.Progress(statut=self.translate("Patch des textes " + bmglang), add=1)

        subprocess.run(["./tools/szs/wszst", "EXTRACT", get_nodir(gamefile), "-d", get_nodir(gamefile) + ".d",
                       "--overwrite"], creationflags=CREATE_NO_WINDOW, cwd=get_dir(gamefile))

        # Common.bmg
        bmgtracks = subprocess.run(["./tools/szs/wbmgt", "CAT", get_nodir(gamefile) + ".d/message/Common.bmg"],
                                   creationflags=CREATE_NO_WINDOW, cwd=get_dir(gamefile),
                                   check=True, stdout=subprocess.PIPE).stdout
        bmgtracks = bmgtracks.decode()
        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= {self.translate('Aléatoire: Toutes les pistes', lang=bmglang)}\n"
                    f"  703f\t= {self.translate('Aléatoire: Pistes Originales', lang=bmglang)}\n"
                    f"  7040\t= {self.translate('Aléatoire: Custom Tracks', lang=bmglang)}\n"
                    f"  7041\t= {self.translate('Aléatoire: Pistes Nouvelles', 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 = "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")
        bmgtext = subprocess.run(["tools/szs/wctct", "bmg", "--le-code", "--long", "./file/CTFILE.txt", "--patch-bmg",
                                 "OVERWRITE=./file/tmp/Common.bmg", "--patch-bmg", "OVERWRITE=./file/ExtraCommon.txt"],
                                 creationflags=CREATE_NO_WINDOW, check=True, stdout=subprocess.PIPE).stdout.decode()
        rbmgtext = subprocess.run(["tools/szs/wctct", "bmg", "--le-code", "--long", "./file/RCTFILE.txt", "--patch-bmg",
                                  "OVERWRITE=./file/tmp/Common.bmg", "--patch-bmg", "OVERWRITE=./file/ExtraCommon.txt"],
                                  creationflags=CREATE_NO_WINDOW, check=True, stdout=subprocess.PIPE).stdout.decode()

        shutil.rmtree(gamefile + ".d")
        os.remove("./file/tmp/Common.bmg")
        os.remove("./file/ExtraCommon.txt")

        def finalise(common_file, bmgtext):
            for console in trackname_color: bmgtext = bmgtext.replace(console, trackname_color[console])
            with open(common_file, "w", encoding="utf-8") as f: f.write(bmgtext)
            subprocess.run(["./tools/szs/wbmgt", "ENCODE", get_nodir(common_file), "--overwrite"],
                           creationflags=CREATE_NO_WINDOW, cwd=get_dir(common_file))
            os.remove(common_file)

        finalise(f"./file/Common_{bmglang}.txt", bmgtext)
        finalise(f"./file/Common_R{bmglang}.txt", rbmgtext)

    except:
        self.log_error()