Atlas/Atlas-Install
1
0
Fork 0
mirror of https://github.com/Faraphel/Atlas-Install.git synced 2025-07-03 11:18:26 +02:00
Code Issues Projects Releases Packages Wiki Activity Actions

added ctfile code, fixed issue, load from json function

Browse source
This commit is contained in:
raphael60650 2021-07-16 22:44:41 +02:00
parent 7b1b2dc73e
commit c041f55cb9
4 changed files with 114 additions and 41 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

102
source/CT_Config.py
View file

@ -5,8 +5,8 @@ import os
def get_cup_icon(i):
if os.path.exists(f"./file/cup_icon/{id}.png"):
cup_icon = Image.open(f"./file/cup_icon/{id}.png").resize((128, 128))
if os.path.exists(f"./file/cup_icon/{i}.png"):
cup_icon = Image.open(f"./file/cup_icon/{i}.png").resize((128, 128))
else:
cup_icon = Image.new("RGBA", (128, 128))
@ -19,51 +19,105 @@ def get_cup_icon(i):
draw.text((4, 4), "CT", (255, 165, 0), font=font)
font = ImageFont.truetype("./file/SuperMario256.ttf", 60)
draw.text((5 - 2, 80 - 2), "%03i" % (i - 10), (0, 0, 0), font=font) # i-10 because first 8 cup are not
draw.text((5 + 2, 80 - 2), "%03i" % (i - 10), (0, 0, 0), font=font) # counted as new, random cup,
draw.text((5 - 2, 80 + 2), "%03i" % (i - 10), (0, 0, 0), font=font) # left and right arrow
draw.text((5 + 2, 80 + 2), "%03i" % (i - 10), (0, 0, 0), font=font)
draw.text((5 - 2, 80 - 2), "%03i" % i, (0, 0, 0), font=font)
draw.text((5 + 2, 80 - 2), "%03i" % i, (0, 0, 0), font=font)
draw.text((5 - 2, 80 + 2), "%03i" % i, (0, 0, 0), font=font)
draw.text((5 + 2, 80 + 2), "%03i" % i, (0, 0, 0), font=font)
draw.text((5, 80), "%03i" % (i - 10), (255, 165, 0), font=font)
return cup_icon
draw.text((5, 80), "%03i" % i, (255, 165, 0), font=font)
return cup_icon
class CT_Config:
def __init__(self, version):
def __init__(self, version: str = None):
self.version = version
self.ordered_cups = []
self.unordered_tracks = []
self.all_tracks = []
self.all_version: set = {version}
def load_ctconfig_json(self, ctconfig_json: dict):
"""
:param ctconfig_json: json of the ctconfig to load
:return: ?
"""
self.ordered_cups = []
self.unordered_tracks = []
self.all_tracks = []
for cup_json in ctconfig_json["cup"].values(): # tracks with defined order
cup = Cup()
cup.load_from_json(cup_json)
if not cup.locked: # locked cup are not useful (they are original track or random track)
self.ordered_cups.append(cup)
self.all_tracks.extend(cup.tracks)
for track_json in ctconfig_json["tracks_list"]: # unordered tracks
track = Track()
track.load_from_json(track_json)
self.unordered_tracks.append(track)
self.all_tracks.append(track)
self.version = ctconfig_json["version"]
self.all_version = set()
for track in self.all_tracks:
self.all_version.add(track.since_version)
self.all_version = sorted(self.all_version)
def load_ctconfig_file(self, ctconfig_file: str = "./ct_config.json"):
"""
:param ctconfig_file: path to the ctconfig file
:return: ?
"""
with open(ctconfig_file, encoding="utf-8") as f:
ctconfig_json = json.load(f)
self.load_ctconfig_json(ctconfig_json)
def add_ordered_cup(self, cup: Cup):
"""
:param cup: a Cup object to add as an ordered cup
:return: ?
"""
self.ordered_cups.append(cup)
for track in cup.tracks:
self.all_version.add(track.since_version)
self.all_tracks.append(track)
def add_unordered_track(self, track: Track):
"""
:param track: a Track object to add as an unordered tracks
:return: ?
"""
self.unordered_tracks.append(track)
self.all_version.add(track.since_version)
self.all_tracks.append(track)
def get_tracks(self):
def search_tracks(self, values_list=False, not_value=False, **kwargs):
"""
:return: all tracks from the CT_Config
:param values_list: search track with a value list instead of a single value
:param not_value: search track that does not have value
:param kwargs: any track property = any value
:return: track list respecting condition
"""
tracks = []
for cup in self.ordered_cups:
tracks.extend(*cup.get_tracks())
tracks.extend(self.unordered_tracks)
track = self.all_tracks.copy()
return tracks
if values_list:
if not_value: filter_func = lambda track: getattr(track, keyword) not in value
else: filter_func = lambda track: getattr(track, keyword) in value
else:
if not_value: filter_func = lambda track: getattr(track, keyword) != value
else: filter_func = lambda track: getattr(track, keyword) == value
def search_tracks(self, **kwargs):
tracks = self.get_tracks()
for keyword, value in kwargs.items():
filter(lambda track: getattr(track, keyword) == value, tracks)
def get_total_tracks_count(self):
return (len(self.ordered_cups) * 4) + len(self.unordered_tracks)
track = list(filter(filter_func, track))
return track
def create_ctfile(self, directory="./file/"):
"""
:param directory: create CTFILE.txt and RCTFILE.txt in this directory
:return: None
"""
with open(directory+"CTFILE.txt", "w", encoding="utf-8") as ctfile, \
open(directory+"RCTFILE.txt", "w", encoding="utf-8") as rctfile:
header = (
@ -95,10 +149,10 @@ class CT_Config:
CT_ICON_WIDTH = 128
icon_files = ["left", "right"]
total_cup_count = math.ceil(self.get_total_tracks_count() / 4)
total_cup_count = math.ceil(len(self.all_tracks) / 4)
ct_icon = Image.new("RGBA", (CT_ICON_WIDTH, CT_ICON_WIDTH * (total_cup_count + 2))) # +2 because of left and right arrow
icon_files.extend([cup.id for cup in self.ordered_cups]) # adding ordered cup id
icon_files.extend([str(i) for i, cup in enumerate(self.ordered_cups)]) # adding ordered cup id
icon_files.extend(["_"] * ((len(self.unordered_tracks) // 4) + 1)) # creating unordered track icon
for i, id in enumerate(icon_files):

28
source/Cup.py
View file

@ -4,16 +4,30 @@ from .patch_ct_icon import get_cup_icon
class Cup:
def __init__(self, name: str,
track1: Track = EMPTY_TRACK,
track2: Track = EMPTY_TRACK,
track3: Track = EMPTY_TRACK,
track4: Track = EMPTY_TRACK,
icon: Image = None):
def __init__(self, name: str = None,
track1: Track = None,
track2: Track = None,
track3: Track = None,
track4: Track = None,
icon: Image = None, locked: bool = False,
*args, **kwargs):
self.name = name
self.tracks = [track1, track2, track3, track4]
self.tracks = [
track1 if track1 else Track(),
track2 if track2 else Track(),
track3 if track3 else Track(),
track4 if track4 else Track()
]
self.icon = icon
self.locked = locked
def load_from_json(self, cup: dict):
for key, value in cup.items(): # load all value in the json as class attribute
if key != "tracks": setattr(self, key, value)
else: # if the key is tracks
for i, track_json in value.items(): # load all tracks from their json
self.tracks[int(i)].load_from_json(track_json)
def get_ctfile_cup(self, race=False):
"""

7
source/Game.py
View file

@ -6,17 +6,17 @@ import os
class Game:
def __init__(self, path: str, region: str = "PAL", game_ID: str = "RMCP01"):
self.extension = get_extension(path)
self.path = path
self.region_ID = region_ID[region]
self.region = region
self.game_ID = game_ID
def extract_game(self):
extension = get_extension(self.path)
if extension.upper() == "DOL":
if self.extension.upper() == "DOL":
self.path = os.path.realpath(self.path + "/../../") # main.dol is in PATH/sys/, so go back 2 dir upper
elif extension.upper() in ["ISO", "WBFS", "CSIO"]:
elif self.extension.upper() in ["ISO", "WBFS", "CSIO"]:
# Fiding a directory name that doesn't already exist
directory_name, i = "MKWiiFaraphel", 1
while True:
@ -26,6 +26,7 @@ class Game:
wszst.extract(self.path, self.path)
if os.path.exists(self.path + "/DATA"): self.path += "/DATA"
self.extension = "DOL"
else:
raise Exception("This format is not supported !")

18
source/Track.py
View file

@ -3,9 +3,9 @@ import source.wszst
class Track:
def __init__(self, name: str, file_wu8: str = None, file_szs: str = None, prefix: str = None, suffix: str = None,
def __init__(self, name: str = "_", file_wu8: str = None, file_szs: str = None, prefix: str = None, suffix: str = None,
author="Nintendo", special="T11", music="T11", new=True, sha1: str = None, since_version: str = None,
score: int = 0, warning: int = 0, note: str = ""):
score: int = 0, warning: int = 0, note: str = "", *args, **kwargs):
self.name = name # Track name
self.prefix = prefix # Prefix, often used for game or original console like Wii U, DS, ...
@ -16,11 +16,14 @@ class Track:
self.music = music # Music of the track
self.new = new # Is the track new
self.since_version = since_version # Since which version is this track available
self.file_wu8 = file_wu8
self.file_szs = file_szs
self.score = score # Track score between 1 and 3 stars
self.warning = warning # Track bug level (1 = minor, 2 = major)
self.note = note # Note about the track
self.file_wu8 = file_wu8
self.file_szs = file_szs
def __repr__(self):
return f"{self.get_track_name()} sha1={self.sha1}"
def get_track_name(self):
prefix = self.prefix + " " if self.prefix else ""
@ -29,6 +32,10 @@ class Track:
name = (prefix + self.name + suffix)
return name
def load_from_json(self, track: dict):
for key, value in track.items(): # load all value in the json as class attribute
setattr(self, key, value)
def get_track_formatted_name(self, highlight_track_from_version: str = None):
"""
:param highlight_track_from_version: if a specific version need to be highlighted.
@ -89,6 +96,3 @@ class Track:
)
return ctfile_text
EMPTY_TRACK = Track("_")