[10539] | 1 | #region License Information
|
---|
| 2 | /* HeuristicLab
|
---|
[14186] | 3 | * Copyright (C) 2002-2016 Heuristic and Evolutionary Algorithms Laboratory (HEAL)
|
---|
[10539] | 4 | *
|
---|
| 5 | * This file is part of HeuristicLab.
|
---|
| 6 | *
|
---|
| 7 | * HeuristicLab is free software: you can redistribute it and/or modify
|
---|
| 8 | * it under the terms of the GNU General Public License as published by
|
---|
| 9 | * the Free Software Foundation, either version 3 of the License, or
|
---|
| 10 | * (at your option) any later version.
|
---|
| 11 | *
|
---|
| 12 | * HeuristicLab is distributed in the hope that it will be useful,
|
---|
| 13 | * but WITHOUT ANY WARRANTY; without even the implied warranty of
|
---|
| 14 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
---|
| 15 | * GNU General Public License for more details.
|
---|
| 16 | *
|
---|
| 17 | * You should have received a copy of the GNU General Public License
|
---|
| 18 | * along with HeuristicLab. If not, see <http://www.gnu.org/licenses/>.
|
---|
| 19 | */
|
---|
| 20 | #endregion
|
---|
| 21 |
|
---|
[10987] | 22 | using System.Collections.Generic;
|
---|
[10539] | 23 | using System.Drawing;
|
---|
[13502] | 24 | using System.Linq;
|
---|
[10987] | 25 | using HeuristicLab.Analysis;
|
---|
[10539] | 26 | using HeuristicLab.Common;
|
---|
[10245] | 27 | using HeuristicLab.Core;
|
---|
[10242] | 28 |
|
---|
[10539] | 29 | namespace HeuristicLab.DataPreprocessing {
|
---|
[10709] | 30 |
|
---|
[10882] | 31 | [Item("ScatterPlot", "Represents a scatter plot.")]
|
---|
[10992] | 32 | public class ScatterPlotContent : PreprocessingChartContent {
|
---|
[10252] | 33 |
|
---|
[10915] | 34 | public string SelectedXVariable { get; set; }
|
---|
| 35 | public string SelectedYVariable { get; set; }
|
---|
[13502] | 36 | public string SelectedColorVariable { get; set; }
|
---|
[10915] | 37 |
|
---|
[10992] | 38 | public ScatterPlotContent(IFilteredPreprocessingData preprocessingData)
|
---|
| 39 | : base(preprocessingData) {
|
---|
[10252] | 40 | }
|
---|
| 41 |
|
---|
[10882] | 42 | public ScatterPlotContent(ScatterPlotContent content, Cloner cloner)
|
---|
[10539] | 43 | : base(content, cloner) {
|
---|
[10987] | 44 | this.SelectedXVariable = content.SelectedXVariable;
|
---|
| 45 | this.SelectedYVariable = content.SelectedYVariable;
|
---|
[13502] | 46 | this.SelectedColorVariable = content.SelectedColorVariable;
|
---|
[10245] | 47 | }
|
---|
[11001] | 48 |
|
---|
[13502] | 49 | public static new Image StaticItemImage {
|
---|
[11001] | 50 | get { return HeuristicLab.Common.Resources.VSImageLibrary.Performance; }
|
---|
| 51 | }
|
---|
| 52 |
|
---|
[10987] | 53 | public override IDeepCloneable Clone(Cloner cloner) {
|
---|
| 54 | return new ScatterPlotContent(this, cloner);
|
---|
| 55 | }
|
---|
[10245] | 56 |
|
---|
[13502] | 57 | public ScatterPlot CreateScatterPlot(string variableNameX, string variableNameY, string variableNameColor = "-") {
|
---|
[10987] | 58 | ScatterPlot scatterPlot = new ScatterPlot();
|
---|
| 59 |
|
---|
[10992] | 60 | IList<double> xValues = PreprocessingData.GetValues<double>(PreprocessingData.GetColumnIndex(variableNameX));
|
---|
| 61 | IList<double> yValues = PreprocessingData.GetValues<double>(PreprocessingData.GetColumnIndex(variableNameY));
|
---|
[13502] | 62 | if (variableNameColor == null || variableNameColor == "-") {
|
---|
| 63 | List<Point2D<double>> points = new List<Point2D<double>>();
|
---|
[10987] | 64 |
|
---|
[13502] | 65 | for (int i = 0; i < xValues.Count; i++) {
|
---|
| 66 | Point2D<double> point = new Point2D<double>(xValues[i], yValues[i]);
|
---|
| 67 | points.Add(point);
|
---|
| 68 | }
|
---|
[10987] | 69 |
|
---|
[13502] | 70 | ScatterPlotDataRow scdr = new ScatterPlotDataRow(variableNameX + " - " + variableNameY, "", points);
|
---|
| 71 | scatterPlot.Rows.Add(scdr);
|
---|
| 72 |
|
---|
| 73 | } else {
|
---|
| 74 | var colorValues = PreprocessingData.GetValues<double>(PreprocessingData.GetColumnIndex(variableNameColor));
|
---|
| 75 | var data = xValues.Zip(yValues, (x, y) => new { x, y }).Zip(colorValues, (v, c) => new { v.x, v.y, c }).ToList();
|
---|
| 76 | var gradients = ColorGradient.Colors;
|
---|
| 77 | int curGradient = 0;
|
---|
| 78 | int numColors = colorValues.Distinct().Count();
|
---|
| 79 | foreach (var colorValue in colorValues.Distinct()) {
|
---|
| 80 | var values = data.Where(x => x.c == colorValue);
|
---|
| 81 | var row = new ScatterPlotDataRow(
|
---|
| 82 | variableNameX + " - " + variableNameY + " (" + colorValue + ")",
|
---|
| 83 | "",
|
---|
| 84 | values.Select(v => new Point2D<double>(v.x, v.y)),
|
---|
| 85 | new ScatterPlotDataRowVisualProperties() { Color = gradients[curGradient] });
|
---|
| 86 | curGradient += gradients.Count / numColors;
|
---|
| 87 | scatterPlot.Rows.Add(row);
|
---|
| 88 | }
|
---|
[10987] | 89 | }
|
---|
| 90 | return scatterPlot;
|
---|
| 91 | }
|
---|
[10242] | 92 | }
|
---|
| 93 | }
|
---|