sm_rando/map_rom_test.py at master · randomcodegen/sm_rando · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
from world_rando import coord
from world_rando import concrete_map
from world_rando import item_order_graph
from world_rando import map_gen
from world_rando import map_viz
from rom_tools import rom_manager
from encoding import sm_global
import random
import collections
import matplotlib
matplotlib.use('Agg')
import matplotlib.pyplot as plt

def rand_m(p1, p2):
    return concrete_map.manhattan(p1, p2) + random.uniform(0,6)

def less_rand_d(p1, p2):
    return concrete_map.euclidean(p1, p2) + random.uniform(0, 6)

def get_path_info(paths):
    npaths = len(paths)
    lens = [len(p) for p in paths]
    total_length = sum(lens)
    return total_length, npaths

if __name__ == "__main__":
    o, g, rsg, es, ro = item_order_graph.abstract_map()
    print(ro)
    print(o)
    cmap = {}
    ntiles = 0
    room_dims = []
    path_length = 0
    npaths = 0
    for region, graph in rsg.items():
        graph.visualize("output/a_" + region)
        print("Generating map for " + region)
        # Can be as large as 64/32
        dimensions = coord.Coord(54,30)
        cmap[region], rooms, paths = map_gen.less_naive_gen(dimensions, less_rand_d, graph, es)
        ntiles += len(cmap[region])
        for room in rooms.values():
            room_dims.append(len(room))
        l,n = get_path_info(paths)
        path_length += l
        npaths += n

    print("Tiles: " + str(ntiles))
    print("Average room size: " + str(sum(room_dims)/len(room_dims)))
    print("Average path length: " + str(path_length/npaths))
    rom = rom_manager.RomManager("../sm_guinea_pig_map_edit_copy.smc")
    for region, rcmap in cmap.items():
        # put it on the ROM
        hidden, tiles = sm_global.region_map_locs[region]
        tmap = map_viz.tiles_parse("encoding/dsl/tiles.txt")
        rcmap_ts = map_viz.cmap_to_tuples(rcmap, tmap) #TODO: negative numbers?
        rom.placeCmap(rcmap_ts, tiles, hidden)
        #output
        map_viz.map_viz(rcmap, "output/" + region + ".png", "encoding/map_tiles")
    rom.saveRom()