143 lines
4.9 KiB
Python
143 lines
4.9 KiB
Python
# noinspection PyPackageRequirements
|
|
import wx
|
|
|
|
import gui.mainFrame
|
|
from gui.auxFrame import AuxiliaryFrame
|
|
from gui.contextMenu import ContextMenuSingle
|
|
from service.market import Market
|
|
|
|
|
|
class GraphFitAmmoPicker(ContextMenuSingle):
|
|
|
|
def __init__(self):
|
|
self.mainFrame = gui.mainFrame.MainFrame.getInstance()
|
|
|
|
def display(self, callingWindow, srcContext, mainItem):
|
|
if srcContext != 'graphFitList':
|
|
return False
|
|
if mainItem is None or not mainItem.isFit:
|
|
return False
|
|
if callingWindow.graphFrame.getView().internalName != 'dmgStatsGraph':
|
|
return False
|
|
return True
|
|
|
|
def getText(self, callingWindow, itmContext, mainItem):
|
|
return 'Plot with Different Ammo...'
|
|
|
|
def activate(self, callingWindow, fullContext, mainItem, i):
|
|
window = AmmoPicker(self.mainFrame, mainItem.item)
|
|
window.Show()
|
|
|
|
|
|
GraphFitAmmoPicker.register()
|
|
|
|
|
|
class AmmoPicker(AuxiliaryFrame):
|
|
|
|
def __init__(self, parent, fit):
|
|
super().__init__(parent, title='Choose Different Ammo', style=wx.DEFAULT_DIALOG_STYLE)
|
|
|
|
mods = self.getMods(fit)
|
|
drones = self.getDrones(fit)
|
|
fighters = self.getFighters(fit)
|
|
|
|
mainSizer = wx.BoxSizer(wx.VERTICAL)
|
|
|
|
firstRadio = True
|
|
|
|
def addRadioButton(text):
|
|
nonlocal firstRadio
|
|
if not firstRadio:
|
|
rb = wx.RadioButton(self, wx.ID_ANY, text, style=wx.RB_GROUP)
|
|
rb.SetValue(True)
|
|
firstRadio = True
|
|
else:
|
|
rb = wx.RadioButton(self, wx.ID_ANY, text)
|
|
rb.SetValue(False)
|
|
mainSizer.Add(rb, 0, wx.EXPAND | wx.ALL, 5)
|
|
|
|
for modInfo, ammo in mods:
|
|
text = '\n'.join('{}x {}'.format(amount, item.name) for item, amount in modInfo)
|
|
addRadioButton(text)
|
|
if drones:
|
|
addRadioButton('Drones')
|
|
if fighters:
|
|
addRadioButton('Fighters')
|
|
|
|
self.SetSizer(mainSizer)
|
|
self.SetMinSize((346, 156))
|
|
self.Bind(wx.EVT_KEY_UP, self.kbEvent)
|
|
|
|
def kbEvent(self, event):
|
|
if event.GetKeyCode() == wx.WXK_ESCAPE and event.GetModifiers() == wx.MOD_NONE:
|
|
self.Close()
|
|
return
|
|
event.Skip()
|
|
|
|
def getMods(self, fit):
|
|
sMkt = Market.getInstance()
|
|
loadableCharges = {}
|
|
# Modules, format: {frozenset(ammo): {item: count}}
|
|
modsPrelim = {}
|
|
if fit is not None:
|
|
for mod in fit.modules:
|
|
if not mod.canDealDamage():
|
|
continue
|
|
typeID = mod.item.ID
|
|
if typeID in loadableCharges:
|
|
charges = loadableCharges[typeID]
|
|
else:
|
|
charges = loadableCharges.setdefault(typeID, set())
|
|
for charge in mod.getValidCharges():
|
|
if sMkt.getPublicityByItem(charge):
|
|
charges.add(charge)
|
|
# We're not interested in modules which contain no charges
|
|
if charges:
|
|
data = modsPrelim.setdefault(frozenset(charges), {})
|
|
if mod.item not in data:
|
|
data[mod.item] = 0
|
|
data[mod.item] += 1
|
|
# Format: [([(item, count), ...], frozenset(ammo)), ...]
|
|
modsFinal = []
|
|
for charges, itemCounts in modsPrelim.items():
|
|
modsFinal.append((
|
|
# Sort items within group
|
|
sorted(itemCounts.items(), key=lambda i: sMkt.itemSort(i[0]), reverse=True),
|
|
charges))
|
|
# Sort item groups
|
|
modsFinal.sort(key=lambda i: sMkt.itemSort(i[0][0][0]), reverse=True)
|
|
return modsFinal
|
|
|
|
def getDrones(self, fit):
|
|
drones = set()
|
|
if fit is not None:
|
|
for drone in fit.drones:
|
|
if drone.item is None:
|
|
continue
|
|
# Drones are our "ammo", so we want to pick even those which are inactive
|
|
if drone.canDealDamage(ignoreState=True):
|
|
drones.add(drone)
|
|
continue
|
|
if {'remoteWebifierEntity', 'remoteTargetPaintEntity'}.intersection(drone.item.effects):
|
|
drones.add(drone)
|
|
continue
|
|
return drones
|
|
|
|
def getFighters(self, fit):
|
|
fighters = set()
|
|
if fit is not None:
|
|
for fighter in fit.fighters:
|
|
if fighter.item is None:
|
|
continue
|
|
# Fighters are our "ammo" as well
|
|
if fighter.canDealDamage(ignoreState=True):
|
|
fighters.add(fighter)
|
|
continue
|
|
for ability in fighter.abilities:
|
|
if not ability.active:
|
|
continue
|
|
if ability.effect.name == 'fighterAbilityStasisWebifier':
|
|
fighters.add(fighter)
|
|
break
|
|
return fighters
|