Graphsearch A*, heurystyka, action cost
This commit is contained in:
parent
a5a2820eca
commit
9ebb4a0f15
@ -1,4 +1,4 @@
|
||||
<?xml version="1.0" encoding="UTF-8"?>
|
||||
<project version="4">
|
||||
<component name="ProjectRootManager" version="2" project-jdk-name="Python 3.7 (venv)" project-jdk-type="Python SDK" />
|
||||
<component name="ProjectRootManager" version="2" project-jdk-name="Python 3.8" project-jdk-type="Python SDK" />
|
||||
</project>
|
@ -2,7 +2,7 @@
|
||||
<module type="PYTHON_MODULE" version="4">
|
||||
<component name="NewModuleRootManager">
|
||||
<content url="file://$MODULE_DIR$" />
|
||||
<orderEntry type="jdk" jdkName="Python 3.7 (venv)" jdkType="Python SDK" />
|
||||
<orderEntry type="inheritedJdk" />
|
||||
<orderEntry type="sourceFolder" forTests="false" />
|
||||
</component>
|
||||
</module>
|
@ -1,11 +1,15 @@
|
||||
from resources.Globals import *
|
||||
|
||||
images_coord_list = []
|
||||
cell_expense_list = []
|
||||
|
||||
|
||||
class Node:
|
||||
def __init__(self):
|
||||
self.state = State()
|
||||
self.parent = None
|
||||
self.parent = []
|
||||
self.action = ""
|
||||
self.priority = 0
|
||||
|
||||
|
||||
class State:
|
||||
@ -14,6 +18,14 @@ class State:
|
||||
self.direction = ""
|
||||
|
||||
|
||||
def init_data(coord_list, expense_list):
|
||||
global images_coord_list
|
||||
global cell_expense_list
|
||||
|
||||
images_coord_list = coord_list
|
||||
cell_expense_list = expense_list
|
||||
|
||||
|
||||
def successor(state):
|
||||
|
||||
node_state_left = Node()
|
||||
@ -98,11 +110,86 @@ def successor(state):
|
||||
return [node_state_left, node_state_right]
|
||||
|
||||
|
||||
def graph_search(fringe, explored, start_state, end_state_coord):
|
||||
def get_cell_expense(node):
|
||||
global images_coord_list
|
||||
global cell_expense_list
|
||||
|
||||
for i in range(0, len(images_coord_list)):
|
||||
if (images_coord_list[i][0] <= node.state.coord[0] and node.state.coord[0] <= images_coord_list[i][0] + IMAGE_SIZE) and (images_coord_list[i][1] <= node.state.coord[1] and node.state.coord[1] <= images_coord_list[i][1] + IMAGE_SIZE):
|
||||
return cell_expense_list[i]
|
||||
|
||||
|
||||
def heurystyka(node_now):
|
||||
|
||||
if node_now.action == "Left" or node_now.action == "Right":
|
||||
return node_now.parent[2] + (get_cell_expense(node_now) / 2)
|
||||
elif node_now.action == "Up":
|
||||
return node_now.parent[2] + (get_cell_expense(node_now) * 2)
|
||||
elif node_now.action == "":
|
||||
return get_cell_expense(node_now)
|
||||
|
||||
|
||||
# def graph_search(fringe, explored, start_state, end_state_coord):
|
||||
#
|
||||
# node = Node()
|
||||
# node.state = start_state
|
||||
# node.parent = node.state
|
||||
# fringe.append(node)
|
||||
# iterator = 0
|
||||
#
|
||||
# end_loop = True
|
||||
# while end_loop:
|
||||
# if len(fringe) == 0:
|
||||
# end_loop = False
|
||||
# #return False
|
||||
#
|
||||
# elem = fringe[iterator]
|
||||
#
|
||||
# if elem.state.coord == end_state_coord:
|
||||
# return fringe
|
||||
#
|
||||
# explored.append(elem)
|
||||
#
|
||||
# another_states = successor(elem.state)
|
||||
# for i in range(0, len(another_states)):
|
||||
# n = len(fringe)
|
||||
# for j in range(0, n):
|
||||
# if another_states[i].state.coord[0] == fringe[j].state.coord[0] and another_states[i].state.coord[1] == fringe[j].state.coord[1]:
|
||||
# if another_states[i].state.direction == fringe[j].state.direction:
|
||||
# break
|
||||
# else:
|
||||
# states = []
|
||||
# for k in range(0, len(fringe)):
|
||||
# new_state = [fringe[k].state.coord, fringe[k].state.direction]
|
||||
# states.append(new_state)
|
||||
# now_state = [another_states[i].state.coord, another_states[i].state.direction]
|
||||
# if now_state in states:
|
||||
# break
|
||||
#
|
||||
# another_states[i].parent = elem.state
|
||||
# fringe.append(another_states[i])
|
||||
# else:
|
||||
# states = []
|
||||
# for k in range(0, len(fringe)):
|
||||
# new_state = [fringe[k].state.coord, fringe[k].state.direction]
|
||||
# states.append(new_state)
|
||||
# now_state = [another_states[i].state.coord, another_states[i].state.direction]
|
||||
#
|
||||
# if now_state in states:
|
||||
# break
|
||||
#
|
||||
# if another_states[i].state.direction == fringe[j].state.direction:
|
||||
# another_states[i].parent = elem.state
|
||||
# fringe.append(another_states[i])
|
||||
# iterator += 1
|
||||
|
||||
|
||||
def graph_search_A(fringe, explored, start_state, end_state_coord):
|
||||
node = Node()
|
||||
node.state = start_state
|
||||
node.parent = node.state
|
||||
node.priority = heurystyka(node)
|
||||
node.parent = [node.state.coord, node.state.direction, node.priority]
|
||||
|
||||
fringe.append(node)
|
||||
iterator = 0
|
||||
|
||||
@ -110,7 +197,7 @@ def graph_search(fringe, explored, start_state, end_state_coord):
|
||||
while end_loop:
|
||||
if len(fringe) == 0:
|
||||
end_loop = False
|
||||
#return False
|
||||
# return False
|
||||
|
||||
elem = fringe[iterator]
|
||||
|
||||
@ -121,10 +208,15 @@ def graph_search(fringe, explored, start_state, end_state_coord):
|
||||
|
||||
another_states = successor(elem.state)
|
||||
for i in range(0, len(another_states)):
|
||||
another_states[i].parent = [elem.state.coord, elem.state.direction, elem.priority]
|
||||
p = heurystyka(another_states[i])
|
||||
|
||||
n = len(fringe)
|
||||
for j in range(0, n):
|
||||
if another_states[i].state.coord[0] == fringe[j].state.coord[0] and another_states[i].state.coord[1] == fringe[j].state.coord[1]:
|
||||
if another_states[i].state.direction == fringe[j].state.direction:
|
||||
if another_states[i].state.direction == fringe[j].state.direction and p < fringe[j].priority:
|
||||
another_states[i].priority = p
|
||||
fringe[j] = another_states[i]
|
||||
break
|
||||
else:
|
||||
states = []
|
||||
@ -133,10 +225,30 @@ def graph_search(fringe, explored, start_state, end_state_coord):
|
||||
states.append(new_state)
|
||||
now_state = [another_states[i].state.coord, another_states[i].state.direction]
|
||||
if now_state in states:
|
||||
break
|
||||
index = states.index(now_state)
|
||||
if p < fringe[index].priority:
|
||||
another_states[i].priority = p
|
||||
fringe[index] = another_states[i]
|
||||
break
|
||||
else:
|
||||
break
|
||||
|
||||
another_states[i].parent = elem.state
|
||||
another_states[i].priority = p
|
||||
fringe.append(another_states[i])
|
||||
|
||||
n1 = len(fringe)
|
||||
|
||||
while n1 > 1:
|
||||
change = False
|
||||
for l in range(0, n1 - 1):
|
||||
if fringe[l].priority > fringe[l + 1].priority:
|
||||
fringe[l], fringe[l + 1] = fringe[l + 1], fringe[l]
|
||||
change = True
|
||||
|
||||
n1 -= 1
|
||||
|
||||
if not change:
|
||||
break
|
||||
else:
|
||||
states = []
|
||||
for k in range(0, len(fringe)):
|
||||
@ -145,9 +257,29 @@ def graph_search(fringe, explored, start_state, end_state_coord):
|
||||
now_state = [another_states[i].state.coord, another_states[i].state.direction]
|
||||
|
||||
if now_state in states:
|
||||
break
|
||||
index = states.index(now_state)
|
||||
if p < fringe[index].priority:
|
||||
another_states[i].priority = p
|
||||
fringe[index] = another_states[i]
|
||||
break
|
||||
else:
|
||||
break
|
||||
|
||||
if another_states[i].state.direction == fringe[j].state.direction:
|
||||
another_states[i].parent = elem.state
|
||||
another_states[i].priority = p
|
||||
fringe.append(another_states[i])
|
||||
|
||||
n2 = len(fringe)
|
||||
|
||||
while n2 > 1:
|
||||
change = False
|
||||
for h in range(0, n2 - 1):
|
||||
if fringe[h].priority > fringe[h + 1].priority:
|
||||
fringe[h], fringe[h + 1] = fringe[h + 1], fringe[h]
|
||||
change = True
|
||||
|
||||
n2 -= 1
|
||||
|
||||
if not change:
|
||||
break
|
||||
iterator += 1
|
||||
|
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
Binary file not shown.
@ -28,6 +28,7 @@ field = Field()
|
||||
fringe = []
|
||||
explored = []
|
||||
action_list = []
|
||||
images_coord = []
|
||||
|
||||
|
||||
def Arrow(direction):
|
||||
@ -46,9 +47,16 @@ def Arrow(direction):
|
||||
|
||||
# Putting images
|
||||
def Fill(bool):
|
||||
global images_coord
|
||||
if bool:
|
||||
field.PuttingSmallImages()
|
||||
|
||||
for i in range(0, len(field.canvas_small_images)):
|
||||
images_coord.append(field.small_field_canvas.coords(field.canvas_small_images[i]))
|
||||
print("Coords List: ", images_coord)
|
||||
|
||||
nd.init_data(images_coord, field.cell_expense)
|
||||
|
||||
# Drawing red/green rectangles
|
||||
for el in field.state_of_cell_array:
|
||||
if el[0] != 0:
|
||||
@ -171,7 +179,7 @@ def create_action_list(states, index):
|
||||
action_list.reverse()
|
||||
return True
|
||||
action_list.append(fringe[index].action)
|
||||
state_parent = [fringe[index].parent.coord, fringe[index].parent.direction]
|
||||
state_parent = [fringe[index].parent[0], fringe[index].parent[1]]
|
||||
create_action_list(states, states.index(state_parent))
|
||||
|
||||
|
||||
@ -189,6 +197,7 @@ def MouseClickEvent(event):
|
||||
img_coords = field.small_field_canvas.coords(field.canvas_small_images[i])
|
||||
if (img_coords[0] <= event.x and event.x <= img_coords[0] + IMAGE_SIZE) and (img_coords[1] <= event.y and event.y <= img_coords[1] + IMAGE_SIZE):
|
||||
end_position = img_coords
|
||||
print("Color cost: ", field.cell_expense[i])
|
||||
|
||||
# if len(end_position) == 2:
|
||||
# print("Koncowa pozycja: {} {}".format(end_position[0], end_position[1]))
|
||||
@ -202,8 +211,7 @@ def MouseClickEvent(event):
|
||||
for k in range(0, len(fringe)):
|
||||
new_state = fringe[k].state.coord
|
||||
states.append(new_state)
|
||||
index = states.index(start_position)
|
||||
start_node = fringe[index]
|
||||
start_node = fringe[-1]
|
||||
|
||||
node.state.coord = start_node.state.coord
|
||||
node.state.direction = start_node.state.direction
|
||||
@ -211,21 +219,28 @@ def MouseClickEvent(event):
|
||||
fringe.clear()
|
||||
explored.clear()
|
||||
action_list.clear()
|
||||
|
||||
fringe = nd.graph_search(fringe, explored, node.state, end_position)
|
||||
fringe = nd.graph_search_A(fringe, explored, node.state, end_position)
|
||||
# fringe = nd.graph_search(fringe, explored, node.state, end_position)
|
||||
|
||||
states = []
|
||||
goal_all = []
|
||||
for i in range(0, len(fringe)):
|
||||
new_state = [fringe[i].state.coord, fringe[i].state.direction]
|
||||
states.append(new_state)
|
||||
if end_position[0] == fringe[i].state.coord[0] and end_position[1] == fringe[i].state.coord[1]:
|
||||
fringe = fringe[:i + 1]
|
||||
break
|
||||
goal_all.append(fringe[i])
|
||||
|
||||
elem_min = goal_all[0]
|
||||
for i in range(1, len(goal_all)):
|
||||
if elem_min.priority > goal_all[i].priority:
|
||||
elem_min = goal_all[i]
|
||||
index = fringe.index(elem_min)
|
||||
fringe = fringe[:index + 1]
|
||||
|
||||
create_action_list(states, -1)
|
||||
|
||||
# for i in range(0, len(fringe)):
|
||||
# print('Node{} = State: {} {}, Parent: {} {}, Action: {}'.format(i + 1, fringe[i].state.coord, fringe[i].state.direction, fringe[i].parent.coord, fringe[i].parent.direction, fringe[i].action))
|
||||
# print('Node{} = State: {} {}, Parent: {} {} {}, Action: {}'.format(i + 1, fringe[i].state.coord, fringe[i].state.direction, fringe[i].parent[0], fringe[i].parent[1], fringe[i].parent[2], fringe[i].action))
|
||||
|
||||
print(action_list)
|
||||
|
||||
@ -310,25 +325,26 @@ def DrawRectangle():
|
||||
y = 4
|
||||
|
||||
color = None
|
||||
|
||||
# Chose color for rectangle
|
||||
for i in range(len(field.cell_expense)):
|
||||
if field.cell_expense[i] == 10:
|
||||
color = "None"
|
||||
elif field.cell_expense[i] == 20:
|
||||
color = "yellow"
|
||||
elif field.cell_expense[i] == 30:
|
||||
color = "dodger blue"
|
||||
elif field.cell_expense[i] == 40:
|
||||
color = "dodger blue"
|
||||
elif field.cell_expense[i] == 80:
|
||||
color = "green4"
|
||||
if color != "None":
|
||||
field.small_field_canvas.create_rectangle(x, y, x + IMAGE_SIZE + 2, y + IMAGE_SIZE + 2, width=2, outline=color)
|
||||
x += player.step
|
||||
if i > 0 and i % 10 == 0:
|
||||
if x + IMAGE_SIZE + 2 > field.width:
|
||||
x = 4
|
||||
y += player.step
|
||||
|
||||
|
||||
|
||||
|
||||
def AddCostCellsToArray(amount, cost):
|
||||
counter = 0
|
||||
while counter < amount:
|
||||
|
@ -19,12 +19,10 @@ amount_of_sand_cells = 10
|
||||
sand_cell_cost = 20
|
||||
|
||||
amount_of_water_cells = 10
|
||||
water_cell_cost = 30
|
||||
water_cell_cost = 40
|
||||
|
||||
amount_of_swamp_cells = 10
|
||||
swamp_cell_cost = 40
|
||||
|
||||
|
||||
swamp_cell_cost = 80
|
||||
|
||||
x_start = 5
|
||||
y_start = 5
|
||||
|
BIN
resources/__pycache__/Globals.cpython-38.pyc
Normal file
BIN
resources/__pycache__/Globals.cpython-38.pyc
Normal file
Binary file not shown.
BIN
resources/__pycache__/__init__.cpython-38.pyc
Normal file
BIN
resources/__pycache__/__init__.cpython-38.pyc
Normal file
Binary file not shown.
Loading…
Reference in New Issue
Block a user