33 lines
958 B
Python
33 lines
958 B
Python
from functools import partial
|
|
from parse.generic import tokenize
|
|
from parse.generic import number
|
|
|
|
def tokenize_lines(lines):
|
|
for line in lines:
|
|
if '_Coll:' in line:
|
|
yield tokenize.block(line, delim='::')
|
|
elif 'coll_tiles' in line:
|
|
tokens, = tokenize.block(line, delim='::')
|
|
yield tokenize.line(tokens)
|
|
|
|
def flatten(tokens):
|
|
stack = []
|
|
for t in tokens:
|
|
if t[0] == 'coll_tiles':
|
|
tile_ids = t[1] if len(t) == 2 else []
|
|
for name in stack:
|
|
yield name, list(map(number.parse, tile_ids))
|
|
stack = []
|
|
elif len(t) == 1:
|
|
name, = t
|
|
stack.append(name)
|
|
|
|
def parse(prefix):
|
|
path = prefix / 'data/tilesets/collision_tile_ids.asm'
|
|
with open(path) as f:
|
|
tokens = tokenize_lines(f.read().split('\n'))
|
|
l = list(flatten(tokens))
|
|
d = dict(l)
|
|
assert len(l) == len(d)
|
|
return d
|