urho3d/Extras/LuaScript/LuaScript.cpp

//
// Copyright (c) 2008-2013 the Urho3D project.
//
// Permission is hereby granted, free of charge, to any person obtaining a copy
// of this software and associated documentation files (the "Software"), to deal
// in the Software without restriction, including without limitation the rights
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
// copies of the Software, and to permit persons to whom the Software is
// furnished to do so, subject to the following conditions:
//
// The above copyright notice and this permission notice shall be included in
// all copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
// THE SOFTWARE.
//

#include "Precompiled.h"
#include "Context.h"
#include "EngineEvents.h"
#include "File.h"
#include "Log.h"
#include "Profiler.h"
#include "LuaScript.h"
#include "ResourceCache.h"
#include "Scene.h"

extern "C"
{
#include <lua.h>
#include <lualib.h>
#include <lauxlib.h>
}
#include "tolua++.h"

#include "DebugNew.h"

extern int tolua_Audio_open(lua_State*);
extern int tolua_Container_open(lua_State*);
extern int tolua_Core_open(lua_State*);
extern int tolua_Engine_open(lua_State*);
extern int tolua_Graphics_open(lua_State*);
extern int tolua_Input_open(lua_State*);
extern int tolua_IO_open(lua_State*);
extern int tolua_Math_open(lua_State*);
extern int tolua_Navigation_open(lua_State*);
extern int tolua_Network_open(lua_State*);
extern int tolua_Physics_open(lua_State*);
extern int tolua_Resource_open(lua_State*);
extern int tolua_Scene_open(lua_State*);
extern int tolua_UI_open(lua_State*);
extern int  tolua_LuaScript_open(lua_State*);
namespace Urho3D
{

OBJECTTYPESTATIC(LuaScript);

static Context* scriptContext_ = 0;

LuaScript::LuaScript(Context* context) :
    Object(context),
    luaState_(0)
{
    scriptContext_ = context_;

    luaState_ = luaL_newstate();
    if (!luaState_)
    {
        LOGERROR("Could not create Lua state");
        return;
    }

    luaL_openlibs(luaState_);
    ReplacePrintFunction();

    tolua_Audio_open(luaState_);
    tolua_Container_open(luaState_);
    tolua_Core_open(luaState_);
    tolua_Engine_open(luaState_);
    tolua_Graphics_open(luaState_);
    tolua_Input_open(luaState_);
    tolua_IO_open(luaState_);
    tolua_Math_open(luaState_);
    tolua_Navigation_open(luaState_);
    tolua_Network_open(luaState_);
    tolua_Physics_open(luaState_);
    tolua_Resource_open(luaState_);
    tolua_Scene_open(luaState_);
    tolua_UI_open(luaState_);
    tolua_LuaScript_open(luaState_);

    // Subscribe to console commands
    SubscribeToEvent(E_CONSOLECOMMAND, HANDLER(LuaScript, HandleConsoleCommand));
}

LuaScript::~LuaScript()
{
    lua_close(luaState_);
}

bool LuaScript::ExecuteFile(const char* fileName)
{
    if (!fileName)
        return false;

    PROFILE(ExecuteFile);

    ResourceCache* cache = GetSubsystem<ResourceCache>();
    if (!cache)
        return false;
    
    SharedPtr<File> file = cache->GetFile(fileName);
    if (!file)
        return false;

    unsigned size = file->GetSize();
    char* buffer = new char[size];
    file->Read(buffer, size);
    
    int top = lua_gettop(luaState_);

    int error = luaL_loadbuffer(luaState_, buffer, size, fileName);
    delete [] buffer;

    if (error)
    {
        String message = lua_tostring(luaState_, -1);
        lua_settop(luaState_, top);
        LOGRAW(String("Lua: Unable to execute Lua file '") + fileName + "'. ");
        LOGRAW("Lua: " + message);
        return false;
    }

    if (lua_pcall(luaState_, 0, 0, 0))
    {
        String message = lua_tostring(luaState_, -1);
        lua_settop(luaState_, top);
        LOGRAW(String("Lua: Unable to execute Lua script file '") + fileName + "'.");
        LOGRAW("Lua: " + message);
        return false;
    }
    
    lua_settop(luaState_, top);

    return true;
}

bool LuaScript::ExecuteString(const char* string)
{
    if (string == 0)
        return false;

    PROFILE(ExecuteString);

    int top = lua_gettop(luaState_);
    if (luaL_dostring(luaState_, string) != 0)
    {
        String message = lua_tostring(luaState_, -1);
        lua_settop(luaState_, top);
        LOGRAW(String("Lua: Unable to execute Lua string '") + string + "'.");
        LOGRAW("Lua: " + message);
        return false;
    }

    lua_settop(luaState_, top);

    return true;
}

bool LuaScript::ExecuteFunction(const char* funcName)
{
    if (funcName == 0)
        return false;

    PROFILE(ExecuteFunction);

    int top = lua_gettop(luaState_);
    lua_getglobal(luaState_, funcName);    
    if (!lua_isfunction(luaState_, -1))
    {
        lua_settop(luaState_, top);
        LOGRAW(String("Lua: Unable to get lua global: '") + funcName + "'.");
        return false;
    }

    if (lua_pcall(luaState_, 0, 0, 0))
    {
        String message = lua_tostring(luaState_, -1);
        lua_settop(luaState_, top);
        LOGRAW(String("Lua: Unable to execute function '") + funcName + "'.");
        LOGRAW("Lua: " + message);
        return false;
    }

    return true;
}

void LuaScript::SetDefaultScene( Scene* scene )
{
    defaultScene_ = scene;
}

Scene* LuaScript::GetDefaultScene() const
{
    return defaultScene_;
}

void LuaScript::SubscribeLuaEvent(const char* event, const char* funcName)
{
    StringHash eventType(event);
    eventFunctionMap_[eventType].Push(String(funcName));

    SubscribeToEvent(eventType, HANDLER(LuaScript, HandleEvent));
}

void LuaScript::ReplacePrintFunction()
{
    static const struct luaL_reg printlib[] = 
    {
        {"print", &LuaScript::Print},
        { NULL, NULL}
    };

    lua_getglobal(luaState_, "_G");
    luaL_register(luaState_, NULL, printlib);
    lua_pop(luaState_, 1);
}

int LuaScript::Print(lua_State *L)
{
    String string;
    int n = lua_gettop(L);
    lua_getglobal(L, "tostring");    
    for (int i = 1; i <= n; i++)
    {
        const char *s;
        lua_pushvalue(L, -1);  /* function to be called */
        lua_pushvalue(L, i);   /* value to print */
        lua_call(L, 1, 1);
        s = lua_tostring(L, -1);  /* get result */
        if (s == NULL)
            return luaL_error(L, LUA_QL("tostring") " must return a string to " LUA_QL("print"));

        if (i > 1)
            string.Append("    "); // fputs("\t", stdout);

        string.Append(s); // fputs(s, stdout);
        lua_pop(L, 1);  /* pop result */
    }

    LOGRAW(String("Lua: ") + string); // fputs("\n", stdout);

    return 0;
}

int LuaScript::PCallCallback(lua_State* L)
{
    String message = lua_tostring(L, -1);
    LOGRAW("Lua pcall error: " + message);
    return 0;
}

void LuaScript::HandleEvent(StringHash eventType, VariantMap& eventData)
{
    HashMap<StringHash, Vector<String> >::ConstIterator it = eventFunctionMap_.Find(eventType);
    if (it == eventFunctionMap_.End())
        return;

    for (unsigned i = 0; i < it->second_.Size(); ++i)
    {
        const String& funcName = it->second_[i];        
        int top = lua_gettop(luaState_);
        lua_getglobal(luaState_, funcName.CString());    
        if (!lua_isfunction(luaState_, -1))
        {
            lua_settop(luaState_, top);
            LOGRAW(String("Lua: Unable to get lua global: '") + funcName + "'.");
            return;
        }

        tolua_pushusertype(luaState_, (void*)&eventType, "StringHash");
        tolua_pushusertype(luaState_, (void*)&eventData, "VariantMap");
        if (lua_pcall(luaState_, 2, 0, 0))
        {
            String message = lua_tostring(luaState_, -1);
            lua_settop(luaState_, top);
            LOGRAW(String("Lua: Unable to execute function '") + funcName + "'.");
            LOGRAW("Lua: " + message);
            return;
        }
    }
}

void LuaScript::HandleConsoleCommand(StringHash eventType, VariantMap& eventData)
{
    using namespace ConsoleCommand;
    ExecuteString(eventData[P_COMMAND].GetString().CString());
}

Context* GetLuaScriptContext()
{
    return scriptContext_;
}

void SubscribeToEvent(const char* eventType, const char* funcName)
{
    LuaScript* luaScript = scriptContext_->GetSubsystem<LuaScript>();
    luaScript->SubscribeLuaEvent(eventType, funcName);
}

}