int CLuaFunctionDefs::GetResourceRootElement ( lua_State* luaVM )
{
// Resource given?
CResource* pResource = NULL;
if ( lua_istype ( luaVM, 1, LUA_TLIGHTUSERDATA ) )
{
pResource = lua_toresource ( luaVM, 1 );
}
// No resource given, get this resource's root
else if ( lua_istype ( luaVM, 1, LUA_TNONE ) )
{
// Find our vm and get the root
CLuaMain* pLuaMain = m_pLuaManager->GetVirtualMachine ( luaVM );
if ( pLuaMain )
{
pResource = pLuaMain->GetResource ();
}
}
else
m_pScriptDebugging->LogBadPointer ( luaVM, "getResourceRootElement", "resource", 1 );
// Did we find a resource?
if ( pResource )
{
// Grab the root element of it and return it if it existed
CClientEntity* pEntity = pResource->GetResourceEntity ();
if ( pEntity )
{
lua_pushelement ( luaVM, pEntity );
return 1;
}
}
else
m_pScriptDebugging->LogBadType ( luaVM, "getResourceRootElement" );
// Failed
lua_pushboolean ( luaVM, false );
return 1;
}
int CLuaFunctionDefs::GetResourceRootElement ( lua_State* luaVM )
{
// Verify arguments
CResource* pResource = NULL;
CScriptArgReader argStream ( luaVM );
argStream.ReadUserData ( pResource, NULL );
// No resource given, get this resource's root
if ( !argStream.HasErrors() )
{
if ( !pResource )
{
// Find our vm and get the root
CLuaMain* pLuaMain = m_pLuaManager->GetVirtualMachine ( luaVM );
if ( pLuaMain )
{
pResource = pLuaMain->GetResource ();
}
}
// Did we find a resource?
if ( pResource )
{
// Grab the root element of it and return it if it existed
CClientEntity* pEntity = pResource->GetResourceEntity ();
if ( pEntity )
{
lua_pushelement ( luaVM, pEntity );
return 1;
}
}
}
else
m_pScriptDebugging->LogCustom ( luaVM, argStream.GetFullErrorMessage() );
// Failed
lua_pushboolean ( luaVM, false );
return 1;
}
int CLuaFunctionDefs::Call ( lua_State* luaVM )
{
CResource * pResource = NULL;
SString strFunctionName = "";
CScriptArgReader argStream ( luaVM );
argStream.ReadUserData ( pResource );
argStream.ReadString ( strFunctionName );
if ( !argStream.HasErrors ( ) )
{
// Grab our VM
CLuaMain* pLuaMain = m_pLuaManager->GetVirtualMachine ( luaVM );
if ( pLuaMain )
{
// Grab this resource
CResource* pThisResource = pLuaMain->GetResource ();
if ( pThisResource )
{
if ( pResource )
{
//Get the target Lua VM
lua_State* targetLuaVM = pResource->GetVM()->GetVM();
// Read out the vargs
CLuaArguments args;
args.ReadArguments ( luaVM, 3 );
CLuaArguments returns;
LUA_CHECKSTACK ( targetLuaVM, 1 ); // Ensure some room
//Lets grab the original hidden variables so we can restore them later
lua_getglobal ( targetLuaVM, "sourceResource" );
CLuaArgument OldResource ( luaVM, -1 );
lua_pop( targetLuaVM, 1 );
lua_getglobal ( targetLuaVM, "sourceResourceRoot" );
CLuaArgument OldResourceRoot ( luaVM, -1 );
lua_pop( targetLuaVM, 1 );
//Set the new values for the current sourceResource, and sourceResourceRoot
lua_pushresource ( targetLuaVM, pThisResource );
lua_setglobal ( targetLuaVM, "sourceResource" );
lua_pushelement ( targetLuaVM, pThisResource->GetResourceEntity() );
lua_setglobal ( targetLuaVM, "sourceResourceRoot" );
// Call the exported function with the given name and the args
if ( pResource->CallExportedFunction ( strFunctionName, args, returns, *pThisResource ) )
{
// Push return arguments
returns.PushArguments ( luaVM );
//Restore the old variables
OldResource.Push ( targetLuaVM );
lua_setglobal ( targetLuaVM, "sourceResource" );
OldResourceRoot.Push ( targetLuaVM );
lua_setglobal ( targetLuaVM, "sourceResourceRoot" );
return returns.Count ();
}
else
{
//Restore the old variables
OldResource.Push ( targetLuaVM );
lua_setglobal ( targetLuaVM, "sourceResource" );
OldResourceRoot.Push ( targetLuaVM );
lua_setglobal ( targetLuaVM, "sourceResourceRoot" );
m_pScriptDebugging->LogError ( luaVM, "call: failed to call '%s:%s'", pResource->GetName (), *strFunctionName );
}
}
else
{
m_pScriptDebugging->LogBadType ( luaVM );
}
}
}
}
else
m_pScriptDebugging->LogCustom ( luaVM, argStream.GetFullErrorMessage() );
// Failed
lua_pushboolean ( luaVM, false );
return 1;
}
int CLuaFunctionDefs::Call ( lua_State* luaVM )
{
// Grab our VM
CLuaMain* pLuaMain = m_pLuaManager->GetVirtualMachine ( luaVM );
if ( pLuaMain )
{
// Grab this resource
CResource* pThisResource = pLuaMain->GetResource ();
if ( pThisResource )
{
// Typechecking
if ( lua_istype ( luaVM, 1, LUA_TLIGHTUSERDATA ) &&
lua_istype ( luaVM, 2, LUA_TSTRING ) )
{
// Grab the resource
CResource* pResource = lua_toresource ( luaVM, 1 );
if ( pResource )
{
//Get the target Lua VM
lua_State* targetLuaVM = pResource->GetVM()->GetVM();
// The function name
const char* szFunctionName = lua_tostring ( luaVM, 2 );
// Read out the vargs
CLuaArguments args;
args.ReadArguments ( luaVM, 3 );
CLuaArguments returns;
//Lets grab the original hidden variables so we can restore them later
lua_getglobal ( targetLuaVM, "sourceResource" );
CLuaArgument OldResource ( luaVM, -1 );
lua_getglobal ( targetLuaVM, "sourceResourceRoot" );
CLuaArgument OldResourceRoot ( luaVM, -1 );
//Set the new values for the current sourceResource, and sourceResourceRoot
lua_pushresource ( targetLuaVM, pThisResource );
lua_setglobal ( targetLuaVM, "sourceResource" );
lua_pushelement ( targetLuaVM, pThisResource->GetResourceEntity() );
lua_setglobal ( targetLuaVM, "sourceResourceRoot" );
// Call the exported function with the given name and the args
if ( pResource->CallExportedFunction ( szFunctionName, args, returns, *pThisResource ) )
{
// Push return arguments
returns.PushArguments ( luaVM );
//Restore the old variables
OldResource.Push ( targetLuaVM );
lua_setglobal ( targetLuaVM, "sourceResource" );
OldResourceRoot.Push ( targetLuaVM );
lua_setglobal ( targetLuaVM, "sourceResourceRoot" );
return returns.Count ();
}
else
{
//Restore the old variables
OldResource.Push ( targetLuaVM );
lua_setglobal ( targetLuaVM, "sourceResource" );
OldResourceRoot.Push ( targetLuaVM );
lua_setglobal ( targetLuaVM, "sourceResourceRoot" );
m_pScriptDebugging->LogError ( luaVM, "call: failed to call '%s:%s'", pResource->GetName (), szFunctionName );
}
}
else
{
m_pScriptDebugging->LogBadPointer ( luaVM, "call", "resource", 1 );
}
}
else
{
m_pScriptDebugging->LogBadType ( luaVM, "call" );
}
}
}
// Failed
lua_pushboolean ( luaVM, false );
return 1;
}
