source: branches/HeuristicLab.EvolutionTracking/HeuristicLab.Encodings.SymbolicExpressionTreeEncoding.Views/3.4/SymbolicExpressionTreeTile.cs @ 10514

#region License Information
/* HeuristicLab
 * Copyright (C) 2002-2014 Heuristic and Evolutionary Algorithms Laboratory (HEAL)
 *
 * This file is part of HeuristicLab.
 *
 * HeuristicLab is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * HeuristicLab is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with HeuristicLab. If not, see <http://www.gnu.org/licenses/>.
 */
#endregion

using System.Collections.Generic;
using System.Drawing;
using System.Linq;
using HeuristicLab.Visualization;
using HeuristicLab.Visualization.Primitives;
using Rectangle = HeuristicLab.Visualization.Rectangle;

namespace HeuristicLab.Encodings.SymbolicExpressionTreeEncoding.Views {
  public class SymbolicExpressionTreeTile : Group {
    public double PreferredNodeWidth { get; set; }
    public double PreferredNodeHeight { get; set; }
    private ISymbolicExpressionTree symbolicExpressionTree;
    public ISymbolicExpressionTree SymbolicExpressionTree {
      get { return symbolicExpressionTree; }
      set {
        symbolicExpressionTree = value;
        GeneratePrimitives(PreferredNodeWidth, PreferredNodeHeight);
      }
    }

    public ReingoldTilfordLayoutEngine<ISymbolicExpressionTreeNode> LayoutEngine { get; set; }
    public SymbolicExpressionTreeLayoutAdapter LayoutAdapter { get; set; }
    public SymbolicExpressionTreeTile(IChart chart) : base(chart) { }
    public SymbolicExpressionTreeTile(IChart chart, ISymbolicExpressionTree tree)
      : base(chart) {
      SymbolicExpressionTree = tree;
      PreferredNodeWidth = 80;
      PreferredNodeHeight = 40;
    }
    private void GeneratePrimitives(double preferredNodeWidth, double preferredNodeHeight) {
      Clear();
      var layoutNodes = LayoutAdapter.Convert(SymbolicExpressionTree).ToList();
      layoutNodes.RemoveAt(0);
      LayoutEngine.Root = layoutNodes[0];
      LayoutEngine.CalculateLayout();

      var primitivesMap = new Dictionary<LayoutNode<ISymbolicExpressionTreeNode>, LabeledPrimitive>(); // both Ellipse and Rectangle are derived from the RectangularPrimitiveBase
      var font = new Font(FontFamily.GenericSansSerif, 10, GraphicsUnit.Pixel);

      foreach (var node in layoutNodes) {
        LabeledPrimitive labeledPrimitive;
        if (node.Children == null || node.Children.Count == 0) {
          var rectangle = new Rectangle(Chart, node.X, node.Y, node.X + preferredNodeWidth, node.Y + preferredNodeHeight);
          var textPrimitive = new TextPrimitive(Chart, rectangle.LowerLeft, rectangle.UpperRight) {
            Font = font,
            Text = node.Content.ToString()
          };
          labeledPrimitive = new LabeledPrimitive(Chart, rectangle, textPrimitive);
        } else {
          var ellipse = new Ellipse(Chart, node.X, node.Y, node.X + preferredNodeWidth, node.Y + preferredNodeHeight);
          var textPrimitive = new TextPrimitive(Chart, ellipse.LowerLeft, ellipse.UpperRight) {
            Font = font,
            Text = node.Content.ToString()
          };
          labeledPrimitive = new LabeledPrimitive(Chart, ellipse, textPrimitive);
        }
        this.Add(labeledPrimitive);
        primitivesMap.Add(node, labeledPrimitive);
      }

      foreach (var node in layoutNodes.Where(n => n.Children != null)) {
        foreach (var child in node.Children) {
          var start = new PointD(node.X + preferredNodeWidth / 2, node.Y + preferredNodeHeight);
          var end = new PointD(child.X + preferredNodeWidth / 2, child.Y);
          var line = new Line(this.Chart, start, end);
          this.Add(line);
        }
      }
    }
  }
}