First slice: classes, combat, questing, Deadmines

Co-Authored-By: Claude Opus 5.5 <noreply@anthropic.com>

tools/balance.py

@@ -0,0 +1,99 @@
+#!/usr/bin/env python3
+"""Balance report: solo hunts per class and level, and all-auto Deadmines runs.
+
+    python3 tools/balance.py [solo|dungeon] [runs]
+"""
+
+from __future__ import annotations
+
+import random
+import statistics
+import sys
+from pathlib import Path
+
+sys.path.insert(0, str(Path(__file__).resolve().parent.parent))
+
+import combat  # noqa: E402
+import game  # noqa: E402
+from content import CLASSES, DUNGEONS, MOBS  # noqa: E402
+
+SOLO_TARGETS = {1: "young_wolf", 5: "defias_thug", 10: "riverpaw_gnoll", 15: "riverpaw_brute",
+                19: "defias_highwayman"}
+
+
+def hero(cls: str, level: int, uid: str = "p") -> dict:
+    player = game.new_player(uid, cls.title(), cls, 0)
+    game.dev_level({"config": {}}, player, level, 0)
+    return player
+
+
+def solo(runs: int) -> None:
+    print(f"{'class':9} {'lvl':>3} {'mob':20} {'win':>5} {'secs':>5} {'hp lost':>7}")
+    for cls in CLASSES:
+        for level, mob_id in SOLO_TARGETS.items():
+            wins, times, lost = 0, [], []
+            for seed in range(runs):
+                player = hero(cls, level)
+                rng = random.Random(seed)
+                mob_level = level
+                enemies = [combat.mob_unit("e0", mob_id, mob_level)]
+                fight = combat.new_fight("f", seed, 0, game.player_unit(player, 0, False), enemies, "hunt")
+                combat.advance(fight, 400)
+                me = fight["units"]["p"]
+                if fight["state"] == "won":
+                    wins += 1
+                    times.append(fight["t"])
+                    lost.append(1 - me["hp"] / me["max_hp"])
+            print(f"{cls:9} {level:>3} {MOBS[mob_id]['name']:20} {wins / runs:>5.0%} "
+                  f"{statistics.mean(times) if times else 0:>5.1f} {statistics.mean(lost) if lost else 0:>7.0%}")
+
+
+def dungeon(runs: int, party: list[str], level: int) -> None:
+    pulls = DUNGEONS["deadmines"]["pulls"]
+    results = {p["name"]: {"won": 0, "t": [], "hp": [], "low": [], "mana": []} for p in pulls}
+    for seed in range(runs):
+        rng = random.Random(seed)
+        heroes = [hero(cls, level, f"p{i}") for i, cls in enumerate(party)]
+        for pull in pulls:
+            units = []
+            for h in heroes:
+                units += game.player_unit(h, 0, True)
+            hp_scale, dmg_scale = game.dungeon_scale(len(heroes))
+            enemies = []
+            for index, mob_id in enumerate(pull["mobs"]):
+                low, high = MOBS[mob_id]["lv"]
+                unit = combat.mob_unit(f"e{index}", mob_id, rng.randint(low, high), hp_scale, dmg_scale)
+                unit["swing"]["next"] = 0.8 + 0.4 * index
+                enemies.append(unit)
+            fight = combat.new_fight("f", rng.randrange(2 ** 31), 0, units, enemies, "dungeon", pull["name"],
+                                     (hp_scale, dmg_scale))
+            while fight["state"] == "on":
+                combat.advance(fight, fight["t"] + combat.STEP)
+                tank = fight["units"]["p0"]
+                fight["low"] = min(fight.get("low", 1), tank["hp"] / tank["max_hp"])
+            row = results[pull["name"]]
+            row["low"].append(fight.get("low", 1))
+            row["mana"].append(fight["units"]["p1"]["res"]["cur"] / fight["units"]["p1"]["res"]["max"])
+            if fight["state"] == "won":
+                row["won"] += 1
+                row["t"].append(fight["t"])
+                row["hp"].append(sum(1 for u in combat.players(fight) if u["dead"]))
+    print(f"party: {', '.join(party)} at {level}")
+    print(f"{'pull':20} {'win':>5} {'secs':>5} {'dead':>5} {'tank low':>8} {'heal mana':>9}")
+    for name, row in results.items():
+        print(f"{name:20} {row['won'] / runs:>5.0%} {statistics.mean(row['t']) if row['t'] else 0:>5.0f} "
+              f"{statistics.mean(row['hp']) if row['hp'] else 0:>5.1f} {statistics.mean(row['low']):>8.0%} "
+              f"{statistics.mean(row['mana']):>9.0%}")
+
+
+if __name__ == "__main__":
+    mode = sys.argv[1] if len(sys.argv) > 1 else "solo"
+    runs = int(sys.argv[2]) if len(sys.argv) > 2 else 20
+    if mode == "solo":
+        solo(runs)
+    else:
+        if "perfect" in sys.argv:
+            combat.AUTO_CD, combat.AUTO_DELAY, combat.AUTO_REACT = 1.0, 0.0, 0.25
+            print("-- perfect hands --")
+        dungeon(runs, ["warrior", "priest", "mage", "rogue", "hunter"], 18)
+        dungeon(runs, ["paladin", "druid", "warlock", "shaman", "rogue"], 18)