source: branches/HeuristicLab.EvolutionTracking/HeuristicLab.Problems.DataAnalysis.Symbolic.Views/3.4/SymbolicDataAnalysisExpressionLineageExplorerView.cs @ 10517

using System.Collections.Generic;
using System.Globalization;
using System.Linq;
using System.Windows.Forms;
using HeuristicLab.Encodings.SymbolicExpressionTreeEncoding;
using HeuristicLab.Encodings.SymbolicExpressionTreeEncoding.Views;
using HeuristicLab.EvolutionTracking;
using HeuristicLab.MainForm;
using HeuristicLab.MainForm.WindowsForms;
using HeuristicLab.Visualization;

namespace HeuristicLab.Problems.DataAnalysis.Symbolic.Views {
  [View("LineageExplorerView")]
  [Content(typeof(IGenealogyGraph), false)]
  public partial class SymbolicDataAnalysisExpressionLineageExplorerView : AsynchronousContentView {
    private readonly ReingoldTilfordLayoutEngine<ISymbolicExpressionTreeNode> symbolicExpressionTreeNodeLayoutEngine;
    private readonly SymbolicExpressionTreeLayoutAdapter symbolicExpressionTreeLayoutAdapter;

    private readonly Dictionary<TreeNode, ISymbolicExpressionTree> treeMap;

    private const int MinHorizontalSpacing = 50;
    private const int MinVerticalSpacing = 50;
    private const int PreferredNodeWidth = 80;
    private const int PreferredNodeHeight = 40;

    public new IGenealogyGraph Content {
      get { return (IGenealogyGraph)base.Content; }
      set { base.Content = value; }
    }

    public SymbolicDataAnalysisExpressionLineageExplorerView() {
      InitializeComponent();
      symbolicExpressionTreeNodeLayoutEngine = new ReingoldTilfordLayoutEngine<ISymbolicExpressionTreeNode> {
        MinHorizontalSpacing = MinHorizontalSpacing + PreferredNodeWidth,
        MinVerticalSpacing = MinVerticalSpacing + PreferredNodeHeight
      };
      symbolicExpressionTreeLayoutAdapter = new SymbolicExpressionTreeLayoutAdapter();
      treeMap = new Dictionary<TreeNode, ISymbolicExpressionTree>();
      double width = lineageExplorerChart.PreferredSize.Width;
      double height = lineageExplorerChart.PreferredSize.Height;
      if (lineageExplorerChart.Chart == null) {
        lineageExplorerChart.Chart = new Chart(0, 0, width, height);
      }
    }


    #region events
    protected override void OnContentChanged() {
      base.OnContentChanged();
      if (Content != null) { updateTreeView(); }
    }
    #endregion

    protected override void SetEnabledStateOfControls() {
      base.SetEnabledStateOfControls();
      treeView.Enabled = Content != null;
    }

    private void updateTreeView() {
      var lastGen = Content.Ranks.OrderBy(x => x.Key).Last().Value.ToList();
      treeMap.Clear();
      lineageExplorerChart.Clear();
      foreach (var g in lastGen) {
        var treeNode = new TreeNode(g.Quality.ToString(CultureInfo.InvariantCulture));
        treeView.Nodes.Add(treeNode);
        treeMap.Add(treeNode, (ISymbolicExpressionTree)g.Content);
      }
    }

    private void treeView_AfterSelect(object sender, TreeViewEventArgs e) {
      lineageExplorerChart.Clear();
      var selectedNode = treeView.SelectedNode;
      var symbolicExpressionTree = treeMap[selectedNode];
      var chart = lineageExplorerChart.Chart;
      var tile = new SymbolicExpressionTreeTile(chart) {
        LayoutAdapter = symbolicExpressionTreeLayoutAdapter,
        LayoutEngine = symbolicExpressionTreeNodeLayoutEngine,
        PreferredNodeWidth = PreferredNodeWidth,
        PreferredNodeHeight = PreferredNodeHeight,
        SymbolicExpressionTree = symbolicExpressionTree
      };
      lineageExplorerChart.UpdateEnabled = false;
      lineageExplorerChart.Add(tile);
      lineageExplorerChart.FlipVertical();
      lineageExplorerChart.UpdateEnabled = true;
      lineageExplorerChart.EnforceUpdate();
    }
  }
}