--[[

  ########################################################################

  Name:      Battalion

  Made by:   Anachron 

  ########################################################################

]]--

--[[

  Readme:

  https://hedgewars.org/wiki/Battalion

  ########################################################################

]]--

--[[

  ########################################################################

  TODO / ideas list

  ########################################################################

  - Make Hogs sorted by rareness for teams with less hogs (more fair)

  - Keep first picked up unused crate utitlity until next round

  - Ship default scheme but let user overwrite it

  - Make SuddenDeathWaterRise dynamic

  - Make SuddenDeathTurns dynamic

  - Add Hog Variants like Crazy Scientist or Astronaut

  ########################################################################

]]--

--[[

  ##############################################################################

  ### GENERAL SCRIPT LOADING AND VARIABLE INITIALISATION                     ###

  ##############################################################################

]]--

HedgewarsScriptLoad("/Scripts/Locale.lua")

HedgewarsScriptLoad("/Scripts/Tracker.lua")

HedgewarsScriptLoad("/Scripts/Params.lua")

-- List of all hog variants with belonging weapons,

--  hitpoints, chances and more

local variants = {}

local varName = ""

local newLine = ""

local gmAny = 0xFFFFFFFF

--[[

  ##############################################################################

  ### VARIANT SETUP                                                          ###

  ##############################################################################

]]--

varName = "Pyromancer"

variants[varName] = {}

variants[varName]["name"] = loc("Pyromancer")

variants[varName]["chance"] = 7

variants[varName]["hat"] = "Gasmask"

variants[varName]["hp"] = 70

variants[varName]["hogLimit"] = 2

variants[varName]["weaponLimit"] = 1

variants[varName]["weapons"] = {amFlamethrower, amMolotov, amWhip}

variants[varName]["helperLimit"] = 1

variants[varName]["helpers"] = {amRope, amParachute}

variants[varName]["special"] = false

varName = "Builder"

variants[varName] = {}

variants[varName]["name"] = loc("Builder")

variants[varName]["chance"] = 10

variants[varName]["hat"] = "constructor"

variants[varName]["hp"] = 100

variants[varName]["hogLimit"] = 1

variants[varName]["weaponLimit"] = 1

variants[varName]["weapons"] = {amDynamite, amWhip, amHammer}

variants[varName]["helperLimit"] = 1

variants[varName]["helpers"] = {amGirder, amBlowTorch}

variants[varName]["special"] = false

varName = "Rifleman"

variants[varName] = {}

variants[varName]["name"] = loc("Rifleman")

variants[varName]["chance"] = 7

variants[varName]["hat"] = "Sniper"

variants[varName]["hp"] = 70

variants[varName]["hogLimit"] = 2

variants[varName]["weaponLimit"] = 1

variants[varName]["weapons"] = {amRCPlane, amShotgun, amSniperRifle}

variants[varName]["helperLimit"] = 1

variants[varName]["helpers"] = {amLowGravity, amParachute}

variants[varName]["special"] = false

varName = "Warrior"

variants[varName] = {}

variants[varName]["name"] = loc("Warrior")

variants[varName]["chance"] = 12

variants[varName]["hat"] = "spartan"

variants[varName]["hp"] = 120

variants[varName]["hogLimit"] = 2

variants[varName]["weaponLimit"] = 1

variants[varName]["weapons"] = {amClusterBomb, amGrenade, amBazooka}

variants[varName]["helperLimit"] = 1

variants[varName]["helpers"] = {amParachute, amRope}

variants[varName]["special"] = false

varName = "Chef"

variants[varName] = {}

variants[varName]["name"] = loc("Chef")

variants[varName]["chance"] = 7

variants[varName]["hat"] = "chef"

variants[varName]["hp"] = 70

variants[varName]["hogLimit"] = 1

variants[varName]["weaponLimit"] = 1

variants[varName]["weapons"] = {amCake, amKnife, amWhip}

variants[varName]["helperLimit"] = 1

variants[varName]["helpers"] = {amRubber, amParachute}

variants[varName]["special"] = false

varName = "Medic"

variants[varName] = {}

variants[varName]["name"] = loc("Medic")

variants[varName]["chance"] = 12

variants[varName]["hat"] = "war_desertmedic"

variants[varName]["hp"] = 120

variants[varName]["hogLimit"] = 1

variants[varName]["weaponLimit"] = 1

variants[varName]["weapons"] = {amResurrector, amMine, amGasBomb}

variants[varName]["helperLimit"] = 1

variants[varName]["helpers"] = {amTeleport, amParachute}

variants[varName]["special"] = false

varName = "Ninja"

variants[varName] = {}

variants[varName]["name"] = loc("Ninja")

variants[varName]["chance"] = 8

variants[varName]["hat"] = "NinjaTriangle"

variants[varName]["hp"] = 80

variants[varName]["hogLimit"] = 2

variants[varName]["weaponLimit"] = 1

variants[varName]["weapons"] = {amSMine, amMine, amFirePunch}

variants[varName]["helperLimit"] = 1

variants[varName]["helpers"] = {amRope, amParachute}

variants[varName]["special"] = false

varName = "Athlete"

variants[varName] = {}

variants[varName]["name"] = loc("Athlete")

variants[varName]["chance"] = 8

variants[varName]["hat"] = "footballhelmet"

variants[varName]["hp"] = 80

variants[varName]["hogLimit"] = 1

variants[varName]["weaponLimit"] = 1

variants[varName]["weapons"] = {amBaseballBat, amFirePunch, amSeduction}

variants[varName]["helperLimit"] = 1

variants[varName]["helpers"] = {amRope, amPickHammer}

variants[varName]["special"] = false

varName = "Scientist"

variants[varName] = {}

variants[varName]["name"] = loc("Scientist")

variants[varName]["chance"] = 7

variants[varName]["hat"] = "doctor"

variants[varName]["hp"] = 80

variants[varName]["hogLimit"] = 1

variants[varName]["weaponLimit"] = 1

variants[varName]["weapons"] = {amPortalGun, amSineGun, amIceGun}

variants[varName]["helperLimit"] = 1

variants[varName]["helpers"] = {amTeleport, amJetpack}  

variants[varName]["special"] = false

varName = "Air General"

variants[varName] = {}

variants[varName]["name"] = loc("Air General")

variants[varName]["chance"] = 5

variants[varName]["hat"] = "war_desertofficer"

variants[varName]["hp"] = 50

variants[varName]["hogLimit"] = 1

variants[varName]["weaponLimit"] = 1

variants[varName]["weapons"] = {amMineStrike, amNapalm, amAirAttack}

variants[varName]["helperLimit"] = 1

variants[varName]["helpers"] = {amRope, amParachute}

variants[varName]["special"] = true

varName = "Hunter"

variants[varName] = {}

variants[varName]["name"] = loc("Hunter")

variants[varName]["chance"] = 10

variants[varName]["hat"] = "Skull"

variants[varName]["hp"] = 100

variants[varName]["hogLimit"] = 1

variants[varName]["weaponLimit"] = 1

variants[varName]["weapons"] = {amBee, amMortar, amDrill}

variants[varName]["helperLimit"] = 1

variants[varName]["helpers"] = {amRope, amParachute}

variants[varName]["special"] = false

varName = "King"

variants[varName] = {}

variants[varName]["name"] = loc("King")

variants[varName]["chance"] = 3

variants[varName]["hat"] = "crown"

variants[varName]["hp"] = 60

variants[varName]["hogLimit"] = 1

variants[varName]["weaponLimit"] = 1

variants[varName]["weapons"] = {amWatermelon, amHellishBomb, amBallgun}

variants[varName]["helperLimit"] = 1

variants[varName]["helpers"] = {amRope, amParachute}

variants[varName]["special"] = true

varName = "Knight"

variants[varName] = {}

variants[varName]["name"] = loc("Knight")

variants[varName]["chance"] = 0

variants[varName]["hat"] = "knight"

variants[varName]["hp"] = 80

variants[varName]["hogLimit"] = 1

variants[varName]["weaponLimit"] = 1

variants[varName]["weapons"] = {amShotgun, amBazooka, amMine}

variants[varName]["helperLimit"] = 1

variants[varName]["helpers"] = {amParachute, amRope}

variants[varName]["special"] = true

--[[

  ##############################################################################

  ### GENERAL VARIABLES USED FOR GAMEPLAY                                    ###

  ##############################################################################

]]--

local unused = {amSnowball, amDrillStrike, amTardis}

local lowWeaps = {amKamikaze}

local lowTresh = 25

local counter = {} -- Saves how many hogs of a variant a team has

local group = {} -- Saves randomized variants for all teams

local teamIndex = {} -- Temporary counter for amount of mutated hogs in team

local teamHogs = {} -- Saves a list of all hogs belonging to a team

local hogCount = {} -- Saves how many hogs a team has

local teamNames = {} -- Saves all teams and names

local hogInfo = {} -- Saves all hogs with their original values

local LastHog = nil -- Last Hedgehog

local CurHog = nil -- Current Hedgehog

local LastTeam = nil -- Last Team

local CurTeam = nil -- Current Team

local mode = 'default' -- Which game type to play

local modeExplicit = false -- Whether the mode was set in script param

local luck = 100 -- Multiplier for bonuses like crates

local strength = 1 -- Multiplier for more weapons

local useVariantHats = true -- Whether to overwrite the hog hats to those of their variants

                            -- In King Mode, crowns are always enforced regardless of this setting

local useVariantNames = false -- Whether to overwrite the hog names to those of their variants

local highHasBonusWeps = false -- whether or not a hog got bonus weapons on current turn

local highHasBonusHelp = false -- whether or not a hog got bonus helpers on current turn

local highPickupCount = 1

local highPickupSDCount = 2

local highHelperCount = 1

local highHelperSDCount = 1

local highEnemyKillHPBonus = 10

local highFriendlyKillHPBonus = 15

local highWeapons = {} -- Saves the weapons from kills

local highHelpers = {} -- Saves the helpers from kills

local highSpecialBonus = {amTeleport, amJetpack}

local highSpecialPool = {amExtraDamage, amVampiric}

local kingLinkPerc = 50 -- Percentage of life to share from the team

local pointsWepBase = 5 -- Game start points weapons

local pointsHlpBase = 2 -- Game start points helpers

local pointsKeepPerc = 80 -- Percentage of points to take to next round

local pointsWepTurn = 5 -- Round bonus points weapons

local pointsHlpTurn = 2 -- Round bonus points helpers

local pointsWepMax = 25 -- Maximum points for weapons

local pointsHlpMax = 10 -- Maximum points for helpers

local pointsKeepSDPerc = 60 -- Percentage of points to take to next round on SD

local pointsWepSDTurn = 7 -- Round bonus points weapons on SD

local pointsHlpSDTurn = 3 -- Round bonus points helpers on SD

local pointsWepSDMax = 35 -- Maximum points for weapons on SD

local pointsHlpSDMax = 15 -- Maximum points for helpers on SD

local pointsWeaponVal = {}

pointsWeaponVal[amBazooka] = 5

pointsWeaponVal[amShotgun] = 4

pointsWeaponVal[amFirePunch] = 3

pointsWeaponVal[amMine] = 5

--pointsWeaponVal[amAirAttack] = 10

pointsWeaponVal[amBee] = 6

pointsWeaponVal[amClusterBomb] = 7

pointsWeaponVal[amGrenade] = 5

pointsWeaponVal[amDEagle] = 3

pointsWeaponVal[amWhip] = 3

pointsWeaponVal[amDynamite] = 7

--pointsWeaponVal[amMineStrike] = 14

pointsWeaponVal[amMortar] = 4

pointsWeaponVal[amWatermelon] = 30

pointsWeaponVal[amSniperRifle] = 3

pointsWeaponVal[amBaseballBat] = 3

pointsWeaponVal[amCake] = 7

--pointsWeaponVal[amNapalm] = 11

pointsWeaponVal[amDrill] = 6

pointsWeaponVal[amHellishBomb] = 20

pointsWeaponVal[amSineGun] = 4

--pointsWeaponVal[amKamikaze] = 3

--pointsWeaponVal[amBallgun] = 12

--pointsWeaponVal[amPianoStrike] = 15

pointsWeaponVal[amSnowball] = 2

pointsWeaponVal[amMolotov] = 3

pointsWeaponVal[amFlamethrower] = 4

pointsWeaponVal[amRCPlane] = 7

--pointsWeaponVal[amDrillStrike] = 12

pointsWeaponVal[amGasBomb] = 2

pointsWeaponVal[amHammer] = 3

pointsWeaponVal[amSMine] = 4

pointsWeaponVal[amAirMine] = 3

pointsWeaponVal[amKnife] = 3

pointsWeaponVal[amPortalGun] = 5

--pointsWeaponVal[amIceGun] = 6

pointsWeaponVal[amSeduction] = 2

local pointsHelperVal = {}

pointsHelperVal[amRope] = 5

pointsHelperVal[amParachute] = 2

--pointsHelperVal[amGirder] = 3

pointsHelperVal[amBlowTorch] = 2

pointsHelperVal[amLowGravity] = 3

--pointsHelperVal[amRubber] = 4

pointsHelperVal[amPickHammer] = 2

pointsHelperVal[amTeleport] = 10

pointsHelperVal[amJetpack] = 8

local pointsPerTeam = {}

local pointsToWep = {} -- List of [points] = {ammo1, ammo2}

local pointsToHlp = {} -- List of [points] = {ammo1, ammo2}

local wepPoints = {}

local hlpPoints = {}

local firstTurnOver = false

local suddenDeath = false

local healthCrateChance = 7

local utilCrateChance = 9

local weaponCrateChance = 12

local healthCrateChanceSD = 12

local utilCrateChanceSD = 16

local weaponCrateChanceSD = 21

local emptyCrateChance = 7

local bonusCrateChance = 5

local cratePickupGap = 35

local utilities = {amInvulnerable, amVampiric, amExtraTime, amExtraDamage, amRope, amLandGun}

local autoSelectHelpers = {amRope, amParachute}

local LastWaterLine = 0 -- Saves WaterLine to make sure a water rise wont trigger highland kill

local helpers = {}

helpers[amSkip] = true

helpers[amRope] = true

helpers[amParachute] = true

helpers[amBlowTorch] = true

helpers[amGirder] = true

helpers[amTeleport] = true

helpers[amSwitch] = true

helpers[amJetpack] = true

helpers[amBirdy] = true

helpers[amPortalGun] = true

helpers[amResurrector] = true

helpers[amTardis] = true

helpers[amLandGun] = true

helpers[amRubber] = true

--helpers[amKamikaze] = true

local posCaseAmmo    = 1

local posCaseHealth  = 2

local posCaseUtility = 4

local posCaseDummy   = 8

--[[

  ##############################################################################

  ### GENERAL BONUS LUA FUNCTIONS                                            ###

  ##############################################################################

]]--

function swap(array, index1, index2)

    array[index1], array[index2] = array[index2], array[index1]

end

function shuffle(array)

    local cnt = #array

    while cnt > 1 do

        local index = GetRandom(cnt) +1

        swap(array, index, cnt)

        cnt = cnt - 1

    end

end

function table.clone(org)

  local copy = {}

  for orig_key, orig_value in pairs(org) do

      copy[orig_key] = orig_value

  end

  return copy

end

--[[

  ##############################################################################

  ### WEAPON, UTILITY AND AMMO FUNCTIONS                                     ###

  ##############################################################################

]]--

function clearHogAmmo(hog)

  for val=0, AmmoTypeMax do

    AddAmmo(hog, val, 0)

  end

end

function autoSelectAmmo(hog, var)

  -- Check if hog has any "useful" helper, select helper, if yes

  for key, val in pairs(autoSelectHelpers) do

    if GetAmmoCount(hog, val) > 0 then

      SetWeapon(val)

      return

    end

  end

end

function AddHogAmmo(hog, ammo)

  -- Add weapons of variant

  for key, val in pairs(ammo) do

    AddAmmo(hog, val, GetAmmoCount(hog, val) +1)

  end

end

function GetRandomAmmo(hog, sourceType)

  local var = getHogInfo(hog, 'variant')

  ammo = {}

  local source = ''

  if variants[var] == nil then

    return ammo

  end

  if sourceType == "weapons" then

    source = variants[var][sourceType]

    sourceLimit = variants[var]["weaponLimit"]

  elseif sourceType == "helpers" then

    source = variants[var][sourceType]

    sourceLimit = variants[var]["helperLimit"]

  elseif sourceType == 'poolWeapons' then

    if highWeapons[hog] == nil then

      highWeapons[hog] = {}

    end

    source = highWeapons[hog]

    if suddenDeath == false then

      sourceLimit = highPickupCount

    else

      sourceLimit = highPickupSDCount

    end

  elseif sourceType == 'poolHelpers' then

    if highHelpers[hog] == nil then

      highHelpers[hog] = {}

    end

    source = highHelpers[hog]

    if suddenDeath == false then

      sourceLimit = highHelperCount

    else

      sourceLimit = highHelperSDCount

    end

  else

    return ammo

  end

  local varAmmo = {}

  for key, val in pairs(source) do

      varAmmo[key] = val

  end

  -- If the amount of random weapons is equally to the amount of weapons possible

  -- We don't need to randomize

  if sourceLimit >= table.getn(source) then

    return varAmmo

  end

  local randIndex = 0

  local i = 0

  while i < sourceLimit and #varAmmo > 0 do

    randIndex = GetRandom(#varAmmo) +1

    ammo[i] = varAmmo[randIndex]

    -- Shift last value to the current index

    varAmmo[randIndex] = varAmmo[#varAmmo]

    -- And remove the last index from the array

    varAmmo[#varAmmo] = nil

    i = i +1

  end

  return ammo

end

function updatePointsLabel(team)

  SetTeamLabel(team, string.format(loc("%d / %d"), pointsPerTeam[team]["weapons"], pointsPerTeam[team]["helpers"]))

end

function addTurnAmmo(hog)

  -- Check if hog is valid

  if hog == nil then

    return

  end

  -- Check if hog is alive

  local hp = GetHealth(hog)

  if hp == nil or hp <= 0 then

    return

  end

  -- Unless its points mode, get weapons normally by variant

  if mode ~= "points" then

    local maxHp = getHogInfo(hog, 'maxHp')

    local hpPer = div(hp * 100, maxHp)

    local wep = getHogInfo(hog, 'weapons')

    local hlp = getHogInfo(hog, 'helpers')