--
-- traffic_sensor_client.lua
--

---------------------------------------------------------------------------
--
-- Handle events from Race
--
---------------------------------------------------------------------------

addEventHandler('onClientResourceStart', resourceRoot,
	function()
		Bigdar.setHotKey( 'F4' )
	end
)

addEvent('onClientMapStarting', true)
addEventHandler('onClientMapStarting', root,
	function(mapinfo)
		outputDebug( 'BIGDAR', 'onClientMapStarting' )
		if mapinfo.modename == "Destruction derby" or mapinfo.modename == "Freeroam" then
			Bigdar.allowed = false
		else
			Bigdar.allowed = true
		end
		Bigdar.finishedPlayers = {}
		Bigdar.allPlayers = {}
	end
)

addEvent('onClientMapStopping', true)
addEventHandler('onClientMapStopping', root,
	function()
		outputDebug( 'BIGDAR', 'onClientMapStopping' )
	end
)

addEvent('onClientPlayerFinish', true)
addEventHandler('onClientPlayerFinish', root,
	function()
		outputDebug( 'BIGDAR', 'onClientPlayerFinish' )
		Bigdar.finishedPlayers[source] = true
	end
)

---------------------------------------------------------------------------
-- Bigdar - Big radar
---------------------------------------------------------------------------
Bigdar = {}
Bigdar.smoothList = {}			-- {player = rrz}
Bigdar.allPlayers = {}
Bigdar.finishedPlayers = {}
Bigdar.lastSmoothSeconds = 0
Bigdar.beginValidSeconds = nil
Bigdar.enabled = true		-- Manual override
Bigdar.allowed = true		-- Map override
Bigdar.hidden = false		-- Black screen override
Bigdar.hotkey = nil		-- Current hotkey

function Bigdar.setHotKey ( hotkey )
	if Bigdar.hotkey then
		unbindKey ( Bigdar.hotkey, 'down', "showsensor" )
	end
	if hotkey and Bigdar.hotkey and hotkey ~= Bigdar.hotkey then
		outputConsole( "Race Traffic Sensor hotkey is now '" .. tostring(hotkey) .. "'" )
	end
	Bigdar.hotkey = hotkey
	if Bigdar.hotkey then
		bindKey ( Bigdar.hotkey, 'down', "showsensor" )
	end
end

function Bigdar.toggle()
	Bigdar.enabled = not Bigdar.enabled
	outputChatBox( 'Traffic Sensor is now ' .. (Bigdar.enabled and 'on' or 'off') )
end

function isPlayerFinished(player)
	return Bigdar.finishedPlayers[player]
end

function Bigdar.render()
	-- Ensure map allows it, and player not dead, and in a vehicle and not spectating
	local vehicle = getPedOccupiedVehicle(localPlayer)
	if	 not Bigdar.allowed
		or isPedDead(localPlayer)
		or isPlayerFinished(localPlayer)
		or not vehicle
		or getCameraTarget() ~= vehicle
		then
			Bigdar.beginValidSeconds = nil
			return
	end

	-- Ensure at least 1 second since g_BeginValidSeconds was set
	local timeSeconds = getSecondCount()
	if not Bigdar.beginValidSeconds then
		Bigdar.beginValidSeconds = timeSeconds
	end
	if timeSeconds - Bigdar.beginValidSeconds < 1 then
		return
	end

	-- No draw if faded out or not enabled
	if
		Bigdar.hidden or
		not Bigdar.enabled then
			return
	end

	-- Icon definition
	local icon = { file='img/arrow_ts.png', w=80, h=56 }

	-- Calc smoothing vars
	local delta = timeSeconds - Bigdar.lastSmoothSeconds
	Bigdar.lastSmoothSeconds = timeSeconds
	local timeslice = math.clamp(0,delta*14,1)

	-- Get screen dimensions
	local screenX,screenY = guiGetScreenSize()
	local halfScreenX = screenX * 0.5
	local halfScreenY = screenY * 0.5

	-- Get my pos and rot
	local mx, my, mz = getElementPosition(localPlayer)
	local _, _, mrz	= getCameraRot()

	-- To radians
	mrz = math.rad(-mrz)

	if #Bigdar.allPlayers == 0 then
		Bigdar.allPlayers = getElementsByType('player')
	end

	-- For each 'other player'
	for i,player in ipairs(Bigdar.allPlayers) do
		if player ~= localPlayer and not isPedDead(player) and not isPlayerFinished(player) then

			-- Get other pos
			local ox, oy, oz = getElementPosition(player)

			-- Only draw marker if other player it is close enough, and not on screen
			local maxDistance = 60
			local alpha = 1 - getDistanceBetweenPoints3D( mx, my, mz, ox, oy, oz ) / maxDistance
			local onScreen = getScreenFromWorldPosition ( ox, oy, oz )

			if onScreen or alpha <= 0 then
				-- If no draw, reset smooth position
				Bigdar.smoothList[player] = nil
			else
				-- Calc arrow color
				local r,g,b = 255,220,210
				local team = getPlayerTeam(player)
				if team then
					r,g,b = getTeamColor(team)
				end

				-- Calc draw scale
				local scalex = alpha * 0.5 + 0.5
				local scaley = alpha * 0.25 + 0.75

				-- Calc dir to
				local dx = ox - mx
				local dy = oy - my
				-- Calc rotz to
				local drz = math.atan2(dx,dy)
				-- Calc relative rotz to
				local rrz = drz - mrz

				-- Add smoothing to the relative rotz
				local smooth = Bigdar.smoothList[player] or rrz
				smooth = math.wrapdifference(-math.pi, smooth, rrz, math.pi)
				if math.abs(smooth-rrz) > 1.57 then
					smooth = rrz	-- Instant jump if more than 1/4 of a circle to go
				end
				smooth = math.lerp( smooth, rrz, timeslice )
				Bigdar.smoothList[player] = smooth
				rrz = smooth

				-- Calc on screen pos for relative rotz
				local sx = math.sin(rrz)
				local sy = math.cos(rrz)

				-- Draw at edge of screen
				local X1 = halfScreenX
				local Y1 = halfScreenY
				local X2 = sx * halfScreenX + halfScreenX
				local Y2 = -sy * halfScreenY + halfScreenY
				local X
				local Y
				if math.abs(sx) > math.abs(sy) then
					-- Left or right
					if X2 < X1 then
						-- Left
						X = 32
						Y = Y1+ (Y2-Y1)* (X-X1) / (X2-X1)
					else
						-- right
						X = screenX-32
						Y = Y1+ (Y2-Y1)* (X-X1) / (X2-X1)
					end
				else
					-- Top or bottom
					if Y2 < Y1 then
						-- Top
						Y = 32
						X = X1+ (X2-X1)* (Y-Y1) / (Y2 - Y1)
					else
						-- bottom
						Y = screenY-32
						X = X1+ (X2-X1)* (Y-Y1) / (Y2 - Y1)
					end
				end
				dxDrawImage ( X-icon.w/2*scalex, Y-icon.h/2*scaley, icon.w*scalex, icon.h*scaley, icon.file, 180 + rrz * 180 / math.pi, 0, 0, tocolor(r,g,b,255*alpha), false )
			end
		end
	end
end
addEventHandler('onClientRender', root, Bigdar.render)


---------------------------------------------------------------------------
-- Various events

addEventHandler('onClientPlayerJoin', root,
	function()
		table.insertUnique(Bigdar.allPlayers, source)
	end
)

addEventHandler('onClientPlayerQuit', root,
	function()
		table.removevalue(Bigdar.finishedPlayers,source)
		table.removevalue(Bigdar.allPlayers,source)
		Bigdar.smoothList[source] = nil
	end
)


addEvent ( "onClientScreenFadedOut", true )
addEventHandler ( "onClientScreenFadedOut", root,
	function()
		Bigdar.hidden = true
	end
)

addEvent ( "onClientScreenFadedIn", true )
addEventHandler ( "onClientScreenFadedIn", root,
	function()
		Bigdar.hidden = false
	end
)


---------------------------------------------------------------------------
--
-- Commands and binds
--
--
--
---------------------------------------------------------------------------

function onHotKey()
	Bigdar.toggle()
end
addCommandHandler ( "showsensor", onHotKey )

addCommandHandler('doF4',
	function(player,command,...)
		outputDebugString('doF4')
		Bigdar.toggle()
	end
)