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.
48 lines
1.6 KiB
Python
48 lines
1.6 KiB
Python
from gui.contextMenu import ContextMenu
|
|
import gui.mainFrame
|
|
import service
|
|
import gui.globalEvents as GE
|
|
import wx
|
|
from gui import bitmapLoader
|
|
|
|
class DamagePattern(ContextMenu):
|
|
def __init__(self):
|
|
self.mainFrame = gui.mainFrame.MainFrame.getInstance()
|
|
|
|
def display(self, srcContext, selection):
|
|
return srcContext in ("resistancesViewFull",) and self.mainFrame.getActiveFit() is not None
|
|
|
|
def getText(self, itmContext, selection):
|
|
sDP = service.DamagePattern.getInstance()
|
|
self.patterns = sDP.getDamagePatternList()
|
|
self.patterns.sort( key=lambda p: (p.name in ["Selected Ammo",
|
|
"Uniform"], p.name) )
|
|
m = map(lambda p: p.name, self.patterns)
|
|
return m
|
|
|
|
def activate(self, fullContext, selection, i):
|
|
sDP = service.DamagePattern.getInstance()
|
|
sFit = service.Fit.getInstance()
|
|
fitID = self.mainFrame.getActiveFit()
|
|
sFit.setDamagePattern(fitID, self.patterns[i])
|
|
setattr(self.mainFrame,"_activeDmgPattern",self.patterns[i])
|
|
wx.PostEvent(self.mainFrame, GE.FitChanged(fitID=fitID))
|
|
|
|
def getBitmap(self, context, selection):
|
|
sFit = service.Fit.getInstance()
|
|
fitID = self.mainFrame.getActiveFit()
|
|
f = sFit.getFit(fitID)
|
|
dp = f.damagePattern
|
|
if dp is None:
|
|
return None
|
|
|
|
index = self.patterns.index(dp)
|
|
bitmap = bitmapLoader.getBitmap("state_active_small", "icons")
|
|
l = [None] * len(self.patterns)
|
|
l[index] = bitmap
|
|
|
|
return l
|
|
|
|
|
|
DamagePattern.register()
|