Changeset d3d67f0 in sasview for src/sas/plottools/PlotPanel.py

Timestamp:

Jul 10, 2015 4: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

File:

• src/sas/plottools/PlotPanel.py (modified) (8 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)