--[[                  ___                   ___

                    (   )                 (   )

___ .-. .-. ___  ___ | |_    .---. ___ .-. | |_

(   )   '   (   )(   (   __) / .-, (   )   (   __)

|  .-.  .-. | |  | | | |   (__) ; ||  .-. .| |

| |  | |  | | |  | | | | ___ .'`  || |  | || | ___

| |  | |  | | |  | | | |(   / .'| || |  | || |(   )

| |  | |  | | |  | | | | | | /  | || |  | || | | |

| |  | |  | | |  ; ' | ' | ; |  ; || |  | || ' | |

| |  | |  | ' `-'  / ' `-' ' `-'  || |  | |' `-' ;

(___)(___)(___'.__.'   `.__.`.__.'_(___)(___)`.__.

----  Recommended settings:

----    * one hedgehog per team (forced by game)

----    * one team per clan

----    * 'Small' one-island map

--]]

HedgewarsScriptLoad("/Scripts/Locale.lua")

HedgewarsScriptLoad("/Scripts/Tracker.lua")

HedgewarsScriptLoad("/Scripts/Params.lua")

--[[

    MUTANT SCRIPT

]]

local hhs = {}

local crates = {}

local numhhs = 0

local meh = false

local gameOver=false

local mutant = nil

local mutant_base_health = 200

local mutant_base_disease = 25

local disease_timer = 2000

local kill_reward = nil

local mt_hurt=false

local killsCounter = 0

local team_fire_punishment = 3

local mutant_kill_reward = 2

local hh_weapons = { amBazooka, amGrenade, amShotgun, amMine}

local mt_weapons = {amWatermelon, amHellishBomb, amBallgun, amRCPlane, amTeleport}

local disease=0

local timer=0

local winScore = 15

local hogsLimit = 1

local teamsDead = {}

local teamsDeleted = {}

local hogLimitHit = false

local teamLimitHit = false

local cnthhs

local circles = {}

local circleFrame = -1

-- Variables for custom achievements

-- Most kills in 1 turn

local recordKills = 0

local recordKillsHogName = nil

local recordKillsTeamName = nil

-- Most suicides

local recordSuicides = 0

local recordSuicidesHogName = nil

local recordSuicidesTeamName = nil

-- Most skips

local recordSkips = 0

local recordSkipsHogName = nil

local recordSkipsTeamName = nil

-- Most crates collected

local recordCrates = 0

local recordCratesHogName = nil

local recordCratesTeamName = nil

-- Most deaths

local recordDeaths = 0

local recordDeathsHogName = nil

local recordDeathsTeamName = nil

-- Total killed hedgehogs

local totalKills = 0

-- Total damage

local totalDamage = 0

local mutantHat = "WhySoSerious"

local feederHat = "poke_slowpoke"

function rules()

    local mineStr

    if MinesTime < 0 then

        mineStr = loc("Mines time: 0s-5s")

    else

        mineStr = string.format(loc("Mines explode after %d s."), div(MinesTime, 1000))

    end

    local ruleSet = loc("Hedgehogs will be revived after their death.") .. "|" ..

    mineStr .. "|" ..

    loc("The first hedgehog to kill someone becomes the Mutant.") .. "|" ..

    loc("The Mutant has super weapons and a lot of health.") .. "|" ..

    loc("The Mutant loses health quickly, but gains health by killing.") .. "|" ..

    " |" ..

    loc("Score points by killing other hedgehogs.") .. "|" ..

    loc("The hedgehog with least points (or most deaths) becomes the Bottom Feeder.") .. "|" ..

    loc("The score and deaths are shown next to the team bar.") .. "|" ..

    string.format(loc("Goal: Score %d points or more to win!"), winScore) .. "|" ..

        " |" ..

    loc("Scoring: ") .. "|" ..

    loc("+2 for becoming the Mutant") .. "|" ..

    loc("+1 to the Mutant for killing anyone") .. "|" ..

    loc("+1 to the Bottom Feeder for killing anyone") .. "|" ..

    loc("-1 to anyone for a suicide")

    return ruleSet

end

function showStartingInfo()

    ShowMission(loc("Mutant"), loc("A Hedgewars tag game"), rules(), -amWatermelon, 5000)

end

function onGameInit()

    -- Sudden Death would be weird

    WaterRise = 0

    HealthDecrease = 0

    -- Weapons must be reset for the Mutant mechanic to work

    EnableGameFlags(gfResetWeps)

    -- King Mode messes with game too much

    DisableGameFlags(gfKing)

end

function limitHogsTeam(gear)

    cnthhs = cnthhs + 1

    if cnthhs > 1 then

        hogLimitHit = true

        SetEffect(gear, heResurrectable, 0)

        setGearValue(gear, "excess", true)

        DeleteGear(gear)

    end

end

function limitHogsClan(gear)

    teamLimitHit = true

    SetEffect(gear, heResurrectable, 0)

    setGearValue(gear, "excess", true)

    DeleteGear(gear)

end

function onGameStart()

    if ClansCount >= 2 then

        SendHealthStatsOff()

        SendAchievementsStatsOff()

    end

    SendRankingStatsOff()

    trackTeams()

    teamScan()

    runOnHogs(saveStuff)

    -- Enforce team and hog limits

    hogLimitHit = false

    -- Rule 1: One team per clan

    if TeamsCount > ClansCount then

        local usedClans = {}

        for i=0, TeamsCount - 1 do

            local teamName = GetTeamName(i)

            local clanNumber = GetTeamClan(teamName)

            if not usedClans[clanNumber] then

                usedClans[clanNumber] = true

            else

                runOnHogsInTeam(limitHogsClan, teamName)

                teamsDeleted[teamName] = true

                setTeamValue(teamName, "Score", getTeamValue(teamName, "Score") -99999)

            end

        end

    end

    -- Rule 2: One hog per team

    for i=0 , TeamsCount - 1 do

        cnthhs = 0

        runOnHogsInTeam(limitHogsTeam, GetTeamName(i))

    end

    if teamLimitHit then

        WriteLnToChat(loc("Only one team per clan allowed! Excess teams will be removed."))

    end

    if hogLimitHit then

        WriteLnToChat(loc("Only one hog per team allowed! Excess hogs will be removed."))

    end

    trackTeams()

    showStartingInfo()

end

function giveWeapons(gear)

    if gear == mutant then

        AddAmmo(gear, amRope)

        for i=1, #mt_weapons do

            AddAmmo(gear, mt_weapons[i])

        end

    else

        for i=1, #hh_weapons do

            AddAmmo(gear,hh_weapons[i])

        end

    end

end

function onAmmoStoreInit()

    SetAmmo(amSkip, 9, 0, 0, 0)

    SetAmmo(amRope,0,1,0,5)

    SetAmmo(amSnowball,0,1,0,1)

    for i=1, #hh_weapons do

        SetAmmo(hh_weapons[i], 0, 0, 0, 1)

    end

    for i=1, #mt_weapons do

        SetAmmo(mt_weapons[i], 0, 3, 0, 1)

    end

end

function drawCircles()

    for i = 0, #hhs do

        if circles[hhs[i]] ~= nil then

            DeleteVisualGear(circles[hhs[i]])

            circles[hhs[i]] = nil

        end

        if hhs[i] ~= CurrentHedgehog then

            if mutant == nil then

                circles[hhs[i]] = AddVisualGear(0, 0, vgtCircle, 0, false)

                SetVisualGearValues(circles[hhs[i]], 0, 0, 0, 0, 0, 0, 0, 22, 5, 0xff000080)

            elseif CurrentHedgehog == mutant then

                circles[hhs[i]] = AddVisualGear(0, 0, vgtCircle, 0, false)

                SetVisualGearValues(circles[hhs[i]], 0, 0, 0, 0, 0, 0, 0, 22, 3, 0xaa000070)

            elseif getGearValue(CurrentHedgehog, "Feeder") and hhs[i] ~= mutant then

                circles[hhs[i]] = AddVisualGear(0, 0, vgtCircle, 0, false)

                SetVisualGearValues(circles[hhs[i]], 0, 0, 0, 0, 0, 0, 0, 22, 3, 0xaa000070)

            elseif hhs[i] == mutant then

                circles[hhs[i]] = AddVisualGear(0, 0, vgtCircle, 0, false)

                SetVisualGearValues(circles[hhs[i]], 0, 0, 0, 0, 0, 0, 0, 22, 5, 0xff000080)

            end

        end

    end

    circleFrame = 0

end

function onNewTurn()

    killsCounter = 0

    if mutant == nil and TotalRounds >= 0 then

        AddCaption( loc("First killer will mutate"), capcolDefault, capgrpGameState )

    end

    checkScore()

    for i=0, TeamsCount-1 do

        local teamName = GetTeamName(i)

        if not teamsDeleted[teamName] then

            SendStat(siClanHealth, getTeamValue(teamName, "Score"), teamName)

        end

    end

    giveWeapons(CurrentHedgehog)

    drawCircles()

    setAIHints()

    kill_reward= numhhs*10

    if CurrentHedgehog == mutant then

        mt_hurt=true

        disease= mutant_base_disease - numhhs

    else

        mt_hurt=false

    end

    setGearValue(CurrentHedgehog, "Alive", true)

end

function countBodies()

        if killsCounter == 2 then

            AddCaption(loc("Double kill!"), capcolDefault, capgrpGameState )

        elseif killsCounter == 3 then

            AddCaption(loc("Mega kill!"), capcolDefault, capgrpGameState )

            PlaySound(sndRegret)

        elseif killsCounter == 4 then

            AddCaption(loc("Ultra kill!"), capcolDefault, capgrpGameState )

        elseif killsCounter == 5 then

            AddCaption(loc("Monster kill!"), capcolDefault, capgrpGameState )

            PlaySound(sndIllGetYou)

        elseif killsCounter == 6 then

            AddCaption(loc("Ludicrous kill!"), capcolDefault, capgrpGameState )

            PlaySound(sndNutter)

        elseif killsCounter == 7 then

            AddCaption(loc("Holy shit!"), capcolDefault, capgrpGameState )

            PlaySound(sndLaugh)

        elseif killsCounter > 8 then

            AddCaption(loc("Insanity!"), capcolDefault, capgrpGameState )

        end

        if killsCounter > recordKills then

            recordKills = killsCounter

            recordKillsHogName = getGearValue(CurrentHedgehog, "Name")

            recordKillsTeamName = GetHogTeamName(CurrentHedgehog)

        end

end

function onGameTick()

    if circleFrame > -1 then

        for i = 0, #hhs do

            if circles[hhs[i]] ~= nil and hhs[i]~= nil then

                hhx, hhy = GetGearPosition(hhs[i])

                SetVisualGearValues(circles[hhs[i]], hhx + 1, hhy - 3, 0, 0, 0, 0, 0, 40 - (circleFrame % 25))

            end

        end

        circleFrame = circleFrame + 0.06

        if circleFrame >= 25 then

            for i = 0, #hhs do

                if circles[hhs[i]] ~= nil then

                    DeleteVisualGear(circles[hhs[i]])

                    circles[hhs[i]] = nil

                end

            end

        end

    end

    if (TurnTimeLeft==0 or band(GetState(mutant), gstHHDriven) == 0) and mt_hurt then

        mt_hurt = false

    end

    -- Mutant's disease

    -- Hurt Mutant during its turn time

    -- Mutant's health is safe in ready phase

    if mt_hurt and mutant~=nil and ReadyTimeLeft == 0 then

        timer = timer + 1

        if timer > disease_timer then

            timer = 0

            local h = GetHealth(mutant)-disease

            SetHealth(mutant, h)

            -- Low health warning

            if h <= 75 then

                PlaySound(sndPoisonMoan, mutant)

            elseif h <= 150 then

                PlaySound(sndPoisonCough, mutant)

            end

            local tag = AddVisualGear(GetX(mutant), GetY(mutant)-5, vgtHealthTag, disease, true)

            SetVisualGearValues(tag, nil, nil, nil, nil, nil, nil, nil, nil, nil, GetClanColor(GetHogClan(mutant)))

            if GetHealth(mutant)<=0 then

                SetHealth(mutant,0)

                mt_hurt= false

                setGearValue(mutant,"SelfDestruct",true)

                EndTurn()

            end

        end

    end

end

--[[

Forces the special mutant/feeder names and hats only to be

taken by those who deserved it.

Names and hats will be changed (and ridiculed) if neccesary.

]]

function exposeIdentityTheft(gear)

    local lon = string.lower(GetHogName(gear)) -- lowercase origina name

    local name, hat

    -- Change name if hog uses a reserved one

    if lon == "mutant" or lon == string.lower(loc("Mutant")) then

       SetHogName(gear, loc("Identity Thief"))

       SetHogHat(gear, "Disguise")

    elseif lon == "bottom feeder" or lon == string.lower(loc("Bottom Feeder")) then

       -- Word play on "Bottom Feeder". Someone who is low on cotton. :D

       -- Either translate literally or make up your ow word play

       SetHogName(gear, loc("Cotton Needer"))

       SetHogHat(gear, "StrawHat")

    end

    -- Strip hog off its special hat

    if GetHogHat(gear) == mutantHat or GetHogHat(gear) == feederHat then

       SetHogHat(gear, "NoHat")

    end

end

function saveStuff(gear)

    exposeIdentityTheft(gear)

    setGearValue(gear,"Name",GetHogName(gear))

    setGearValue(gear,"Hat",GetHogHat(gear))

end

function armageddon(gear)

    SetState(gear, gstLoser)

    SetEffect(gear, heResurrectable, 0)

    SetHealth(gear, 0)

end

function renderScores()

    for i=0, TeamsCount-1 do

        local name = GetTeamName(i)

        SetTeamLabel(name, string.format(loc("%d | %d"), getTeamValue(name, "Score"), getTeamValue(name, "DeadHogs")))

    end

end

function createEndGameStats()

    SendStat(siGraphTitle, loc("Score graph"))

    local teamsSorted = {}

    for i=0, TeamsCount-1, 1 do

        teamsSorted[i+1] = GetTeamName(i)

    end

    -- Achievements stuff

    local achievements = 0

    --- Most kills per turn

    if recordKills >= 3 then

        SendStat(siMaxStepKills, string.format("%d %s (%s)", recordKills, recordKillsHogName, recordKillsTeamName))

        achievements = achievements + 1

    end

    --- Most crates collected

    if recordCrates >= 5 then

        SendStat(siCustomAchievement, string.format(loc("%s (%s) was the greediest hedgehog and collected %d crates."), recordCratesHogName, recordCratesTeamName, recordCrates))

        achievements = achievements + 1

    end

    --- Most suicides

    if recordSuicides >= 5 then

        SendStat(siCustomAchievement, string.format(loc("%s (%s) hate life and suicided %d times."), recordSuicidesHogName, recordSuicidesTeamName, recordSuicides))

        achievements = achievements + 1

    end

    --- Most deaths

    if recordDeaths >= 5 then

        SendStat(siCustomAchievement, string.format(loc("Poor %s (%s) died %d times."), recordDeathsHogName, recordDeathsTeamName, recordDeaths))

        achievements = achievements + 1

    end

    --- Most skips

    if recordSkips >= 3 then

        SendStat(siMaxTurnSkips, string.format("%d %s (%s)", recordSkips, recordSkipsHogName, recordSkipsTeamName))

        achievements = achievements + 1

    end

    --- Total damage

    if totalDamage >= 900 then

        SendStat(siCustomAchievement, string.format(loc("%d damage was dealt in this game."), totalDamage))

        achievements = achievements + 1

    end

    --- Total kills

    if totalKills >= 20 or (achievements <= 0 and totalKills >= 1) then

        SendStat(siKilledHHs, tostring(totalKills))

        achievements = achievements + 1

    end

    -- Score and stats stuff

    local showScore = ""

    local rank = 0

    local rankPlus = 1

    local prevScore

    table.sort(teamsSorted, function(team1, team2) return getTeamValue(team1, "Score") > getTeamValue(team2, "Score") end)

    for i=1, TeamsCount do

        local score = getTeamValue(teamsSorted[i], "Score")

        local deaths = getTeamValue(teamsSorted[i], "DeadHogs")

        if i == 1 or score < prevScore then

            rank = rank + rankPlus

            rankPlus = 1

            prevScore = score

        else

            rankPlus = rankPlus + 1

        end