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

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

-- Colors

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 = 0x00FF0080

local colorGeneratorParticle = 0xFFFF00FF

local colorMessageError = 0xFFFFFFFF

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

local amCMStructurePlacer = amAirAttack

local amCMCratePlacer = amMineStrike

local amCMObjectPlacer = amNapalm

-- Config variables (script parameter)

-- NOTE: If you change this, also change the default game scheme “Construction Mode”

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

	{amDuck,	 2*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},

	{amMinigun,	13*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},

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

}

-- Frames in sprTarget for the structure images

local catFrames = {

	["Respawner"] = 1,

	["Generator"] = 2,

	["Healing Station"] = 3,

	["Support Station"] = 4,

	["Weapon Filter"] = 5,

	["Teleportation Node"] = 6,

	["Bio-Filter"] = 7,

	["Construction Station"] = 8,

	["Reflector Shield"] = 9,

}

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

-- Returns true if ammoType is an ammo type with a special meaning

-- in Construction Mode.

function IsConstructionModeAmmo(ammoType)

	return ammoType == amCMStructurePlacer or

	ammoType == amCMObjectPlacer or

	ammoType == amCMCratePlacer or

	ammoType == amGirder or

	ammoType == amRubber or

	ammoType == amTeleport

end

function RenderClanPower()

	for i=0, TeamsCount-1 do

		local name = GetTeamName(i)

		SetTeamLabel(name, clanPower[GetTeamClan(name)])

	end

	DrawClanPowerTag()

end

function DrawClanPowerTag()

	local zoomL = 1.1

	local xOffset = 45

	local yOffset = 70

	local tValue = clanPower[GetHogClan(CurrentHedgehog)]

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

	DeleteVisualGear(clanPowerTag)

	clanPowerTag = nil

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)