From 436bee1e85f2379d012b133d4afdcd4b626c922b Mon Sep 17 00:00:00 2001
From: blitzmann <holmes.ryan.90@gmail.com>
Date: Sat, 5 Jan 2019 22:26:17 -0500
Subject: [PATCH] Start working on a gui to set spoolup options

---
 gui/builtinContextMenus/spoolUp.py | 122 +++++++++++++++++++++++++++++
 gui/contextMenu.py                 |   1 +
 2 files changed, 123 insertions(+)
 create mode 100644 gui/builtinContextMenus/spoolUp.py

diff --git a/gui/builtinContextMenus/spoolUp.py b/gui/builtinContextMenus/spoolUp.py
new file mode 100644
index 000000000..0d6f04452
--- /dev/null
+++ b/gui/builtinContextMenus/spoolUp.py
@@ -0,0 +1,122 @@
+import gui.mainFrame
+from gui.contextMenu import ContextMenu
+# noinspection PyPackageRequirements
+from service.settings import ContextMenuSettings
+import wx
+from eos.utils.spoolSupport import SpoolType
+
+class SpoolUp(ContextMenu):
+    def __init__(self):
+        self.mainFrame = gui.mainFrame.MainFrame.getInstance()
+        self.settings = ContextMenuSettings.getInstance()
+
+    def display(self, srcContext, selection):
+        if not self.settings.get('project'):
+            return False
+
+        if srcContext not in ("fittingModule") or self.mainFrame.getActiveFit() is None:
+            return False
+
+        item = selection[0]
+
+        return item.item.group.name in ('Precursor Weapon')
+
+    def getText(self, itmContext, selection):
+        return "Set Spoolup".format(itmContext)
+
+    def activate(self, fullContext, selection, i):
+        thing = selection[0]
+        mainFrame = gui.mainFrame.MainFrame.getInstance()
+        fitID = mainFrame.getActiveFit()
+        srcContext = fullContext[0]
+
+        while True:
+            dlg = SpoolUpChanger(self.mainFrame, thing)
+            if dlg.ShowModal() != wx.ID_OK:
+                break
+
+            type = dlg.spoolChoice.GetClientData(dlg.spoolChoice.GetSelection())
+            amount = dlg.input.GetValue()
+
+            if type == SpoolType.SCALE:
+                if amount < 0 or amount > 100:
+                    dlg = wx.MessageDialog(self.mainFrame, "The amount provided  isn't within the range of 0 - 100.", "Error", wx.OK | wx.ICON_ERROR)
+                    dlg.ShowModal()
+                    continue
+                amount = amount / 100
+            if type is None:
+                amount = None
+
+            thing.spoolType = type
+            thing.spoolAmount = amount
+            break
+
+
+
+SpoolUp.register()
+
+
+class SpoolUpChanger(wx.Dialog):
+
+    spoolTypes = {
+        None: ("Default", "pyfa defaults to using 'scale' calculation with 100%"),
+        SpoolType.SCALE: ("Scale", "Amount is a percentage from 0 - 100; rounds down to closest module cycle."),
+        SpoolType.TIME: ("Time (s)", "Amount defines time in seconds since spoolup module was activated."),
+        SpoolType.CYCLES: ("Cycles", "Amount is number of cycles module went through since being activated.")
+    }
+
+    def __init__(self, parent, module):
+        wx.Dialog.__init__(self, parent, title="Change Spool-Up")
+        self.SetMinSize((346, 156))
+
+        bSizer1 = wx.BoxSizer(wx.VERTICAL)
+
+        bSizer2 = wx.BoxSizer(wx.VERTICAL)
+
+        text = wx.StaticText(self, wx.ID_ANY, "Type:")
+        bSizer2.Add(text, 0)
+
+        self.spoolChoice = wx.Choice(self, wx.ID_ANY, style=0)
+
+        for k, v in self.spoolTypes.items():
+            i = self.spoolChoice.Append(v[0], k)
+            if module.spoolType == k:
+                self.spoolChoice.SetSelection(i)
+
+        self.spoolChoice.Bind(wx.EVT_CHOICE, self.spoolTypeChanged)
+
+
+        bSizer2.Add(self.spoolChoice, 0, wx.TOP | wx.BOTTOM | wx.EXPAND, 5)
+
+        self.spoolDesc = wx.StaticText(self, wx.ID_ANY, self.spoolTypes[module.spoolType][1])
+        bSizer2.Add(self.spoolDesc, 1)
+
+        text1 = wx.StaticText(self, wx.ID_ANY, "Amount:")
+        bSizer2.Add(text1, 0, wx.TOP, 10)
+
+        bSizer1.Add(bSizer2, 0, wx.ALL, 10)
+
+        self.input = wx.SpinCtrlDouble(self, min=0, max=1000)
+
+        bSizer1.Add(self.input, 0, wx.LEFT | wx.RIGHT | wx.EXPAND, 10)
+
+        bSizer3 = wx.BoxSizer(wx.VERTICAL)
+        bSizer3.Add(wx.StaticLine(self, wx.ID_ANY), 0, wx.BOTTOM | wx.EXPAND, 15)
+
+        bSizer3.Add(self.CreateStdDialogButtonSizer(wx.OK | wx.CANCEL), 0, wx.EXPAND)
+        bSizer1.Add(bSizer3, 0, wx.ALL | wx.EXPAND, 10)
+
+        self.input.SetFocus()
+        self.input.Bind(wx.EVT_TEXT_ENTER, self.processEnter)
+        self.SetSizer(bSizer1)
+        self.CenterOnParent()
+        self.Fit()
+
+    def spoolTypeChanged(self, evt):
+        if evt.ClientData is None:
+            self.input.Enable(False)
+        self.spoolDesc.SetLabel(self.spoolTypes[evt.ClientData][1])
+        self.Layout()
+
+    def processEnter(self, evt):
+        self.EndModal(wx.ID_OK)
diff --git a/gui/contextMenu.py b/gui/contextMenu.py
index be972b0f8..6a97481a2 100644
--- a/gui/contextMenu.py
+++ b/gui/contextMenu.py
@@ -184,6 +184,7 @@ from gui.builtinContextMenus import (  # noqa: E402,F401
     moduleGlobalAmmoPicker,
     moduleAmmoPicker,
     itemStats,
+    spoolUp,
     damagePattern,
     marketJump,
     # droneSplit,