ynerant/squirrel-battle
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 3 Wiki Activity

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

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

8
squirrelbattle/mapgeneration/broguelike.py
View File

@@ -29,15 +29,17 @@ class Generator:
@staticmethod @staticmethod
def room_fits(level, y, x, room, door_y, door_x, dy, dx): 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]) lh, lw = len(level), len(level[0])
rh, rw = len(room), len(room[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 ry in range(rh):
for rx in range(rw): for rx in range(rw):
if room[ry][rx] == Tile.FLOOR: if room[ry][rx] == Tile.FLOOR:
ly, lx = y + ry - door_y, x + rx - door_x 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: level[ly][lx] == Tile.FLOOR:
return False return False
return True return True