From de75ae3122e2aad73a51e90d57c36047e18ac66d Mon Sep 17 00:00:00 2001 From: Darek Golomski Date: Mon, 12 Apr 2021 16:10:14 +0200 Subject: [PATCH] Agent defuse mine under him --- src/BFS.py | 2 ++ src/main.py | 6 ++++-- 2 files changed, 6 insertions(+), 2 deletions(-) diff --git a/src/BFS.py b/src/BFS.py index 3036c03..abb064f 100644 --- a/src/BFS.py +++ b/src/BFS.py @@ -1,5 +1,6 @@ import queue from typing import List +import pygame as pg from node import Node from tile import Tile @@ -36,6 +37,7 @@ def successor(x: int, y: int, direction: int): def breadth_first_search(field: List[List[Tile]], start_x: int, start_y: int, start_direction: int): + pg.time.delay(50) explored = [(start_x, start_y)] node_queue = queue.SimpleQueue() node_queue.put(Node(start_x, start_y, start_direction)) diff --git a/src/main.py b/src/main.py index 1ec8d80..b00d900 100644 --- a/src/main.py +++ b/src/main.py @@ -57,12 +57,14 @@ def main(): # petla while dopóki env.mine_count != 0 # uwzglednienie czy bfs zwraca False i wtedy break # for i in range(1): + # while env.mine_count: pg.time.delay(50) path, actions = breadth_first_search(env.field, agent.x, agent.y, agent.direction) print(path, actions) if not path: - print("CHUJA") - break + if not env.field[agent.y][agent.x].mine: + print("CHUJA") + break for action in actions: print(action) pg.event.post(pg.event.Event(pg.KEYDOWN, {'key': action}))