Check version before putting columns in legend.

Author: alexras

Plot.py

@@ -395,11 +395,22 @@ def drawPlot(self, ax):
         if self.yLabel is not None:
             ax.set_ylabel(self.yLabel)
 
+        legendKeywords = {}
+        
+        if self.legendCols > 0:
+            (superMajor, major, minor) = [int(x) for x in matplotlib.__version__.split('.')]
+
+            if superMajor == 0 and major < 98:
+                print >>sys.stderr, "Number of columns support not available in versions of matplotlib prior to 0.98"
+            else:
+                legendKeywords["ncol"] = self.legendCols
+
+        
         if self.legend:
             pylab.legend(plotHandles, plotLabels, loc=self.legendLoc, 
-                         ncol=self.legendCols)
+                         **legendKeywords)
         if self.figLegend:
             pylab.figlegend(plotHandles, plotLabels, loc=self.legendLoc, 
-                            ncol=self.legendCols)
+                            **legendKeywords)
 
         return (plotHandles, plotLabels)

PlotLayout.py

@@ -1,4 +1,5 @@
 import pylab
+import matplotlib
 from matplotlib import pyplot
 import sys
 
@@ -218,9 +219,19 @@ def __doPlot(self):
                 ungroupedPlotsRemaining -= 1
 
         if self.figLegendLoc is not None:
+            figLegendKeywords = {}
+
+            if self.figLegendCols is not None:
+                (superMajor, major, minor) = [int(x) for x in matplotlib.__version__.split('.')]
+
+                if superMajor == 0 and major < 98:
+                    print >>sys.stderr, "Number of columns support not available in versions of matplotlib prior to 0.98"
+                else:
+                    figLegendKeywords["ncol"] = self.figLegendCols
+            
             pylab.figlegend(plotHandles, plotLabels, 
                             self.figLegendLoc, 
-                            ncol=self.figLegendCols)
+                            **figLegendKeywords)
 
         if self.plotParams is not None:
             pylab.subplots_adjust(left=self.plotParams["left"],