From 012c5f536b46a4509a31052fd8213aa8f6b0b944 Mon Sep 17 00:00:00 2001
From: blitzmann <holmes.ryan.90@gmail.com>
Date: Sun, 3 Jun 2018 16:29:00 -0400
Subject: [PATCH] Add mutaplasmid context menu

---
 gui/builtinContextMenus/mutaplasmids.py | 44 +++++++++++++++++++++++++
 gui/contextMenu.py                      |  3 +-
 2 files changed, 46 insertions(+), 1 deletion(-)
 create mode 100644 gui/builtinContextMenus/mutaplasmids.py

diff --git a/gui/builtinContextMenus/mutaplasmids.py b/gui/builtinContextMenus/mutaplasmids.py
new file mode 100644
index 000000000..87d5e12ea
--- /dev/null
+++ b/gui/builtinContextMenus/mutaplasmids.py
@@ -0,0 +1,44 @@
+from gui.contextMenu import ContextMenu
+import gui.mainFrame
+# noinspection PyPackageRequirements
+import wx
+import gui.globalEvents as GE
+from service.fit import Fit
+from service.settings import ContextMenuSettings
+
+
+class MutaplasmidCM(ContextMenu):
+    def __init__(self):
+        self.mainFrame = gui.mainFrame.MainFrame.getInstance()
+        self.settings = ContextMenuSettings.getInstance()
+
+    def display(self, srcContext, selection):
+
+        # if not self.settings.get('ammoPattern'):
+        #     return False
+
+        if srcContext not in ("fittingModule") or self.mainFrame.getActiveFit() is None:
+            return False
+
+        item = selection[0]
+        for attr in ("emDamage", "thermalDamage", "explosiveDamage", "kineticDamage"):
+            if item.getAttribute(attr) is not None:
+                return True
+
+        return False
+
+    def getText(self, itmContext, selection):
+        return "Apply Mutaplasmid"
+
+    def activate(self, fullContext, selection, i):
+        item = selection[0]
+        fit = self.mainFrame.getActiveFit()
+        sFit = Fit.getInstance()
+        sFit.setAsPattern(fit, item)
+        wx.PostEvent(self.mainFrame, GE.FitChanged(fitID=fit))
+
+    def getBitmap(self, context, selection):
+        return None
+
+
+MutaplasmidCM.register()
diff --git a/gui/contextMenu.py b/gui/contextMenu.py
index 11d574157..08ea258d5 100644
--- a/gui/contextMenu.py
+++ b/gui/contextMenu.py
@@ -208,5 +208,6 @@ from gui.builtinContextMenus import (  # noqa: E402,F401
     fighterAbilities,
     boosterSideEffects,
     commandFits,
-    tabbedFits
+    tabbedFits,
+    mutaplasmids,
 )