diff --git a/gui/builtinContextMenus/fitAddBrowse.py b/gui/builtinContextMenus/fitAddBrowse.py index 836121f40..0f6eae676 100644 --- a/gui/builtinContextMenus/fitAddBrowse.py +++ b/gui/builtinContextMenus/fitAddBrowse.py @@ -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() diff --git a/gui/graphFrame/lists.py b/gui/graphFrame/lists.py index 9888bb812..bdf3d2b52 100644 --- a/gui/graphFrame/lists.py +++ b/gui/graphFrame/lists.py @@ -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):