import random as r import sys def createDungeon(x=None, y=None, seed=None): """ Initializes an x by y grid. x is width, y is height seed is the chance that a given cell will be "live" and should be an integer between 1-99. If True is equivalent to "walkable", then lower seeds make more walls. """ x = 10 if x == None else int(x) y = 10 if y == None else int(y) seed = 45 if seed == None else int(seed) new_map = [] for j in range(y): new_row = [] for i in range(x): new_row.append(False if r.randint(1,99) > seed else True) new_map.append(new_row) return new_map def refineDungeon(d_map, d_lmt=None, a_lmt=None): """ Refines the grid. """ d_lmt = 3 if d_lmt == None else int(d_lmt) a_lmt = 4 if a_lmt == None else int(a_lmt) new_map = [] for j in range(len(d_map)): new_line = [] for i in range(len(d_map[j])): x, y = i, j n_count = countAliveNeighbors(d_map, x, y) if d_map[y][x]: if n_count <= d_lmt: new_line.append(False) else: new_line.append(True) else: if n_count >= a_lmt: new_line.append(True) else: new_line.append(False) new_map.append(new_line) return new_map def countAliveNeighbors(d_map, x, y): count = 0 for j in range(-1,2): for i in range(-1,2): n_x, n_y = x+i, y+j if i == 0 and j == 0: continue if n_x < 0 or n_x >= len(d_map[j]) or n_y == 0 or n_y >= len(d_map): count += 1 elif d_map[n_y][n_x]: count += 1 return count def printDungeon(d_map, wall=None, path=None): wall = "II" if wall == None else wall path = " " if path == None else path for line in d_map: print("".join([wall if x == True else path for x in line])) print() def main(x=None, y=None, seed=None, d_lmt=None, a_lmt=None): # Initialize x = 20 if x == None else int(x) y = 20 if y == None else int(y) seed = 45 if seed == None else int(seed) d_lmt = 4 if d_lmt == None else int(d_lmt) a_lmt = 4 if a_lmt == None else int(a_lmt) my_map = createDungeon(x,y,seed) printDungeon(my_map) my_map = refineDungeon(my_map, d_lmt, d_lmt) printDungeon(my_map) my_map = refineDungeon(my_map, d_lmt, d_lmt) printDungeon(my_map) if __name__ == "__main__": args = sys.argv try: x = args[1] except: x = None try: y = args[2] except: y = None try: seed = args[3] except: seed = None try: d_lmt = args[4] except: d_lmt = None try: a_lmt = args[5] except: a_lmt = None main(x, y, seed, d_lmt, a_lmt)