172 lines
8.3 KiB
Python
172 lines
8.3 KiB
Python
from functools import reduce
|
|
|
|
from eos.saveddata.damagePattern import DamagePattern
|
|
|
|
from gui.utils.numberFormatter import formatAmount
|
|
|
|
|
|
tankTypes = ("shield", "armor", "hull")
|
|
damageTypes = ("em", "thermal", "kinetic", "explosive")
|
|
damagePatterns = [DamagePattern.oneType(damageType) for damageType in damageTypes]
|
|
damageTypeResonanceNames = [damageType.capitalize() + "DamageResonance" for damageType in damageTypes]
|
|
resonanceNames = {"shield": ["shield" + s for s in damageTypeResonanceNames],
|
|
"armor": ["armor" + s for s in damageTypeResonanceNames],
|
|
"hull": [s[0].lower() + s[1:] for s in damageTypeResonanceNames]}
|
|
|
|
|
|
def firepowerSection(fit):
|
|
""" Returns the text of the firepower section"""
|
|
firepower = [fit.totalDPS, fit.weaponDPS, fit.droneDPS, fit.totalVolley]
|
|
firepowerStr = [formatAmount(dps, 3, 0, 0) for dps in firepower]
|
|
showWeaponAndDroneDps = (fit.weaponDPS > 0) and (fit.droneDPS > 0)
|
|
if sum(firepower) == 0:
|
|
return ""
|
|
return "DPS: {} (".format(firepowerStr[0]) + \
|
|
("Weapon: {}, Drone: {}, ".format(*firepowerStr[1:3]) if showWeaponAndDroneDps else "") + \
|
|
("Volley: {})\n".format(firepowerStr[3]))
|
|
|
|
|
|
def tankSection(fit):
|
|
""" Returns the text of the tank section"""
|
|
ehp = [fit.ehp[tank] for tank in tankTypes] if fit.ehp is not None else [0, 0, 0]
|
|
ehp.append(sum(ehp))
|
|
ehpStr = [formatAmount(ehpVal, 3, 0, 9) for ehpVal in ehp]
|
|
resists = {tankType: [1 - fit.ship.getModifiedItemAttr(s) for s in resonanceNames[tankType]] for tankType in tankTypes}
|
|
ehpAgainstDamageType = [sum(pattern.calculateEhp(fit).values()) for pattern in damagePatterns]
|
|
ehpAgainstDamageTypeStr = [formatAmount(ehpVal, 3, 0, 9) for ehpVal in ehpAgainstDamageType]
|
|
|
|
return \
|
|
" {:>7} {:>7} {:>7} {:>7} {:>7}\n".format("TOTAL", "EM", "THERM", "KIN", "EXP") + \
|
|
"EHP {:>7} {:>7} {:>7} {:>7} {:>7}\n".format(ehpStr[3], *ehpAgainstDamageTypeStr) + \
|
|
"Shield {:>7} {:>7.0%} {:>7.0%} {:>7.0%} {:>7.0%}\n".format(ehpStr[0], *resists["shield"]) + \
|
|
"Armor {:>7} {:>7.0%} {:>7.0%} {:>7.0%} {:>7.0%}\n".format(ehpStr[1], *resists["armor"]) + \
|
|
"Hull {:>7} {:>7.0%} {:>7.0%} {:>7.0%} {:>7.0%}\n".format(ehpStr[2], *resists["hull"])
|
|
|
|
|
|
def repsSection(fit):
|
|
""" Returns the text of the repairs section"""
|
|
selfRep = [fit.effectiveTank[tankType + "Repair"] for tankType in tankTypes]
|
|
sustainRep = [fit.effectiveSustainableTank[tankType + "Repair"] for tankType in tankTypes]
|
|
remoteRep = [fit.remoteReps[tankType.capitalize()] for tankType in tankTypes]
|
|
shieldRegen = [fit.effectiveSustainableTank["passiveShield"], 0, 0]
|
|
shieldRechargeModuleMultipliers = [module.item.attributes["shieldRechargeRateMultiplier"].value for module in
|
|
fit.modules if
|
|
module.item and "shieldRechargeRateMultiplier" in module.item.attributes]
|
|
shieldRechargeMultiplierByModules = reduce(lambda x, y: x * y, shieldRechargeModuleMultipliers, 1)
|
|
if shieldRechargeMultiplierByModules >= 0.9: # If the total affect of modules on the shield recharge is negative or insignificant, we don't care about it
|
|
shieldRegen[0] = 0
|
|
totalRep = list(zip(selfRep, remoteRep, shieldRegen))
|
|
totalRep = list(map(sum, totalRep))
|
|
|
|
selfRep.append(sum(selfRep))
|
|
sustainRep.append(sum(sustainRep))
|
|
remoteRep.append(sum(remoteRep))
|
|
shieldRegen.append(sum(shieldRegen))
|
|
totalRep.append(sum(totalRep))
|
|
|
|
totalSelfRep = selfRep[-1]
|
|
totalRemoteRep = remoteRep[-1]
|
|
totalShieldRegen = shieldRegen[-1]
|
|
|
|
text = ""
|
|
|
|
if sum(totalRep) > 0: # Most commonly, there are no reps at all; then we skip this section
|
|
singleTypeRep = None
|
|
singleTypeRepName = None
|
|
if totalRemoteRep == 0 and totalShieldRegen == 0: # Only self rep
|
|
singleTypeRep = selfRep[:-1]
|
|
singleTypeRepName = "Self"
|
|
if totalSelfRep == 0 and totalShieldRegen == 0: # Only remote rep
|
|
singleTypeRep = remoteRep[:-1]
|
|
singleTypeRepName = "Remote"
|
|
if totalSelfRep == 0 and totalRemoteRep == 0: # Only shield regen
|
|
singleTypeRep = shieldRegen[:-1]
|
|
singleTypeRepName = "Regen"
|
|
if singleTypeRep and sum(
|
|
x > 0 for x in singleTypeRep) == 1: # Only one type of reps and only one tank type is repaired
|
|
index = next(i for i, v in enumerate(singleTypeRep) if v > 0)
|
|
if singleTypeRepName == "Regen":
|
|
text += "Shield regeneration: {} EHP/s".format(formatAmount(singleTypeRep[index], 3, 0, 9))
|
|
else:
|
|
text += "{} {} repair: {} EHP/s".format(singleTypeRepName, tankTypes[index],
|
|
formatAmount(singleTypeRep[index], 3, 0, 9))
|
|
if (singleTypeRepName == "Self") and (sustainRep[index] != singleTypeRep[index]):
|
|
text += " (Sustained: {} EHP/s)".format(formatAmount(sustainRep[index], 3, 0, 9))
|
|
text += "\n"
|
|
else: # Otherwise show a table
|
|
selfRepStr = [formatAmount(rep, 3, 0, 9) for rep in selfRep]
|
|
sustainRepStr = [formatAmount(rep, 3, 0, 9) for rep in sustainRep]
|
|
remoteRepStr = [formatAmount(rep, 3, 0, 9) for rep in remoteRep]
|
|
shieldRegenStr = [formatAmount(rep, 3, 0, 9) if rep != 0 else "" for rep in shieldRegen]
|
|
totalRepStr = [formatAmount(rep, 3, 0, 9) for rep in totalRep]
|
|
|
|
header = "REPS "
|
|
lines = [
|
|
"Shield ",
|
|
"Armor ",
|
|
"Hull ",
|
|
"Total "
|
|
]
|
|
|
|
showSelfRepColumn = totalSelfRep > 0
|
|
showSustainRepColumn = sustainRep != selfRep
|
|
showRemoteRepColumn = totalRemoteRep > 0
|
|
showShieldRegenColumn = totalShieldRegen > 0
|
|
|
|
if showSelfRepColumn + showSustainRepColumn + showRemoteRepColumn + showShieldRegenColumn > 1:
|
|
header += "{:>7} ".format("TOTAL")
|
|
lines = [line + "{:>7} ".format(rep) for line, rep in zip(lines, totalRepStr)]
|
|
if showSelfRepColumn:
|
|
header += "{:>7} ".format("SELF")
|
|
lines = [line + "{:>7} ".format(rep) for line, rep in zip(lines, selfRepStr)]
|
|
if showSustainRepColumn:
|
|
header += "{:>7} ".format("SUST")
|
|
lines = [line + "{:>7} ".format(rep) for line, rep in zip(lines, sustainRepStr)]
|
|
if showRemoteRepColumn:
|
|
header += "{:>7} ".format("REMOTE")
|
|
lines = [line + "{:>7} ".format(rep) for line, rep in zip(lines, remoteRepStr)]
|
|
if showShieldRegenColumn:
|
|
header += "{:>7} ".format("REGEN")
|
|
lines = [line + "{:>7} ".format(rep) for line, rep in zip(lines, shieldRegenStr)]
|
|
|
|
text += header + "\n"
|
|
repsByTank = zip(totalRep, selfRep, sustainRep, remoteRep, shieldRegen)
|
|
for line in lines:
|
|
reps = next(repsByTank)
|
|
if sum(reps) > 0:
|
|
text += line + "\n"
|
|
return text
|
|
|
|
|
|
def miscSection(fit):
|
|
text = ""
|
|
text += "Speed: {} m/s\n".format(formatAmount(fit.maxSpeed, 3, 0, 0))
|
|
text += "Signature: {} m\n".format(formatAmount(fit.ship.getModifiedItemAttr("signatureRadius"), 3, 0, 9))
|
|
|
|
text += "Capacitor: {} GJ".format(formatAmount(fit.ship.getModifiedItemAttr("capacitorCapacity"), 3, 0, 9))
|
|
capState = fit.capState
|
|
if fit.capStable:
|
|
text += " (Stable at {0:.0f}%)".format(capState)
|
|
else:
|
|
text += " (Lasts {})".format("%ds" % capState if capState <= 60 else "%dm%ds" % divmod(capState, 60))
|
|
text += "\n"
|
|
|
|
text += "Targeting range: {} km\n".format(formatAmount(fit.maxTargetRange / 1000, 3, 0, 0))
|
|
text += "Scan resolution: {0:.0f} mm\n".format(fit.ship.getModifiedItemAttr("scanResolution"))
|
|
text += "Sensor strength: {}\n".format(formatAmount(fit.scanStrength, 3, 0, 0))
|
|
|
|
return text
|
|
|
|
|
|
def statsExportText(fit):
|
|
""" Returns the text of the stats export of the given fit"""
|
|
sections = filter(None, (firepowerSection(fit), # Prune empty sections
|
|
tankSection(fit),
|
|
repsSection(fit),
|
|
miscSection(fit)))
|
|
|
|
text = "{} ({})\n".format(fit.name, fit.ship.name) + "\n"
|
|
text += "\n".join(sections)
|
|
|
|
return text
|