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

-- 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")

local strucID = {}

local strucGear = {}

local strucClan = {}

local strucType = {}

local strucCost = {}

local strucHealth = {}

local strucCirc = {}

local strucCircCol = {}

local strucCircRadius = {}

local strucCircType = {}

local strucAltDisplay = {}

local clanBoundsSX = {}

local clanBoundsSY = {}

local clanBoundsEX = {}

local clanBoundsEY = {}

local teamLStructIndex = {}

local teamLObjectMode = {}

local teamLCrateMode = {}

local teamLMineIndex = {}

local teamLWeapIndex = {}

local teamLUtilIndex = {}

local wallsVisible = false

local wX = {}

local wY = {}

local wWidth = {}

local wHeight = {}

local wCol = {}

-- 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:

-- { ammoType, ammoTypeString, unused, cost }

-- WEAPON CRATES 

-- 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 = {

	"Girder Placement Mode",

	"Rubber Placement Mode",

	"Mine Placement Mode",

	"Sticky Mine Placement Mode",

	"Barrel Placement Mode",

	"Weapon Crate Placement Mode",

	"Utility Crate Placement Mode",

	"Health Crate Placement Mode",

	"Structure Placement Mode"

}

local catReverse = {}

for c=1, #cat do

	catReverse[cat[c]] = c

end

local sProx = {

	{loc("Girder Placement Mode"),false},

	{loc("Rubber Placement Mode"),false},

	{loc("Mine Placement Mode"),false},

	{loc("Sticky Mine Placement Mode"),false},

	{loc("Barrel Placement Mode"),false},

	{loc("Weapon Crate Placement Mode"),false},

	{loc("Utility Crate Placement Mode"),false},

	{loc("Health Crate Placement Mode"),false},

	{loc("Structure Placement Mode"),false},

	{loc("Teleportation Mode"),false},

}

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

local pIndex = 1

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()

		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)

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

	SetGearPosition(tempG, pX, pY)

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

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

	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)

	if pType == loc("Reflector Shield") then

		table.insert(strucHealth,255)

	else

		table.insert(strucHealth,1)

	end

	table.insert(strucCirc,tempCirc)

	table.insert(strucCircType, 1)

	if pType == loc("Bio-Filter") then

		table.insert(strucCircCol,colorRed)

		table.insert(strucCircRadius,1000)

		frameID = 7

	elseif pType == loc("Healing Station") then

		table.insert(strucCircCol,0xFF00FF00)

		table.insert(strucCircRadius,500)

		frameID = 3

	elseif pType == loc("Respawner") then

		table.insert(strucCircCol,0xFF00FF00)

		table.insert(strucCircRadius,75)

		runOnHogs(EnableHogResurrectionForThisClan)

		frameID = 1

	elseif pType == loc("Teleportation Node") then

		table.insert(strucCircCol,0x0000FFFF)

		table.insert(strucCircRadius,350)

		frameID = 6

	elseif pType == loc("Generator") then

		table.insert(strucCircCol,0xFFFF00FF)

		table.insert(strucCircRadius,75)

		setGearValue(tempG, "power", 0)