------------------- ABOUT ----------------------

--

-- In this cold planet hero seeks for a part of the

-- antigravity device. He has to capture Thanta who

-- knows where the device is hidden. Hero will be

-- able to use only the ice gun for this mission.

HedgewarsScriptLoad("/Scripts/Locale.lua")

HedgewarsScriptLoad("/Scripts/Animate.lua")

HedgewarsScriptLoad("/Missions/Campaign/A_Space_Adventure/global_functions.lua")

----------------- VARIABLES --------------------

-- globals

local missionName = loc("A frozen adventure")

local heroAtAntiFlyArea = false

local heroVisitedAntiFlyArea = false

local heroAtFinalStep = false

local iceGunTaken = false

local checkPointReached = 1 -- 1 is normal spawn

local heroDamageAtCurrentTurn = 0

-- dialogs

local dialog01 = {}

local dialog02 = {}

-- mission objectives

local goToThantaString = loc("Go to Thanta and get the device part!")

local goals = {

	[dialog01] = {missionName, loc("Getting ready"), loc("Collect the freezer and get the device part from Thanta.") .. "|" .. loc("Mines time: 0 seconds"), 1, 4500},

	["checkpoint"] = {missionName, loc("Objectives"), goToThantaString .. "|" .. loc("Mines time: 0 seconds"), 1, 4500},

}

-- crates

local icegunY = 1950

local icegunX = 260

-- hogs

local hero = {}

local ally = {}

local bandit1 = {}

local bandit2 = {}

local bandit3 = {}

local bandit4 = {}

local bandit5 = {}

-- teams

local teamA = {}

local teamB = {}

local teamC = {}

-- hedgehogs values

hero.name = loc("Hog Solo")

hero.x = 340

hero.y = 1840

hero.dead = false

ally.name = loc("Paul McHoggy")

ally.x = 300

ally.y = 1840

bandit1.name = loc("Thanta")

bandit1.x = 3240

bandit1.y = 1280

bandit1.dead = false

bandit1.frozen = false

bandit1.roundsToUnfreeze = 0

bandit2.name = loc("Billy Frost")

bandit2.x = 1480

bandit2.y = 1990

bandit3.name = loc("Ice Jake")

bandit3.x = 1860

bandit3.y = 1150

bandit4.name = loc("John Snow")

bandit4.x = 3200

bandit4.y = 970

bandit4.frozen = false

bandit4.roundsToUnfreeze = 0

bandit5.name = loc("White Tee")

bandit5.x = 3280

bandit5.y = 600

bandit5.frozen = false

bandit5.roundsToUnfreeze = 0

teamA.name = loc("Allies")

teamA.color = -6

teamB.name = loc("Frozen Bandits")

teamB.color = -2

teamC.name = loc("Hog Solo")

teamC.color = -6

-------------- LuaAPI EVENT HANDLERS ------------------

function onGameInit()

	Seed = 1

	TurnTime = 25000

	CaseFreq = 0

	MinesNum = 0

	MinesTime = 1

	Explosives = 0

	Map = "ice01_map"

	Theme = "Snow"

	-- Disable Sudden Death

	WaterRise = 0

	HealthDecrease = 0

	-- get the check point

	checkPointReached = initCheckpoint("ice01")

	-- get hero health

	local heroHealth = 100

	if tonumber(GetCampaignVar("HeroHealth")) then

		heroHealth = tonumber(GetCampaignVar("HeroHealth"))

	end

	if heroHealth ~= 100 then

		heroHealth = heroHealth + 5

		if heroHealth > 100 then

			heroHealth = 100

		end

		SaveCampaignVar("HeroHealth", heroHealth)

	end

	-- Hero

	teamC.name = AddMissionTeam(teamC.color)

	hero.gear = AddMissionHog(heroHealth)

	hero.name = GetHogName(hero.gear)

	AnimSetGearPosition(hero.gear, hero.x, hero.y)

	HogTurnLeft(hero.gear, true)

	-- Ally

	teamA.name = AddTeam(teamA.name, teamA.color, "heart", "Island", "Default", "cm_face")

	SetTeamPassive(teamA.name, true)

	ally.gear = AddHog(ally.name, 0, 100, "war_airwarden02")

	AnimSetGearPosition(ally.gear, ally.x, ally.y)

	-- Frozen Bandits

	teamB.name = AddTeam(teamB.name, teamB.color, "plant2", "Island", "Default", "cm_pirate")

	bandit1.gear = AddHog(bandit1.name, 1, 120, "Santa")

	AnimSetGearPosition(bandit1.gear, bandit1.x, bandit1.y)

	HogTurnLeft(bandit1.gear, true)

	bandit2.gear = AddHog(bandit2.name, 1, 100, "ushanka")

	AnimSetGearPosition(bandit2.gear, bandit2.x, bandit2.y)

	bandit3.gear = AddHog(bandit3.name, 1, 100, "thug")

	AnimSetGearPosition(bandit3.gear, bandit3.x, bandit3.y)

	bandit4.gear = AddHog(bandit4.name, 1, 40, "tophats")

	AnimSetGearPosition(bandit4.gear, bandit4.x, bandit4.y)

	HogTurnLeft(bandit4.gear, true)

	bandit5.gear = AddHog(bandit5.name, 1, 40, "Sniper")

	AnimSetGearPosition(bandit5.gear, bandit5.x, bandit5.y)

	HogTurnLeft(bandit5.gear, true)

	if checkPointReached == 1 then

		-- Start of the game

	elseif checkPointReached == 2 then

		iceGunTaken = true

		AnimSetGearPosition(hero.gear, 840, 1650)

	elseif checkPointReached == 3 then

		iceGunTaken = true

		heroAtFinalStep = true

		heroVisitedAntiFlyArea = true

		AnimSetGearPosition(hero.gear, 1450, 910)

	end

	AnimInit(checkPointReached == 1)

	AnimationSetup()

end

function onGameStart()

	AnimWait(hero.gear, 3000)

	FollowGear(hero.gear)

	-- Add mines

	AddGear(1612, 940, gtMine, 0, 0, 0, 0)

	AddGear(1622, 945, gtMine, 0, 0, 0, 0)

	AddGear(1645, 950, gtMine, 0, 0, 0, 0)

	AddGear(1655, 960, gtMine, 0, 0, 0, 0)

	AddGear(1665, 965, gtMine, 0, 0, 0, 0)

	AddGear(1800, 1000, gtMine, 0, 0, 0, 0)

	AddGear(1810, 1005, gtMine, 0, 0, 0, 0)

	AddGear(1820, 1010, gtMine, 0, 0, 0, 0)

	AddGear(1830, 1015, gtMine, 0, 0, 0, 0)

	AddGear(1840, 1020, gtMine, 0, 0, 0, 0)

	AddGear(1900, 1020, gtMine, 0, 0, 0, 0)

	AddGear(1910, 1020, gtMine, 0, 0, 0, 0)

	AddGear(1920, 1020, gtMine, 0, 0, 0, 0)

	AddGear(1930, 1030, gtMine, 0, 0, 0, 0)

	AddGear(1940, 1040, gtMine, 0, 0, 0, 0)

	AddGear(2130, 1110, gtMine, 0, 0, 0, 0)

	AddGear(2140, 1120, gtMine, 0, 0, 0, 0)

	AddGear(2180, 1120, gtMine, 0, 0, 0, 0)

	AddGear(2200, 1130, gtMine, 0, 0, 0, 0)

	AddGear(2210, 1130, gtMine, 0, 0, 0, 0)

	local x=2300

	local step=0

	while x<3100 do

		AddGear(x, 1150, gtMine, 0, 0, 0, 0)

		step = step + 1

		if step == 5 then

			step = 0

			x = x + GetRandom(201)+100

		else

			x = x + GetRandom(21)+10

		end

	end

	AddEvent(onHeroDeath, {hero.gear}, heroDeath, {hero.gear}, 0)

	AddEvent(onHeroFinalStep, {hero.gear}, heroFinalStep, {hero.gear}, 0)

	AddEvent(onAntiFlyArea, {hero.gear}, antiFlyArea, {hero.gear}, 1)

	AddEvent(onAntiFlyAreaVelocity, {hero.gear}, antiFlyAreaVelocity, {hero.gear}, 1)

	AddEvent(onNonAntiFlyArea, {hero.gear}, nonAntiFlyArea, {hero.gear}, 1)

	AddEvent(onThantaDeath, {bandit1.gear}, thantaDeath, {bandit1.gear}, 0)

	AddEvent(onHeroWin, {hero.gear}, heroWin, {hero.gear}, 0)

	AddAmmo(hero.gear, amJetpack, 99)

	AddAmmo(bandit1.gear, amBazooka, 5)

	AddAmmo(bandit2.gear, amBazooka, 4)

	AddAmmo(bandit3.gear, amMine, 2)

	AddAmmo(bandit3.gear, amGrenade, 3)

	AddAmmo(bandit4.gear, amBazooka, 5)

	AddAmmo(bandit5.gear, amBazooka, 5)

	if checkPointReached == 1 then

		AddAmmo(hero.gear, amBazooka, 1)

		SpawnSupplyCrate(icegunX, icegunY, amIceGun)

		AddEvent(onColumnCheckPoint, {hero.gear}, columnCheckPoint, {hero.gear}, 0)

		AddEvent(onHeroAtIceGun, {hero.gear}, heroAtIceGun, {hero.gear}, 0)

		AddAnim(dialog01)

	elseif checkPointReached == 2 then

		local baz = tonumber(GetCampaignVar("HeroAmmoIce01Bazooka")) or 1

		AddAmmo(hero.gear, amBazooka, baz)

		AddAmmo(hero.gear, amIceGun, 8)

		AnimCaption(hero.gear, goToThantaString, 5000)

		ShowMission(unpack(goals["checkpoint"]))

	elseif checkPointReached == 3 then

		local baz = tonumber(GetCampaignVar("HeroAmmoIce01Bazooka")) or 0

		local ice = math.max(6, tonumber(GetCampaignVar("HeroAmmoIce01IceGun")) or 6)

		AddAmmo(hero.gear, amBazooka, baz)

		AddAmmo(hero.gear, amIceGun, ice)

		AnimCaption(hero.gear, goToThantaString, 5000)

		ShowMission(unpack(goals["checkpoint"]))

	end

	SendHealthStatsOff()

end

function getNextBandit(hog)

	local continue = true

	local startHog = hog

	while true do

		if (GetHealth(hog) and GetEffect(hog, heFrozen) < 256) and

		( (heroAtFinalStep and (hog == bandit1.gear or hog == bandit4.gear or hog == bandit5.gear)) or

		((not heroAtFinalStep) and (hog == bandit2.gear or hog == bandit3.gear)) ) then

			return hog

		end

		if hog == bandit1.gear then

			hog = bandit2.gear

		elseif hog == bandit2.gear then

			hog = bandit3.gear

		elseif hog == bandit3.gear then

			hog = bandit4.gear

		elseif hog == bandit4.gear then

			hog = bandit5.gear

		elseif hog == bandit5.gear then

			hog = bandit1.gear

		end

		if startHog == hog then

			return nil

		end

	end

end

function onNewTurn()

	heroDamageAtCurrentTurn = 0

	-- round has to start if hero goes near the column

	if not heroVisitedAntiFlyArea and CurrentHedgehog ~= hero.gear then

		SkipTurn()

	elseif not heroVisitedAntiFlyArea and CurrentHedgehog == hero.gear then

		SetTurnTimeLeft(MAX_TURN_TIME)

	elseif (GetHogTeamName(CurrentHedgehog) == teamB.name) then

		local nextBandit = getNextBandit(CurrentHedgehog)

		if nextBandit ~= nil then

			if CurrentHedgehog ~= nextBandit then

				AnimSwitchHog(nextBandit)

			end

		else

			SkipTurn()

		end

	end

	-- frozen hogs accounting

	if CurrentHedgehog == hero.gear and heroAtFinalStep and TurnTimeLeft > 0 then

		if bandit1.frozen then

			if bandit1.roundsToUnfreeze == 0 then

				SetEffect(bandit1.gear, heFrozen, 255)

				bandit1.frozen = false

			else

				bandit1.roundsToUnfreeze = bandit1.roundsToUnfreeze - 1

			end

		end

		if bandit4.frozen then

			if bandit4.roundsToUnfreeze == 0 then

				SetEffect(bandit4.gear, heFrozen, 255)

				bandit4.frozen = false

			else

				bandit4.roundsToUnfreeze = bandit4.roundsToUnfreeze - 1

			end

		end

		if bandit5.frozen then

			if bandit5.roundsToUnfreeze == 0 then

				SetEffect(bandit5.gear, heFrozen, 255)

				bandit5.frozen = false

			else

				bandit5.roundsToUnfreeze = bandit5.roundsToUnfreeze - 1

			end

		end

	else

		if bandit1.frozen then

			SetEffect(bandit1.gear, heFrozen, 9999999999)

		end

		if bandit4.frozen then

			SetEffect(bandit4.gear, heFrozen, 9999999999)

		end

		if bandit5.frozen then

			SetEffect(bandit5.gear, heFrozen, 9999999999)

		end

	end

end

function onGameTick()

	AnimUnWait()

	if ShowAnimation() == false then

		return

	end

	ExecuteAfterAnimations()

	CheckEvents()

	if GetEffect(bandit1.gear, heFrozen) > 256 and not bandit1.frozen then

		bandit1.frozen = true

		SetEffect(bandit1.gear, heFrozen, 9999999999)

		bandit1.roundsToUnfreeze = 1

	end

	if GetEffect(bandit4.gear, heFrozen) > 256 and not bandit4.frozen then

		bandit4.frozen = true

		SetEffect(bandit4.gear, heFrozen, 9999999999)

		bandit4.roundsToUnfreeze = 2

	end

	if GetEffect(bandit5.gear, heFrozen) > 256 and not bandit5.frozen then

		bandit5.frozen = true

		SetEffect(bandit5.gear, heFrozen, 9999999999)

		bandit5.roundsToUnfreeze = 2

	end

end

function onAmmoStoreInit()

	SetAmmo(amIceGun, 0, 0, 0, 8)

end

function onGearDelete(gear)

	if gear == hero.gear then

		hero.dead = true

	elseif gear == bandit1.gear then

		bandit1.dead = true

	end

end

	if GameTime > 3000 then

		SetAnimSkip(true)

	end

end

function onGearDamage(gear, damage)

	if gear == hero.gear then

		heroDamageAtCurrentTurn = heroDamageAtCurrentTurn + damage

	end

end

-------------- EVENTS ------------------

function onAntiFlyArea(gear)

	if not hero.dead and (GetX(gear) > 860 or GetY(gear) < 1400) and not heroAtAntiFlyArea then

		return true

	end

	return false

end

function onAntiFlyAreaVelocity(gear)

	if not hero.dead and GetY(gear) < 1300 and GetX(gear) < 1190 then

		return true

	end

	return false

end

function onNonAntiFlyArea(gear)

	if not hero.dead and (GetX(gear) < 860 and GetY(gear) > 1400) and heroAtAntiFlyArea then

		return true

	end

	return false

end

function onHeroDeath(gear)

	if hero.dead then

		return true

	end

	return false

end

function onHeroFinalStep(gear)

	if not hero.dead and GetY(gear) < 960 and GetX(gear) > 1400 then

		return true

	end

	return false

end

function onColumnCheckPoint(gear)

	if not hero.dead and iceGunTaken and GetX(gear) < 870 and GetX(gear) > 850 and GetY(gear) > 1500 and StoppedGear(gear) then

		return true

	end

	return false

end

function onHeroAtIceGun(gear)

	if not hero.dead and GetX(gear) < icegunX+15 and GetX(gear) > icegunX-15 and GetY(gear) > icegunY-15 and GetY(gear) < icegunY+15 then

		return true

	end

	return false

end

function onThantaDeath(gear)

	if bandit1.dead then

		return true

	end

	return false

end

function onHeroWin(gear)

	if (not hero.dead and not bandit1.dead) and heroDamageAtCurrentTurn == 0 and (GetX(hero.gear)>=GetX(bandit1.gear)-80

		and GetX(hero.gear)<=GetX(bandit1.gear)+80)	and (GetY(hero.gear)>=GetY(bandit1.gear)-30 and GetY(hero.gear)<=GetY(bandit1.gear)+30) then

		return true

	end

	return false

end

-------------- ACTIONS ------------------

function antiFlyArea(gear)

	heroAtAntiFlyArea = true

	if not heroVisitedAntiFlyArea then

		EndTurn(true)

		FollowGear(hero.gear)

		SetGearMessage(hero.gear, band(GetGearMessage(hero.gear), bnot(gmAllStoppable)))

		AnimSwitchHog(bandit1.gear)

		FollowGear(hero.gear)

		HogSay(hero.gear, loc("My flying saucer stopped working!"), SAY_THINK)

		EndTurn(true)

	end

	AddAmmo(hero.gear, amJetpack, 0)

	AddAmmo(hero.gear, amSkip, 100)

	heroVisitedAntiFlyArea = true

end

function antiFlyAreaVelocity(gear)

	dx, dy = GetGearVelocity(hero.gear)

	SetGearVelocity(hero.gear, dx, math.max(dy, 0))

end

function nonAntiFlyArea(gear)

	heroAtAntiFlyArea = false

	AddAmmo(hero.gear, amJetpack, 99)

end

function heroDeath(gear)

	SendStat(siGameResult, string.format(loc("%s lost, try again!"), hero.name))

	SendStat(siCustomAchievement, loc("To win the game you have to stand next to Thanta."))

	SendStat(siCustomAchievement, loc("Most of the time you'll be able to use the freezer only."))

	SendStat(siCustomAchievement, loc("Use the bazooka and the flying saucer to get the freezer."))

	sendSimpleTeamRankings({teamB.name, teamC.name, teamA.name})

	EndGame()

end

function heroFinalStep(gear)

	heroAtFinalStep = true

	saveCheckpoint("3")

	SaveCampaignVar("HeroHealth", GetHealth(hero.gear))

	SaveCampaignVar("HeroAmmoIce01IceGun", GetAmmoCount(hero.gear, amIceGun))

	SaveCampaignVar("HeroAmmoIce01Bazooka", GetAmmoCount(hero.gear, amBazooka))

	AddCaption(loc("Checkpoint reached!"), capcolDefault, capgrpMessage2)

end

function columnCheckPoint(gear)

	saveCheckpoint("2")

	SaveCampaignVar("HeroHealth", GetHealth(hero.gear))