diff --git a/gui/builtinContextMenus/graphFitAmmoPicker.py b/gui/builtinContextMenus/graphFitAmmoPicker.py index 06049ed7d..de112d697 100644 --- a/gui/builtinContextMenus/graphFitAmmoPicker.py +++ b/gui/builtinContextMenus/graphFitAmmoPicker.py @@ -1,20 +1,40 @@ +# noinspection PyPackageRequirements +import wx + +import gui.mainFrame from gui.contextMenu import ContextMenuSingle class GraphFitAmmoPicker(ContextMenuSingle): + def __init__(self): + self.mainFrame = gui.mainFrame.MainFrame.getInstance() + def display(self, callingWindow, srcContext, mainItem): if srcContext != 'graphFitList': return False if mainItem is None or not mainItem.isFit: return False + if callingWindow.graphFrame.getView().internalName != 'dmgStatsGraph': + return False return True def getText(self, callingWindow, itmContext, mainItem): - return 'Duplicate Fit with Ammo...' + return 'Plot with Different Ammo...' def activate(self, callingWindow, fullContext, mainItem, i): - pass + with AmmoPicker(self.mainFrame) as dlg: + if dlg.ShowModal() == wx.ID_OK: + pass + else: + pass GraphFitAmmoPicker.register() + + +class AmmoPicker(wx.Dialog): + + def __init__(self, parent): + super().__init__(parent, title='Choose Different Ammo', style=wx.DEFAULT_DIALOG_STYLE) + self.SetMinSize((346, 156))