# =============================================================================
# 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)