--[[-----------------------------------------------------

-- CONSTRUCTION MODE --

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

A Hedgewars gameplay mode by mikade.

Special thanks to all who helped test and offered suggestions.

Additional thanks to sheepluva/nemo for adding some extra hooks.

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

-- STRUCTURES LIST --

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

* Healing Station: Heals hogs to 150 health

* Teleportation Node: Allows teleporting to any other teleporter nodes

* Bio-filter: Explodes enemy hogs

* Respawner: If you have one of these, any slain hogs are resurrected here

* Generator: Generates power (used to buy stuff)

* Support Station: Allows purchasing crates

* Construction Station: Allows purchasing girders, rubber, mines, sticky mines, barrels

* Reflector Shield: Reflect projectiles

* Weapon Filter: Removes all equipement of enemy hogs passing through this area

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

-- SCRIPT PARAMETER

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

The script parameter can be used to configure the energy

of the game. It is a comma-seperated list of key=value pairs, where each

key is a word and each value is an integer between 0 and 4294967295.

Possible keys:

* initialenergy:  Amount of energy that each team starts with (default: 550)

                  Note: Must be smaller than or equal to maxenergy

* energyperround: Amount of energy that each team gets per round (default: 50)

* maxenergy:      Maximum amount of energy each team can hold (default: 1000)

* cratesperround: Maximum number of crates you can place per round (default: 5)

For the previous 2 keys, you can use the value “inf” for an unlimited amount.

Example: “initialenergy=750, maxenergy=2000” starts thee game with 750 energy

         and sets the maximum energy to 2000.

Example: “craterperround=inf” disables the crate placement limit.

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

-- Ideas list --

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

* To make the weapon filter more attractive, make it vaporize flying saucers

  and also rope, and maybe incoming gears

* Make healing thing also cure poison?

* Maybe make poison more virulent and dangerous

]]

HedgewarsScriptLoad("/Scripts/Locale.lua")

HedgewarsScriptLoad("/Scripts/Tracker.lua")

HedgewarsScriptLoad("/Scripts/Params.lua")

-- Structures stuff

local strucID = {}

local strucGear = {}

local strucClan = {}

local strucType = {}

local strucCost = {}

local strucHealth = {}

local strucCirc = {}

local strucCircCol = {}

local strucCircRadius = {}

local strucCircType = {}

local strucAltDisplay = {}

-- Clan stuff

local clanPower = {} -- current power for each clan. Used to build stuff

local clanPowerTag = nil -- visual gear ID of displayed clan power

local clanUsedExtraTime = {} -- has used extra time in this round?

local clanCratesSpawned = {} -- number of crates spawned in this round

local clanFirstTurn = {}

local clanBoundsSX = {}

local clanBoundsSY = {}

local clanBoundsEX = {}

local clanBoundsEY = {}

-- For tracking previous mode selection per-team

local teamLStructIndex = {}

local teamLObjectMode = {}

local teamLCrateMode = {}

local teamLMineIndex = {}

local teamLWeapIndex = {}

local teamLUtilIndex = {}

-- Wall stuff

local wallsVisible = false

local wX = {}

local wY = {}

local wWidth = {}

local wHeight = {}

local wCol = {}

local wMargin = 20

local borderEffectTimer = 0 -- timer for border clan sparkles

-- Other stuff

local placedExpense = 0 -- Cost of current selected thing

local curWep = amNothing -- current weapon, used to reduce # of calls to GetCurAmmoType()

local fortMode = false -- is using a fort map?

local tempID_CheckProximity = nil -- temporary structure variable for CheckProximity

local cGear = nil -- detects placement of girders and objects (using airattack)

local uniqueStructureID = 0 -- Counter and ID for structures. Is incremented each time a structure spawns

--[[ Hacky workaround for object placer: Since this thing is

based on the drill strike, it allows the 5 timer keys to be

pressed, causing the announcer to show up.

This variable counts the number of ticks to count down until

to overwrite this anouncer message.

-1 means “do nothing”. ]]

local checkForSpecialWeaponsIn = -1

local lastWep = amNothing -- helper variable to track the previous hack

-- Colors

local colorRed = 0xff0000ff

local colorGreen = 0x00ff00ff

-- Fake ammo types, for the overwritten weapons in Construction Mode

local amCMStructurePlacer = amAirAttack

local amCMCratePlacer = amNapalm

local amCMObjectPlacer = amDrillStrike

-- Config variables (script parameter)

local conf_initialEnergy = 550

local conf_energyPerRound = 50

local conf_maxEnergy = 1000

local conf_cratesPerRound = 5

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

-- CRATE DEFINITIONS --

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

-- format:

local costFactor = 20

-- Weapons which shouldn't be aded:

-- Air attack, napalm, drillstrike: Overwritten weapons for the Construction Mode tools

local atkArray = {

}

-- UTILITY CRATES --

-- Utilities which shouldn't be added:

-- * Teleport: We have teleportation node

-- * Switch: Always infinite

-- * Girder, rubber: Requires construction station

-- * Resurrector: We have the resurrector structure for this

-- Utilities which might be weird for this mode:

-- * Tardis: Randomly teleports hog, maybe even into enemy clan's area

local utilArray = {

}

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

-- Placement stuff

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

-- primary placement categories

local cIndex = 1 -- category index

local cat = {

}

local catReverse = {}

for c=1, #cat do

	catReverse[cat[c]] = c

end

local sProx = {

}

local pMode = {}	-- pMode contains custom subsets of the main categories

local pIndex = 1

local currentGirderRotation = 1 -- current girder rotation, we actually need this as HW remembers what rotation you last used

function DrawClanPowerTag()

	local zoomL = 1.1

	local xOffset = 45

	local yOffset = 70

	local tValue = clanPower[GetHogClan(CurrentHedgehog)]

	local tCol = 0x00ff00ff

	DeleteVisualGear(clanPowerTag)

	clanPowerTag = AddVisualGear(-div(ScreenWidth, 2) + xOffset, ScreenHeight - yOffset, vgtHealthTag, tValue, false)

	SetVisualGearValues(

		clanPowerTag, 	-- id

		nil,		-- x offset (set above)

		nil,		-- y offset (set above)

		0, 		-- dx

		0, 		-- dy

		zoomL, 		-- zoom

		1, 		-- ~= 0 means align to screen

		nil, 		-- frameticks

		nil, 		-- value (set above)

		240000, 	-- timer

		tCol		-- color.   -- GetClanColor( GetHogClan(CurrentHedgehog) )

	)

end

function XYisInRect(px, py, psx, psy, pex, pey)

	if (px > psx) and (px < pex) and (py > psy) and (py < pey) then

		return(true)

	else

		return(false)

	end

end

function AddWall(zXMin, zYMin, zWidth, zHeight, zCol)

	table.insert(wX, zXMin)

	table.insert(wY, zYMin)

	table.insert(wWidth, zWidth)

	table.insert(wHeight, zHeight)

	table.insert(wCol, zCol)

end

function BorderSpark(zXMin,zYMin, zWidth, zHeight, bCol)

	local eX = zXMin + GetRandom(zWidth+10)

	local eY = zYMin + GetRandom(zHeight+10)

	local tempE = AddVisualGear(eX, eY, vgtDust, 0, false)

	if tempE ~= 0 then

		SetVisualGearValues(tempE, eX, eY, nil, nil, nil, nil, nil, 1, nil, bCol)

	end

end

function HandleBorderEffects()

	borderEffectTimer = borderEffectTimer + 1

	if borderEffectTimer > 15 then

		borderEffectTimer = 1

		for i = 1, #wX do

			BorderSpark(wX[i],wY[i],wWidth[i],wHeight[i], wCol[i])

		end

	end

end

----

-- old reflecting stuff from like 3 years ago lol

---

function gearCanBeDeflected(gear)

	if 	(GetGearType(gear) == gtShell) or

		(GetGearType(gear) == gtGrenade) or

		(GetGearType(gear) == gtAirBomb) or

		(GetGearType(gear) == gtClusterBomb) or

		(GetGearType(gear) == gtCluster) or

		(GetGearType(gear) == gtGasBomb) or

		(GetGearType(gear) == gtMine) or

		(GetGearType(gear) == gtMortar) or

		(GetGearType(gear) == gtHellishBomb) or

		(GetGearType(gear) == gtWatermelon) or

		(GetGearType(gear) == gtMelonPiece)	or

		(GetGearType(gear) == gtEgg) or

		(GetGearType(gear) == gtDrill) or

		(GetGearType(gear) == gtBall) or

		(GetGearType(gear) == gtExplosives) or

		(GetGearType(gear) == gtFlame) or

		(GetGearType(gear) == gtPortal) or

		(GetGearType(gear) == gtDynamite) or

		(GetGearType(gear) == gtSMine) or

		(GetGearType(gear) == gtKnife) or

		(GetGearType(gear) == gtJetpack) or

		(GetGearType(gear) == gtBirdy) or

		(GetGearType(gear) == gtSnowball) or

		(GetGearType(gear) == gtMolotov)

	then

		return(true)

	else

		return(false)

	end

end

function getThreatDamage(gear)

	local dmg

	--- damage amounts for weapons

	if 	(GetGearType(gear) == gtGrenade) or

		(GetGearType(gear) == gtClusterBomb) or

		(GetGearType(gear) == gtGasBomb) or

		(GetGearType(gear) == gtKnife) or

		(GetGearType(gear) == gtEgg) or

		(GetGearType(gear) == gtMolotov) or

		(GetGearType(gear) == gtHellishBomb) or

		(GetGearType(gear) == gtWatermelon) or

		(GetGearType(gear) == gtSMine) then

		dmg = 30

	elseif (GetGearType(gear) == gtMelonPiece) then

		dmg = 40

	elseif (GetGearType(gear) == gtAirBomb) or

			(GetGearType(gear) == gtDrill) or

			(GetGearType(gear) == gtMine) or

			(GetGearType(gear) == gtCluster) then

		dmg = 20

	elseif (GetGearType(gear) == gtFlame) or

			(GetGearType(gear) == gtPortal) or

			(GetGearType(gear) == gtDynamite) then

		dmg = 0

	elseif (GetGearType(gear) == gtBall) then

		dmg = 1

	else	-- normal shell, snowball etc

		dmg = 65

	end

	return(dmg)

end

function setGearReflectionValues(gear)

	local dmg = getThreatDamage(gear)

	setGearValue(gear,"damage",dmg)

	setGearValue(gear,"deflects",0)

	if (CurrentHedgehog ~= nil) then

		setGearValue(gear,"owner",GetHogClan(CurrentHedgehog)) -- NEW NEEDS CHANGE?

	else

		setGearValue(gear,"owner",10)

	end

end

function isATrackedGear(gear)

	if 	(GetGearType(gear) == gtHedgehog) or

		(GetGearType(gear) == gtTarget) or

		(GetGearType(gear) == gtCase)

	then

		return(true)

	else

		return(false)

	end

end

function AddStruc(pX,pY, pType, pClan)

	uniqueStructureID = uniqueStructureID + 1

	local tempG = AddGear(0, 0, gtTarget, 0, 0, 0, 0)

	SetGearPosition(tempG, pX, pY)

	setGearValue(tempG, "uniqueStructureID", uniqueStructureID)

	local tempCirc = AddVisualGear(0,0,vgtCircle,0,true)

	SetVisualGearValues(tempCirc, 0, 0, 100, 255, 1, 100, 0, 500, 1, 0xFFFFFF00)

	table.insert(strucID, uniqueStructureID)

	table.insert(strucType, pType)

	table.insert(strucGear,tempG)

	table.insert(strucClan,pClan)

	table.insert(strucCost,2)

	local frameID = 0

	local visualSprite = sprTarget

	local madness = AddVisualGear(GetX(tempG), GetY(tempG), vgtStraightShot, 1, true,1)

		table.insert(strucHealth,255)

	else

		table.insert(strucHealth,1)

	end

	table.insert(strucCirc,tempCirc)

	table.insert(strucCircType, 1)

		table.insert(strucCircCol,colorRed)

		table.insert(strucCircRadius,1000)

		frameID = 7

		table.insert(strucCircCol,0xFF00FF00)

		table.insert(strucCircRadius,500)

		frameID = 3

		table.insert(strucCircCol,0xFF00FF00)

		table.insert(strucCircRadius,75)

		runOnHogs(EnableHogResurrectionForThisClan)

		frameID = 1

		table.insert(strucCircCol,0x0000FFFF)

		table.insert(strucCircRadius,350)

		frameID = 6

		table.insert(strucCircCol,0xFFFF00FF)