Fix issue #292: Can't send arrays to pixeleffects.

Arrays of numbers, vectors and matrices are supported.
This commit is contained in:
vrld
2011-09-07 23:32:54 +02:00
parent 2eb9ee20d5
commit eec0250545
5 changed files with 177 additions and 71 deletions
+109 -27
View File
@@ -2,6 +2,8 @@
#include "wrap_Image.h"
#include "wrap_Canvas.h"
#include <string>
#include <iostream>
using namespace std;
namespace love
{
@@ -21,48 +23,128 @@ namespace opengl
return 1;
}
int w_PixelEffect_sendFloat(lua_State * L)
static int _sendScalars(lua_State * L, PixelEffect * effect, const char * name, int count)
{
size_t count = lua_gettop(L) - 2;
PixelEffect * effect = luax_checkpixeleffect(L, 1);
const char* name = luaL_checkstring(L, 2);
if (count < 1 || count > 4)
return luaL_error(L, "Invalid variable count (expected 1-4, got %d).", count);
float values[4] = {0,0,0,0};
for (unsigned int i = 0; i < count; ++i)
values[i] = (float) luaL_checknumber(L, i+1 + 2);
try {
effect->sendFloat(name, count, values);
} catch(love::Exception& e) {
luaL_error(L, e.what());
float * values = new float[count];
for (int i = 0; i < count; ++i) {
if (!lua_isnumber(L, 3 + i)) {
delete[] values;
return luaL_typerror(L, 3 + i, "number");
}
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, e.what());
}
delete[] values;
return 0;
}
static int _sendVectors(lua_State * L, PixelEffect * effect, const char * name, int count)
{
int 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 (int k = 1; k <= dimension; ++k) {
lua_rawgeti(L, 3 + i, k);
values[i * dimension + k - 1] = (float)lua_tonumber(L, -1);
}
lua_pop(L, dimension);
}
try {
effect->sendFloat(name, dimension, values, count);
} catch(love::Exception& e) {
delete[] values;
return luaL_error(L, e.what());
}
delete[] values;
return 0;
}
int w_PixelEffect_sendFloat(lua_State * L)
{
PixelEffect * effect = luax_checkpixeleffect(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))
return _sendScalars(L, effect, name, count);
else if (lua_istable(L, 3))
return _sendVectors(L, effect, name, count);
return luaL_typerror(L, 3, "number or table");
}
int w_PixelEffect_sendMatrix(lua_State * L)
{
size_t count = lua_gettop(L) - 3;
int count = lua_gettop(L) - 2;
PixelEffect * effect = luax_checkpixeleffect(L, 1);
const char* name = luaL_checkstring(L, 2);
int size = luaL_checkinteger(L, 3);
if (size < 2 || size > 4)
return luaL_error(L, "Invalid matrix size: %dx%d (only 2x2, 3x3 and 4x4 matrices are supported).", count, count);
if (!lua_istable(L, 3))
return luaL_typerror(L, 3, "matrix table");
float values[16] = {0,0,0,0,0,0,0,0,0,0,0,0,0,0,0,0};
for (unsigned int i = 0; i < count; ++i)
values[i] = (float) luaL_checknumber(L, i+1 + 3);
lua_getfield(L, 3, "dimension");
int dimension = lua_tointeger(L, -1);
lua_pop(L, 1);
try {
effect->sendFloat(name, size, values);
} catch(love::Exception& e) {
luaL_error(L, e.what());
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);
cout << i << "." << k << " = " << (float)lua_tonumber(L, -1) << endl;
}
lua_pop(L, 1 + dimension);
}
try {
effect->sendMatrix(name, dimension, values, count);
} catch(love::Exception& e) {
delete[] values;
return luaL_error(L, e.what());
}
delete[] values;
return 0;
}