WickedEngine/WickedEngine/wiResourceManager_BindLua.cpp

#include "wiResourceManager_BindLua.h"
#include "wiSound_BindLua.h"
#include "wiHelper.h"
#include "Texture_BindLua.h"
#include "wiRenderer.h"

using namespace wiGraphicsTypes;

const char wiResourceManager_BindLua::className[] = "Resource";

Luna<wiResourceManager_BindLua>::FunctionType wiResourceManager_BindLua::methods[] = {
	lunamethod(wiResourceManager_BindLua, Get),
	lunamethod(wiResourceManager_BindLua, Add),
	lunamethod(wiResourceManager_BindLua, Del),
	lunamethod(wiResourceManager_BindLua, List),
	{ NULL, NULL }
};
Luna<wiResourceManager_BindLua>::PropertyType wiResourceManager_BindLua::properties[] = {
	{ NULL, NULL }
};

wiResourceManager_BindLua::wiResourceManager_BindLua(wiResourceManager* resources) :resources(resources)
{
}
wiResourceManager_BindLua::wiResourceManager_BindLua(lua_State *L)
{
	resources = nullptr;
}
wiResourceManager_BindLua::~wiResourceManager_BindLua()
{
}

int wiResourceManager_BindLua::Get(lua_State *L)
{
	if (resources == nullptr)
	{
		wiLua::SError(L, "Get(string name) resources is empty!");
		return 0;
	}
	int argc = wiLua::SGetArgCount(L);
	if (argc > 0)
	{
		string name = wiLua::SGetString(L, 1);
		const wiResourceManager::Resource* data = resources->get(name);
		if (data != nullptr)
		{
			switch (data->type)
			{
			case wiResourceManager::Data_Type::IMAGE:
				Luna<Texture_BindLua>::push(L, new Texture_BindLua((Texture2D*)data->data));
				return 1;
				break;
			case wiResourceManager::Data_Type::MUSIC:
			case wiResourceManager::Data_Type::SOUND:
				Luna<wiSound_BindLua>::push(L, new wiSound_BindLua((wiSound*)data->data));
				return 1;
				break;
			default:
				wiLua::SError(L, "Get(string name) resource type not supported in scripts!");
				break;
			}
		}
		else
		{
			wiLua::SError(L, "Get(string name) resource not found!");
		}
		return 0;
	}
	else
	{
		wiLua::SError(L, "Get(string name) not enough arguments!");
	}
	return 0;
}
int wiResourceManager_BindLua::Add(lua_State *L)
{
	if (resources == nullptr)
	{
		wiLua::SError(L, "Add(string name) resources is empty!");
		return 0;
	}
	int argc = wiLua::SGetArgCount(L);
	if (argc > 0)
	{
		string name = wiLua::SGetString(L, 1);
		wiResourceManager::Data_Type type = wiResourceManager::Data_Type::DYNAMIC;
		if (argc > 1) //type info also provided in this case
		{
			string typeStr = wiHelper::toUpper( wiLua::SGetString(L, 2) );
			if (!typeStr.compare("SOUND"))
				type = wiResourceManager::Data_Type::SOUND;
			else if (!typeStr.compare("MUSIC"))
				type = wiResourceManager::Data_Type::MUSIC;
		}
		void* data = resources->add(name, type);
		wiLua::SSetString(L, (data != nullptr ? "ok" : "not found"));
		return 1;
	}
	else
	{
		wiLua::SError(L, "Resource:Add(string name, (opt) string type) not enough arguments!");
	}
	return 0;
}
int wiResourceManager_BindLua::Del(lua_State *L)
{
	if (resources == nullptr)
	{
		wiLua::SError(L, "Del(string name) resources is empty!");
		return 0;
	}
	int argc = wiLua::SGetArgCount(L);
	if (argc > 0)
	{
		string name = wiLua::SGetString(L, 1);
		wiLua::SSetString(L, (resources->del(name) ? "ok" : "not found"));
		return 1;
	}
	else
	{
		wiLua::SError(L, "Del(string name) not enough arguments!");
	}
	return 0;
}
int wiResourceManager_BindLua::List(lua_State *L)
{
	if (resources == nullptr)
	{
		wiLua::SError(L, "List(string name) resources is empty!");
		return 0;
	}
	stringstream ss("");
	for (auto x : resources->resources)
	{
		ss << x.first.GetString() << endl;
	}
	wiLua::SSetString(L, ss.str());
	return 1;
}

void wiResourceManager_BindLua::Bind()
{
	static bool initialized = false;
	if (!initialized)
	{
		Texture_BindLua::Bind();
		Luna<wiResourceManager_BindLua>::Register(wiLua::GetGlobal()->GetLuaState());
		wiLua::GetGlobal()->RegisterObject(className, "globalResources", new wiResourceManager_BindLua(wiResourceManager::GetGlobal()));
		initialized = true;
	}
}