MiniMapGen/main.py

import sys
import functools
import random
from noise import pnoise2
from termcolor import colored

def addition(a, b):
    return a + b

seed = 0
seed_codes = sys.argv[1].encode("utf-8")
if seed_codes:
    seed = functools.reduce(addition, seed_codes)

scale = 25.0
world_w, world_h = 16, 16

water_deepest = -0.3
water_deep = -0.2
water = -0.1
grass = 0.3
grass_high = 0.6
mountains = 1
block = "█"
block = block + block

def generate_world(seed):
    try:
        print("Seed: " + str(seed))
        random.seed(seed)
        world = []
        for y in range(1, world_h + 1):
            row = []
            for x in range(1, world_w + 1):
                row_noise = pnoise2(
                        x / scale,
                        y / scale,
                        octaves=10,
                        persistence=0.5,
                        lacunarity=2.0,
                        base=seed
                    )
                row.append(row_noise)
            world.append(row)
        return world
    except Exception as e:
        print(e)

def terrain_matches(cell1, cell2):
    return cell1["is_land"] == cell2["is_land"] or cell1["is_water"] == cell2["is_water"]
        
def group_terrain(world):
    world_terrain_without_group = []
    group_id = 0
    for y in range(world_h):
        row = []
        for x in range(world_w):
            cell = world[y][x]
            terrain = {
                "v": cell,
                "x": x,
                "y": y,
                "is_land": cell > water,
                "is_water": cell <= water
                }
            row.append(terrain)            
        world_terrain_without_group.append(row)

    world_terrain = []
    for y in range(world_h):
        row = []
        for x in range(world_w):
            cell = world_terrain_without_group[y][x]
            north = world_terrain_without_group[y - 1][x] if y > 0 else None
            south = world_terrain_without_group[y][x] if y < world_h else None
            east = world_terrain_without_group[y][x] if x < world_w else None
            west = world_terrain_without_group[y][x - 1] if x > 0 else None
            if north and terrain_matches(cell, north):
                print(f"{x}:{y} - north")
            elif east and terrain_matches(cell, east):
                print(f"{x}:{y} - east")
            elif south and terrain_matches(cell, south):
                print(f"{x}:{y} - south")
            elif west and terrain_matches(cell, west):
                print(f"{x}:{y} - west")
            # else:
            #     print("new group")
            #     cell["group_id"] = group_id
            #     group_id += 1
            #print(cell)
            row.append(cell)
        world_terrain.append(row)

    print(f"Number of groups: {group_id}")
    return world_terrain

def add_home(world):
    print("Choosing home location")
    home = (0, 0)

    # build only on "grass"
    valid_locations = []
    for y in range(world_h):
        if y > 0 and y < world_h - 1:
            for x in range(world_w):
                if x > 0 and x < world_w - 1:
                    cell = world[y][x]
                    if cell > water and cell <= grass:
                        valid_locations.append((x, y))

    location = random.randrange(len(valid_locations))
    home = valid_locations[location]
    print(home)
    return home

def add_dock(world, home):
    print("Adding a dock")
    # find the closest bit of water to the house?
    # find the biggest body of water

def distribution(world):
    total = world_w * world_h
    above_water = 0

    for y in range(world_h):
        for x in range(world_w):
            cell = world[y][x]
            if cell > water:
                above_water = above_water + 1

    return above_water / total

def print_map(world, home):
    print("  ", end="", flush=True)
    for x in range(world_w):
        to_print = str(x) + " " if x < 10 else x
        print(to_print, end="", flush=True)
    print("")
    for y in range(world_h):
        to_print = str(y) + " " if y < 10 else y
        print(to_print, end="", flush=True)
        for x in range(world_w):
            cell = world[y][x]
            if home[0] == x and home[1] == y:
                print(colored(block, "magenta"), end="", flush=True)
            elif cell <= water_deepest:
                print(colored(block, "grey"), end="", flush=True)
            elif cell <= water_deep:
                print(colored(block, "blue", attrs=["dark"]), end="", flush=True)
            elif cell <= water:
                print(colored(block, "blue"), end="", flush=True)
            elif cell <= grass:
                print(colored(block, "green"), end="", flush=True)
            elif cell <= grass_high:
                print(colored(block, "green", attrs=["dark"]), end="", flush=True)
            else:
                print(colored(block, "white"), end="", flush=True)
        print("")

def print_terrain(world):
    colours = ["grey", "red", "green", "yellow", "blue", "magenta", "cyan", "white"]
    print("  ", end="", flush=True)
    for x in range(world_w):
        to_print = str(x) + " " if x < 10 else x
        print(to_print, end="", flush=True)
    print("")
    for y in range(world_h):
        to_print = str(y) + " " if y < 10 else y
        print(to_print, end="", flush=True)
        for x in range(world_w):
            cell = world[y][x]
            print(colored(block, colours[cell["group_id"]]), end="", flush=True)
        print("")

world = generate_world(seed)
i = 1
land_percentage = distribution(world)
while (land_percentage < 0.25 or 1 - land_percentage < 0.25):
    world = generate_world(seed + i)
    land_percentage = distribution(world)
    i = i + 1

world_terrain = group_terrain(world)

print(str(i - 1) + " seed iterations")

home = add_home(world)
dock = add_dock(world, home)
print_map(world, home)
print_terrain(world_terrain)