Merge pull request #539 from efiring/qt4agg_draw

mdboom · mdboom · commit 29a43dbdc119 · 2011-11-08T11:01:06.000-08:00
Qt4agg draw
diff --git a/lib/matplotlib/backends/backend_qt4agg.py b/lib/matplotlib/backends/backend_qt4agg.py
@@ -59,7 +59,6 @@ def __init__( self, figure ):
         self.drawRect = False
         self.rect = []
         self.blitbox = None
-        self.replot = True
         self.setAttribute(QtCore.Qt.WA_OpaquePaintEvent)
 
     def drawRectangle( self, rect ):
@@ -69,7 +68,7 @@ def drawRectangle( self, rect ):
 
     def paintEvent( self, e ):
         """
-        Draw to the Agg backend and then copy the image to the qt.drawable.
+        Copy the image from the Agg canvas to the qt.drawable.
         In Qt, all drawing should be done inside of here when a widget is
         shown onscreen.
         """
@@ -78,10 +77,6 @@ def paintEvent( self, e ):
         if DEBUG: print 'FigureCanvasQtAgg.paintEvent: ', self, \
            self.get_width_height()
 
-        if self.replot:
-            FigureCanvasAgg.draw(self)
-            self.replot = False
-
         if self.blitbox is None:
             # matplotlib is in rgba byte order.  QImage wants to put the bytes
             # into argb format and is in a 4 byte unsigned int.  Little endian
@@ -121,11 +116,13 @@ def paintEvent( self, e ):
 
     def draw( self ):
         """
-        Draw the figure when xwindows is ready for the update
+        Draw the figure with Agg, and queue a request
+        for a Qt draw.
         """
-
-        if DEBUG: print "FigureCanvasQtAgg.draw", self
-        self.replot = True
+        # The Agg draw is done here; delaying it until the paintEvent
+        # causes problems with code that uses the result of the
+        # draw() to update plot elements.
+        FigureCanvasAgg.draw(self)
         self.update()
 
     def blit(self, bbox=None):
