# HG changeset patch
# User Wuzzy <Wuzzy2@mail.ru>
# Date 1518206693 -3600
# Node ID c8b7bc04eb4c4a6bf93580980d8e064f113a1448
# Parent  1a1a514aef2e986a80962d409baf7353a286a0cc
Lua API: AddAmmo unselects weapons before it would remove current ammo from current hog

diff -r 1a1a514aef2e -r c8b7bc04eb4c ChangeLog.txt
--- a/ChangeLog.txt	Fri Feb 09 19:58:55 2018 +0100
+++ b/ChangeLog.txt	Fri Feb 09 21:04:53 2018 +0100
@@ -43,6 +43,7 @@
  + New call: SpawnSupplyCrate(x, y, content, [, amount]): Spawn ammo or utility crate, depending on content
  + New call: HealHog(gearUid, healthBoost[, showMessage[, tint]]): Heal hedgehog with graphical effects and message
  + New callback: onEndTurn(): Called at the end of a turn (when gears have settled)
+ * AddAmmo now automatically unselects weapon if it would remove current ammo from current hedgehog
  * Fix call: SetWeapon(amNothing) now unselects weapon
  * Fix global: TotalRounds was stuck at -1 for several turns
 
diff -r 1a1a514aef2e -r c8b7bc04eb4c hedgewars/uScript.pas
--- a/hedgewars/uScript.pas	Fri Feb 09 19:58:55 2018 +0100
+++ b/hedgewars/uScript.pas	Fri Feb 09 21:04:53 2018 +0100
@@ -1689,7 +1689,7 @@
 
 function lc_addammo(L : Plua_State) : LongInt; Cdecl;
 var gear : PGear;
-    at, n: LongInt;
+    at, n, c: LongInt;
 const
     call = 'AddAmmo';
     params = 'gearUid, ammoType [, ammoCount]';
@@ -1697,14 +1697,19 @@
     if CheckAndFetchParamCount(L, 2, 3, call, params, n) then
         begin
         at:= LuaToAmmoTypeOrd(L, 2, call, params);
-        if at >= 0 then
+        if (at >= 0) and (TAmmoType(at) <> amNothing) then
             begin
             gear:= GearByUID(Trunc(lua_tonumber(L, 1)));
             if (gear <> nil) and (gear^.Hedgehog <> nil) then
                 if n = 2 then
                     AddAmmo(gear^.Hedgehog^, TAmmoType(at))
                 else
-                    SetAmmo(gear^.Hedgehog^, TAmmoType(at), Trunc(lua_tonumber(L, 3)))
+                    begin
+                    c:= Trunc(lua_tonumber(L, 3));
+                    if (c = 0) and (CurrentHedgehog = gear^.Hedgehog) and (gear^.Hedgehog^.CurAmmoType = TAmmoType(at)) then
+                        ParseCommand('setweap ' + char(0), true, true);
+                    SetAmmo(gear^.Hedgehog^, TAmmoType(at), c);
+                    end;
             end;
         end;
     lc_addammo:= 0