From 9d8c499afc09ee9ad34bf6f2b427d6c80b0e6f78 Mon Sep 17 00:00:00 2001
From: Firew0lf <Firew0lf@users.noreply.github.com>
Date: Sat, 12 Sep 2015 12:05:04 +0200
Subject: [PATCH] Fixed the QTM __index method

---
 source/qtm.c | 36 ++++++++++++++++++++++++++----------
 1 file changed, 26 insertions(+), 10 deletions(-)

diff --git a/source/qtm.c b/source/qtm.c
index 948789e..45ce66b 100644
--- a/source/qtm.c
+++ b/source/qtm.c
@@ -5,10 +5,14 @@
 #include <lapi.h>
 #include <lauxlib.h>
 
+#include <string.h>
+
 typedef struct {
 	qtmHeadtrackingInfo *info;
 } qtm_userdata;
 
+static const struct luaL_Reg qtm_methods[];
+
 static int qtm_init(lua_State *L) {
 	Result ret = qtmInit();
 	if (ret) {
@@ -54,18 +58,30 @@ static int qtm_checkHeadFullyDetected(lua_State *L) {
 
 static int qtm___index(lua_State *L) {
 	qtm_userdata *info = luaL_checkudata(L, 1, "LQTM");
-	lua_Integer index = luaL_checkinteger(L, 2);
-	index = index - 1; // Lua index begins at 1
-	if (index > 3 || index < 0) {
-		lua_pushnil(L);
-		lua_pushnil(L);
+	if (lua_isinteger(L, 2)) { // index
+		lua_Integer index = luaL_checkinteger(L, 2);
+		index = index - 1; // Lua index begins at 1
+		if (index > 3 || index < 0) {
+			lua_pushnil(L);
+			lua_pushnil(L);
+			return 2;
+		}
+	
+		lua_pushnumber(L, info->info->coords0[index].x);
+		lua_pushnumber(L, info->info->coords0[index].y);
 		return 2;
+		
+	} else if (lua_isstring(L, 2)) { //method
+		const char *mname = luaL_checkstring(L, 2);
+		for (u8 i=0;qtm_methods[i].name;i++) {
+			if (strcmp(qtm_methods[i].name, mname) == 0) {
+				lua_pushcfunction(L, qtm_methods[i].func);
+				return 1;
+			}
+		}
 	}
-	
-	lua_pushnumber(L, info->info->coords0[index].x);
-	lua_pushnumber(L, info->info->coords0[index].y);
-	
-	return 2;
+	lua_pushnil(L);
+	return 1;
 }
 
 static int qtm_convertCoordToScreen(lua_State *L) {