Free cookie consent management tool by TermsFeed Policy Generator

source: trunk/sources/HeuristicLab.Problems.DataAnalysis.Views/3.4/Classification/DiscriminantFunctionClassificationSolutionThresholdView.cs @ 9426

Last change on this file since 9426 was 8868, checked in by sforsten, 12 years ago

#1949:

  • TextAnnotations are now placed in an easier way
  • Y axis doesn't start at zero any more
File size: 13.3 KB
Line 
1#region License Information
2/* HeuristicLab
3 * Copyright (C) 2002-2012 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
21
22using System;
23using System.Collections.Generic;
24using System.Drawing;
25using System.Linq;
26using System.Windows.Forms;
27using System.Windows.Forms.DataVisualization.Charting;
28using HeuristicLab.Common;
29using HeuristicLab.MainForm;
30using HeuristicLab.MainForm.WindowsForms;
31
32namespace HeuristicLab.Problems.DataAnalysis.Views {
33  [View("Classification Threshold")]
34  [Content(typeof(IDiscriminantFunctionClassificationSolution), false)]
35  public sealed partial class DiscriminantFunctionClassificationSolutionThresholdView : DataAnalysisSolutionEvaluationView {
36    private const double TrainingAxisValue = 0.0;
37    private const double TestAxisValue = 10.0;
38    private const double TrainingTestBorder = (TestAxisValue - TrainingAxisValue) / 2;
39    private const string TrainingLabelText = "Training Samples";
40    private const string TestLabelText = "Test Samples";
41
42    public new IDiscriminantFunctionClassificationSolution Content {
43      get { return (IDiscriminantFunctionClassificationSolution)base.Content; }
44      set { base.Content = value; }
45    }
46
47    private Dictionary<double, Series> classValueSeriesMapping;
48    private Random random;
49    private bool updateInProgress;
50
51    public DiscriminantFunctionClassificationSolutionThresholdView()
52      : base() {
53      InitializeComponent();
54
55      classValueSeriesMapping = new Dictionary<double, Series>();
56      random = new Random();
57      updateInProgress = false;
58
59      this.chart.CustomizeAllChartAreas();
60      this.chart.ChartAreas[0].CursorX.IsUserSelectionEnabled = true;
61      this.chart.ChartAreas[0].AxisX.ScaleView.Zoomable = true;
62      this.chart.ChartAreas[0].AxisX.Minimum = TrainingAxisValue - TrainingTestBorder;
63      this.chart.ChartAreas[0].AxisX.Maximum = TestAxisValue + TrainingTestBorder;
64      AddCustomLabelToAxis(this.chart.ChartAreas[0].AxisX);
65
66      this.chart.ChartAreas[0].AxisY.Title = "Estimated Values";
67      this.chart.ChartAreas[0].AxisY.IsStartedFromZero = false;
68      this.chart.ChartAreas[0].CursorY.IsUserSelectionEnabled = true;
69      this.chart.ChartAreas[0].AxisX.ScaleView.Zoomable = true;
70    }
71
72    private void AddCustomLabelToAxis(Axis axis) {
73      CustomLabel trainingLabel = new CustomLabel();
74      trainingLabel.Text = TrainingLabelText;
75      trainingLabel.FromPosition = TrainingAxisValue - TrainingTestBorder;
76      trainingLabel.ToPosition = TrainingAxisValue + TrainingTestBorder;
77      axis.CustomLabels.Add(trainingLabel);
78
79      CustomLabel testLabel = new CustomLabel();
80      testLabel.Text = TestLabelText;
81      testLabel.FromPosition = TestAxisValue - TrainingTestBorder;
82      testLabel.ToPosition = TestAxisValue + TrainingTestBorder;
83      axis.CustomLabels.Add(testLabel);
84    }
85
86    protected override void RegisterContentEvents() {
87      base.RegisterContentEvents();
88      Content.ModelChanged += new EventHandler(Content_ModelChanged);
89      Content.ProblemDataChanged += new EventHandler(Content_ProblemDataChanged);
90    }
91    protected override void DeregisterContentEvents() {
92      base.DeregisterContentEvents();
93      Content.ModelChanged -= new EventHandler(Content_ModelChanged);
94      Content.ProblemDataChanged -= new EventHandler(Content_ProblemDataChanged);
95    }
96
97    private void Content_ProblemDataChanged(object sender, EventArgs e) {
98      UpdateChart();
99    }
100    private void Content_ModelChanged(object sender, EventArgs e) {
101      Content.Model.ThresholdsChanged += new EventHandler(Model_ThresholdsChanged);
102      UpdateChart();
103    }
104    private void Model_ThresholdsChanged(object sender, EventArgs e) {
105      AddThresholds();
106    }
107    protected override void OnContentChanged() {
108      base.OnContentChanged();
109      UpdateChart();
110    }
111
112    private void UpdateChart() {
113      if (InvokeRequired) Invoke((Action)UpdateChart);
114      else if (!updateInProgress) {
115        updateInProgress = true;
116        chart.Series.Clear();
117        classValueSeriesMapping.Clear();
118        if (Content != null) {
119          IEnumerator<string> classNameEnumerator = Content.ProblemData.ClassNames.GetEnumerator();
120          IEnumerator<double> classValueEnumerator = Content.ProblemData.ClassValues.OrderBy(x => x).GetEnumerator();
121          while (classNameEnumerator.MoveNext() && classValueEnumerator.MoveNext()) {
122            Series series = new Series(classNameEnumerator.Current);
123            series.ChartType = SeriesChartType.FastPoint;
124            series.Tag = classValueEnumerator.Current;
125            chart.Series.Add(series);
126            classValueSeriesMapping.Add(classValueEnumerator.Current, series);
127            FillSeriesWithDataPoints(series);
128          }
129          AddThresholds();
130        }
131        chart.ChartAreas[0].RecalculateAxesScale();
132        updateInProgress = false;
133      }
134    }
135
136    private void FillSeriesWithDataPoints(Series series) {
137      List<double> estimatedValues = Content.EstimatedValues.ToList();
138      var targetValues = Content.ProblemData.Dataset.GetDoubleValues(Content.ProblemData.TargetVariable).ToList();
139
140      foreach (int row in Content.ProblemData.TrainingIndices) {
141        double estimatedValue = estimatedValues[row];
142        double targetValue = targetValues[row];
143        if (targetValue.IsAlmost((double)series.Tag)) {
144          double jitterValue = random.NextDouble() * 2.0 - 1.0;
145          DataPoint point = new DataPoint();
146          point.XValue = TrainingAxisValue + 0.01 * jitterValue * JitterTrackBar.Value * (TrainingTestBorder * 0.9);
147          point.YValues[0] = estimatedValue;
148          point.Tag = new KeyValuePair<double, double>(TrainingAxisValue, jitterValue);
149          series.Points.Add(point);
150        }
151      }
152
153      foreach (int row in Content.ProblemData.TestIndices) {
154        double estimatedValue = estimatedValues[row];
155        double targetValue = targetValues[row];
156        if (targetValue.IsAlmost((double)series.Tag)) {
157          double jitterValue = random.NextDouble() * 2.0 - 1.0;
158          DataPoint point = new DataPoint();
159          point.XValue = TestAxisValue + 0.01 * jitterValue * JitterTrackBar.Value * (TrainingTestBorder * 0.9);
160          point.YValues[0] = estimatedValue;
161          point.Tag = new KeyValuePair<double, double>(TestAxisValue, jitterValue);
162          series.Points.Add(point);
163        }
164      }
165
166      UpdateCursorInterval();
167    }
168
169    private void AddThresholds() {
170      chart.Annotations.Clear();
171      int classIndex = 1;
172      IClassificationProblemData problemData = Content.ProblemData;
173      var classValues = Content.Model.ClassValues.ToArray();
174      Axis y = chart.ChartAreas[0].AxisY;
175      Axis x = chart.ChartAreas[0].AxisX;
176      string name;
177      foreach (double threshold in Content.Model.Thresholds) {
178        if (!double.IsInfinity(threshold)) {
179          HorizontalLineAnnotation annotation = new HorizontalLineAnnotation();
180          annotation.AllowMoving = true;
181          annotation.AllowResizing = false;
182          annotation.LineWidth = 2;
183          annotation.LineColor = Color.Red;
184          annotation.IsInfinitive = true;
185          annotation.ClipToChartArea = chart.ChartAreas[0].Name;
186          annotation.Tag = classIndex;  //save classIndex as Tag to avoid moving the threshold accross class bounderies
187          annotation.AxisX = chart.ChartAreas[0].AxisX;
188          annotation.AxisY = y;
189          annotation.Y = threshold;
190
191          name = problemData.GetClassName(classValues[classIndex - 1]);
192          TextAnnotation beneathLeft = CreateTextAnnotation(name, classIndex, x, y, x.Minimum, threshold, ContentAlignment.TopLeft);
193          TextAnnotation beneathRight = CreateTextAnnotation(name, classIndex, x, y, x.Maximum, threshold, ContentAlignment.TopRight);
194
195          name = problemData.GetClassName(classValues[classIndex]);
196          TextAnnotation aboveLeft = CreateTextAnnotation(name, classIndex, x, y, x.Minimum, threshold, ContentAlignment.BottomLeft);
197          TextAnnotation aboveRight = CreateTextAnnotation(name, classIndex, x, y, x.Maximum, threshold, ContentAlignment.BottomRight);
198
199          chart.Annotations.Add(annotation);
200          chart.Annotations.Add(beneathLeft);
201          chart.Annotations.Add(aboveLeft);
202          chart.Annotations.Add(beneathRight);
203          chart.Annotations.Add(aboveRight);
204
205          beneathLeft.ResizeToContent();
206          beneathRight.ResizeToContent();
207          aboveLeft.ResizeToContent();
208          aboveRight.ResizeToContent();
209
210          beneathRight.Width = -beneathRight.Width;
211          aboveLeft.Height = -aboveLeft.Height;
212          aboveRight.Height = -aboveRight.Height;
213          aboveRight.Width = -aboveRight.Width;
214
215          classIndex++;
216        }
217      }
218    }
219
220    private TextAnnotation CreateTextAnnotation(string name, int classIndex, Axis axisX, Axis axisY, double x, double y, ContentAlignment alignment) {
221      TextAnnotation annotation = new TextAnnotation();
222      annotation.Text = name;
223      annotation.AllowMoving = true;
224      annotation.AllowResizing = false;
225      annotation.AllowSelecting = false;
226      annotation.IsSizeAlwaysRelative = true;
227      annotation.ClipToChartArea = chart.ChartAreas[0].Name;
228      annotation.Tag = classIndex;
229      annotation.AxisX = axisX;
230      annotation.AxisY = axisY;
231      annotation.Alignment = alignment;
232      annotation.X = x;
233      annotation.Y = y;
234      return annotation;
235    }
236
237    private void JitterTrackBar_ValueChanged(object sender, EventArgs e) {
238      foreach (Series series in chart.Series) {
239        foreach (DataPoint point in series.Points) {
240          double value = ((KeyValuePair<double, double>)point.Tag).Key;
241          double jitterValue = ((KeyValuePair<double, double>)point.Tag).Value; ;
242          point.XValue = value + 0.01 * jitterValue * JitterTrackBar.Value * (TrainingTestBorder * 0.9);
243        }
244      }
245    }
246
247    private void chart_CustomizeLegend(object sender, CustomizeLegendEventArgs e) {
248      foreach (LegendItem legendItem in e.LegendItems) {
249        var series = chart.Series[legendItem.SeriesName];
250        if (series != null) {
251          bool seriesIsInvisible = series.Points.Count == 0;
252          foreach (LegendCell cell in legendItem.Cells)
253            cell.ForeColor = seriesIsInvisible ? Color.Gray : Color.Black;
254        }
255      }
256    }
257
258    private void chart_MouseMove(object sender, MouseEventArgs e) {
259      HitTestResult result = chart.HitTest(e.X, e.Y);
260      if (result.ChartElementType == ChartElementType.LegendItem)
261        this.Cursor = Cursors.Hand;
262      else
263        this.Cursor = Cursors.Default;
264    }
265
266    private void ToggleSeries(Series series) {
267      if (series.Points.Count == 0)
268        FillSeriesWithDataPoints(series);
269      else
270        series.Points.Clear();
271    }
272
273    private void chart_MouseDown(object sender, MouseEventArgs e) {
274      HitTestResult result = chart.HitTest(e.X, e.Y);
275      if (result.ChartElementType == ChartElementType.LegendItem) {
276        if (result.Series != null) ToggleSeries(result.Series);
277      }
278    }
279
280    private void chart_AnnotationPositionChanging(object sender, AnnotationPositionChangingEventArgs e) {
281      int classIndex = (int)e.Annotation.Tag;
282      double[] thresholds = Content.Model.Thresholds.ToArray();
283      thresholds[classIndex] = e.NewLocationY;
284      Array.Sort(thresholds);
285      Content.Model.SetThresholdsAndClassValues(thresholds, Content.Model.ClassValues);
286    }
287
288    private void UpdateCursorInterval() {
289      Series series = chart.Series[0];
290      double[] xValues = (from point in series.Points
291                          where !point.IsEmpty
292                          select point.XValue)
293                    .DefaultIfEmpty(1.0)
294                    .ToArray();
295      double[] yValues = (from point in series.Points
296                          where !point.IsEmpty
297                          select point.YValues[0])
298                    .DefaultIfEmpty(1.0)
299                    .ToArray();
300
301      double xRange = xValues.Max() - xValues.Min();
302      double yRange = yValues.Max() - yValues.Min();
303      if (xRange.IsAlmost(0.0)) xRange = 1.0;
304      if (yRange.IsAlmost(0.0)) yRange = 1.0;
305      double xDigits = (int)Math.Log10(xRange) - 3;
306      double yDigits = (int)Math.Log10(yRange) - 3;
307      double xZoomInterval = Math.Pow(10, xDigits);
308      double yZoomInterval = Math.Pow(10, yDigits);
309      this.chart.ChartAreas[0].CursorX.Interval = xZoomInterval;
310      this.chart.ChartAreas[0].CursorY.Interval = yZoomInterval;
311    }
312  }
313}
Note: See TracBrowser for help on using the repository browser.