Adds Link's Awakening: DX. Fully imports and forks LADXR, with permission - https://github.com/daid/LADXR
		
			
				
	
	
		
			285 lines
		
	
	
		
			13 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
			
		
		
	
	
			285 lines
		
	
	
		
			13 KiB
		
	
	
	
		
			Python
		
	
	
	
	
	
from . import overworld
 | 
						|
from . import dungeon1
 | 
						|
from . import dungeon2
 | 
						|
from . import dungeon3
 | 
						|
from . import dungeon4
 | 
						|
from . import dungeon5
 | 
						|
from . import dungeon6
 | 
						|
from . import dungeon7
 | 
						|
from . import dungeon8
 | 
						|
from . import dungeonColor
 | 
						|
from .requirements import AND, OR, COUNT, COUNTS, FOUND, RequirementsSettings
 | 
						|
from .location import Location
 | 
						|
from ..locations.items import *
 | 
						|
from ..locations.keyLocation import KeyLocation
 | 
						|
from ..worldSetup import WorldSetup
 | 
						|
from .. import itempool
 | 
						|
from .. import mapgen
 | 
						|
 | 
						|
 | 
						|
class Logic:
 | 
						|
    def __init__(self, configuration_options, *, world_setup):
 | 
						|
        self.world_setup = world_setup
 | 
						|
        r = RequirementsSettings(configuration_options)
 | 
						|
 | 
						|
        if configuration_options.overworld == "dungeondive":
 | 
						|
            world = overworld.DungeonDiveOverworld(configuration_options, r)
 | 
						|
        elif configuration_options.overworld == "random":
 | 
						|
            world = mapgen.LogicGenerator(configuration_options, world_setup, r, world_setup.map)
 | 
						|
        else:
 | 
						|
            world = overworld.World(configuration_options, world_setup, r)
 | 
						|
 | 
						|
        if configuration_options.overworld == "nodungeons":
 | 
						|
            world.updateIndoorLocation("d1", dungeon1.NoDungeon1(configuration_options, world_setup, r).entrance)
 | 
						|
            world.updateIndoorLocation("d2", dungeon2.NoDungeon2(configuration_options, world_setup, r).entrance)
 | 
						|
            world.updateIndoorLocation("d3", dungeon3.NoDungeon3(configuration_options, world_setup, r).entrance)
 | 
						|
            world.updateIndoorLocation("d4", dungeon4.NoDungeon4(configuration_options, world_setup, r).entrance)
 | 
						|
            world.updateIndoorLocation("d5", dungeon5.NoDungeon5(configuration_options, world_setup, r).entrance)
 | 
						|
            world.updateIndoorLocation("d6", dungeon6.NoDungeon6(configuration_options, world_setup, r).entrance)
 | 
						|
            world.updateIndoorLocation("d7", dungeon7.NoDungeon7(configuration_options, world_setup, r).entrance)
 | 
						|
            world.updateIndoorLocation("d8", dungeon8.NoDungeon8(configuration_options, world_setup, r).entrance)
 | 
						|
            world.updateIndoorLocation("d0", dungeonColor.NoDungeonColor(configuration_options, world_setup, r).entrance)
 | 
						|
        elif configuration_options.overworld != "random":
 | 
						|
            world.updateIndoorLocation("d1", dungeon1.Dungeon1(configuration_options, world_setup, r).entrance)
 | 
						|
            world.updateIndoorLocation("d2", dungeon2.Dungeon2(configuration_options, world_setup, r).entrance)
 | 
						|
            world.updateIndoorLocation("d3", dungeon3.Dungeon3(configuration_options, world_setup, r).entrance)
 | 
						|
            world.updateIndoorLocation("d4", dungeon4.Dungeon4(configuration_options, world_setup, r).entrance)
 | 
						|
            world.updateIndoorLocation("d5", dungeon5.Dungeon5(configuration_options, world_setup, r).entrance)
 | 
						|
            world.updateIndoorLocation("d6", dungeon6.Dungeon6(configuration_options, world_setup, r).entrance)
 | 
						|
            world.updateIndoorLocation("d7", dungeon7.Dungeon7(configuration_options, world_setup, r).entrance)
 | 
						|
            world.updateIndoorLocation("d8", dungeon8.Dungeon8(configuration_options, world_setup, r).entrance)
 | 
						|
            world.updateIndoorLocation("d0", dungeonColor.DungeonColor(configuration_options, world_setup, r).entrance)
 | 
						|
 | 
						|
        if configuration_options.overworld != "random":
 | 
						|
            for k in world.overworld_entrance.keys():
 | 
						|
                assert k in world_setup.entrance_mapping, k
 | 
						|
            for k in world_setup.entrance_mapping.keys():
 | 
						|
                assert k in world.overworld_entrance, k
 | 
						|
 | 
						|
            for entrance, indoor in world_setup.entrance_mapping.items():
 | 
						|
                exterior = world.overworld_entrance[entrance]
 | 
						|
                if world.indoor_location[indoor] is not None:
 | 
						|
                    exterior.location.connect(world.indoor_location[indoor], exterior.requirement)
 | 
						|
                    if exterior.enterIsSet():
 | 
						|
                        exterior.location.connect(world.indoor_location[indoor], exterior.one_way_enter_requirement, one_way=True)
 | 
						|
                    if exterior.exitIsSet():
 | 
						|
                        world.indoor_location[indoor].connect(exterior.location, exterior.one_way_exit_requirement, one_way=True)
 | 
						|
 | 
						|
        egg_trigger = AND(OCARINA, SONG1)
 | 
						|
        if configuration_options.logic == 'glitched' or configuration_options.logic == 'hell':
 | 
						|
            egg_trigger = OR(AND(OCARINA, SONG1), BOMB)
 | 
						|
 | 
						|
        if world_setup.goal == "seashells":
 | 
						|
            world.nightmare.connect(world.egg, COUNT(SEASHELL, 20))
 | 
						|
        elif world_setup.goal in ("raft", "bingo", "bingo-full"):
 | 
						|
            world.nightmare.connect(world.egg, egg_trigger)
 | 
						|
        else:
 | 
						|
            goal = int(world_setup.goal)
 | 
						|
            if goal < 0:
 | 
						|
                world.nightmare.connect(world.egg, None)
 | 
						|
            elif goal == 0:
 | 
						|
                world.nightmare.connect(world.egg, egg_trigger)
 | 
						|
            elif goal == 8:
 | 
						|
                world.nightmare.connect(world.egg, AND(egg_trigger, INSTRUMENT1, INSTRUMENT2, INSTRUMENT3, INSTRUMENT4, INSTRUMENT5, INSTRUMENT6, INSTRUMENT7, INSTRUMENT8))
 | 
						|
            else:
 | 
						|
                world.nightmare.connect(world.egg, AND(egg_trigger, COUNTS([INSTRUMENT1, INSTRUMENT2, INSTRUMENT3, INSTRUMENT4, INSTRUMENT5, INSTRUMENT6, INSTRUMENT7, INSTRUMENT8], goal)))
 | 
						|
 | 
						|
        # if configuration_options.dungeon_items == 'keysy':
 | 
						|
        #     for n in range(9):
 | 
						|
        #         for count in range(9):
 | 
						|
        #             world.start.add(KeyLocation("KEY%d" % (n + 1)))
 | 
						|
        #         world.start.add(KeyLocation("NIGHTMARE_KEY%d" % (n + 1)))
 | 
						|
 | 
						|
        self.world = world
 | 
						|
        self.start = world.start
 | 
						|
        self.windfish = world.windfish
 | 
						|
        self.location_list = []
 | 
						|
        self.iteminfo_list = []
 | 
						|
 | 
						|
        self.__location_set = set()
 | 
						|
        self.__recursiveFindAll(self.start)
 | 
						|
        del self.__location_set
 | 
						|
 | 
						|
        for ii in self.iteminfo_list:
 | 
						|
            ii.configure(configuration_options)
 | 
						|
 | 
						|
    def dumpFlatRequirements(self):
 | 
						|
        def __rec(location, req):
 | 
						|
            if hasattr(location, "flat_requirements"):
 | 
						|
                new_flat_requirements = requirements.mergeFlat(location.flat_requirements, requirements.flatten(req))
 | 
						|
                if new_flat_requirements == location.flat_requirements:
 | 
						|
                    return
 | 
						|
                location.flat_requirements = new_flat_requirements
 | 
						|
            else:
 | 
						|
                location.flat_requirements = requirements.flatten(req)
 | 
						|
            for connection, requirement in location.simple_connections:
 | 
						|
                __rec(connection, AND(req, requirement) if req else requirement)
 | 
						|
            for connection, requirement in location.gated_connections:
 | 
						|
                __rec(connection, AND(req, requirement) if req else requirement)
 | 
						|
        __rec(self.start, None)
 | 
						|
        for ii in self.iteminfo_list:
 | 
						|
            print(ii)
 | 
						|
            for fr in ii._location.flat_requirements:
 | 
						|
                print("    " + ", ".join(sorted(map(str, fr))))
 | 
						|
 | 
						|
    def __recursiveFindAll(self, location):
 | 
						|
        if location in self.__location_set:
 | 
						|
            return
 | 
						|
        self.location_list.append(location)
 | 
						|
        self.__location_set.add(location)
 | 
						|
        for ii in location.items:
 | 
						|
            self.iteminfo_list.append(ii)
 | 
						|
        for connection, requirement in location.simple_connections:
 | 
						|
            self.__recursiveFindAll(connection)
 | 
						|
        for connection, requirement in location.gated_connections:
 | 
						|
            self.__recursiveFindAll(connection)
 | 
						|
 | 
						|
 | 
						|
class MultiworldLogic:
 | 
						|
    def __init__(self, settings, rnd=None, *, world_setups=None):
 | 
						|
        assert rnd or world_setups
 | 
						|
        self.worlds = []
 | 
						|
        self.start = Location()
 | 
						|
        self.location_list = [self.start]
 | 
						|
        self.iteminfo_list = []
 | 
						|
 | 
						|
        for n in range(settings.multiworld):
 | 
						|
            options = settings.multiworld_settings[n]
 | 
						|
            world = None
 | 
						|
            if world_setups:
 | 
						|
                world = Logic(options, world_setup=world_setups[n])
 | 
						|
            else:
 | 
						|
                for cnt in range(1000):  # Try the world setup in case entrance randomization generates unsolvable logic
 | 
						|
                    world_setup = WorldSetup()
 | 
						|
                    world_setup.randomize(options, rnd)
 | 
						|
                    world = Logic(options, world_setup=world_setup)
 | 
						|
                    if options.entranceshuffle not in ("advanced", "expert", "insanity") or len(world.iteminfo_list) == sum(itempool.ItemPool(options, rnd).toDict().values()):
 | 
						|
                        break
 | 
						|
 | 
						|
            for ii in world.iteminfo_list:
 | 
						|
                ii.world = n
 | 
						|
 | 
						|
            req_done_set = set()
 | 
						|
            for loc in world.location_list:
 | 
						|
                loc.simple_connections = [(target, addWorldIdToRequirements(req_done_set, n, req)) for target, req in loc.simple_connections]
 | 
						|
                loc.gated_connections = [(target, addWorldIdToRequirements(req_done_set, n, req)) for target, req in loc.gated_connections]
 | 
						|
                loc.items = [MultiworldItemInfoWrapper(n, options, ii) for ii in loc.items]
 | 
						|
                self.iteminfo_list += loc.items
 | 
						|
 | 
						|
            self.worlds.append(world)
 | 
						|
            self.start.simple_connections += world.start.simple_connections
 | 
						|
            self.start.gated_connections += world.start.gated_connections
 | 
						|
            self.start.items += world.start.items
 | 
						|
            world.start.items.clear()
 | 
						|
            self.location_list += world.location_list
 | 
						|
 | 
						|
        self.entranceMapping = None
 | 
						|
 | 
						|
 | 
						|
class MultiworldMetadataWrapper:
 | 
						|
    def __init__(self, world, metadata):
 | 
						|
        self.world = world
 | 
						|
        self.metadata = metadata
 | 
						|
 | 
						|
    @property
 | 
						|
    def name(self):
 | 
						|
        return self.metadata.name
 | 
						|
 | 
						|
    @property
 | 
						|
    def area(self):
 | 
						|
        return "P%d %s" % (self.world + 1, self.metadata.area)
 | 
						|
 | 
						|
 | 
						|
class MultiworldItemInfoWrapper:
 | 
						|
    def __init__(self, world, configuration_options, target):
 | 
						|
        self.world = world
 | 
						|
        self.world_count = configuration_options.multiworld
 | 
						|
        self.target = target
 | 
						|
        self.dungeon_items = configuration_options.dungeon_items
 | 
						|
        self.MULTIWORLD_OPTIONS = None
 | 
						|
        self.item = None
 | 
						|
 | 
						|
    @property
 | 
						|
    def nameId(self):
 | 
						|
        return self.target.nameId
 | 
						|
 | 
						|
    @property
 | 
						|
    def forced_item(self):
 | 
						|
        if self.target.forced_item is None:
 | 
						|
            return None
 | 
						|
        if "_W" in self.target.forced_item:
 | 
						|
            return self.target.forced_item
 | 
						|
        return "%s_W%d" % (self.target.forced_item, self.world)
 | 
						|
 | 
						|
    @property
 | 
						|
    def room(self):
 | 
						|
        return self.target.room
 | 
						|
 | 
						|
    @property
 | 
						|
    def metadata(self):
 | 
						|
        return MultiworldMetadataWrapper(self.world, self.target.metadata)
 | 
						|
 | 
						|
    @property
 | 
						|
    def MULTIWORLD(self):
 | 
						|
        return self.target.MULTIWORLD
 | 
						|
 | 
						|
    def read(self, rom):
 | 
						|
        world = rom.banks[0x3E][0x3300 + self.target.room] if self.target.MULTIWORLD else self.world
 | 
						|
        return "%s_W%d" % (self.target.read(rom), world)
 | 
						|
 | 
						|
    def getOptions(self):
 | 
						|
        if self.MULTIWORLD_OPTIONS is None:
 | 
						|
            options = self.target.getOptions()
 | 
						|
            if self.target.MULTIWORLD and len(options) > 1:
 | 
						|
                self.MULTIWORLD_OPTIONS = []
 | 
						|
                for n in range(self.world_count):
 | 
						|
                    self.MULTIWORLD_OPTIONS += ["%s_W%d" % (t, n) for t in options if n == self.world or self.canMultiworld(t)]
 | 
						|
            else:
 | 
						|
                self.MULTIWORLD_OPTIONS = ["%s_W%d" % (t, self.world) for t in options]
 | 
						|
        return self.MULTIWORLD_OPTIONS
 | 
						|
 | 
						|
    def patch(self, rom, option):
 | 
						|
        idx = option.rfind("_W")
 | 
						|
        world = int(option[idx+2:])
 | 
						|
        option = option[:idx]
 | 
						|
        if not self.target.MULTIWORLD:
 | 
						|
            assert self.world == world
 | 
						|
            self.target.patch(rom, option)
 | 
						|
        else:
 | 
						|
            self.target.patch(rom, option, multiworld=world)
 | 
						|
 | 
						|
    # Return true if the item is allowed to be placed in any world, or false if it is
 | 
						|
    # world specific for this check.
 | 
						|
    def canMultiworld(self, option):
 | 
						|
        if self.dungeon_items in {'', 'smallkeys'}:
 | 
						|
            if option.startswith("MAP"):
 | 
						|
                return False
 | 
						|
            if option.startswith("COMPASS"):
 | 
						|
                return False
 | 
						|
            if option.startswith("STONE_BEAK"):
 | 
						|
                return False
 | 
						|
        if self.dungeon_items in {'', 'localkeys'}:
 | 
						|
            if option.startswith("KEY"):
 | 
						|
                return False
 | 
						|
        if self.dungeon_items in {'', 'localkeys', 'localnightmarekey', 'smallkeys'}:
 | 
						|
            if option.startswith("NIGHTMARE_KEY"):
 | 
						|
                return False
 | 
						|
        return True
 | 
						|
 | 
						|
    @property
 | 
						|
    def location(self):
 | 
						|
        return self.target.location
 | 
						|
 | 
						|
    def __repr__(self):
 | 
						|
        return "W%d:%s" % (self.world, repr(self.target))
 | 
						|
 | 
						|
 | 
						|
def addWorldIdToRequirements(req_done_set, world, req):
 | 
						|
    if req is None:
 | 
						|
        return None
 | 
						|
    if isinstance(req, str):
 | 
						|
        return "%s_W%d" % (req, world)
 | 
						|
    if req in req_done_set:
 | 
						|
        return req
 | 
						|
    return req.copyWithModifiedItemNames(lambda item: "%s_W%d" % (item, world))
 |