Added buffer_vector::erase_if.

2 files changed, 30 insertions, 0 deletions

diff --git a/base/base_tests/buffer_vector_test.cpp b/base/base_tests/buffer_vector_test.cpp
index 40a34b5615..b63f3c929f 100644
--- a/base/base_tests/buffer_vector_test.cpp
+++ b/base/base_tests/buffer_vector_test.cpp
@@ -331,3 +331,23 @@ UNIT_TEST(BufferVectorMove)
   v1 = move(v2);
   TestVector(v1, 3);
 }
+
+UNIT_TEST(BufferVector_EraseIf)
+{
+  buffer_vector<int, 2> v;
+  v.push_back(1);
+  v.push_back(2);
+  v.erase_if([] (int x) { return x == 1; });
+  TEST_EQUAL(v.size(), 1, ());
+  TEST_EQUAL(v[0], 2, ());
+
+  v.push_back(3);
+  v.push_back(4);
+  v.erase_if([] (int x) { return x == 3; });
+  TEST_EQUAL(v.size(), 2, ());
+  TEST_EQUAL(v[0], 2, ());
+  TEST_EQUAL(v[1], 4, ());
+
+  v.erase_if([] (int x) { return true; });
+  TEST_EQUAL(v.size(), 0, ());
+}
diff --git a/base/buffer_vector.hpp b/base/buffer_vector.hpp
index 0d012209dd..df303b12a5 100644
--- a/base/buffer_vector.hpp
+++ b/base/buffer_vector.hpp
@@ -373,6 +373,16 @@ public:
     insert(where, &value, &value + 1);
   }
 
+  template <class TFn>
+  void erase_if(TFn fn)
+  {
+    iterator b = begin();
+    iterator e = end();
+    iterator i = remove_if(b, e, fn);
+    if (i != e)
+      resize(distance(b, i));
+  }
+
 private:
   void SwitchToDynamic()
   {