PokeBot/main.lua

-- OPTIONS

RESET_FOR_TIME = true -- Set to false if you just want to see the bot finish a run

local CUSTOM_SEED = nil -- Set to a known seed to replay it, or leave nil for random runs
local PAINT_ON    = true -- Display contextual information while the bot runs

-- START CODE (hard hats on)

VERSION = "1.3"
GAME_NAME = "red"
YELLOW = GAME_NAME == "yellow"
INTERNAL = false

local START_WAIT = 99

local Battle = require "action.battle"
local Textbox = require "action.textbox"
local Walk = require "action.walk"

local Combat = require "ai.combat"
local Control = require "ai.control"
local Strategies = require("ai."..GAME_NAME..".strategies")

local Bridge = require "util.bridge"
local Input = require "util.input"
local Memory = require "util.memory"
local Menu = require "util.menu"
local Paint = require "util.paint"
local Utils = require "util.utils"
local Settings = require "util.settings"

local Pokemon = require "storage.pokemon"

local hasAlreadyStartedPlaying = false
local inBattle, oldSecs
local running = true
local previousPartySize = 0
local lastHP
local criticaled = false

local function startNewAdventure()
	local startMenu, withBattleStyle
	if YELLOW then
		startMenu = Memory.raw(0x0F95) == 0
		withBattleStyle = "battle_style"
	else
		startMenu = Memory.value("player", "name") ~= 0
	end
	if startMenu and Menu.getCol() ~= 0 then
		if Settings.set("text_speed", "battle_animation", withBattleStyle) then
			Menu.select(0)
		end
	elseif math.random(0, START_WAIT) == 0 then
		Input.press("Start")
	end
end

local function choosePlayerNames()
	local name
	if Memory.value("player", "name2") == 80 then
		name = "E"
	else
		name = "B"
	end
	Textbox.name(name, true)
end

local function pollForResponse()
	local response = Bridge.process()
	if response then
		Bridge.polling = false
		Textbox.setName(tonumber(response))
	end
end

local function resetAll()
	Strategies.softReset()
	Combat.reset()
	Control.reset()
	Walk.reset()
	Paint.reset()
	Bridge.reset()
	oldSecs = 0
	running = false
	previousPartySize = 0
	-- client.speedmode = 200

	if CUSTOM_SEED then
		Strategies.seed = CUSTOM_SEED
		print("RUNNING WITH A FIXED SEED ("..Strategies.seed.."), every run will play out identically!")
	else
		Strategies.seed = os.time()
	end
	math.randomseed(Strategies.seed)
end

-- EXECUTE

Control.init()

print("Welcome to PokeBot "..GAME_NAME.." version "..VERSION)
STREAMING_MODE = not Walk.init()
if INTERNAL and STREAMING_MODE then
	RESET_FOR_TIME = true
end

if CUSTOM_SEED then
	client.reboot_core()
else
	hasAlreadyStartedPlaying = Utils.ingame()
end

Strategies.init(hasAlreadyStartedPlaying)
if RESET_FOR_TIME and hasAlreadyStartedPlaying then
	RESET_FOR_TIME = false
	print("Disabling time-limit resets as the game is already running. Please reset the emulator and restart the script if you'd like to go for a fast time.")
end
if STREAMING_MODE then
	Bridge.init()
else
	Input.setDebug(true)
end

-- Main loop

local previousMap

while true do
	local currentMap = Memory.value("game", "map")
	if currentMap ~= previousMap then
		Input.clear()
		previousMap = currentMap
	end
	if Strategies.frames then
		if Memory.value("game", "battle") == 0 then
			Strategies.frames = Strategies.frames + 1
		end
		gui.text(0, 80, Strategies.frames)
	end
	if Bridge.polling then
		pollForResponse()
	end

	if not Input.update() then
		if not Utils.ingame() then
			if currentMap == 0 then
				if running then
					if not hasAlreadyStartedPlaying then
						client.reboot_core()
						hasAlreadyStartedPlaying = true
					else
						resetAll()
					end
				else
					startNewAdventure()
				end
			else
				if not running then
					Bridge.liveSplit()
					running = true
				end
				choosePlayerNames()
			end
		else
			local battleState = Memory.value("game", "battle")
			if battleState > 0 then
				if battleState == 1 then
					if not inBattle then
						Control.wildEncounter()
						inBattle = true
					end
				end
				local isCritical
				local battleMenu = Memory.value("battle", "menu")
				if battleMenu == 94 then
					isCritical = false
				elseif Memory.double("battle", "our_hp") == 0 then
					if Memory.value("battle", "critical") == 1 then
						isCritical = true
					end
				end
				if isCritical ~= nil and isCritical ~= criticaled then
					criticaled = isCritical
					Strategies.criticaled = criticaled
				end
			else
				inBattle = false
			end
			local currentHP = Pokemon.index(0, "hp")
			-- if currentHP ~= lastHP then
			-- 	Bridge.hp(currentHP, Pokemon.index(0, "max_hp"))
			-- 	lastHP = currentHP
			-- end
			if currentHP == 0 and not Control.canDie() and Pokemon.index(0) > 0 then
				Strategies.death(currentMap)
			elseif Walk.strategy then
				if Strategies.execute(Walk.strategy) then
					Walk.traverse(currentMap)
				end
			elseif battleState > 0 then
				if not Control.shouldCatch(partySize) then
					Battle.automate()
				end
			elseif Textbox.handle() then
				Walk.traverse(currentMap)
			end
		end
	end

	if STREAMING_MODE then
		local newSecs = Memory.raw(0x1A44)
		if newSecs ~= oldSecs and (newSecs > 0 or Memory.raw(0x1A45) > 0) then
			Bridge.time(Utils.elapsedTime())
			oldSecs = newSecs
		end
	elseif PAINT_ON then
		Paint.draw(currentMap)
	end

	Input.advance()
	emu.frameadvance()
end

Bridge.close()