47 lines
1.5 KiB
Python
47 lines
1.5 KiB
Python
from source.dungeon.EnemyList import EnemySprite, SpriteType, sprite_translation, enemy_names
|
|
|
|
def can_combine_req(req1, req2):
|
|
for i in range(0, 4):
|
|
if req1.sub_groups[i] and req2.sub_groups[i]:
|
|
if len(set(req1.sub_groups[i]).intersection(req2.sub_groups[i])) == 0:
|
|
return False
|
|
return True
|
|
|
|
def get_enemy_map(mode, reqs, vanilla_map):
|
|
data = {}
|
|
|
|
if mode != "mimics":
|
|
return data
|
|
|
|
green_mimic = reqs.get((EnemySprite.GreenMimic, 0))
|
|
|
|
for room_id, sprites in vanilla_map.items():
|
|
data[room_id] = {}
|
|
for idx, sprite in enumerate(sprites):
|
|
subtype = 0 if sprite.sub_type != SpriteType.Overlord else sprite.sub_type
|
|
req = reqs.get((sprite.kind, subtype))
|
|
if not req or isinstance(req, dict) or req.boss:
|
|
continue
|
|
if req.static:
|
|
if not can_combine_req(green_mimic, req):
|
|
data[room_id] = {}
|
|
break
|
|
continue
|
|
if req.killable:
|
|
data[room_id][idx] = 'GreenMimic'
|
|
else:
|
|
data[room_id][idx] = enemy_names[sprite.kind]
|
|
if len(data[room_id]) == 0:
|
|
del data[room_id]
|
|
return data
|
|
|
|
|
|
def get_enemy_map_ow(mode, data_tables):
|
|
reqs = data_tables.sprite_requirements
|
|
return get_enemy_map(mode, reqs, data_tables.ow_enemy_table)
|
|
|
|
def get_enemy_map_uw(mode, data_tables):
|
|
reqs = data_tables.sprite_requirements
|
|
return get_enemy_map(mode, reqs, data_tables.uw_enemy_table.room_map)
|
|
|