fb20847920
Added the ability to remove charges and projected items using the itemRemove built in context menu. In this commit I fixed a minor bug in the projected view that caused the right clicked item not to be SHOWN as selected until after the context menu had been spawned and closed. I also removed all parent references from the ContextMenu class and sub classes. I am instead using the fit service to do all the work of removing the items in the itemRemove class.
55 lines
1.7 KiB
Python
55 lines
1.7 KiB
Python
from gui.contextMenu import ContextMenu
|
|
import gui.mainFrame
|
|
import gui.globalEvents as GE
|
|
import service
|
|
import wx
|
|
|
|
class WhProjector(ContextMenu):
|
|
def __init__(self):
|
|
self.mainFrame = gui.mainFrame.MainFrame.getInstance()
|
|
|
|
def display(self, srcContext, selection):
|
|
return srcContext == "projected"
|
|
|
|
def getText(self, itmContext, selection):
|
|
return "Add System Effects"
|
|
|
|
def activate(self, fullContext, selection, i):
|
|
pass
|
|
|
|
def getSubMenu(self, context, selection, menu, i):
|
|
self.idmap = {}
|
|
menu.Bind(wx.EVT_MENU, self.handleSelection)
|
|
m = wx.Menu()
|
|
sMkt = service.Market.getInstance()
|
|
effdata = sMkt.getSystemWideEffects()
|
|
for swType in sorted(effdata):
|
|
item = wx.MenuItem(m, wx.ID_ANY, swType)
|
|
sub = wx.Menu()
|
|
sub.Bind(wx.EVT_MENU, self.handleSelection)
|
|
item.SetSubMenu(sub)
|
|
m.AppendItem(item)
|
|
for swData in sorted(effdata[swType], key=lambda tpl: tpl[2]):
|
|
wxid = wx.NewId()
|
|
swObj, swName, swClass = swData
|
|
self.idmap[wxid] = (swObj, swName)
|
|
subitem = wx.MenuItem(sub, wxid, swClass)
|
|
sub.AppendItem(subitem)
|
|
return m
|
|
|
|
|
|
def handleSelection(self, event):
|
|
#Skip events ids that aren't mapped
|
|
|
|
swObj, swName = self.idmap.get(event.Id, (False, False))
|
|
if not swObj and not swName:
|
|
event.Skip()
|
|
return
|
|
|
|
sFit = service.Fit.getInstance()
|
|
fitID = self.mainFrame.getActiveFit()
|
|
sFit.project(fitID, swObj)
|
|
wx.PostEvent(self.mainFrame, GE.FitChanged(fitID=fitID))
|
|
|
|
WhProjector.register()
|