damage refactor, new averages

bin/_pathfix.py

@@ -9,3 +9,6 @@ from os.path import dirname, join, abspath
 bin_path = dirname(__file__)
 project_path = abspath(join(bin_path, ".."))
 sys.path.insert(0, project_path)
+
+db_path = join(project_path, "db", "mh4u.db")
+motion_values_path = join(project_path, "db", "motion_values.json")

bin/mhdamage.py

@@ -0,0 +1,107 @@
+#!/usr/bin/env python
+
+import sys
+import os
+
+import _pathfix
+
+from mhapi.db import MHDB
+from mhapi.damage import MotionValueDB, WeaponMonsterDamage
+
+
+def percent_change(a, b):
+    if a == 0:
+        return b
+    return (100.0 * (b-a) / a)
+
+
+if __name__ == '__main__':
+    if len(sys.argv) < 4:
+        print "Usage: %s 'monster name' 'weapon name'+" % sys.argv[0]
+        sys.exit(os.EX_USAGE)
+
+    sharp_plus = bool(int(sys.argv[1]))
+    monster_name = sys.argv[2]
+    weapon_names = sys.argv[3:]
+
+    db = MHDB(_pathfix.db_path)
+    motiondb = MotionValueDB(_pathfix.motion_values_path)
+
+    monster = db.get_monster_by_name(monster_name)
+    if not monster:
+        raise ValueError("Monster '%s' not found" % monster_name)
+    monster_damage = db.get_monster_damage(monster["_id"])
+    weapons = []
+    for name in weapon_names:
+        weapon = db.get_weapon_by_name(name)
+        if not weapon:
+            raise ValueError("Weapon '%s' not found" % name)
+        weapons.append(weapon)
+
+    monster_breaks = db.get_monster_breaks(monster["_id"])
+    weapon_type = weapons[0]["wtype"]
+    motion = motiondb[weapon_type].average
+    print "Weapon Type: %s" % weapon_type
+    print "Average Motion: %0.1f" % motion
+    print "Monster Breaks: %s" % ", ".join(monster_breaks)
+    weapon_damage_map = dict()
+    for name, row in zip(weapon_names, weapons):
+        row_type = row["wtype"]
+        if row_type != weapon_type:
+            raise ValueError("Weapon '%s' is different type" % name)
+        try:
+            weapon_damage_map[name] = WeaponMonsterDamage(row,
+                                            monster, monster_damage,
+                                            motion, sharp_plus,
+                                            monster_breaks)
+        except ValueError as e:
+            print str(e)
+            sys.exit(1)
+
+    damage_map_base = weapon_damage_map[weapon_names[0]]
+    parts = damage_map_base.parts
+
+    for part in parts:
+        tdiffs = [percent_change(
+                    damage_map_base[part].total,
+                    weapon_damage_map[w][part].total
+                  )
+                  for w in weapon_names[1:]]
+        ediffs = [percent_change(
+                    damage_map_base[part].element,
+                    weapon_damage_map[w][part].element
+                  )
+                  for w in weapon_names[1:]]
+        bdiffs = [percent_change(
+                    damage_map_base[part].break_diff(),
+                    weapon_damage_map[w][part].break_diff()
+                  )
+                  for w in weapon_names[1:]]
+        tdiff_s = ",".join("%+0.1f%%" % i for i in tdiffs)
+        ediff_s = ",".join("%+0.1f%%" % i for i in ediffs)
+        bdiff_s = ",".join("%+0.1f%%" % i for i in bdiffs)
+        damage = damage_map_base[part]
+        print "%22s%s h%02d %0.2f (%s) h%02d %0.2f (%s) %+0.2f (%s)" \
+            % (part, "*" if damage.is_breakable() else " ",
+               damage.hitbox,
+               damage.total,
+               tdiff_s,
+               damage.ehitbox,
+               damage.element,
+               ediff_s,
+               damage.break_diff(),
+               bdiff_s)
+
+    print "            --------------------"
+
+    for avg_type in "uniform raw weakpart_raw element weakpart_element break_raw break_element break_only".split():
+        base = damage_map_base.averages[avg_type]
+        diffs = [percent_change(
+                    base,
+                    weapon_damage_map[w].averages[avg_type]
+                 )
+                 for w in weapon_names[1:]]
+
+        diff_s = ",".join("%+0.1f%%" % i for i in diffs)
+
+        print "%22s %0.2f (%s)" % (avg_type, base, diff_s)

bin/mhrewards.py

@@ -26,10 +26,7 @@ if __name__ == '__main__':
     out = get_utf8_writer(sys.stdout)
     err_out = get_utf8_writer(sys.stderr)
 
-    # TODO: doesn't work if script is symlinked
-    db_path = os.path.dirname(sys.argv[0])
-    db_path = os.path.join(db_path, "..", "db", "mh4u.db")
-    db = MHDB(db_path)
+    db = MHDB(_pathfix.db_path)
 
     item_row = rewards.find_item(db, item_name, err_out)
     if item_row is None: