32 lines
1.1 KiB
Python
Executable File
32 lines
1.1 KiB
Python
Executable File
from gui.contextMenu import ContextMenu
|
|
from gui.itemStats import ItemStatsDialog
|
|
from gui.patternEditor import DmgPatternEditorDlg
|
|
import gui.mainFrame
|
|
import service
|
|
import gui.fittingView
|
|
import wx
|
|
|
|
class DamagePattern(ContextMenu):
|
|
def __init__(self):
|
|
self.mainFrame = gui.mainFrame.MainFrame.getInstance()
|
|
|
|
def display(self, context, selection):
|
|
return context in ("resistancesViewFull",)
|
|
|
|
def getText(self, context, selection):
|
|
sDP = service.DamagePattern.getInstance()
|
|
self.patterns = sDP.getDamagePatternList()
|
|
self.patterns.sort(key=lambda p: p.name)
|
|
m = map(lambda p: p.name, self.patterns)
|
|
m.append("Raw")
|
|
return m
|
|
|
|
def activate(self, context, selection, i):
|
|
sDP = service.DamagePattern.getInstance()
|
|
sFit = service.Fit.getInstance()
|
|
fitID = self.mainFrame.getActiveFit()
|
|
sFit.setDamagePattern(fitID, self.patterns[i] if i < len(self.patterns) else None)
|
|
wx.PostEvent(self.mainFrame, gui.fittingView.FitChanged(fitID=fitID))
|
|
|
|
DamagePattern.register()
|