source: trunk/sources/HeuristicLab.Problems.DataAnalysis.Views/3.3/ScatterPlotView.cs @ 4068

Last change on this file since 4068 was 4068, checked in by swagner, 9 years ago

Sorted usings and removed unused usings in entire solution (#1094)

File size: 9.9 KB
Line 
1#region License Information
2/* HeuristicLab
3 * Copyright (C) 2002-2010 Heuristic and Evolutionary Algorithms Laboratory (HEAL)
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
21using System;
22using System.Collections.Generic;
23using System.Drawing;
24using System.Linq;
25using System.Windows.Forms;
26using System.Windows.Forms.DataVisualization.Charting;
27using HeuristicLab.MainForm;
28using HeuristicLab.MainForm.WindowsForms;
29
30namespace HeuristicLab.Problems.DataAnalysis.Views {
31  [View("Scatter Plot View")]
32  [Content(typeof(DataAnalysisSolution), true)]
33  public partial class ScatterPlotView : AsynchronousContentView {
34    private const string ALL_SERIES = "All Samples";
35    private const string TRAINING_SERIES = "Training Samples";
36    private const string TEST_SERIES = "Test Samples";
37
38    public new DataAnalysisSolution Content {
39      get { return (DataAnalysisSolution)base.Content; }
40      set { base.Content = value; }
41    }
42
43    public ScatterPlotView()
44      : base() {
45      InitializeComponent();
46
47      this.chart.Series.Add(ALL_SERIES);
48      this.chart.Series[ALL_SERIES].LegendText = ALL_SERIES;
49      this.chart.Series[ALL_SERIES].ChartType = SeriesChartType.FastPoint;
50
51      this.chart.Series.Add(TRAINING_SERIES);
52      this.chart.Series[TRAINING_SERIES].LegendText = TRAINING_SERIES;
53      this.chart.Series[TRAINING_SERIES].ChartType = SeriesChartType.FastPoint;
54      this.chart.Series[TRAINING_SERIES].Points.Add(1.0);
55
56      this.chart.Series.Add(TEST_SERIES);
57      this.chart.Series[TEST_SERIES].LegendText = TEST_SERIES;
58      this.chart.Series[TEST_SERIES].ChartType = SeriesChartType.FastPoint;
59
60      this.chart.TextAntiAliasingQuality = TextAntiAliasingQuality.High;
61      this.chart.AxisViewChanged += new EventHandler<System.Windows.Forms.DataVisualization.Charting.ViewEventArgs>(chart_AxisViewChanged);
62
63      //configure axis                 
64      this.chart.ChartAreas[0].AxisX.Title = "Estimated Values";
65      this.chart.ChartAreas[0].CursorX.IsUserSelectionEnabled = true;
66      this.chart.ChartAreas[0].AxisX.ScaleView.Zoomable = true;
67      this.chart.ChartAreas[0].CursorX.Interval = 1;
68      this.chart.ChartAreas[0].CursorY.Interval = 1;
69
70      this.chart.ChartAreas[0].AxisY.Title = "Target Values";
71      this.chart.ChartAreas[0].CursorY.IsUserSelectionEnabled = true;
72      this.chart.ChartAreas[0].AxisY.ScaleView.Zoomable = true;
73      this.chart.ChartAreas[0].AxisY.IsStartedFromZero = true;
74    }
75
76    protected override void RegisterContentEvents() {
77      base.RegisterContentEvents();
78      Content.EstimatedValuesChanged += new EventHandler(Content_EstimatedValuesChanged);
79      Content.ProblemDataChanged += new EventHandler(Content_ProblemDataChanged);
80    }
81    protected override void DeregisterContentEvents() {
82      base.DeregisterContentEvents();
83      Content.EstimatedValuesChanged -= new EventHandler(Content_EstimatedValuesChanged);
84      Content.ProblemDataChanged -= new EventHandler(Content_ProblemDataChanged);
85    }
86
87
88    private void Content_ProblemDataChanged(object sender, EventArgs e) {
89      UpdateChart();
90    }
91    private void Content_EstimatedValuesChanged(object sender, EventArgs e) {
92      UpdateSeries();
93    }
94
95    protected override void OnContentChanged() {
96      base.OnContentChanged();
97      UpdateChart();
98    }
99
100    private void UpdateChart() {
101      if (InvokeRequired) Invoke((Action)UpdateChart);
102      else {
103        if (Content != null) {
104          this.UpdateSeries();
105          if (!this.chart.Series.Any(s => s.Points.Count > 0))
106            this.ClearChart();
107        }
108      }
109    }
110
111    private void UpdateCursorInterval() {
112      var estimatedValues = this.chart.Series[ALL_SERIES].Points.Select(x => x.XValue).DefaultIfEmpty(1.0);
113      var targetValues = this.chart.Series[ALL_SERIES].Points.Select(x => x.YValues[0]).DefaultIfEmpty(1.0);
114      double estimatedValuesRange = estimatedValues.Max() - estimatedValues.Min();
115      double targetValuesRange = targetValues.Max() - targetValues.Min();
116      double interestingValuesRange = Math.Min(Math.Max(targetValuesRange, 1.0), Math.Max(estimatedValuesRange, 1.0));
117      double digits = (int)Math.Log10(interestingValuesRange) - 3;
118      double zoomInterval = Math.Max(Math.Pow(10, digits), 10E-5);
119      this.chart.ChartAreas[0].CursorX.Interval = zoomInterval;
120      this.chart.ChartAreas[0].CursorY.Interval = zoomInterval;
121    }
122
123
124    private void UpdateSeries() {
125      if (InvokeRequired) Invoke((Action)UpdateSeries);
126      else {
127        string targetVariableName = Content.ProblemData.TargetVariable.Value;
128        Dataset dataset = Content.ProblemData.Dataset;
129        int trainingStart = Content.ProblemData.TrainingSamplesStart.Value;
130        int trainingEnd = Content.ProblemData.TrainingSamplesEnd.Value;
131        int testStart = Content.ProblemData.TestSamplesStart.Value;
132        int testEnd = Content.ProblemData.TestSamplesEnd.Value;
133        if (this.chart.Series[ALL_SERIES].Points.Count > 0)
134          this.chart.Series[ALL_SERIES].Points.DataBindXY(Content.EstimatedValues.ToArray(), "",
135            dataset.GetVariableValues(targetVariableName), "");
136        if (this.chart.Series[TRAINING_SERIES].Points.Count > 0)
137          this.chart.Series[TRAINING_SERIES].Points.DataBindXY(Content.EstimatedTrainingValues.ToArray(), "",
138            dataset.GetVariableValues(targetVariableName, trainingStart, trainingEnd), "");
139        if (this.chart.Series[TEST_SERIES].Points.Count > 0)
140          this.chart.Series[TEST_SERIES].Points.DataBindXY(Content.EstimatedTestValues.ToArray(), "",
141            dataset.GetVariableValues(targetVariableName, testStart, testEnd), "");
142
143        double max = Math.Max(Content.EstimatedValues.Max(), dataset.GetVariableValues(targetVariableName).Max());
144        double min = Math.Min(Content.EstimatedValues.Min(), dataset.GetVariableValues(targetVariableName).Min());
145
146        max = Math.Ceiling(max) * 1.2;
147        min = Math.Floor(min) * 0.8;
148
149        this.chart.ChartAreas[0].AxisX.Maximum = max;
150        this.chart.ChartAreas[0].AxisX.Minimum = min;
151        this.chart.ChartAreas[0].AxisY.Maximum = max;
152        this.chart.ChartAreas[0].AxisY.Minimum = min;
153        UpdateCursorInterval();
154      }
155    }
156
157    private void ClearChart() {
158      this.chart.Series[ALL_SERIES].Points.Clear();
159      this.chart.Series[TRAINING_SERIES].Points.Clear();
160      this.chart.Series[TEST_SERIES].Points.Clear();
161    }
162
163    private void ToggleSeriesData(Series series) {
164      if (series.Points.Count > 0) {  //checks if series is shown
165        if (this.chart.Series.Any(s => s != series && s.Points.Count > 0)) {
166          series.Points.Clear();
167        }
168      } else if (Content != null) {
169        string targetVariableName = Content.ProblemData.TargetVariable.Value;
170        Dataset dataset = Content.ProblemData.Dataset;
171        int trainingStart = Content.ProblemData.TrainingSamplesStart.Value;
172        int trainingEnd = Content.ProblemData.TrainingSamplesEnd.Value;
173        int testStart = Content.ProblemData.TestSamplesStart.Value;
174        int testEnd = Content.ProblemData.TestSamplesEnd.Value;
175
176        IEnumerable<double> predictedValues = null;
177        IEnumerable<double> targetValues = null;
178        switch (series.Name) {
179          case ALL_SERIES:
180            predictedValues = Content.EstimatedValues;
181            targetValues = dataset.GetVariableValues(targetVariableName);
182            break;
183          case TRAINING_SERIES:
184            predictedValues = Content.EstimatedTrainingValues;
185            targetValues = dataset.GetVariableValues(targetVariableName, trainingStart, trainingEnd);
186            break;
187          case TEST_SERIES:
188            predictedValues = Content.EstimatedTestValues;
189            targetValues = dataset.GetVariableValues(targetVariableName, testStart, testEnd);
190            break;
191        }
192        series.Points.DataBindXY(predictedValues, "", targetValues, "");
193        this.chart.Legends[series.Legend].ForeColor = Color.Black;
194        UpdateCursorInterval();
195      }
196    }
197
198    private void chart_MouseDown(object sender, MouseEventArgs e) {
199      HitTestResult result = chart.HitTest(e.X, e.Y);
200      if (result.ChartElementType == ChartElementType.LegendItem) {
201        this.ToggleSeriesData(result.Series);
202      }
203    }
204
205    private void chart_MouseMove(object sender, MouseEventArgs e) {
206      HitTestResult result = chart.HitTest(e.X, e.Y);
207      if (result.ChartElementType == ChartElementType.LegendItem)
208        this.Cursor = Cursors.Hand;
209      else
210        this.Cursor = Cursors.Default;
211    }
212
213    private void chart_AxisViewChanged(object sender, System.Windows.Forms.DataVisualization.Charting.ViewEventArgs e) {
214      this.chart.ChartAreas[0].AxisX.ScaleView.Size = e.NewSize;
215      this.chart.ChartAreas[0].AxisY.ScaleView.Size = e.NewSize;
216    }
217
218    private void chart_CustomizeLegend(object sender, CustomizeLegendEventArgs e) {
219      e.LegendItems[0].Cells[1].ForeColor = this.chart.Series[ALL_SERIES].Points.Count == 0 ? Color.Gray : Color.Black;
220      e.LegendItems[1].Cells[1].ForeColor = this.chart.Series[TRAINING_SERIES].Points.Count == 0 ? Color.Gray : Color.Black;
221      e.LegendItems[2].Cells[1].ForeColor = this.chart.Series[TEST_SERIES].Points.Count == 0 ? Color.Gray : Color.Black;
222    }
223  }
224}
Note: See TracBrowser for help on using the repository browser.