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

-- 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 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 colorClanTag = 0x00ff00ff

local colorSupportStation = 0xFFFF00FF

local colorConstructionStation = 0xFFFFFFFF

local colorTeleportationNode = 0x0000FFFF

local colorHealingStation = 0xFF808040 -- Just a slight glow

local colorBioFilter = 0xFF0000FF

local colorReflectorShield = 0xFFAE00FF

local colorWeaponFilter =  0xA800FFFF

local colorHealingStationParticle = 0x00FF00FF

local colorMessageError = 0xFFFFFFFF

-- 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, cost }

local costFactor = 20

-- WEAPON CRATES

-- Weapons which shouldn't be aded:

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

-- Mine strike: Is currently broken

-- Piano strike: Hog is resurrected by respawner. Not strictly prohibited, however.

local atkArray = {

	{amBazooka,	 2*costFactor},

	--{amBee,	 4*costFactor},

	{amMortar,	 1*costFactor},

	{amDrill,	 3*costFactor},

	{amSnowball,	 3*costFactor},

	{amGrenade,	 2*costFactor},

	{amClusterBomb,	 3*costFactor},

	{amWatermelon,	25*costFactor},

	{amHellishBomb,	25*costFactor},

	{amMolotov,	 3*costFactor},

	{amGasBomb,	 3*costFactor},

	{amShotgun,	 2*costFactor},

	{amDEagle,	 2*costFactor},

	{amSniperRifle,	 3*costFactor},

	--{amSineGun,	 6*costFactor},

	{amFlamethrower, 4*costFactor},

	{amIceGun,	15*costFactor},

	{amFirePunch,	 3*costFactor},

	{amWhip,	 1*costFactor},

	{amBaseballBat,	 7*costFactor},

	--{amKamikaze,	 1*costFactor},

	{amSeduction,	 1*costFactor},

	{amHammer,	 1*costFactor},

	{amMine,	 1*costFactor},

	{amDynamite,	 9*costFactor},

	{amCake,	25*costFactor},

	{amBallgun,	40*costFactor},

	--{amRCPlane,	25*costFactor},

	{amSMine,	 5*costFactor},

	--{amMineStrike,15*costFactor},

	--{amPiano,	40*costFactor},

	{amPickHammer,	 2*costFactor},

	{amBlowTorch,	 4*costFactor},

	{amKnife,	 2*costFactor},

	{amBirdy,	 7*costFactor},

}

-- UTILITY CRATES --

-- Utilities which shouldn't be added:

-- * Teleport: We have teleportation node

-- * Switch: Infinite in default Construction Mode weapons scheme

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

 	{amLandGun,	 5*costFactor},

	{amRope,	 7*costFactor},

	{amParachute,	 2*costFactor},

	{amJetpack,	 8*costFactor},

	{amPortalGun,	 15*costFactor},

	{amInvulnerable, 5*costFactor},

	{amLaserSight,	 2*costFactor},

	{amVampiric,	 6*costFactor},

	{amLowGravity,	 4*costFactor},

	{amExtraDamage,	 6*costFactor},

	{amExtraTime,	 8*costFactor}

}

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

-- Placement stuff

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

-- primary placement categories

local cIndex = 1 -- category index

local cat = {

	loc_noop("Girder Placement Mode"),

	loc_noop("Rubber Placement Mode"),

	loc_noop("Mine Placement Mode"),

	loc_noop("Sticky Mine Placement Mode"),

	loc_noop("Barrel Placement Mode"),

	loc_noop("Weapon Crate Placement Mode"),

	loc_noop("Utility Crate Placement Mode"),

	loc_noop("Health Crate Placement Mode"),

	loc_noop("Structure Placement Mode"),

}

local catReverse = {}

for c=1, #cat do

	catReverse[cat[c]] = c

end

-- Track girders in proximity of CurrentHedgehog

local sProx = {

	["Girder Placement Mode"] = false,

	["Rubber Placement Mode"] = false,

	["Mine Placement Mode"] = false,

	["Sticky Mine Placement Mode"] = false,

	["Barrel Placement Mode"] = false,

	["Weapon Crate Placement Mode"] = false,

	["Utility Crate Placement Mode"] = false,

	["Health Crate Placement Mode"] = false,

	["Structure Placement Mode"] = false,

	["Teleportation Mode"] = false,

}

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

	-- alternatively:  tCol = GetClanColor(GetHogClan(CurrentHedgehog))

	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

	)

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)

	if pType ~= "Respawner" and pType ~= "Generator" then

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

		table.insert(strucCirc, tempCirc)

	else

		table.insert(strucCirc, false)

	end

	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)

	if pType == "Reflector Shield" then

		table.insert(strucHealth,255)

	else

		table.insert(strucHealth,1)

	end

	if pType == "Bio-Filter" then

		table.insert(strucCircCol, colorBioFilter)

		table.insert(strucCircRadius,1000)