(*

 * Hedgewars, a free turn based strategy game

 * Copyright (c) 2004-2011 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

 *)

{$INCLUDE "options.inc"}

unit uScript;

(*

 * This unit defines, implements and registers functions and

 * variables/constants bindings for usage in Lua scripts.

 *

 * Please keep http://code.google.com/p/hedgewars/wiki/LuaAPI up to date!

 *

 * Note: If you add a new function, make sure to test if _all_ parameters

 *       work as intended! (Especially conversions errors can sneak in

 *       unnoticed and render the parameter useless!)

 *)

interface

procedure ScriptPrintStack;

procedure ScriptClearStack;

procedure ScriptLoad(name : shortstring);

procedure ScriptOnGameInit;

procedure ScriptOnScreenResize();

procedure ScriptCall(fname : shortstring);

function ScriptCall(fname : shortstring; par1: LongInt) : LongInt;

function ScriptCall(fname : shortstring; par1, par2: LongInt) : LongInt;

function ScriptCall(fname : shortstring; par1, par2, par3: LongInt) : LongInt;

function ScriptCall(fname : shortstring; par1, par2, par3, par4 : LongInt) : LongInt;

function ScriptExists(fname : shortstring) : boolean;

function ParseCommandOverride(key, value : shortstring) : shortstring;

procedure initModule;

procedure freeModule;

implementation

{$IFNDEF LUA_DISABLED}

    uConsole,

    uConsts,

    uVisualGears,

    uGears,

    uFloat,

    uWorld,

    uAmmos,

    uSound,

    uChat,

    uStats,

    uStore,

    uRandom,

    uTypes,

    uVariables,

    uCommands,

    uUtils,

    uCaptions,

    uDebug,

    uCollisions,

    uRenderUtils,

    uTextures,

    uLandGraphics,

    SDLh,

    sysutils; 

var luaState : Plua_State;

    ScriptAmmoLoadout : shortstring;

    ScriptAmmoProbability : shortstring;

    ScriptAmmoDelay : shortstring;

    ScriptAmmoReinforcement : shortstring;

    ScriptLoaded : boolean;

procedure ScriptPrepareAmmoStore; forward;

procedure ScriptApplyAmmoStore; forward;

procedure ScriptSetAmmo(ammo : TAmmoType; count, propability, delay, reinforcement: Byte); forward;

procedure LuaError(s: shortstring);

begin

    WriteLnToConsole(s);

    AddChatString(#5 + s);

end;

// wrapped calls //

// functions called from Lua:

// function(L : Plua_State) : LongInt; Cdecl;

// where L contains the state, returns the number of return values on the stack

// call lua_gettop(L) to receive number of parameters passed

function lc_band(L: PLua_State): LongInt; Cdecl;

begin

    if lua_gettop(L) <> 2 then 

        begin

        LuaError('Lua: Wrong number of parameters passed to band!');

        lua_pushnil(L);

        end

    else lua_pushinteger(L, lua_tointeger(L, 2) and lua_tointeger(L, 1));

    lc_band := 1;

end;

function lc_bor(L: PLua_State): LongInt; Cdecl;

begin

    if lua_gettop(L) <> 2 then 

        begin

        LuaError('Lua: Wrong number of parameters passed to bor!');

        lua_pushnil(L);

        end

    else lua_pushinteger(L, lua_tointeger(L, 2) or lua_tointeger(L, 1));

    lc_bor := 1;

end;

function lc_bnot(L: PLua_State): LongInt; Cdecl;

begin

    if lua_gettop(L) <> 1 then 

        begin

        LuaError('Lua: Wrong number of parameters passed to bnot!');

        lua_pushnil(L);

        end

    else lua_pushinteger(L, not lua_tointeger(L, 1));

    lc_bnot := 1;

end;

function lc_getinputmask(L : Plua_State) : LongInt; Cdecl;

begin

    if lua_gettop(L) <> 0 then

        LuaError('Lua: Wrong number of parameters passed to GetInputMask!')

    else lua_pushinteger(L, InputMask);

    lc_getinputmask:= 1

end;

function lc_setinputmask(L : Plua_State) : LongInt; Cdecl;

begin

    if lua_gettop(L) <> 1 then

        LuaError('Lua: Wrong number of parameters passed to SetInputMask!')

    else InputMask:= lua_tointeger(L, 1);

    lc_setinputmask:= 0

end;

function lc_writelntoconsole(L : Plua_State) : LongInt; Cdecl;

begin

    if lua_gettop(L) = 1 then

        begin

        WriteLnToConsole('Lua: ' + lua_tostring(L ,1));

        end

    else

        LuaError('Lua: Wrong number of parameters passed to WriteLnToConsole!');

    lc_writelntoconsole:= 0;

end;

function lc_parsecommand(L : Plua_State) : LongInt; Cdecl;

begin

    if lua_gettop(L) = 1 then

        begin

        ParseCommand(lua_tostring(L ,1), true);

        end

    else

        LuaError('Lua: Wrong number of parameters passed to ParseCommand!');

    lc_parsecommand:= 0;

end;

function lc_showmission(L : Plua_State) : LongInt; Cdecl;

begin

    if lua_gettop(L) = 5 then

        begin

        ShowMission(lua_tostring(L, 1), lua_tostring(L, 2), lua_tostring(L, 3), lua_tointeger(L, 4), lua_tointeger(L, 5));

        end

    else

        LuaError('Lua: Wrong number of parameters passed to ShowMission!');

    lc_showmission:= 0;

end;

function lc_hidemission(L : Plua_State) : LongInt; Cdecl;

begin

    L:= L; // avoid compiler hint

    HideMission;

    lc_hidemission:= 0;

end;

function lc_addcaption(L : Plua_State) : LongInt; Cdecl;

begin

    if lua_gettop(L) = 1 then

        AddCaption(lua_tostring(L, 1), cWhiteColor, capgrpMessage)

    else if lua_gettop(L) = 3 then

        begin

        AddCaption(lua_tostring(L, 1), lua_tointeger(L, 2) shr 8, TCapGroup(lua_tointeger(L, 3)));

        end

    else

        LuaError('Lua: Wrong number of parameters passed to AddCaption!');

    lc_addcaption:= 0;

end;

function lc_campaignlock(L : Plua_State) : LongInt; Cdecl;

begin

    if lua_gettop(L) = 1 then

        begin

        // to be done

        end

    else

        LuaError('Lua: Wrong number of parameters passed to CampaignLock!');

    lc_campaignlock:= 0;

end;

function lc_campaignunlock(L : Plua_State) : LongInt; Cdecl;

begin

    if lua_gettop(L) = 1 then

        begin

        // to be done

        end

    else

        LuaError('Lua: Wrong number of parameters passed to CampaignUnlock!');

    lc_campaignunlock:= 0;

end;

function lc_spawnfakehealthcrate(L: Plua_State) : LongInt; Cdecl;

var gear: PGear;

begin

    if lua_gettop(L) <> 4 then begin

        LuaError('Lua: Wrong number of parameters passed to SpawnFakeHealthCrate!');

        lua_pushnil(L);

    end

    else begin

        gear := SpawnFakeCrateAt(lua_tointeger(L, 1), lua_tointeger(L, 2),

            HealthCrate, lua_toboolean(L, 3), lua_toboolean(L, 4));

        lua_pushinteger(L, gear^.uid);

    end;

    lc_spawnfakehealthcrate := 1;        

end;

function lc_spawnfakeammocrate(L: PLua_State): LongInt; Cdecl;

var gear: PGear;

begin

    if lua_gettop(L) <> 4 then begin

        LuaError('Lua: Wrong number of parameters passed to SpawnFakeAmmoCrate!');

        lua_pushnil(L);

    end

    else begin

        gear := SpawnFakeCrateAt(lua_tointeger(L, 1), lua_tointeger(L, 2),

            AmmoCrate, lua_toboolean(L, 3), lua_toboolean(L, 4));

        lua_pushinteger(L, gear^.uid);

    end;

    lc_spawnfakeammocrate := 1;

end;

function lc_spawnfakeutilitycrate(L: PLua_State): LongInt; Cdecl;

var gear: PGear;

begin

    if lua_gettop(L) <> 4 then begin

        LuaError('Lua: Wrong number of parameters passed to SpawnFakeUtilityCrate!');

        lua_pushnil(L);

    end

    else begin  

        gear := SpawnFakeCrateAt(lua_tointeger(L, 1), lua_tointeger(L, 2),

            UtilityCrate, lua_toboolean(L, 3), lua_toboolean(L, 4));

        lua_pushinteger(L, gear^.uid);

    end;

    lc_spawnfakeutilitycrate := 1;

end;

function lc_spawnhealthcrate(L: Plua_State) : LongInt; Cdecl;

var gear: PGear;

var health: LongInt;

begin

    if (lua_gettop(L) < 2) or (lua_gettop(L) > 3) then begin

        LuaError('Lua: Wrong number of parameters passed to SpawnHealthCrate!');

        lua_pushnil(L);

    end

    else begin

        if lua_gettop(L) = 3 then health:= lua_tointeger(L, 3)

        else health:= cHealthCaseAmount;

        gear := SpawnCustomCrateAt(lua_tointeger(L, 1), lua_tointeger(L, 2), HealthCrate, health);

        if gear <> nil then lua_pushinteger(L, gear^.uid)

        else lua_pushnil(L);

    end;

    lc_spawnhealthcrate := 1;        

end;

function lc_spawnammocrate(L: PLua_State): LongInt; Cdecl;

var gear: PGear;

begin

    if lua_gettop(L) <> 3 then begin

        LuaError('Lua: Wrong number of parameters passed to SpawnAmmoCrate!');

        lua_pushnil(L);

    end

    else begin

        gear := SpawnCustomCrateAt(lua_tointeger(L, 1), lua_tointeger(L, 2), AmmoCrate, lua_tointeger(L, 3));

        if gear <> nil then lua_pushinteger(L, gear^.uid)

        else lua_pushnil(L);

    end;

    lc_spawnammocrate := 1;

end;

function lc_spawnutilitycrate(L: PLua_State): LongInt; Cdecl;

var gear: PGear;

begin

    if lua_gettop(L) <> 3 then begin

        LuaError('Lua: Wrong number of parameters passed to SpawnUtilityCrate!');

        lua_pushnil(L);

    end

    else begin  

        gear := SpawnCustomCrateAt(lua_tointeger(L, 1), lua_tointeger(L, 2),

            UtilityCrate, lua_tointeger(L, 3));

        if gear <> nil then lua_pushinteger(L, gear^.uid)

        else lua_pushnil(L);

    end;

    lc_spawnutilitycrate := 1;

end;

function lc_addgear(L : Plua_State) : LongInt; Cdecl;

var gear : PGear;

    x, y, s, t: LongInt;

    dx, dy: hwFloat;

    gt: TGearType;

begin

    if lua_gettop(L) <> 7 then

        begin

        LuaError('Lua: Wrong number of parameters passed to AddGear!');

        lua_pushnil(L); // return value on stack (nil)

        end

    else

        begin

        x:= lua_tointeger(L, 1);

        y:= lua_tointeger(L, 2);

        gt:= TGearType(lua_tointeger(L, 3));

        s:= lua_tointeger(L, 4);

        dx:= int2hwFloat(lua_tointeger(L, 5)) / 1000000;

        dy:= int2hwFloat(lua_tointeger(L, 6)) / 1000000;

        t:= lua_tointeger(L, 7);

        gear:= AddGear(x, y, gt, s, dx, dy, t);

        lastGearByUID:= gear;

        lua_pushinteger(L, gear^.uid)

        end;

    lc_addgear:= 1; // 1 return value

end;

function lc_deletegear(L : Plua_State) : LongInt; Cdecl;

var gear : PGear;

begin

    if lua_gettop(L) <> 1 then

        begin

        LuaError('Lua: Wrong number of parameters passed to DeleteGear!');

        end

    else

        begin

        gear:= GearByUID(lua_tointeger(L, 1));

        if gear <> nil then DeleteGear(gear);

        end;

    lc_deletegear:= 0

end;

function lc_addvisualgear(L : Plua_State) : LongInt; Cdecl;

var vg : PVisualGear;

    x, y, s: LongInt;

    c: Boolean;

    vgt: TVisualGearType;

begin

    if lua_gettop(L) <> 5 then

        begin

        LuaError('Lua: Wrong number of parameters passed to AddVisualGear!');

        lua_pushnil(L); // return value on stack (nil)

        end

    else

        begin

        x:= lua_tointeger(L, 1);

        y:= lua_tointeger(L, 2);

        vgt:= TVisualGearType(lua_tointeger(L, 3));

        s:= lua_tointeger(L, 4);

        c:= lua_toboolean(L, 5);

        vg:= AddVisualGear(x, y, vgt, s, c);

        if vg <> nil then 

            begin

            lastVisualGearByUID:= vg;

            lua_pushinteger(L, vg^.uid)

            end

        else lua_pushinteger(L, 0)

        end;

    lc_addvisualgear:= 1; // 1 return value

end;

function lc_deletevisualgear(L : Plua_State) : LongInt; Cdecl;

var vg : PVisualGear;

begin

    if lua_gettop(L) <> 1 then

        begin

        LuaError('Lua: Wrong number of parameters passed to DeleteVisualGear!');

        end

    else

        begin

        vg:= VisualGearByUID(lua_tointeger(L, 1));

        if vg <> nil then DeleteVisualGear(vg);

        end;

    lc_deletevisualgear:= 0

end;

function lc_getvisualgearvalues(L : Plua_State) : LongInt; Cdecl;

var vg: PVisualGear;

begin

    if lua_gettop(L) <> 1 then

        begin

        LuaError('Lua: Wrong number of parameters passed to GetVisualGearValues!');

        lua_pushnil(L); lua_pushnil(L); lua_pushnil(L); lua_pushnil(L); lua_pushnil(L);

        lua_pushnil(L); lua_pushnil(L); lua_pushnil(L); lua_pushnil(L); lua_pushnil(L)

        end

    else

        begin

        vg:= VisualGearByUID(lua_tointeger(L, 1));

        if vg <> nil then

            begin

            lua_pushinteger(L, round(vg^.X));

            lua_pushinteger(L, round(vg^.Y));

            lua_pushnumber(L, vg^.dX);

            lua_pushnumber(L, vg^.dY);

            lua_pushnumber(L, vg^.Angle);

            lua_pushinteger(L, vg^.Frame);

            lua_pushinteger(L, vg^.FrameTicks);

            lua_pushinteger(L, vg^.State);

            lua_pushinteger(L, vg^.Timer);

            lua_pushinteger(L, vg^.Tint);

            end

        else

            begin

            lua_pushnil(L); lua_pushnil(L); lua_pushnil(L); lua_pushnil(L); lua_pushnil(L);

            lua_pushnil(L); lua_pushnil(L); lua_pushnil(L); lua_pushnil(L); lua_pushnil(L)

            end

        end;

    lc_getvisualgearvalues:= 10;

end;

function lc_setvisualgearvalues(L : Plua_State) : LongInt; Cdecl;

var vg : PVisualGear;

begin

    if lua_gettop(L) <> 11 then

        begin

        LuaError('Lua: Wrong number of parameters passed to SetVisualGearValues!');

        lua_pushnil(L); // return value on stack (nil)

        end

    else

        begin

        vg:= VisualGearByUID(lua_tointeger(L, 1));

        if vg <> nil then

            begin

            vg^.X:= lua_tointeger(L, 2);

            vg^.Y:= lua_tointeger(L, 3);

            vg^.dX:= lua_tonumber(L, 4);

            vg^.dY:= lua_tonumber(L, 5);

            vg^.Angle:= lua_tonumber(L, 6);

            vg^.Frame:= lua_tointeger(L, 7);

            if lua_tointeger(L, 8) <> 0 then vg^.FrameTicks:= lua_tointeger(L, 8);  // find a better way to do this. maybe need to break all these up.

            vg^.State:= lua_tointeger(L, 9);

            vg^.Timer:= lua_tointeger(L, 10);

            vg^.Tint:= lua_tointeger(L, 11);

            end

        end;

    lc_setvisualgearvalues:= 0;

end;

function lc_getfollowgear(L : Plua_State) : LongInt; Cdecl;

begin

    if lua_gettop(L) <> 0 then

        begin

        LuaError('Lua: Wrong number of parameters passed to GetFollowGear!');

        lua_pushnil(L); // return value on stack (nil)

        end

    else

        if FollowGear = nil then

            lua_pushnil(L)

        else

            lua_pushinteger(L, FollowGear^.uid);

    lc_getfollowgear:= 1; // 1 return value

end;