source: branches/PushGP/HeuristicLab.PushGP/HeuristicLab.Problems.ProgramSynthesis/Push/Expressions/YankExpressions.cs @ 14834

namespace HeuristicLab.Problems.ProgramSynthesis.Push.Expressions {
  using System.Collections.Generic;

  using HeuristicLab.Problems.ProgramSynthesis.Push.Attributes;
  using HeuristicLab.Problems.ProgramSynthesis.Push.Interpreter;
  using HeuristicLab.Problems.ProgramSynthesis.Push.Stack;

  /// <summary>
  ///     Removes an indexed item from "deep" in the stack and pushes it on top of the stack.
  ///     The index is taken from the INTEGER stack, and the indexing is done after the index is removed.
  /// </summary>
  /// <typeparam name="T">Stacktype</typeparam>
  public abstract class YankExpression<T> : StatelessExpression {
    public bool Eval(IPushStack<T> stack, IPushStack<long> integerStack, int count = 1) {
      if (integerStack.Count == 0) return false;

      var index = integerStack.Top;

      if ((index > stack.Count - count) || (index < 0) || (stack.Count < 2)) return false;

      integerStack.Pop();
      stack.Yank((int)index);

      return true;
    }
  }

  [PushExpression(StackTypes.Integer, "INTEGER.YANK")]
  public class IntegerYankExpression : YankExpression<long> {
    public override bool Eval(IInternalPushInterpreter interpreter) {
      return this.Eval(interpreter.IntegerStack, interpreter.IntegerStack, 2);
    }
  }

  [PushExpression(StackTypes.Float, "FLOAT.YANK", StackTypes.Integer)]
  public class FloatYankExpression : YankExpression<double> {
    public override bool Eval(IInternalPushInterpreter interpreter) {
      return this.Eval(interpreter.FloatStack, interpreter.IntegerStack);
    }
  }

  [PushExpression(StackTypes.Boolean, "BOOLEAN.YANK", StackTypes.Integer)]
  public class BooleanYankExpression : YankExpression<bool> {
    public override bool Eval(IInternalPushInterpreter interpreter) {
      return this.Eval(interpreter.BooleanStack, interpreter.IntegerStack);
    }
  }

  [PushExpression(StackTypes.Name, "NAME.YANK", StackTypes.Integer)]
  public class NameYankExpression : YankExpression<string> {
    public override bool Eval(IInternalPushInterpreter interpreter) {
      return this.Eval(interpreter.NameStack, interpreter.IntegerStack);
    }
  }

  [PushExpression(StackTypes.Exec, "EXEC.YANK", StackTypes.Integer)]
  public class ExecYankExpression : YankExpression<Expression> {
    public override bool Eval(IInternalPushInterpreter interpreter) {
      return this.Eval(interpreter.ExecStack, interpreter.IntegerStack);
    }
  }

  [PushExpression(StackTypes.Code, "CODE.YANK", StackTypes.Integer)]
  public class CodeYankExpression : YankExpression<Expression> {
    public override bool Eval(IInternalPushInterpreter interpreter) {
      return this.Eval(interpreter.CodeStack, interpreter.IntegerStack);
    }
  }

  [PushExpression(StackTypes.Char, "CHAR.YANK", StackTypes.Integer)]
  public class CharYankExpression : YankExpression<char> {
    public override bool Eval(IInternalPushInterpreter interpreter) {
      return Eval(interpreter.CharStack, interpreter.IntegerStack);
    }
  }

  [PushExpression(StackTypes.String, "STRING.YANK", StackTypes.Integer)]
  public class StringYankExpression : YankExpression<string> {
    public override bool Eval(IInternalPushInterpreter interpreter) {
      return Eval(interpreter.StringStack, interpreter.IntegerStack);
    }
  }

  [PushExpression(StackTypes.IntegerVector, "INTEGER[].YANK", StackTypes.Integer)]
  public class IntegerVectorYankExpression : YankExpression<List<long>> {
    public override bool Eval(IInternalPushInterpreter interpreter) {
      return Eval(interpreter.IntegerVectorStack, interpreter.IntegerStack);
    }
  }
}