forked from s464965/WMICraft
121 lines
3.8 KiB
Python
121 lines
3.8 KiB
Python
from __future__ import annotations
|
|
from typing import List
|
|
|
|
from common.constants import ACTION, Direction, ROWS, COLUMNS
|
|
|
|
|
|
class State:
|
|
def __init__(self, row, column, direction):
|
|
self.row = row
|
|
self.column = column
|
|
self.direction = direction
|
|
|
|
|
|
class Node:
|
|
def __init__(self, state, parent=None, action=None):
|
|
self.state = state
|
|
self.parent = parent
|
|
self.action = action
|
|
|
|
|
|
def goal_test(goal_list, state: State):
|
|
if (state.row, state.column) in goal_list:
|
|
return True
|
|
return False
|
|
|
|
|
|
def get_successors(state: State, map):
|
|
successors = list()
|
|
|
|
state_left = State(state.row, state.column, state.direction.left())
|
|
successors.append((ACTION.get("rotate_left"), state_left))
|
|
|
|
state_right = State(state.row, state.column, state.direction.right())
|
|
successors.append((ACTION.get("rotate_right"), state_right))
|
|
|
|
target = go(state.row, state.column, state.direction)
|
|
|
|
if is_valid_move(map, target[0], target[1]):
|
|
state_go = State(target[0], target[1], state.direction)
|
|
successors.append((ACTION.get("go"), state_go))
|
|
|
|
return successors
|
|
|
|
|
|
def graphsearch(initial_state: State, map, goal_list, fringe: List[Node] = None, explored: List[Node] = None):
|
|
# fringe and explored initialization
|
|
if fringe is None:
|
|
fringe = list()
|
|
if explored is None:
|
|
explored = list()
|
|
explored_states = set()
|
|
fringe_states = set()
|
|
|
|
# train Node
|
|
fringe.append(Node(initial_state))
|
|
fringe_states.add((initial_state.row, initial_state.column, initial_state.direction))
|
|
|
|
while True:
|
|
# fringe empty -> solution not found
|
|
if not any(fringe):
|
|
print("Brak rozwiazania")
|
|
return []
|
|
|
|
# get first element from fringe
|
|
element = fringe.pop(0)
|
|
fringe_states.remove((element.state.row, element.state.column, element.state.direction))
|
|
|
|
# if solution was found, prepare and return actions sequence
|
|
if goal_test(goal_list, element.state):
|
|
actions_sequence = [element.action]
|
|
parent = element.parent
|
|
|
|
while parent is not None:
|
|
# train's action will be None, don't add it
|
|
if parent.action is not None:
|
|
actions_sequence.append(parent.action)
|
|
parent = parent.parent
|
|
|
|
actions_sequence.reverse()
|
|
return actions_sequence
|
|
|
|
# add current node to explored (prevents infinite cycles)
|
|
explored.append(element)
|
|
explored_states.add((element.state.row, element.state.column, element.state.direction))
|
|
|
|
# loop through every possible next action
|
|
for successor in get_successors(element.state, map):
|
|
|
|
# make sure not to fall into a cycle
|
|
successor_state = (successor[1].row, successor[1].column, successor[1].direction)
|
|
if successor_state not in fringe_states and successor_state not in explored_states:
|
|
# create new Node and add it at the end of fringe
|
|
new_node = Node(state=successor[1],
|
|
parent=element,
|
|
action=successor[0])
|
|
fringe.append(new_node)
|
|
fringe_states.add((new_node.state.row, new_node.state.column, new_node.state.direction))
|
|
|
|
|
|
# TEMPORARY METHOD
|
|
def go(row, column, direction):
|
|
target = tuple()
|
|
|
|
if direction == Direction.RIGHT:
|
|
target = row, column + 1
|
|
elif direction == Direction.LEFT:
|
|
target = row, column - 1
|
|
elif direction == Direction.UP:
|
|
target = row - 1, column
|
|
elif direction == Direction.DOWN:
|
|
target = row + 1, column
|
|
|
|
return target
|
|
|
|
|
|
def is_valid_move(map, target_row, target_column):
|
|
if 0 <= target_row < ROWS and 0 <= target_column < COLUMNS and map[target_row][target_column] in ['g', 's', ' ']:
|
|
return True
|
|
|
|
return False
|