From 402ae4638a513d7bd367346c864c31702ef92078 Mon Sep 17 00:00:00 2001
From: Bart van Strien <bart.bes+projects@gmail.com>
Date: Fri, 10 May 2013 17:28:55 +0200
Subject: [PATCH] Make ImageData:setPixel accept a table (issue #582)

---
 src/modules/image/wrap_ImageData.cpp | 25 +++++++++++++++++++++----
 1 file changed, 21 insertions(+), 4 deletions(-)

diff --git a/src/modules/image/wrap_ImageData.cpp b/src/modules/image/wrap_ImageData.cpp
index d09c2fc0c..7605d7876 100644
--- a/src/modules/image/wrap_ImageData.cpp
+++ b/src/modules/image/wrap_ImageData.cpp
@@ -82,10 +82,27 @@ int w_ImageData_setPixel(lua_State *L)
 	int x = luaL_checkint(L, 2);
 	int y = luaL_checkint(L, 3);
 	pixel c;
-	c.r = luaL_checkint(L, 4);
-	c.g = luaL_checkint(L, 5);
-	c.b = luaL_checkint(L, 6);
-	c.a = luaL_optint(L, 7, 255);
+
+	if (lua_istable(L, 4))
+	{
+		for (int i = 1; i <= 4; i++)
+			lua_rawgeti(L, 4, i);
+
+		c.r = (unsigned char)luaL_checkint(L, -4);
+		c.g = (unsigned char)luaL_checkint(L, -3);
+		c.b = (unsigned char)luaL_checkint(L, -2);
+		c.a = (unsigned char)luaL_optint(L, -1, 255);
+
+		lua_pop(L, 4);
+	}
+	else
+	{
+		c.r = (unsigned char)luaL_checkint(L, 4);
+		c.g = (unsigned char)luaL_checkint(L, 5);
+		c.b = (unsigned char)luaL_checkint(L, 6);
+		c.a = (unsigned char)luaL_optint(L, 7, 255);
+	}
+
 	try
 	{
 		t->setPixel(x, y, c);