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 CakeWarning = false

local CakeFireWarning = false

local CakeTries = 0

local addCake = true

local takeASeat = false

local Stars = {}

local jokeAwardNavy = nil

local jokeAwardSpeed = nil

local jokeAwardDamage = nil

local recordBroken = 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

local racing = false

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

    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

    -- 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"), capcolDefault, capgrpMessage2)

        MrMine = nil

    elseif GetGearType(gear) == gtCake then

        Cake = nil

        CakeWarning = false

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

    local recordInfo = ""

    if isSinglePlayer then

        recordInfo = getReadableChallengeRecord("Highscore")

    end

    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.")

                .. "|" .. recordInfo,

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

    for i=0, TeamsCount-1 do

        SetTeamLabel(GetTeamName(i), "0")

    end

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

    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

    takeASeat = 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)

        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

    PlaySound(sndExplosion)

    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

        SkipTurn()

        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

        local _, Y = GetVisualGearValues(s)

        if Y ~= nil and Y > WaterLine + 500 then

            DeleteVisualGear(s)

            Stars[s] = nil

        elseif Y == nil then

            Stars[s] = nil

        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)

        if Cake ~= nil then

            local cx,cy = GetGearPosition(Cake)

            if y < cy-1500 then DeleteGear(Cake) end

            if Cake ~= nil and GetHealth(Cake) < 999980 then

                if not CakeWarning and gearIsInCircle(CurrentHedgehog,cx,cy,1350) then

                    AddCaption(loc("Warning: Fire cake detected"))

                    CakeWarning = true

                end

                if gearIsInCircle(CurrentHedgehog,cx,cy,450) then

                    if not CakeFireWarning then

                        AddCaption(loc("Don't touch the flames!"))

                        CakeFireWarning = true

                    end

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

                    DeleteGear(Cake)

                end

            end

        end

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

            end

        end

     end

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

        if MaxHeight < delayHeight and

           TurnTimeLeft<(MAX_TURN_TIME-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, false)

            if s then

                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

        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 rawFinishTime = GameTime-startTime

        local finishTime = rawFinishTime/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

                SaveMissionVar("Won", "true")

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

                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)

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

                updateChallengeRecord("TimeRecord", rawFinishTime, false)

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

                updateChallengeRecord("Highscore", getActualHeight(RecordHeight))

                SendStat(siPointType, "!TIME")

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

                EndGame()

                SetState(CurrentHedgehog, gstWinner)

                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), capcolDefault, capgrpGameState)

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

                makeMultiPlayerWinnerStat(CurrentHedgehog)

                PlaySound(sndVictory, CurrentHedgehog)

		SetWeapon(amNothing)

                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