PokeBot/main.lua

-- OPTIONS

RESET_FOR_TIME = false -- 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 NIDORAN_NAME = "A" -- Set this to the single character to name Nidoran (note, to replay a seed, it MUST match!)
local PAINT_ON    = true -- Display contextual information while the bot runs

-- START CODE (hard hats on)

VERSION = "1.4.3"
YELLOW = memory.getcurrentmemorydomainsize() > 30000
GAME_NAME = YELLOW and "yellow" or "red"

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 Data = require "data.data"
local Pokemon = require "storage.pokemon"

local hasAlreadyStartedPlaying = false
local oldSeconds
local running = true
local lastHP

-- HELPERS

local function resetAll()
	Strategies.softReset()
	Combat.reset()
	Control.reset()
	Walk.reset()
	Paint.reset()
	Bridge.reset()
	oldSeconds = 0
	running = false

	if CUSTOM_SEED then
		Data.run.seed = CUSTOM_SEED
		Strategies.replay = true
		p("RUNNING WITH A FIXED SEED ("..NIDORAN_NAME.." "..Data.run.seed.."), every run will play out identically!", true)
	else
		Data.run.seed = os.time()
		print("PokeBot v"..VERSION..": starting a new run with seed "..Data.run.seed)
	end
	math.randomseed(Data.run.seed)
end

-- EXECUTE

p("Welcome to PokeBot "..GAME_NAME.." version "..VERSION, true)

Control.init()

STREAMING_MODE = not Walk.init()

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
	p("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.", true)
end
if STREAMING_MODE then
	if not CUSTOM_SEED and INTERNAL then
		RESET_FOR_TIME = true
	end
	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
		Utils.drawText(0, 80, Strategies.frames)
	end
	if Bridge.polling then
		Settings.pollForResponse(NIDORAN_NAME)
	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
					Settings.startNewAdventure(START_WAIT)
				end
			else
				if not running then
					Bridge.liveSplit()
					running = true
				end
				Settings.choosePlayerNames()
			end
		else
			local battleState = Memory.value("game", "battle")
			Control.encounter(battleState)
			local curr_hp = Pokemon.index(0, "hp")
			-- if curr_hp ~= lastHP then
			-- 	Bridge.hp(curr_hp, Pokemon.index(0, "max_hp"))
			-- 	lastHP = curr_hp
			-- end
			if curr_hp == 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 newSeconds = Memory.value("time", "seconds")
		if newSeconds ~= oldSeconds and (newSeconds > 0 or Memory.value("time", "frames") > 0) then
			Bridge.time(Utils.elapsedTime())
			oldSeconds = newSeconds
		end
	elseif PAINT_ON then
		Paint.draw(currentMap)
	end

	Input.advance()
	emu.frameadvance()
end

Bridge.close()