267 lines
11 KiB
Python
267 lines
11 KiB
Python
# ===============================================================================
|
|
# Copyright (C) 2010 Diego Duclos
|
|
#
|
|
# This file is part of eos.
|
|
#
|
|
# eos is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU Lesser General Public License as published by
|
|
# the Free Software Foundation, either version 2 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# eos is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU Lesser General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU Lesser General Public License
|
|
# along with eos. If not, see <http://www.gnu.org/licenses/>.
|
|
# ===============================================================================
|
|
|
|
import re
|
|
from collections import OrderedDict
|
|
|
|
import eos.db
|
|
|
|
|
|
BUILTINS = OrderedDict([
|
|
(1, ('Uniform', 25, 25, 25, 25)),
|
|
(2, ('[Generic]EM', 1, 0, 0, 0)),
|
|
(3, ('[Generic]Thermal', 0, 1, 0, 0)),
|
|
(4, ('[Generic]Kinetic', 0, 0, 1, 0)),
|
|
(5, ('[Generic]Explosive', 0, 0, 0, 1)),
|
|
(6, ('[Bombs]Electron Bomb', 6400, 0, 0, 0)),
|
|
(7, ('[Bombs]Scorch Bomb', 0, 6400, 0, 0)),
|
|
(8, ('[Bombs]Concussion Bomb', 0, 0, 6400, 0)),
|
|
(9, ('[Bombs]Shrapnel Bomb', 0, 0, 0, 6400)),
|
|
(10, ('[Frequency Crystals][T2] Conflagration', 7.7, 7.7, 0, 0)),
|
|
(11, ('[Frequency Crystals][T2] Scorch', 9, 2, 0, 0)),
|
|
(12, ('[Frequency Crystals][T2] Gleam', 7, 7, 0, 0)),
|
|
(13, ('[Frequency Crystals][T2] Aurora', 5, 3, 0, 0)),
|
|
(14, ('[Frequency Crystals]Multifrequency', 7, 5, 0, 0)),
|
|
(15, ('[Frequency Crystals]Gamma', 7, 4, 0, 0)),
|
|
(16, ('[Frequency Crystals]Xray', 6, 4, 0, 0)),
|
|
(17, ('[Frequency Crystals]Ultraviolet', 6, 3, 0, 0)),
|
|
(18, ('[Frequency Crystals]Standard', 5, 3, 0, 0)),
|
|
(19, ('[Frequency Crystals]Infrared', 5, 2, 0, 0)),
|
|
(20, ('[Frequency Crystals]Microwave', 4, 2, 0, 0)),
|
|
(21, ('[Frequency Crystals]Radio', 5, 0, 0, 0)),
|
|
(22, ('[Hybrid Charges][T2] Void', 0, 7.7, 7.7, 0)),
|
|
(23, ('[Hybrid Charges][T2] Null', 0, 6, 5, 0)),
|
|
(24, ('[Hybrid Charges][T2] Javelin', 0, 8, 6, 0)),
|
|
(25, ('[Hybrid Charges][T2] Spike', 0, 4, 4, 0)),
|
|
(26, ('[Hybrid Charges]Antimatter', 0, 5, 7, 0)),
|
|
(27, ('[Hybrid Charges]Plutonium', 0, 5, 6, 0)),
|
|
(28, ('[Hybrid Charges]Uranium', 0, 4, 6, 0)),
|
|
(29, ('[Hybrid Charges]Thorium', 0, 4, 5, 0)),
|
|
(30, ('[Hybrid Charges]Lead', 0, 3, 5, 0)),
|
|
(31, ('[Hybrid Charges]Iridium', 0, 3, 4, 0)),
|
|
(32, ('[Hybrid Charges]Tungsten', 0, 2, 4, 0)),
|
|
(33, ('[Hybrid Charges]Iron', 0, 2, 3, 0)),
|
|
(34, ('[Missiles]Mjolnir', 1, 0, 0, 0)),
|
|
(35, ('[Missiles]Inferno', 0, 1, 0, 0)),
|
|
(36, ('[Missiles]Scourge', 0, 0, 1, 0)),
|
|
(37, ('[Missiles]Nova', 0, 0, 0, 1)),
|
|
(38, ('[Missiles][Structure] Standup Missile', 1, 1, 1, 1)),
|
|
(39, ('[Projectile Ammo][T2] Hail', 0, 0, 3.3, 12.1)),
|
|
(40, ('[Projectile Ammo][T2] Barrage', 0, 0, 5, 6)),
|
|
(41, ('[Projectile Ammo][T2] Quake', 0, 0, 5, 9)),
|
|
(42, ('[Projectile Ammo][T2] Tremor', 0, 0, 3, 5)),
|
|
(43, ('[Projectile Ammo]EMP', 9, 0, 1, 2)),
|
|
(44, ('[Projectile Ammo]Phased Plasma', 0, 10, 2, 0)),
|
|
(45, ('[Projectile Ammo]Fusion', 0, 0, 2, 10)),
|
|
(46, ('[Projectile Ammo]Depleted Uranium', 0, 3, 2, 3)),
|
|
(47, ('[Projectile Ammo]Titanium Sabot', 0, 0, 6, 2)),
|
|
(48, ('[Projectile Ammo]Proton', 3, 0, 2, 0)),
|
|
(49, ('[Projectile Ammo]Carbonized Lead', 0, 0, 4, 1)),
|
|
(50, ('[Projectile Ammo]Nuclear', 0, 0, 1, 4)),
|
|
# Different sizes of plasma do different damage, the values here are
|
|
# average of proportions across sizes
|
|
(51, ('[Exotic Plasma][T2] Occult', 0, 55863, 0, 44137)),
|
|
(52, ('[Exotic Plasma][T2] Mystic', 0, 66319, 0, 33681)),
|
|
(53, ('[Exotic Plasma]Tetryon', 0, 69208, 0, 30792)),
|
|
(54, ('[Exotic Plasma]Baryon', 0, 59737, 0, 40263)),
|
|
(55, ('[Exotic Plasma]Meson', 0, 60519, 0, 39481)),
|
|
(56, ('[NPC][Asteroid] Angel Cartel', 1838, 562, 2215, 3838)),
|
|
(57, ('[NPC][Asteroid] Blood Raiders', 5067, 4214, 0, 0)),
|
|
(58, ('[NPC][Asteroid] Guristas', 0, 1828, 7413, 0)),
|
|
(59, ('[NPC][Asteroid] Rogue Drone', 394, 666, 1090, 1687)),
|
|
(60, ('[NPC][Asteroid] Sanshas Nation', 5586, 4112, 0, 0)),
|
|
(61, ('[NPC][Asteroid] Serpentis', 0, 5373, 4813, 0)),
|
|
(62, ('[NPC][Deadspace] Angel Cartel', 369, 533, 1395, 3302)),
|
|
(63, ('[NPC][Deadspace] Blood Raiders', 6040, 5052, 10, 15)),
|
|
(64, ('[NPC][Deadspace] Guristas', 0, 1531, 9680, 0)),
|
|
(65, ('[NPC][Deadspace] Rogue Drone', 276, 1071, 1069, 871)),
|
|
(66, ('[NPC][Deadspace] Sanshas Nation', 3009, 2237, 0, 0)),
|
|
(67, ('[NPC][Deadspace] Serpentis', 0, 3110, 1929, 0)),
|
|
(68, ('[NPC][Mission] Amarr Empire', 4464, 3546, 97, 0)),
|
|
(69, ('[NPC][Mission] Caldari State', 0, 2139, 4867, 0)),
|
|
(70, ('[NPC][Mission] CONCORD', 336, 134, 212, 412)),
|
|
(71, ('[NPC][Mission] Gallente Federation', 9, 3712, 2758, 0)),
|
|
(72, ('[NPC][Mission] Khanid', 612, 483, 43, 6)),
|
|
(73, ('[NPC][Mission] Minmatar Republic', 1024, 388, 1655, 4285)),
|
|
(74, ('[NPC][Mission] Mordus Legion', 25, 262, 625, 0)),
|
|
(75, ('[NPC][Mission] Thukker', 0, 52, 10, 79)),
|
|
(76, ('[NPC][Burner] Cruor (Blood Raiders)', 90, 90, 0, 0)),
|
|
(77, ('[NPC][Burner] Dramiel (Angel)', 55, 0, 20, 96)),
|
|
(78, ('[NPC][Burner] Daredevil (Serpentis)', 0, 110, 154, 0)),
|
|
(79, ('[NPC][Burner] Succubus (Sanshas Nation)', 135, 30, 0, 0)),
|
|
(80, ('[NPC][Burner] Worm (Guristas)', 0, 0, 228, 0)),
|
|
(81, ('[NPC][Burner] Enyo', 0, 147, 147, 0)),
|
|
(82, ('[NPC][Burner] Hawk', 0, 0, 247, 0)),
|
|
(83, ('[NPC][Burner] Jaguar', 36, 0, 50, 182)),
|
|
(84, ('[NPC][Burner] Vengeance', 232, 0, 0, 0)),
|
|
(85, ('[NPC][Burner] Ashimmu (Blood Raiders)', 260, 100, 0, 0)),
|
|
(86, ('[NPC][Burner] Talos', 0, 413, 413, 0)),
|
|
(87, ('[NPC][Burner] Sentinel', 0, 75, 0, 90)),
|
|
(88, ('[NPC][Other] Sleepers', 1472, 1472, 1384, 1384)),
|
|
(89, ('[NPC][Other] Sansha Incursion', 1682, 1347, 3678, 3678))])
|
|
|
|
|
|
class DamagePattern:
|
|
|
|
DAMAGE_TYPES = ('em', 'thermal', 'kinetic', 'explosive')
|
|
_builtins = None
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
self.update(*args, **kwargs)
|
|
|
|
def update(self, emAmount=25, thermalAmount=25, kineticAmount=25, explosiveAmount=25):
|
|
self.emAmount = emAmount
|
|
self.thermalAmount = thermalAmount
|
|
self.kineticAmount = kineticAmount
|
|
self.explosiveAmount = explosiveAmount
|
|
|
|
@classmethod
|
|
def getBuiltins(cls):
|
|
if cls._builtins is None:
|
|
cls._builtins = OrderedDict()
|
|
for id, (name, em, therm, kin, explo) in BUILTINS.items():
|
|
pattern = DamagePattern(emAmount=em, thermalAmount=therm, kineticAmount=kin, explosiveAmount=explo)
|
|
pattern.ID = id
|
|
pattern.name = name
|
|
cls._builtins[id] = pattern
|
|
return list(cls._builtins.values())
|
|
|
|
def calculateEhp(self, fit):
|
|
ehp = {}
|
|
for (type, attr) in (('shield', 'shieldCapacity'), ('armor', 'armorHP'), ('hull', 'hp')):
|
|
rawCapacity = fit.ship.getModifiedItemAttr(attr)
|
|
ehp[type] = self.effectivify(fit, rawCapacity, type)
|
|
|
|
return ehp
|
|
|
|
def calculateEffectiveTank(self, fit, tankInfo):
|
|
typeMap = {
|
|
"passiveShield": "shield",
|
|
"shieldRepair": "shield",
|
|
"armorRepair": "armor",
|
|
"armorRepairPreSpool": "armor",
|
|
"armorRepairFullSpool": "armor",
|
|
"hullRepair": "hull"}
|
|
ereps = {}
|
|
for field in tankInfo:
|
|
if field in typeMap:
|
|
ereps[field] = self.effectivify(fit, tankInfo[field], typeMap[field])
|
|
return ereps
|
|
|
|
def effectivify(self, fit, amount, type):
|
|
type = type if type != "hull" else ""
|
|
totalDamage = sum((self.emAmount, self.thermalAmount, self.kineticAmount, self.explosiveAmount))
|
|
specificDivider = 0
|
|
for damageType in self.DAMAGE_TYPES:
|
|
# Compose an attribute name, then make sure the first letter is NOT capitalized
|
|
attrName = "%s%sDamageResonance" % (type, damageType.capitalize())
|
|
attrName = attrName[0].lower() + attrName[1:]
|
|
|
|
resonance = fit.ship.getModifiedItemAttr(attrName)
|
|
damage = getattr(self, "%sAmount" % damageType)
|
|
|
|
specificDivider += damage / float(totalDamage or 1) * resonance
|
|
|
|
return amount / (specificDivider or 1)
|
|
|
|
importMap = {
|
|
"em" : "em",
|
|
"therm": "thermal",
|
|
"kin" : "kinetic",
|
|
"exp" : "explosive"
|
|
}
|
|
|
|
@classmethod
|
|
def oneType(cls, damageType, amount=100):
|
|
pattern = DamagePattern()
|
|
pattern.update(amount if damageType == "em" else 0,
|
|
amount if damageType == "thermal" else 0,
|
|
amount if damageType == "kinetic" else 0,
|
|
amount if damageType == "explosive" else 0)
|
|
return pattern
|
|
|
|
@classmethod
|
|
def importPatterns(cls, text):
|
|
lines = re.split('[\n\r]+', text)
|
|
patterns = []
|
|
numPatterns = 0
|
|
|
|
# When we import damage profiles, we create new ones and update old ones. To do this, get a list of current
|
|
# patterns to allow lookup
|
|
lookup = {}
|
|
current = eos.db.getDamagePatternList()
|
|
for pattern in current:
|
|
lookup[pattern.name] = pattern
|
|
|
|
for line in lines:
|
|
try:
|
|
if line.strip()[0] == "#": # comments
|
|
continue
|
|
line = line.split('#', 1)[0] # allows for comments
|
|
type, data = line.rsplit('=', 1)
|
|
type, data = type.strip(), data.split(',')
|
|
except:
|
|
# Data isn't in correct format, continue to next line
|
|
continue
|
|
|
|
if type != "DamageProfile":
|
|
continue
|
|
|
|
numPatterns += 1
|
|
name, data = data[0], data[1:5]
|
|
fields = {}
|
|
|
|
for index, val in enumerate(data):
|
|
try:
|
|
fields["%sAmount" % cls.DAMAGE_TYPES[index]] = int(val)
|
|
except:
|
|
continue
|
|
|
|
if len(fields) == 4: # Avoid possible blank lines
|
|
if name.strip() in lookup:
|
|
pattern = lookup[name.strip()]
|
|
pattern.update(**fields)
|
|
eos.db.save(pattern)
|
|
else:
|
|
pattern = DamagePattern(**fields)
|
|
pattern.name = name.strip()
|
|
eos.db.save(pattern)
|
|
patterns.append(pattern)
|
|
|
|
eos.db.commit()
|
|
|
|
return patterns, numPatterns
|
|
|
|
EXPORT_FORMAT = "DamageProfile = %s,%d,%d,%d,%d\n"
|
|
|
|
@classmethod
|
|
def exportPatterns(cls, *patterns):
|
|
out = "# Exported from pyfa\n#\n"
|
|
out += "# Values are in following format:\n"
|
|
out += "# DamageProfile = [name],[EM amount],[Thermal amount],[Kinetic amount],[Explosive amount]\n\n"
|
|
for dp in patterns:
|
|
out += cls.EXPORT_FORMAT % (dp.name, dp.emAmount, dp.thermalAmount, dp.kineticAmount, dp.explosiveAmount)
|
|
|
|
return out.strip()
|
|
|
|
def __deepcopy__(self, memo):
|
|
p = DamagePattern(self.emAmount, self.thermalAmount, self.kineticAmount, self.explosiveAmount)
|
|
p.name = "%s copy" % self.name
|
|
return p
|