VM: use std::erase_if() for VM::sweep()

mrunix00 · Aug 1, 2024 · 2eef162 · 2eef162
1 parent 4cfc210
commit 2eef162
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/src/vm.cpp b/src/vm.cpp
@@ -175,15 +175,15 @@ void VM::markAll() {
     }
 }
 void VM::sweep() {
-    for (auto it = objects.begin(); it != objects.end();) {
-        if (!(*it)->marked) {
-            delete *it;
-            it = objects.erase(it);
+    std::erase_if(objects, [](Object *obj) {
+        if (!obj->marked) {
+            delete obj;
+            return true;
         } else {
-            (*it)->marked = false;
-            it++;
+            obj->marked = false;
+            return false;
         }
-    }
+    });
 }
 
 void VM::run(const Program &program) {