Correct out of bounds errors and add missing arguments to range call

This commit is contained in:
Charles Peyrat 2021-01-08 06:54:01 +01:00
parent c959a9d865
commit 9c252a2bbc
1 changed files with 5 additions and 3 deletions

View File

@ -29,15 +29,17 @@ class Generator:
@staticmethod
def room_fits(level, y, x, room, door_y, door_x, dy, dx):
if level[y][x] != Tile.EMPTY or level[y-dy][x-dx] != Tile.FLOOR:
return False
lh, lw = len(level), len(level[0])
rh, rw = len(room), len(room[0])
if not(0 < y+dy < lh and 0 < x+dx < lw):
return False
if level[y][x] != Tile.EMPTY or level[y+dy][x+dx] != Tile.FLOOR:
return False
for ry in range(rh):
for rx in range(rw):
if room[ry][rx] == Tile.FLOOR:
ly, lx = y + ry - door_y, x + rx - door_x
if not(0 <= ly <= rh and 0 <= lx <= rw) or \
if not(0 <= ly < lh and 0 <= lx < lw) or \
level[ly][lx] == Tile.FLOOR:
return False
return True