Make it possible to save image sequence to a file

Author: tkralphs

setup.py

@@ -4,7 +4,7 @@
 import setuptools
 
 setup(name='coinor.grumpy',
-      version='0.9.3',
+      version='0.9.4',
       description='Graphics for Understanding Mathematical Programming (GrUMPy)',
       long_description='''GrUMPy is a class for visualizing various algorithm used in solving discrete optimization problem. It has a class for dynamically generating and visualizing branch-and-bound trees that is derived from the GiMPy graph class. Using the branch-and-bound class, a user can visualize the branch-and-bound process in a number of different ways either by building the tree dynamically through direct calls to Python from the solver or by piping the output of an instrumented solver to GrUMPy for parsing. The branch-and-bound class also includes a pure Python implementation of branch and bound that is targeted at educational use.
 

src/grumpy/polyhedron2D.py

@@ -316,15 +316,18 @@ def set_xlim(self, xlim):
     def set_ylim(self, ylim):
         self.ax.set_ylim(ylim)
 
-    def show(self, pause = True, wait_for_click = True):
+    def show(self, pause = True, wait_for_click = True, filename = None):
         plt.legend()
-        if wait_for_click == True:
-            plt.draw()
-            if plt.waitforbuttonpress(timeout = 10000):
-                plt.close()
-                exit()
+        if filename is not None:
+            plt.savefig(filename, bbox_inches='tight')
         else:
-            plt.show(block=pause)
+            if wait_for_click == True:
+                plt.draw()
+                if plt.waitforbuttonpress(timeout = 10000):
+                    plt.close()
+                    exit()
+                else:
+                    plt.show(block=pause)
         self.fig = None
 
 if __name__ == '__main__':