~ubuntu-branches/ubuntu/raring/scummvm/raring

« back to all changes in this revision

Viewing changes to engines/sword25/script/luabindhelper.cpp

  • Committer: Bazaar Package Importer
  • Author(s): Moritz Muehlenhoff
  • Date: 2011-05-25 19:02:23 UTC
  • mto: (21.1.2 sid)
  • mto: This revision was merged to the branch mainline in revision 24.
  • Revision ID: james.westby@ubuntu.com-20110525190223-fiqm0oaec714xk31
Tags: upstream-1.3.0
Import upstream version 1.3.0

Show diffs side-by-side

added added

removed removed

Lines of Context:
 
1
/* ScummVM - Graphic Adventure Engine
 
2
 *
 
3
 * ScummVM is the legal property of its developers, whose names
 
4
 * are too numerous to list here. Please refer to the COPYRIGHT
 
5
 * file distributed with this source distribution.
 
6
 *
 
7
 * This program is free software; you can redistribute it and/or
 
8
 * modify it under the terms of the GNU General Public License
 
9
 * as published by the Free Software Foundation; either version 2
 
10
 * of the License, or (at your option) any later version.
 
11
 
 
12
 * This program is distributed in the hope that it will be useful,
 
13
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 
14
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 
15
 * GNU General Public License for more details.
 
16
 
 
17
 * You should have received a copy of the GNU General Public License
 
18
 * along with this program; if not, write to the Free Software
 
19
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
 
20
 *
 
21
 * $URL$
 
22
 * $Id$
 
23
 *
 
24
 */
 
25
 
 
26
/*
 
27
 * This code is based on Broken Sword 2.5 engine
 
28
 *
 
29
 * Copyright (c) Malte Thiesen, Daniel Queteschiner and Michael Elsdoerfer
 
30
 *
 
31
 * Licensed under GNU GPL v2
 
32
 *
 
33
 */
 
34
 
 
35
#include "sword25/kernel/kernel.h"
 
36
#include "sword25/script/luabindhelper.h"
 
37
#include "sword25/script/luascript.h"
 
38
 
 
39
namespace {
 
40
const char *METATABLES_TABLE_NAME = "__METATABLES";
 
41
const char *PERMANENTS_TABLE_NAME = "Permanents";
 
42
 
 
43
bool registerPermanent(lua_State *L, const Common::String &name) {
 
44
        // A C function has to be on the stack
 
45
        if (!lua_iscfunction(L, -1))
 
46
                return false;
 
47
 
 
48
        // Make sure that the Permanents-Table is on top of the stack
 
49
        lua_getfield(L, LUA_REGISTRYINDEX, PERMANENTS_TABLE_NAME);
 
50
        if (lua_isnil(L, -1)) {
 
51
                // Permanents-Table does not yet exist, so it has to be created
 
52
 
 
53
                // Pop nil from the stack
 
54
                lua_pop(L, 1);
 
55
 
 
56
                // Create Permanents-Table and insert a second reference to it on the stack
 
57
                lua_newtable(L);
 
58
                lua_pushvalue(L, -1);
 
59
 
 
60
                // Store the Permanents-Table in the registry. The second reference is left
 
61
                // on the stack to be used in the connection
 
62
                lua_setfield(L, LUA_REGISTRYINDEX, PERMANENTS_TABLE_NAME);
 
63
        }
 
64
 
 
65
        // C function with the name of an index in the Permanents-Table
 
66
        lua_insert(L, -2);
 
67
        lua_setfield(L, -2, name.c_str());
 
68
 
 
69
        // Remove the Permanents-Table from the stack
 
70
        lua_pop(L, 1);
 
71
 
 
72
        return true;
 
73
}
 
74
}
 
75
 
 
76
namespace Sword25 {
 
77
 
 
78
/**
 
79
 * Registers a set of functions into a Lua library.
 
80
 * @param L             A pointer to the Lua VM
 
81
 * @param LibName       The name of the library.
 
82
 * If this is an empty string, the functions will be added to the global namespace.
 
83
 * @param Functions     An array of function pointers along with their names.
 
84
 * The array must be terminated with the enry (0, 0)
 
85
 * @return              Returns true if successful, otherwise false.
 
86
 */
 
87
bool LuaBindhelper::addFunctionsToLib(lua_State *L, const Common::String &libName, const luaL_reg *functions) {
 
88
#ifdef DEBUG
 
89
        int __startStackDepth = lua_gettop(L);
 
90
#endif
 
91
 
 
92
        // If the table name is empty, the functions are to be added to the global namespace
 
93
        if (libName.size() == 0) {
 
94
                for (; functions->name; ++functions) {
 
95
                        lua_pushstring(L, functions->name);
 
96
                        lua_pushcclosure(L, functions->func, 0);
 
97
                        lua_settable(L, LUA_GLOBALSINDEX);
 
98
 
 
99
                        // Function is being permanently registed, so persistence can be ignored
 
100
                        lua_pushstring(L, functions->name);
 
101
                        lua_gettable(L, LUA_GLOBALSINDEX);
 
102
                        registerPermanent(L, functions->name);
 
103
                }
 
104
        } else { // If the table name is not empty, the functions are added to the given table
 
105
                // Ensure that the library table exists
 
106
                if (!createTable(L, libName)) return false;
 
107
 
 
108
                // Register each function into the table
 
109
                for (; functions->name; ++functions) {
 
110
                        // Function registration
 
111
                        lua_pushstring(L, functions->name);
 
112
                        lua_pushcclosure(L, functions->func, 0);
 
113
                        lua_settable(L, -3);
 
114
 
 
115
                        // Function is being permanently registed, so persistence can be ignored
 
116
                        lua_pushstring(L, functions->name);
 
117
                        lua_gettable(L, -2);
 
118
                        registerPermanent(L, libName + "." + functions->name);
 
119
                }
 
120
 
 
121
                // Remove the library table from the Lua stack
 
122
                lua_pop(L, 1);
 
123
        }
 
124
 
 
125
#ifdef DEBUG
 
126
        assert(__startStackDepth == lua_gettop(L));
 
127
#endif
 
128
 
 
129
        return true;
 
130
}
 
131
 
 
132
/**
 
133
 * Adds a set of constants to the Lua library
 
134
 * @param L             A pointer to the Lua VM
 
135
 * @param LibName       The name of the library.
 
136
 * If this is an empty string, the functions will be added to the global namespace.
 
137
 * @param Constants     An array of the constant values along with their names.
 
138
 * The array must be terminated with the enry (0, 0)
 
139
 * @return              Returns true if successful, otherwise false.
 
140
 */
 
141
bool LuaBindhelper::addConstantsToLib(lua_State *L, const Common::String &libName, const lua_constant_reg *constants) {
 
142
#ifdef DEBUG
 
143
        int __startStackDepth = lua_gettop(L);
 
144
#endif
 
145
 
 
146
        // If the table is empty, the constants are added to the global namespace
 
147
        if (libName.size() == 0) {
 
148
                for (; constants->Name; ++constants) {
 
149
                        lua_pushstring(L, constants->Name);
 
150
                        lua_pushnumber(L, constants->Value);
 
151
                        lua_settable(L, LUA_GLOBALSINDEX);
 
152
                }
 
153
        }
 
154
        // If the table name is nto empty, the constants are added to that table
 
155
        else {
 
156
                // Ensure that the library table exists
 
157
                if (!createTable(L, libName)) return false;
 
158
 
 
159
                // Register each constant in the table
 
160
                for (; constants->Name; ++constants) {
 
161
                        lua_pushstring(L, constants->Name);
 
162
                        lua_pushnumber(L, constants->Value);
 
163
                        lua_settable(L, -3);
 
164
                }
 
165
 
 
166
                // Remove the library table from the Lua stack
 
167
                lua_pop(L, 1);
 
168
        }
 
169
 
 
170
#ifdef DEBUG
 
171
        assert(__startStackDepth == lua_gettop(L));
 
172
#endif
 
173
 
 
174
        return true;
 
175
}
 
176
 
 
177
/**
 
178
 * Adds a set of methods to a Lua class
 
179
 * @param L             A pointer to the Lua VM
 
180
 * @param ClassName     The name of the class
 
181
 * When the class name specified does not exist, it is created.
 
182
 * @param Methods       An array of function pointers along with their method names.
 
183
 * The array must be terminated with the enry (0, 0)
 
184
 * @return              Returns true if successful, otherwise false.
 
185
 */
 
186
bool LuaBindhelper::addMethodsToClass(lua_State *L, const Common::String &className, const luaL_reg *methods) {
 
187
#ifdef DEBUG
 
188
        int __startStackDepth = lua_gettop(L);
 
189
#endif
 
190
 
 
191
        // Load the metatable onto the Lua stack
 
192
        if (!getMetatable(L, className)) return false;
 
193
 
 
194
        // Register each method in the Metatable
 
195
        for (; methods->name; ++methods) {
 
196
                lua_pushstring(L, methods->name);
 
197
                lua_pushcclosure(L, methods->func, 0);
 
198
                lua_settable(L, -3);
 
199
 
 
200
                // Function is being permanently registed, so persistence can be ignored
 
201
                lua_pushstring(L, methods->name);
 
202
                lua_gettable(L, -2);
 
203
                registerPermanent(L, className + "." + methods->name);
 
204
        }
 
205
 
 
206
        // Remove the metatable from the stack
 
207
        lua_pop(L, 1);
 
208
 
 
209
#ifdef DEBUG
 
210
        assert(__startStackDepth == lua_gettop(L));
 
211
#endif
 
212
 
 
213
        return true;
 
214
}
 
215
 
 
216
/**
 
217
 * Sets the garbage collector callback method when items of a particular class are deleted
 
218
 * @param L             A pointer to the Lua VM
 
219
 * @param ClassName     The name of the class
 
220
 * When the class name specified does not exist, it is created.
 
221
 * @param GCHandler     A function pointer
 
222
 * @return              Returns true if successful, otherwise false.
 
223
 */
 
224
bool LuaBindhelper::setClassGCHandler(lua_State *L, const Common::String &className, lua_CFunction GCHandler) {
 
225
#ifdef DEBUG
 
226
        int __startStackDepth = lua_gettop(L);
 
227
#endif
 
228
 
 
229
        // Load the metatable onto the Lua stack
 
230
        if (!getMetatable(L, className)) return false;
 
231
 
 
232
        // Add the GC handler to the Metatable
 
233
        lua_pushstring(L, "__gc");
 
234
        lua_pushcclosure(L, GCHandler, 0);
 
235
        lua_settable(L, -3);
 
236
 
 
237
        // Function is being permanently registed, so persistence can be ignored
 
238
        lua_pushstring(L, "__gc");
 
239
        lua_gettable(L, -2);
 
240
        registerPermanent(L, className + ".__gc");
 
241
 
 
242
        // Remove the metatable from the stack
 
243
        lua_pop(L, 1);
 
244
 
 
245
#ifdef DEBUG
 
246
        assert(__startStackDepth == lua_gettop(L));
 
247
#endif
 
248
 
 
249
        return true;
 
250
}
 
251
 
 
252
} // End of namespace Sword25
 
253
 
 
254
namespace {
 
255
void pushMetatableTable(lua_State *L) {
 
256
        // Push the Metatable table onto the stack
 
257
        lua_getglobal(L, METATABLES_TABLE_NAME);
 
258
 
 
259
        // If the table doesn't yet exist, it must be created
 
260
        if (lua_isnil(L, -1)) {
 
261
                // Pop nil from stack
 
262
                lua_pop(L, 1);
 
263
 
 
264
                // New table has been created, so add it to the global table and leave reference on stack
 
265
                lua_newtable(L);
 
266
                lua_pushvalue(L, -1);
 
267
                lua_setglobal(L, METATABLES_TABLE_NAME);
 
268
        }
 
269
}
 
270
}
 
271
 
 
272
namespace Sword25 {
 
273
 
 
274
bool LuaBindhelper::getMetatable(lua_State *L, const Common::String &tableName) {
 
275
        // Push the Metatable table onto the stack
 
276
        pushMetatableTable(L);
 
277
 
 
278
        // Versuchen, die gew�nschte Metatabelle auf den Stack zu legen. Wenn sie noch nicht existiert, muss sie erstellt werden.
 
279
        lua_getfield(L, -1, tableName.c_str());
 
280
        if (lua_isnil(L, -1)) {
 
281
                // Pop nil from stack
 
282
                lua_pop(L, 1);
 
283
 
 
284
                // Create new table
 
285
                lua_newtable(L);
 
286
 
 
287
                // Set the __index field in the table
 
288
                lua_pushvalue(L, -1);
 
289
                lua_setfield(L, -2, "__index");
 
290
 
 
291
                // Flag the table as persisted. This ensures that objects within this table get stored
 
292
                lua_pushbooleancpp(L, true);
 
293
                lua_setfield(L, -2, "__persist");
 
294
 
 
295
                // Set the table name and push it onto the stack
 
296
                lua_pushvalue(L, -1);
 
297
                lua_setfield(L, -3, tableName.c_str());
 
298
        }
 
299
 
 
300
        // Remove the Metatable table from the stack
 
301
        lua_remove(L, -2);
 
302
 
 
303
        return true;
 
304
}
 
305
 
 
306
// Like luaL_checkudata, only without that no error is generated.
 
307
void *LuaBindhelper::my_checkudata(lua_State *L, int ud, const char *tname) {
 
308
        int top = lua_gettop(L);
 
309
 
 
310
        void *p = lua_touserdata(L, ud);
 
311
        if (p != NULL) { /* value is a userdata? */
 
312
                if (lua_getmetatable(L, ud)) { /* does it have a metatable? */
 
313
                        // lua_getfield(L, LUA_REGISTRYINDEX, tname);  /* get correct metatable */
 
314
                        LuaBindhelper::getMetatable(L, tname);
 
315
                        if (lua_rawequal(L, -1, -2)) { /* does it have the correct mt? */
 
316
                                lua_settop(L, top);
 
317
                                return p;
 
318
                        }
 
319
                }
 
320
        }
 
321
 
 
322
        lua_settop(L, top);
 
323
        return NULL;
 
324
}
 
325
 
 
326
 
 
327
bool LuaBindhelper::createTable(lua_State *L, const Common::String &tableName) {
 
328
        const char *partBegin = tableName.c_str();
 
329
 
 
330
        while (partBegin - tableName.c_str() < (int)tableName.size()) {
 
331
                const char *partEnd = strchr(partBegin, '.');
 
332
                if (!partEnd)
 
333
                        partEnd = partBegin + strlen(partBegin);
 
334
                Common::String subTableName(partBegin, partEnd);
 
335
 
 
336
                // Tables with an empty string as the name are not allowed
 
337
                if (subTableName.size() == 0)
 
338
                        return false;
 
339
 
 
340
                // Verify that the table with the name already exists
 
341
                // The first round will be searched in the global namespace, with later passages
 
342
                // in the corresponding parent table in the stack
 
343
                if (partBegin == tableName.c_str()) {
 
344
                        lua_pushstring(L, subTableName.c_str());
 
345
                        lua_gettable(L, LUA_GLOBALSINDEX);
 
346
                } else {
 
347
                        lua_pushstring(L, subTableName.c_str());
 
348
                        lua_gettable(L, -2);
 
349
                        if (!lua_isnil(L, -1))
 
350
                                lua_remove(L, -2);
 
351
                }
 
352
 
 
353
                // If it doesn't exist, create table
 
354
                if (lua_isnil(L, -1)) {
 
355
                        // Pop nil from stack
 
356
                        lua_pop(L, 1);
 
357
 
 
358
                        // Create new table
 
359
                        lua_newtable(L);
 
360
                        lua_pushstring(L, subTableName.c_str());
 
361
                        lua_pushvalue(L, -2);
 
362
                        if (partBegin == tableName.c_str())
 
363
                                lua_settable(L, LUA_GLOBALSINDEX);
 
364
                        else {
 
365
                                lua_settable(L, -4);
 
366
                                lua_remove(L, -2);
 
367
                        }
 
368
                }
 
369
 
 
370
                partBegin = partEnd + 1;
 
371
        }
 
372
 
 
373
        return true;
 
374
}
 
375
 
 
376
} // End of namespace Sword25
 
377
 
 
378
namespace {
 
379
Common::String getLuaValueInfo(lua_State *L, int stackIndex) {
 
380
        switch (lua_type(L, stackIndex)) {
 
381
        case LUA_TNUMBER:
 
382
                lua_pushstring(L, lua_tostring(L, stackIndex));
 
383
                break;
 
384
 
 
385
        case LUA_TSTRING:
 
386
                lua_pushfstring(L, "\"%s\"", lua_tostring(L, stackIndex));
 
387
                break;
 
388
 
 
389
        case LUA_TBOOLEAN:
 
390
                lua_pushstring(L, (lua_toboolean(L, stackIndex) ? "true" : "false"));
 
391
                break;
 
392
 
 
393
        case LUA_TNIL:
 
394
                lua_pushliteral(L, "nil");
 
395
                break;
 
396
 
 
397
        default:
 
398
                lua_pushfstring(L, "%s: %p", luaL_typename(L, stackIndex), lua_topointer(L, stackIndex));
 
399
                break;
 
400
        }
 
401
 
 
402
        Common::String result(lua_tostring(L, -1));
 
403
        lua_pop(L, 1);
 
404
 
 
405
        return result;
 
406
}
 
407
}
 
408
 
 
409
namespace Sword25 {
 
410
 
 
411
Common::String LuaBindhelper::stackDump(lua_State *L) {
 
412
        Common::String oss;
 
413
 
 
414
        int i = lua_gettop(L);
 
415
        oss += "------------------- Stack Dump -------------------\n";
 
416
 
 
417
        while (i) {
 
418
                oss += i + ": " + getLuaValueInfo(L, i) + "\n";
 
419
                i--;
 
420
        }
 
421
 
 
422
        oss += "-------------- Stack Dump Finished ---------------\n";
 
423
 
 
424
        return oss;
 
425
}
 
426
 
 
427
Common::String LuaBindhelper::tableDump(lua_State *L) {
 
428
        Common::String oss;
 
429
 
 
430
        oss += "------------------- Table Dump -------------------\n";
 
431
 
 
432
        lua_pushnil(L);
 
433
        while (lua_next(L, -2) != 0) {
 
434
                // Get the value of the current element on top of the stack, including the index
 
435
                oss += getLuaValueInfo(L, -2) + " : " + getLuaValueInfo(L, -1) + "\n";
 
436
 
 
437
                // Pop value from the stack. The index is then ready for the next call to lua_next()
 
438
                lua_pop(L, 1);
 
439
        }
 
440
 
 
441
        oss += "-------------- Table Dump Finished ---------------\n";
 
442
 
 
443
        return oss;
 
444
}
 
445
 
 
446
} // End of namespace Sword25