s [skip ci]

Author: firewave

lib/programmemory.cpp

@@ -173,15 +173,27 @@ void ProgramMemory::erase_if(const std::function<bool(const ExprIdToken&)>& pred
     if (mValues->empty())
         return;
 
-    // TODO: messes with the use count
-    const auto values = mValues;
+    if (mValues.use_count() == 1)
+    {
+        // we are the only user so can use it directly
+        for (auto it = mValues->begin(); it != mValues->end();) {
+            if (pred(it->first))
+                it = mValues->erase(it);
+            else
+                ++it;
+        }
+    }
+    else
+    {
+        const auto values = mValues;
 
-    for (auto it = values->begin(); it != values->end(); ++it) {
-        if (!pred(it->first))
-            continue;
+        for (auto it = values->begin(); it != values->end(); ++it) {
+            if (!pred(it->first))
+                continue;
 
-        copyOnWrite();
-        mValues->erase(it->first);
+            copyOnWrite();
+            mValues->erase(it->first);
+        }
     }
 }