Fixed linewidth in selection in scatter plots.

fluents/plots.py

@@ -349,6 +349,8 @@ class ScatterPlot(Plot):
     """The ScatterPlot is slower than scattermarker, but has size option."""
     @plotlogger
     def __init__(self, dataset_1, dataset_2, id_dim, sel_dim, id_1, id_2, c='b', s=30, sel_dim_2=None, name="Scatter plot", **kw):
+        """Initializes a scatter plot.
+        """
 
         Plot.__init__(self, name)
         self.dataset_1 = dataset_1
@@ -407,6 +409,7 @@ class ScatterPlot(Plot):
                                                       self.yaxis_data,
                                                       alpha=0,
                                                       c='w',s=self.s,
+                                                      edgecolor='r',
                                                       linewidth=0,
                                                       zorder=4)
         self._background = self.canvas.copy_from_bbox(self.axes.bbox)
@@ -480,7 +483,7 @@ class ScatterPlot(Plot):
         linewidth = scipy.zeros(self.xaxis_data.shape, 'f')
         index = self.get_index_from_selection(self.dataset_1, selection)
         if len(index) > 0:
-            linewidth.put(1, index)
+            linewidth[index] = 1.5
         self.selection_collection.set_linewidth(linewidth)
             
         if self._use_blit and len(index)>0 :
@@ -527,9 +530,14 @@ class HistogramPlot(Plot):
             self.current_dim = dataset.get_dim_name(1)
         if dataset.shape[1]==1:
             self.current_dim = dataset.get_dim_name(0)
+
+        # Set default paramteters
+        if not kw.has_key('bins'):
+            kw['bins'] = 20
+
         # Initial draw
         self.axes.grid(False)
-        bins = min(self._data.size, 20)
+        bins = min(self._data.size, kw['bins'])
         count, lims, self.patches = self.axes.hist(self._data, bins=bins)
 
         # Add identifiers to the individual patches