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

-- HIGHLANDER / HOGS OF WAR

-- by mikade

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

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

-- ideas for the future

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

-- add structure

-- allow switcher, resurrector

-- nerf teleport

-- balance weapon distribution across entire team / all teams

-- add other inequalities/bonuses like... ???

HedgewarsScriptLoad("/Scripts/Locale.lua")

HedgewarsScriptLoad("/Scripts/Tracker.lua")

HedgewarsScriptLoad("/Scripts/Params.lua")

-- These define weps allowed by the script.

-- These were arbitrarily defined out-of-order in initial script, so that was preserved here, resulting 

-- in a moderately odd syntax.

local utilWeps =  {

-- Intentionally left out:

-- * Resurrector (guaranteed to screw up the game)

-- * Time Box

-- * Switch Hedgehog (not sure why)

local wepArray = {}

local atkChoices = {}

local utilChoices = {}

local currHog

local lastHog

local started = false

local switchStage = 0

local lastWep = amNothing

local shotsFired = 0

local probability = {1,2,5,10,20,50,200,500,1000000};

local atktot = 0

local utiltot = 0

local someHog = nil -- just for looking up the weps

-- Script parameter stuff

--[[ Loyal Highlander.

If true, killing hogs of your own clan doesn't give you their weapons.

Otherwise, killing any hog gives you their weapons. ]]

local loyal = false

--[[ Multiple weapon usages.

This is a bit tricky to explain.

First, remind yourselves that hogs can never hold more than 1 of the same ammo type.

This param changes how ammo will be restocked after killing a hog if you

already owned this ammo.

Basically this is about if you can use the same weapon multiple times in a

turn by killing enemies in a clever way.

We need to distinguish between your current inventory and the “reset inventory”,

that is, the state to which your inventory will get reset in the next turn.

No Multi-Use (default):

Multi-Use:

Example 1:

Example 2:

]]

local multiUse = false

function onParameters()

end

function CheckForWeaponSwap()

	if GetCurAmmoType() ~= lastWep then

		shotsFired = 0

	end

	lastWep = GetCurAmmoType()

end

function onSlot()

	CheckForWeaponSwap()

end

function onSetWeapon()

	CheckForWeaponSwap()

end

function onHogAttack()

	CheckForWeaponSwap()

	shotsFired = shotsFired + 1

end

function StartingSetUp(gear)

	end

	setGearValue(gear,amSkip,100)

end

	end

end

-- this is called when a hog dies

function TransferWeps(gear)

	if CurrentHedgehog ~= nil and CurrentHedgehog ~= gear and (not loyal or (GetHogClan(CurrentHedgehog) ~= GetHogClan(gear))) then

			val = getGearValue(gear,w)

			if val ~= 0 and (multiUse or (wepArray[w] ~= 9 and getGearValue(CurrentHedgehog, w) == 0))  then

				setGearValue(CurrentHedgehog, w, val)

				-- if you are using multi-shot weapon, gimme one more

				if (GetCurAmmoType() == w) and (shotsFired ~= 0) then

					AddAmmo(CurrentHedgehog, w, val+1)

				-- assign ammo as per normal

				else

					AddAmmo(CurrentHedgehog, w, val)

				end

			end

		end

	end

end

function onGameInit()

	EnableGameFlags(gfInfAttack, gfRandomOrder, gfPerHogAmmo)

	DisableGameFlags(gfResetWeps, gfSharedAmmo)

	HealthCaseProb = 100

	if loyal then

		Goals = loc("Loyal Highlander: Eliminate enemy hogs to take their weapons") .. "|"

	else

		Goals = loc("Highlander: Eliminate hogs to take their weapons") .. "|"

	end

	Goals = Goals .. loc("Replenishment: Weapons are restocked on turn start of a new hog") .. "|" ..

	loc("Ammo Limit: Hogs can’t have more than 1 ammo per type") .. "|"

	if multiUse then

		Goals = Goals .. loc("Multi-Use: You can take and use the same ammo type multiple times in a turn")

	else

		Goals = Goals .. loc("No Multi-Use: Once you used an ammo, you can’t take it again in this turn")

	end

end

function onGameStart()

	runOnGears(StartingSetUp)

	runOnGears(ConvertValues)

end

function CheckForHogSwitch()

	--[[ Restock the weapons of the hog on turn start, provided it is not the same hog as before.

	This exception is done do avoid a single hog receiving tons of weapons when it is the only unfrozen

	hog and takes consecutive turns. ]]

	if (CurrentHedgehog ~= nil) then

		currHog = CurrentHedgehog

		if currHog ~= lastHog then

			-- re-assign ammo to this guy, so that his entire ammo set will

			-- be visible during another player's turn

			if lastHog ~= nil then

				ConvertValues(lastHog)

			end

			-- give the new hog what he is supposed to have, too

			ConvertValues(CurrentHedgehog)

		end

		lastHog = currHog

	end

end

function onNewTurn()

	CheckForHogSwitch()

end

function onGearAdd(gear)

	if (GetGearType(gear) == gtHedgehog) then

		trackGear(gear)

	end

end

function onGearDelete(gear)

		TransferWeps(gear)

		trackDeletion(gear)

	end

end