refactor: overhaul ascendancy fixer script
This commit is contained in:
ppoelzl
@@ -1,58 +1,91 @@
|
||||
from __future__ import annotations
|
||||
|
||||
import dataclasses
|
||||
import json
|
||||
import logging
|
||||
import os
|
||||
import pathlib
|
||||
|
||||
logging.basicConfig(level=logging.INFO)
|
||||
|
||||
|
||||
@dataclasses.dataclass(frozen=True, slots=True)
|
||||
class Point2D:
|
||||
"""Two-dimensional point. Supports subtracting points."""
|
||||
x: int
|
||||
y: int
|
||||
|
||||
def __sub__(self, other: Point2D) -> Point2D:
|
||||
return Point2D(self.x - other.x, self.y - other.y)
|
||||
|
||||
|
||||
NODE_GROUPS = {
|
||||
"Juggernaut": {"x": -10400, "y": 5200},
|
||||
"Berserker": {"x": -10400, "y": 3700},
|
||||
"Chieftain": {"x": -10400, "y": 2200},
|
||||
"Raider": {"x": 10200, "y": 5200},
|
||||
"Deadeye": {"x": 10200, "y": 2200},
|
||||
"Pathfinder": {"x": 10200, "y": 3700},
|
||||
"Occultist": {"x": -1500, "y": -9850},
|
||||
"Elementalist": {"x": 0, "y": -9850},
|
||||
"Necromancer": {"x": 1500, "y": -9850},
|
||||
"Slayer": {"x": 1500, "y": 9800},
|
||||
"Gladiator": {"x": -1500, "y": 9800},
|
||||
"Champion": {"x": 0, "y": 9800},
|
||||
"Inquisitor": {"x": -10400, "y": -2200},
|
||||
"Hierophant": {"x": -10400, "y": -3700},
|
||||
"Guardian": {"x": -10400, "y": -5200},
|
||||
"Assassin": {"x": 10200, "y": -5200},
|
||||
"Trickster": {"x": 10200, "y": -3700},
|
||||
"Saboteur": {"x": 10200, "y": -2200},
|
||||
"Ascendant": {"x": -7800, "y": 7200},
|
||||
"Juggernaut": Point2D(-10400, 5200),
|
||||
"Berserker": Point2D(-10400, 3700),
|
||||
"Chieftain": Point2D(-10400, 2200),
|
||||
"Raider": Point2D(10200, 5200),
|
||||
"Deadeye": Point2D(10200, 2200),
|
||||
"Pathfinder": Point2D(10200, 3700),
|
||||
"Occultist": Point2D(-1500, -9850),
|
||||
"Elementalist": Point2D(0, -9850),
|
||||
"Necromancer": Point2D(1500, -9850),
|
||||
"Slayer": Point2D(1500, 9800),
|
||||
"Gladiator": Point2D(-1500, 9800),
|
||||
"Champion": Point2D(0, 9800),
|
||||
"Inquisitor": Point2D(-10400, -2200),
|
||||
"Hierophant": Point2D(-10400, -3700),
|
||||
"Guardian": Point2D(-10400, -5200),
|
||||
"Assassin": Point2D(10200, -5200),
|
||||
"Trickster": Point2D(10200, -3700),
|
||||
"Saboteur": Point2D(10200, -2200),
|
||||
"Ascendant": Point2D(-7800, 7200),
|
||||
}
|
||||
ascLocations = {}
|
||||
GroupOffset = {}
|
||||
|
||||
|
||||
def main():
|
||||
with open("data.json") as f:
|
||||
data = json.loads(f.read())
|
||||
for group in data["groups"]:
|
||||
if (
|
||||
data["groups"][group]["nodes"]
|
||||
and "ascendancyName" in data["nodes"][data["groups"][group]["nodes"][0]]
|
||||
):
|
||||
for node in data["groups"][group]["nodes"]:
|
||||
if "isAscendancyStart" in data["nodes"][node]:
|
||||
ascLocations[data["nodes"][data["groups"][group]["nodes"][0]]["ascendancyName"]] = {"x": data["groups"][group]["x"], "y" : data["groups"][group]["y"]}
|
||||
break
|
||||
for group in data["groups"]:
|
||||
if (
|
||||
data["groups"][group]["nodes"]
|
||||
and "ascendancyName" in data["nodes"][data["groups"][group]["nodes"][0]]
|
||||
):
|
||||
asc = data["nodes"][data["groups"][group]["nodes"][0]]["ascendancyName"]
|
||||
GroupOffset[group] = {"x": (ascLocations[asc]["x"] - data["groups"][group]["x"]), "y": (ascLocations[asc]["y"] - data["groups"][group]["y"])}
|
||||
for group in data["groups"]:
|
||||
if (
|
||||
data["groups"][group]["nodes"]
|
||||
and "ascendancyName" in data["nodes"][data["groups"][group]["nodes"][0]]
|
||||
):
|
||||
asc = data["nodes"][data["groups"][group]["nodes"][0]]["ascendancyName"]
|
||||
data["groups"][group]["x"] = NODE_GROUPS[asc]["x"] - GroupOffset[group]["x"]
|
||||
data["groups"][group]["y"] = NODE_GROUPS[asc]["y"] - GroupOffset[group]["y"]
|
||||
with open("data_fixed.json", "w") as o:
|
||||
o.write(json.dumps(data, indent=4))
|
||||
def fix_ascendancy_positions(path: os.PathLike) -> None:
|
||||
"""Normalise the relative positions of ascendancy nodes on the passive skill tree.
|
||||
|
||||
Ascendancy positions in the passive skill tree data we receive from GGG look
|
||||
scrambled, which is why we have to fix them before importing the skill tree in PoB.
|
||||
|
||||
.. warning:: Overwrites the input file in-place.
|
||||
|
||||
:param path: File path to a JSON passive skill tree data file.
|
||||
:return:
|
||||
"""
|
||||
with open(path, "rb") as f:
|
||||
data = json.load(f)
|
||||
ascendancy_groups = [
|
||||
(data["nodes"][group["nodes"][0]]["ascendancyName"], group)
|
||||
for group in data["groups"].values()
|
||||
if "ascendancyName" in data["nodes"][group["nodes"][0]]
|
||||
]
|
||||
ascendancy_starting_point = {
|
||||
ascendancy: Point2D(group["x"], group["y"])
|
||||
for ascendancy, group in ascendancy_groups
|
||||
for node in group["nodes"]
|
||||
if "isAscendancyStart" in data["nodes"][node]
|
||||
}
|
||||
for ascendancy, group in ascendancy_groups:
|
||||
offset = NODE_GROUPS[ascendancy] - ascendancy_starting_point[ascendancy]
|
||||
group["x"] += offset.x
|
||||
group["y"] += offset.y
|
||||
with open(path, "w", encoding="utf-8") as o:
|
||||
json.dump(data, o, indent=4)
|
||||
|
||||
|
||||
def main(root: pathlib.Path) -> None:
|
||||
"""Fix all passive skill tree JSONs found in root directory.
|
||||
|
||||
.. warning: Overwrites all matched files in-place.
|
||||
|
||||
:param root: File path to root directory.
|
||||
:return:
|
||||
"""
|
||||
for file in root.glob("**/data.json"):
|
||||
fix_ascendancy_positions(file)
|
||||
logging.info(f"Found and processed file '{file}'.")
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
main(pathlib.Path("src/TreeData"))
|
||||
|
||||
Reference in New Issue
Block a user