From 004cfe0a9f1c5df53cb535e854f9360e1a9ffea5 Mon Sep 17 00:00:00 2001
From: flatberg <flatberg@pvv.ntnu.no>
Date: Tue, 14 Aug 2007 16:10:35 +0000
Subject: [PATCH] Quick fix for gray on NaNs

---
 fluents/plots.py | 7 +++++++
 1 file changed, 7 insertions(+)

diff --git a/fluents/plots.py b/fluents/plots.py
index 3dedc0a..41aa91b 100644
--- a/fluents/plots.py
+++ b/fluents/plots.py
@@ -101,6 +101,7 @@ class Plot(view.View):
         control : intersection
 
         """
+        print "Inside update selection, key: %s" %key
         if key == 'shift':
             ids = set(ids).union(self._current_selection[self.current_dim])
         elif key == 'control':
@@ -491,6 +492,12 @@ class ScatterPlot(Plot):
         self.sc.set_array(map_vec)
         self.sc.set_clim(vec_min, vec_max)
         self.sc.update_scalarmappable() #sets facecolors from array
+        if hasattr(self.sc.cmap, "_lut"):
+            print "changing lut"
+            self.sc.cmap._lut[-1,:] = [.5,.5,.5,1]
+            self.sc.cmap._lut[0,:] = [.5,.5,.5,1]
+        else:
+            print "No lut present"
         self.canvas.draw()
                     
     def rectangle_select_callback(self, x1, y1, x2, y2, key):