bfs.py

@@ -1,15 +1,21 @@
 from succ import succ as successors
 
-def bfs(istate, goalstate):
+def bfs(istate, goalx, goaly):
     fringe = [istate]
     explored = []
+    steps = []
     while(fringe):
         state = fringe.pop(0)
-        if state == goalstate : 
+        if state.xpos == goalx and state.ypos == goaly: 
-                return 
+                steps.insert(0, state)
+                while(state.parent != None):
+                    state = state.parent
+                    steps.insert(0, state)
+                return steps
+        
         element = successors(state)
         explored.append(state)
         for value in element :
-            if value not in explored :
+            if value not in explored and value not in fringe:
                 fringe.append(value)
     return False

main.py

@@ -1,5 +1,8 @@
 import pygame
 import random
+from bfs import bfs
+from state import State
+import time
 
 pygame.init()
 WIDTH, HEIGHT = 800, 800
@@ -17,6 +20,10 @@ COBBLE_IMG = pygame.image.load("cobble.jpeg")
 COBBLE = pygame.transform.scale(COBBLE_IMG, (50, 50))
 FPS = 10
 
+class Agent:
+    def __init__(self, rect, direction):
+        self.rect = rect
+        self.direction = direction
 
 def randomize_map():        # tworzenie mapy z losowymi polami
     fields_list = [DIRT, GRASS, SAND, COBBLE]
@@ -34,37 +41,57 @@ def draw_window(agent, fields):
     for i in range(16):
         for j in range(16):
             window.blit(fields[i][j], (i * 50, j * 50))
-    window.blit(AGENT, (agent.x, agent.y))      # wyswietlanie agenta
+    window.blit(AGENT, (agent.rect.x, agent.rect.y))      # wyswietlanie agenta
     pygame.display.update()
 
 
-def agent_movement(keys_pressed, agent):    # sterowanie
+#def agent_movement(keys_pressed, agent):    # sterowanie
-    if keys_pressed[pygame.K_LEFT] and agent.x > 0:
+#    if keys_pressed[pygame.K_LEFT] and agent.x > 0:
-        agent.x -= 50
+#        agent.x -= 50
-    if keys_pressed[pygame.K_RIGHT] and agent.x < 750:
+#    if keys_pressed[pygame.K_RIGHT] and agent.x < 750:
-        agent.x += 50
+#        agent.x += 50
-    if keys_pressed[pygame.K_UP] and agent.y > 0:
+#    if keys_pressed[pygame.K_UP] and agent.y > 0:
-        agent.y -= 50
+#        agent.y -= 50
-    if keys_pressed[pygame.K_DOWN] and agent.y < 750:
+#    if keys_pressed[pygame.K_DOWN] and agent.y < 750:
-        agent.y += 50
+#        agent.y += 50
 
 
 def main():
     clock = pygame.time.Clock()
     run = True
-    agent = pygame.Rect(0, 0, 50, 50)   # tworzenie pola dla agenta
+    agent = Agent(pygame.Rect(0, 0, 50, 50), 0)   # tworzenie pola dla agenta
     fields = randomize_map()
     while run:
         clock.tick(FPS)
         for event in pygame.event.get():    # przechwycanie zamknięcia okna
             if event.type == pygame.QUIT:
                 run = False
-        keys_pressed = pygame.key.get_pressed()
+        #keys_pressed = pygame.key.get_pressed()
-        draw_window(agent, fields)
+        
-        agent_movement(keys_pressed, agent)
+        steps = bfs(State(None, None, 0, 0, 'E'), 450, 100)
+        for interm in steps:
+            if interm.action == 'LEFT':
+                agent.direction = (agent.direction - 1) % 4
+            if interm.action == 'RIGHT':
+                agent.direction = (agent.direction + 1) % 4
+            if interm.action == 'FORWARD':
+                if agent.direction == 0:
+                    agent.rect.x += 50
+                elif agent.direction == 1:
+                    agent.rect.y += 50
+                elif agent.direction == 2:
+                    agent.rect.x -= 50
+                else:
+                    agent.rect.y -= 50
+            draw_window(agent, fields)
+            time.sleep(1)
+    
+        while True:
+            pass  
 
     pygame.quit()
 
 
+
 if __name__ == "__main__":
     main()

succ.py

@@ -26,5 +26,4 @@ def succ(st: State):
         successors.append(State(st, 'RIGHT', st.xpos, st.ypos, 'S'))
         if st.xpos < 750:
             successors.append(State(st, 'FORWARD', st.xpos + 50, st.ypos, 'E'))
-
     return successors