Add all-mimics mode

source/enemizer/SpecialEnemyModes.py

@@ -0,0 +1,46 @@
+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)
+