diff options
Diffstat (limited to 'main.py')
| -rw-r--r-- | main.py | 75 |
1 files changed, 62 insertions, 13 deletions
@@ -1,4 +1,6 @@ import math + +from numpy._core import multiarray from character import CharacterPlan from distribution import Distribution from game import Difficulty, Roll, d @@ -62,22 +64,27 @@ def main(): show_stat("Treat Wounds 20 +1", treat) class StatsTest: - def __init__(self) -> None: + def __init__(self, difficulty = Difficulty.Moderate) -> None: # self.char_js = pb2e.download_character(pfb_id=160180) self.char_js = pb2e.open_character("quizrel2") self.char_plan = CharacterPlan(self.char_js) - self.creature_plan = CreaturePlan(default_difficulty=Difficulty.Moderate) + self.creature_plan = CreaturePlan(default_difficulty=difficulty) # self.creature_plan["reflex"] = Difficulty.High # self.creature_plan["ac"] = Difficulty.High self.char = self.char_plan.build(1) def add_weapons(self): - falcata = Weapon("falcata", "martial", "sword", 8, fatal=12) - sword = Weapon("pick", "martial", "pick", 6, fatal=10) - rapier = Weapon("rapier", "martial", "sword", 6, finesse=True, deadly=8) - urumi = Weapon("urumi", "martial", "flail", 6, deadly=10) - sword2 = Weapon("tricky pick", "martial", "pick", 6, fatal=10, backstabber=True) - self.char.weapons.extend([sword, falcata, rapier, sword2, urumi]) + wps = [Weapon("falcata", "martial", "sword", 8, fatal=12), + Weapon("trident", "martial", "spear", 8), + Weapon("pick", "martial", "pick", 6, fatal=10), + Weapon("rapier", "martial", "sword", 6, finesse=True, deadly=8), + Weapon("urumi", "martial", "flail", 6, deadly=10), + Weapon("tricky pick", "martial", "pick", 6, fatal=10, backstabber=True), + Weapon("shortsword", "simple", "sword", 6, agile=True), + Weapon("gauntlet", "simple", "brawling", 4, agile=True), + Weapon("light pick", "martial", "pick", 6, agile=True, fatal=8) + ] + self.char.weapons.extend(wps) def get_weapon(self, name): for wp in self.char.weapons: @@ -110,7 +117,7 @@ def make_check(name, bonus, dc): if isinstance(bonus, Distribution): bonus = bonus.expected() print(f"{name} {bonus:+} vs DC {dc}:", - Roll.short_str(roll), f"Total: {dc-bonus-10:+}") + Roll.short_str(roll), f"Total: {10+bonus-dc:+}") return roll def demoralize_test(): @@ -139,7 +146,7 @@ def demoralize_test(): frightened = roll.apply(lambda x: x if x >= Roll.Success else 0).expected() print("Frightened:", frightened) -def test3(): +def weapon_damage(): t = StatsTest() def calc_damage(level): t.set_level(level) @@ -152,15 +159,55 @@ def test3(): res = dict() for weapon in t.char.weapons: - attack = Attack(t.char, t.creature, weapon) + attack = Attack(t.char, t.creature, weapon, sneak=False) res[weapon.name] = attack.strike().expected() # res["hp"] = creature["hp"] return res # calc_damage(20) + levelup_view("Weapons Comparison", "Expected Damage", calc_damage) + +def twin_takedown_test(): + t = StatsTest() + # t.creature_plan["ac"] = Difficulty.Extreme + def calc_damage(level): + t.set_level(level) + res = dict() + wp1 = t.get_weapon("tricky pick") + wp2 = t.get_weapon("gauntlet") + + # Twin takedown + attack = Attack(t.char, t.creature, wp1, sneak=True) + dmg = attack.strike().expected() + attack.previous_attacks += 1 + attack.weapon = wp2 + dmg += attack.strike().expected() + res["Twin Takedown"] = dmg + + # Double slice + attack = Attack(t.char, t.creature, wp1, sneak=True) + dmg = attack.strike().expected() + attack.sneak = False + attack.weapon = wp2 + dmg += attack.strike().expected() + res["Double Slice"] = dmg + + return res levelup_view("Weapons Comparison", "Expected Damage", calc_damage) +def multiple_attacks(): + t = StatsTest() + t.set_level(5) + wp = t.get_weapon("Shortsword") + wp.agile = False + # t.creature["ac"] -= 1 + attack = Attack(t.char, t.creature, wp, sneak=True) + print(attack) + for _ in range(4): + attack.previous_attacks += 1 + print(attack) + def levelup_view(title, ylabel, fn): @@ -192,5 +239,7 @@ def levelup_view(title, ylabel, fn): if __name__ == '__main__': - # test4() - demoralize_test() + # weapon_damage() + twin_takedown_test() + # multiple_attacks() + # demoralize_test() |