Changeset 9691

Timestamp:

Feb 25, 2015, 9:50:59 AM (9 years ago)

Author:

davies

Message:

Another fix

File:

• trunk/anuga_core/anuga/utilities/plot_utils.py (modified) (1 diff)

trunk/anuga_core/anuga/utilities/plot_utils.py

@@ -1180 +1180 @@
         all_poly.set_facecolor('none')
     else:
-        msg = 'len(values) must either be 1, or the same as len(p.vols)'
-        assert len(values)==len(p.vols), msg
+        try:
+            lv = len(values)
+        except:
+            values = numpy.array(len(p.vols)*[values])
+            lv = len(values)
+
+        msg = 'len(values) must be the same as len(p.vols) (or values can be a constant)'
+        assert lv==len(p.vols), msg
         all_poly = PolyCollection( vertices, array = values, cmap = values_cmap, 
             edgecolors=edgecolors)