From a92ca7ceacbd71341aa8a94dee712be243625dcc Mon Sep 17 00:00:00 2001
From: Blixibon <acj3017@gmail.com>
Date: Sat, 24 Apr 2021 11:44:51 -0500
Subject: [PATCH] Added weakref() and getclass() for Vector and other C++
 instances in VScript

---
 sp/src/vscript/vscript_squirrel.cpp | 36 +++++++++++++++++++++++++++++
 1 file changed, 36 insertions(+)

diff --git a/sp/src/vscript/vscript_squirrel.cpp b/sp/src/vscript/vscript_squirrel.cpp
index 79160a49..b1e250b1 100644
--- a/sp/src/vscript/vscript_squirrel.cpp
+++ b/sp/src/vscript/vscript_squirrel.cpp
@@ -495,6 +495,19 @@ namespace SQVector
 		return 1;
 	}
 
+	SQInteger weakref(HSQUIRRELVM vm)
+	{
+		sq_weakref(vm, 1);
+		return 1;
+	}
+
+	SQInteger getclass(HSQUIRRELVM vm)
+	{
+		sq_getclass(vm, 1);
+		sq_push(vm, -1);
+		return 1;
+	}
+
 	// multi purpose - copy from input vector, or init with 3 float input
 	SQInteger Set(HSQUIRRELVM vm)
 	{
@@ -985,6 +998,8 @@ namespace SQVector
 		{_SC("_mul"), _multiply, 2, _SC("..")},
 		{_SC("_div"), _divide, 2, _SC("..")},
 		{_SC("_unm"), _unm, 1, _SC(".")},
+		{_SC("weakref"), weakref, 1, _SC(".")},
+		{_SC("getclass"), getclass, 1, _SC(".")},
 		{_SC("Set"), Set, -2, _SC("..nn")},
 		{_SC("Add"), Add, 2, _SC("..")},
 		{_SC("Subtract"), Subtract, 2, _SC("..")},
@@ -1648,6 +1663,19 @@ SQInteger IsValid_stub(HSQUIRRELVM vm)
 	return 1;
 }
 
+SQInteger weakref_stub(HSQUIRRELVM vm)
+{
+	sq_weakref(vm, 1);
+	return 1;
+}
+
+SQInteger getclass_stub(HSQUIRRELVM vm)
+{
+	sq_getclass(vm, 1);
+	sq_push(vm, -1);
+	return 1;
+}
+
 struct SquirrelSafeCheck
 {
 	SquirrelSafeCheck(HSQUIRRELVM vm, int outputCount = 0) :
@@ -2390,6 +2418,14 @@ bool SquirrelVM::RegisterClass(ScriptClassDesc_t* pClassDesc)
 	sq_newclosure(vm_, IsValid_stub, 0);
 	sq_newslot(vm_, -3, SQFalse);
 
+	sq_pushstring(vm_, "weakref", -1);
+	sq_newclosure(vm_, weakref_stub, 0);
+	sq_newslot(vm_, -3, SQFalse);
+
+	sq_pushstring(vm_, "getclass", -1);
+	sq_newclosure(vm_, getclass_stub, 0);
+	sq_newslot(vm_, -3, SQFalse);
+
 
 	for (int i = 0; i < pClassDesc->m_FunctionBindings.Count(); ++i)
 	{