606 lines
26 KiB
Python
606 lines
26 KiB
Python
import copy
|
|
import random
|
|
from ..logic.logic import Logic
|
|
from ..utils.parameters import Knows
|
|
from ..graph.location import locationsDict
|
|
from ..rom.rom import snes_to_pc
|
|
from ..utils import log
|
|
|
|
# order expected by ROM patches
|
|
graphAreas = [
|
|
"Ceres",
|
|
"Crateria",
|
|
"GreenPinkBrinstar",
|
|
"RedBrinstar",
|
|
"WreckedShip",
|
|
"Kraid",
|
|
"Norfair",
|
|
"Crocomire",
|
|
"LowerNorfair",
|
|
"WestMaridia",
|
|
"EastMaridia",
|
|
"Tourian"
|
|
]
|
|
|
|
vanillaTransitions = [
|
|
('Lower Mushrooms Left', 'Green Brinstar Elevator'),
|
|
('Morph Ball Room Left', 'Green Hill Zone Top Right'),
|
|
('Moat Right', 'West Ocean Left'),
|
|
('Keyhunter Room Bottom', 'Red Brinstar Elevator'),
|
|
('Noob Bridge Right', 'Red Tower Top Left'),
|
|
('Crab Maze Left', 'Le Coude Right'),
|
|
('Kronic Boost Room Bottom Left', 'Lava Dive Right'),
|
|
('Crocomire Speedway Bottom', 'Crocomire Room Top'),
|
|
('Three Muskateers Room Left', 'Single Chamber Top Right'),
|
|
('Warehouse Entrance Left', 'East Tunnel Right'),
|
|
('East Tunnel Top Right', 'Crab Hole Bottom Left'),
|
|
('Caterpillar Room Top Right', 'Red Fish Room Left'),
|
|
('Glass Tunnel Top', 'Main Street Bottom'),
|
|
('Green Pirates Shaft Bottom Right', 'Golden Four'),
|
|
('Warehouse Entrance Right', 'Warehouse Zeela Room Left'),
|
|
('Crab Shaft Right', 'Aqueduct Top Left')
|
|
]
|
|
|
|
vanillaBossesTransitions = [
|
|
('KraidRoomOut', 'KraidRoomIn'),
|
|
('PhantoonRoomOut', 'PhantoonRoomIn'),
|
|
('DraygonRoomOut', 'DraygonRoomIn'),
|
|
('RidleyRoomOut', 'RidleyRoomIn')
|
|
]
|
|
|
|
# vanilla escape transition in first position
|
|
vanillaEscapeTransitions = [
|
|
('Tourian Escape Room 4 Top Right', 'Climb Bottom Left'),
|
|
('Brinstar Pre-Map Room Right', 'Green Brinstar Main Shaft Top Left'),
|
|
('Wrecked Ship Map Room', 'Basement Left'),
|
|
('Norfair Map Room', 'Business Center Mid Left'),
|
|
('Maridia Map Room', 'Crab Hole Bottom Right')
|
|
]
|
|
|
|
vanillaEscapeAnimalsTransitions = [
|
|
('Flyway Right 0', 'Bomb Torizo Room Left'),
|
|
('Flyway Right 1', 'Bomb Torizo Room Left'),
|
|
('Flyway Right 2', 'Bomb Torizo Room Left'),
|
|
('Flyway Right 3', 'Bomb Torizo Room Left'),
|
|
('Bomb Torizo Room Left Animals', 'Flyway Right')
|
|
]
|
|
|
|
escapeSource = 'Tourian Escape Room 4 Top Right'
|
|
escapeTargets = ['Green Brinstar Main Shaft Top Left', 'Basement Left', 'Business Center Mid Left', 'Crab Hole Bottom Right']
|
|
|
|
locIdsByAreaAddresses = {
|
|
"Ceres": snes_to_pc(0xA1F568),
|
|
"Crateria": snes_to_pc(0xA1F569),
|
|
"GreenPinkBrinstar": snes_to_pc(0xA1F57B),
|
|
"RedBrinstar": snes_to_pc(0xA1F58C),
|
|
"WreckedShip": snes_to_pc(0xA1F592),
|
|
"Kraid": snes_to_pc(0xA1F59E),
|
|
"Norfair": snes_to_pc(0xA1F5A2),
|
|
"Crocomire": snes_to_pc(0xA1F5B2),
|
|
"LowerNorfair": snes_to_pc(0xA1F5B8),
|
|
"WestMaridia": snes_to_pc(0xA1F5C3),
|
|
"EastMaridia": snes_to_pc(0xA1F5CB),
|
|
"Tourian": snes_to_pc(0xA1F5D7)
|
|
}
|
|
|
|
def getAccessPoint(apName, apList=None):
|
|
if apList is None:
|
|
apList = Logic.accessPoints
|
|
return next(ap for ap in apList if ap.Name == apName)
|
|
|
|
class GraphUtils:
|
|
log = log.get('GraphUtils')
|
|
|
|
def getStartAccessPointNames():
|
|
return [ap.Name for ap in Logic.accessPoints if ap.Start is not None]
|
|
|
|
def getStartAccessPointNamesCategory():
|
|
ret = {'regular': [], 'custom': [], 'area': []}
|
|
for ap in Logic.accessPoints:
|
|
if ap.Start == None:
|
|
continue
|
|
elif 'areaMode' in ap.Start and ap.Start['areaMode'] == True:
|
|
ret['area'].append(ap.Name)
|
|
elif GraphUtils.isStandardStart(ap.Name):
|
|
ret['regular'].append(ap.Name)
|
|
else:
|
|
ret['custom'].append(ap.Name)
|
|
return ret
|
|
|
|
def isStandardStart(startApName):
|
|
return startApName == 'Ceres' or startApName == 'Landing Site'
|
|
|
|
def getPossibleStartAPs(areaMode, maxDiff, morphPlacement, player):
|
|
ret = []
|
|
refused = {}
|
|
allStartAPs = GraphUtils.getStartAccessPointNames()
|
|
for apName in allStartAPs:
|
|
start = getAccessPoint(apName).Start
|
|
ok = True
|
|
cause = ""
|
|
if 'knows' in start:
|
|
for k in start['knows']:
|
|
if not Knows.knowsDict[player].knows(k, maxDiff):
|
|
ok = False
|
|
cause += Knows.desc[k]['display']+" is not known. "
|
|
break
|
|
if 'areaMode' in start and start['areaMode'] != areaMode:
|
|
ok = False
|
|
cause += "Start location available only with area randomization enabled. "
|
|
if 'forcedEarlyMorph' in start and start['forcedEarlyMorph'] == True and morphPlacement == 'late':
|
|
ok = False
|
|
cause += "Start location unavailable with late morph placement. "
|
|
if ok:
|
|
ret.append(apName)
|
|
else:
|
|
refused[apName] = cause
|
|
return ret, refused
|
|
|
|
def updateLocClassesStart(startGraphArea, split, possibleMajLocs, preserveMajLocs, nLocs):
|
|
locs = locationsDict
|
|
preserveMajLocs = [locs[locName] for locName in preserveMajLocs if locs[locName].isClass(split)]
|
|
possLocs = [locs[locName] for locName in possibleMajLocs][:nLocs]
|
|
GraphUtils.log.debug("possLocs="+str([loc.Name for loc in possLocs]))
|
|
candidates = [loc for loc in locs.values() if loc.GraphArea == startGraphArea and loc.isClass(split) and loc not in preserveMajLocs]
|
|
remLocs = [loc for loc in locs.values() if loc not in possLocs and loc not in candidates and loc.isClass(split)]
|
|
newLocs = []
|
|
while len(newLocs) < nLocs:
|
|
if len(candidates) == 0:
|
|
candidates = remLocs
|
|
loc = possLocs.pop(random.randint(0,len(possLocs)-1))
|
|
newLocs.append(loc)
|
|
loc.setClass([split])
|
|
if not loc in preserveMajLocs:
|
|
GraphUtils.log.debug("newMajor="+loc.Name)
|
|
loc = candidates.pop(random.randint(0,len(candidates)-1))
|
|
loc.setClass(["Minor"])
|
|
GraphUtils.log.debug("replaced="+loc.Name)
|
|
|
|
def getGraphPatches(startApName):
|
|
ap = getAccessPoint(startApName)
|
|
return ap.Start['patches'] if 'patches' in ap.Start else []
|
|
|
|
def createBossesTransitions():
|
|
transitions = vanillaBossesTransitions
|
|
def isVanilla():
|
|
for t in vanillaBossesTransitions:
|
|
if t not in transitions:
|
|
return False
|
|
return True
|
|
while isVanilla():
|
|
transitions = []
|
|
srcs = []
|
|
dsts = []
|
|
for (src,dst) in vanillaBossesTransitions:
|
|
srcs.append(src)
|
|
dsts.append(dst)
|
|
while len(srcs) > 0:
|
|
src = srcs.pop(random.randint(0,len(srcs)-1))
|
|
dst = dsts.pop(random.randint(0,len(dsts)-1))
|
|
transitions.append((src,dst))
|
|
return transitions
|
|
|
|
def createAreaTransitions(lightAreaRando=False):
|
|
if lightAreaRando:
|
|
return GraphUtils.createLightAreaTransitions()
|
|
else:
|
|
return GraphUtils.createRegularAreaTransitions()
|
|
|
|
def createRegularAreaTransitions(apList=None, apPred=None):
|
|
if apList is None:
|
|
apList = Logic.accessPoints
|
|
if apPred is None:
|
|
apPred = lambda ap: ap.isArea()
|
|
tFrom = []
|
|
tTo = []
|
|
apNames = [ap.Name for ap in apList if apPred(ap) == True]
|
|
transitions = []
|
|
|
|
def findTo(trFrom):
|
|
ap = getAccessPoint(trFrom, apList)
|
|
fromArea = ap.GraphArea
|
|
targets = [apName for apName in apNames if apName not in tTo and getAccessPoint(apName, apList).GraphArea != fromArea]
|
|
if len(targets) == 0: # fallback if no area transition is found
|
|
targets = [apName for apName in apNames if apName != ap.Name]
|
|
if len(targets) == 0: # extreme fallback: loop on itself
|
|
targets = [ap.Name]
|
|
return random.choice(targets)
|
|
|
|
def addTransition(src, dst):
|
|
tFrom.append(src)
|
|
tTo.append(dst)
|
|
|
|
while len(apNames) > 0:
|
|
sources = [apName for apName in apNames if apName not in tFrom]
|
|
src = random.choice(sources)
|
|
dst = findTo(src)
|
|
transitions.append((src, dst))
|
|
addTransition(src, dst)
|
|
addTransition(dst, src)
|
|
toRemove = [apName for apName in apNames if apName in tFrom and apName in tTo]
|
|
for apName in toRemove:
|
|
apNames.remove(apName)
|
|
return transitions
|
|
|
|
def getAPs(apPredicate, apList=None):
|
|
if apList is None:
|
|
apList = Logic.accessPoints
|
|
return [ap for ap in apList if apPredicate(ap) == True]
|
|
|
|
def loopUnusedTransitions(transitions, apList=None):
|
|
if apList is None:
|
|
apList = Logic.accessPoints
|
|
usedAPs = set()
|
|
for (src,dst) in transitions:
|
|
usedAPs.add(getAccessPoint(src, apList))
|
|
usedAPs.add(getAccessPoint(dst, apList))
|
|
unusedAPs = [ap for ap in apList if not ap.isInternal() and ap not in usedAPs]
|
|
for ap in unusedAPs:
|
|
transitions.append((ap.Name, ap.Name))
|
|
|
|
# crateria can be forced in corner cases
|
|
def createMinimizerTransitions(startApName, locLimit, forcedAreas=None):
|
|
if forcedAreas is None:
|
|
forcedAreas = []
|
|
if startApName == 'Ceres':
|
|
startApName = 'Landing Site'
|
|
startAp = getAccessPoint(startApName)
|
|
def getNLocs(locsPredicate, locList=None):
|
|
if locList is None:
|
|
locList = Logic.locations
|
|
# leave out bosses and count post boss locs systematically
|
|
return len([loc for loc in locList if locsPredicate(loc) == True and not loc.SolveArea.endswith(" Boss") and not loc.isBoss()])
|
|
availAreas = list(sorted({ap.GraphArea for ap in Logic.accessPoints if ap.GraphArea != startAp.GraphArea and getNLocs(lambda loc: loc.GraphArea == ap.GraphArea) > 0}))
|
|
areas = [startAp.GraphArea]
|
|
if startAp.GraphArea in forcedAreas:
|
|
forcedAreas.remove(startAp.GraphArea)
|
|
GraphUtils.log.debug("availAreas: {}".format(availAreas))
|
|
GraphUtils.log.debug("forcedAreas: {}".format(forcedAreas))
|
|
GraphUtils.log.debug("areas: {}".format(areas))
|
|
inBossCheck = lambda ap: ap.Boss and ap.Name.endswith("In")
|
|
nLocs = 0
|
|
transitions = []
|
|
usedAPs = []
|
|
trLimit = 5
|
|
locLimit -= 3 # 3 "post boss" locs will always be available, and are filtered out in getNLocs
|
|
def openTransitions():
|
|
nonlocal areas, inBossCheck, usedAPs
|
|
return GraphUtils.getAPs(lambda ap: ap.GraphArea in areas and not ap.isInternal() and not inBossCheck(ap) and not ap in usedAPs)
|
|
while nLocs < locLimit or len(openTransitions()) < trLimit or len(forcedAreas) > 0:
|
|
GraphUtils.log.debug("openTransitions="+str([ap.Name for ap in openTransitions()]))
|
|
fromAreas = availAreas
|
|
if len(openTransitions()) <= 1: # dont' get stuck by adding dead ends
|
|
GraphUtils.log.debug("avoid being stuck")
|
|
fromAreas = [area for area in fromAreas if len(GraphUtils.getAPs(lambda ap: ap.GraphArea == area and not ap.isInternal())) > 1]
|
|
elif len(forcedAreas) > 0: # no risk to get stuck, we can pick a forced area if necessary
|
|
GraphUtils.log.debug("add forced area")
|
|
fromAreas = forcedAreas
|
|
elif nLocs >= locLimit: # we just need transitions, avoid adding a huge area
|
|
GraphUtils.log.debug("not enough open transitions")
|
|
fromAreas = []
|
|
n = trLimit - len(openTransitions())
|
|
while len(fromAreas) == 0:
|
|
fromAreas = [area for area in availAreas if len(GraphUtils.getAPs(lambda ap: not ap.isInternal())) > n]
|
|
n -= 1
|
|
minLocs = min([getNLocs(lambda loc: loc.GraphArea == area) for area in fromAreas])
|
|
fromAreas = [area for area in fromAreas if getNLocs(lambda loc: loc.GraphArea == area) == minLocs]
|
|
nextArea = random.choice(fromAreas)
|
|
if nextArea in forcedAreas:
|
|
forcedAreas.remove(nextArea)
|
|
GraphUtils.log.debug("nextArea="+str(nextArea))
|
|
apCheck = lambda ap: not ap.isInternal() and not inBossCheck(ap) and ap not in usedAPs
|
|
possibleSources = GraphUtils.getAPs(lambda ap: ap.GraphArea in areas and apCheck(ap))
|
|
possibleTargets = GraphUtils.getAPs(lambda ap: ap.GraphArea == nextArea and apCheck(ap))
|
|
src = random.choice(possibleSources)
|
|
dst = random.choice(possibleTargets)
|
|
usedAPs += [src,dst]
|
|
GraphUtils.log.debug("add transition: (src: {}, dst: {})".format(src.Name, dst.Name))
|
|
transitions.append((src.Name,dst.Name))
|
|
availAreas.remove(nextArea)
|
|
areas.append(nextArea)
|
|
GraphUtils.log.debug("areas: {}".format(areas))
|
|
nLocs = getNLocs(lambda loc:loc.GraphArea in areas)
|
|
GraphUtils.log.debug("nLocs: {}".format(nLocs))
|
|
# we picked the areas, add transitions (bosses and tourian first)
|
|
sourceAPs = openTransitions()
|
|
random.shuffle(sourceAPs)
|
|
targetAPs = GraphUtils.getAPs(lambda ap: (inBossCheck(ap) or ap.Name == "Golden Four") and not ap in usedAPs)
|
|
random.shuffle(targetAPs)
|
|
assert len(sourceAPs) >= len(targetAPs), "Minimizer: less source than target APs"
|
|
while len(targetAPs) > 0:
|
|
transitions.append((sourceAPs.pop().Name, targetAPs.pop().Name))
|
|
transitions += GraphUtils.createRegularAreaTransitions(sourceAPs, lambda ap: not ap.isInternal())
|
|
GraphUtils.log.debug("FINAL MINIMIZER transitions: {}".format(transitions))
|
|
GraphUtils.loopUnusedTransitions(transitions)
|
|
GraphUtils.log.debug("FINAL MINIMIZER nLocs: "+str(nLocs+3))
|
|
GraphUtils.log.debug("FINAL MINIMIZER areas: "+str(areas))
|
|
return transitions
|
|
|
|
def createLightAreaTransitions():
|
|
# group APs by area
|
|
aps = {}
|
|
totalCount = 0
|
|
for ap in Logic.accessPoints:
|
|
if not ap.isArea():
|
|
continue
|
|
if not ap.GraphArea in aps:
|
|
aps[ap.GraphArea] = {'totalCount': 0, 'transCount': {}, 'apNames': []}
|
|
aps[ap.GraphArea]['apNames'].append(ap.Name)
|
|
# count number of vanilla transitions between each area
|
|
for (srcName, destName) in vanillaTransitions:
|
|
srcAP = getAccessPoint(srcName)
|
|
destAP = getAccessPoint(destName)
|
|
aps[srcAP.GraphArea]['transCount'][destAP.GraphArea] = aps[srcAP.GraphArea]['transCount'].get(destAP.GraphArea, 0) + 1
|
|
aps[srcAP.GraphArea]['totalCount'] += 1
|
|
aps[destAP.GraphArea]['transCount'][srcAP.GraphArea] = aps[destAP.GraphArea]['transCount'].get(srcAP.GraphArea, 0) + 1
|
|
aps[destAP.GraphArea]['totalCount'] += 1
|
|
totalCount += 1
|
|
|
|
transitions = []
|
|
while totalCount > 0:
|
|
# choose transition
|
|
srcArea = random.choice(list(aps.keys()))
|
|
srcName = random.choice(aps[srcArea]['apNames'])
|
|
src = getAccessPoint(srcName)
|
|
destArea = random.choice(list(aps[src.GraphArea]['transCount'].keys()))
|
|
destName = random.choice(aps[destArea]['apNames'])
|
|
transitions.append((srcName, destName))
|
|
|
|
# update counts
|
|
totalCount -= 1
|
|
aps[srcArea]['totalCount'] -= 1
|
|
aps[destArea]['totalCount'] -= 1
|
|
aps[srcArea]['transCount'][destArea] -= 1
|
|
if aps[srcArea]['transCount'][destArea] == 0:
|
|
del aps[srcArea]['transCount'][destArea]
|
|
aps[destArea]['transCount'][srcArea] -= 1
|
|
if aps[destArea]['transCount'][srcArea] == 0:
|
|
del aps[destArea]['transCount'][srcArea]
|
|
aps[srcArea]['apNames'].remove(srcName)
|
|
aps[destArea]['apNames'].remove(destName)
|
|
|
|
if aps[srcArea]['totalCount'] == 0:
|
|
del aps[srcArea]
|
|
if aps[destArea]['totalCount'] == 0:
|
|
del aps[destArea]
|
|
|
|
return transitions
|
|
|
|
def getVanillaExit(apName):
|
|
allVanillaTransitions = vanillaTransitions + vanillaBossesTransitions + vanillaEscapeTransitions
|
|
for (src,dst) in allVanillaTransitions:
|
|
if apName == src:
|
|
return dst
|
|
if apName == dst:
|
|
return src
|
|
return None
|
|
|
|
def isEscapeAnimals(apName):
|
|
return 'Flyway Right' in apName or 'Bomb Torizo Room Left' in apName
|
|
|
|
# gets dict like
|
|
# (RoomPtr, (vanilla entry screen X, vanilla entry screen Y)): AP
|
|
def getRooms():
|
|
rooms = {}
|
|
for ap in Logic.accessPoints:
|
|
if ap.Internal == True:
|
|
continue
|
|
# special ap for random escape animals surprise
|
|
if GraphUtils.isEscapeAnimals(ap.Name):
|
|
continue
|
|
|
|
roomPtr = ap.RoomInfo['RoomPtr']
|
|
|
|
vanillaExitName = GraphUtils.getVanillaExit(ap.Name)
|
|
# special ap for random escape animals surprise
|
|
if GraphUtils.isEscapeAnimals(vanillaExitName):
|
|
continue
|
|
|
|
connAP = getAccessPoint(vanillaExitName)
|
|
entryInfo = connAP.ExitInfo
|
|
rooms[(roomPtr, entryInfo['screen'], entryInfo['direction'])] = ap
|
|
rooms[(roomPtr, entryInfo['screen'], (ap.EntryInfo['SamusX'], ap.EntryInfo['SamusY']))] = ap
|
|
# for boss rando with incompatible ridley transition, also register this one
|
|
if ap.Name == 'RidleyRoomIn':
|
|
rooms[(roomPtr, (0x0, 0x1), 0x5)] = ap
|
|
rooms[(roomPtr, (0x0, 0x1), (0xbf, 0x198))] = ap
|
|
|
|
return rooms
|
|
|
|
def escapeAnimalsTransitions(graph, possibleTargets, firstEscape):
|
|
n = len(possibleTargets)
|
|
assert (n < 4 and firstEscape is not None) or (n <= 4 and firstEscape is None), "Invalid possibleTargets list: " + str(possibleTargets)
|
|
GraphUtils.log.debug("escapeAnimalsTransitions. possibleTargets="+str(possibleTargets)+", firstEscape="+str(firstEscape))
|
|
if n >= 1:
|
|
# complete possibleTargets. we need at least 2: one to
|
|
# hide the animals in, and one to connect the vanilla
|
|
# animals door to
|
|
if not any(t[1].Name == 'Climb Bottom Left' for t in graph.InterAreaTransitions):
|
|
# add standard Climb if not already in graph: it can be in Crateria-less minimizer + Disabled Tourian case
|
|
possibleTargets.append('Climb Bottom Left')
|
|
# make the escape possibilities loop by adding back the first escape
|
|
if firstEscape is not None:
|
|
possibleTargets.append(firstEscape)
|
|
poss = possibleTargets[:]
|
|
while len(possibleTargets) < 4:
|
|
possibleTargets.append(poss.pop(random.randint(0, len(poss)-1)))
|
|
n = len(possibleTargets)
|
|
# check if we can both hide the animals and connect the vanilla animals door to a cycling escape
|
|
if n >= 2:
|
|
# get actual animals: pick the first of the remaining targets (will contain a map room AP)
|
|
animalsAccess = possibleTargets.pop(0)
|
|
graph.EscapeAttributes['Animals'] = animalsAccess
|
|
# poss will contain the remaining map room AP(s) + optional AP(s) added above, to
|
|
# get the cycling 4 escapes from vanilla animals room
|
|
poss = possibleTargets[:]
|
|
GraphUtils.log.debug("escapeAnimalsTransitions. poss="+str(poss))
|
|
while len(possibleTargets) < 4:
|
|
if len(poss) > 1:
|
|
possibleTargets.append(poss.pop(random.randint(0, len(poss)-1)))
|
|
else:
|
|
# no more possible variety, spam the last possible escape
|
|
possibleTargets.append(poss[0])
|
|
|
|
else:
|
|
# failsafe: if not enough targets left, abort and do vanilla animals
|
|
animalsAccess = 'Flyway Right'
|
|
possibleTargets = ['Bomb Torizo Room Left'] * 4
|
|
GraphUtils.log.debug("escapeAnimalsTransitions. animalsAccess="+animalsAccess)
|
|
assert len(possibleTargets) == 4, "Invalid possibleTargets list: " + str(possibleTargets)
|
|
# actually add the 4 connections for successive escapes challenge
|
|
basePtr = 0xADAC
|
|
btDoor = getAccessPoint('Flyway Right')
|
|
for i in range(len(possibleTargets)):
|
|
ap = copy.copy(btDoor)
|
|
ap.Name += " " + str(i)
|
|
ap.ExitInfo['DoorPtr'] = basePtr + i*24
|
|
graph.addAccessPoint(ap)
|
|
target = possibleTargets[i]
|
|
graph.addTransition(ap.Name, target)
|
|
# add the connection for animals access
|
|
bt = getAccessPoint('Bomb Torizo Room Left')
|
|
btCpy = copy.copy(bt)
|
|
btCpy.Name += " Animals"
|
|
btCpy.ExitInfo['DoorPtr'] = 0xAE00
|
|
graph.addAccessPoint(btCpy)
|
|
graph.addTransition(animalsAccess, btCpy.Name)
|
|
|
|
def isHorizontal(dir):
|
|
# up: 0x3, 0x7
|
|
# down: 0x2, 0x6
|
|
# left: 0x1, 0x5
|
|
# right: 0x0, 0x4
|
|
return dir in [0x1, 0x5, 0x0, 0x4]
|
|
|
|
def removeCap(dir):
|
|
if dir < 4:
|
|
return dir
|
|
return dir - 4
|
|
|
|
def getDirection(src, dst):
|
|
exitDir = src.ExitInfo['direction']
|
|
entryDir = dst.EntryInfo['direction']
|
|
# compatible transition
|
|
if exitDir == entryDir:
|
|
return exitDir
|
|
# if incompatible but horizontal we keep entry dir (looks more natural)
|
|
if GraphUtils.isHorizontal(exitDir) and GraphUtils.isHorizontal(entryDir):
|
|
return entryDir
|
|
# otherwise keep exit direction and remove cap
|
|
return GraphUtils.removeCap(exitDir)
|
|
|
|
def getBitFlag(srcArea, dstArea, origFlag):
|
|
flags = origFlag
|
|
if srcArea == dstArea:
|
|
flags &= 0xBF
|
|
else:
|
|
flags |= 0x40
|
|
return flags
|
|
|
|
def getDoorConnections(graph, areas=True, bosses=False,
|
|
escape=True, escapeAnimals=True):
|
|
transitions = []
|
|
if areas:
|
|
transitions += vanillaTransitions
|
|
if bosses:
|
|
transitions += vanillaBossesTransitions
|
|
if escape:
|
|
transitions += vanillaEscapeTransitions
|
|
if escapeAnimals:
|
|
transitions += vanillaEscapeAnimalsTransitions
|
|
for srcName, dstName in transitions:
|
|
src = graph.accessPoints[srcName]
|
|
dst = graph.accessPoints[dstName]
|
|
dst.EntryInfo.update(src.ExitInfo)
|
|
src.EntryInfo.update(dst.ExitInfo)
|
|
connections = []
|
|
for src, dst in graph.InterAreaTransitions:
|
|
if not (escape and src.Escape and dst.Escape):
|
|
# area only
|
|
if not bosses and src.Boss:
|
|
continue
|
|
# boss only
|
|
if not areas and not src.Boss:
|
|
continue
|
|
# no random escape
|
|
if not escape and src.Escape:
|
|
continue
|
|
|
|
conn = {}
|
|
conn['ID'] = str(src) + ' -> ' + str(dst)
|
|
# remove duplicates (loop transitions)
|
|
if any(c['ID'] == conn['ID'] for c in connections):
|
|
continue
|
|
# print(conn['ID'])
|
|
# where to write
|
|
conn['DoorPtr'] = src.ExitInfo['DoorPtr']
|
|
# door properties
|
|
conn['RoomPtr'] = dst.RoomInfo['RoomPtr']
|
|
conn['doorAsmPtr'] = dst.EntryInfo['doorAsmPtr']
|
|
if 'exitAsmPtr' in src.ExitInfo:
|
|
conn['exitAsmPtr'] = src.ExitInfo['exitAsmPtr']
|
|
conn['direction'] = GraphUtils.getDirection(src, dst)
|
|
conn['bitFlag'] = GraphUtils.getBitFlag(src.RoomInfo['area'], dst.RoomInfo['area'],
|
|
dst.EntryInfo['bitFlag'])
|
|
conn['cap'] = dst.EntryInfo['cap']
|
|
conn['screen'] = dst.EntryInfo['screen']
|
|
if conn['direction'] != src.ExitInfo['direction']: # incompatible transition
|
|
conn['distanceToSpawn'] = 0
|
|
conn['SamusX'] = dst.EntryInfo['SamusX']
|
|
conn['SamusY'] = dst.EntryInfo['SamusY']
|
|
if dst.Name == 'RidleyRoomIn': # special case: spawn samus on ridley platform
|
|
conn['screen'] = (0x0, 0x1)
|
|
else:
|
|
conn['distanceToSpawn'] = dst.EntryInfo['distanceToSpawn']
|
|
if 'song' in dst.EntryInfo:
|
|
conn['song'] = dst.EntryInfo['song']
|
|
conn['songs'] = dst.RoomInfo['songs']
|
|
connections.append(conn)
|
|
return connections
|
|
|
|
def getDoorsPtrs2Aps():
|
|
ret = {}
|
|
for ap in Logic.accessPoints:
|
|
if ap.Internal == True:
|
|
continue
|
|
ret[ap.ExitInfo["DoorPtr"]] = ap.Name
|
|
return ret
|
|
|
|
def getAps2DoorsPtrs():
|
|
ret = {}
|
|
for ap in Logic.accessPoints:
|
|
if ap.Internal == True:
|
|
continue
|
|
ret[ap.Name] = ap.ExitInfo["DoorPtr"]
|
|
return ret
|
|
|
|
def getTransitions(addresses):
|
|
# build address -> name dict
|
|
doorsPtrs = GraphUtils.getDoorsPtrs2Aps()
|
|
|
|
transitions = []
|
|
# (src.ExitInfo['DoorPtr'], dst.ExitInfo['DoorPtr'])
|
|
for (srcDoorPtr, destDoorPtr) in addresses:
|
|
transitions.append((doorsPtrs[srcDoorPtr], doorsPtrs[destDoorPtr]))
|
|
|
|
return transitions
|
|
|
|
def hasMixedTransitions(areaTransitions, bossTransitions):
|
|
vanillaAPs = []
|
|
for (src, dest) in vanillaTransitions:
|
|
vanillaAPs += [src, dest]
|
|
|
|
vanillaBossesAPs = []
|
|
for (src, dest) in vanillaBossesTransitions:
|
|
vanillaBossesAPs += [src, dest]
|
|
|
|
for (src, dest) in areaTransitions:
|
|
if src in vanillaBossesAPs or dest in vanillaBossesAPs:
|
|
return True
|
|
|
|
for (src, dest) in bossTransitions:
|
|
if src in vanillaAPs or dest in vanillaAPs:
|
|
return True
|
|
|
|
return False
|