ae7ce076af
Spoiler lists bosses Enemizer settings fix Swamp flooded ladder's fix (Crossed mostly) Spoiler encoding issue fixed
276 lines
11 KiB
Python
276 lines
11 KiB
Python
from tkinter import ttk, messagebox, StringVar, Button, Entry, Frame, Label, E, W, LEFT, RIGHT, X
|
|
from argparse import Namespace
|
|
import logging
|
|
import os
|
|
import random
|
|
import re
|
|
from CLI import parse_cli
|
|
from Fill import FillError
|
|
from Main import main, EnemizerError
|
|
from Utils import local_path, output_path, open_file, update_deprecated_args
|
|
import source.classes.constants as CONST
|
|
from source.gui.randomize.multiworld import multiworld_page
|
|
import source.gui.widgets as widgets
|
|
from source.classes.Empty import Empty
|
|
|
|
|
|
def bottom_frame(self, parent, args=None):
|
|
# Bottom Frame
|
|
self = ttk.Frame(parent)
|
|
|
|
# Bottom Frame options
|
|
self.widgets = {}
|
|
|
|
mw,_ = multiworld_page(self, parent.settings)
|
|
mw.pack(fill=X, expand=True)
|
|
self.widgets = mw.widgets
|
|
|
|
# Seed input
|
|
# widget ID
|
|
widget = "seed"
|
|
|
|
# Empty object
|
|
self.widgets[widget] = Empty()
|
|
# pieces
|
|
self.widgets[widget].pieces = {}
|
|
|
|
# frame
|
|
self.widgets[widget].pieces["frame"] = Frame(self)
|
|
# frame: label
|
|
self.widgets[widget].pieces["frame"].label = Label(self.widgets[widget].pieces["frame"], text="Seed #")
|
|
self.widgets[widget].pieces["frame"].label.pack(side=LEFT)
|
|
# storagevar
|
|
savedSeed = parent.settings["seed"]
|
|
self.widgets[widget].storageVar = StringVar(value=savedSeed)
|
|
# textbox
|
|
self.widgets[widget].type = "textbox"
|
|
self.widgets[widget].pieces["textbox"] = Entry(self.widgets[widget].pieces["frame"], width=15, textvariable=self.widgets[widget].storageVar)
|
|
self.widgets[widget].pieces["textbox"].pack(side=LEFT)
|
|
|
|
def saveSeed(caller,_,mode):
|
|
savedSeed = self.widgets["seed"].storageVar.get()
|
|
parent.settings["seed"] = int(savedSeed) if savedSeed.isdigit() else None
|
|
self.widgets[widget].storageVar.trace_add("write",saveSeed)
|
|
# frame: pack
|
|
self.widgets[widget].pieces["frame"].pack(side=LEFT)
|
|
|
|
## Number of Generation attempts
|
|
key = "generationcount"
|
|
self.widgets[key] = widgets.make_widget(
|
|
self,
|
|
"spinbox",
|
|
self,
|
|
"Count",
|
|
None,
|
|
None,
|
|
{"label": {"side": LEFT}, "spinbox": {"side": RIGHT}}
|
|
)
|
|
self.widgets[key].pack(side=LEFT)
|
|
|
|
def generateRom():
|
|
guiargs = create_guiargs(parent)
|
|
# get default values for missing parameters
|
|
for k,v in vars(parse_cli(['--multi', str(guiargs.multi)])).items():
|
|
if k not in vars(guiargs):
|
|
setattr(guiargs, k, v)
|
|
elif type(v) is dict: # use same settings for every player
|
|
setattr(guiargs, k, {player: getattr(guiargs, k) for player in range(1, guiargs.multi + 1)})
|
|
argsDump = vars(guiargs)
|
|
hasEnemizer = "enemizercli" in argsDump and os.path.isfile(argsDump["enemizercli"])
|
|
needEnemizer = False
|
|
if hasEnemizer:
|
|
falsey = ["none", "default", False, 0]
|
|
for enemizerOption in [ "shufflepots", "shuffleenemies", "enemy_damage", "shufflebosses", "enemy_health" ]:
|
|
if enemizerOption in argsDump:
|
|
if isinstance(argsDump[enemizerOption], dict):
|
|
for playerID,playerSetting in argsDump[enemizerOption].items():
|
|
if not playerSetting in falsey:
|
|
needEnemizer = True
|
|
elif not argsDump[enemizerOption] in falsey:
|
|
needEnemizer = True
|
|
seeds = []
|
|
if not needEnemizer or (needEnemizer and hasEnemizer):
|
|
try:
|
|
if guiargs.count is not None and guiargs.seed:
|
|
seed = guiargs.seed
|
|
for _ in range(guiargs.count):
|
|
seeds.append(seed)
|
|
main(seed=seed, args=guiargs, fish=parent.fish)
|
|
seed = random.randint(0, 999999999)
|
|
else:
|
|
if guiargs.seed:
|
|
seeds.append(guiargs.seed)
|
|
else:
|
|
random.seed(None)
|
|
guiargs.seed = random.randint(0, 999999999)
|
|
seeds.append(guiargs.seed)
|
|
main(seed=guiargs.seed, args=guiargs, fish=parent.fish)
|
|
except (FillError, EnemizerError, Exception, RuntimeError) as e:
|
|
logging.exception(e)
|
|
messagebox.showerror(title="Error while creating seed", message=str(e))
|
|
else:
|
|
YES = parent.fish.translate("cli","cli","yes")
|
|
NO = parent.fish.translate("cli","cli","no")
|
|
successMsg = ""
|
|
made = {}
|
|
for k in [ "rom", "playthrough", "spoiler" ]:
|
|
made[k] = parent.fish.translate("cli","cli","made." + k)
|
|
made["enemizer"] = parent.fish.translate("cli","cli","used.enemizer")
|
|
for k in made:
|
|
v = made[k]
|
|
pattern = "([\w]+)(:)([\s]+)(.*)"
|
|
m = re.search(pattern,made[k])
|
|
made[k] = m.group(1) + m.group(2) + ' ' + m.group(4)
|
|
successMsg += (made["rom"] % (YES if (guiargs.create_rom) else NO)) + "\n"
|
|
successMsg += (made["playthrough"] % (YES if (guiargs.calc_playthrough) else NO)) + "\n"
|
|
successMsg += (made["spoiler"] % (YES if (not guiargs.jsonout and guiargs.create_spoiler) else NO)) + "\n"
|
|
successMsg += (made["enemizer"] % (YES if needEnemizer else NO)) + "\n"
|
|
# FIXME: English
|
|
successMsg += ("Seed%s: %s" % ('s' if len(seeds) > 1 else "", ','.join(str(x) for x in seeds)))
|
|
|
|
messagebox.showinfo(title="Success", message=successMsg)
|
|
|
|
## Generate Button
|
|
# widget ID
|
|
widget = "go"
|
|
|
|
# Empty object
|
|
self.widgets[widget] = Empty()
|
|
# pieces
|
|
self.widgets[widget].pieces = {}
|
|
|
|
# button
|
|
self.widgets[widget].type = "button"
|
|
self.widgets[widget].pieces["button"] = Button(self, text='Generate Patched Rom', command=generateRom)
|
|
# button: pack
|
|
self.widgets[widget].pieces["button"].pack(side=LEFT)
|
|
|
|
def open_output():
|
|
if args and args.outputpath:
|
|
open_file(output_path(args.outputpath))
|
|
else:
|
|
open_file(output_path(parent.settings["outputpath"]))
|
|
|
|
## Output Button
|
|
# widget ID
|
|
widget = "outputdir"
|
|
|
|
# Empty object
|
|
self.widgets[widget] = Empty()
|
|
# pieces
|
|
self.widgets[widget].pieces = {}
|
|
|
|
# storagevar
|
|
self.widgets[widget].storageVar = StringVar(value=parent.settings["outputpath"])
|
|
|
|
# button
|
|
self.widgets[widget].type = "button"
|
|
self.widgets[widget].pieces["button"] = Button(self, text='Open Output Directory', command=open_output)
|
|
# button: pack
|
|
self.widgets[widget].pieces["button"].pack(side=RIGHT)
|
|
|
|
## Documentation Button
|
|
# widget ID
|
|
widget = "docs"
|
|
|
|
# Empty object
|
|
self.widgets[widget] = Empty()
|
|
# pieces
|
|
self.widgets[widget].pieces = {}
|
|
# button
|
|
self.widgets[widget].type = "button"
|
|
self.widgets[widget].selectbox = Empty()
|
|
self.widgets[widget].selectbox.storageVar = Empty()
|
|
if os.path.exists(local_path('README.html')):
|
|
def open_readme():
|
|
open_file(local_path('README.html'))
|
|
self.widgets[widget].pieces["button"] = Button(self, text='Open Documentation', command=open_readme)
|
|
# button: pack
|
|
self.widgets[widget].pieces["button"].pack(side=RIGHT)
|
|
|
|
return self
|
|
|
|
|
|
def create_guiargs(parent):
|
|
guiargs = Namespace()
|
|
|
|
# set up settings to gather
|
|
# Page::Subpage::GUI-id::param-id
|
|
options = CONST.SETTINGSTOPROCESS
|
|
|
|
# Cycle through each page
|
|
for mainpage in options:
|
|
# Cycle through each subpage (in case of Item Randomizer)
|
|
for subpage in options[mainpage]:
|
|
# Cycle through each widget
|
|
for widget in options[mainpage][subpage]:
|
|
# Get the value and set it
|
|
arg = options[mainpage][subpage][widget]
|
|
setattr(guiargs, arg, parent.pages[mainpage].pages[subpage].widgets[widget].storageVar.get())
|
|
|
|
# Get EnemizerCLI setting
|
|
guiargs.enemizercli = parent.pages["randomizer"].pages["enemizer"].widgets["enemizercli"].storageVar.get()
|
|
|
|
# Get Multiworld Worlds count
|
|
guiargs.multi = int(parent.pages["bottom"].pages["content"].widgets["worlds"].storageVar.get())
|
|
|
|
# Get baserom path
|
|
guiargs.rom = parent.pages["randomizer"].pages["generation"].widgets["rom"].storageVar.get()
|
|
|
|
# Get if we're using the Custom Item Pool
|
|
guiargs.custom = bool(parent.pages["randomizer"].pages["generation"].widgets["usecustompool"].storageVar.get())
|
|
|
|
# Get Seed ID
|
|
guiargs.seed = None
|
|
if parent.pages["bottom"].pages["content"].widgets["seed"].storageVar.get():
|
|
guiargs.seed = parent.pages["bottom"].pages["content"].widgets["seed"].storageVar.get()
|
|
|
|
# Get number of generations to run
|
|
guiargs.count = 1
|
|
if parent.pages["bottom"].pages["content"].widgets["generationcount"].storageVar.get():
|
|
guiargs.count = int(parent.pages["bottom"].pages["content"].widgets["generationcount"].storageVar.get())
|
|
|
|
# Get Adjust settings
|
|
adjustargs = {
|
|
"nobgm": "disablemusic",
|
|
"quickswap": "quickswap",
|
|
"heartcolor": "heartcolor",
|
|
"heartbeep": "heartbeep",
|
|
"menuspeed": "fastmenu",
|
|
"owpalettes": "ow_palettes",
|
|
"uwpalettes": "uw_palettes"
|
|
}
|
|
for adjustarg in adjustargs:
|
|
internal = adjustargs[adjustarg]
|
|
setattr(guiargs,"adjust." + internal, parent.pages["adjust"].content.widgets[adjustarg].storageVar.get())
|
|
|
|
# Get Custom Items and Starting Inventory Items
|
|
customitems = CONST.CUSTOMITEMS
|
|
guiargs.startinventory = []
|
|
guiargs.customitemarray = {}
|
|
guiargs.startinventoryarray = {}
|
|
for customitem in customitems:
|
|
if customitem not in CONST.CANTSTARTWITH:
|
|
# Starting Inventory is a CSV
|
|
amount = int(parent.pages["startinventory"].content.startingWidgets[customitem].storageVar.get())
|
|
guiargs.startinventoryarray[customitem] = amount
|
|
for _ in range(0, amount):
|
|
label = CONST.CUSTOMITEMLABELS[customitems.index(customitem)]
|
|
guiargs.startinventory.append(label)
|
|
# Custom Item Pool is a dict of ints
|
|
guiargs.customitemarray[customitem] = int(parent.pages["custom"].content.customWidgets[customitem].storageVar.get())
|
|
|
|
# Starting Inventory is a CSV
|
|
guiargs.startinventory = ','.join(guiargs.startinventory)
|
|
|
|
# Get Sprite Selection (set or random)
|
|
guiargs.sprite = parent.pages["randomizer"].pages["gameoptions"].widgets["sprite"]["spriteObject"]
|
|
guiargs.randomSprite = parent.randomSprite.get()
|
|
|
|
# Get output path
|
|
guiargs.outputpath = parent.outputPath.get()
|
|
|
|
guiargs = update_deprecated_args(guiargs)
|
|
|
|
return guiargs
|