Initial commit

engines/grim/lua/lfunc.cpp

@@ -0,0 +1,87 @@
+/*
+** Auxiliary functions to manipulate prototypes and closures
+** See Copyright Notice in lua.h
+*/
+
+#define FORBIDDEN_SYMBOL_EXCEPTION_setjmp
+#define FORBIDDEN_SYMBOL_EXCEPTION_longjmp
+
+#include "engines/grim/lua/lfunc.h"
+#include "engines/grim/lua/lmem.h"
+#include "engines/grim/lua/lstate.h"
+
+namespace Grim {
+
+#define gcsizeproto(p)   5  /* approximate "weight" for a prototype */
+#define gcsizeclosure(c) 1  /* approximate "weight" for a closure */
+
+Closure *luaF_newclosure(int32 nelems) {
+	Closure *c = (Closure *)luaM_malloc(sizeof(Closure) + nelems * sizeof(TObject));
+	luaO_insertlist(&rootcl, (GCnode *)c);
+	nblocks += gcsizeclosure(c);
+	c->nelems = nelems;
+	return c;
+}
+
+TProtoFunc *luaF_newproto() {
+	TProtoFunc *f = luaM_new(TProtoFunc);
+	f->code = nullptr;
+	f->lineDefined = 0;
+	f->fileName = nullptr;
+	f->consts = nullptr;
+	f->nconsts = 0;
+	f->locvars = nullptr;
+	luaO_insertlist(&rootproto, (GCnode *)f);
+	nblocks += gcsizeproto(f);
+	return f;
+}
+
+static void freefunc(TProtoFunc *f) {
+	luaM_free(f->code);
+	luaM_free(f->locvars);
+	luaM_free(f->consts);
+	luaM_free(f);
+}
+
+void luaF_freeproto(TProtoFunc *l) {
+	while (l) {
+		TProtoFunc *next = (TProtoFunc *)l->head.next;
+		nblocks -= gcsizeproto(l);
+		freefunc(l);
+		l = next;
+	}
+}
+
+void luaF_freeclosure(Closure *l) {
+	while (l) {
+		Closure *next = (Closure *)l->head.next;
+		nblocks -= gcsizeclosure(l);
+		luaM_free(l);
+		l = next;
+	}
+}
+
+/*
+** Look for n-th local variable at line "line" in function "func".
+** Returns NULL if not found.
+*/
+char *luaF_getlocalname (TProtoFunc *func, int32 local_number, int32 line) {
+	int32 count = 0;
+	char *varname = nullptr;
+	LocVar *lv = func->locvars;
+	if (!lv)
+		return nullptr;
+	for (; lv->line != -1 && lv->line < line; lv++) {
+		if (lv->varname) {  // register
+			if (++count == local_number)
+				varname = lv->varname->str;
+		} else {  // unregister
+			if (--count < local_number) {
+				varname = nullptr;
+			}
+		}
+	}
+	return varname;
+}
+
+} // end of namespace Grim