1 files changed, 114 insertions, 0 deletions
diff --git a/src/script/squirrel_std.cpp b/src/script/squirrel_std.cpp
new file mode 100644
index 000000000..3850b7b54
--- /dev/null
+++ b/src/script/squirrel_std.cpp
@@ -0,0 +1,114 @@
+
+#include <squirrel.h>
+#include "../stdafx.h"
+#include "../debug.h"
+#include "squirrel.hpp"
+#include "squirrel_std.hpp"
+#include "../core/alloc_func.hpp"
+#include "../core/math_func.hpp"
+
+/* abs() is normally defined to myabs(), which we don't want in this file */
+#undef abs
+
+SQInteger SquirrelStd::abs(HSQUIRRELVM vm)
+{
+	SQInteger tmp;
+
+	sq_getinteger(vm, 2, &tmp);
+	sq_pushinteger(vm, ::abs(tmp));
+	return 1;
+}
+
+SQInteger SquirrelStd::min(HSQUIRRELVM vm)
+{
+	SQInteger tmp1, tmp2;
+
+	sq_getinteger(vm, 2, &tmp1);
+	sq_getinteger(vm, 3, &tmp2);
+	sq_pushinteger(vm, ::min(tmp1, tmp2));
+	return 1;
+}
+
+SQInteger SquirrelStd::max(HSQUIRRELVM vm)
+{
+	SQInteger tmp1, tmp2;
+
+	sq_getinteger(vm, 2, &tmp1);
+	sq_getinteger(vm, 3, &tmp2);
+	sq_pushinteger(vm, ::max(tmp1, tmp2));
+	return 1;
+}
+
+SQInteger SquirrelStd::require(HSQUIRRELVM vm)
+{
+	SQInteger top = sq_gettop(vm);
+	const SQChar *filename;
+	SQChar *real_filename;
+
+	sq_getstring(vm, 2, &filename);
+
+	/* Get the script-name of the current file, so we can work relative from it */
+	SQStackInfos si;
+	sq_stackinfos(vm, 1, &si);
+	if (si.source == NULL) {
+		DEBUG(misc, 0, "[squirrel] Couldn't detect the script-name of the 'require'-caller; this should never happen!");
+		return SQ_ERROR;
+	}
+	real_filename = scstrdup(si.source);
+	/* Keep the dir, remove the rest */
+	SQChar *s = scstrrchr(real_filename, PATHSEPCHAR);
+	if (s != NULL) {
+		/* Keep the PATHSEPCHAR there, remove the rest */
+		*s++;
+		*s = '\0';
+	}
+	/* And now we concat, so we are relative from the current script
+	 * First, we have to make sure we have enough space for the full path */
+	real_filename = ReallocT(real_filename, scstrlen(real_filename) + scstrlen(filename) + 1);
+	scstrcat(real_filename, filename);
+	/* Tars dislike opening files with '/' on Windows.. so convert it to '\\' ;) */
+	char *filen = strdup(FS2OTTD(real_filename));
+#if (PATHSEPCHAR != '/')
+	for (char *n = filen; *n != '\0'; n++) if (*n == '/') *n = PATHSEPCHAR;
+#endif
+
+	bool ret = Squirrel::LoadScript(vm, filen);
+
+	/* Reset the top, so the stack stays correct */
+	sq_settop(vm, top);
+	free(real_filename);
+	free(filen);
+
+	return ret ? 0: SQ_ERROR;
+}
+
+SQInteger SquirrelStd::notifyallexceptions(HSQUIRRELVM vm)
+{
+	SQBool b;
+
+	if (sq_gettop(vm) >= 1) {
+		if (SQ_SUCCEEDED(sq_getbool(vm, -1, &b))) {
+			sq_notifyallexceptions(vm, b);
+			return 0;
+		}
+	}
+
+	return SQ_ERROR;
+}
+
+void squirrel_register_global_std(Squirrel *engine)
+{
+	/* We don't use squirrel_helper here, as we want to register to the global
+	 *  scope and not to a class. */
+	engine->AddMethod("require",             &SquirrelStd::require,             2, "?s");
+	engine->AddMethod("notifyallexceptions", &SquirrelStd::notifyallexceptions, 2, "?b");
+}
+
+void squirrel_register_std(Squirrel *engine)
+{
+	/* We don't use squirrel_helper here, as we want to register to the global
+	 *  scope and not to a class. */
+	engine->AddMethod("abs", &SquirrelStd::abs, 2, "?i");
+	engine->AddMethod("min", &SquirrelStd::min, 3, "?ii");
+	engine->AddMethod("max", &SquirrelStd::max, 3, "?ii");
+}