def _interior_garden(w, h, wall_material, floor_material): M = Map(w, h, fill_cell=C.floor_flagged) garden_part = Map(w // 2, h, fill_cell=C.flora_grass) for y in range(0, h): garden_part[w // 2 - 1, y] = C.floor_flagged() for x in range(1, w // 2 - 1): for y in range(1, 4): garden_part[x, y] = C.floor_flagged() for x in range(2, w // 2): garden_part[x, 2].put(T.water_trough()) for y in range(h - 3, h): garden_part[0, y] = C.flora_tree() garden_part[1, y].put(T.water_trough()) garden_part[2, y] = C.flora_flower() garden_part[w // 2 - 2, 0] = C.flora_flower() garden_part[0, h - 5].put(T.furniture_sofa()) garden_part[1, h - 5].put(T.furniture_table_round()) garden_part[2, h - 6].put(T.urn()) M.meld(garden_part, 0, 0) garden_part2 = deepcopy(garden_part) garden_part2.hmirror() M.meld(garden_part2, w // 2 + w % 2, 0) for x in range(0, w // 2 - 1): M[x, h - 4] = C.floor_flagged() M.scatter(0, 0, w - 1, h - 1, [A.animal_cat()]) return M
def _room_torture(w=5, h=5): M = Map(w, h, fill_cell=C.floor_flagged) # Create walls for x in range(0, w): M[x, 0] = C.wall_stone() M[x, h - 1] = C.wall_stone() for y in range(0, h): M[0, y] = C.wall_stone() M[w - 1, y] = C.wall_stone() M[w // 2, h - 1] = C.door_closed() M[w // 2 + 1, h - 2] = C.stairs_up() # Place furniture and items in the room M[w // 2, h // 2].put(T.furniture_torture()) all_coord = [(w // 2, h - 2), (w // 2, h // 2), (w // 2 + 1, h - 2)] for item_class in (T.bones, T.bones_skull, T.tool_tongs): while True: x = random.randint(1, w - 2) y = random.randint(1, h - 2) if (x, y) not in all_coord: M[x, y].put(item_class()) all_coord.append((x, y)) break return M
def _room_jailer(w=5, h=5): M = Map(w, h, fill_cell=C.floor_flagged) # Create walls for x in range(0, w): M[x, 0] = C.wall_stone() M[x, h - 1] = C.wall_stone() for y in range(0, h): M[0, y] = C.wall_stone() M[w - 1, y] = C.wall_stone() M[w // 2, 0] = C.door_closed() # Place furniture and items in the room all_coord = [(w // 2, 1)] for item_class in (T.furniture_bed_single, T.furniture_chest, T.furniture_chair, T.furniture_table, T.furniture_torch): while True: x = random.randint(1, w - 2) y = random.randint(1, h - 2) if (x, y) not in all_coord: M[x, y].put(item_class()) all_coord.append((x, y)) break return M
def world_test(w, h): heightmap = [] for y in range(h): line = [] for x in range(w): distance_corners = min([ sqrt(x**2 + y**2), sqrt((w-x)**2 + y**2), sqrt(x**2 + (h-y)**2), sqrt((w-x)**2 + (h-y)**2) ]) distance_edges = min([x*0.7, y*0.7, (w-x)*0.7, (h-y)*0.7]) minus = (w + h) / 3 c = random.randint(-w*0.7, -w*0.6) + distance_corners + distance_edges #print('{:> .1f}'.format(c), '', end='') line.append(c) heightmap.append(line) heightmap = diamond_square_2x(heightmap) heightmap = diamond_square_2x(heightmap, smooth_iteration=2) heightmap = diamond_square_2x(heightmap, smooth_iteration=3) heightmap = diamond_square_2x(heightmap, smooth_iteration=4) M = Map(len(heightmap[0]), len(heightmap), fill_cell=C.overworld_ocean) for y, line in enumerate(M.cells): for x, cell in enumerate(line): M[x, y] = C.overworld_ocean() if heightmap[y][x] < 0 else C.overworld_forest() M = _smooth_map(M) M = _smooth_map(M) return M
def _room_cattle_pens(w, h, wall_material, floor_material): M = room_default(w, h, wall_type=C.wall_fence, floor_type=floor_material) M[w - 7, 0] = C.door_close_fence() for i in range(w * h // 3): grass_x = random.randint(1, w - 2) grass_y = random.randint(1, h - 2) M[grass_x, grass_y] = random.choice( [C.flora_grass, C.flora_cane, C.floor_grass])() num_cattles = h // 4 + 1 cowshed = Map(4, 3, fill_cell=floor_material) for y in (1, 2): cowshed[0, y] = C.wall_fence() for y in range(0, 3): cowshed[3, y].put(T.water_trough()) for x in (1, 2): cowshed[x, 2] = wall_material() cowshed[1, 1].put(T.bucket()) cowshed_y = 1 for x in range(num_cattles): copied_cowshed = deepcopy(cowshed) M.meld(copied_cowshed, w - 5, cowshed_y) cowshed_y += 3 cows = [A.animal_cow() for _ in range(num_cattles)] M.scatter(1, 1, w - 5, h - 1, cows) return M
def room_default(w, h, wall_type, floor_type): """ Create an empty room with given wall type, floor type. Parameters ---------- w : int Map width h : int Map height wall_type : Cell Wall cell type floor_typr : Cell Floor cell type """ M = Map(w, h, fill_cell=floor_type) # Create walls for x in range(0, w): M[x, 0] = wall_type() M[x, h - 1] = wall_type() for y in range(0, h): M[0, y] = wall_type() M[w - 1, y] = wall_type() return M
def dungeon_bsp_tree(w=30, h=30, optimal_block_size=10): """ Construct the dungeon map using binary space partitioning (BSP) algorithm. Visual ------ Rectangular square-like rooms connected with 1-pixel corridors. These rooms evenly distributed across the map. All corridors are straight. Parameters ---------- w : int Map width h : int Map height optimal_block_size : int Optimal block size. Approximately equals to room size. """ M = Map(w, h, fill_cell=C.wall_dungeon_rough) nodes = {} root = BSPNode('v', 1, 1, w - 1, h - 1) _recursive_split_tree_node(root, optimal_block_size) _load_leafs(root, nodes) _fill_rooms(M, nodes.values()) all_edges = _get_all_edges(nodes.values()) st_edges = _construct_spanning_tree(list(nodes.keys()), all_edges) _create_corridors(M, nodes, st_edges) return M
def _interior_garden(w, h, wall_material, floor_material): M = Map(w, h, fill_cell=C.floor_rocks) for x in range(w): M[x, h-1] = wall_material() M[w//2, h-1] = C.door_closed_wooden() M[w//2-1, h-1] = C.door_closed_wooden() return M
def dungeon_grid_simple(w, h, room_size=4, delete_chance=0.33): """Dungeon map generator based on square room grid.""" M = Map(w, h, fill_cell=C.floor_flagged) _create_room_grid(M, room_size=room_size) _create_doors(M, room_size=room_size) _crush_walls(M, room_size=room_size, delete_chance=delete_chance) return M
def _gen_main(self, xsize, ysize, length, turn_chance=0.4): M = Map(xsize, ysize, fill_symbol='#') worm_x = random.randint(int(xsize * 0.3), int(xsize * 0.6)) worm_y = random.randint(int(ysize * 0.3), int(ysize * 0.6)) move = random.choice([NORTH, SOUTH, EAST, WEST]) for _ in range(length): worm_x, worm_y, move = self._move_worm(M, worm_x, worm_y, move, turn_chance) return M
def _room_outdoor(w, h): M = Map(w, h, fill_cell=C.floor_rocks) for i in range(w*h//3): grass_x = random.randint(0, w-1) grass_y = random.randint(0, h-1) M[grass_x, grass_y] = random.choice([C.flora_grass, C.flora_tree, C.floor_grass])() for x in (1, 8): M[x, 0].put(T.furniture_table()) for x in (0, 2, 7, 9): M[x, 0].put(T.furniture_stool()) if w > 13: M[11, 0].put(T.furniture_table()) M[10, 0].put(T.furniture_stool()) M[12, 0].put(T.furniture_stool()) for y in range(0, h): M[4, y] = C.floor_cobblestone() num_stables = (h - 5) // 2 stables = Map(3, 2, fill_cell=C.void) for x in range(3): stables[x, 0] = C.wall_fence_thin() stables[2, 1] = C.wall_fence_thin() stables[0, 1] = C.door_close_fence() stables[1, 1].put(A.animal_horse()) last_stable_y = h - 1 for i in range(num_stables): stable_x = w - 3 stable_y = 4 + (i * 2) M.meld(stables, stable_x, stable_y) last_stable_y = stable_y for x in range(w-3, w): M[x, last_stable_y + 2] = C.wall_fence_thin() M[2, h-1].put(T.sign_pointer()) M[3, 0].put(T.light_torch()) for y in (h//2-1, h//2): M[0, y].put(T.washtub()) if w > 9: M[0, h//2+1].put(T.washtub()) M[5, 0] = C.flora_flower() return M
def dungeon_drunkard(w, h, length=None, turn_chance=0.4): M = Map(w, h, fill_cell=C.wall_dungeon_smooth) if not length: length = int(w * h / 2) worm_x = random.randint(int(w * 0.3), int(w * 0.6)) worm_y = random.randint(int(h * 0.3), int(h * 0.6)) move = random.choice([NORTH, SOUTH, EAST, WEST]) for _ in range(length): M[worm_x, worm_y] = C.floor_flagged() worm_x, worm_y, move = _move_worm(M, worm_x, worm_y, move, turn_chance) return M
def building_stables(w=16, h=16): """ Construct stables with storage and two rows of horse boxes. Constraints: - Map width and map height must be >= 16 and <=23. Parameters ---------- w : int Map width h : int Map height """ # Initial checks. Don't accept too small/big stables if w < 16 or h < 16: raise ValueError('Building is too small: w or h < 16') elif w > 23 or h > 23: raise ValueError('Building is too big: w or h > 25') M = Map(w, h, fill_cell=C.floor_rocks) for x in range(w): for y in range(h): if random.random() > 0.75: M[x, y] = C.flora_grass() # Calculate w and h for storage, horse boxes and stables. horse_box_size_w = w // 2 - 4 horse_box_size_h = 2 stables_size_w = horse_box_size_w * 2 + 3 stables_shift_h = (h - 1) % 3 stables_size_h = h - stables_shift_h storage_size_w = w - stables_size_w + 1 storage_size_h = h * 2 // 3 storage_start_w = w - storage_size_w # Meld stables, storage, add dog. main_stables = room_horse_stables(stables_size_w, stables_size_h, horse_box_size_w, horse_box_size_h) M.meld(main_stables, 0, 0) main_storage = room_storage(storage_size_w, storage_size_h) M.meld(main_storage, storage_start_w, 0) M[w - (w - stables_size_w) // 2, storage_size_h].put(T.well()) dog_place_x = random.randint(stables_size_w, w - 1) dog_place_y = random.randint(storage_size_h + 1, h - 1) M[dog_place_x, dog_place_y].put(A.animal_dog()) if random.choice([True, False]): M.hmirror() return M
def _interior_pantry(w, h, wall_material, floor_material): M = Map(w, h, fill_cell=floor_material) if random.random() < 0.5: for y in range(h): M[2, y] = wall_material() items = [T.tool_wateringcan(), T.tool_pitchfork(), T.tool_fishingrod()] M.scatter(0, 0, w - 1, h, items) else: M[0, 0] = C.stairs_down() M[0, 1].put(T.furniture_box_filled()) return M
def _interior_bar(w, h, floor_material): M = Map(w, h, fill_cell=floor_material) M[0, 0].put(T.furniture_table()) M[1, 0].put(T.furniture_stool()) for x in range(w//3, w): for y in (h//3, h//3*2+1): M[x, y].put(T.furniture_longtable()) M.scatter(w//3, h//3-1, w, h//3+2, [T.furniture_stool() for _ in range(w-w//3)], exclude=[(w//3, h//3-1)]) M.scatter(w//3, h//3*2, w, h//3*2+3, [T.furniture_stool() for _ in range(w-w//3)], exclude=[(w//3, h//3*2+2)]) M[0, h-1].put(T.furniture_table()) M[1, h-1].put(T.furniture_stool()) M[w-1, h-1].put(T.light_lantern_oil()) return M
def world_perlin_noise(w, h, scale=10.0, octaves=6, persistence=0.5, lacunarity=2.0): M = Map(w, h) min_value = None max_value = None startx = random.randint(0, w * 100) starty = random.randint(0, h * 100) for y, line in enumerate(M.cells): for x, cell in enumerate(line): n = noise.pnoise2((startx + x) / scale, (starty + y) / scale, octaves=octaves, persistence=persistence, lacunarity=lacunarity, repeatx=w, repeaty=h, base=2) M[x, y].noise = n if min_value is None: min_value = n if max_value is None: max_value = n if n < min_value: min_value = n if n > max_value: max_value = n delta = (max_value - min_value) / 12 for y, line in enumerate(M.cells): for x, cell in enumerate(line): if M[x, y].noise < min_value + delta: M[x, y] = C.overworld_ocean() elif M[x, y].noise < min_value + 5 * delta: M[x, y] = C.overworld_ocean() elif M[x, y].noise < min_value + 8 * delta: M[x, y] = C.overworld_plains() elif M[x, y].noise < min_value + 10 * delta: M[x, y] = C.overworld_forest() elif M[x, y].noise < min_value + 12 * delta: M[x, y] = C.overworld_mountain() else: M[x, y] = C.overworld_pinnacle() return M
def _room_outdoor(w, h): M = Map(w, h, fill_cell=C.floor_rocks) for i in range(w * h // 3): grass_x = random.randint(1, w - 1) grass_y = random.randint(0, h - 2) M[grass_x, grass_y] = random.choice( [C.flora_grass, C.flora_tree, C.floor_grass])() M[w - 1, 0].put(T.washtub()) for x in range(w): M[x, h - 1] = C.wall_fence_thin() for y in range(h - 1): M[0, y] = C.wall_fence_thin() for y in range(h - 1): M[3, y] = C.floor_rocks() M[3, h - 1] = C.door_close_fence() return M
def world_test2(w=400, h=200): M = Map(w, h, fill_cell=C.overworld_ocean) ''' plates = {i: (random.randint(0, w-1), random.randint(0, h-1)) for i in range(30)} for y, line in enumerate(M.cells): for x, cell in enumerate(line): intervals = {i: (plates[i][0] - x) ** 2 + (plates[i][1] - y) ** 2 for i in plates} #print(intervals) M[x, y].plate = min(intervals.items(), key=lambda item: item[1])[0] for y, line in enumerate(M.cells): for x, cell in enumerate(line): M[x, y] = P[M[x, y].plate]() for i in plates: M[plates[i]] = C.void() ''' for y, line in enumerate(M.cells): for x, cell in enumerate(line): distance_corners = min([ sqrt(x**2 + y**2), sqrt((w-x)**2 + y**2), sqrt(x**2 + (h-y)**2), sqrt((w-x)**2 + (h-y)**2) ]) distance_edges = min([x, y, (w-x), (h-y)]) minus = (w + h) / 3 c = random.randint(-w*0.8, -w*0.6) + distance_corners + distance_edges #print('{:> .1f}'.format(c), '', end='') M[x, y] = C.overworld_ocean() if c < 0 else C.overworld_forest() #print() #for y in [0, h-1]: # for x in range(M.w): # M[x, y] = C.overworld_ocean() #for x in [0, w-1]: # for y in range(M.h): # M[x, y] = C.overworld_ocean() #for x in range(w//2-2, w//2+3): # for y in range(h//2-2, h//2+3): # M[x, y] = C.overworld_forest() #for y, line in enumerate(M.cells): # for x, cell in enumerate(line): # if sum(c.cname == 'overworld_forest' for c in M.surrounding(x, y)) >= 7: # M[x, y] = C.overworld_forest() #M = _smooth_map(M) return M
def room_horse_box(w, h, orientation='left'): """ Construct small horse box. """ M = Map(w, h, fill_cell=C.floor_rocks) # Place watertrough and horse food. M[0, 0].put(T.water_trough()) M[w - 1, 0] = C.door_closed() M[0, h - 1].put(T.water_trough()) M[w - 1, h - 1] = C.wall_fence() if h > 2: for y in range(1, h - 1): M[0, y].put(T.water_trough()) M[w - 1, y] = C.wall_fence() # Create horse box with animal or without. stable_with_horse_chance = random.random() all_coord = [] while True: x = random.randint(1, w - 2) y = random.randint(0, h - 1) if (x, y) not in all_coord: M[x, y] = C.flora_grass() all_coord.append((x, y)) break if stable_with_horse_chance > 0.3: while True: x = random.randint(1, w - 2) y = random.randint(0, h - 1) if (x, y) not in all_coord: M[x, y].put(T.farm_mangler()) all_coord.append((x, y)) break while True: x = random.randint(1, w - 2) y = random.randint(0, h - 1) if (x, y) not in all_coord: M[x, y].put(A.animal_horse()) all_coord.append((x, y)) break if orientation == 'right': M.hmirror() return M
def dungeon_cellular_simple(w=30, h=30, start_floor_chance=0.55, smooth_level=3): """ Construct the dungeon map using simple cellular automata. Note that this map can contain disconnected areas. Visual ------ Natural-looking cave-like map. Parameters ---------- w : int Map width h : int Map height start_floor_chance : float Chance of the floor cell in the first map fill smooth_level : int Number of sequential smooth functions """ M = Map(w, h, fill_cell=C.wall_cave) # Randomly fill all-but-border cells by floor with start_floor_chance probability for y, line in enumerate(M.cells[1:-1]): for x, _ in enumerate(line[1:-1]): chance = random.random() if chance <= start_floor_chance: M.cells[y + 1][x + 1] = C.floor_dirt() # Sequentially smooth the map smooth_level times for _ in range(smooth_level): M = _smooth_map(M) return M
def _interior_vending(w, h, wall_material, floor_material): M = Map(w, h, fill_cell=floor_material) for x in range(w): M[x, 2].put(T.furniture_longtable_showcase()) M[0, 1].put(T.money_pile()) M[w//2, 1].put(T.dining_mug()) for x in range(1, w, 2): M[x, 3].put(T.furniture_stool()) M[1, h-1].put(T.furniture_barrel()) M[2, h-1].put(T.furniture_barrel()) if w > 7: M[w-2, h-1].put(T.furniture_table()) M[w-1, h-1].put(T.furniture_stool()) M[w-3, h-1].put(T.furniture_stool()) if h > 6: for x in range(1, w-2, 3): for y in range(5, h-2, 3): M[x, y].put(T.furniture_table()) M[x+1, y].put(T.furniture_stool()) return M
def _room_prison_center(w=5, h=5): M = Map(w, h, fill_cell=C.floor_flagged) if w > 8 and h > 4: jailer_room = _room_jailer(w // 2 + 1, h) M.meld(jailer_room, 0, 0) torture_room = _room_torture(w - w // 2, h) M.meld(torture_room, (w - w // 2) - 1, 0) elif h > 8 and w > 4: jailer_room = _room_jailer(w, h // 2 + 1) M.meld(jailer_room, 0, 0) torture_room = _room_torture(w, h - h // 2) M.meld(torture_room, 0, (h - h // 2) - 1) else: for x in range(0, w): M[x, 0] = C.wall_stone() M[x, h - 1] = C.wall_stone() for y in range(0, h): M[0, y] = C.wall_stone() M[w - 1, y] = C.wall_stone() M[w // 2, 0] = C.door_closed() M[w // 2, h // 2] = C.stairs_up() return M
def building_prison_linear(w=21, h=12, orientation='horizontal'): """ Construct a linear prison building interior. Parameters ---------- w : int Map width h : int Map height """ # Initial checks. Don't accept: # - Too small prisons # - Too wide prisons (for both horizontal and vertical orientations) if w < 11 or h < 11: raise ValueError('Building is too small: w or h < 11') if h > 16 and orientation == 'horizontal': raise ValueError( 'Building is too big: h > 16 and orientation == "horizontal"') if w > 16 and orientation == 'vertical': raise ValueError( 'Building is too big: w > 16 and orientation == "vertical"') if orientation == 'vertical': w, h = h, w M = Map(w, h, fill_cell=C.void) # Randomly choose where torture room and jailer room are. torture_left, jailer_right = None, None torture_left = random.choice([True, False]) if torture_left: jailer_right = True else: jailer_right = False # Create jailer room. We have two situations: jailer room left/right. jailer_y_start = h // 4 jailer_y_end = h // 3 * 2 if jailer_right: # Create walls, floor and door for y in range(jailer_y_start, jailer_y_end + 1): M[w - 1, y] = C.wall_stone() M[w - 5, y] = C.wall_stone() for x in range(w - 5, w): M[x, jailer_y_start] = C.wall_stone() M[x, jailer_y_end] = C.wall_stone() for y in range(jailer_y_start + 1, jailer_y_end): for x in range(w - 4, w - 1): M[x, y] = C.floor_flagged() M[w - 5, h // 2] = C.door_closed() # Place some furniture M[w - 4, jailer_y_start + 1].put(T.furniture_table()) M[w - 3, jailer_y_start + 1].put(T.furniture_table()) M[w - 2, jailer_y_start + 1].put(T.furniture_chair()) M[w - 4, jailer_y_end - 1].put(T.furniture_torch()) M[w - 3, jailer_y_end - 1].put(T.furniture_bed_single()) M[w - 2, jailer_y_end - 1].put(T.furniture_chest()) else: # Create walls, floor and door for y in range(jailer_y_start, jailer_y_end + 1): M[0, y] = C.wall_stone() M[4, y] = C.wall_stone() for x in range(0, 5): M[x, jailer_y_start] = C.wall_stone() M[x, jailer_y_end] = C.wall_stone() for y in range(jailer_y_start + 1, jailer_y_end): for x in range(1, 4): M[x, y] = C.floor_flagged() M[4, h // 2] = C.door_closed() # Place some furniture M[1, jailer_y_start + 1].put(T.furniture_table()) M[2, jailer_y_start + 1].put(T.furniture_table()) M[3, jailer_y_start + 1].put(T.furniture_chair()) M[1, jailer_y_end - 1].put(T.furniture_chest()) M[2, jailer_y_end - 1].put(T.furniture_bed_single()) M[3, jailer_y_end - 1].put(T.furniture_torch()) # Create torture room. We have two situations: torture room left/right. torture_start and torture_end - x-coord. # If torture_end = 0 or 1, there is no place for room (only one or two walls) # So we expand torture room for a one cell's width (+4) if jailer_right: torture_start = 0 torture_end = (w - 1) % 4 if torture_end == 0: torture_end = 4 if torture_end == 1: torture_end = 5 # Create walls, floor and door for x in range(torture_start, torture_end + 1): M[x, 0] = C.wall_stone() M[x, h - 1] = C.wall_stone() for y in range(0, h - 1): M[torture_start, y] = C.wall_stone() M[torture_end, y] = C.wall_stone() for x in range(torture_start + 1, torture_end): for y in range(1, h - 1): M[x, y] = C.floor_flagged() M[torture_end, h // 2] = C.door_closed() # Place some furniture. If torture_end == 2 (just a corridor), then we set only stairs. M[torture_end - 1, h - 2] = C.stairs_up() if torture_end != 2: M[(torture_end - torture_start) // 2, h // 2].put(T.furniture_torture()) all_coord = [(torture_end - 1, h - 2), ((torture_end - torture_start) // 2, h // 2)] for item_class in (T.bones, T.bones_skull, T.tool_tongs): while True: x = random.randint(1, torture_end - 1) y = random.randint(1, h - 2) if (x, y) not in all_coord: M[x, y].put(item_class()) all_coord.append((x, y)) break else: # If torture room is right, we are using the torture room width for calculations. # If torture_width = 7, then we reduce torture room for a one cell's width (-4). torture_width = w % 4 + 4 if torture_width == 7: torture_width = 3 torture_end = w - 1 # Create walls, floor and door for x in range(w - torture_width, torture_end): M[x, 0] = C.wall_stone() M[x, h - 1] = C.wall_stone() for y in range(0, h): M[w - torture_width, y] = C.wall_stone() M[torture_end, y] = C.wall_stone() for x in range(w - torture_width + 1, torture_end): for y in range(1, h - 1): M[x, y] = C.floor_flagged() M[w - torture_width, h // 2] = C.door_closed() # Place some furniture. If torture_width = 3 (just a corridor), then we set only stairs. M[w - 2, h - 2] = C.stairs_up() if torture_width != 3: M[w - 2, h // 2].put(T.furniture_torture()) all_coord = [(w - 1, h - 2), (w - 2, h // 2)] for item_class in (T.bones, T.bones_skull, T.tool_tongs): while True: x = random.randint(w - torture_width + 1, w - 2) y = random.randint(1, h - 2) if (x, y) not in all_coord: M[x, y].put(item_class()) all_coord.append((x, y)) break # Fill corridor with a floor if jailer_right: cor_start = torture_end + 1 cor_end = w - 6 else: cor_start = 5 cor_end = w - torture_width - 1 if h % 2 == 1: for x in range(cor_start, cor_end + 1): M[x, h // 2] = C.floor_flagged() else: for x in range(cor_start, cor_end + 1): M[x, h // 2 - 1] = C.floor_flagged() M[x, h // 2] = C.floor_flagged() # Place prison cells number_of_cells = (cor_end - cor_start + 2) // 4 for cell_index in range(number_of_cells): cell_x = (cor_start - 1) + (cell_index * 4) # Place upper cell M_cell = room_prison_cell(w=5, h=(h - 1) // 2) M.meld(M_cell, cell_x, 0) # Place lower cell M_cell = room_prison_cell(w=5, h=(h - 1) // 2, direction='up') M.meld(M_cell, cell_x, h // 2 + 1) if orientation == 'vertical': M.transpose() return M
def building_house_tworoom(w=12, h=9, wall_material=None, floor_material=None, has_exterior=True, direction='down'): """ Construct house with living room, kitchen and outdoor. Constraints: - Map width and map height must be >= 9 - Map width and map height must be <= 15 - Wall material must be 'block', 'plank', 'brick' or 'stone'. - Floor material must be 'dirt', 'parquet' or 'cobblestone'. Parameters ---------- w : int Map width h : int Map height wall_material : str Wall's material. floor_material : str Floor's material. has_exterior : bool Flag that generates an exterior of the house. direction : str Direction of the house. Can be 'up', 'down', 'left' or 'right'. """ # Initial checks. Don't accept too small/big house if w < 9 or h < 9: raise ValueError('Building is too small: w or h < 9') elif w > 12 or h > 12: raise ValueError('Building is too big: w or h > 15') # Choose materials if not wall_material: wall_material = random.choice( [C.wall_block, C.wall_plank, C.wall_brick, C.wall_stone]) elif wall_material not in (['block', 'plank', 'brick', 'stone']): raise ValueError( 'Wall material should be "block", "plank", "brick" or "stone"') if wall_material == 'block': wall_material = C.wall_block elif wall_material == 'plank': wall_material = C.wall_plank elif wall_material == 'brick': wall_material = C.wall_brick elif wall_material == 'stone': wall_material = C.wall_stone if not floor_material: floor_material = random.choice( [C.floor_dirt, C.floor_parquet, C.floor_cobblestone]) elif floor_material not in (['dirt', 'parquet', 'cobblestone']): raise ValueError( 'Floor material should be "dirt", "parquet" or "cobblestone"') if floor_material == 'dirt': floor_material = C.floor_dirt elif floor_material == 'parquet': floor_material = C.floor_parquet elif floor_material == 'cobblestone': floor_material = C.floor_cobblestone M = Map(w, h, fill_cell=C.void) main_room_h = h // 3 * 2 living_room_w = w // 2 + 1 living_room_h = h - main_room_h + 1 main_room = _room_main(w, main_room_h, wall_material, floor_material) M.meld(main_room, 0, 0) living_room = _room_living(living_room_w, living_room_h, wall_material, floor_material) M.meld(living_room, w - living_room_w, main_room_h - 1) if has_exterior: outdoor = _room_outdoor(w - living_room_w, living_room_h - 1) M.meld(outdoor, 0, main_room_h) if random.choice([True, False]): M.hmirror() if direction == 'up': M.vmirror() elif direction == 'left': M.transpose() elif direction == 'right': M.transpose() M.hmirror() return M
def building_medieval_mansion(w=25, h=25, wall_material=None, floor_material=None, direction='down'): """ Construct medieval mansion with living rooms, kitchen, library, treasury, servant's room and outdoor. Constraints: - Map width and map height must be >= 20 - Map width and map height must be <= 30 - Wall material must be 'block', 'plank', 'brick' or 'stone'. - Floor material must be 'dirt', 'parquet' or 'cobblestone'. Parameters ---------- w : int Map width h : int Map height wall_material : str Wall's material. floor_material : str Floor's material. direction : str Direction of the house. Can be 'up', 'down', 'left' or 'right'. """ # Initial checks. Don't accept too small/big house if w < 20 or h < 20: raise ValueError('Building is too small: w or h < 20') elif w > 30 or h > 30: raise ValueError('Building is too big: w or h > 30') # Choose materials if not wall_material: wall_material = random.choice([C.wall_block, C.wall_plank, C.wall_brick, C.wall_stone]) elif wall_material not in (['block', 'plank', 'brick', 'stone']): raise ValueError('Wall material should be "block", "plank", "brick" or "stone"') if wall_material == 'block': wall_material = C.wall_block elif wall_material == 'plank': wall_material = C.wall_plank elif wall_material == 'brick': wall_material = C.wall_brick elif wall_material == 'stone': wall_material = C.wall_stone if not floor_material: floor_material = random.choice([C.floor_dirt, C.floor_parquet, C.floor_cobblestone]) elif floor_material not in (['dirt', 'parquet', 'cobblestone']): raise ValueError('Floor material should be "dirt", "parquet" or "cobblestone"') if floor_material == 'dirt': floor_material = C.floor_dirt elif floor_material == 'parquet': floor_material = C.floor_parquet elif floor_material == 'cobblestone': floor_material = C.floor_cobblestone M = Map(w, h, fill_cell=C.void) default_room_w = w // 4 + 1 default_room_h = h // 4 library_h = h // 2 - 1 second_bedroom_h = h // 4 + 2 treasury = _room_treasury(default_room_w, default_room_h, wall_material, floor_material) M.meld(treasury, 0, 0) bedroom = _room_bedroom(default_room_w, h-second_bedroom_h-default_room_h+2, wall_material, floor_material) M.meld(bedroom, 0, default_room_h-1) second_bedroom = _room_second_bedroom(default_room_w, second_bedroom_h, wall_material, floor_material) M.meld(second_bedroom, 0, h-second_bedroom_h) sacrifice = _room_of_sacrifice(default_room_w, default_room_h, wall_material, floor_material) M.meld(sacrifice, w-default_room_w, 0) kitchen = _room_kitchen(default_room_w, h-default_room_h*2+1, wall_material, floor_material) M.meld(kitchen, w-default_room_w, default_room_h-1) servant = _room_servant(default_room_w, default_room_h+1, wall_material, floor_material) M.meld(servant, w-default_room_w, h-default_room_h-1) library = _room_library(w-default_room_w*2+2, library_h, wall_material, floor_material) M.meld(library, default_room_w-1, 0) garden = _interior_garden(w-default_room_w*2, h-library_h, wall_material, floor_material) M.meld(garden, default_room_w, library_h) if random.choice([True, False]): M.hmirror() if direction == 'up': M.vmirror() elif direction == 'left': M.transpose() elif direction == 'right': M.transpose() M.hmirror() return M
def room_prison_cell(w=5, h=5, direction='down'): if direction == 'left' or direction == 'right': w, h = h, w M = Map(w, h, fill_cell=C.floor_flagged) prison_cell_type = random.choice(['with_prisoner', 'with_bones', 'with_blood', 'with_spider']) # Create walls for x in range(0, w): M[x, 0] = C.wall_stone() M[x, h-1] = C.wall_stone() for y in range(0, h): M[0, y] = C.wall_stone() M[w-1, y] = C.wall_stone() # Create prison bars and door M[1, h-1] = C.wall_bars() M[3, h-1] = C.wall_bars() M[2, h-1] = C.door_closed_bars() # Place some things all_coord = [] if prison_cell_type == 'with_prisoner': for item_class in (T.furniture_napsack, T.bucket, T.furniture_torch): while True: x = random.randint(1, 3) y = random.randint(1, h-2) if (x, y) not in all_coord: M[x, y].put(item_class()) all_coord.append((x,y)) break elif prison_cell_type == 'with_bones': for item_class in (T.bones, T.bones_skull): while True: x = random.randint(1, 3) y = random.randint(1, h-2) if (x, y) not in all_coord: M[x, y].put(item_class()) all_coord.append((x, y)) break elif prison_cell_type == 'with_blood': for item_class in (T.bones_remains, T.effect_blood): while True: x = random.randint(1, 3) y = random.randint(1, h-2) if (x, y) not in all_coord: M[x, y].put(item_class()) all_coord.append((x, y)) break elif prison_cell_type == 'with_spider': for item_class in (T.web, T.web, A.animal_spider): while True: x = random.randint(1, 3) y = random.randint(1, h-2) if (x, y) not in all_coord: M[x, y].put(item_class()) all_coord.append((x, y)) break if direction == 'up': M.vmirror() elif direction == 'left': M.transpose() elif direction == 'right': M.transpose() M.hmirror() return M
def building_prison_rectangular(w=17, h=17): """ Construct a rectangular prison interior. Parameters ---------- w : int Map width h : int Map height """ # Initial checks. Don't accept too small prisons if w < 17 or h < 17: raise ValueError('Building is too small: w or h < 17') M = Map(w, h, fill_cell=C.void) # Calculate prison cells sizes depends on prison size. cell_shift_w = (w - 1) % 4 if cell_shift_w == 1: left_w = 4 right_w = 5 elif cell_shift_w == 2: left_w = 5 right_w = 5 elif cell_shift_w == 3: left_w = 5 right_w = 6 else: left_w = 4 right_w = 4 cell_shift_h = (h - 1) % 4 if cell_shift_h == 1: top_h = 4 bottom_h = 5 elif cell_shift_h == 2: top_h = 5 bottom_h = 5 elif cell_shift_h == 3: top_h = 5 bottom_h = 6 else: top_h = 4 bottom_h = 4 # Calculate the number of prison cells fits horizontally/vertically num_cell_w = (w - left_w - right_w) // 4 num_cell_h = (h - top_h - bottom_h) // 4 # Place cells for cell_index in range(num_cell_w): cell_x = left_w + (cell_index * 4) # Place upper cell M_cell = room_prison_cell(w=5, h=top_h + 1) M.meld(M_cell, cell_x, 0) # Place lower cell M_cell = room_prison_cell(w=5, h=bottom_h + 1, direction='up') M.meld(M_cell, cell_x, h - bottom_h - 1) for cell_index in range(num_cell_h): cell_y = top_h + (cell_index * 4) # Place left cell M_cell = room_prison_cell(w=left_w + 1, h=5, direction='left') M.meld(M_cell, 0, cell_y) # Place right cell M_cell = room_prison_cell(w=right_w + 1, h=5, direction='right') M.meld(M_cell, w - right_w - 1, cell_y) center_room_w = w - (left_w + 1) - (right_w + 1) - 2 center_room_h = h - (top_h + 1) - (bottom_h + 1) - 2 center_room = _room_prison_center(center_room_w, center_room_h) M.meld(center_room, left_w + 2, top_h + 2) # Construct the corridor for x in range(left_w + 1, w - right_w - 1): M[x, top_h + 1] = C.floor_flagged() M[x, h - bottom_h - 2] = C.floor_flagged() for y in range(top_h + 1, h - bottom_h - 1): M[left_w + 1, y] = C.floor_flagged() M[w - right_w - 2, y] = C.floor_flagged() return M
def building_hunter_robber_house(size=10, material=None, house_type=None): """ Construct the hunter or robber house. Parameters ---------- size : int Square map size. This attribute will be applied for both `w` and `h`. material : string Wall material. Can be "wooden", "stone" or None. If None, a random material will be chosen. house_type : string Type of the house. Can be "hunter" or "robber". If None, a random type will be chosen. """ # Initial check. Don't accept too small/big building if size < 8 or size > 17: raise ValueError( 'Building is too small or too big: size < 8 or size > 17') # Choose materials if not material: material = random.choice(['wooden', 'stone']) if material not in ('wooden', 'stone'): raise ValueError('Material should be "stone" or "wooden"') wall_cell_type = C.wall_stone if material == 'stone' else C.wall_plank floor_cell_type = C.floor_flagged if material == 'stone' else C.floor_plank # Choose between robber house and hunter house if not house_type: house_type = random.choice(['robber', 'hunter']) if house_type not in ('robber', 'hunter'): raise ValueError('Type should be "robber" or "hunter"') M = Map(size, size, fill_cell=floor_cell_type) # Create outward walls for x in range(size): M[x, 0] = wall_cell_type() M[x, size - 1] = wall_cell_type() for y in range(size): M[0, y] = wall_cell_type() M[size - 1, y] = wall_cell_type() # Create door door_random = random.choice([True, False]) door_param = size // 3 * 2 if door_random: M[door_param, size - 1] = C.door_closed() else: M[0, door_param] = C.door_closed() # Place bonfire or hearth in the middle of the room. Place chairs M[size // 2 - 1, size // 2].put(T.furniture_chair()) M[size // 2 + 1, size // 2].put(T.furniture_chair()) if house_type == 'hunter': M[size // 2, size // 2].put(T.furniture_hearth()) else: M[size // 2, size // 2].put(T.bonfire()) # Randomly choose where escape is. Place stairs and wardrobes. escape = random.choice([True, False]) if escape: M[size - 2, 1] = C.stairs_down() if house_type == 'robber': M[size - 3, 1].put(T.furniture_closet()) M[size - 3, 2].put(T.furniture_closet()) M[size - 2, 2].put(T.furniture_closet()) elif house_type == 'hunter': M[size - 2, size - 2].put(T.furniture_closet()) else: M[1, 1] = C.stairs_down() if house_type == 'robber': M[2, 1].put(T.furniture_closet()) M[2, 2].put(T.furniture_closet()) M[1, 2].put(T.furniture_closet()) elif house_type == 'hunter': M[size - 2, size - 2].put(T.furniture_closet()) # Place beds near walls beds_start = 1 if escape else size // 3 + 1 beds_end = size // 2 if escape else size - 1 if escape: for x in range(beds_start, beds_end + 1, 2): M[x, 1].put(T.furniture_bed_single()) else: for x in range(beds_start + 1, beds_end, 2): M[x, 1].put(T.furniture_bed_single()) # Place chests M[size - 2, size // 2 - 1].put(T.furniture_chest()) M[size - 2, size // 2].put(T.furniture_chest()) # Place table with chairs for x in range(size // 5, size // 2): M[x, size - 2].put(T.furniture_longtable()) for x in range(size // 5 + 1, size // 2, 2): M[x, size - 3].put(T.furniture_chair()) return M
def building_housebarn(w=30, h=15, material=None): """ Construct the housebarn (also known as longhouse) building interior. Parameters ---------- w : int Map width h : int Map height material : string Wall material. Can be "wooden", "stone" or None. If None, a random material will be chosen. """ # Initial checks. Don't accept: # - Too small buildings # - Too long/square buildings # - Wall types that are not "stone" or "wooden" if w < 10 or h < 10: raise ValueError('Building is too small: w/h < 10') if not material: material = random.choice(['wooden', 'stone']) if material not in ('wooden', 'stone'): raise ValueError('Material should be "stone" or "wooden"') wall_cell_type = C.wall_stone if material == 'stone' else C.wall_plank is_horizontal = True if w >= h else False if is_horizontal: if w < h * 2 or w > h * 3: raise ValueError('Building is too long or too short.') else: if h < w * 2 or h > w * 3: raise ValueError('Building is too long or too short.') # If parameters are vertial, we firstly construct horizontal building then transpose it. # It allows not to use two additional different subtypes of the building which will simplify the code. if not is_horizontal: w, h = h, w M = Map(w, h, fill_cell=C.floor_flagged) # Create outward walls for x in range(w): M[x, 0] = wall_cell_type() M[x, h - 1] = wall_cell_type() for y in range(h): M[0, y] = wall_cell_type() M[w - 1, y] = wall_cell_type() # Randomly choose where the living part is living_left = random.choice([True, False]) living_wall_x = None barn_wall_x = None # Place central doors/corridor and calculate X-positions for vertical walls if w % 2 == 0: M[w // 2, 0] = C.floor_flagged() M[w // 2 - 1, 0] = C.floor_flagged() M[w // 2, h - 1] = C.floor_flagged() M[w // 2 - 1, h - 1] = C.floor_flagged() living_wall_x = (w // 2 - 3) if living_left else (w // 2 + 2) barn_wall_x = (w // 2 + 2) if living_left else (w // 2 - 3) else: M[w // 2, 0] = C.door_closed_wooden() M[w // 2, h - 1] = C.door_closed_wooden() living_wall_x = (w // 2 - 2) if living_left else (w // 2 + 2) barn_wall_x = (w // 2 + 2) if living_left else (w // 2 - 2) # Place vertical walls for i in range(1, h // 3): M[living_wall_x, i] = wall_cell_type() M[living_wall_x, h - i - 1] = wall_cell_type() for i in range(1, h - 1): M[barn_wall_x, i] = C.wall_fence_thin() M[barn_wall_x, h // 2] = C.door_closed_wooden() # Create living room: # Set initial coordinates lx_start = 1 if living_left else living_wall_x + 1 lx_end = living_wall_x - 1 if living_left else w - 1 beds_dx = int((lx_end - lx_start) % 2 == 0 and not living_left) beds_dy = random.choice([0, 1]) # Place beds near walls or at 1 cell from walls for bed_x in range(lx_start + beds_dx, lx_end, 2): M[bed_x, 1 + beds_dy].put(T.furniture_bed_single()) M[bed_x, h - 2 - beds_dy].put(T.furniture_bed_single()) # Place bonfire in the middle of the room or hearth on the side of the room is_bonfire = random.choice([True, False]) if is_bonfire: M[(lx_start + lx_end) // 2, h // 2].put(T.bonfire()) elif living_left: M[1, h // 2].put(T.furniture_hearth()) else: M[w - 2, h // 2].put(T.furniture_hearth()) # Create barn: # Set initial coordinates bx_start = 1 if not living_left else barn_wall_x + 1 bx_end = barn_wall_x - 1 if not living_left else w - 2 # Fill the barn floor with dirt for x in range(bx_start, bx_end + 1): for y in range(1, h - 1): M[x, y] = C.floor_dirt() is_central_barn = random.choice([True, False]) if is_central_barn: # Central barn: stalls in the center, two waterthroughs on the side for y in range(h // 3, h * 2 // 3): M[bx_start + 2, y] = C.wall_fence_thin() for x in range(bx_start + 3, bx_end - 2, 2): M[x, h // 2] = C.wall_fence_thin() M[x, h // 2 - 1].put(T.farm_mangler()) M[x, h // 2 + 1].put(T.farm_mangler()) for y in range(h // 3, h * 2 // 3): M[x + 1, y] = C.wall_fence_thin() for x in range(bx_start + 1, bx_end): M[x, 1].put(T.water_trough()) M[x, h - 2].put(T.water_trough()) else: # Side barn: stalls on the side, one waterthrough in the center for x in range(bx_start, bx_end - 1, 2): M[x, 1].put(T.farm_mangler()) M[x, h - 2].put(T.farm_mangler()) for y in range(1, h // 3): M[x + 1, y] = C.wall_fence_thin() for y in range(h * 2 // 3, h - 1): M[x + 1, y] = C.wall_fence_thin() for x in range(bx_start + 2, bx_end - 1): M[x, h // 2].put(T.water_trough()) # Transpose the building if it should be vertical if not is_horizontal: M.transpose() return M
def building_kitchengarden(w=17, h=17, wall_material=None, floor_material=None, direction='down'): """ Construct kitchen garden with big storage, garden beds and cattle pens. Constraints: - Map width and map height must be >= 17 - Map width and map height must be <= 22 - Wall material must be 'block', 'plank' or 'stone'. - Floor material must be 'grit' or 'rocks'. Parameters ---------- w : int Map width h : int Map height wall_material : str Wall's material. floor_material : str Floor's material. direction : str Direction of the kitchen garden. Can be 'up', 'down', 'left' or 'right'. """ # Initial checks. Don't accept too small/big kitchen garden if w < 17 or h < 17: raise ValueError('Building is too small: w or h < 17') elif w > 22 or h > 22: raise ValueError('Building is too big: w or h > 22') # Choose materials if not wall_material: wall_material = random.choice( [C.wall_block, C.wall_plank, C.wall_stone]) elif wall_material not in (['block', 'plank', 'stone']): raise ValueError('Wall material should be "block", "plank" or "stone"') if wall_material == 'block': wall_material = C.wall_block elif wall_material == 'plank': wall_material = C.wall_plank elif wall_material == 'stone': wall_material = C.wall_stone if not floor_material: floor_chance = random.random() if floor_chance > 0.3: floor_material = C.floor_rocks else: floor_material = C.floor_grit elif floor_material not in (['grit', 'rocks']): raise ValueError('Floor material should be "grit" or "rocks"') if floor_material == 'grit': floor_material = C.floor_grit elif floor_material == 'rocks': floor_material = C.floor_rocks M = Map(w, h, fill_cell=floor_material) storage_w = w // 3 * 2 storage_h = h // 4 + 1 garden_beds_w = w // 2 for i in range(w * h // 3): grass_x = random.randint(1, w - 1) grass_y = random.randint(0, h - 2) M[grass_x, grass_y] = random.choice( [C.flora_grass, C.flora_tree, C.floor_grass])() garden_beds = _room_garden_beds(garden_beds_w, h - storage_h + 1, wall_material, floor_material) M.meld(garden_beds, 0, storage_h - 1) cattle_pens = _room_cattle_pens(w - garden_beds_w + 1, h - storage_h + 1, wall_material, floor_material) M.meld(cattle_pens, garden_beds_w - 1, storage_h - 1) storage = _room_storage(storage_w, storage_h, wall_material, floor_material, garden_beds_w) M.meld(storage, 0, 0) for x in range(storage_w, w - 1): M[x, storage_h // 2] = floor_material() for y in range(storage_h // 2, storage_h - 1): M[storage_w + 1, y] = floor_material() if random.choice([True, False]): M.hmirror() if direction == 'up': M.vmirror() elif direction == 'left': M.transpose() elif direction == 'right': M.transpose() M.hmirror() return M