added agg buffer to numpy array example

jdh2358 · jdh2358 · commit 078d6c03cb74 · 2007-07-15T15:33:02.000Z
svn path=/trunk/matplotlib/; revision=3532
diff --git a/examples/agg_buffer_to_array.py b/examples/agg_buffer_to_array.py
@@ -0,0 +1,24 @@
+import matplotlib
+matplotlib.use('Agg')
+from pylab import figure, show
+import numpy as npy
+
+# make an agg figure
+fig = figure()
+ax = fig.add_subplot(111)
+ax.plot([1,2,3])
+ax.set_title('a simple figure')
+fig.canvas.draw()
+
+# grab rhe pixel buffer and dumpy it into a numpy array
+buf = fig.canvas.buffer_rgba(0,0)
+l, b, w, h = fig.bbox.get_bounds()
+X = npy.fromstring(buf, npy.uint8)
+X.shape = h,w,4
+
+# now display the array X as an Axes in a new figure
+fig2 = figure()
+ax2 = fig2.add_subplot(111, frameon=False)
+ax2.imshow(X)
+fig2.savefig('simple.png')
+show()
diff --git a/lib/matplotlib/backends/backend_agg.py b/lib/matplotlib/backends/backend_agg.py
@@ -382,8 +382,8 @@ def draw(self):
         """
         if __debug__: verbose.report('FigureCanvasAgg.draw', 'debug-annoying')
 
-        renderer = self.get_renderer()
-        self.figure.draw(renderer)
+        self.renderer = self.get_renderer()
+        self.figure.draw(self.renderer)
 
     def get_renderer(self):
         l,b,w,h = self.figure.bbox.get_bounds()
diff --git a/lib/matplotlib/image.py b/lib/matplotlib/image.py
@@ -264,7 +264,9 @@ def get_extent(self):
         if self._extent is not None:
             return self._extent
         else:
-            numrows, numcols = self.get_size()
+            sz = self.get_size()
+            #print 'sz', sz
+            numrows, numcols = sz
             if self.origin == 'upper':
                 return (-0.5, numcols-0.5, numrows-0.5, -0.5)
             else:
