diff --git a/gui/graphFrame.py b/gui/graphFrame.py index baf197bde..9a39d8af2 100644 --- a/gui/graphFrame.py +++ b/gui/graphFrame.py @@ -162,6 +162,7 @@ class GraphFrame(wx.Frame): self.mainSizer.Add(self.fitList, 0, wx.EXPAND) self.fitList.fitList.Bind(wx.EVT_LEFT_DCLICK, self.OnLeftDClick) + self.fitList.fitList.Bind(wx.EVT_CONTEXT_MENU, self.OnContextMenu) self.mainFrame.Bind(GE.FIT_CHANGED, self.draw) self.Bind(wx.EVT_CLOSE, self.closeEvent) self.Bind(wx.EVT_CHAR_HOOK, self.kbEvent) @@ -169,6 +170,11 @@ class GraphFrame(wx.Frame): from gui.builtinStatsViews.resistancesViewFull import EFFECTIVE_HP_TOGGLED # Grr crclar gons self.mainFrame.Bind(EFFECTIVE_HP_TOGGLED, self.ehpToggled) + self.contextMenu = wx.Menu() + removeItem = wx.MenuItem(self.contextMenu, 1, 'Remove Fit') + self.contextMenu.Append(removeItem) + self.contextMenu.Bind(wx.EVT_MENU, self.ContextMenuHandler, removeItem) + self.Fit() self.SetMinSize(self.GetSize()) @@ -192,6 +198,16 @@ class GraphFrame(wx.Frame): self.removeFits(self.getSelectedFits()) event.Skip() + def OnContextMenu(self, event): + if self.getSelectedFits(): + self.PopupMenu(self.contextMenu) + + def ContextMenuHandler(self, event): + selectedMenuItem = event.GetId() + if selectedMenuItem == 1: # Copy was chosen + fits = self.getSelectedFits() + self.removeFits(fits) + def ehpToggled(self, event): event.Skip() view = self.getView() @@ -401,6 +417,7 @@ class GraphFrame(wx.Frame): class FitList(wx.Panel): + def __init__(self, parent): wx.Panel.__init__(self, parent) self.mainSizer = wx.BoxSizer(wx.VERTICAL) @@ -418,4 +435,3 @@ class FitDisplay(gui.display.Display): def __init__(self, parent): gui.display.Display.__init__(self, parent) -