(*

 * Hedgewars, a free turn based strategy game

 * Copyright (c) 2004-2015 Andrey Korotaev <unC0Rr@gmail.com>

 *

 * This program is free software; you can redistribute it and/or modify

 * it under the terms of the GNU General Public License as published by

 * the Free Software Foundation; version 2 of the License

 *

 * This program is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 * GNU General Public License for more details.

 *

 * You should have received a copy of the GNU General Public License

 * along with this program; if not, write to the Free Software

 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA

 *)

{$INCLUDE "options.inc"}

unit uTeams;

interface

uses uConsts, uInputHandler, uRandom, uFloat, uStats,

     uCollisions, uSound, uStore, uTypes, uScript

     {$IFDEF USE_TOUCH_INTERFACE}, uWorld{$ENDIF};

procedure initModule;

procedure freeModule;

function  AddTeam(TeamColor: Longword): PTeam;

procedure SwitchHedgehog;

procedure AfterSwitchHedgehog;

procedure InitTeams;

function  TeamSize(p: PTeam): Longword;

procedure RecountTeamHealth(team: PTeam);

procedure RecountAllTeamsHealth();

procedure RestoreHog(HH: PHedgehog);

procedure RestoreTeamsFromSave;

function  CheckForWin: boolean;

procedure TeamGoneEffect(var Team: TTeam);

procedure SwitchCurrentHedgehog(newHog: PHedgehog);

var MaxTeamHealth: LongInt;

implementation

uses uLocale, uAmmos, uChat, uVariables, uUtils, uIO, uCaptions, uCommands, uDebug,

    uGearsUtils, uGearsList, uVisualGearsList, uTextures

    {$IFDEF USE_TOUCH_INTERFACE}, uTouch{$ENDIF};

var TeamsGameOver: boolean;

    NextClan: boolean;

function CheckForWin: boolean;

var AliveClan: PClan;

    s, cap: ansistring;

    ts: array[0..(cMaxTeams - 1)] of ansistring;

    t, AliveCount, i, j: LongInt;

begin

CheckForWin:= false;

AliveCount:= 0;

for t:= 0 to Pred(ClansCount) do

    if ClansArray[t]^.ClanHealth > 0 then

        begin

        inc(AliveCount);

        AliveClan:= ClansArray[t]

        end;

if (AliveCount > 1) or ((AliveCount = 1) and ((GameFlags and gfOneClanMode) <> 0)) then

    exit;

CheckForWin:= true;

TurnTimeLeft:= 0;

ReadyTimeLeft:= 0;

// if the game ends during a multishot, do last TurnReaction

if (not bBetweenTurns) and isInMultiShoot then

    TurnReaction();

if not TeamsGameOver then

    begin

    if AliveCount = 0 then

        begin // draw

        AddCaption(GetEventString(eidRoundDraw), cWhiteColor, capgrpGameState);

        if SendGameResultOn then

            SendStat(siGameResult, shortstring(trmsg[sidDraw]));

        AddGear(0, 0, gtATFinishGame, 0, _0, _0, 3000);

        end

    else // win

        begin

        with AliveClan^ do

            begin

            if TeamsNumber = 1 then // single team wins

                begin

                s:= ansistring(Teams[0]^.TeamName);

                // Victory caption is randomly selected

                cap:= FormatA(GetEventString(eidRoundWin), s);

                AddCaption(cap, cWhiteColor, capgrpGameState);

                s:= FormatA(trmsg[sidWinner], s);

                end

            else // clan with at least 2 teams wins

                begin

                s:= '';

                for j:= 0 to Pred(TeamsNumber) do

                    begin

                    ts[j] := Teams[j]^.TeamName;

                    end;

                // Write victory message for caption and stats page

                if (TeamsNumber = cMaxTeams) or (TeamsCount = TeamsNumber) then

                    // No enemies for some reason … Everyone wins!!1!

                    s:= trmsg[sidWinnerAll]

                else if (TeamsNumber >= 2) and (TeamsNumber < cMaxTeams) then

                    // List all winning teams in a list

                    s:= FormatA(trmsg[TMsgStrId(Ord(sidWinner2) + (TeamsNumber - 2))], ts);

                // The winner caption is the same as the stats message and not randomized

                cap:= s;

                AddCaption(cap, cWhiteColor, capgrpGameState);

                // TODO (maybe): Show victory animation/captions per-team instead of all winners at once?

                end;

            for j:= 0 to Pred(TeamsNumber) do

                with Teams[j]^ do

                    for i:= 0 to cMaxHHIndex do

                        with Hedgehogs[i] do

                            if (Gear <> nil) then

                                Gear^.State:= gstWinner;

            if Flawless then

                AddVoice(sndFlawless, Teams[0]^.voicepack)

            else

                AddVoice(sndVictory, Teams[0]^.voicepack);

            end;

        if SendGameResultOn then

            SendStat(siGameResult, shortstring(s));

        AddGear(0, 0, gtATFinishGame, 0, _0, _0, 3000)

        end;

    SendStats;

    end;

TeamsGameOver:= true;

GameOver:= true

end;

procedure SwitchHedgehog;

var c, i, t: LongWord;

    PrevHH, PrevTeam : LongWord;

begin

TargetPoint.X:= NoPointX;

if checkFails(CurrentTeam <> nil, 'nil Team', true) then exit;

with CurrentHedgehog^ do

    if (PreviousTeam <> nil) and PlacingHogs and Unplaced then

        begin

        Unplaced:= false;

        if Gear <> nil then

           begin

           DeleteCI(Gear);

           FindPlace(Gear, false, 0, LAND_WIDTH, true);

           if Gear <> nil then

               AddCI(Gear)

           end

        end;

PreviousTeam:= CurrentTeam;

with CurrentHedgehog^ do

    begin

    if Gear <> nil then

        begin

        MultiShootAttacks:= 0;

        Gear^.Message:= 0;

        Gear^.Z:= cHHZ;

        RemoveGearFromList(Gear);

        InsertGearToList(Gear)

        end

    end;

// Try to make the ammo menu viewed when not your turn be a bit more useful for per-hog-ammo mode

with CurrentTeam^ do

    if ((GameFlags and gfPerHogAmmo) <> 0) and (not ExtDriven) and (CurrentHedgehog^.BotLevel = 0) then

        begin

        c:= CurrHedgehog;

        repeat

            begin

            inc(c);

            if c > cMaxHHIndex then

                c:= 0

            end

        until (c = CurrHedgehog) or (Hedgehogs[c].Gear <> nil) and (Hedgehogs[c].Effects[heFrozen] < 50255);

        LocalAmmo:= Hedgehogs[c].AmmoStore

        end;

c:= CurrentTeam^.Clan^.ClanIndex;

repeat

    with ClansArray[c]^ do

        if (GameFlags and gfTagTeam) <> 0 then

            begin

            if (CurrTeam = TagTeamIndex) then

                begin

                if (c = 0) and (not PlacingHogs) then

                    inc(TotalRounds);

                TagTeamIndex:= Pred(TagTeamIndex) mod TeamsNumber;

                CurrTeam:= Pred(CurrTeam) mod TeamsNumber;

                inc(c);

                NextClan:= true;

                end;

            end

        else if (c = 0) and (not PlacingHogs) then

            inc(TotalRounds);

    if (GameFlags and gfTagTeam) = 0 then

        inc(c);

    if c = ClansCount then

        c:= 0;

    with ClansArray[c]^ do

        begin

        PrevTeam:= CurrTeam;

        repeat

            CurrTeam:= Succ(CurrTeam) mod TeamsNumber;

            CurrentTeam:= Teams[CurrTeam];

            with CurrentTeam^ do

                begin

                PrevHH:= CurrHedgehog mod HedgehogsNumber; // prevent infinite loop when CurrHedgehog = 7, but HedgehogsNumber < 8 (team is destroyed before its first turn)

                repeat

                    CurrHedgehog:= Succ(CurrHedgehog) mod HedgehogsNumber;

                until ((Hedgehogs[CurrHedgehog].Gear <> nil) and (Hedgehogs[CurrHedgehog].Effects[heFrozen] < 256)) or (CurrHedgehog = PrevHH)

                end

        until ((CurrentTeam^.Hedgehogs[CurrentTeam^.CurrHedgehog].Gear <> nil) and (CurrentTeam^.Hedgehogs[CurrentTeam^.CurrHedgehog].Effects[heFrozen] < 256)) or (PrevTeam = CurrTeam) or ((CurrTeam = TagTeamIndex) and ((GameFlags and gfTagTeam) <> 0))

        end;

        if (CurrentTeam^.Hedgehogs[CurrentTeam^.CurrHedgehog].Gear = nil) or (CurrentTeam^.Hedgehogs[CurrentTeam^.CurrHedgehog].Effects[heFrozen] > 255) then

            begin

            with CurrentTeam^.Clan^ do

                for t:= 0 to Pred(TeamsNumber) do

                    with Teams[t]^ do

                        for i:= 0 to Pred(HedgehogsNumber) do

                            with Hedgehogs[i] do

                                begin

                                if Effects[heFrozen] > 255 then Effects[heFrozen]:= max(255,Effects[heFrozen]-50000);

                                if (Gear <> nil) and (Effects[heFrozen] < 256) and (CurrentTeam^.Hedgehogs[CurrentTeam^.CurrHedgehog].Effects[heFrozen] > 255) then

                                    CurrHedgehog:= i

                                end;

            if (CurrentTeam^.Hedgehogs[CurrentTeam^.CurrHedgehog].Gear = nil) or (CurrentTeam^.Hedgehogs[CurrentTeam^.CurrHedgehog].Effects[heFrozen] > 255) then

                inc(CurrentTeam^.Clan^.TurnNumber);

            end

until (CurrentTeam^.Hedgehogs[CurrentTeam^.CurrHedgehog].Gear <> nil) and (CurrentTeam^.Hedgehogs[CurrentTeam^.CurrHedgehog].Effects[heFrozen] < 256);

SwitchCurrentHedgehog(@(CurrentTeam^.Hedgehogs[CurrentTeam^.CurrHedgehog]));

{$IFDEF USE_TOUCH_INTERFACE}

if (Ammoz[CurrentHedgehog^.CurAmmoType].Ammo.Propz and ammoprop_NoCrosshair) = 0 then

    begin

    if not(arrowUp.show) then

        begin

        animateWidget(@arrowUp, true, true);

        animateWidget(@arrowDown, true, true);

        end;

    end

else

    if arrowUp.show then

        begin

        animateWidget(@arrowUp, true, false);

        animateWidget(@arrowDown, true, false);

        end;

{$ENDIF}

AmmoMenuInvalidated:= true;

end;

procedure AfterSwitchHedgehog;

var i, t: LongInt;

    CurWeapon: PAmmo;

    w: real;

    vg: PVisualGear;

    g: PGear;

    s: ansistring;

begin

if PlacingHogs then

    begin

    PlacingHogs:= false;

    for t:= 0 to Pred(TeamsCount) do

        for i:= 0 to cMaxHHIndex do

            if (TeamsArray[t]^.Hedgehogs[i].Gear <> nil) and (TeamsArray[t]^.Hedgehogs[i].Unplaced) then

                PlacingHogs:= true;

    if not PlacingHogs then // Reset  various things I mucked with

        begin

        for i:= 0 to ClansCount do

            if ClansArray[i] <> nil then

                ClansArray[i]^.TurnNumber:= 0;

        ResetWeapons

        end

    end;

inc(CurrentTeam^.Clan^.TurnNumber);

with CurrentTeam^.Clan^ do

    for t:= 0 to Pred(TeamsNumber) do

        with Teams[t]^ do

            for i:= 0 to Pred(HedgehogsNumber) do

                with Hedgehogs[i] do

                    if Effects[heFrozen] > 255 then

                        Effects[heFrozen]:= max(255,Effects[heFrozen]-50000);

CurWeapon:= GetCurAmmoEntry(CurrentHedgehog^);

if CurWeapon^.Count = 0 then

    CurrentHedgehog^.CurAmmoType:= amNothing;

with CurrentHedgehog^ do

    begin

    with Gear^ do

        begin

        Z:= cCurrHHZ;

        State:= gstHHDriven;

        Active:= true;

        Power:= 0;

        LastDamage:= nil

        end;

    RemoveGearFromList(Gear);

    InsertGearToList(Gear);

    FollowGear:= Gear

    end;

if (GameFlags and gfDisableWind) = 0 then

    begin

    cWindSpeed:= rndSign(GetRandomf * 2 * cMaxWindSpeed);

    w:= hwFloat2Float(cWindSpeed);

    vg:= AddVisualGear(0, 0, vgtSmoothWindBar);

    if vg <> nil then vg^.dAngle:= w;

    AddFileLog('Wind = '+FloatToStr(cWindSpeed));

    end;

ApplyAmmoChanges(CurrentHedgehog^);

if (not CurrentTeam^.ExtDriven) and (CurrentHedgehog^.BotLevel = 0) then

    SetBinds(CurrentTeam^.Binds);

if PlacingHogs then

    begin

    if CurrentHedgehog^.Unplaced then

        TurnTimeLeft:= 15000

    else TurnTimeLeft:= 0

    end

else

    begin

    if ((GameFlags and gfTagTeam) <> 0) and (not NextClan) then

        begin

        if TagTurnTimeLeft <> 0 then

            TurnTimeLeft:= TagTurnTimeLeft;

        TagTurnTimeLeft:= 0;

        end

    else

        begin

        TurnTimeLeft:= cHedgehogTurnTime;

        TagTurnTimeLeft:= 0;

        NextClan:= false;

        end;

    if (GameFlags and gfSwitchHog) <> 0 then

        begin

        g:= AddGear(hwRound(CurrentHedgehog^.Gear^.X), hwRound(CurrentHedgehog^.Gear^.Y), gtSwitcher, 0, _0, _0, 0);

        CurAmmoGear:= g;

        lastGearByUID:= g;

        end

    else

        bShowFinger:= true;

    end;

IsGetAwayTime:= false;

if (TurnTimeLeft > 0) and (CurrentHedgehog^.BotLevel = 0) then

    begin

    if CurrentTeam^.ExtDriven then

        begin

        if GetRandom(2) = 0 then

             AddVoice(sndIllGetYou, CurrentTeam^.voicepack)

        else AddVoice(sndJustYouWait, CurrentTeam^.voicepack)

        end

    else

        begin

        GetRandom(2); // needed to avoid extdriven desync

        AddVoice(sndYesSir, CurrentTeam^.voicepack);

        end;

    if cHedgehogTurnTime < 1000000 then

        ReadyTimeLeft:= cReadyDelay;

    s:= ansistring(CurrentTeam^.TeamName);

    AddCaption(FormatA(trmsg[sidReady], s), cWhiteColor, capgrpGameState)

    end

else

    begin

    if TurnTimeLeft > 0 then

        begin

        if GetRandom(2) = 0 then

             AddVoice(sndIllGetYou, CurrentTeam^.voicepack)

        else AddVoice(sndJustYouWait, CurrentTeam^.voicepack)

        end;

    ReadyTimeLeft:= 0

    end;

end;

function AddTeam(TeamColor: Longword): PTeam;

var team: PTeam;

    c: LongInt;

begin

if checkFails(TeamsCount < cMaxTeams, 'Too many teams', true) then exit(nil);

New(team);

if checkFails(team <> nil, 'AddTeam: team = nil', true) then exit(nil);

FillChar(team^, sizeof(TTeam), 0);

team^.AttackBar:= 2;

team^.CurrHedgehog:= 0;

team^.Flag:= 'hedgewars';

TeamsArray[TeamsCount]:= team;

inc(TeamsCount);

inc(VisibleTeamsCount);

team^.Binds:= DefaultBinds;

c:= Pred(ClansCount);

while (c >= 0) and (ClansArray[c]^.Color <> TeamColor) do dec(c);

if c < 0 then

    begin

    new(team^.Clan);

    FillChar(team^.Clan^, sizeof(TClan), 0);

    ClansArray[ClansCount]:= team^.Clan;

    inc(ClansCount);

    with team^.Clan^ do

        begin

        ClanIndex:= Pred(ClansCount);

        Color:= TeamColor;

        TagTeamIndex:= 0;

        Flawless:= true

        end

    end

else

    begin

    team^.Clan:= ClansArray[c];

    end;

with team^.Clan^ do

    begin

    Teams[TeamsNumber]:= team;

    inc(TeamsNumber)

    end;

// mirror changes into array for clans to spawn

SpawnClansArray:= ClansArray;

CurrentTeam:= team;

AddTeam:= team;

end;

procedure RecountAllTeamsHealth;

var t: LongInt;

begin

for t:= 0 to Pred(TeamsCount) do

    RecountTeamHealth(TeamsArray[t])

end;

procedure InitTeams;

var i, t: LongInt;

    th, h: LongInt;

begin

for t:= 0 to Pred(TeamsCount) do

    with TeamsArray[t]^ do

        begin

        if (not ExtDriven) and (Hedgehogs[0].BotLevel = 0) then

            begin

            LocalClan:= Clan^.ClanIndex;

            LocalTeam:= t;

            LocalAmmo:= Hedgehogs[0].AmmoStore

            end;

        th:= 0;

        for i:= 0 to cMaxHHIndex do

            if Hedgehogs[i].Gear <> nil then

                inc(th, Hedgehogs[i].Gear^.Health);

        if th > MaxTeamHealth then

            MaxTeamHealth:= th;

        // Some initial King buffs

        if (GameFlags and gfKing) <> 0 then

            begin

            Hedgehogs[0].King:= true;