from utilities import movement,check_moves from DataModels.House import House from DataModels.Container import Container def DFS(grid, available_movement, gc_moveset, depth=0): possible_goals = [] a = gc_moveset[-1][0] b = gc_moveset[-1][1] possible_goals.append([a+1,b]) possible_goals.append([a-1,b]) possible_goals.append([a,b+1]) possible_goals.append([a,b-1]) house_in_area = False for location in possible_goals: try: if(type(grid[location[0]][location[1]]) == House and grid[location[0]][location[1]].container.is_full and grid[location[0]][location[1]].unvisited): grid[location[0]][location[1]].unvisited = False house_in_area = True print("dupa") break except: continue if(house_in_area or len(available_movement) == 0): print("Do zwrocenia: ",gc_moveset) xy = gc_moveset[-1] print(available_movement) gc_moveset.append("pick_garbage") return (xy, gc_moveset) x,y = gc_moveset[-1] for direction in available_movement: x_next, y_next = movement(grid,x,y)[0][direction] available_movement_next = check_moves(grid, x_next,y_next,direction) gc_moveset_next = gc_moveset.copy() gc_moveset_next.append([x_next,y_next]) return DFS(grid, available_movement_next, gc_moveset_next)