Merge branch 'master' of hongabar.org:robotgame

This commit is contained in:
Sakse Dalum 2012-08-07 15:40:50 +02:00
commit b0062a5ba7
3 changed files with 115 additions and 9 deletions

View File

@ -52,3 +52,14 @@ class Left(Direction):
def next_pos(pos): def next_pos(pos):
x, y = pos x, y = pos
return x - 1, y return x - 1, y
succ = {Up: Right,
Right: Down,
Down: Left,
Left: Up}.__getitem__
pred = {Right: Up,
Down: Right,
Left: Down,
Up: Left}.__getitem__

View File

@ -22,13 +22,21 @@
# copyright : (C) 2012 Niels G. W. Serup # copyright : (C) 2012 Niels G. W. Serup
# maintained by : Niels G. W. Serup <ns@metanohi.name> # maintained by : Niels G. W. Serup <ns@metanohi.name>
"""Logic for rolling.""" """
Logic for a rolling stone on a playfield of movement-stopping stones and
direction-changing turns. Also has a pseudo-random playfield generator.
"""
from __future__ import print_function from __future__ import print_function
from robotgame.logic.direction import *
import random
class RollingStoneError(Exception): class RollingStoneError(Exception):
pass pass
class WouldHitWall(RollingStoneError):
pass
class Field(object): class Field(object):
def next_posdir(self): def next_posdir(self):
raise NotImplementedError raise NotImplementedError
@ -56,21 +64,34 @@ class Stone(Field):
return pos, direc return pos, direc
def step(playfield, pos, direc): def step(playfield, old_pos, old_direc):
field = _at(playfield, pos) """
Return a new (position, direction) tuple based on the location on the
playfield.
"""
field = _at(playfield, old_pos)
if field is not None: if field is not None:
return field.next_posdir(pos, direc) (x, y), direc = field.next_posdir(old_pos, old_direc)
return direc.next_pos(pos), direc else:
(x, y), direc = old_direc.next_pos(old_pos), old_direc
if x < 0 or x >= len(playfield[y]) or y < 0 or y >= len(playfield):
return old_pos, old_direc
return (x, y), direc
def reaches_goal(playfield, max_steps): def reaches_goal(playfield, max_steps):
"""
Determine if the rolling stone reaches the goal within range(max_steps).
"""
pos = _find_start(playfield) pos = _find_start(playfield)
direc = None direc = None
for i in range(max_steps): for i in range(max_steps):
pos, direc = step(playfield, pos, direc) new_pos, new_direc = step(playfield, pos, direc)
if isGoal(playfield, pos): if isGoal(playfield, pos):
return True return True
if isStone(playfield, pos): if new_pos == pos:
return False return False
pos, direc = new_pos, new_direc
return False return False
def _find_start(playfield): def _find_start(playfield):
@ -84,10 +105,80 @@ def _at(playfield, pos):
x, y = pos x, y = pos
return playfield[y][x] return playfield[y][x]
def _set(playfield, pos, val):
x, y = pos
playfield[y][x] = val
_is = lambda t: lambda playfield, pos: isinstance(_at(playfield, pos), t) _is = lambda t: lambda playfield, pos: isinstance(_at(playfield, pos), t)
(isGoal, isTurn, isStart, isStone) = (_is(Goal), _is(Turn), _is(Start), _is(Stone)) isGoal, isTurn, isStart, isStone = _is(Goal), _is(Turn), _is(Start), _is(Stone)
def generate_playfield(height, width, start_pos, start_direc, goal_pos, nstones, nturns=None):
"""
Generate a completable playfield.
The generated playfield will have nstones stones nturns turns. A
completable playfield will always be completable in either zero, one, or
two steps.
"""
playfield = [[None for i in range(width)] for i in range(height)]
_set(playfield, start_pos, Start(start_direc))
_set(playfield, goal_pos, Goal())
def _find_min_turns(from_pos, from_direc):
x0, y0 = from_pos
x2, y2 = goal_pos
turns = []
if from_direc in (Up, Left):
def get_turns(x0, y0, x2, y2):
if y0 == 0:
raise WouldHitWall
elif y0 < y2:
turns.append(((x0, y0 - 1), succ(succ(from_direc))))
turns.extend(_find_min_turns(*turns[-1]))
elif y0 > y2 and x0 != x2:
turns.append(((x0, y2), succ(from_direc) if x0 < x2 else pred(from_direc)))
elif y0 == y2 and x0 != x2:
turns.append(((x0, y0 - 1), succ(from_direc) if x0 < x2 else pred(from_direc)))
turns.append(((x2, y0 - 1), succ(succ(from_direc))))
return turns
if from_direc is Up:
turns = get_turns(x0, y0, x2, y2)
else:
turns = [((x, y), direc) for ((y, x), direc) in get_turns(y0, x0, y2, x2)]
else:
def get_turns(x0, y0, x2, y2):
if x0 > x2:
turns.append(((x0 + 1, y0), succ(succ(from_direc))))
turns.extend(_find_min_turns(*turns[-1]))
elif x0 < x2 and y0 != y2:
turns.append(((x2, y0), pred(from_direc) if y0 < y2 else succ(from_direc)))
elif x0 == x2 and y0 != y2:
turns.append(((x0 + 1, y0), pred(from_direc) if y0 < y2 else succ(from_direc)))
turns.append(((x0 + 1, y2), succ(succ(from_direc))))
return turns
if from_direc is Right:
if x0 == len(playfield[y0]):
raise WouldHitWall
turns = get_turns(x0, y0, x2, y2)
else:
if y0 == len(playfield):
raise WouldHitWall
turns = [((x, y), direc) for ((y, x), direc) in get_turns(y0, x0, y2, x2)]
return turns
def _randomize_path(turns):
pass
def _insert_stones(turns):
pass
turns = _find_min_turns(start_pos, start_direc)
if nturns is not None:
if len(turns) > nturns:
raise RollingStoneError("Too few steps allocated.")
_randomize_path(turns)
_insert_stones()
return playfield, 3
def generate_playfield():

View File

@ -23,5 +23,9 @@ class RollingStoneTest(unittest.TestCase):
self.assertTrue(reaches_goal(playfield_example_succeed, 100)) self.assertTrue(reaches_goal(playfield_example_succeed, 100))
self.assertFalse(reaches_goal(playfield_example_fail, 100)) self.assertFalse(reaches_goal(playfield_example_fail, 100))
def test_playfield_generation(self):
playfield, min_steps = generate_playfield(10, 10, (0, 0), Down, (9, 9), 10, 5)
# self.assertTrue(reaches_goal(playfield, min_steps))
if __name__ == '__main__': if __name__ == '__main__':
unittest.main() unittest.main()