230 lines
8.8 KiB
Python
230 lines
8.8 KiB
Python
import json
|
|
import logging
|
|
import os
|
|
from argparse import Namespace
|
|
from tkinter import (
|
|
BOTTOM,
|
|
LEFT,
|
|
RIGHT,
|
|
Button,
|
|
E,
|
|
Entry,
|
|
Frame,
|
|
Label,
|
|
StringVar,
|
|
W,
|
|
X,
|
|
filedialog,
|
|
messagebox,
|
|
ttk,
|
|
)
|
|
|
|
import source.gui.widgets as widgets
|
|
from AdjusterMain import adjust, patch
|
|
from source.classes.ItemGfxSelector import ItemGfxSelector
|
|
from source.classes.SpriteSelector import SpriteSelector
|
|
from Utils import output_path
|
|
|
|
|
|
def adjust_page(top, parent, settings):
|
|
# Adjust page
|
|
self = ttk.Frame(parent)
|
|
|
|
# Adjust options
|
|
self.widgets = {}
|
|
|
|
# Adjust option sections
|
|
self.frames = {}
|
|
self.frames["checkboxes"] = Frame(self)
|
|
self.frames["checkboxes"].pack(anchor=W)
|
|
|
|
# Adjust option frames
|
|
self.frames["selectOptionsFrame"] = Frame(self)
|
|
self.frames["leftAdjustFrame"] = Frame(self.frames["selectOptionsFrame"])
|
|
self.frames["rightAdjustFrame"] = Frame(self.frames["selectOptionsFrame"])
|
|
self.frames["bottomAdjustFrame"] = Frame(self)
|
|
self.frames["selectOptionsFrame"].pack(fill=X)
|
|
self.frames["leftAdjustFrame"].pack(side=LEFT)
|
|
self.frames["rightAdjustFrame"].pack(side=RIGHT)
|
|
self.frames["bottomAdjustFrame"].pack(fill=X)
|
|
|
|
# Load Adjust option widgets as defined by JSON file
|
|
# Defns include frame name, widget type, widget options, widget placement attributes
|
|
with open(os.path.join("resources","app","gui","adjust","overview","widgets.json")) as widgetDefns:
|
|
myDict = json.load(widgetDefns)
|
|
for framename,theseWidgets in myDict.items():
|
|
dictWidgets = widgets.make_widgets_from_dict(self, theseWidgets, self.frames[framename])
|
|
for key in dictWidgets:
|
|
self.widgets[key] = dictWidgets[key]
|
|
packAttrs = {"anchor":E}
|
|
if self.widgets[key].type == "checkbox":
|
|
packAttrs["anchor"] = W
|
|
self.widgets[key].pack(packAttrs)
|
|
|
|
# Sprite Selection
|
|
# This one's more-complicated, build it and stuff it
|
|
self.spriteNameVar2 = StringVar()
|
|
spriteDialogFrame2 = Frame(self.frames["leftAdjustFrame"])
|
|
baseSpriteLabel2 = Label(spriteDialogFrame2, text='Sprite:')
|
|
spriteEntry2 = Label(spriteDialogFrame2, textvariable=self.spriteNameVar2)
|
|
self.sprite = None
|
|
|
|
def set_sprite(sprite_param, random_sprite=False):
|
|
if sprite_param is None or not sprite_param.valid:
|
|
self.sprite = None
|
|
self.spriteNameVar2.set('(unchanged)')
|
|
else:
|
|
self.sprite = sprite_param
|
|
self.spriteNameVar2.set(self.sprite.name)
|
|
top.randomSprite.set(random_sprite)
|
|
|
|
def SpriteSelectAdjuster():
|
|
SpriteSelector(parent, set_sprite, adjuster=True)
|
|
|
|
spriteSelectButton2 = Button(spriteDialogFrame2, text='...', command=SpriteSelectAdjuster)
|
|
|
|
baseSpriteLabel2.pack(side=LEFT)
|
|
spriteEntry2.pack(side=LEFT)
|
|
spriteSelectButton2.pack(side=LEFT)
|
|
spriteDialogFrame2.pack(anchor=E)
|
|
|
|
# Triforce Piece GFX Selection
|
|
self.triforceGfxNameVar = StringVar()
|
|
self.triforceGfxNameVar.set('(unchanged)')
|
|
triforceGfxDialogFrame = Frame(self.frames["leftAdjustFrame"])
|
|
triforceGfxLabel = Label(triforceGfxDialogFrame, text='Triforce Piece:')
|
|
triforceGfxEntry = Label(triforceGfxDialogFrame, textvariable=self.triforceGfxNameVar)
|
|
self.triforce_gfx = None
|
|
|
|
def set_triforce_gfx(item_name):
|
|
self.triforce_gfx = item_name
|
|
self.triforceGfxNameVar.set(item_name if item_name else '(unchanged)')
|
|
|
|
def TriforceGfxSelectAdjuster():
|
|
from Tables import item_gfx_table
|
|
valid_items = list(item_gfx_table.keys())
|
|
ItemGfxSelector(parent, set_triforce_gfx, valid_items=valid_items, adjuster=True)
|
|
|
|
triforceGfxSelectButton = Button(triforceGfxDialogFrame, text='...', command=TriforceGfxSelectAdjuster)
|
|
|
|
triforceGfxLabel.pack(side=LEFT)
|
|
triforceGfxEntry.pack(side=LEFT)
|
|
triforceGfxSelectButton.pack(side=LEFT)
|
|
triforceGfxDialogFrame.pack(anchor=E)
|
|
|
|
# Path to game file to Adjust
|
|
# This one's more-complicated, build it and stuff it
|
|
adjustRomFrame = Frame(self.frames["bottomAdjustFrame"])
|
|
adjustRomLabel = Label(adjustRomFrame, text='Rom to adjust: ')
|
|
self.romVar2 = StringVar(value=settings["rom"])
|
|
romEntry2 = Entry(adjustRomFrame, textvariable=self.romVar2)
|
|
|
|
def RomSelect2():
|
|
initdir = os.path.join(os.getcwd(), settings['outputpath']) if 'outputpath' in settings else os.getcwd()
|
|
rom = filedialog.askopenfilename(filetypes=[("Rom Files", (".sfc", ".smc")), ("All Files", "*")],
|
|
initialdir=initdir)
|
|
if rom:
|
|
settings["rom"] = rom
|
|
self.romVar2.set(rom)
|
|
romSelectButton2 = Button(adjustRomFrame, text='Select Rom', command=RomSelect2)
|
|
|
|
adjustRomLabel.pack(side=LEFT)
|
|
romEntry2.pack(side=LEFT, fill=X, expand=True)
|
|
romSelectButton2.pack(side=LEFT)
|
|
adjustRomFrame.pack(fill=X)
|
|
|
|
# These are the options to Adjust
|
|
def adjustRom():
|
|
if output_path.cached_path is None:
|
|
output_path.cached_path = top.settings["outputpath"]
|
|
options = {
|
|
"heartbeep": "heartbeep",
|
|
"heartcolor": "heartcolor",
|
|
"menuspeed": "fastmenu",
|
|
"owpalettes": "ow_palettes",
|
|
"uwpalettes": "uw_palettes",
|
|
"quickswap": "quickswap",
|
|
"nobgm": "disablemusic",
|
|
"reduce_flashing": "reduce_flashing",
|
|
'msu_resume': 'msu_resume',
|
|
"shuffle_sfx": "shuffle_sfx",
|
|
"shuffle_sfxinstruments": "shuffle_sfxinstruments",
|
|
"shuffle_songinstruments": "shuffle_songinstruments",
|
|
}
|
|
guiargs = Namespace()
|
|
for option in options:
|
|
arg = options[option]
|
|
setattr(guiargs, arg, self.widgets[option].storageVar.get())
|
|
guiargs.rom = self.romVar2.get()
|
|
guiargs.baserom = top.pages["randomizer"].pages["generation"].widgets["rom"].storageVar.get()
|
|
guiargs.sprite = self.sprite
|
|
guiargs.triforce_gfx = self.triforce_gfx
|
|
guiargs.outputpath = os.path.dirname(guiargs.rom)
|
|
try:
|
|
adjust(args=guiargs)
|
|
except Exception as e:
|
|
logging.exception(e)
|
|
messagebox.showerror(title="Error while creating seed", message=str(e))
|
|
else:
|
|
messagebox.showinfo(title="Success", message="Rom patched successfully")
|
|
|
|
adjustButton = Button(self.frames["bottomAdjustFrame"], text='Adjust Rom', command=adjustRom)
|
|
adjustButton.pack(padx=(5, 0))
|
|
|
|
patchFileFrame = Frame(self.frames["bottomAdjustFrame"])
|
|
patchFileLabel = Label(patchFileFrame, text='BPS Patch: ')
|
|
self.patchVar = StringVar(value=settings["patch"])
|
|
patchEntry = Entry(patchFileFrame, textvariable=self.patchVar)
|
|
|
|
def PatchSelect():
|
|
initdir = os.path.join(os.getcwd(), settings['outputpath']) if 'outputpath' in settings else os.getcwd()
|
|
file = filedialog.askopenfilename(filetypes=[("BPS Files", ".bps"), ("All Files", "*")], initialdir=initdir)
|
|
if file:
|
|
settings["patch"] = file
|
|
self.patchVar.set(file)
|
|
patchSelectButton = Button(patchFileFrame, text='Select Patch', command=PatchSelect)
|
|
|
|
patchFileLabel.pack(side=LEFT)
|
|
patchEntry.pack(side=LEFT, fill=X, expand=True)
|
|
patchSelectButton.pack(side=LEFT)
|
|
patchFileFrame.pack(fill=X)
|
|
|
|
def patchRom():
|
|
if output_path.cached_path is None:
|
|
output_path.cached_path = top.settings["outputpath"]
|
|
options = {
|
|
"heartbeep": "heartbeep",
|
|
"heartcolor": "heartcolor",
|
|
"menuspeed": "fastmenu",
|
|
"owpalettes": "ow_palettes",
|
|
"uwpalettes": "uw_palettes",
|
|
"quickswap": "quickswap",
|
|
"nobgm": "disablemusic",
|
|
"reduce_flashing": "reduce_flashing",
|
|
"shuffle_sfx": "shuffle_sfx",
|
|
"shuffle_sfxinstruments": "shuffle_sfxinstruments",
|
|
"shuffle_songinstruments": "shuffle_songinstruments",
|
|
"msu_resume": "msu_resume"
|
|
}
|
|
guiargs = Namespace()
|
|
for option in options:
|
|
arg = options[option]
|
|
setattr(guiargs, arg, self.widgets[option].storageVar.get())
|
|
guiargs.patch = self.patchVar.get()
|
|
guiargs.baserom = top.pages["randomizer"].pages["generation"].widgets["rom"].storageVar.get()
|
|
guiargs.sprite = self.sprite
|
|
guiargs.triforce_gfx = self.triforce_gfx
|
|
guiargs.outputpath = os.path.dirname(guiargs.patch)
|
|
try:
|
|
patch(args=guiargs)
|
|
except Exception as e:
|
|
logging.exception(e)
|
|
messagebox.showerror(title="Error while creating seed", message=str(e))
|
|
else:
|
|
messagebox.showinfo(title="Success", message="Rom patched successfully")
|
|
|
|
patchButton = Button(self.frames["bottomAdjustFrame"], text='Patch Rom', command=patchRom)
|
|
patchButton.pack(side=BOTTOM, padx=(5, 0))
|
|
|
|
return self,settings
|