70 lines
2.4 KiB
Python
70 lines
2.4 KiB
Python
from collections import OrderedDict
|
|
|
|
# noinspection PyPackageRequirements
|
|
import wx
|
|
|
|
import gui.mainFrame
|
|
from graphs.events import ResistModeChanged
|
|
from graphs.wrapper import TargetWrapper
|
|
from gui.contextMenu import ContextMenuCombined
|
|
from service.const import TargetResistMode
|
|
from service.settings import GraphSettings
|
|
|
|
|
|
optionMap = OrderedDict((
|
|
('Auto', TargetResistMode.auto),
|
|
('Shield', TargetResistMode.shield),
|
|
('Armor', TargetResistMode.armor),
|
|
('Hull', TargetResistMode.hull),
|
|
('Weighted Average', TargetResistMode.weightedAverage)))
|
|
|
|
|
|
class TargetWrapperResists(ContextMenuCombined):
|
|
|
|
def __init__(self):
|
|
self.mainFrame = gui.mainFrame.MainFrame.getInstance()
|
|
|
|
def display(self, callingWindow, srcContext, mainItem, selection):
|
|
if srcContext != 'graphTgtList':
|
|
return False
|
|
if GraphSettings.getInstance().get('ignoreResists'):
|
|
return False
|
|
if not isinstance(mainItem, TargetWrapper) or not mainItem.isFit:
|
|
return False
|
|
self.callingWindow = callingWindow
|
|
self.selection = selection
|
|
return True
|
|
|
|
def getText(self, callingWindow, itmContext, mainItem, selection):
|
|
return 'Resist Mode'
|
|
|
|
def addOption(self, menu, optionLabel):
|
|
id = ContextMenuCombined.nextID()
|
|
self.optionIds[id] = optionLabel
|
|
menuItem = wx.MenuItem(menu, id, optionLabel, kind=wx.ITEM_CHECK)
|
|
menu.Bind(wx.EVT_MENU, self.handleMode, menuItem)
|
|
return menuItem
|
|
|
|
def getSubMenu(self, callingWindow, context, mainItem, selection, rootMenu, i, pitem):
|
|
msw = True if "wxMSW" in wx.PlatformInfo else False
|
|
self.optionIds = {}
|
|
sub = wx.Menu()
|
|
for optionLabel, optionValue in optionMap.items():
|
|
menuItem = self.addOption(rootMenu if msw else sub, optionLabel)
|
|
sub.Append(menuItem)
|
|
menuItem.Check(mainItem.resistMode == optionValue)
|
|
return sub
|
|
|
|
def handleMode(self, event):
|
|
optionLabel = self.optionIds[event.Id]
|
|
optionValue = optionMap[optionLabel]
|
|
changedFitIDs = set()
|
|
for wrapper in self.selection:
|
|
if wrapper.isFit and wrapper.resistMode != optionValue:
|
|
wrapper.resistMode = optionValue
|
|
changedFitIDs.add(wrapper.item.ID)
|
|
wx.PostEvent(gui.mainFrame.MainFrame.getInstance(), ResistModeChanged(fitIDs=changedFitIDs))
|
|
|
|
|
|
TargetWrapperResists.register()
|