-- Luascript for bizhawk 2.9.1
-- by mugg1991
-- Motocross Maniacs gb

function text(x, y, text, color)
	gui.pixelText(x, y, text,color,0x00000000)
end

function drawButton(posx,posy,width,height,label,color,frequency,clickedfunction)

	if X>posx and X<posx+width and Y>posy and Y<posy+height then

		if clicked and clickedFrames%frequency==1 then
		    clickedfunction()
		end
		gui.drawBox(posx,posy,posx+width,posy+height,0x00606060,color+0xFF484848)
	else
		gui.drawBox(posx,posy,posx+width,posy+height,0x00606060,color)
	end
	
	text(posx+2,posy+2,label,0xFFFFFFFF)
	
end

local drawFrameCountAndInput = function(x,y)

	if movie.mode()=="PLAY" then
		text(x, y,emu.framecount().."/"..movie.length(),0xFFFFFFFF)
	else
		text(x, y,emu.framecount(),0xFFFFFFFF)
	end
		
	if emu.islagged() then 
		text(x, y+8,emu.lagcount() .. " *",0xFFF05050)
	else
		text(x, y+8,emu.lagcount(),0xFFF05050)
	end
	
	local inputtable = {}
	
	if movie.mode()=="INACTIVE" then 
		inputtable = joypad.getimmediate()
	elseif movie.mode()=="PLAY" or movie.mode()=="RECORD" then
		inputtable = movie.getinput(emu.framecount()-1) 
	end
	
	local buttons = {["Up"]="^", ["Down"]="v", ["Left"]="<", ["Right"]=">", ["Select"]="s", ["Start"]="S", ["A"]="A", ["B"]="B", ["L"]="L", ["R"]="R"}
	local s = ""
	for k,v in pairs(inputtable) do
		if v==true then
			s=s..buttons[k]
		end
	end
	text(x,y+16,s,0xFFffffff)

end

function hex(num)
   local dig = 2
   local hexstr = '0123456789ABCDEF'
   local s = ''
   --while num > 0 do
   while dig > 0 do -- use digit instead so there is padding
       local mod = math.fmod(num, 16)
       s = string.sub(hexstr, mod+1, mod+1) .. s
       num = math.floor(num / 16)
       dig = dig - 1
   end
   if s == '' then s = '0' end
   return s
end

local displayItems = function()
	hiddenjets=0
	items=0
	for i=0,192,8 do
	
			item_type = memory.read_u8(0xca00+i)
			item_status = memory.read_u8(0xca01+i)
			item_x = memory.read_u8(0xca03+i)+memory.read_u8(0xca04+i)*256
			item_y = memory.read_u8(0xca02+i)
			item_xcam = item_x-Xcam/256
			item_ycam = item_y-Ycam/256+28
			exists=false
			if item_status==3 and item_y~=0 then exists=true end
			
			if item_type>0 and item_type<8 and exists then
				items=items+1
			end
			
			if item_type==6 and exists then 
				hiddenjets=hiddenjets+1
			end
			
			if exists then
				for p=0,4096,4096 do --4096 is the length of the course. We want to display items for both laps.
					if item_ycam < 24 then -- don't obscure lua hud
						gui.drawBox(item_xcam+20+p,item_ycam,item_xcam+36+p,item_ycam+16,0x00FF0000,0x40FFFFFF)				
					else
						gui.drawBox(item_xcam+20+p,item_ycam,item_xcam+36+p,item_ycam+16,0x60FF0000,0x20FF0000)
						text(item_xcam+21+p,item_ycam+1,itemTable[item_type],0xFF000000)
					end
				end
			end
	end
	text(110,30,"jets: "..hiddenjets,0x80000000)
	text(156,30,"items: "..items,0x80000000)
end

function file_exists(name)
   local f=io.open(name,"r")
   local exists=false
   if f~=nil then 
		exists=true 
		io.close(f)
   else 
		exists=false
   end
	return exists
end

local writeGhostToFile = function()

	FileName=LevelName.."_Ghost_"..hex(Minutes).."_"..hex(Seconds).."_"..hex(Milliseconds)..".txt"
	
	if file_exists(FileName) then
			
		print("NOTE: The ghost file was not written because there already exists a ghost file with the same record time.")
		
	else
	
		io.output(FileName)
		
		table.sort(ghostTable)

		
		for key,value in pairs(ghostTable) do
			io.write(key.." "..ghostDataTable[value].."\n")
		end
			
		io.close()
		
		if file_exists(FileName)==true then
			print("NOTE: File "..FileName.." has been written.")
		elseif file_exists(FileName)==false then
			print("NOTE: File has not been written because of an unknown error.")		
		end	
	
	end
	
end

local writeGhostToTable = function()

	ghostFrames=ghostFrames+1
	ghostTable[ghostFrames]=TotalTime
	ghostDataTable[TotalTime]="Time:"..TotalTime.." X:"..Xscreen.." Y:"..Yscreen.." Xcam:"..Xcam.." Ycam:"..Ycam.." Speed:"..Speed2.." Angle:"..Angle.." Nitro:"..Nitros -- extra table because table.sort didn't work in my favor
	
	if TotalTime>50 then
		if TotalTimeUnchangedCount>20 or recordingGhost==false then --recording finished
			if Xpos~=0 then
				writeGhostToFile() 
			else
				print("NOTE: The ghost file wasn't written because you haven't moved.")
			end
			recordingGhost=false
			ghostFrames=0
		end
	end
	
end

function doLine(filename)
	local f = io.open(filename, "r")

	local iterator = function(f, prevLine)
		if f==nil then
			return nil
		else
			local success, line = pcall(function()
				return f:read("*l"):gsub("\n",""):gsub("\r","")
			end)
			if not success or line == nil then
				f:close()
				line = nil
			end

			return line
		end
	end

	return iterator, f, ""
end

local readGhostFromFile = function()

		local TimeIndex
		local XscreenIndex
		local YscreenIndex
		local XcamIndex
		local YcamIndex
		local SpeedIndex
		local AngleIndex
		local NitroIndex
		
		local highestTimeInGhost = 0
		
		for line in doLine(PlayFilePath) do  -- finding the settings and assigning them to variables

			TimeIndex = string.find(line,"Time:")
			XscreenIndex = string.find(line,"X:")
			YscreenIndex = string.find(line,"Y:")
			XcamIndex = string.find(line,"Xcam:")
			YcamIndex = string.find(line,"Ycam:")
			SpeedIndex = string.find(line,"Speed:")
			AngleIndex = string.find(line,"Angle:")
			NitroIndex = string.find(line,"Nitro:")
			
			Time = tonumber( string.sub(line, TimeIndex + 5, XscreenIndex - 1) )
			Xscreen = tonumber( string.sub(line, XscreenIndex + 2, YscreenIndex - 1) )
			Yscreen = tonumber( string.sub(line, YscreenIndex + 2, XcamIndex - 1) )
			Xcam = tonumber( string.sub(line, XcamIndex + 5, YcamIndex - 1) )
			Ycam = tonumber( string.sub(line, YcamIndex + 5, SpeedIndex - 1) )
			Speed = tonumber( string.sub(line, SpeedIndex + 6, AngleIndex - 1) )
			Angle = tonumber( string.sub(line, AngleIndex + 6, NitroIndex - 1) )
			Nitros = tonumber( string.sub(line, NitroIndex + 6) )
						
			playTable[Time] = {
				Xscreen = Xscreen,
				Yscreen = Yscreen,
				Xcam = Xcam,
				Ycam = Ycam,
				Speed = Speed,
				Angle = Angle,
				Nitros = Nitros
			}
			
			if Time > highestTimeInGhost then highestTimeInGhost = Time end
			
		end
		
		if TotalTime > highestTimeInGhost then
			playTable = {}
			playingGhost = false
			print("NOTE: Cannot play ghost, since its data is in the past.")
		end
end

local readGhostFromTable = function()
	if playTable[2]~=nil then -- ghost is loaded		
		if playTable[TotalTime] == nil then
			-- ghost file reached its end
			playTable = {}
			playingGhost = false
			print("NOTE: End of ghost file reached.")
			
		else
			Xscreen = playTable[TotalTime].Xscreen
			Yscreen = playTable[TotalTime].Yscreen
			Xcam = playTable[TotalTime].Xcam / 256
			Ycam = playTable[TotalTime].Ycam / 256
			Speed = playTable[TotalTime].Speed
			Angle = playTable[TotalTime].Angle
			Nitros = playTable[TotalTime].Nitros

			-- draw ghost
			gui.drawBox(Xcam+20,Ycam,Xcam+36,Ycam+16,0xFFFF0000,0xFFFFFFFF)	
			print(Xcam, Ycam, Xscreen, Yscreen)
		end
	else -- ghost not loaded yet
		readGhostFromFile() 
	end
end

local displayGhosts = function()
	
	gui.drawBox(200,-1,300,172,0x00000000,0xFF000000)
	gui.drawLine(205,12,296,12,0xFF303030)
	
	--title lol
	text(205,4,"Ghost menu",0xFFFFFFFF)
	-- mode
	if playingGhost then
		text(205,16,"(playback)",0x60FFFFFF)
	elseif recordingGhost then
		text(205,16,"(recording)",0x60FFFFFF)	
	else
		text(205,16,"(idle)",0x60FFFFFF)
	end
	
	--recording button
	if recordingGhost then
	
		drawButton(205,52,35,10,"Stop",0xFF484848,20,function()
			recordingGhost=false
		end)
		
		writeGhostToTable()
		
		gui.drawEllipse(245,54,6,6,0xFFFF0000,0xFFFF0000)
		
	else
	
		drawButton(205,52,35,10,"Record",0xFF484848,20,function()
			if playingGhost then
				print("NOTE: Stop playing back before recording.")
				ghostTable = {}
				ghostDataTable = {}
				recordingGhost=false
			else
				if TotalTime~=0 or Level==0 then 
					print("NOTE: Not recording. Click Record before a course starts (time is 0).")
					ghostTable = {}
					ghostDataTable = {}
					recordingGhost=false
				else
					recordingGhost=true
				end
			end
		end)
		
	end
	
	-- selected playback file
	if PlayFilePath=="" then 
		t="-" 
	else
		t=PlayFilePath:sub(-21)
	end
	
	drawButton(205,82,95,10,t,0xFF181818,20,function()
		FormsSelectedPath = forms.openfile()
	end)
	
	if FormsSelectedPath ~= nil then PlayFilePath=FormsSelectedPath end
	
	-- playback buttons
	if playingGhost then
	
		drawButton(205,70,35,10,"Stop",0xFF484848,20,function()
			playingGhost=false
		end)
		
		readGhostFromTable()
				
		gui.drawLine(245,72,245,78,0xFF00FF00)
		gui.drawLine(246,73,246,77,0xFF00FF00)
		gui.drawLine(247,74,247,76,0xFF00FF00)
		gui.drawPixel(248,75,0xFF00FF00)
	
	else
	
		drawButton(205,70,35,10,"Play",0xFF484848,20,function()
			if recordingGhost then
			
				print("NOTE: Stop recording before playing back a ghost.")
				playingGhost=false
				playTable={}
				
			elseif FormsSelectedPath==nil then
			
				print("NOTE: Please select a file before playing back.")
				playingGhost=false
				playTable={}
				
			elseif tostring(PlayFilePath):sub(-4)~=".txt" then
			
				print("NOTE: The selected file is not a .txt file.")
				playingGhost=false
				playTable={}
				
			elseif not file_exists(PlayFilePath) then
			
				print("NOTE: The selected file doesn't exist.")
				playingGhost=false
				playTable={}
				
			else
				playingGhost=true
			end
	

		end)
	
	end
	
end

local getMouse = function()
    X = input.getmouse().X+20
    Y = input.getmouse().Y+28
	clicked = input.getmouse().Left
	
	if clicked then clickedFrames = clickedFrames + 1 
	else clickedFrames = 0 end
end

event.onexit(function()
	client.SetGameExtraPadding(0,0,0,0)
end)

event.onloadstate(function()

	if recordingGhost then
		recordingGhost=false
		print("NOTE: The recording has stopped because you have loaded a state.")
	end
	
end)

memory.usememorydomain("System Bus")

itemTable = {
[-1] = "???",
[2] = "Speed", 
[3] = "Time", 
[4] = "Nitro",
[5] = "Rig",
[6] = "Jet", 
[7] = "Bike",
}

difficultyTable = {
[-1]="?",
[0]="A",
[1]="B",
[2]="C"
}

ghostTable = {}
ghostDataTable = {}
playTable = {}

BOOL_items=false
BOOL_ghosts=false
X = 0
Y = 0
clicked=false
clickedFrames=0
recordingGhost=false
playingGhost=false
TotalTimeUnchangedCount=0
TotalTimeBefore=0
TotalTime=0
ghostFrames=0
PlayFilePath="" -- the file path name of the ghost file that should be played back

if BOOL_ghosts then client.SetGameExtraPadding(20,28,130,0) 
else client.SetGameExtraPadding(20,28,20,0) end
console.clear()

-----

while true do
	
	---- Mouse 
	getMouse()
	
	if emu.framecount()>400 then		
		--declare values:
		Minutes = memory.read_u8(0xc804)
		Seconds = memory.read_u8(0xc803)
		Milliseconds = memory.read_u8(0xc802)
		MinutesLeft = memory.read_u8(0xc882)
		SecondsLeft = memory.read_u8(0xc881)
		MillisecondsLeft = memory.read_u8(0xc880)
		Speed1 = memory.read_s8(0xc810) -- pixels to advance screen 
		Speed2 = memory.read_u16_le(0xc38a) 
		Nitros1 = memory.read_u8(0xc792) 
		Nitros2 = memory.read_u8(0xc812) 
		Indicator = memory.read_u8(0xc780) 
		if Indicator==1 then Nitros=Nitros1 else Nitros=Nitros2 end
		Xpos = memory.read_u16_le(0xe903)
		Xcam = memory.read_u24_le(0xc902)
		Ycam = memory.read_u16_le(0xc900)
		
		Xscreen = memory.read_u8(0xcc01)+8
		Yscreen = memory.read_u8(0xcc00)+8
		
		Level = memory.read_u8(0xc018)
		Difficulty = memory.read_u8(0xc01B)
		if Difficulty<0 or Difficulty>2 then Difficulty=-1 end
		LevelName=Level..difficultyTable[Difficulty]
		
		if Level~=0 then 
			Angle = memory.read_u8(0xcc02) -- this is upper left corner, value is a pointer for what graphic to load I guess 
			if Angle%2==0 then Angle=1+Angle/4
			else Angle=16+(Angle+1)/4 end
			if Speed2==0 then Angle=0 end
			TotalTime = hex(Milliseconds) + hex(Seconds)*100 + hex(Minutes)*6000
		else
			Angle = -1
			TotalTime = -1
		end
		
		Angle = math.floor(Angle) -- thanks newer Bizhawk lua
		
		if TotalTimeBefore==TotalTime then
			TotalTimeUnchangedCount=TotalTimeUnchangedCount+1
		else
			TotalTimeUnchangedCount=0
		end
		
		TotalTimeBefore=TotalTime
		
		--lines boxes
		gui.drawBox(-1,27,20,172,0x00000000,0xFFFFFFFF)
		gui.drawBox(179,27,201,172,0x00000000,0xFFFFFFFF)
		gui.drawBox(-1,-1,201,28,0x00000000,0xFF202020)
		gui.drawLine(34,0,34,27,0xFF404040)
		gui.drawLine(156,0,156,27,0xFF404040)
		
		--racetrack visualisation
		gui.drawBox(30,27,160,28,0xFFA0A0A0)--track
		gui.drawLine(30,27,30,28,0xFFa0FF00)--start
		gui.drawLine(95,27,95,28,0xFFa0FF00)--middle
		gui.drawLine(160,27,160,28,0xFFa0FF00)--finish
		Xtrack=30+(Xpos/8192)*130
		gui.drawLine(Xtrack,27,Xtrack,28,0xFFFF0000)--you
		
		--display values
		text(45,2,"Time",0x80FFFFFF)
		text(49,10,"Gas",0x80FFFFFF)
		text(103,2,"Spd",0x80FFFFFF)
		text(103,10,"Ntr   Angl",0x80FFFFFF)	
		text(114,18,"pos",0x80FFFFFF)	
		text(38,18,"cam",0x80FFFFFF)
		
		if Level ~=0 then
			text(1,30,"Lv: "..LevelName,0x80000000)	
			--text(74,30,TotalTime,0x80000000)
			--text(46,30,Angle,0x80000000)
			text(65,2,hex(Minutes)..":"..hex(Seconds).."."..hex(Milliseconds),0xFFFFFFFF)
			text(65,10,hex(MinutesLeft)..":"..hex(SecondsLeft).."."..hex(MillisecondsLeft),0xFFFFFFFF)
			text(119,2,Speed2,0xFFFFFFFF)	
			text(119,10,Nitros,0xFFFFFFFF)	
			text(147,10,Angle,0xFFFFFFFF)			
			text(134,18,Xpos,0xFFFFFFFF)	
			text(53,18,Xcam,0xFFFFFFFF)
			text(88,18,Ycam,0xFFFFFFFF)	
		end		
		
		drawButton(160,2,35,10,"ITEMS",0xFF484848,15,function()
			BOOL_items= not BOOL_items
		end)
		
		drawButton(160,14,35,10,"GHOSTS",0xFF484848,15,function()
			BOOL_ghosts= not BOOL_ghosts
			if BOOL_ghosts then 
				client.SetGameExtraPadding(20,28,130,0)
			else
				client.SetGameExtraPadding(20,28,20,0)
			end
		end)
		
		if BOOL_items==true then 
			text(189,4,"o",0xFF00c000)
			displayItems() 
		else	
			text(189,4,"x",0xFFF05050)
		end
		
		if BOOL_ghosts==true then
			text(189,16,"o",0xFF00C000)
			displayGhosts()
		else	
			text(189,16,"x",0xFFF05050)
		end
	end
	
	drawFrameCountAndInput(2,2)
	
	--emu.frameadvance()
	emu.yield()
end