--------------

-- TECH RACER

--------------

-- DEVELOPER WARNING - FOR OFFICIAL DEVELOPMENT --

-- Be careful when editig this script, do not introduce changes lightly!

-- This script is used for time records on the official Hedgewars server.

-- Introducing breaking changes means we have to invalidate past time records!

--------------

-- TO DO

--------------

-- allow scrolling of maps (was going to add this in the engine itself, but it can be done now by refreshing preview)

-----------------------------

-- SCRIPT BEGINS

-----------------------------

HedgewarsScriptLoad("/Scripts/Locale.lua")

HedgewarsScriptLoad("/Scripts/OfficialChallenges.lua")

HedgewarsScriptLoad("/Scripts/Tracker.lua")

HedgewarsScriptLoad("/Scripts/Params.lua")

HedgewarsScriptLoad("/Scripts/Utils.lua")

HedgewarsScriptLoad("/Scripts/TechMaps.lua")

------------------

-- Got Variables?

------------------

local atkArray = {

	{amBazooka, 	"amBazooka",		0},

	{amBee, 	"amBee",		0},

	{amMortar, 	"amMortar",		0},

	{amDrill, 	"amDrill",		0},

	{amSnowball, 	"amSnowball",		0},

	{amGrenade,	"amGrenade",		0},

	{amClusterBomb,	"amClusterBomb",	0},

	{amMolotov, 	"amMolotov",		0},

	{amWatermelon, 	"amWatermelon",		0},

	{amHellishBomb,	"amHellishBomb",	0},

	{amGasBomb, 	"amGasBomb",		0},

	{amShotgun,	"amShotgun",		0},

	{amDEagle,	"amDEagle",		0},

	{amFlamethrower,"amFlamethrower",	0},

	{amSniperRifle,	"amSniperRifle",	0},

	{amSineGun, 	"amSineGun",		0},

	{amIceGun, 	"amIceGun",		0},

	{amLandGun,	"amLandGun",		0},

	{amFirePunch, 	"amFirePunch",		0},

	{amWhip,	"amWhip",		0},

	{amBaseballBat, "amBaseballBat",	0},

	{amKamikaze, 	"amKamikaze",		0},

	{amSeduction, 	"amSeduction",		0},

	{amHammer,	"amHammer",		0},

	{amMine, 	"amMine",		0},

	{amDynamite, 	"amDynamite",		0},

	{amCake, 	"amCake",		0},

	{amBallgun, 	"amBallgun",		0},

	{amRCPlane,	"amRCPlane",		0},

	{amSMine,	"amSMine",		0},

	{amAirMine,	"amAirMine",		0},

	{amAirAttack,	"amAirAttack",		0},

	{amMineStrike,	"amMineStrike",		0},

	{amDrillStrike,	"amDrillStrike",	0},

	{amAirMine,	"amAirMine",		0},

	{amNapalm, 	"amNapalm",		0},

	{amPiano,	"amPiano",		0},

	{amKnife,	"amKnife",		0},

	{amBirdy,	"amBirdy",		0}

}

local utilArray = {

	{amBlowTorch, 	"amBlowTorch",		0},

	{amPickHammer,	"amPickHammer",		0},

	{amGirder, 	"amGirder",		0},

	{amRubber, 	"amRubber",		0},

	{amPortalGun,	"amPortalGun",		0},

	{amRope, 	"amRope",		0},

	{amParachute, 	"amParachute",		0},

	{amTeleport,	"amTeleport",		0},

	{amJetpack,	"amJetpack",		0},

	{amInvulnerable,"amInvulnerable",	0},

	{amLaserSight,	"amLaserSight",		0},

	{amVampiric,	"amVampiric",		0},

	{amLowGravity, 	"amLowGravity",		0},

	{amExtraDamage, "amExtraDamage",	0},

	{amExtraTime,	"amExtraTime",		0},

	{amResurrector, "amResurrector",	0},

	{amTardis, 	"amTardis",		0},

	{amSwitch,	"amSwitch",		0}

}

local activationStage = 0

local jet = nil

portalDistance = 5000

ufoFuel = 0

local roundLimit = 3

local roundNumber = 0

local firstClan = 10

local fastX = {}

local fastY = {}

local fastCount = 0

local fastIndex = 0

local fastColour = 0xffffffff

local currX = {}

local currY = {}

local currCount = 0

local specialPointsX = {}

local specialPointsY = {}

local specialPointsFlag = {}

local specialPointsCount = 0

mapID = nil

--------------------------

-- hog and team tracking variales

--------------------------

local numhhs = 0 -- store number of hedgehogs

local hhs = {} -- store hedgehog gears

local numTeams --  store the number of teams in the game

local teamNameArr = {}  -- store the list of teams

local teamClan = {}

local teamSize = {}     -- store how many hogs per team

local teamIndex = {} -- at what point in the hhs{} does each team begin

local teamComment = {}

local teamScore = {}

-------

-- racer vars

--------

local cGear = nil

local cameraGear = nil

local bestClan = 10

local bestTime = MAX_TURN_TIME

local gameBegun = false

local gameOver = false

local racerActive = false

local trackTime = 0

local wpCirc = {}

local wpX = {}

local wpY = {}

local wpCol = {}

local wpActive = {}

local wpRad = 450

local wpCount = 0

local wpLimit = 20

local usedWeapons = {}

local roundN

local lastRound

local RoundHasChanged

local cnthhs = 0

-------------------

-- general methods

-------------------

function RebuildTeamInfo()

	-- make a list of individual team names

	for i = 0, (TeamsCount-1) do

		teamNameArr[i] = " " -- = i

		teamSize[i] = 0

		teamIndex[i] = 0

		teamScore[i] = MAX_TURN_TIME

	end

	numTeams = 0

	for i = 0, (numhhs-1) do

		local z = 0

		local unfinished = true

		while(unfinished == true) do

			local newTeam = true

			local tempHogTeamName = GetHogTeamName(hhs[i]) -- this is the new name

			if tempHogTeamName == teamNameArr[z] then

				newTeam = false

				unfinished = false

			end

			z = z + 1

			if z == TeamsCount then

				unfinished = false

				if newTeam == true then

					teamNameArr[numTeams] = tempHogTeamName

					numTeams = numTeams + 1

				end

			end

		end

	end

	-- find out how many hogs per team, and the index of the first hog in hhs

	for i = 0, (numTeams-1) do

		for z = 0, (numhhs-1) do

			if GetHogTeamName(hhs[z]) == teamNameArr[i] then

				teamClan[i] = GetHogClan(hhs[z])

				if teamSize[i] == 0 then

					teamIndex[i] = z -- should give starting index

				end

				teamSize[i] = teamSize[i] + 1

				--add a pointer so this hog appears at i in hhs

			end

		end

	end

end

-----------------

-- RACER METHODS

-----------------

-- Returns min opacity, max opacity and flashing speed (`FrameTicks`)

-- for the waypoint visual gears

function FlashingHelper(wpIndex)

	local minO, maxO, flashing

	if wpIndex == 0 then

		-- Notable flashing of first waypoint

		minO, maxO = 92, 255

		flashing = 2

	else

		-- Slow pulsation

		minO, maxO = 164, 224

		flashing = 10

	end

	return minO, maxO, flashing

end

function CheckWaypoints()

	trackFinished = true

	for i = 0, (wpCount-1) do

		local g1X, g1Y = GetGearPosition(CurrentHedgehog)

		local g2X, g2Y = wpX[i], wpY[i]

		local g1X = g1X - g2X

		local g1Y = g1Y - g2Y

		local dist = (g1X*g1X) + (g1Y*g1Y)

		local NR = (48/100*wpRad)/2

		if dist < (NR*NR) and not gameOver then

			wpCol[i] = GetClanColor(GetHogClan(CurrentHedgehog))

			SetVisualGearValues(wpCirc[i], wpX[i], wpY[i], 64, 64, 1, 10, 0, wpRad, 5, wpCol[i])

			local wpRem = 0

			for k = 0, (wpCount-1) do

				if wpActive[k] == false then

					wpRem = wpRem + 1

				end

			end

			if not wpActive[i] then

				local wpMessage = ""

				if wpRem-1 == 0 then

					wpMessage = loc("Track completed!")

				else

					wpMessage = string.format(loc("Waypoints remaining: %d"), wpRem-1)

				end

				AddCaption(wpMessage, 0xffba00ff, capgrpGameState)

			end

			wpActive[i] = true

		end

		if wpActive[i] == false then

			trackFinished = false

		end

	end

	return(trackFinished)

end

function AdjustScores()

	local bestTimeComment = loc("Did not finish")

	local newScore = false

	-- update this clan's time if the new track is better

	for i = 0, (numTeams-1) do

		if teamClan[i] == GetHogClan(CurrentHedgehog) then

			if trackTime < teamScore[i] then

				teamScore[i] = trackTime

				newScore = true

			else

				newScore = false

			end

		end

	end

	-- find the best time out of those so far

	for i = 0, (numTeams-1) do

		if teamScore[i] < bestTime then

			bestTime = teamScore[i]

			bestClan = teamClan[i]

		end

	end

	if bestTime ~= MAX_TURN_TIME then

		bestTimeComment = string.format(loc("%.1fs"), (bestTime/1000))

	end

	if newScore == true then

		if trackTime == bestTime then -- best time of the race

			ShowMission(loc("TechRacer"),

			loc("Track completed!"),

			string.format(loc("New race record: %.1fs"), (trackTime/1000)) .. "|" ..

			string.format(loc("Winning time: %s"), bestTimeComment), 0, 4000)

			PlaySound(sndHomerun)

		else    -- best time for the clan

			ShowMission(loc("TechRacer"),

			loc("Track completed!"),

			string.format(loc("New clan record: %.1fs"), (trackTime/1000)) .. "|" ..

			string.format(loc("Winning time: %s"), bestTimeComment), 4, 4000)

		end

	else -- not any kind of new score

		ShowMission(loc("TechRacer"),

			loc("Track completed!"),

			string.format(loc("Time: %.1fs"), (trackTime/1000)) .. "|" ..

			string.format(loc("Winning time: %s"), bestTimeComment), -amSkip, 4000)

		PlaySound(sndHellish)

	end

	for i = 0, (TeamsCount-1) do

		if teamNameArr[i] ~= " " and teamScore[i] ~= MAX_TURN_TIME then

			SetTeamLabel(teamNameArr[i], string.format(loc("%.1fs"), teamScore[i]/1000))

		end

	end

	if bestTime == trackTime then

		fastColour = GetClanColor(GetHogClan(CurrentHedgehog))

		for i = 0, (currCount-1) do

			fastX[i] = currX[i]

			fastY[i] = currY[i]

		end

		fastCount = currCount

		fastIndex = 0

	else

		currCount = 0

		fastIndex = 0

	end

end

function onNewRound()

	roundNumber = roundNumber + 1

	local totalComment = ""

	for i = 0, (TeamsCount-1) do

		if teamNameArr[i] ~= " " then

			if teamScore[i] ~= MAX_TURN_TIME then

				teamComment[i] = string.format(loc("%s: %.1fs"), teamNameArr[i], (teamScore[i]/1000)) .. "|"

			else

				teamComment[i] = string.format(loc("%s: Did not finish"), teamNameArr[i]) .. "|"

			end

			totalComment = totalComment .. teamComment[i]

		elseif teamNameArr[i] == " " then

			teamComment[i] = "|"

		end

	end

	local icon

	if roundNumber >= roundLimit then

		icon = 0

	else

		icon = 2

	end

	ShowMission(loc("TechRacer"),

		loc("Status update"),

		string.format(loc("Rounds complete: %d/%d"), roundNumber, roundLimit) .. "| |" ..

		loc("Best team times: ") .. "|" ..

		totalComment,

		icon, 4000)

	-- end game if its at round limit

	if roundNumber >= roundLimit then

		-- Sort the scores for the ranking list

		local unfinishedArray = {}

		local sortedTeams = {}

		local k = 1

		local c = 1

		local clanScores = {}

		local previousClan

		for i = 0, TeamsCount-1 do

			local clan = GetTeamClan(teamNameArr[i])

			if not clanScores[clan+1] then

				clanScores[clan+1] = {}

				clanScores[clan+1].index = clan

				clanScores[clan+1].score = teamScore[i]

			end

			if teamScore[i] ~= MAX_TURN_TIME and teamNameArr[i] ~= " " then

				sortedTeams[k] = {}

				sortedTeams[k].name = teamNameArr[i]

				sortedTeams[k].score = teamScore[i]

				sortedTeams[k].clan = clan

				k = k + 1

			else

				table.insert(unfinishedArray, string.format(loc("%s did not finish the race."), teamNameArr[i]))

			end

		end

		table.sort(sortedTeams, function(team1, team2)

			if team1.score == team2.score then

				return team1.clan < team2.clan

			else

				return team1.score < team2.score

			end

		end)

		table.sort(clanScores, function(clan1, clan2) return clan1.score < clan2.score end)

		local rank = 0

		local rankPlus = 0

		local prevScore

		local clanRanks = {}

		for c = 1, #clanScores do

			rankPlus = rankPlus + 1

			if clanScores[c].score ~= prevScore then

				rank = rank + rankPlus

				rankPlus = 0

			end

			prevScore = clanScores[c].score

			clanRanks[clanScores[c].index] = rank

		end

		-- Write all the stats!

		for i = 1, #sortedTeams do

			SendStat(siPointType, "!TIME")

			SendStat(siTeamRank, tostring(clanRanks[GetTeamClan(sortedTeams[i].name)]))

			SendStat(siPlayerKills, sortedTeams[i].score, sortedTeams[i].name)

		end

		local roundDraw = false

		if #clanScores >= 2 and clanScores[1].score == clanScores[2].score and clanScores[1].score ~= MAX_TURN_TIME then

			roundDraw = true

			SendStat(siGameResult, GetEngineString("TMsgStrId", sidDraw))

			SendStat(siCustomAchievement, loc("The teams are tied for the fastest time."))

		elseif #sortedTeams >= 1 then

			SendStat(siGameResult, formatEngineString(GetEngineString("TMsgStrId", sidWinner), sortedTeams[1].name))

			SendStat(siCustomAchievement, string.format(loc("%s wins with a best time of %.1fs."), sortedTeams[1].name, (sortedTeams[1].score/1000)))

			for i=1,#unfinishedArray do

				 SendStat(siCustomAchievement, unfinishedArray[i])

			end

		else

			roundDraw = true

			SendStat(siGameResult, GetEngineString("TMsgStrId", sidDraw))

			SendStat(siCustomAchievement, loc("Nobody managed to finish the race. What a shame!"))