PokeBot/util/menu.lua

local menu = {}

local input = require "util.input"
local memory = require "util.memory"

local yellow = GAME_NAME == "yellow"

local sliding = false

-- Private functions

local function getRow(menuType, scrolls)
	if menuType and menuType == "settings" then
		menuType = menuType.."_row"
	else
		menuType = "row"
	end
	local row = memory.value("menu", menuType)
	if scrolls then
		row = row + memory.value("menu", "scroll_offset")
	end
	return row
end

local function setRow(desired, throttle, scrolls, menuType, loop)
	local currentRow = getRow(menuType, scrolls)
	if throttle == "accelerate" then
		if sliding then
			throttle = false
		else
			local dist = math.abs(desired - currentRow)
			if dist < 15 then
				throttle = true
			else
				throttle = false
				sliding = true
			end
		end
	else
		sliding = false
	end
	return menu.balance(currentRow, desired, true, loop, throttle)
end

local function isCurrently(desired, menuType)
	if menuType then
		menuType = menuType.."_current"
	else
		menuType = "current"
	end
	return memory.value("menu", menuType) == desired
end
menu.isCurrently = isCurrently

-- Menu

function menu.getCol()
	return memory.value("menu", "column")
end

function menu.open(desired, atIndex, menuType)
	if isCurrently(desired, menuType) then
		return true
	end
	menu.select(atIndex, false, false, menuType)
	return false
end

function menu.select(option, throttle, scrolls, menuType, dontPress, loop)
	if setRow(option, throttle, scrolls, menuType, loop) then
		local delay = 1
		if throttle then
			delay = 2
		end
		if not dontPress then
			input.press("A", delay)
		end
		return true
	end
end

function menu.cancel(desired, menuType)
	if not isCurrently(desired, menuType) then
		return true
	end
	input.press("B")
	return false
end

-- Selections

function menu.balance(current, desired, inverted, looping, throttle)
	if current == desired then
		sliding = false
		return true
	end
	if not throttle then
		throttle = 0
	else
		throttle = 1
	end
	local goUp = current > desired == inverted
	if looping and math.abs(current - desired) > math.floor(looping / 2) then
		goUp = not goUp
	end
	if goUp then
		input.press("Up", throttle)
	else
		input.press("Down", throttle)
	end
	return false
end

function menu.sidle(current, desired, looping, throttle)
	if current == desired then
		return true
	end
	if not throttle then
		throttle = 0
	else
		throttle = 1
	end
	local goLeft = current > desired
	if looping and math.abs(current - desired) > math.floor(looping / 2) then
		goLeft = not goLeft
	end
	if goLeft then
		input.press("Left", throttle)
	else
		input.press("Right", throttle)
	end
	return false
end

function menu.setCol(desired)
	return menu.sidle(menu.getCol(), desired)
end

-- Options

function menu.setOption(name, desired)
	if yellow then
		local rowFor = {
			text_speed = 0,
			battle_animation = 1,
			battle_style = 2
		}
		local currentRow = memory.raw(0x0D3D, true)
		if menu.balance(currentRow, rowFor[name], true, false, true) then
			input.press("Left")
		end
	else
		local rowFor = {
			text_speed = 3,
			battle_animation = 8,
			battle_style = 13
		}
		if memory.value("setting", name) == desired then
			return true
		end
		if setRow(rowFor[name], true, false, "settings") then
			menu.setCol(desired)
		end
	end
	return false
end

-- Pause menu

function menu.isOpen()
	return memory.value("game", "textbox") == 1 or memory.value("menu", "current") == 24
end

function menu.close()
	if memory.value("game", "textbox") == 0 and memory.value("menu", "main") < 8 then
		return true
	end
	input.press("B")
end

function menu.pause()
	if memory.value("game", "textbox") == 1 then
		if memory.value("battle", "menu") == 95 then
			input.cancel()
		else
			local main = memory.value("menu", "main")
			if main > 2 and main ~= 64 then
				return true
			end
			input.press("B")
		end
	else
		input.press("Start", 2)
	end
end

return menu