source: branches/Robocode.TrunkInt/HeuristicLab.Problems.Robocode/3.3/Grammar.cs @ 10011

#region License Information
/* HeuristicLab
 * Copyright (C) 2002-2013 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 HeuristicLab.Common;
using HeuristicLab.Core;
using HeuristicLab.Encodings.SymbolicExpressionTreeEncoding;
using HeuristicLab.Persistence.Default.CompositeSerializers.Storable;

namespace HeuristicLab.Problems.Robocode {
  [StorableClass]
  [Item("Robocode Grammar", "The grammar for the Robocode GP problem.")]
  public class Grammar : SymbolicExpressionGrammar {
    [StorableConstructor]
    protected Grammar(bool deserializing) : base(deserializing) { }
    protected Grammar(Grammar original, Cloner cloner) : base(original, cloner) { }

    public Grammar()
      : base("Robocode Grammar", "The grammar for the Robocode GP problem.") {
      Initialize();
    }

    public override IDeepCloneable Clone(Cloner cloner) {
      return new Grammar(this, cloner);
    }

    // initialize set of allowed symbols and define 
    // the allowed combinations of symbols
    private void Initialize() {
      #region Symbols
      var block = new Block();
      var stat = new Stat();
      var ifStat = new IfStat();
      var elseStat = new ElseStat();
      var whileStat = new WhileStat();

      var logicalExpr = new LogicalExpression();
      var numericalExpr = new NumericalExpression();

      var equal = new Equal();
      var lessThan = new LessThan();
      var lessThanOrEqual = new LessThanOrEqual();
      var greaterThan = new GreaterThan();
      var greaterThanOrEqual = new GreaterThanOrEqual();

      var conjunction = new Conjunction();
      var disjunction = new Disjunction();
      var negation = new Negation();

      var addition = new Addition();
      var subtraction = new Subtraction();
      var multiplication = new Multiplication();
      var division = new Division();
      var modulus = new Modulus();

      var number = new Number();
      var logicalVal = new LogicalValue();

      var ahead = new Ahead();
      var back = new Back();
      var fire = new Fire();
      var shotPower = new ShotPower();

      var getEnergy = new GetEnergy();
      var getGunHeading = new GetGunHeading();
      var getHeading = new GetHeading();
      var getRadarHeading = new GetRadarHeading();
      var getX = new GetX();
      var getY = new GetY();

      //var setAdjustGunForRobotTurn = new SetAdjustGunForRobotTurn();
      //var setAdjustRadarForGunTurn = new SetAdjustRadarForGunTurn();
      //var setAdjustRadarForRobotTurn = new SetAdjustRadarForRobotTurn();
      //var independent = new Independent();

      var turnLeft = new TurnLeft();
      var turnRight = new TurnRight();
      var turnGunLeft = new TurnGunLeft();
      var turnGunRight = new TurnGunRight();
      var turnRadarLeft = new TurnRadarLeft();
      var turnRadarRight = new TurnRadarRight();

      var onBulletHit = new OnBulletHit();
      var onBulletMissed = new OnBulletMissed();
      var onHitByBullet = new OnHitByBullet();
      var onHitRobot = new OnHitRobot();
      var onHitWall = new OnHitWall();
      var onScannedRobot = new OnScannedRobot();

      var run = new Run();
      var tank = new Tank();

      var doNothing = new DoNothing();
      var emptyEvent = new EmptyEvent();
      #endregion

      #region Symbol Collections
      var eventSymbols = new ISymbol[] { onScannedRobot, onBulletHit, onBulletMissed, onHitByBullet, onHitRobot, onHitWall };
      var controlSymbols = new ISymbol[] { ifStat, whileStat };
      var functionSymbols = new ISymbol[] {
        ahead, back, fire, turnGunLeft, turnGunRight, turnLeft, turnRadarLeft, turnRadarRight, turnRight
        //setAdjustGunForRobotTurn, setAdjustRadarForGunTurn, setAdjustRadarForRobotTurn, 
      };
      var relationalOperators = new ISymbol[] { equal, lessThan, lessThanOrEqual, greaterThan, greaterThanOrEqual };
      var logicalOperators = new ISymbol[] { conjunction, disjunction, negation };
      var numericalExpressions = new ISymbol[] { number, getEnergy, getGunHeading, getHeading, getRadarHeading, getX, getY };
      var numericalOperators = new ISymbol[] { addition, subtraction, multiplication, division, modulus };
      #endregion

      #region Adding Symbols
      foreach (var s in eventSymbols)
        AddSymbol(s);
      foreach (var s in controlSymbols)
        AddSymbol(s);
      foreach (var s in functionSymbols)
        AddSymbol(s);
      foreach (var s in relationalOperators)
        AddSymbol(s);
      foreach (var s in logicalOperators)
        AddSymbol(s);
      foreach (var s in numericalExpressions)
        AddSymbol(s);
      foreach (var s in numericalOperators)
        AddSymbol(s);

      AddSymbol(block);
      AddSymbol(stat);
      AddSymbol(elseStat);
      AddSymbol(shotPower);
      AddSymbol(logicalVal);
      AddSymbol(logicalExpr);
      AddSymbol(numericalExpr);
      AddSymbol(run);
      AddSymbol(tank);
      AddSymbol(emptyEvent);
      AddSymbol(doNothing);
      #endregion

      #region Grammar Definition
      // StartSymbol
      AddAllowedChildSymbol(StartSymbol, tank);

      // Tank
      AddAllowedChildSymbol(tank, run, 0);
      AddAllowedChildSymbol(tank, onScannedRobot, 1);
      AddAllowedChildSymbol(tank, onBulletHit, 2);
      AddAllowedChildSymbol(tank, onBulletMissed, 3);
      AddAllowedChildSymbol(tank, onHitByBullet, 4);
      AddAllowedChildSymbol(tank, onHitRobot, 5);
      AddAllowedChildSymbol(tank, onHitWall, 6);

      // Run
      AddAllowedChildSymbol(run, stat);

      // Event
      foreach (var s in eventSymbols)
        AddAllowedChildSymbol(s, stat);

      // Block
      AddAllowedChildSymbol(block, stat);

      // Stat
      AddAllowedChildSymbol(stat, stat);
      AddAllowedChildSymbol(stat, block);
      foreach (var s in controlSymbols)
        AddAllowedChildSymbol(stat, s);
      foreach (var s in functionSymbols)
        AddAllowedChildSymbol(stat, s);
      AddAllowedChildSymbol(stat, emptyEvent);
      AddAllowedChildSymbol(stat, doNothing);

      // IfStat
      AddAllowedChildSymbol(ifStat, logicalExpr, 0);
      AddAllowedChildSymbol(ifStat, stat, 1);
      AddAllowedChildSymbol(ifStat, emptyEvent, 1);
      AddAllowedChildSymbol(ifStat, doNothing, 1);
      AddAllowedChildSymbol(ifStat, elseStat, 2);
      AddAllowedChildSymbol(ifStat, emptyEvent, 2);
      AddAllowedChildSymbol(ifStat, doNothing, 2);

      // ElseStat
      AddAllowedChildSymbol(elseStat, stat);
      AddAllowedChildSymbol(elseStat, emptyEvent);
      AddAllowedChildSymbol(elseStat, doNothing);

      // WhileStat
      AddAllowedChildSymbol(whileStat, logicalExpr, 0);
      AddAllowedChildSymbol(whileStat, stat, 1);
      AddAllowedChildSymbol(whileStat, emptyEvent, 1);
      AddAllowedChildSymbol(whileStat, doNothing, 1);

      // Numerical Expressions
      foreach (var s in numericalExpressions)
        AddAllowedChildSymbol(numericalExpr, s);
      foreach (var s in numericalOperators) {
        AddAllowedChildSymbol(numericalExpr, s);
        foreach (var ne in numericalExpressions)
          AddAllowedChildSymbol(s, ne);
        foreach (var no in numericalOperators) {
          AddAllowedChildSymbol(s, no);
        }
      }

      // Logical Expressions
      AddAllowedChildSymbol(logicalExpr, logicalVal);
      foreach (var s in logicalOperators) {
        AddAllowedChildSymbol(logicalExpr, s);
        AddAllowedChildSymbol(s, logicalVal);
        foreach (var lo in logicalOperators)
          AddAllowedChildSymbol(s, lo);
        foreach (var ro in relationalOperators)
          AddAllowedChildSymbol(s, ro);
      }
      foreach (var s in relationalOperators)
        AddAllowedChildSymbol(s, numericalExpr);

      // Functions
      foreach (var f in functionSymbols) {
        if (f is Fire)
          AddAllowedChildSymbol(f, shotPower);
        else
          AddAllowedChildSymbol(f, numericalExpr);
      }
      #endregion
    }
  }
}