Add undo support to muta->normal conversion

gui/fitCommands/__init__.py

@@ -38,3 +38,4 @@ from .guiImportMutatedModule import GuiImportMutatedModuleCommand
 from .guiSetSpoolup import GuiSetSpoolup
 from .guiRebaseItems import GuiRebaseItemsCommand
 from .guiMutaConvert import GuiMutaConvertCommand
+from .guiMutaRevert import GuiMutaRevertCommand

gui/fitCommands/guiMutaConvert.py

@@ -1,6 +1,5 @@
 import wx
 
-import eos.db
 import gui.mainFrame
 from gui import globalEvents as GE
 from service.fit import Fit
@@ -18,8 +17,13 @@ class GuiMutaConvertCommand(wx.Command):
         self.mutaplasmid = mutaplasmid
 
     def Do(self):
-        fit = eos.db.getFit(self.fitID)
+        sFit = Fit.getInstance()
+        fit = sFit.getFit(self.fitID)
         oldMod = fit.modules[self.position]
+        if oldMod.isEmpty:
+            return False
+        if oldMod.isMutated:
+            return False
 
         success = self.internal_history.Submit(FitReplaceModuleCommand(
             fitID=self.fitID,
@@ -33,7 +37,7 @@ class GuiMutaConvertCommand(wx.Command):
         if not success:
             return False
 
-        Fit.getInstance().recalc(self.fitID)
+        sFit.recalc(self.fitID)
         wx.PostEvent(self.mainFrame, GE.FitChanged(fitID=self.fitID))
         return True
 

gui/fitCommands/guiMutaRevert.py

@@ -0,0 +1,48 @@
+import wx
+
+import gui.mainFrame
+from gui import globalEvents as GE
+from service.fit import Fit
+from .calc.fitReplaceModule import FitReplaceModuleCommand
+
+
+class GuiMutaRevertCommand(wx.Command):
+
+    def __init__(self, fitID, position):
+        wx.Command.__init__(self, True, "Convert Item to Normal")
+        self.mainFrame = gui.mainFrame.MainFrame.getInstance()
+        self.internal_history = wx.CommandProcessor()
+        self.fitID = fitID
+        self.position = position
+
+    def Do(self):
+        sFit = Fit.getInstance()
+        fit = sFit.getFit(self.fitID)
+        oldMod = fit.modules[self.position]
+        if oldMod.isEmpty:
+            return False
+        if not oldMod.isMutated:
+            return False
+
+        success = self.internal_history.Submit(FitReplaceModuleCommand(
+            fitID=self.fitID,
+            position=self.position,
+            newItemID=oldMod.baseItemID,
+            newBaseItemID=None,
+            newMutaplasmidID=None,
+            newMutations=None,
+            newState=oldMod.state,
+            newCharge=oldMod.charge))
+        if not success:
+            return False
+
+        sFit.recalc(self.fitID)
+        wx.PostEvent(self.mainFrame, GE.FitChanged(fitID=self.fitID))
+        return True
+
+    def Undo(self):
+        for _ in self.internal_history.Commands:
+            self.internal_history.Undo()
+        Fit.getInstance().recalc(self.fitID)
+        wx.PostEvent(self.mainFrame, GE.FitChanged(fitID=self.fitID))
+        return True