Changeset 14514 for branches/DataPreprocessing Enhancements/HeuristicLab.DataPreprocessing.Views/3.4/ScatterPlotMultiView.cs

Timestamp:

12/21/16 11:26:54 (7 years ago)

Author:

pfleck

Message:

#2709

• Added a VerticalLabel for the multi-scatterplot.
• Added regression options for single- and multi-scatterplot

File:

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

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

@@ -10 +10 @@
 using HeuristicLab.MainForm;
 using HeuristicLab.MainForm.WindowsForms;
+using RegressionType = HeuristicLab.Analysis.ScatterPlotDataRowVisualProperties.ScatterPlotDataRowRegressionType;
 
 namespace HeuristicLab.DataPreprocessing.Views {
@@ -16 +17 @@
   public partial class ScatterPlotMultiView : PreprocessingCheckedVariablesView {
     private readonly IDictionary<string, Label> columnHeaderCache;
-    private readonly IDictionary<string, Label> rowHeaderCache;
+    private readonly IDictionary<string, VerticalLabel> rowHeaderCache;
     private readonly IDictionary<Tuple<string/*col*/, string/*row*/>, Control> bodyCache;
 
     public ScatterPlotMultiView() {
       InitializeComponent();
+
+      regressionTypeComboBox.DataSource = Enum.GetValues(typeof(RegressionType));
+      regressionTypeComboBox.SelectedItem = RegressionType.None;
 
       #region Initialize Scrollbars
@@ -41 +45 @@
 
       columnHeaderCache = new Dictionary<string, Label>();
-      rowHeaderCache = new Dictionary<string, Label>();
+      rowHeaderCache = new Dictionary<string, VerticalLabel>();
       bodyCache = new Dictionary<Tuple<string, string>, Control>();
 
@@ -57 +61 @@
         GenerateCharts();
       }
+    }
+
+    protected override void SetEnabledStateOfControls() {
+      base.SetEnabledStateOfControls();
+      var regressionType = (RegressionType)regressionTypeComboBox.SelectedValue;
+      polynomialRegressionOrderNumericUpDown.Enabled = regressionType == RegressionType.Polynomial;
     }
 
@@ -239 +249 @@
     private Label GetRowHeader(string variable) {
       if (!rowHeaderCache.ContainsKey(variable)) {
-        rowHeaderCache.Add(variable, new Label() {
+        rowHeaderCache.Add(variable, new VerticalLabel() {
           Text = variable,
           TextAlign = ContentAlignment.MiddleCenter,
           Name = variable,
           Width = rowHeaderTableLayoutPanel.Width,
+          Height = columnHeaderScrollPanel.Width,
           Dock = DockStyle.Fill,
           Margin = new Padding(3)
@@ -269 +280 @@
         } else { //scatter plot
           var scatterPlot = Content.CreateScatterPlot(colVariable, rowVariable);
-          foreach (var row in scatterPlot.Rows)
+          var regressionType = (RegressionType)regressionTypeComboBox.SelectedValue;
+          int order = (int)polynomialRegressionOrderNumericUpDown.Value;
+          foreach (var row in scatterPlot.Rows) {
             row.VisualProperties.PointSize = 3;
+            row.VisualProperties.IsRegressionVisibleInLegend = false;
+            row.VisualProperties.RegressionType = regressionType;
+            row.VisualProperties.PolynomialRegressionOrder = order;
+          }
           scatterPlot.VisualProperties.Title = string.Empty;
-          var scatterPlotControl = new /*Preprocessing*/ScatterPlotControl {
+          var scatterPlotControl = new ScatterPlotControl {
             Name = key.ToString(),
             Content = scatterPlot,
@@ -279 +296 @@
             //XAxisFormat = "G3"
           };
-          scatterPlotControl.DoubleClick += ScatterPlotDoubleClick; // ToDo: not working; double click is already handled by the chart
+          //scatterPlotControl.DoubleClick += ScatterPlotDoubleClick; // ToDo: not working; double click is already handled by the chart
           bodyCache.Add(key, scatterPlotControl);
         }
@@ -348 +365 @@
         c++;
       }
+      UpdateRegressionLine();
     }
 
@@ -444 +462 @@
     }
     #endregion
+
+    #region Regression Line
+    private void regressionTypeComboBox_SelectedValueChanged(object sender, EventArgs e) {
+      var regressionType = (RegressionType)regressionTypeComboBox.SelectedValue;
+      polynomialRegressionOrderNumericUpDown.Enabled = regressionType == RegressionType.Polynomial;
+      UpdateRegressionLine();
+    }
+
+    private void polynomialRegressionOrderNumericUpDown_ValueChanged(object sender, EventArgs e) {
+      UpdateRegressionLine();
+    }
+
+    private void UpdateRegressionLine() {
+      var regressionType = (RegressionType)regressionTypeComboBox.SelectedValue;
+      int order = (int)polynomialRegressionOrderNumericUpDown.Value;
+
+      foreach (var control in bodyCache.Values) {
+        var scatterPlotControl = control as ScatterPlotControl;
+        if (scatterPlotControl != null) {
+          foreach (var row in scatterPlotControl.Content.Rows) {
+            row.VisualProperties.IsRegressionVisibleInLegend = false;
+            row.VisualProperties.RegressionType = regressionType;
+            row.VisualProperties.PolynomialRegressionOrder = order;
+          }
+        }
+      }
+    }
+    #endregion
   }
 }
+