Implement drone mutation support with some exceptions

gui/builtinContextMenus/itemMutations.py

@@ -0,0 +1,94 @@
+import re
+
+# noinspection PyPackageRequirements
+import wx
+
+import gui.mainFrame
+from gui.contextMenu import ContextMenuSingle
+from gui.fitCommands import (
+    GuiConvertMutatedLocalModuleCommand, GuiRevertMutatedLocalModuleCommand,
+    GuiConvertMutatedLocalDroneCommand, GuiRevertMutatedLocalDroneCommand)
+from service.fit import Fit
+
+_t = wx.GetTranslation
+
+
+class ChangeItemMutation(ContextMenuSingle):
+
+    def __init__(self):
+        self.mainFrame = gui.mainFrame.MainFrame.getInstance()
+        self.eventIDs = {}
+
+    def display(self, callingWindow, srcContext, mainItem):
+
+        if srcContext not in ("fittingModule", "droneItem") or self.mainFrame.getActiveFit() is None:
+            return False
+
+        if mainItem is None or getattr(mainItem, 'isEmpty', False):
+            return False
+
+        if len(mainItem.item.mutaplasmids) == 0 and not mainItem.isMutated:
+            return False
+
+        return True
+
+    def getText(self, callingWindow, itmContext, mainItem):
+        return _t("Apply Mutaplasmid") if not mainItem.isMutated else _t("Revert to {}").format(mainItem.baseItem.name)
+
+    def getSubMenu(self, callingWindow, context, mainItem, rootMenu, i, pitem):
+        if mainItem.isMutated:
+            return None
+
+        msw = True if "wxMSW" in wx.PlatformInfo else False
+        self.skillIds = {}
+        sub = wx.Menu()
+
+        menu = rootMenu if msw else sub
+
+        for item in mainItem.item.mutaplasmids:
+            label = item.item.name
+            keywords = ('Decayed', 'Gravid', 'Unstable', 'Radical')
+            for kw in keywords:
+                if item.item.name.startswith(f'{kw} '):
+                    label = kw
+            m = re.match('(?P<mutagrade>\S+) (?P<dronetype>\S+) Drone (?P<mutatype>\S+) Mutaplasmid', label)
+            if m:
+                label = '{} {}'.format(m.group('mutagrade'), m.group('mutatype'))
+            id = ContextMenuSingle.nextID()
+            self.eventIDs[id] = (item, mainItem)
+            mItem = wx.MenuItem(menu, id, label)
+            menu.Bind(wx.EVT_MENU, self.handleMenu, mItem)
+            sub.Append(mItem)
+
+        return sub
+
+    def handleMenu(self, event):
+        mutaplasmid, item = self.eventIDs[event.Id]
+        fitID = self.mainFrame.getActiveFit()
+        fit = Fit.getInstance().getFit(fitID)
+        if item in fit.modules:
+            position = fit.modules.index(item)
+            self.mainFrame.command.Submit(GuiConvertMutatedLocalModuleCommand(
+                    fitID=fitID, position=position, mutaplasmid=mutaplasmid))
+        elif item in fit.drones:
+            position = fit.drones.index(item)
+            self.mainFrame.command.Submit(GuiConvertMutatedLocalDroneCommand(
+                    fitID=fitID, position=position, mutaplasmid=mutaplasmid))
+
+    def activate(self, callingWindow, fullContext, mainItem, i):
+        fitID = self.mainFrame.getActiveFit()
+        fit = Fit.getInstance().getFit(fitID)
+        if mainItem in fit.modules:
+            position = fit.modules.index(mainItem)
+            self.mainFrame.command.Submit(GuiRevertMutatedLocalModuleCommand(
+                    fitID=fitID, position=position))
+        elif mainItem in fit.drones:
+            position = fit.drones.index(mainItem)
+            self.mainFrame.command.Submit(GuiRevertMutatedLocalDroneCommand(
+                    fitID=fitID, position=position))
+
+    def getBitmap(self, callingWindow, context, mainItem):
+        return None
+
+
+ChangeItemMutation.register()