@Override public String visitDiceobject(@NotNull DiceGameParser.DiceobjectContext ctx) { if (ctx.NAME != null) { return "[dice for dice in self.dices if dice.name == '" + ctx.NAME.getText() + "'][0]"; } else if (ctx.POS != null) { return "self.dices[" + ctx.POS.getText() + "]"; } return "visitDiceobject"; }
@Override public String visitDiceinit(@NotNull DiceGameParser.DiceinitContext ctx) { StringBuilder result = new StringBuilder(); result.append("Dice('" + ctx.NAME.getText() + "',["); for (ParseTree val : ctx.children) { if (val.getClass() == DiceGameParser.FaceContext.class) { result.append(val.getText() + ", "); } } result.append("])"); return result.toString(); }
@Override public String visitGame(@NotNull DiceGameParser.GameContext ctx) { String imports = "#!/usr/bin/env python\n# -*- coding: utf-8 -*-\nimport random\n\n"; StringBuilder playerClass = new StringBuilder(); playerClass.append("class Player:\n"); playerClass.append( indent( "def __str__(self): return self.name + ', ' + ', '.join(map(str, [value for key, value in self.__dict__.items() if key not in ['name']]))\n")); StringBuilder playerInit = new StringBuilder(); playerInit.append("def __init__(self, name): "); playerInit.append("self.name = name; "); StringBuilder playerActive = new StringBuilder(); for (ParseTree init : ctx.children) { if (init.getClass() == DiceGameParser.PlayerinitContext.class) { if (((DiceGameParser.PlayerinitContext) init).ASSN != null) { playerInit.append(init.accept(this) + "; "); } else if (((DiceGameParser.PlayerinitContext) init).PLAYERACTIVECOND != null) { playerActive.append("def isActive(self): "); playerActive.append(init.accept(this) + "; "); } } } playerClass.append(indent(playerInit.toString()) + "\n"); playerClass.append(indent(playerActive.toString()) + "\n"); playerClass.append(indent("\n")); StringBuilder diceClass = new StringBuilder(); diceClass.append("class Dice:\n"); diceClass.append( indent( "def __init__(self, name, faces): self.name = name; self.faces = faces; self.roll()")); diceClass.append(indent("def roll(self): self.value = random.choice(self.faces)")); diceClass.append(indent("def __str__(self): return self.name+': '+str(self.value)")); diceClass.append(indent("\n")); StringBuilder gameClass = new StringBuilder(); gameClass.append("class Game:\n"); gameClass.append(indent("# Static methods")); gameClass.append( indent( "def status(self): return 'Status[name, ' + ', '.join([key for key, value in self.players[0].__dict__.items() if key not in ['name']]).lower()+']: '+' - '.join(map(str, self.players))")); gameClass.append( indent( "def rightPlayer(self): return self.players[ (self.players.index(self.activePlayer) - 1) % self.playerCount]")); gameClass.append( indent( "def leftPlayer(self): return self.players[ (self.players.index(self.activePlayer) + 1) % self.playerCount]")); gameClass.append(indent("def playerNum (self, num): return self.players[num]")); gameClass.append(indent("def rollDices(self): map(Dice.roll, self.dices)")); gameClass.append( indent( "def sortDices(self, desc=False): self.dices = sorted(self.dices, key=lambda dice: dice.value, reverse=desc)")); StringBuilder gameSetupMethod = new StringBuilder(); gameSetupMethod.append("def setup(self):\n"); gameSetupMethod.append( indent("self.playerCount = int(raw_input('Enter number of players: '))")); gameSetupMethod.append( indent( "if self.playerCount < self.min: print(self.name+' is made for more than '+str(self.min)+' players. Bring some friends ;-)'); exit()")); gameSetupMethod.append( indent( "if self.playerCount > self.max: print(str(self.playerCount)+' players? Thats too much for '+self.name+'... Maximum: '+str(self.max)); exit()")); gameSetupMethod.append(indent("")); gameSetupMethod.append( indent( "for p in range(self.playerCount): name = raw_input('Enter name of player #'+str(p)+': '); self.players.append(Player(name))")); gameSetupMethod.append(indent("self.activePlayer = self.players[0]")); gameSetupMethod.append(indent("print('Game initialized '+self.status())")); gameSetupMethod.append(indent("")); gameClass.append(indent(gameSetupMethod.toString())); StringBuilder gameInitMethod = new StringBuilder(); gameInitMethod.append("def __init__(self):\n"); gameInitMethod.append(indent("self.players = []")); gameInitMethod.append(indent("# Dynamic inits")); gameInitMethod.append(indent("self.name = '" + ctx.NAME.getText() + "'\n")); gameInitMethod.append(indent("")); boolean hasRunningCond = false; for (ParseTree init : ctx.children) { if (init.getClass() == DiceGameParser.GameinitContext.class) { DiceGameParser.GameinitContext initCtx = (DiceGameParser.GameinitContext) init; if (initCtx.COND != null) { gameClass.append(indent(initCtx.accept(this))); hasRunningCond = true; } else { gameInitMethod.append(indent(init.accept(this))); } } } if (!hasRunningCond) { gameClass.append(indent("def isRunning(self): return True")); } gameClass.append(indent(gameInitMethod.toString())); StringBuilder gameLoop = new StringBuilder(); gameLoop.append("while self.isRunning():\n"); gameLoop.append(indent("\nprint(self.status())")); gameLoop.append(indent("print('\\n'+self.activePlayer.name+' ist dran.'),")); gameLoop.append(indent("")); for (ParseTree init : ctx.children) { if (init.getClass() == DiceGameParser.ActionContext.class) { DiceGameParser.ActionContext loopCtx = (DiceGameParser.ActionContext) init; gameLoop.append(indent(loopCtx.accept(this))); } } String gameLoopMethod = "def loop(self):\n" + indent(gameLoop.toString()); gameClass.append(indent(gameLoopMethod)); StringBuilder main = new StringBuilder(); main.append("\nif __name__ == '__main__':\n"); main.append(indent("game = Game()")); main.append(indent("game.setup()")); main.append(indent("game.loop()")); main.append(indent("print('\\nSpiel beendet: '+game.status())")); main.append(indent(ctx.GAMEEND.accept(this))); return imports + playerClass.toString() + diceClass.toString() + gameClass.toString() + main.toString(); }