Subversion Repositories programming

#!/usr/bin/env python

__author__    = "Ira W. Snyder (devel@irasnyder.com)"

__copyright__ = "Copyright (c) 2006, Ira W. Snyder (devel@irasnyder.com)"

__license__   = "GNU GPL v2 (or, at your option, any later version)"

from PyCompat import *

from Actions import *

from Menu import Menu

import sys

import copy

(WALK, PUSH, CLIMB, GRAB) = ('Walk', 'Push', 'Climb', 'Grab')

def findNextMove (State):

	"""Prioritized Search for the best next move"""

	monkeyPosition = Action(State).findMonkeyPosition ()

	boxPosition = Action(State).findBoxPosition ()

	# Try to Grab

	if Grab(State).meetsPreconditions():

		return (GRAB,)

	# Try to push box to middle

	if Push(State, boxPosition, Push.MIDDLE).meetsPreconditions():

		if boxPosition != Push.MIDDLE:

			return (PUSH, boxPosition, Push.MIDDLE)

	# Try to walk to box

	if Walk(State, monkeyPosition, boxPosition).meetsPreconditions():

		if monkeyPosition != boxPosition:

			return (WALK, monkeyPosition, boxPosition)

	# Try to climb the box

	if Climb(State, Climb.UP).meetsPreconditions():

		if not Action(State).monkeyOnBox ():

			return (CLIMB, Climb.UP)

	# Try to climb down from the box

	if Climb(State, Climb.DOWN).meetsPreconditions():

		if Action(State).monkeyOnBox ():

			return (CLIMB, Climb.DOWN)

def isValidInitialState (State):

	monkeyPos = Action(State).findMonkeyPosition()

	boxPos = Action(State).findBoxPosition()

	onBox = Action(State).monkeyOnBox()

	if onBox and (monkeyPos != boxPos):

		return False

	return True

def enter_initial_state ():

	State = set([])

	# Get Monkey Position

	m = Menu(autorun=True, name='Enter Monkey Position')

	m.add_entry ('1', 'Window', lambda: Walk.WINDOW)

	m.add_entry ('2', 'Door', lambda: Walk.DOOR)

	m.add_entry ('3', 'Middle', lambda: Walk.MIDDLE)

	m.add_entry ('Q', 'Quit', sys.exit)

	pos = m.run_menu()[1]

	State.add ('At(Monkey,%s)' % pos)

	# Get Box Position

	m = Menu(autorun=True, name='Enter Box Position')

	m.add_entry ('1', 'Window', lambda: Push.WINDOW)

	m.add_entry ('2', 'Door', lambda: Push.DOOR)

	m.add_entry ('3', 'Middle', lambda: Push.MIDDLE)

	m.add_entry ('Q', 'Quit', sys.exit)

	pos = m.run_menu()[1]

	State.add ('At(Box,%s)' % pos)

	# Get Monkey Status

	m = Menu(autorun=True, name='Enter Monkey Height Status')

	m.add_entry ('1', 'Monkey on Box', lambda: 'OnBox()')

	m.add_entry ('2', 'Monkey on Floor', lambda: 'OnFloor()')

	m.add_entry ('Q', 'Quit', sys.exit)

	State.add (m.run_menu()[1])

	# Get Banana Status

	m = Menu(autorun=True, name='Enter Banana Status')

	m.add_entry ('1', 'Monkey has Banana', lambda: 'HasBanana()')

	m.add_entry ('2', 'Monkey does not have Banana', lambda: 'NoBanana()')

	m.add_entry ('Q', 'Quit', sys.exit)

	State.add (m.run_menu()[1])

	return State

def find_plan (State):

	if not isValidInitialState (State):

		print 'Bad initial state\n'

		return

	if Action(State).isGoal():

		print 'Initial State is already a goal state\n'

		return

	print 'Plan:'

	while not Action(State).isGoal():

		next = findNextMove (State)

		action = next[0]

		if action == WALK:

			print '%s from %s to %s' % next

			State = Walk(State, next[1], next[2]).takeAction()

		elif action == PUSH:

			print '%s Box from %s to %s' % next

			State = Push(State, next[1], next[2]).takeAction()

		elif action == CLIMB:

			print '%s %s' % next

			State = Climb(State, next[1]).takeAction()

		elif action == GRAB:

			print '%s Banana' % next

			State = Grab(State).takeAction()

		else:

			print 'BAD STATE'

			return

	print

def main ():

	# Create the main menu

	m = Menu (name='Main Menu')

	m.add_entry ('1', 'Enter Initial State', enter_initial_state)

	m.add_entry ('2', 'Find Plan', find_plan)

	m.add_entry ('Q', 'Quit', sys.exit)

	m.hide_entry ('2')

	# Run the menu

	while True:

		(num, func) = m.run_menu ()

		if num == '1':

			State = func()

			m.unhide_entry ('2')

		elif num == '2':

			func(copy.deepcopy(State))

		else:

			func()

if __name__ == '__main__':

	main ()

# vim: set ts=4 sts=4 sw=4: