Temporarily set DISPLAY environment variable if not set before saving. See Issue 17.

Author: alexras

PlotLayout.py

@@ -2,6 +2,7 @@
 import matplotlib
 from matplotlib import pyplot
 import sys
+import os
 
 from Utils import getGoldenRatioDimensions
 
@@ -252,6 +253,16 @@ def plot(self):
 
     def save(self, filename):
         print "Saving %s ..." % filename
+        
+        tempDisplayHack = False
+        
+        if "DISPLAY" not in os.environ:
+            tempDisplayHack = True
+            os.environ["DISPLAY"] = ":0.0"
+        
         self.__doPlot()
         pylab.savefig(filename)
         pylab.clf()
+        
+        if tempDisplayHack == True:
+            del os.environ["DISPLAY"]