Changeset d3d67f0 in sasview for src

Timestamp:

Jul 10, 2015 6:27:04 PM (9 years ago)

Author:

Paul Kienzle <pkienzle@…>

Branches:

master, ESS_GUI, ESS_GUI_Docs, ESS_GUI_batch_fitting, ESS_GUI_bumps_abstraction, ESS_GUI_iss1116, ESS_GUI_iss879, ESS_GUI_iss959, ESS_GUI_opencl, ESS_GUI_ordering, ESS_GUI_sync_sascalc, costrafo411, magnetic_scatt, release-4.1.1, release-4.1.2, release-4.2.2, release_4.0.1, ticket-1009, ticket-1094-headless, ticket-1242-2d-resolution, ticket-1243, ticket-1249, ticket885, unittest-saveload

Children:

a7aa5c7

Parents:

098f3d2

Message:

fix toolbar on bumps/tools graphs

Location:

src/sas/plottools

Files:

• PlotPanel.py (modified) (8 diffs)
• config.py (modified) (1 diff)
• toolbar.py (modified) (4 diffs)

src/sas/plottools/PlotPanel.py

@@ -22 +22 @@
 
 from plottables import Graph
-from plottables import Text
 from TextDialog import TextDialog
 from LabelDialog import LabelDialog
@@ -33 +32 @@
 import numpy
 
-from sas.guiframe.events import StatusEvent
+from ..guiframe.events import StatusEvent
+from .toolbar import NavigationToolBar, PlotPrintout, bind
 
 def show_tree(obj, d=0):
@@ -89 +89 @@
             lo, hi = math.pow(10., lo), math.pow(10., hi)
     return (lo, hi)
-
-
-def CopyImage(canvas):
-    """
-    0: matplotlib plot
-    1: wx.lib.plot
-    2: other
-    """
-    bmp = wx.BitmapDataObject()
-    bmp.SetBitmap(canvas.bitmap)
-
-    wx.TheClipboard.Open()
-    wx.TheClipboard.SetData(bmp)
-    wx.TheClipboard.Close()
 
 
@@ -334 +320 @@
         """
         self.enable_toolbar = True
-        from toolbar import NavigationToolBar
         self.toolbar = NavigationToolBar(parent=self, canvas=self.canvas)
+        bind(self.toolbar, wx.EVT_TOOL, self.onResetGraph, id=self.toolbar._NTB2_RESET)
+        bind(self.toolbar, wx.EVT_TOOL, self.onContextMenu, id=self.toolbar._NTB2_HOME)
         self.toolbar.Realize()
         ## The 'SetToolBar()' is not working on MAC: JHC
@@ -1966 +1953 @@
         self.toolbar.update()
 
+    def onPrint(self, event=None):
+        self.toolbar.print_figure(event)
+
     def onPrinterSetup(self, event=None):
         """
@@ -1988 +1978 @@
 
             #now create the printpeview object
-            _preview = wx.PrintPreview(PlotPrintout(self,'test'),
-                                       PlotPrintout(self,'test'))
+            _preview = wx.PrintPreview(PlotPrintout(self.canvas),
+                                       PlotPrintout(self.canvas))
             #and tie it to a printpreview frame then show it
             _frame = wx.PreviewFrame(_preview, _plot, "Print Preview", wx.Point(100, 100), wx.Size(600, 650))
@@ -1999 +1989 @@
             pass
 
-    def onPrint(self, event=None):
-        """
-        Matplotlib canvas can no longer print itself so using wx.Printer.  
-        Need therefore to also define a Printout object.
+    def OnCopyFigureMenu(self, evt):
+        """
+        Copy the current figure to clipboard
         """
         try:
-            _printer = wx.Printer()
-            _printer.Print(self.canvas, PlotPrintout(self,'test'), True)
-        except:
-            traceback.print_exc()
-            pass
-
-    def OnCopyFigureMenu(self, evt):
-        """
-        Copy the current figure to clipboard
-        """
-        try:
-            CopyImage(self.canvas)
+            self.toolbar.copy_figure()
         except:
             print "Error in copy Image"
@@ -2046 +2024 @@
             self._drawn += 1
         self.gui_repaint(drawDC=wx.PaintDC(self))
-
-class PlotPrintout(wx.Printout):
-    """
-    Create the wx.Printout object for matplotlib figure from the PlotPanel.
-    Provides the required OnPrintPage and HasPage overrides.  Other methods
-    may be added/overriden in the future.
-    :TODO: this needs LOTS of TLC .. but fixes immediate problem
-    """
-    def __init__(self, figure, title, size=None, dpi=None, **kwargs):
-        """
-        Initialize wx.Printout and get passed figure object
-        """
-        wx.Printout.__init__(self)
-        self.figure = figure
-
-    def OnPrintPage(self, page):
-        """
-        Most rudimentry OnPrintPage overide.  instatiates a dc object, gets
-        its size, gets the size of the figure object, scales it to the dc
-        canvas size keeping the aspect ratio intact, then prints as bitmap
-        """
-        _dc = self.GetDC()
-        (_dcX, _dcY) = _dc.GetSizeTuple()
-        (_bmpX,_bmpY) = self.figure.canvas.GetSize()
-        _scale = min(_dcX/_bmpX, _dcY/_bmpY)
-        _dc.SetUserScale(_scale, _scale)
-        _dc.DrawBitmap(self.figure.canvas.bitmap, 0, 0, False,)
-        return True
-
-    def GetPageInfo(self):
-        """
-        just sets the page to 1 - no flexibility for now
-        """
-        return (1, 1, 1, 1)

src/sas/plottools/config.py

@@ -58 +58 @@
 matplotlib.rcParams[param] = 0.05
 matplotlib.rcParams['legend.numpoints'] = 1
+
+# this should happen after initial matplotlib configuration
+from .toolbar import NavigationToolBar
+from matplotlib.backends import backend_wxagg
+backend_wxagg.NavigationToolbar2WxAgg = NavigationToolBar
+
+# CRUFT: bumps 0.7.5.6 and older uses wrong toolbar
+backend_wxagg.NavigationToolbar2Wx = NavigationToolBar

src/sas/plottools/toolbar.py

@@ -31 +31 @@
     def _init_toolbar(self):
         self._parent = self.canvas.GetParent()
-        _NTB2_HOME = wx.NewId()
+        self._NTB2_HOME = wx.NewId()
         self._NTB2_BACK = wx.NewId()
         self._NTB2_FORWARD = wx.NewId()
         self._NTB2_PAN = wx.NewId()
         self._NTB2_ZOOM = wx.NewId()
-        _NTB2_SAVE = wx.NewId()
-        _NTB2_PRINT = wx.NewId()
-        _NTB2_RESET = wx.NewId()
+        self._NTB2_SAVE = wx.NewId()
+        self._NTB2_PRINT = wx.NewId()
+        self._NTB2_RESET = wx.NewId()
 
         # for mpl 1.2+ compatibility
@@ -53 +53 @@
         context_tip += '    right-click the data symbols.'
         context = wx.ArtProvider.GetBitmap(wx.ART_LIST_VIEW, wx.ART_TOOLBAR)
-        self.AddSimpleTool(_NTB2_HOME, context, context_tip, context_tip)
+        self.AddSimpleTool(self._NTB2_HOME, context, context_tip, context_tip)
 
         self.InsertSeparator(1)
@@ -69 +69 @@
 
         self.AddSeparator()
-        self.AddSimpleTool(_NTB2_SAVE, _load_bitmap('filesave.png'),
+        self.AddSimpleTool(self._NTB2_SAVE, _load_bitmap('filesave.png'),
                            'Save', 'Save plot contents to file')
 
         print_bmp = wx.ArtProvider.GetBitmap(wx.ART_PRINT, wx.ART_TOOLBAR)
-        self.AddSimpleTool(_NTB2_PRINT, print_bmp, 'Print', 'Print plot')
+        self.AddSimpleTool(self._NTB2_PRINT, print_bmp, 'Print', 'Print plot')
 
         reset_bmp = wx.ArtProvider.GetBitmap(wx.ART_GO_HOME, wx.ART_TOOLBAR)
-        self.AddSimpleTool(_NTB2_RESET, reset_bmp, 'Reset', 'Reset graph range')
+        self.AddSimpleTool(self._NTB2_RESET, reset_bmp, 'Reset', 'Reset graph range')
 
-        bind(self, wx.EVT_TOOL, self.on_menu, id=_NTB2_HOME)
+        bind(self, wx.EVT_TOOL, self.context_menu, id=self._NTB2_HOME)
         bind(self, wx.EVT_TOOL, self.forward, id=self._NTB2_FORWARD)
         bind(self, wx.EVT_TOOL, self.back, id=self._NTB2_BACK)
         bind(self, wx.EVT_TOOL, self.zoom, id=self._NTB2_ZOOM)
         bind(self, wx.EVT_TOOL, self.pan, id=self._NTB2_PAN)
-        bind(self, wx.EVT_TOOL, self.save_figure, id=_NTB2_SAVE)
-        bind(self, wx.EVT_TOOL, self.on_print, id=_NTB2_PRINT)
-        bind(self, wx.EVT_TOOL, self.on_reset, id=_NTB2_RESET)
+        bind(self, wx.EVT_TOOL, self.save_figure, id=self._NTB2_SAVE)
+        bind(self, wx.EVT_TOOL, self.print_figure, id=self._NTB2_PRINT)
+        bind(self, wx.EVT_TOOL, self.home, id=self._NTB2_RESET)
 
         self.Realize()
 
     def on_menu(self, event):
-        """
-            Plot menu
-        """
         try:
             self._parent.onToolContextMenu(event=event)
@@ -98 +95 @@
             logging.error("Plot toolbar could not show menu")
 
-    def on_reset(self, event):
+    def context_menu(self, event):
         """
-            Reset plot
+        Default context menu for a plot panel
+
         """
+        # Slicer plot popup menu
+        wx_id = wx.NewId()
+        popup = wx.Menu()
+        popup.Append(wx_id, '&Save image', 'Save image as PNG')
+        wx.EVT_MENU(self, wx_id, self.save_figure)
+
+        wx_id = wx.NewId()
+        popup.Append(wx_id, '&Print image', 'Print image ')
+        wx.EVT_MENU(self, wx_id, self.print_figure)
+
+        wx_id = wx.NewId()
+        popup.Append(wx_id, '&Copy to Clipboard', 'Copy image to the clipboard')
+        wx.EVT_MENU(self, wx_id, self.copy_figure)
+
+        # Show the popup menu relative to the location of the toolbar
+        self.PopupMenu(popup, (0,0))
+
+
+    def print_figure(self, event):
         try:
-            self._parent.onResetGraph(event=event)
+            _printer = wx.Printer()
+            _printer.Print(self.canvas, PlotPrintout(self.canvas), True)
         except:
-            logging.error("Plot toolbar could not reset plot")
+            import traceback
+            logging.error(traceback.format_exc())
 
-    def on_print(self, event):
+    def copy_figure(self, event):
+        copy_image_to_clipboard(self.canvas)
+
+class PlotPrintout(wx.Printout):
+    """
+    Create the wx.Printout object for matplotlib figure from the PlotPanel.
+    Provides the required OnPrintPage and HasPage overrides.  Other methods
+    may be added/overriden in the future.
+    :TODO: this needs LOTS of TLC .. but fixes immediate problem
+    """
+    def __init__(self, canvas):
         """
-            Print
+        Initialize wx.Printout and get passed figure object
         """
-        try:
-            self._parent.onPrint(event=event)
-        except:
-            logging.error("Plot toolbar could not print")
+        wx.Printout.__init__(self)
+        self.canvas = canvas
+
+    def OnPrintPage(self, page):
+        """
+        Most rudimentry OnPrintPage overide.  instatiates a dc object, gets
+        its size, gets the size of the figure object, scales it to the dc
+        canvas size keeping the aspect ratio intact, then prints as bitmap
+        """
+        _dc = self.GetDC()
+        (_dcX, _dcY) = _dc.GetSizeTuple()
+        (_bmpX,_bmpY) = self.canvas.GetSize()
+        _scale = min(_dcX/_bmpX, _dcY/_bmpY)
+        _dc.SetUserScale(_scale, _scale)
+        _dc.DrawBitmap(self.canvas.bitmap, 0, 0, False,)
+        return True
+
+    def GetPageInfo(self):
+        """
+        just sets the page to 1 - no flexibility for now
+        """
+        return (1, 1, 1, 1)
+
+
+def copy_image_to_clipboard(canvas):
+    bmp = wx.BitmapDataObject()
+    bmp.SetBitmap(canvas.bitmap)
+
+    wx.TheClipboard.Open()
+    wx.TheClipboard.SetData(bmp)
+    wx.TheClipboard.Close()
+
+