basic_test.py

from godcomplex import World
from godcomplex import PrintUtil
from godcomplex import Navigate
import numpy
from PIL import Image
import sqlite3
from colors import color

MAX_HISTORY_SHOWN = 30

def print_at_cell(coord, text):
    x, y = coord
    print('\033[{};{}H{}'.format(y + 2, x + 1, text))

if __name__ == '__main__':
    db = sqlite3.connect('file:worldgen.db?mode=ro', uri=True)
    world = World(db=db)

    print('Generating terrain...')
    world.init_terrain()
    PrintUtil.print_terrain(world)
    print('========================')

    print('Generating moisture...')
    world.init_moisture()
    PrintUtil.print_digit_layer(world, 'moisture')
    print('========================')

    print('Generating biomes...')
    world.init_biomes()
    PrintUtil.print_terrain(world)
    print('========================')

    print('Generating peoples...')
    world.init_peoples()
    print('========================')

    print('The world has started. Input newline to run a single simulation step.')

    current_view = 'activity'
    while True:
        cmd = input()
        if cmd.startswith('ag'):
            print('\n'.join([str(a) for a in world.agents.values()]))
            input()
        elif cmd.startswith('p'):
            try:
                args = [int(v) for v in cmd.split()[1:]]
                if len(args) == 2:
                    args = args + args
                print(args)
                x1, y1, x2, y2 = [int(v) for v in args]
                c1 = (x1, y1)
                c2 = (x2, y2)
                path = Navigate.path(world, c1, lambda _, coord: coord == c2)
                if path:
                    print(path)
                    for c in path:
                        print_at_cell(c, color('>', fg='red'))
                else:
                    print('No path found from', c1, 'to', c2)
            except ValueError as e:
                print(e)
                print('USAGE: p <x1> <x2> <y1> <y2>')
            input()
        elif cmd.startswith('se'):
            print('\n'.join([str(a) for a in world.agents.values() if a.agent_type == 'SETTLEMENT']))
            input()
        elif cmd.startswith('h'):
            print('\n'.join([str(e) for e in world.history.events[-MAX_HISTORY_SHOWN:]]))
            input()
        elif cmd.startswith('f'):
            print('\n'.join([str(a) for a in world.factions.values()]))
            input()
        elif cmd.startswith('a'):
            current_view = 'activity'
        elif cmd.startswith('t'):
            current_view = 'terrain'
        elif cmd.startswith('g'):
            Image.fromarray(numpy.asarray(world.elevation) * 255) \
                .convert('RGB') \
                .save('test.bmp')
        else:
            world.step()

        print('\033[;HYear {}, First Age of the World\033[0K'.format(
            world.history.current_timestamp))
        if current_view == 'activity':
            PrintUtil.print_activity(world)
        elif current_view == 'terrain':
            PrintUtil.print_terrain(world)

        print('Currently in {} view. '.format(current_view) +
            'Commands:' + '  '.join(['[ag]ent list',
                '[se]ttlement list',
                '[f]action list',
                '[h]istorical events',
                '[a]ctivity view',
                '[t]errain view',
                '[g]reyscale export',
                '[p]ath']) + '\033[0J')