Add browser lite handlers to graphs window

This commit is contained in:
DarkPhoenix
2019-08-01 11:54:33 +03:00
parent ee2193e1bb
commit 2e9b024390
2 changed files with 12 additions and 2 deletions

View File

@@ -28,7 +28,8 @@ class AddBrowsedFits(ContextMenuUnconditional):
excludedFitIDs = callingWindow.getExistingFitIDs()
dlg = FitBrowserLiteDialog(self.mainFrame, title=titles[fullContext[0]], excludedFitIDs=excludedFitIDs)
if dlg.ShowModal() == wx.ID_OK:
pass
fitIDs = dlg.getFitIDsToAdd()
callingWindow.addFitsByIDs(fitIDs)
AddBrowsedFits.register()

View File

@@ -22,7 +22,6 @@
import wx
import gui.display
import gui.globalEvents as GE
from eos.saveddata.targetProfile import TargetProfile
from gui.contextMenu import ContextMenu
from service.const import GraphCacheCleanupReason
@@ -169,6 +168,16 @@ class BaseList(gui.display.Display):
def getExistingFitIDs(self):
return [f.ID for f in self.fits]
def addFitsByIDs(self, fitIDs):
sFit = Fit.getInstance()
for fitID in fitIDs:
fit = sFit.getFit(fitID)
if fit is not None:
self.fits.append(fit)
self.updateView()
self.graphFrame.draw()
class FitList(BaseList):
def __init__(self, graphFrame, parent):