Changeset 10987 for branches/DataPreprocessing/HeuristicLab.DataPreprocessing

Timestamp:

06/11/14 14:23:19 (10 years ago)

Author:

aesterer

Message:

Refactored scatter plot

File:

• branches/DataPreprocessing/HeuristicLab.DataPreprocessing/3.4/Implementations/ScatterPlotContent.cs (modified) (2 diffs)

branches/DataPreprocessing/HeuristicLab.DataPreprocessing/3.4/Implementations/ScatterPlotContent.cs

@@ -20 +20 @@
 #endregion
 
+using System.Collections.Generic;
 using System.Drawing;
+using HeuristicLab.Analysis;
 using HeuristicLab.Common;
 using HeuristicLab.Core;
@@ -27 +29 @@
 
   [Item("ScatterPlot", "Represents a scatter plot.")]
-  public class ScatterPlotContent : PreprocessingChartContent {
+  public class ScatterPlotContent : Item, IViewChartShortcut {
+    public static new Image StaticItemImage {
+      get { return HeuristicLab.Common.Resources.VSImageLibrary.Performance; }
+    }
 
     public string SelectedXVariable { get; set; }
-
     public string SelectedYVariable { get; set; }
 
-    public ScatterPlotContent(IChartLogic chartlogic)
-      : base(chartlogic) {
+    public IPreprocessingContext Context {
+      get;
+      private set;
+    }
+    public ITransactionalPreprocessingData PreprocessingData {
+      get { return Context.Data; }
+    }
+
+    public ScatterPlotContent(IPreprocessingContext context)
+      : base() {
+      this.Context = context;
     }
 
     public ScatterPlotContent(ScatterPlotContent content, Cloner cloner)
       : base(content, cloner) {
-        this.SelectedXVariable = content.SelectedXVariable;
-        this.SelectedYVariable = content.SelectedYVariable;
+      this.Context = content.Context;
+      this.SelectedXVariable = content.SelectedXVariable;
+      this.SelectedYVariable = content.SelectedYVariable;
     }
-
-    public static new Image StaticItemImage {
-      get { return HeuristicLab.Common.Resources.VSImageLibrary.Performance; }
-    }
-
     public override IDeepCloneable Clone(Cloner cloner) {
       return new ScatterPlotContent(this, cloner);
     }
 
+    public DataRow CreateDataRow(string variableName, DataRowVisualProperties.DataRowChartType chartType) {
+      int columnIndex = PreprocessingData.GetColumnIndex(variableName);
+      IList<double> values = PreprocessingData.GetValues<double>(columnIndex);
+      DataRow row = new DataRow(variableName, "", values);
+      row.VisualProperties.ChartType = chartType;
+      return row;
+    }
+
+    public IEnumerable<string> GetVariableNames() {
+      List<string> doubleVariableNames = new List<string>();
+
+      //only return variable names from type double
+      foreach (string variableName in PreprocessingData.VariableNames) {
+        if (PreprocessingData.IsType<double>(PreprocessingData.GetColumnIndex(variableName)))
+          doubleVariableNames.Add(variableName);
+      }
+
+      return doubleVariableNames;
+    }
+
+    public ScatterPlot CreateScatterPlot(string variableNameX, string variableNameY) {
+      ScatterPlot scatterPlot = new ScatterPlot();
+
+      int xColumnIndex = PreprocessingData.GetColumnIndex(variableNameX);
+      int yColumnIndex = PreprocessingData.GetColumnIndex(variableNameY);
+
+      IList<double> xValues = PreprocessingData.GetValues<double>(xColumnIndex);
+      IList<double> yValues = PreprocessingData.GetValues<double>(yColumnIndex);
+
+      List<Point2D<double>> points = new List<Point2D<double>>();
+
+      for (int i = 0; i < xValues.Count; i++) {
+        Point2D<double> point = new Point2D<double>(xValues[i], yValues[i]);
+        points.Add(point);
+      }
+
+      ScatterPlotDataRow scdr = new ScatterPlotDataRow(variableNameX + " - " + variableNameY, "", points);
+      scatterPlot.Rows.Add(scdr);
+      return scatterPlot;
+    }
+
   }
 }