mirror of
https://github.com/mhwikicn/mhdos-armor-set-searcher
synced 2025-12-06 13:39:03 +08:00
114 lines
3.4 KiB
Python
114 lines
3.4 KiB
Python
import json
|
|
|
|
def reorder_resistances(raw: list) -> list:
|
|
return [
|
|
raw[0],
|
|
raw[1],
|
|
raw[3],
|
|
raw[2],
|
|
raw[4],
|
|
]
|
|
|
|
def map_skills_map(raw: dict, skill_ids: dict) -> dict:
|
|
skills = {}
|
|
for s in raw:
|
|
n = s["k"]
|
|
id = skill_ids[n]
|
|
try:
|
|
v = int(s["q"])
|
|
except:
|
|
v = 0 # torso up
|
|
skills[id] = v
|
|
return skills
|
|
|
|
def map_deco(raw: dict, skill_ids: dict) -> dict:
|
|
return {
|
|
"name": raw["name"],
|
|
"rarity": raw["rarity"],
|
|
"requiredSlots": raw["slots"],
|
|
"skills": map_skills_map(raw["skills"], skill_ids),
|
|
}
|
|
|
|
def process_armor_piece(piece, cat_index, skill_ids):
|
|
name = piece["name"]
|
|
if name == "None":
|
|
return []
|
|
|
|
# skip if unacquirable
|
|
if piece.get("acquire", 0) or piece.get("dev", 0):
|
|
return []
|
|
|
|
# skip if only female
|
|
if piece.get("sex", "") == "Female":
|
|
return []
|
|
|
|
# get type
|
|
x = piece.get("hunterClass", "")
|
|
type = 1 if x == "Blademaster" else 2 if x == "Gunner" else 0
|
|
|
|
# get unique slots and their first occurrence positions
|
|
slots = piece["slots"]
|
|
defe = piece["defense"]
|
|
skills = piece.get("skills", [])
|
|
|
|
# create a dictionary to track first occurrence of each slot value
|
|
slot_variants = {}
|
|
for pos, slot_value in enumerate(slots):
|
|
if slot_value not in slot_variants:
|
|
slot_variants[slot_value] = pos + 1 # LV starts from 1
|
|
|
|
# generate variants for each unique slot value
|
|
variants = []
|
|
for slot_value, lv_pos in slot_variants.items():
|
|
variant_name = f"{name}(LV{lv_pos}+)"
|
|
variant_defense_base = defe[lv_pos - 1] # positions are 0-based in array
|
|
|
|
modeled_piece = {
|
|
"category": cat_index,
|
|
"name": variant_name,
|
|
"rarity": piece["rarity"],
|
|
"skills": map_skills_map(skills, skill_ids),
|
|
"slots": slot_value,
|
|
"defense": {
|
|
"base": variant_defense_base,
|
|
"max": defe[-1],
|
|
},
|
|
"type": type,
|
|
"resistance": reorder_resistances(piece["resistances"]),
|
|
}
|
|
variants.append(modeled_piece)
|
|
|
|
return variants
|
|
|
|
if __name__ == "__main__":
|
|
# read input
|
|
with open("./raw-decos.json") as f:
|
|
decos = list(json.loads(f.read()))
|
|
with open("./raw-armor.json") as f:
|
|
armor = list(json.loads(f.read()))
|
|
with open("../data/skill-names.json") as f:
|
|
skill_names = dict(json.loads(f.read()))
|
|
|
|
# reverse skill names
|
|
skill_ids = {v: int(k) for k, v in skill_names.items()}
|
|
|
|
# decorations
|
|
modeled_decos = [map_deco(x, skill_ids) for x in decos]
|
|
|
|
# iterate over armor categories
|
|
pieces_per_category = []
|
|
for (cat_index, armor_category) in enumerate(armor):
|
|
pieces_of_cat = []
|
|
# iterate over pieces of that category
|
|
for piece in armor_category["armor"]:
|
|
variants = process_armor_piece(piece, cat_index, skill_ids)
|
|
pieces_of_cat.extend(variants)
|
|
pieces_per_category.append(pieces_of_cat)
|
|
|
|
# save files
|
|
for (i, cat_name) in enumerate(["head", "chest", "arms", "waist", "legs"]):
|
|
with open(f"../data/{cat_name}.json", "w") as f:
|
|
f.write(json.dumps(pieces_per_category[i], indent=4, sort_keys=True))
|
|
with open("../data/decorations.json", "w") as f:
|
|
f.write(json.dumps(modeled_decos, indent=4))
|