diff --git a/gui/graphFrame.py b/gui/graphFrame.py index 030cf1776..6db5618e5 100644 --- a/gui/graphFrame.py +++ b/gui/graphFrame.py @@ -145,6 +145,8 @@ class GraphFrame(wx.Frame): self.draw() except: self.SetStatusText("Invalid values") + def AppendFitToList(self, fitID): + print "Append fit %d" % fitID class FitList(wx.Panel): def __init__(self, parent): diff --git a/gui/shipBrowser.py b/gui/shipBrowser.py index d9f6bb2e2..b526c1c19 100644 --- a/gui/shipBrowser.py +++ b/gui/shipBrowser.py @@ -1549,6 +1549,15 @@ class FitItem(wx.Window): self.btnsStatus = "" self.Refresh() + def checkForGraphFrame(self, wnd, gfwnd ): + + while wnd is not None: + if wnd == gfwnd: + return True + else: + wnd = wnd.Parent + return False + def checkPosition(self, event): if self.dragging and self.dragged: self.dragging = False @@ -1563,6 +1572,7 @@ class FitItem(wx.Window): pjWnd = self.mainFrame.additionsPane.projectedPage msWnd = self.mainFrame.fitMultiSwitch cfitWnd = self.mainFrame.fitMultiSwitch.GetCurrentPage() + gfWnd = self.mainFrame.graphFrame if not targetWnd: return @@ -1587,6 +1597,9 @@ class FitItem(wx.Window): fitInst.project(activeFit,draggedFit) wx.PostEvent(self.mainFrame, fv.FitChanged(fitID=activeFit)) + if self.checkForGraphFrame(targetWnd, gfWnd): + self.mainFrame.graphFrame.AppendFitToList(self.fitID) + event.Skip() return