love/src/modules/graphics/opengl/wrap_ShaderEffect.cpp

/**
 * Copyright (c) 2006-2012 LOVE Development Team
 *
 * This software is provided 'as-is', without any express or implied
 * warranty.  In no event will the authors be held liable for any damages
 * arising from the use of this software.
 *
 * Permission is granted to anyone to use this software for any purpose,
 * including commercial applications, and to alter it and redistribute it
 * freely, subject to the following restrictions:
 *
 * 1. The origin of this software must not be misrepresented; you must not
 *    claim that you wrote the original software. If you use this software
 *    in a product, an acknowledgment in the product documentation would be
 *    appreciated but is not required.
 * 2. Altered source versions must be plainly marked as such, and must not be
 *    misrepresented as being the original software.
 * 3. This notice may not be removed or altered from any source distribution.
 **/

#include "wrap_ShaderEffect.h"
#include "wrap_Image.h"
#include "wrap_Canvas.h"
#include <string>
#include <iostream>

namespace love
{
namespace graphics
{
namespace opengl
{

ShaderEffect *luax_checkshadereffect(lua_State *L, int idx)
{
	return luax_checktype<ShaderEffect>(L, idx, "ShaderEffect", GRAPHICS_SHADEREFFECT_T);
}

int w_ShaderEffect_getWarnings(lua_State *L)
{
	ShaderEffect *effect = luax_checkshadereffect(L, 1);
	lua_pushstring(L, effect->getWarnings().c_str());
	return 1;
}

static int _sendScalars(lua_State *L, ShaderEffect *effect, const char *name, int count)
{
	float *values = new float[count];
	for (int i = 0; i < count; ++i)
	{
		if (lua_isnumber(L, 3 + i))
			values[i] = (float)lua_tonumber(L, 3 + i);
		else if (lua_isboolean(L, 3 + i))
			values[i] = (float)lua_toboolean(L, 3 + i);
		else
		{
			delete[] values;
			return luaL_typerror(L, 3 + i, "number or boolean");
		}
		values[i] = (float)lua_tonumber(L, 3 + i);
	}

	try
	{
		effect->sendFloat(name, 1, values, count);
	}
	catch(love::Exception &e)
	{
		delete[] values;
		return luaL_error(L, "%s", e.what());
	}

	delete[] values;
	return 0;
}

static int _sendVectors(lua_State *L, ShaderEffect *effect, const char *name, int count)
{
	size_t dimension = lua_objlen(L, 3);
	float *values = new float[count * dimension];

	for (int i = 0; i < count; ++i)
	{
		if (!lua_istable(L, 3 + i))
		{
			delete[] values;
			return luaL_typerror(L, 3 + i, "table");
		}
		if (lua_objlen(L, 3 + i) != dimension)
		{
			delete[] values;
			return luaL_error(L, "Error in argument %d: Expected table size %d, got %d.",
							  3+i, dimension, lua_objlen(L, 3+i));
		}

		for (size_t k = 1; k <= dimension; ++k)
		{
			lua_rawgeti(L, 3 + i, k);
			if (lua_isboolean(L, -1))
				values[i * dimension + k - 1] = (float)lua_toboolean(L, -1);
			else
				values[i * dimension + k - 1] = (float)lua_tonumber(L, -1);
		}
		lua_pop(L, int(dimension));
	}

	try
	{
		effect->sendFloat(name, dimension, values, count);
	}
	catch(love::Exception &e)
	{
		delete[] values;
		return luaL_error(L, "%s", e.what());
	}

	delete[] values;
	return 0;
}

int w_ShaderEffect_sendFloat(lua_State *L)
{
	ShaderEffect *effect = luax_checkshadereffect(L, 1);
	const char *name = luaL_checkstring(L, 2);
	int count = lua_gettop(L) - 2;

	if (count < 1)
		return luaL_error(L, "No variable to send.");

	if (lua_isnumber(L, 3) || lua_isboolean(L, 3))
		return _sendScalars(L, effect, name, count);
	else if (lua_istable(L, 3))
		return _sendVectors(L, effect, name, count);

	return luaL_typerror(L, 3, "number, boolean or table");
}

int w_ShaderEffect_sendMatrix(lua_State *L)
{
	int count = lua_gettop(L) - 2;
	ShaderEffect *effect = luax_checkshadereffect(L, 1);
	const char *name = luaL_checkstring(L, 2);

	if (!lua_istable(L, 3))
		return luaL_typerror(L, 3, "matrix table");

	lua_getfield(L, 3, "dimension");
	int dimension = lua_tointeger(L, -1);
	lua_pop(L, 1);

	if (dimension < 2 || dimension > 4)
		return luaL_error(L, "Invalid matrix size: %dx%d (only 2x2, 3x3 and 4x4 matrices are supported).",
						  count, count);

	float *values = new float[dimension * dimension * count];
	for (int i = 0; i < count; ++i)
	{
		lua_getfield(L, 3+i, "dimension");
		if (lua_tointeger(L, -1) != dimension)
		{
			// You unlock this door with the key of imagination. Beyond it is
			// another dimension: a dimension of sound, a dimension of sight,
			// a dimension of mind. You're moving into a land of both shadow
			// and substance, of things and ideas. You've just crossed over
			// into... the Twilight Zone.
			int other_dimension = lua_tointeger(L, -1);
			delete[] values;
			return luaL_error(L, "Invalid matrix size at argument %d: Expected size %dx%d, got %dx%d.",
							  3+i, dimension, dimension, other_dimension, other_dimension);
		}

		for (int k = 1; k <= dimension*dimension; ++k)
		{
			lua_rawgeti(L, 3+i, k);
			values[i * dimension * dimension + k - 1] = (float)lua_tonumber(L, -1);
		}

		lua_pop(L, 1 + dimension);
	}

	try
	{
		effect->sendMatrix(name, dimension, values, count);
	}
	catch(love::Exception &e)
	{
		delete[] values;
		return luaL_error(L, "%s", e.what());
	}

	delete[] values;
	return 0;
}

int w_ShaderEffect_sendImage(lua_State *L)
{
	ShaderEffect *effect = luax_checkshadereffect(L, 1);
	const char *name = luaL_checkstring(L, 2);
	Image *img = luax_checkimage(L, 3);

	try
	{
		effect->sendImage(name, *img);
	}
	catch(love::Exception &e)
	{
		luaL_error(L, "%s", e.what());
	}

	return 0;
}

int w_ShaderEffect_sendCanvas(lua_State *L)
{
	ShaderEffect *effect = luax_checkshadereffect(L, 1);
	const char *name = luaL_checkstring(L, 2);
	Canvas *canvas = luax_checkcanvas(L, 3);

	try
	{
		effect->sendCanvas(name, *canvas);
	}
	catch(love::Exception &e)
	{
		luaL_error(L, "%s", e.what());
	}

	return 0;
}


static const luaL_Reg functions[] =
{
	{ "getWarnings", w_ShaderEffect_getWarnings },
	{ "sendFloat",   w_ShaderEffect_sendFloat },
	{ "sendMatrix",  w_ShaderEffect_sendMatrix },
	{ "sendImage",   w_ShaderEffect_sendImage },
	{ "sendCanvas",  w_ShaderEffect_sendCanvas },
	{ 0, 0 }
};

extern "C" int luaopen_shadereffect(lua_State *L)
{
	return luax_register_type(L, "ShaderEffect", functions);
}

} // opengl
} // graphics
} // love