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

import random

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

class Possible:

	def __init__ (self, action, arg1=None, arg2=None):

		self.action = action

		self.arg1 = arg1

		self.arg2 = arg2

	def takeAction (self):

		pass

	def __eq__ (self, rhs):

		return (self.action == rhs.action) and (self.arg1 == rhs.arg1) \

				and (self.arg2 == rhs.arg2)

	def __repr__ (self):

		s = '(%s' % self.action

		if self.arg1:

			s += ', %s' % self.arg1

		if self.arg2:

			s += ', %s' % self.arg2

		return '%s)' % s

def generatePossibleMoves (State):

	moves = []

	# Grab, Priority 1

	if Grab(State).meetsPreconditions():

		moves.append ((1, Possible(GRAB)))

	# Push, Priority 2

	for f in LOCATIONS:

		for t in LOCATIONS:

			if Push(State, f, t).meetsPreconditions():

				moves.append ((2, Possible(PUSH, f, t)))

	# Climb, Priority 2

	for d in BOX_LOC:

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

			moves.append ((2, Possible(CLIMB, d)))

	# Walk, Priority 3

	for f in LOCATIONS:

		for t in LOCATIONS:

			if Walk(State, f, t).meetsPreconditions():

				moves.append ((3, Possible(WALK, f, t)))

	return moves

def chooseMove (moves, priority=None):

	possible = []

	# If priority was not set, choose something from the highest

	# priority that is available

	if not priority:

		if len(moves) > 0:

			moves.sort()

			priority = moves[0][0]

	for e in moves:

		if e[0] == priority:

			possible.append (e[1])

	# Make sure the list is not empty

	if len(possible) < 1:

		return False

	# Return a random choice from the possible moves of this priority

	return random.choice (possible)

# Return (loop_length, first_instruction_in_loop)

def loopDetector (plan_param):

	plan = copy.deepcopy (plan_param)

	plan.reverse()

	isloop = False

	for i in xrange(len(plan)):

		cur = plan[:i]

		rest= plan[i:]

		for i in xrange(len(cur)):

			isloop = True

			if cur[i] != rest[i]:

				isloop = False

				break

		if isloop:

			return (len(cur), cur[0])

	return False

def findNextMove (State):

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

	poss_moves = generatePossibleMoves(State)

	chose_move = chooseMove (poss_moves)

	#print 'POSSIBLE MOVES: ', poss_moves

	#print 'CHOOSE MOVE: ', chose_move

	return chose_move

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: WINDOW)

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

	m.add_entry ('3', 'Middle', lambda: 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: WINDOW)

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

	m.add_entry ('3', 'Middle', lambda: 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.action

		arg1 = next.arg1

		arg2 = next.arg2

		if action == WALK:

			print '%s from %s to %s' % (action, arg1, arg2)

			State = Walk(State, arg1, arg2).takeAction()

		elif action == PUSH:

			print '%s Box from %s to %s' % (action, arg1, arg2)

			State = Push(State, arg1, arg2).takeAction ()

		elif action == CLIMB:

			print '%s %s' % (action, arg1)

			State = Climb(State, arg1).takeAction()

		elif action == GRAB:

			print '%s Banana' % action

			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: