first commit

[rbdr/pico-engine] / LuaVirtualMachine.cpp

// ---------------------------------------------------------------------------\r
// FILE NAME            : LuaVirtualMachine.cpp\r
// ---------------------------------------------------------------------------\r
// DESCRIPTION :\r
//\r
// Lua virtual machine implementation\r
// \r
// ---------------------------------------------------------------------------\r
// VERSION              : 1.00\r
// DATE                 : 1-Sep-2005\r
// AUTHOR               : Richard Shephard\r
// ---------------------------------------------------------------------------\r
// LIBRARY INCLUDE FILES\r
#include "luainc.h"\r
#include "luavirtualmachine.h"\r
#include <assert.h>\r
#include <string.h>\r
#include <stdio.h>\r
// ---------------------------------------------------------------------------\r
\r
\r
//============================================================================\r
// int printMessage\r
//---------------------------------------------------------------------------\r
// Prints a message to the console\r
//\r
// Parameter   Dir      Description\r
// ---------   ---      -----------\r
// lua         IN       State variable\r
//\r
// Return\r
// ------\r
// Number of return varaibles on the stack\r
//\r
//============================================================================\r
static int printMessage (lua_State *lua)\r
{\r
   assert (lua_isstring (lua,1));\r
\r
   const char *msg = lua_tostring (lua, 1);\r
\r
   // get caller\r
   lua_Debug ar;\r
   memset (&ar, 0, sizeof(ar));\r
   lua_getstack (lua, 1, &ar);\r
   lua_getinfo (lua, "Snl", &ar);\r
\r
   // debug output\r
   const char *str = ar.source;\r
   printf ("script: %s -- at %s(%d)\n", msg, str, ar.currentline);\r
   return 0;\r
}\r
\r
//============================================================================\r
// CLuaVirtualMachine::CLuaVirtualMachine\r
//---------------------------------------------------------------------------\r
// Constructor. Setups the default VM state\r
//\r
// Parameter   Dir      Description\r
// ---------   ---      -----------\r
// \r
//\r
// Return\r
// ------\r
// None.\r
//\r
//============================================================================\r
CLuaVirtualMachine::CLuaVirtualMachine (void) : m_pState (NULL), m_pDbg (NULL)\r
{\r
   m_fIsOk = false;\r
}\r
\r
//============================================================================\r
// CLuaVirtualMachine::CLuaVirtualMachine\r
//---------------------------------------------------------------------------\r
// Destructor. Closes the VM\r
//\r
// Parameter   Dir      Description\r
// ---------   ---      -----------\r
// \r
//\r
// Return\r
// ------\r
// None.\r
//\r
//============================================================================\r
CLuaVirtualMachine::~CLuaVirtualMachine (void)\r
{\r
   if (m_pState != NULL)\r
   {\r
      lua_close (m_pState);\r
   }\r
}\r
\r
//============================================================================\r
// CLuaVirtualMachine::Panic\r
//---------------------------------------------------------------------------\r
// When things in Lua go wrong (ever called in protected mode??)\r
//\r
// Parameter   Dir      Description\r
// ---------   ---      -----------\r
// lua         IN       State variable\r
//\r
// Return\r
// ------\r
// None.\r
//\r
//============================================================================\r
void CLuaVirtualMachine::Panic (lua_State *lua)\r
{\r
}\r
\r
//============================================================================\r
// bool CLuaVirtualMachine::InitialiseVM\r
//---------------------------------------------------------------------------\r
// Initialises the VM, open lua, makes sure things are OK\r
//\r
// Parameter   Dir      Description\r
// ---------   ---      -----------\r
// None.\r
//\r
// Return\r
// ------\r
// Success.\r
//\r
//============================================================================\r
bool CLuaVirtualMachine::InitialiseVM (void)\r
{\r
   // Open Lua!\r
   if (Ok ()) DestroyVM ();\r
\r
   m_pState = lua_open ();\r
\r
   if (m_pState) \r
   {\r
      m_fIsOk = true;\r
\r
      // Load util libs into lua\r
      luaL_openlibs (m_pState);\r
\r
      // setup global printing (trace)\r
      lua_pushcclosure (m_pState, printMessage, 0);\r
      lua_setglobal (m_pState, "trace");\r
\r
      lua_atpanic (m_pState, (lua_CFunction) CLuaVirtualMachine::Panic);\r
\r
      return true;\r
   }\r
\r
   return false;\r
}\r
\r
//============================================================================\r
// bool CLuaVirtualMachine::DestroyVM\r
//---------------------------------------------------------------------------\r
// Clears the current Lua state\r
//\r
// Parameter   Dir      Description\r
// ---------   ---      -----------\r
// None.\r
//\r
// Return\r
// ------\r
// Success.\r
//\r
//============================================================================\r
bool CLuaVirtualMachine::DestroyVM (void)\r
{\r
   if (m_pState)\r
   {\r
      lua_close (m_pState);\r
      m_pState = NULL;\r
      m_fIsOk = false;\r
   }\r
   return true;\r
}\r
\r
\r
//============================================================================\r
// bool CLuaVirtualMachine::RunFile \r
//---------------------------------------------------------------------------\r
// Compiles and runs a lua script file\r
//\r
// Parameter   Dir      Description\r
// ---------   ---      -----------\r
// strFilename IN       Filename to compile and run\r
//\r
// Return\r
// ------\r
// Success.\r
//\r
//============================================================================\r
bool CLuaVirtualMachine::RunFile (const char *strFilename)\r
{\r
   bool fSuccess = false;\r
   int iErr = 0;\r
\r
   if ((iErr = luaL_loadfile (m_pState, strFilename)) == 0)\r
   {\r
      // Call main...\r
       if ((iErr = lua_pcall (m_pState, 0, LUA_MULTRET, 0)) == 0)\r
       {\r
          fSuccess = true;\r
       }\r
   }\r
\r
   if (fSuccess == false)\r
   {\r
      if (m_pDbg != NULL) m_pDbg->ErrorRun (iErr);\r
   }\r
\r
   return fSuccess;\r
}\r
\r
//============================================================================\r
// bool CLuaVirtualMachine::RunBuffer \r
//---------------------------------------------------------------------------\r
// Compiles and runs a pre-compiled data buffer\r
//\r
// Parameter   Dir      Description\r
// ---------   ---      -----------\r
// pbBuffer    IN       Buffer to run\r
// szLen       IN       Length of buffer\r
// strName     IN       Name of Buffer\r
//\r
// Return\r
// ------\r
// Success.\r
//\r
//============================================================================\r
bool CLuaVirtualMachine::RunBuffer (const unsigned char *pbBuffer, size_t szLen, const char *strName /* = NULL */)\r
{\r
   bool fSuccess = false;\r
   int iErr = 0;\r
\r
   if (strName == NULL)\r
   {\r
      strName = "Temp";\r
   }\r
\r
   if ((iErr = luaL_loadbuffer (m_pState, (const char *) pbBuffer, szLen, strName)) == 0)\r
   {\r
      // Call main...\r
       if ((iErr = lua_pcall (m_pState, 0, LUA_MULTRET, 0)) == 0)\r
       {\r
          fSuccess = true;\r
       }\r
   }\r
\r
   if (fSuccess == false)\r
   {\r
      if (m_pDbg != NULL) m_pDbg->ErrorRun (iErr);\r
   }\r
\r
   return fSuccess;   \r
\r
}\r
\r
//============================================================================\r
// CLuaVirtualMachine::CallFunction \r
//---------------------------------------------------------------------------\r
// Calls a function that is already on the stack\r
//\r
// Parameter   Dir      Description\r
// ---------   ---      -----------\r
// nArgs       IN       Args that are aleady on the stack\r
// nReturns    IN       Number of expected returns (will be on the stack)\r
//\r
// Return\r
// ------\r
// Success.\r
//\r
//============================================================================\r
bool CLuaVirtualMachine::CallFunction (int nArgs, int nReturns /* = 0 */)\r
{\r
   bool fSuccess = false;\r
   \r
   if (lua_isfunction (m_pState, -nArgs-1))\r
   {\r
      int iErr = 0;\r
      iErr = lua_pcall (m_pState, nArgs, nReturns, 0);\r
\r
      if (iErr == 0)\r
      {\r
         fSuccess = true;\r
      }\r
      else\r
      {\r
         if (m_pDbg != NULL) m_pDbg->ErrorRun (iErr);\r
      }\r
   }\r
\r
   return fSuccess;\r
}\r
\r