Overload grab_frame to close temp files

jenshnielsen · jenshnielsen · commit 63337277da53 · 2014-07-11T14:00:44.000-05:00
diff --git a/lib/matplotlib/animation.py b/lib/matplotlib/animation.py
@@ -208,6 +208,7 @@ def grab_frame(self, **savefig_kwargs):
             # frame format and dpi.
             self.fig.savefig(self._frame_sink(), format=self.frame_format,
                              dpi=self.dpi, **savefig_kwargs)
+
         except RuntimeError:
             out, err = self._proc.communicate()
             verbose.report('MovieWriter -- Error '
@@ -326,6 +327,30 @@ def _frame_sink(self):
         # because it will no longer be referenced and will be gc-ed.
         return open(fname, 'wb')
 
+    def grab_frame(self, **savefig_kwargs):
+        '''
+        Grab the image information from the figure and save as a movie frame.
+        All keyword arguments in savefig_kwargs are passed on to the 'savefig'
+        command that saves the figure.
+        '''
+        #Overloaded to explicitly close temp file.
+        verbose.report('MovieWriter.grab_frame: Grabbing frame.',
+                       level='debug')
+        try:
+            # Tell the figure to save its data to the sink, using the
+            # frame format and dpi.
+            myframesink = self._frame_sink()
+            self.fig.savefig(myframesink, format=self.frame_format,
+                             dpi=self.dpi, **savefig_kwargs)
+            myframesink.close()
+
+        except RuntimeError:
+            out, err = self._proc.communicate()
+            verbose.report('MovieWriter -- Error '
+                           'running proc:\n%s\n%s' % (out,
+                                                      err), level='helpful')
+            raise
+
     def finish(self):
         # Call run here now that all frame grabbing is done. All temp files
         # are available to be assembled.
