diff python/utils.py @ 115:550556378466

added functionalities to indicator maps
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Sat, 23 Jul 2011 02:18:16 -0400
parents f03ec4697a09
children 2bf5b76320c0
line wrap: on
line diff
--- a/python/utils.py	Mon Jul 18 19:24:06 2011 -0400
+++ b/python/utils.py	Sat Jul 23 02:18:16 2011 -0400
@@ -174,7 +174,7 @@
 
 colors = PlottingPropertyValues('brgmyck') # 'w'
 
-def plotIndicatorMap(indicatorMap, squareSize):
+def plotIndicatorMap(indicatorMap, squareSize, masked = True, defaultValue=-1):
     from numpy import array, arange, ones, ma
     from matplotlib.pyplot import pcolor
     coords = array(indicatorMap.keys())
@@ -182,11 +182,13 @@
     minY = min(coords[:,1])
     X = arange(minX, max(coords[:,0])+1.1)*squareSize
     Y = arange(minY, max(coords[:,1])+1.1)*squareSize
-    C = -ones((len(Y), len(X)))
+    C = defaultValue*ones((len(Y), len(X)))
     for k,v in indicatorMap.iteritems():
         C[k[1]-minY,k[0]-minX] = v
-    masked = ma.masked_where(C<0,C)
-    pcolor(X, Y, masked)
+    if masked:
+        pcolor(X, Y, ma.masked_where(C==defaultValue,C))
+    else:
+        pcolor(X, Y, C)
 
 #########################
 # file I/O section