-- kyogre script for sapphire v1.0 --
-- Written by ProjectRevoTPP --
-- For use on Bizhawk. --

-- TO USE: Run as soon as the map loads and your player has control ASAP and let the script do the rest to manip the right Kyogre.--

IWRAM = 0x03000000
EWRAM = 0x02000000

V_BLANK_COUNTER_ADDR = 0x03001790
RTC_ERROR_STATUS = 0x03000458

-- altered sapphire ROM ONLY!
NEW_RTC_COUNTER_SAPPHIRE = 0x0203A3D8

WHICH_KYOGRE = 2 -- 1-4 for which kyogre

A_PRESS_FRAME = 1385 -- kyogre offset before delay, account for bizhawk delay weirdness
KYOGRE_DELAY = 700 -- DON'T ALTER THIS!!!

globaladdrchange = 0 -- used for pcall protect

function FramesAdvance(nframes)
	for frame = 1, nframes do
		emu.frameadvance()
	end
end

function trychangedomain()
	memory.usememorydomain(globaladdrchange)
end

-- this can probably be cleaned up but whatever.
function getMemOffset(domaintype)
	if domaintype == "IWRAM" then
		return IWRAM
	elseif domaintype == "EWRAM" then
		return EWRAM
	else
		print("Unsupported mem offset. Assuming main and returning 0.")
		return 0
	end
end

function readSpecific(domaintype, readtype, endian, addr) -- 0 endian for little, 1 for big
local returnVar = 0xFF
globaladdrchange = domaintype -- set the var for the pcall since pcall does not support passing arguments

if endian == 0 then
	if readtype == 8 then
		if pcall(trychangedomain) then
			returnVar = memory.read_u8_le(addr - getMemOffset(domaintype))
		else
			print("Error changing memory domains.")
			return
		end
	elseif readtype == 16 then
		if pcall(trychangedomain) then
			returnVar = memory.read_u16_le(addr - getMemOffset(domaintype))
		else
			print("Error changing memory domains.")
			return
		end
	elseif readtype == 24 then
		if pcall(trychangedomain) then
			returnVar = memory.read_u24_le(addr - getMemOffset(domaintype))
		else
			print("Error changing memory domains.")
			return
		end
	elseif readtype == 32 then
		if pcall(trychangedomain) then
			returnVar = memory.read_u32_le(addr - getMemOffset(domaintype))
		else
			print("Error changing memory domains.")
			return
		end
	else
		print("Invalid readtype. Exiting script.")
		return
	end
elseif endian == 1 then
	if readtype == 8 then
		if pcall(trychangedomain) then
			returnVar = memory.read_u8_be(addr - getMemOffset(domaintype))
		else
			print("Error changing memory domains.")
			return
		end
	elseif readtype == 16 then
		if pcall(trychangedomain) then
			returnVar = memory.read_u16_be(addr - getMemOffset(domaintype))
		else
			print("Error changing memory domains.")
			return
		end
	elseif readtype == 24 then
		if pcall(trychangedomain) then
			returnVar = memory.read_u24_be(addr - getMemOffset(domaintype))
		else
			print("Error changing memory domains.")
			return
		end
	elseif readtype == 32 then
		if pcall(trychangedomain) then
			returnVar = memory.read_u32_be(addr - getMemOffset(domaintype))
		else
			print("Error changing memory domains.")
			return
		end
	else
		print("Invalid readtype. Exiting script.")
		return
	end
else
	print("Invalid endian type. Exiting script.")
	return
end

return returnVar
end

function checkEmu()
		check = console.getavailabletools --this returns an error if not on Bizhawk
end

if pcall(checkEmu) then
      print("Loaded script on Bizhawk, setting memory domain to IWRAM")
	  memory.usememorydomain("IWRAM");
else
      print("VBA-RR or other emulator detected - will not switch memory domain.") -- dont do anything else, only bizhawk needs the memory domain set
end

if(readSpecific("IWRAM", 16, 0, RTC_ERROR_STATUS) == 0x0000) then
	memory.write_u16_le(RTC_ERROR_STATUS - getMemOffset("IWRAM"), 0x0FF0) -- force RTC to error.
	print("RTC forced into error state.")
end

function drawguitext(framecount, rngcounter)
	gui.text(1,10,string.format("Frame Counter: %i",framecount))
	gui.text(1,18,string.format("RNG Counter: %i",rngcounter))
end

-- press left.
local input = {}
input["Left"] = true
joypad.set(input)

-- wait 120 frames.
FramesAdvance(120)

-- main loop --

repeat
if(readSpecific("IWRAM", 16, 0, RTC_ERROR_STATUS) == 0x0000) then
	memory.write_u16_le(RTC_ERROR_STATUS - getMemOffset("IWRAM"), 0x0FF0) -- force RTC to error.
	print("RTC forced into error state.")
end
local vblankcounter = readSpecific("IWRAM", 32, 0, V_BLANK_COUNTER_ADDR)
local rngcounter = readSpecific("EWRAM", 32, 0, NEW_RTC_COUNTER_SAPPHIRE)
drawguitext(vblankcounter, rngcounter)

if(vblankcounter == (A_PRESS_FRAME - KYOGRE_DELAY - 1 + (WHICH_KYOGRE - 1))) then -- it takes 1 frame to input
	local input = {}
	input["A"] = true
	joypad.set(input)
	print("This should produce the right Kyogre.")
	print(string.format("I inputted on: %i, Kyogre RNG frame is expected to be %i",vblankcounter + 1, vblankcounter + KYOGRE_DELAY + 1)) -- it takes 1 frame to input, account for it.
	print(string.format("Current RNG counter is %i.", rngcounter))
	return -- end script.
end

emu.frameadvance()
until false

gui.register(drawguitext)