Add fighters to dps vs range graph

This commit is contained in:
DarkPhoenix
2019-07-04 14:33:28 +03:00
parent 185cf4f625
commit c4f225003a
2 changed files with 17 additions and 1 deletions

View File

@@ -151,6 +151,10 @@ class Fighter(HandledItem, HandledCharge, ItemAttrShortcut, ChargeAttrShortcut):
def abilities(self):
return self.__abilities or []
@property
def abilityMap(self):
return {a.effectID: a for a in self.abilities}
@property
def charge(self):
return self.__charge

View File

@@ -117,7 +117,19 @@ class FitDamageStatsGraph(FitGraph):
tgtSpeed=miscInputMap['tgtSpeed'],
tgtAngle=miscInputMap['tgtAngle'],
tgtSigRadius=tgtSigRad)
for fighter in fit.fighters:
if not fighter.isDealingDamage():
continue
abilityMap = fighter.abilityMap
for effectID, abilityDps in fighter.getDpsPerEffect().items():
ability = abilityMap[effectID]
totalDps += abilityDps.total * getFighterAbilityMult(
fighter=fighter,
ability=ability,
fit=fit,
distance=distance,
tgtSpeed=miscInputMap['tgtSpeed'],
tgtSigRadius=tgtSigRad)
xs.append(distance)
ys.append(totalDps)
return xs, ys