225 lines
13 KiB
Python
225 lines
13 KiB
Python
#!/usr/bin/python3
|
|
import sys, argparse
|
|
|
|
from solver.interactiveSolver import InteractiveSolver
|
|
from solver.standardSolver import StandardSolver
|
|
from solver.conf import Conf
|
|
import utils.log
|
|
|
|
def interactiveSolver(args):
|
|
# to init, requires interactive/romFileName/presetFileName/output parameters in standard/plando mode
|
|
# to init, requires interactive/presetFileName/output parameters in seedless mode
|
|
# to iterate, requires interactive/state/[loc]/[item]/action/output parameters in item scope
|
|
# to iterate, requires interactive/state/[startPoint]/[endPoint]/action/output parameters in area scope
|
|
if args.action == 'init':
|
|
# init
|
|
if args.mode != 'seedless' and args.romFileName == None:
|
|
print("Missing romFileName parameter for {} mode".format(args.mode))
|
|
sys.exit(1)
|
|
|
|
if args.presetFileName == None or args.output == None:
|
|
print("Missing preset or output parameter")
|
|
sys.exit(1)
|
|
|
|
solver = InteractiveSolver(args.output, args.logic)
|
|
solver.initialize(args.mode, args.romFileName, args.presetFileName, magic=args.raceMagic, fill=args.fill, startLocation=args.startLocation)
|
|
else:
|
|
# iterate
|
|
params = {}
|
|
if args.scope == 'common':
|
|
if args.action == "save":
|
|
params["lock"] = args.lock
|
|
params["escapeTimer"] = args.escapeTimer
|
|
elif args.action == "randomize":
|
|
params["minorQty"] = args.minorQty
|
|
params["energyQty"] = args.energyQty
|
|
params["forbiddenItems"] = args.forbiddenItems.split(',') if args.forbiddenItems is not None else []
|
|
elif args.scope == 'item':
|
|
if args.state == None or args.action == None or args.output == None:
|
|
print("Missing state/action/output parameter")
|
|
sys.exit(1)
|
|
if args.action in ["add", "replace"]:
|
|
if args.mode not in ['seedless', 'race', 'debug'] and args.loc == None:
|
|
print("Missing loc parameter when using action add for item")
|
|
sys.exit(1)
|
|
if args.mode == 'plando':
|
|
if args.item == None:
|
|
print("Missing item parameter when using action add in plando/suitless mode")
|
|
sys.exit(1)
|
|
params = {'loc': args.loc, 'item': args.item, 'hide': args.hide}
|
|
elif args.action == "remove":
|
|
if args.loc != None:
|
|
params = {'loc': args.loc}
|
|
elif args.item != None:
|
|
params = {'item': args.item}
|
|
else:
|
|
params = {'count': args.count}
|
|
elif args.action == "toggle":
|
|
params = {'item': args.item}
|
|
elif args.scope == 'area':
|
|
if args.state == None or args.action == None or args.output == None:
|
|
print("Missing state/action/output parameter")
|
|
sys.exit(1)
|
|
if args.action == "add":
|
|
if args.startPoint == None or args.endPoint == None:
|
|
print("Missing start or end point parameter when using action add for item")
|
|
sys.exit(1)
|
|
params = {'startPoint': args.startPoint, 'endPoint': args.endPoint}
|
|
if args.action == "remove" and args.startPoint != None:
|
|
params = {'startPoint': args.startPoint}
|
|
elif args.scope == 'door':
|
|
if args.state == None or args.action == None or args.output == None:
|
|
print("Missing state/action/output parameter")
|
|
sys.exit(1)
|
|
if args.action == "replace":
|
|
if args.doorName is None or args.newColor is None:
|
|
print("Missing doorName or newColor parameter when using action replace for door")
|
|
sys.exit(1)
|
|
params = {'doorName': args.doorName, 'newColor': args.newColor}
|
|
elif args.action == "toggle":
|
|
if args.doorName is None:
|
|
print("Missing doorName parameter when using action toggle for door")
|
|
sys.exit(1)
|
|
params = {'doorName': args.doorName}
|
|
elif args.scope == 'dump':
|
|
if args.action == "import":
|
|
if args.dump is None:
|
|
print("Missing dump parameter when import a dump")
|
|
params = {'dump': args.dump}
|
|
params["debug"] = args.mode == 'debug'
|
|
|
|
solver = InteractiveSolver(args.output, args.logic)
|
|
solver.iterate(args.state, args.scope, args.action, params)
|
|
|
|
def standardSolver(args):
|
|
if args.romFileName is None:
|
|
print("Parameter --romFileName mandatory when not in interactive mode")
|
|
sys.exit(1)
|
|
|
|
if args.difficultyTarget is None:
|
|
difficultyTarget = Conf.difficultyTarget
|
|
else:
|
|
difficultyTarget = args.difficultyTarget
|
|
|
|
if args.pickupStrategy is None:
|
|
pickupStrategy = Conf.itemsPickup
|
|
else:
|
|
pickupStrategy = args.pickupStrategy
|
|
|
|
# itemsForbidden is like that: [['Varia'], ['Reserve'], ['Gravity']], fix it
|
|
args.itemsForbidden = [item[0] for item in args.itemsForbidden]
|
|
|
|
solver = StandardSolver(args.romFileName, args.presetFileName, difficultyTarget,
|
|
pickupStrategy, args.itemsForbidden, type=args.type,
|
|
firstItemsLog=args.firstItemsLog, extStatsFilename=args.extStatsFilename,
|
|
extStatsStep=args.extStatsStep,
|
|
displayGeneratedPath=args.displayGeneratedPath,
|
|
outputFileName=args.output, magic=args.raceMagic,
|
|
checkDuplicateMajor=args.checkDuplicateMajor, vcr=args.vcr,
|
|
runtimeLimit_s=args.runtimeLimit_s)
|
|
|
|
solver.solveRom()
|
|
|
|
if __name__ == "__main__":
|
|
parser = argparse.ArgumentParser(description="Random Metroid Solver")
|
|
parser.add_argument('--romFileName', '-r', help="the input rom", nargs='?',
|
|
default=None, dest="romFileName")
|
|
parser.add_argument('--preset', '-p', help="the preset file", nargs='?',
|
|
default=None, dest='presetFileName')
|
|
parser.add_argument('--difficultyTarget', '-t',
|
|
help="the difficulty target that the solver will aim for",
|
|
dest='difficultyTarget', nargs='?', default=None, type=int)
|
|
parser.add_argument('--pickupStrategy', '-s', help="Pickup strategy for the Solver",
|
|
dest='pickupStrategy', nargs='?', default=None,
|
|
choices=['all', 'any'])
|
|
parser.add_argument('--itemsForbidden', '-f', help="Item not picked up during solving",
|
|
dest='itemsForbidden', nargs='+', default=[], action='append')
|
|
|
|
parser.add_argument('--type', '-y', help="web or console", dest='type', nargs='?',
|
|
default='console', choices=['web', 'console'])
|
|
parser.add_argument('--checkDuplicateMajor', dest="checkDuplicateMajor", action='store_true',
|
|
help="print a warning if the same major is collected more than once")
|
|
parser.add_argument('--debug', '-d', help="activate debug logging", dest='debug', action='store_true')
|
|
parser.add_argument('--firstItemsLog', '-1',
|
|
help="path to file where for each item type the first time it was found and where will be written (spoilers!)",
|
|
nargs='?', default=None, type=str, dest='firstItemsLog')
|
|
parser.add_argument('--ext_stats', help="Generate extended stats",
|
|
nargs='?', default=None, dest='extStatsFilename')
|
|
parser.add_argument('--ext_stats_step', help="what extended stats to generate",
|
|
nargs='?', default=None, dest='extStatsStep', type=int)
|
|
parser.add_argument('--displayGeneratedPath', '-g', help="display the generated path (spoilers!)",
|
|
dest='displayGeneratedPath', action='store_true')
|
|
parser.add_argument('--race', help="Race mode magic number", dest='raceMagic', type=int)
|
|
parser.add_argument('--vcr', help="Generate VCR output file", dest='vcr', action='store_true')
|
|
# standard/interactive, web site
|
|
parser.add_argument('--output', '-o', help="When called from the website, contains the result of the solver",
|
|
dest='output', nargs='?', default=None)
|
|
# interactive, web site
|
|
parser.add_argument('--interactive', '-i', help="Activate interactive mode for the solver",
|
|
dest='interactive', action='store_true')
|
|
parser.add_argument('--state', help="JSON file of the Solver state (used in interactive mode)",
|
|
dest="state", nargs='?', default=None)
|
|
parser.add_argument('--loc', help="Name of the location to action on (used in interactive mode)",
|
|
dest="loc", nargs='?', default=None)
|
|
parser.add_argument('--action', help="Pickup item at location, remove last pickedup location, clear all (used in interactive mode)",
|
|
dest="action", nargs="?", default=None, choices=['init', 'add', 'remove', 'clear', 'get', 'save', 'replace', 'randomize', 'toggle', 'import'])
|
|
parser.add_argument('--item', help="Name of the item to place in plando mode (used in interactive mode)",
|
|
dest="item", nargs='?', default=None)
|
|
parser.add_argument('--hide', help="Hide the item to place in plando mode (used in interactive mode)",
|
|
dest="hide", action='store_true')
|
|
parser.add_argument('--startPoint', help="The start AP to connect (used in interactive mode)",
|
|
dest="startPoint", nargs='?', default=None)
|
|
parser.add_argument('--endPoint', help="The destination AP to connect (used in interactive mode)",
|
|
dest="endPoint", nargs='?', default=None)
|
|
|
|
parser.add_argument('--mode', help="Solver mode: standard/seedless/plando (used in interactive mode)",
|
|
dest="mode", nargs="?", default=None, choices=['standard', 'seedless', 'plando', 'race', 'debug'])
|
|
parser.add_argument('--scope', help="Scope for the action: common/area/item (used in interactive mode)",
|
|
dest="scope", nargs="?", default=None, choices=['common', 'area', 'item', 'door', 'dump'])
|
|
parser.add_argument('--count', help="Number of item rollback (used in interactive mode)",
|
|
dest="count", type=int)
|
|
parser.add_argument('--lock', help="lock the plando seed (used in interactive mode)",
|
|
dest="lock", action='store_true')
|
|
parser.add_argument('--escapeTimer', help="escape timer like 03:00", dest="escapeTimer", default=None)
|
|
parser.add_argument('--fill', help="in plando load all the source seed locations/transitions as a base (used in interactive mode)",
|
|
dest="fill", action='store_true')
|
|
parser.add_argument('--startLocation', help="in plando/seedless: the start location", dest="startLocation", default="Landing Site")
|
|
parser.add_argument('--minorQty', help="rando plando (used in interactive mode)",
|
|
dest="minorQty", nargs="?", default=None, choices=[str(i) for i in range(0,101)])
|
|
parser.add_argument('--energyQty', help="rando plando (used in interactive mode)",
|
|
dest="energyQty", nargs="?", default=None, choices=["sparse", "medium", "vanilla"])
|
|
parser.add_argument('--forbiddenItems', help="rando plando (used in interactive mode)",
|
|
dest="forbiddenItems", nargs="?", default=None)
|
|
parser.add_argument('--doorName', help="door to replace (used in interactive mode)",
|
|
dest="doorName", nargs="?", default=None)
|
|
parser.add_argument('--newColor', help="new color for door (used in interactive mode)",
|
|
dest="newColor", nargs="?", default=None)
|
|
parser.add_argument('--logic', help='logic to use (used in interactive mode)', dest='logic', nargs='?', default="vanilla", choices=["vanilla", "rotation"])
|
|
parser.add_argument('--runtime',
|
|
help="Maximum runtime limit in seconds. If 0 or negative, no runtime limit.",
|
|
dest='runtimeLimit_s', nargs='?', default=0, type=int)
|
|
parser.add_argument('--dump', help="dump file with autotracker state (used in interactive mode)",
|
|
dest="dump", nargs="?", default=None)
|
|
|
|
args = parser.parse_args()
|
|
|
|
if args.presetFileName is None:
|
|
args.presetFileName = 'worlds/sm/variaRandomizer/standard_presets/regular.json'
|
|
|
|
if args.raceMagic != None:
|
|
if args.raceMagic <= 0 or args.raceMagic >= 0x10000:
|
|
print("Invalid magic")
|
|
sys.exit(-1)
|
|
|
|
if args.count != None:
|
|
if args.count < 1 or args.count > 0x80:
|
|
print("Invalid count")
|
|
sys.exit(-1)
|
|
|
|
utils.log.init(args.debug)
|
|
|
|
if args.interactive == True:
|
|
interactiveSolver(args)
|
|
else:
|
|
standardSolver(args)
|