random goal generator

examples/maze_get_goal.py

@@ -0,0 +1,17 @@
+import logging
+from random import choice
+
+import gym
+import sys
+sys.path.insert(0, '/home/e-dzia/openai-envs')
+# noinspection PyUnresolvedReferences
+import gym_maze
+
+logging.basicConfig(level=logging.DEBUG)
+
+if __name__ == '__main__':
+    maze = gym.make('MazeF1-v0')
+
+    goal = maze.maze.get_goal_state()
+
+    logging.info("Finished")

gym_maze/envs/abstract_maze.py

@@ -165,3 +165,6 @@ def _render_element(el):
             return utils.colorize('A', 'red')
         else:
             return utils.colorize(el, 'cyan')
+
+    def get_goal_state(self):
+        return self.maze.get_goal_state()

gym_maze/maze.py

@@ -1,3 +1,5 @@
+import random
+
 PATH_MAPPING = 0
 WALL_MAPPING = 1
 REWARD_MAPPING = 9
@@ -174,3 +176,13 @@ def get_possible_neighbour_cords(pos_x, pos_y) -> tuple:
         nw = (pos_x - 1, pos_y - 1)
 
         return n, ne, e, se, s, sw, w, nw
+
+    def get_goal_state(self):
+        pos_x = random.randint(0, self.max_x - 1)
+        pos_y = random.randint(0, self.max_y - 1)
+
+        while self.matrix[pos_y, pos_x] == WALL_MAPPING:
+            pos_x = random.randint(0, self.max_x - 1)
+            pos_y = random.randint(0, self.max_y - 1)
+
+        return self.perception(pos_x, pos_y)