(*

 * Hedgewars, a free turn based strategy game

 * Copyright (c) 2006-2008 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., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA

 *)

unit uAmmos;

interface

uses uConsts, uTeams;

{$INCLUDE options.inc}

procedure AddAmmoStore(s: shortstring);

procedure AssignStores;

procedure AddAmmo(var Hedgehog: THedgehog; ammo: TAmmoType);

function  HHHasAmmo(var Hedgehog: THedgehog; Ammo: TAmmoType): boolean;

procedure PackAmmo(Ammo: PHHAmmo; Slot: LongInt);

procedure OnUsedAmmo(var Hedgehog: THedgehog);

procedure ApplyAmmoChanges(var Hedgehog: THedgehog);

procedure SetWeapon(weap: TAmmoType);

procedure DisableSomeWeapons;

implementation

uses uMisc, uGears, uWorld, uLocale, uConsole;

type TAmmoCounts = array[TAmmoType] of Longword;

var StoresList: array[0..Pred(cMaxHHs)] of PHHAmmo;

    StoreCnt: Longword = 0;

procedure FillAmmoStore(Ammo: PHHAmmo; var cnts: TAmmoCounts);

var mi: array[0..cMaxSlotIndex] of byte;

    a: TAmmoType;

begin

FillChar(mi, sizeof(mi), 0);

FillChar(Ammo^, sizeof(Ammo^), 0);

for a:= Low(TAmmoType) to High(TAmmoType) do

    if cnts[a] > 0 then

       begin

       TryDo(mi[Ammoz[a].Slot] <= cMaxSlotAmmoIndex, 'Ammo slot overflow', true);

       Ammo^[Ammoz[a].Slot, mi[Ammoz[a].Slot]]:= Ammoz[a].Ammo;

       Ammo^[Ammoz[a].Slot, mi[Ammoz[a].Slot]].Count:= cnts[a];

       inc(mi[Ammoz[a].Slot])

       end

end;

procedure AddAmmoStore(s: shortstring);

var cnt: Longword;

    a: TAmmoType;

    ammos: TAmmoCounts;

begin

TryDo(byte(s[0]) = byte(ord(High(TAmmoType)) + 1), 'Invalid ammo scheme (incompatible frontend)', true);

inc(StoreCnt);

TryDo(StoreCnt <= cMaxHHs, 'Ammo stores overflow', true);

new(StoresList[Pred(StoreCnt)]);

for a:= Low(TAmmoType) to High(TAmmoType) do

    begin

    cnt:= byte(s[ord(a) + 1]) - byte('0');

    if cnt = 9 then

        begin

        cnt:= AMMO_INFINITE;

        Ammoz[a].Probability:= 0 

        end;

    if ((a = amLowGravity) and ((GameFlags and gfLowGravity) <> 0)) or

       ((a = amInvulnerable) and ((GameFlags and gfInvulnerable) <> 0)) or

       ((a = amLaserSight) and ((GameFlags and gfLaserSight) <> 0)) then

        begin

        cnt:= 0;

        Ammoz[a].Probability:= 0 

        end;

    ammos[a]:= cnt

    end;

FillAmmoStore(StoresList[Pred(StoreCnt)], ammos)

end;

function GetAmmoByNum(num: Longword): PHHAmmo;

begin

TryDo(num < StoreCnt, 'Invalid store number', true);

exit(StoresList[num])

end;

procedure AssignStores;

var t: LongInt;

    i: Longword;

begin

for t:= 0 to Pred(TeamsCount) do

   with TeamsArray[t]^ do

      begin

      for i:= 0 to cMaxHHIndex do

          if Hedgehogs[i].Gear <> nil then

             Hedgehogs[i].Ammo:= GetAmmoByNum(Hedgehogs[i].AmmoStore);

      end

end;

procedure AddAmmo(var Hedgehog: THedgehog; ammo: TAmmoType);

var ammos: TAmmoCounts;

    slot, ami: LongInt;

    hhammo: PHHAmmo;

begin

FillChar(ammos, sizeof(ammos), 0);

hhammo:= Hedgehog.Ammo;

for slot:= 0 to cMaxSlotIndex do

    for ami:= 0 to cMaxSlotAmmoIndex do

        if hhammo^[slot, ami].Count > 0 then

           ammos[hhammo^[slot, ami].AmmoType]:= hhammo^[slot, ami].Count;

if ammos[ammo] <> AMMO_INFINITE then

   begin

   inc(ammos[ammo], Ammoz[ammo].NumberInCase);

   if ammos[ammo] > AMMO_INFINITE then ammos[ammo]:= AMMO_INFINITE

   end;

FillAmmoStore(hhammo, ammos)

end;

procedure PackAmmo(Ammo: PHHAmmo; Slot: LongInt);

var ami: LongInt;

    b: boolean;

begin

    repeat

      b:= false;

      ami:= 0;

      while (not b) and (ami < cMaxSlotAmmoIndex) do

          if (Ammo^[Slot, ami].Count = 0)

             and (Ammo^[Slot, ami + 1].Count > 0) then b:= true

                                                  else inc(ami);

      if b then // there's a free item in ammo stack

         begin

         Ammo^[Slot, ami]:= Ammo^[Slot, ami + 1];

         Ammo^[Slot, ami + 1].Count:= 0

         end;

    until not b;

end;

procedure OnUsedAmmo(var Hedgehog: THedgehog);

begin

with Hedgehog do

	begin

	with Ammo^[CurSlot, CurAmmo] do

		if Count <> AMMO_INFINITE then

			begin

			dec(Count);

			if Count = 0 then

				begin

				PackAmmo(Ammo, CurSlot);

				end

			end

	end

end;

function  HHHasAmmo(var Hedgehog: THedgehog; Ammo: TAmmoType): boolean;

var slot, ami: LongInt;

begin

Slot:= Ammoz[Ammo].Slot;

ami:= 0;

while (ami <= cMaxSlotAmmoIndex) do

      begin

      with Hedgehog.Ammo^[Slot, ami] do

            if (AmmoType = Ammo) then

               exit((Count > 0) and (Hedgehog.Team^.Clan^.TurnNumber > Ammoz[AmmoType].SkipTurns));

      inc(ami)

      end;

HHHasAmmo:= false

end;

procedure ApplyAmmoChanges(var Hedgehog: THedgehog);

var s: shortstring;

begin

TargetPoint.X:= NoPointX;

with Hedgehog do

end;

begin

with Hedgehog do

     if ((Ammo^[CurSlot, CurAmmo].Propz and ammoprop_DontHold) <> 0) or

        (Ammoz[Ammo^[CurSlot, CurAmmo].AmmoType].SkipTurns - CurrentTeam^.Clan^.TurnNumber >= 0) then

        begin

        CurAmmo:= 0;

        CurSlot:= 0;

        while (CurSlot <= cMaxSlotIndex) and 

              (Ammo^[CurSlot, CurAmmo].Count = 0) and

              (Ammoz[Ammo^[CurSlot, CurAmmo].AmmoType].SkipTurns - CurrentTeam^.Clan^.TurnNumber >= 0)

              do inc(CurSlot)

        end

end;

procedure SetWeapon(weap: TAmmoType);

begin

ParseCommand('/setweap ' + char(weap), true)

end;

procedure DisableSomeWeapons;

var i, slot, a: Longword;

	t: TAmmoType;

begin

for i:= 0 to Pred(StoreCnt) do

	for slot:= 0 to cMaxSlotIndex do

		begin

		for a:= 0 to cMaxSlotAmmoIndex do

			with StoresList[i]^[slot, a] do

				if (Propz and ammoprop_NotBorder) <> 0 then Count:= 0;

		PackAmmo(StoresList[i], slot)

		end;

for t:= Low(TAmmoType) to High(TAmmoType) do

	if (Ammoz[t].Ammo.Propz and ammoprop_NotBorder) <> 0 then Ammoz[t].Probability:= 0

end;

end.