HedgewarsScriptLoad("/Scripts/Locale.lua")

HedgewarsScriptLoad("/Scripts/Utils.lua")

HedgewarsScriptLoad("/Scripts/Params.lua")

local hTag = nil

local hTagHeight = 33000

local tTag = nil

local rTag = nil

local startTime = 0

local MaxHeight = 32640

local RecordHeight = 33000

local RecordHeightHogName = nil

local Fire = {}

--local BoomFire = nil

local HH = {}

local totalHedgehogs = 0

local deadHedgehogs = 0

local currTeam = ''

local teams = {}

local teamScoreStats = {}

local teamBests = {}

local teamTimes = {}

local MrMine -- in honour of sparkle's first arrival in the cabin

local YouWon = false

local YouLost = false

local HogsAreInvulnerable = false

local WaterRise = nil

local Cake = nil

local CakeTries = 0

local addCake = true

local Stars = {}

local tauntNoo = false

local jokeAwardNavy = nil

local jokeAwardSpeed = nil

local jokeAwardDamage = nil

local recordBroken = false

local ready = false

local dummyHog = nil

local dummySkip = 0

local baseWaterSpeed = 2

local waterSpeed = 0

local waterAccel = 0

local delayHeight = 32000

local delayTime = 0

local airMineX = {}

local airMineY = {}

local airMine = {}

local init = true

local multiplayerVictoryDelay = -1

local multiplayerWinningHogs = {}

local multiplayerWins = 0

-- Placement positions of winning hogs

local victoryPosses = { }

do

    local m = 0

    for y=108, 39, -32 do

        for x=1820+m, 1972-m, 22 do

            table.insert(victoryPosses, {x=x, y=y})

        end

        m = m + 32

    end

end

function onParameters()

    parseParams()

    if params["speed"] ~= nil then

        baseWaterSpeed = params["speed"]

    end

    if params["accel"] ~= nil then

        waterAccel = params["accel"]

        if waterAccel ~= 0 then waterAccel = div(32640000,waterAccel) end

    end

    if params["delaytime"] ~= nil then

        delayTime = params["delaytime"]

    end

    if params["delayheight"] ~= nil then

        delayHeight = 32768-params["delayheight"]

    end

    if params["nocake"] ~= nil then addCake = false end

end

function onGameInit()

    -- Ensure people get same map for same theme

    TurnTime = 999999999

    CaseFreq = 0

    Explosives = 0

    MineDudPercent = 0

    EnableGameFlags(gfOneClanMode)

    DisableGameFlags(gfBottomBorder+gfBorder)

    --This reduced startup time by only about 15% and looked ugly

    --EnableGameFlags(gfDisableLandObjects) 

    -- force seed instead.  Some themes will still be easier, but at least you won't luck out on the same theme

    Seed = ClimbHome

    -- Disable Sudden Death

    WaterRise = 0

    HealthDecrease = 0

end

function onGearAdd(gear)

    if GetGearType(gear) == gtHedgehog then

        HH[gear] = 1

        totalHedgehogs = totalHedgehogs + 1

        teams[GetHogTeamName(gear)] = 1

    elseif init and GetGearType(gear) == gtAirMine then

        airMine[gear] = 1

    end

end

function onGearDelete(gear)

    if gear == MrMine then

        AddCaption(loc("Once you set off the proximity trigger, Mr. Mine is not your friend"), 0xFFFFFFFF, capgrpMessage)

        MrMine = nil

    elseif gear == Cake then

        Cake = nil

    elseif GetGearType(gear) == gtHedgehog then

	onGameTick20()

	onGearDamage(gear, 0)

        HH[gear] = nil

    end

end

function onGameStart()

    --SetClanColor(ClansCount-1, 0x0000ffff) appears to be broken

    SendHealthStatsOff()

    ShowMission(loc("Climb Home"),

                loc("Challenge"),

                loc("You are far from home, and the water is rising, climb up as high as you can!|Your score will be based on your height."),

                -amRope, 0)

    local x = 1818

    for h,i in pairs(HH) do

        if h ~= nil then

            -- SetGearPosition(h,x,32549)

            SetGearPosition(h,x,108)

            SetHealth(h,1)

            if x < 1978 then x = x+32 else x = 1818 end

            if GetEffect(h,heInvulnerable) == 0 then

                SetEffect(h,heInvulnerable,1)

            else

                HogsAreInvulnerable = true

            end

            SetState(h,bor(GetState(h),gstInvisible))

        end

    end

-- 1925,263 - Mr. Mine position

    MrMine = AddGear(1925,263,gtMine,0,0,0,0)

end

function onAmmoStoreInit()

    SetAmmo(amRope, 9, 0, 0, 0)

end

function onNewTurn()

    if init then

        init = false

        for a,i in pairs(airMine) do

            x,y = GetGearPosition(a)

            airMineX[a] = x

            airMineY[a] = y

        end

    else

        for a,i in pairs(airMine) do

            local x,y = GetGearPosition(a)

            if not x or airMineX[a] ~= x or airMineY[a] ~= y then

                DeleteGear(a)

                AddGear(airMineX[a],airMineY[a], gtAirMine, gsttmpFlag, 0, 0, 0)

            end

        end

    end

    ready = false

    startTime = GameTime

    --disable to preserve highest over multiple turns

    --will need to change water check too ofc

    MaxHeight = 32640

    hTagHeight = 33000

    SetWaterLine(32768)

    YouWon = false

    YouLost = false

    tauntNoo = false

    recordBroken = false

    currTeam = GetHogTeamName(CurrentHedgehog)

    if CurrentHedgehog ~= nil then

        if CurrentHedgehog ~= dummyHog or multiplayerWinningHogs[CurrentHedgehog] == true then

            SetGearPosition(CurrentHedgehog, 1951,32640)

            HogTurnLeft(CurrentHedgehog, true)

            if not HogsAreInvulnerable then SetEffect(CurrentHedgehog,heInvulnerable,0) end

            AddVisualGear(1951,32640,vgtExplosion,0,false)

            SetState(CurrentHedgehog,band(GetState(CurrentHedgehog),bnot(gstInvisible)))

            SetWeapon(amRope)

            ready = true

        else

            dummySkip = GameTime+1

        end

    end

    for hog, _ in pairs(multiplayerWinningHogs) do

        SetEffect(hog, heInvulnerable, 1)

    end

    for f,i in pairs(Fire) do

        DeleteGear(f)

    end

    for s,i in pairs(Stars) do

        DeleteVisualGear(s)

        Stars[s] = nil

    end

    if CurrentHedgehog ~= dummyHog then

        for i = 0,12 do

            flame = AddGear(2000+i*2,308, gtFlame, gsttmpFlag,  0, 0, 0)

            SetTag(flame, 999999+i)

            Fire[flame]=1

        end

    end

    if Cake ~= nil then DeleteGear(Cake) end

    CakeTries = 0

end

--function onGearDelete(gear)

--    if gear == WaterRise and MaxHeight > 500 and CurrentHedgehog ~= nil and band(GetState(CurrentHedgehog),gstHHDriven) ~= 0 then

--        WaterRise = AddGear(0,0,gtWaterUp, 0, 0, 0, 0)

--    end

--end

function FireBoom(x,y,d) -- going to add for rockets too

    AddVisualGear(x,y,vgtExplosion,0,false)

    -- should approximate circle by removing corners

    --if BoomFire == nil then BoomFire = {} end

    for i = 0,50 do

	fx = GetRandom(d)-div(d,2)

	fy = GetRandom(d)-div(d,2)

	if fx<0 then

	   fdx = -5000-GetRandom(3000)

	else

	   fdx = 5000+GetRandom(3000)

	end

	if fy<0 then

	   fdy = -5000-GetRandom(3000)

	else

	   fdy = 5000+GetRandom(3000)

	end

        flame = AddGear(x+fx, y+fy, gtFlame, gsttmpFlag,  fdx, fdy, 0)

        SetTag(flame, 999999+i)

        SetFlightTime(flame, 0)

        Fire[flame]=1

--        BoomFire[flame]=1

    end

end

function onGameTick20()

    local x,y

    if math.random(20) == 1 then AddVisualGear(2012,56,vgtSmoke,0,false) end

    if CurrentHedgehog == dummyHog and dummySkip ~= 0 and dummySkip < GameTime then

        ParseCommand("/skip")

        dummySkip = 0

    end

    --if BoomFire ~= nil then

    --    for f,i in pairs(BoomFire) do

    --        if band(GetState(f),gstCollision~=0) then DeleteGear(f) end

    --    end

    --    BoomFire = nil

    --end

    for s,i in pairs(Stars) do

        g1, g2, g3, g4, g5, g6, g7, g8, g9, g10 = GetVisualGearValues(s)

        if g1 > WaterLine + 500 then

            DeleteVisualGear(s)

            Stars[s] = nil

        end

        --else  wasn't really visible, pointless.

        --    g5 = g5+1

        --    if g5 > 360 then g5 = 0 end

        --    SetVisualGearValues(s, g1, g2, g3, g4, g5, g6, g7, g8, g9, g10)

        --end

    end

    -- This will be executed if a player reached home in multiplayer

    if multiplayerVictoryDelay > 0 then

        multiplayerVictoryDelay = multiplayerVictoryDelay - 20

        if multiplayerVictoryDelay <= 0 then

            -- If delay's over, the game will continue with the next hog

            if CurrentHedgehog then

                multiplayerWinningHogs[CurrentHedgehog] = true

                multiplayerWins = multiplayerWins + 1

                local victoryX, victoryY

                if multiplayerWins <= #victoryPosses then

                    victoryX, victoryY = victoryPosses[multiplayerWins].x, victoryPosses[multiplayerWins].y

                else

                    victoryX, victoryY = victoryPosses[#victoryPosses].x, victoryPosses[#victoryPosses].y

                end

                SetGearPosition(CurrentHedgehog, victoryX, victoryY)

                SetEffect(CurrentHedgehog, heInvulnerable, 1)

                SetHealth(CurrentHedgehog, 1)

                if (deadHedgehogs + multiplayerWins) >= totalHedgehogs then

                    makeFinalMultiPlayerStats()

                    EndGame()

                    onAchievementsDeclaration()

                else

                    EndTurn(true)

                    SetInputMask(0xFFFFFFFF)

                end

                return

            end

        end

    end

    if CurrentHedgehog ~= nil then x,y = GetGearPosition(CurrentHedgehog) end

    if Cake ~= nil and CurrentHedgehog ~= nil then

        local cx,cy = GetGearPosition(Cake)

        if y < cy-1500 then

            DeleteGear(Cake)

            Cake = nil

        end

        if Cake ~= nil and GetHealth(Cake) < 999980 and gearIsInCircle(CurrentHedgehog,cx,cy,450) then

            FireBoom(cx,cy,200) -- todo animate

            DeleteGear(Cake)

            Cake = nil

        end

    end

    if CurrentHedgehog ~= nil and TurnTimeLeft > 0 and band(GetState(CurrentHedgehog),gstHHDriven) ~= 0 then

        if MaxHeight < delayHeight and

           TurnTimeLeft<(999999999-delayTime) and 

            MaxHeight > 286 and WaterLine > 286 then

            if waterAccel ~= 0 then

                SetWaterLine(WaterLine-(baseWaterSpeed+div(getActualHeight(MaxHeight)*100,waterAccel)))

            else

                SetWaterLine(WaterLine-baseWaterSpeed)

            end

        end

        if y > 0 and y < 30000 and MaxHeight > 286 and math.random(y) < 500 then

            local s = AddVisualGear(0, 0, vgtStraightShot, 0, true)

            local c = div(250000,y)

            if c > 255 then c = 255 end

            c = c * 0x10000 + 0xFF0000FF

            SetVisualGearValues(s,

                math.random(2048), -5000, 0, -1-(1/y*1000), 

                math.random(360),

                0,

                999999999, -- frameticks

                sprStar, -- star

                0, c)

                --,  0xFFCC00FF) -- could be fun to make colour shift as you rise...

            Stars[s] = 1

        end    

    end

    if CurrentHedgehog ~= nil and band(GetState(CurrentHedgehog),gstHHDriven) == 0 then

        for f,i in pairs(Fire) do -- takes too long to fall otherwise

            DeleteGear(f)

        end

        if Cake ~= nil then

            DeleteGear(Cake)

            Cake = nil

        end

    end

    if CurrentHedgehog ~= nil and TurnTimeLeft > 0 then

        local vx, vy = GetGearVelocity(CurrentHedgehog)

	local distanceFromWater = WaterLine - y

        --[[ check joke awards ]]

        -- navy award: when distance from main map is over 1000

        local navyDistance = 1250

        if x < -navyDistance or x > LAND_WIDTH+navyDistance then

            local awarded = false

            local dist = 0

            if jokeAwardNavy == nil then

                awarded = true

            else

                if x < 0 then

                    dist = math.abs(x)

                else

                    dist = x - LAND_WIDTH

                end

                if dist > jokeAwardNavy.distance then

                    awarded = true

                end

            end

            if awarded == true then

                jokeAwardNavy = {

                    hogName = GetHogName(CurrentHedgehog),

                    teamName = GetHogTeamName(CurrentHedgehog),

                    distance = dist

                }

            end

        end

        -- Speed award for largest distance from water

        if distanceFromWater > 3000 and WaterLine < 32000 then

            local awarded = false

            if jokeAwardSpeed == nil or distanceFromWater > jokeAwardSpeed.distance then

                awarded = true

            end

            if awarded == true then

                jokeAwardSpeed = {

                    hogName = GetHogName(CurrentHedgehog),

                    teamName = GetHogTeamName(CurrentHedgehog),

                    distance = distanceFromWater

                }

            end

        end

            local finishTime = (GameTime-startTime)/1000

            local roundedFinishTime = math.ceil(math.floor(finishTime+0.5))

            if isSinglePlayer then

                if distanceFromWater < 0 and not YouLost and not YouWon then

                    makeSinglePlayerLoserStats()

                    YouLost = true

                end

                -- FIXME: Hog is also in winning box if it just walks into the chair from the left, touching it. Intentional?

                if not YouWon and not YouLost and gearIsInBox(CurrentHedgehog, 1920, 252, 50, 50) then

                    AddCaption(loc("Victory!"))

                    ShowMission(loc("Climb Home"),

                                loc("Made it!"),

                                string.format(loc("Ahhh, home, sweet home. Made it in %d seconds."), roundedFinishTime),

                                -amRope, 0)

                    PlaySound(sndVictory,CurrentHedgehog)

                    SetState(CurrentHedgehog, gstWinner)

                    SendStat(siGameResult, loc("You have beaten the challenge!"))

                    SendStat(siGraphTitle, loc("Your height over time"))

                    SendStat(siCustomAchievement, string.format(loc("%s reached home in %.3f seconds. Congratulations!"), GetHogName(CurrentHedgehog), finishTime))

                    SendStat(siCustomAchievement, string.format(loc("%s bravely climbed up to a dizzy height of %d to reach home."), GetHogName(CurrentHedgehog), getActualHeight(RecordHeight)))

                    SendStat(siPointType, loc("seconds"))

                    SendStat(siPlayerKills, tostring(roundedFinishTime), GetHogTeamName(CurrentHedgehog))

                    EndGame()

                    onAchievementsDeclaration()

                    YouWon = true

                end

            else

                if distanceFromWater < 0 and not YouLost and not YouWon then

                    makeMultiPlayerLoserStat(CurrentHedgehog)

                    deadHedgehogs = deadHedgehogs + 1

                    YouLost = true

                    if deadHedgehogs >= totalHedgehogs then

                        makeFinalMultiPlayerStats()

                        EndGame()

                        onAchievementsDeclaration()

                    end

                end

                -- Check victory

                if not YouWon and not YouLost and gearIsInBox(CurrentHedgehog, 1920, 252, 50, 50) and

                        -- Delay victory if MrMine is triggered

                        (not MrMine or (MrMine and band(GetState(MrMine), gstAttacking) == 0)) then

                    -- Player managed to reach home in multiplayer.

                    -- Stop hog, disable controls, celebrate victory and continue the game after 4 seconds.

                    AddCaption(string.format(loc("%s climbed home in %d seconds!"), GetHogName(CurrentHedgehog), roundedFinishTime))

                    SendStat(siCustomAchievement, string.format(loc("%s (%s) reached home in %.3f seconds."), GetHogName(CurrentHedgehog), GetHogTeamName(CurrentHedgehog), finishTime))

                    makeMultiPlayerWinnerStat(CurrentHedgehog)

                    PlaySound(sndVictory, CurrentHedgehog)

                    -- TODO: Unselect weapon.

                    -- Note: SetWeapon(amNothing) does not work. :-(

                    SetGearMessage(CurrentHedgehog, band(GetGearMessage(CurrentHedgehog), bnot(gmLeft+gmRight+gmUp+gmDown+gmHJump+gmLJump+gmPrecise)))

                    SetInputMask(0x00)

                    -- TODO: Add stupid winner grin.

                    multiplayerVictoryDelay = 4000

                    YouWon = true

                end

            end

        if GameTime % 500 == 0 then

            if not isSinglePlayer then

	        for t,i in pairs(teams) do

                    if currTeam == t then

                        SendStat(siClanHealth, tostring(getActualHeight(y)), t)

                    else

                        SendStat(siClanHealth, '0', t)

                    end

                end

            elseif CurrentHedgehog ~= nil then

                SendStat(siClanHealth, tostring(getActualHeight(y)), GetHogTeamName(CurrentHedgehog))

            end

            -- play taunts

            if not YouWon and not YouLost then

                local nooDistance = 500

                if ((x < -nooDistance and vx < 0) or (x > LAND_WIDTH+nooDistance and vx > 0)) then

                    if (tauntNoo == false and distanceFromWater > 80) then

                        PlaySound(sndNooo, CurrentHedgehog)

                        tauntNoo = true

                    end

                end

            end

            if addCake and CakeTries < 10 and y < 32600 and y > 3000 and Cake == nil and band(GetState(CurrentHedgehog),gstHHDriven) ~= 0 then 

                -- doing this just after the start the first time to take advantage of randomness sources

                -- Pick a clear y to start with

                if y > 31000 then cy = 24585 elseif

                   y > 28000 then cy = 21500 elseif

                   y > 24000 then cy = 19000 elseif

                   y > 21500 then cy = 16000 elseif

                   y > 19000 then cy = 12265 elseif