Merge pull request #14242 from anntzer/animatecopy

tacaswell · web-flow · commit 4a9b1c1d1a59 · 2019-05-18T11:31:56.000-04:00
PERF: Avoid a buffer copy in PillowWriter.
diff --git a/lib/matplotlib/animation.py b/lib/matplotlib/animation.py
@@ -563,12 +563,10 @@ def grab_frame(self, **savefig_kwargs):
         buf = BytesIO()
         self._fig.savefig(buf, **dict(savefig_kwargs, format="rgba"))
         renderer = self._fig.canvas.get_renderer()
-        # Using frombuffer / getbuffer may be slightly more efficient, but
-        # Py3-only.
-        self._frames.append(Image.frombytes(
+        self._frames.append(Image.frombuffer(
             "RGBA",
-            (int(renderer.width), int(renderer.height)),
-            buf.getvalue()))
+            (int(renderer.width), int(renderer.height)), buf.getbuffer(),
+            "raw", "RGBA", 0, 1))
 
     def finish(self):
         self._frames[0].save(
