Reimplement some of logic used in ammo picker in a service package

service/ammo.py

@@ -0,0 +1,146 @@
+# =============================================================================
+# Copyright (C) 2010 Diego Duclos
+#
+# This file is part of pyfa.
+#
+# pyfa is free software: you can redistribute it and/or modify
+# it under the terms of the GNU General Public License as published by
+# the Free Software Foundation, either version 3 of the License, or
+# (at your option) any later version.
+#
+# pyfa 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 General Public License for more details.
+#
+# You should have received a copy of the GNU General Public License
+# along with pyfa.  If not, see <http://www.gnu.org/licenses/>.
+# =============================================================================
+
+
+import math
+
+from eos.const import FittingHardpoint
+from eos.saveddata.module import Module
+from eos.utils.stats import DmgTypes
+from service.market import Market
+
+
+class Ammo:
+
+    instance = None
+
+    @classmethod
+    def getInstance(cls):
+        if cls.instance is None:
+            cls.instance = Ammo()
+
+        return cls.instance
+
+    @staticmethod
+    def getModuleFlatAmmo(mod):
+        sMkt = Market.getInstance()
+        if mod is None or mod.isEmpty:
+            return set()
+        chargeSet = set()
+        # Do not try to grab it for t3d modes which can also be passed as part of selection
+        if isinstance(mod, Module):
+            for charge in mod.getValidCharges():
+                if sMkt.getPublicityByItem(charge):
+                    chargeSet.add(charge)
+        return chargeSet
+
+    @classmethod
+    def getModuleStructuredAmmo(cls, mod):
+
+        def getChargeDamageInfo(charge):
+            # Set up data storage for missile damage stuff
+            damageMap = {}
+            totalDamage = 0
+            # Fill them with the data about charge
+            for damageType in DmgTypes.names():
+                currentDamage = charge.getAttribute('{}Damage'.format(damageType)) or 0
+                damageMap[damageType] = currentDamage
+                totalDamage += currentDamage
+            # Detect type of ammo
+            chargeDamageType = None
+            for damageType in damageMap:
+                # If all damage belongs to certain type purely, set appropriate
+                # ammoType
+                if damageMap[damageType] == totalDamage:
+                    chargeDamageType = damageType
+                    break
+            # Else consider ammo as mixed damage
+            if chargeDamageType is None:
+                chargeDamageType = 'mixed'
+            return chargeDamageType, totalDamage
+
+        def turretSorter(mod, charge):
+            damage = 0
+            range_ = (mod.item.getAttribute('maxRange')) * \
+                     (charge.getAttribute('weaponRangeMultiplier') or 1)
+            falloff = (mod.item.getAttribute('falloff') or 0) * \
+                      (charge.getAttribute('fallofMultiplier') or 1)
+            for type_ in DmgTypes.names():
+                d = charge.getAttribute('%sDamage' % type_)
+                if d > 0:
+                    damage += d
+            # Take optimal and falloff as range factor
+            rangeFactor = range_ + falloff
+            return -rangeFactor, charge.name.rsplit()[-2:], damage, charge.name
+
+        def missileSorter(mod, charge):
+            # Get charge damage type and total damage
+            chargeDamageType, totalDamage = getChargeDamageInfo(charge)
+            # Find its position in sort list
+            try:
+                position = DmgTypes.names().index(chargeDamageType)
+            # Put charges which have non-standard damage type after charges with
+            # standard damage type
+            except ValueError:
+                position = math.inf
+            return position, totalDamage, charge.name
+
+        def nameSorter(charge):
+            parts = charge.name.split(" ")
+            return [int(p) if p.isdigit() else p for p in parts]
+
+        chargesFlat = cls.getModuleFlatAmmo(mod)
+        # Make sure we do not consider mining turrets as combat turrets
+        if mod.hardpoint == FittingHardpoint.TURRET and mod.getModifiedItemAttr('miningAmount', None) is None:
+            all = []
+            sub = []
+            prevNameBase = None
+            prevRange = None
+            for charge in sorted(chargesFlat, key=turretSorter):
+                if 'civilian' in charge.name.lower():
+                    continue
+                currNameBase = charge.name.rsplit()[-2:]
+                currRange = charge.getAttribute('weaponRangeMultiplier')
+                if prevNameBase is None or currRange != prevRange or currNameBase != prevNameBase:
+                    if sub:
+                        all.append(sub)
+                        sub = []
+                    sub.append(charge)
+                    prevNameBase = currNameBase
+                    prevRange = currRange
+                else:
+                    sub.append(charge)
+            return 'ddTurret', all
+        elif mod.hardpoint == FittingHardpoint.MISSILE and mod.item.name != 'Festival Launcher':
+            all = []
+            sub = []
+            prevType = None
+            for charge in sorted(chargesFlat, key=missileSorter):
+                currType = getChargeDamageInfo(charge)[0]
+                if prevType is None or currType != prevType:
+                    if sub:
+                        all.append(sub)
+                        sub = []
+                    sub.append(charge)
+                    prevType = currType
+                else:
+                    sub.append(charge)
+            return 'ddMissile', all
+        else:
+            return 'general', sorted(chargesFlat, key=nameSorter)