Continue work on projections (adding/removing projected environment and fit)

gui/fitCommands/guiRemoveProjected.py

@@ -0,0 +1,82 @@
+import wx
+from service.fit import Fit
+
+import gui.mainFrame
+from gui import globalEvents as GE
+from eos.saveddata.module import Module
+from .calc.fitRemoveProjectedModule import FitRemoveProjectedModuleCommand
+from .calc.fitRemoveProjectedEnv import FitRemoveProjectedEnvCommand
+from .calc.fitRemoveProjectedFit import FitRemoveProjectedFitCommand
+from logbook import Logger
+import eos.db
+pyfalog = Logger(__name__)
+
+from eos.saveddata.drone import Drone
+from eos.saveddata.module import Module
+from eos.saveddata.fighter import Fighter
+
+
+class GuiRemoveProjectedCommand(wx.Command):
+    def __init__(self, fitID, thing):
+        wx.Command.__init__(self, True, "Projected Add")
+        self.mainFrame = gui.mainFrame.MainFrame.getInstance()
+        self.sFit = Fit.getInstance()
+        self.internal_history = wx.CommandProcessor()
+        self.fitID = fitID
+        fit = self.sFit.getFit(fitID)
+
+        if isinstance(thing, Drone):
+            self.data = fit.projectedDrones.index(thing)
+            self.type = 'drone'
+        elif isinstance(thing, Module):
+            # todo: projected stuff should be wrapped in a projected class wrapper for easier maintainence
+            if thing.item.group.name in Module.SYSTEM_GROUPS:
+                self.type = 'env'
+                self.data = thing.itemID
+            else:
+                self.type = 'module'
+                self.data = fit.projectedModules.index(thing)
+        elif isinstance(thing, Fighter):
+            self.data = fit.projectedFighters.index(thing)
+            self.type = 'fighter'
+        else:
+            # todo: fix!
+            self.data = thing.ID
+            self.type = 'fit'
+
+    def Do(self):
+        result = False
+        # since we can project various types, we need to switch of the fit command. We can't do this switch easily in
+        # the fit command since each type might have a different kind of undo, easier to split it out
+        if self.type == 'module':
+            result = self.internal_history.Submit(FitRemoveProjectedModuleCommand(self.fitID, self.data))
+        elif self.type == 'env':
+            result = self.internal_history.Submit(FitRemoveProjectedEnvCommand(self.fitID, self.data))
+        elif self.type == 'fit':
+            result = self.internal_history.Submit(FitRemoveProjectedFitCommand(self.fitID, self.data))
+
+        # if item.category.name == "Drone":
+        #     pyfalog.warn("DRONE REMOVE PROJECTION NOT IMPLEMENTED")
+        # elif item.category.name == "Fighter":
+        #     pyfalog.warn("FIGHTER REMOVE PROJECTION NOT IMPLEMENTED")
+        # elif item.group.name in Module.SYSTEM_GROUPS:
+        #     result = self.internal_history.Submit(FitRemoveProjectedEnvCommand(self.fitID, self.id))
+        # else:
+        #     # attempt a regular module projection
+        #
+        # elif self.type == 'fit':
+        #     pyfalog.warn("FIT REMOVE PROJECTION NOT IMPLEMENTED")
+
+        if result:
+            self.sFit.recalc(self.fitID)
+            wx.PostEvent(self.mainFrame, GE.FitChanged(fitID=self.fitID))
+            return True
+        return False
+
+    def Undo(self):
+        for _ in self.internal_history.Commands:
+            self.internal_history.Undo()
+        self.sFit.recalc(self.fitID)
+        wx.PostEvent(self.mainFrame, GE.FitChanged(fitID=self.fitID))
+        return True
+