in gym-minigrid/gym_minigrid/roomgrid.py [0:0]
def _gen_grid(self, width, height):
# Create the grid
self.grid = Grid(width, height)
self.room_grid = []
# For each row of rooms
for j in range(0, self.num_rows):
row = []
# For each column of rooms
for i in range(0, self.num_cols):
room = Room(
(i * (self.room_size-1), j * (self.room_size-1)),
(self.room_size, self.room_size)
)
row.append(room)
# Generate the walls for this room
self.grid.wall_rect(*room.top, *room.size)
self.room_grid.append(row)
# For each row of rooms
for j in range(0, self.num_rows):
# For each column of rooms
for i in range(0, self.num_cols):
room = self.room_grid[j][i]
x_l, y_l = (room.top[0] + 1, room.top[1] + 1)
x_m, y_m = (room.top[0] + room.size[0] - 1, room.top[1] + room.size[1] - 1)
# Door positions, order is right, down, left, up
if i < self.num_cols - 1:
room.neighbors[0] = self.room_grid[j][i+1]
room.door_pos[0] = (x_m, self._rand_int(y_l, y_m))
if j < self.num_rows - 1:
room.neighbors[1] = self.room_grid[j+1][i]
room.door_pos[1] = (self._rand_int(x_l, x_m), y_m)
if i > 0:
room.neighbors[2] = self.room_grid[j][i-1]
room.door_pos[2] = room.neighbors[2].door_pos[0]
if j > 0:
room.neighbors[3] = self.room_grid[j-1][i]
room.door_pos[3] = room.neighbors[3].door_pos[1]
# The agent starts in the middle, facing right
self.agent_pos = (
(self.num_cols // 2) * (self.room_size-1) + (self.room_size // 2),
(self.num_rows // 2) * (self.room_size-1) + (self.room_size // 2)
)
self.agent_dir = 0