Changeset 14521 for branches/DataPreprocessing Enhancements/HeuristicLab.DataPreprocessing.Views/3.4/ScatterPlotSingleView.cs

Timestamp:

12/22/16 12:42:09 (7 years ago)

Author:

pfleck

Message:

#2709 Added an option for the preprocessing scatterplot to use a color gradient instead of the chart color palette.

File:

• branches/DataPreprocessing Enhancements/HeuristicLab.DataPreprocessing.Views/3.4/ScatterPlotSingleView.cs (modified) (9 diffs)

branches/DataPreprocessing Enhancements/HeuristicLab.DataPreprocessing.Views/3.4/ScatterPlotSingleView.cs

@@ -22 +22 @@
 using System;
 using System.Collections.Generic;
+using System.Drawing;
 using System.Linq;
 using HeuristicLab.Analysis;
+using HeuristicLab.Common;
 using HeuristicLab.Core.Views;
 using HeuristicLab.MainForm;
+using HeuristicLab.MainForm.WindowsForms;
 using RegressionType = HeuristicLab.Analysis.ScatterPlotDataRowVisualProperties.ScatterPlotDataRowRegressionType;
 
@@ -33 +36 @@
   [Content(typeof(SingleScatterPlotContent), true)]
   public partial class ScatterPlotSingleView : ItemView {
+
+    private readonly string NoGroupItem = "-";
 
     public new SingleScatterPlotContent Content {
@@ -55 +60 @@
       comboBoxXVariable.Items.AddRange(variables.ToArray());
       comboBoxYVariable.Items.AddRange(variables.ToArray());
-      comboBoxGroup.Items.Add("-");
+      comboBoxGroup.Items.Add(NoGroupItem);
       for (int i = 0; i < Content.PreprocessingData.Columns; ++i) {
         if (Content.PreprocessingData.VariableHasType<double>(i)) {
-          double distinctValueCount = Content.PreprocessingData.GetValues<double>(i).GroupBy(x => x).Count();
-          if (distinctValueCount <= 20)
-            comboBoxGroup.Items.Add(Content.PreprocessingData.GetVariableName(i));
+          //double distinctValueCount = Content.PreprocessingData.GetValues<double>(i).GroupBy(x => x).Count();
+          //if (distinctValueCount <= 20)
+          comboBoxGroup.Items.Add(Content.PreprocessingData.GetVariableName(i));
         }
       }
@@ -86 +91 @@
     }
 
+    protected override void SetEnabledStateOfControls() {
+      base.SetEnabledStateOfControls();
+      useGradientCheckBox.Enabled = (string)comboBoxGroup.SelectedItem != NoGroupItem;
+      gradientPanel.Visible = useGradientCheckBox.Enabled && useGradientCheckBox.Checked; ;
+    }
+
     private void UpdateScatterPlot() {
       if (comboBoxXVariable.SelectedItem != null && comboBoxYVariable.SelectedItem != null && comboBoxGroup.SelectedItem != null) {
@@ -91 +102 @@
         var yVariable = (string)comboBoxYVariable.SelectedItem;
         var groupVariable = (string)comboBoxGroup.SelectedItem;
+
+        bool groupingActive = useGradientCheckBox.Checked && groupVariable != NoGroupItem;
+        double min = 0, max = 1;
+        if (groupingActive) {
+          var groupValues = Content.PreprocessingData.GetValues<double>(Content.PreprocessingData.GetColumnIndex(groupVariable))
+            .Distinct().OrderBy(x => x).ToList();
+          min = groupValues.First();
+          max = groupValues.Last();
+        }
         ScatterPlot scatterPlot = Content.CreateScatterPlot(xVariable, yVariable, groupVariable);
+        var rows = scatterPlot.Rows.ToList();
+        scatterPlot.Rows.Clear();
         var regressionType = (RegressionType)regressionTypeComboBox.SelectedValue;
         int order = (int)polynomialRegressionOrderNumericUpDown.Value;
-        foreach (var row in scatterPlot.Rows) {
+        foreach (var row in rows) {
           row.VisualProperties.PointSize = 6;
           row.VisualProperties.IsRegressionVisibleInLegend = false;
           row.VisualProperties.RegressionType = regressionType;
           row.VisualProperties.PolynomialRegressionOrder = order;
-        }
+          row.VisualProperties.IsVisibleInLegend = !useGradientCheckBox.Checked;
+          if (groupingActive)
+            row.VisualProperties.Color = GetColor(double.Parse(row.Name), min, max);
+        }
+        scatterPlot.Rows.AddRange(rows);
         var vp = scatterPlot.VisualProperties;
         vp.Title = string.Empty;
@@ -106 +132 @@
 
         scatterPlotControl.Content = scatterPlot;
+
+        if (groupingActive) {
+          gradientMinimumLabel.Text = min.ToString("G5");
+          gradientMaximumLabel.Text = max.ToString("G5");
+        }
 
         //save selected x and y variable in content
@@ -152 +183 @@
 
     private void comboBoxGroup_SelectedIndexChanged(object sender, EventArgs e) {
+      useGradientCheckBox.Enabled = (string)comboBoxGroup.SelectedItem != NoGroupItem;
+      gradientPanel.Visible = useGradientCheckBox.Enabled && useGradientCheckBox.Checked;
       UpdateScatterPlot();
     }
@@ -159 +192 @@
       var regressionType = (RegressionType)regressionTypeComboBox.SelectedValue;
       polynomialRegressionOrderNumericUpDown.Enabled = regressionType == RegressionType.Polynomial;
+
       UpdateRegressionLine();
     }
@@ -170 +204 @@
       int order = (int)polynomialRegressionOrderNumericUpDown.Value;
 
-      foreach (var row in scatterPlotControl.Content.Rows) {
+      var rows = scatterPlotControl.Content.Rows.ToList();
+      scatterPlotControl.Content.Rows.Clear();
+      foreach (var row in rows) {
         row.VisualProperties.IsRegressionVisibleInLegend = false;
         row.VisualProperties.RegressionType = regressionType;
         row.VisualProperties.PolynomialRegressionOrder = order;
       }
+      scatterPlotControl.Content.Rows.AddRange(rows);
     }
     #endregion
+
+    private void useGradientCheckBox_CheckedChanged(object sender, EventArgs e) {
+      gradientPanel.Visible = useGradientCheckBox.Enabled && useGradientCheckBox.Checked;
+
+      // remove rows and re-add them later to avoid firing visual property changd events
+      var rows = scatterPlotControl.Content.Rows.ToDictionary(r => r.Name, r => r);
+      scatterPlotControl.Content.Rows.Clear();
+
+      if (useGradientCheckBox.Checked) {
+        var groupVariable = (string)comboBoxGroup.SelectedItem;
+        if (groupVariable == NoGroupItem) return;
+
+        var groupValues = Content.PreprocessingData.GetValues<double>(Content.PreprocessingData.GetColumnIndex(groupVariable))
+          .Distinct().OrderBy(x => x).ToList();
+        double min = groupValues.First(), max = groupValues.Last();
+        foreach (var group in groupValues) {
+          ScatterPlotDataRow row;
+          if (rows.TryGetValue(group.ToString("R"), out row)) {
+            row.VisualProperties.Color = GetColor(group, min, max);
+            row.VisualProperties.IsVisibleInLegend = false;
+          }
+        }
+        gradientMinimumLabel.Text = min.ToString("G5");
+        gradientMaximumLabel.Text = max.ToString("G5");
+      } else {
+        foreach (var row in rows.Values) {
+          row.VisualProperties.Color = Color.Empty;
+          row.VisualProperties.IsVisibleInLegend = true;
+        }
+      }
+      scatterPlotControl.Content.Rows.AddRange(rows.Values);
+    }
+
+    private static Color GetColor(double value, double min, double max) {
+      if (double.IsNaN(value)) {
+        return Color.DarkGray;
+      }
+      var colors = ColorGradient.Colors;
+      int index = (int)((colors.Count - 1) * (value - min) / (max - min));
+      if (index >= colors.Count) index = colors.Count - 1;
+      if (index < 0) index = 0;
+      return colors[index];
+    }
+
+    private void BatchRowUpdate(Action<ScatterPlotDataRow> rowAction) {
+      var scatterPlot = scatterPlotControl.Content;
+      var rows = scatterPlot.Rows.ToList();
+      // remove rows and re-add them later to avoid firing visual property changd events
+      scatterPlot.Rows.Clear();
+      foreach (var row in rows) {
+        rowAction(row);
+      }
+      scatterPlot.Rows.AddRange(rows);
+    }
   }
 }