Changeset 15997 for stable/HeuristicLab.Algorithms.DataAnalysis.Views/3.4/GradientBoostedTreesModelView.cs

Timestamp:

07/19/18 15:20:50 (6 years ago)

Author:

mkommend

Message:

#2890: Merged r15730, r15782 into stable.

Location:

stable

Files:

• . (modified) (1 prop)
• HeuristicLab.Algorithms.DataAnalysis.Views (modified) (1 prop)
• HeuristicLab.Algorithms.DataAnalysis.Views/3.4/GradientBoostedTreesModelView.cs (modified) (3 diffs)

stable/HeuristicLab.Algorithms.DataAnalysis.Views/3.4/GradientBoostedTreesModelView.cs

@@ -20 +20 @@
 #endregion
 
+using System;
+using System.Diagnostics;
+using System.Linq;
+using System.Windows.Forms;
 using HeuristicLab.Common;
 using HeuristicLab.Core.Views;
 using HeuristicLab.MainForm;
 using HeuristicLab.MainForm.WindowsForms;
-using HeuristicLab.Problems.DataAnalysis;                  
+using HeuristicLab.Problems.DataAnalysis;
 
 namespace HeuristicLab.Algorithms.DataAnalysis.Views {
@@ -30 +34 @@
   [Content(typeof(IGradientBoostedTreesModel), true)]
   public partial class GradientBoostedTreesModelView : ItemView {
-
+    #region Getter/Setter
     public new IGradientBoostedTreesModel Content {
       get { return (IGradientBoostedTreesModel)base.Content; }
       set { base.Content = value; }
     }
+    #endregion
 
-    protected override void SetEnabledStateOfControls() {
-      base.SetEnabledStateOfControls();
-      listBox.Enabled = Content != null;
-      viewHost.Enabled = Content != null;
-    }
-
+    #region Ctor
     public GradientBoostedTreesModelView()
       : base() {
       InitializeComponent();
+    }
+    #endregion
+
+    #region Events
+    protected override void SetEnabledStateOfControls() {
+      base.SetEnabledStateOfControls();
+      listView.Enabled = Content != null;
+      viewHost.Enabled = Content != null;
     }
 
@@ -51 +59 @@
       if (Content == null) {
         viewHost.Content = null;
-        listBox.Items.Clear();
+        listView.Items.Clear();
       } else {
         viewHost.Content = null;
-        listBox.Items.Clear();
-        foreach (var e in Content.Models) {
-          listBox.Items.Add(e);
-        }
+        listView.BeginUpdate();
+        listView.Items.Clear();
+        int i = 1;
+        listView.Items.AddRange(
+          new ListViewItem(Content.Models.First().ToString()) { Tag = Content.Models.First() }.ToEnumerable()
+          .Union(Content.Models.Skip(1).Select(v => new ListViewItem("Model " + i++) { Tag = v }))
+          .ToArray()
+        );
+        listView.EndUpdate();
       }
     }
 
-    private void listBox_SelectedIndexChanged(object sender, System.EventArgs e) {
-      var model = listBox.SelectedItem;
-      if (model == null) viewHost.Content = null;
-      else {
-        viewHost.Content = ConvertModel(model);
+
+    private void listView_SelectedIndexChanged(object sender, EventArgs e) {
+      if (listView.SelectedItems.Count == 1) {
+        var item = listView.SelectedItems[0];
+        viewHost.Content = ConvertModel(item);
       }
     }
 
-    private void listBox_DoubleClick(object sender, System.EventArgs e) {
-      var selectedItem = listBox.SelectedItem;
-      if (selectedItem == null) return;
-      MainFormManager.MainForm.ShowContent(ConvertModel(selectedItem));
-    }
-
-    private IContent ConvertModel(object model) {
-      var treeModel = model as RegressionTreeModel;
-      if (treeModel != null)
-        return treeModel.CreateSymbolicRegressionModel();
-      else {
-        var regModel = model as IRegressionModel;
-        return regModel;
+    private void listView_DoubleClick(object sender, EventArgs e) {
+      if (listView.SelectedItems.Count == 1) {
+        var item = listView.SelectedItems[0];
+        var content = ConvertModel(item);
+        if (content != null) { MainFormManager.MainForm.ShowContent(content); }
       }
     }
+    #endregion
+
+    #region Helper Methods
+    private IContent ConvertModel(ListViewItem item) {
+      if (item.Tag is RegressionTreeModel) {
+        return (item.Tag as RegressionTreeModel).CreateSymbolicRegressionModel();
+      } else if (item.Tag is IRegressionModel) {
+        return item.Tag as IRegressionModel;
+      } else {
+        return null;
+      }
+    }
+    #endregion
   }
 }