diff --git a/src/ai/api/ai_abstractlist.cpp b/src/ai/api/ai_abstractlist.cpp index 9357f39c50..5fa0a1b238 100644 --- a/src/ai/api/ai_abstractlist.cpp +++ b/src/ai/api/ai_abstractlist.cpp @@ -6,6 +6,7 @@ #include "ai_abstractlist.hpp" #include "../../debug.h" #include "../../core/alloc_func.hpp" +#include "../../script/squirrel.hpp" /** * Base class for any AIAbstractList sorter. @@ -818,6 +819,8 @@ SQInteger AIAbstractList::Valuate(HSQUIRRELVM vm) { (*iter).second = (int32)value; this->buckets[(int32)value].insert((*iter).first); + + Squirrel::DecreaseOps(vm, 5); } sq_release(vm, &obj_func); diff --git a/src/script/squirrel.cpp b/src/script/squirrel.cpp index 1400eaf0a6..602cc593d2 100644 --- a/src/script/squirrel.cpp +++ b/src/script/squirrel.cpp @@ -454,3 +454,8 @@ void Squirrel::InsertResult(int result) vm->GetAt(vm->_stackbase + vm->_suspended_target) = vm->GetUp(-1); vm->Pop(); } + +/* static */ void Squirrel::DecreaseOps(HSQUIRRELVM vm, int ops) +{ + vm->DecreaseOps(ops); +} diff --git a/src/script/squirrel.hpp b/src/script/squirrel.hpp index fc7472e964..b5b352bd02 100644 --- a/src/script/squirrel.hpp +++ b/src/script/squirrel.hpp @@ -186,6 +186,11 @@ public: * Release a SQ object. */ void ReleaseObject(HSQOBJECT *ptr) { sq_release(this->vm, ptr); } + + /** + * Tell the VM to remove \c amount ops from the number of ops till suspend. + */ + static void DecreaseOps(HSQUIRRELVM vm, int amount); }; #endif /* SQUIRREL_HPP */