source: branches/PushGP/HeuristicLab.PushGP/HeuristicLab.Problems.ProgramSynthesis/Push/Problem/BenchmarkSuite/BenchmarkSuitePushSolutionView.cs @ 14952

namespace HeuristicLab.Problems.ProgramSynthesis.Push.Problem.BenchmarkSuite {
  using System;
  using System.Collections.Generic;
  using System.Drawing;
  using System.Globalization;
  using System.Linq;
  using System.Windows.Forms;
  using Core.Views;
  using HeuristicLab.BenchmarkSuite;
  using HeuristicLab.BenchmarkSuite.Views;
  using HeuristicLab.Common;

  using Interpreter;
  using MainForm;
  using Problem;
  using Stack;

  [View("Push Program Debugger")]
  [Content(typeof(PushSolution), true)]
  public partial class BenchmarkSuitePushSolutionView : NamedItemView {

    private const string Separator = ", ";

    private const string exampleSplitter = " => ";
    private const string AbsoluteDiffHeaderText = "Absolute Diff.";
    private const string RelativeDiffHeaderText = "Relative Diff.";
    private const string InputHeaderStringFormat = "Input {0} : {1}";
    private const string EstimatedOutputHeaderStringFormat = "Estimated Output {0} : {1}";
    private const string OutputHeaderStringFormat = "Output {0} : {1}";
    private static readonly Color correctExampleFontColor = Color.FromArgb(0x5cb85c);
    private PooledPushInterpreter interpreter;
    private PushInterpreterPool pool;

    public BenchmarkSuitePushSolutionView() {
      InitializeComponent();
      Name = "Push Program Debugger";
      InitEvents();
    }

    ~BenchmarkSuitePushSolutionView() {
      interpreter.Dispose();
    }

    private void InitEvents() {
      exampleComboBox.SelectedIndexChanged += SelectedExampleIndexChanged;
      pushDebugger.OnReset += PushDebuggerReset;
    }

    private void PushDebuggerReset(object sender, IPushInterpreter interpreter) {
      if (exampleComboBox.SelectedIndex < 0) return;

      var example = Evaluator.Data.Examples[exampleComboBox.SelectedIndex];

      interpreter.BooleanStack.Push(example.InputBoolean);
      interpreter.IntegerStack.Push(example.InputInteger);
      interpreter.FloatStack.Push(example.InputFloat);
      interpreter.CharStack.Push(example.InputChar);
      interpreter.StringStack.Push(example.InputString);
      interpreter.IntegerVectorStack.Push(example.InputIntegerVector.Select(x => x.ToList()).ToArray());
      interpreter.FloatVectorStack.Push(example.InputFloatVector.Select(x => x.ToList()).ToArray());
      interpreter.StringVectorStack.Push(example.InputStringVector.Select(x => x.ToList()).ToArray());
      interpreter.BooleanVectorStack.Push(example.InputBooleanVector.Select(x => x.ToList()).ToArray());
    }

    private void SelectedExampleIndexChanged(object sender, EventArgs e) {
      pushDebugger.ResetDebugging();
    }

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

    public PushBenchmarkSuiteEvaluator Evaluator
    {
      get { return (PushBenchmarkSuiteEvaluator)Content.Evaluator; }
    }

    protected override void OnContentChanged() {
      if (Content == null) return;

      Name = "Push Solution";
      nameTextBox.Text = Name;

      pool = new PushInterpreterPool(Content.Config);

      if (interpreter != null) {
        interpreter.Dispose();
      }

      interpreter = pool.Create(Content.Random);
      UpdateExamples(Evaluator.Data);

      if (exampleComboBox.SelectedIndex < 0) {
        exampleComboBox.SelectedIndex = 0; // Triggers ResetDebugging via event
      }

      InitResultGrid(trainingResultsDataGrid, Evaluator.DataBounds.TrainingRange.Start, Evaluator.DataBounds.TrainingRange.End);
      InitResultGrid(testResultsDataGrid, Evaluator.DataBounds.TestRange.Start, Evaluator.DataBounds.TestRange.End);

      pushDebugger.Content = Content;
    }

    private void InitResultGrid(DataGridView grid, int start, int end) {
      grid.Columns.Clear();
      grid.Rows.Clear();

      var cellTemplate = new DataGridViewTextBoxCell { Style = { WrapMode = DataGridViewTriState.True } };
      var data = Evaluator.Data;

      for (var i = 0; i < data.InputArgumentTypes.Length; i++) {
        var headerTypeName = ViewHelper.GetHeaderTypeName(data.InputArgumentTypes[i]);
        var column = new DataGridViewColumn {
          HeaderText = string.Format(InputHeaderStringFormat, i + 1, headerTypeName),
          AutoSizeMode = DataGridViewAutoSizeColumnMode.Fill,
          CellTemplate = cellTemplate,
          DefaultCellStyle = { Alignment = DataGridViewContentAlignment.MiddleRight }
        };

        grid.Columns.Add(column);
      }

      for (var i = 0; i < data.OutputArgumentTypes.Length; i++) {
        var headerTypeName = ViewHelper.GetHeaderTypeName(data.OutputArgumentTypes[i]);

        var estimatedOutputColumn = new DataGridViewColumn {
          HeaderText = string.Format(EstimatedOutputHeaderStringFormat, i + 1, headerTypeName),
          AutoSizeMode = DataGridViewAutoSizeColumnMode.Fill,
          CellTemplate = cellTemplate,
        };

        var outputColumn = new DataGridViewColumn {
          HeaderText = string.Format(OutputHeaderStringFormat, i + 1, headerTypeName),
          AutoSizeMode = DataGridViewAutoSizeColumnMode.Fill,
          CellTemplate = cellTemplate
        };

        estimatedOutputColumn.DefaultCellStyle.Alignment = DataGridViewContentAlignment.MiddleRight;
        outputColumn.DefaultCellStyle.Alignment = DataGridViewContentAlignment.MiddleRight;

        grid.Columns.Add(estimatedOutputColumn);
        grid.Columns.Add(outputColumn);
      }

      var absoluteDiffColumn = new DataGridViewColumn {
        HeaderText = AbsoluteDiffHeaderText,
        AutoSizeMode = DataGridViewAutoSizeColumnMode.ColumnHeader,
        CellTemplate = cellTemplate
      };

      var relativeDiffColumn = new DataGridViewColumn {
        HeaderText = RelativeDiffHeaderText,
        AutoSizeMode = DataGridViewAutoSizeColumnMode.ColumnHeader,
        CellTemplate = cellTemplate,
      };

      absoluteDiffColumn.DefaultCellStyle.Alignment = DataGridViewContentAlignment.MiddleRight;
      relativeDiffColumn.DefaultCellStyle.Alignment = DataGridViewContentAlignment.MiddleRight;

      grid.Columns.Add(absoluteDiffColumn);
      grid.Columns.Add(relativeDiffColumn);

      using (var pushInterpreter = pool.Create(Content.Random)) {
        var rowIndex = 1;
        for (var i = start; i < end; i++, rowIndex++) {
          var example = data.Examples[i];
          var row = new DataGridViewRow {
            HeaderCell = {
              Value = rowIndex.ToString(),
            }
          };

          row.CreateCells(grid);

          var absoluteDiff = Evaluator.Evaluate(pushInterpreter, Content.Program, i);
          var relativeDiff = absoluteDiff / Math.Abs(data.BestResult - data.WorstResult) * 100d;

          var inputArgumentCountDict = ViewHelper.CreateArgumentCountDict();
          for (var j = 0; j < data.InputArgumentTypes.Length; j++) {
            var type = data.InputArgumentTypes[j];
            var offset = inputArgumentCountDict[type];
            row.Cells[j].Value = ViewHelper.StringifyInput(type, offset, example, Separator);
            inputArgumentCountDict[type]++;
          }

          var outputArgumentCountDict = ViewHelper.CreateArgumentCountDict();
          for (var j = 0; j < data.OutputArgumentTypes.Length; j++) {
            var type = data.OutputArgumentTypes[j];
            var offset = outputArgumentCountDict[type];
            row.Cells[data.InputArgumentTypes.Length + j * 2].Value = ViewHelper.StringifyOutput(type, offset, example, Separator);
            row.Cells[data.InputArgumentTypes.Length + j * 2 + 1].Value = StringifyResult(type, offset, pushInterpreter, example, Separator);
            outputArgumentCountDict[type]++;
          }

          row.Cells[data.TotalArgumentCount + 1].Value = absoluteDiff.ToString("0.####");
          row.Cells[data.TotalArgumentCount + 2].Value = relativeDiff.ToString("0.##") + "%";

          if (absoluteDiff.IsAlmost(data.BestResult)) {
            for (var j = 0; j < row.Cells.Count; j++) {
              row.Cells[j].Style.ForeColor = correctExampleFontColor;
            }
          }

          grid.Rows.Add(row);
          pushInterpreter.Reset();
        }
      }
    }

    private string StringifyResult(ExampleArgumentType type, int offset, IPushInterpreter interpreter, Example example, string valueSeparator) {
      var emptyString = string.Empty;

      switch (type) {
        case ExampleArgumentType.Integer:
          return GetEntryAsString(offset, interpreter.IntegerStack);

        case ExampleArgumentType.IntegerVector:
          return GetVectorEntryAsString(offset, interpreter.IntegerVectorStack);

        case ExampleArgumentType.Float:
          return interpreter.FloatStack.Count > offset
            ? interpreter.FloatStack[offset].ToString(CultureInfo.CurrentUICulture)
            : emptyString;

        case ExampleArgumentType.FloatVector:
          return GetVectorEntryAsString(offset, interpreter.FloatVectorStack);

        case ExampleArgumentType.Boolean:
          return GetEntryAsString(offset, interpreter.BooleanStack);

        case ExampleArgumentType.Char:
          return GetEntryAsString(offset, interpreter.CharStack);

        case ExampleArgumentType.Print:
          var requiredLines = example.OutputPrint.Split(new[] { Environment.NewLine }, StringSplitOptions.None).Length;
          var count = Math.Min(requiredLines, interpreter.PrintStack.Count);
          return string.Join(valueSeparator, string.Join(Environment.NewLine, interpreter.PrintStack.Peek(count)));

        case ExampleArgumentType.String:
          return GetEntryAsString(offset, interpreter.StringStack);

        case ExampleArgumentType.StringVector:
          return GetVectorEntryAsString(offset, interpreter.StringVectorStack);

        default: return string.Empty;
      }
    }

    private static string GetEntryAsString<T>(int offset, IPushStack<T> stack) {
      return stack.Count > offset
       ? stack[offset].ToString()
       : string.Empty;
    }

    private static string GetVectorEntryAsString<T>(int offset, IPushStack<List<T>> vectorStack) {
      return vectorStack.Count > offset
       ? "[" + string.Join(",", vectorStack[offset]) + "]"
       : string.Empty;
    }

    private int GetCount<T>(IPushStack<T> stack, T[] data) {
      return Math.Max(0, Math.Min(data.Length, stack.Count));
    }

    private int GetVectorCount<T>(IPushStack<List<T>> stack, T[][] data) {
      return Math.Max(0, Math.Min(data.Length, stack.Count));
    }

    private void UpdateExamples(ProblemData data) {
      exampleComboBox.Items.Clear();
      if (data == null) return;

      var stringRepresentations = data.Examples.Select(e =>
        string.Join(Separator, e.InputArgs) +
        exampleSplitter +
        string.Join(Separator, e.OutputArgs));

      foreach (var str in stringRepresentations) {
        exampleComboBox.Items.Add(str);
      }
    }
  }
}