script.h

Go to the documentation of this file.

#pragma once

#include "util.h"
#include "framework.h"
#include <lua.h>
#include <lualib.h>
#include <lauxlib.h>
#include "script_debugger.h"
#include "script_export.h"

namespace yappy {
namespace lua {

class LuaError : public std::runtime_error {
public:
    LuaError(const std::string &msg, lua_State *L);
    const char *what() const override
    { return m_what.c_str(); }
private:
    std::string m_what;
};

class Lua : private util::noncopyable {
public:
    Lua(bool debugEnable, size_t maxHeapSize, size_t initHeapSize = 1024 * 1024,
        int instLimit = 10 * 10000);
    ~Lua();

    lua_State *getLuaState() const;

    void loadTraceLib();
    void loadSysLib();
    void loadRandLib();
    void loadResourceLib(framework::Application *app);
    void loadGraphLib(framework::Application *app);
    void loadSoundLib(framework::Application *app);

    void loadFile(const wchar_t *fileName, bool autoBreak, bool prot = true);

    struct doNothing {
        void operator ()(lua_State *L) {}
    };

    template <class ParamFunc = doNothing, class RetFunc = doNothing>
    void callGlobal(const char *funcName, bool autoBreak,
        ParamFunc pushArgFunc = doNothing(), int narg = 0,
        RetFunc getRetFunc = doNothing(), int nret = 0)
    {
        lua_State *L = m_lua.get();
        lua_getglobal(L, funcName);
        // push args
        pushArgFunc(L);
        // pcall
        pcallInternal(narg, nret, autoBreak);
        // get results
        getRetFunc(L);
        // clear stack
        lua_settop(L, 0);
    }

private:
    static const DWORD HeapOption = HEAP_NO_SERIALIZE;

    struct LuaDeleter {
        void operator()(lua_State *L);
    };

    bool m_debugEnable;
    util::HeapPtr m_heap;
    std::unique_ptr<lua_State, LuaDeleter> m_lua;
    std::unique_ptr<debugger::LuaDebugger> m_dbg;

    // custom allocator
    static void *luaAlloc(void *ud, void *ptr, size_t osize, size_t nsize);

    void pcallInternal(int narg, int nret, bool autoBreak);
};

std::vector<std::string> luaValueToStrList(
    lua_State *L, int ind, int maxDepth);

}   // namespace lua
}   // namespace yappy