BLI: add non-const lookup methods for StringMap

1 files changed, 10 insertions, 0 deletions

diff --git a/source/blender/blenlib/BLI_string_map.h b/source/blender/blenlib/BLI_string_map.h
index 0cf0bb40e4d..e7eb1e2635e 100644
--- a/source/blender/blenlib/BLI_string_map.h
+++ b/source/blender/blenlib/BLI_string_map.h
@@ -248,6 +248,11 @@ template<typename T, typename Allocator = GuardedAllocator> class StringMap {
     ITER_SLOTS_END(offset);
   }
 
+  T &lookup(StringRef key)
+  {
+    return const_cast<T &>(const_cast<const StringMap *>(this)->lookup(key));
+  }
+
   /**
    * Get a pointer to the value corresponding to the key. Return nullptr, if the key does not
    * exist.
@@ -266,6 +271,11 @@ template<typename T, typename Allocator = GuardedAllocator> class StringMap {
     ITER_SLOTS_END(offset);
   }
 
+  T *lookup_ptr(StringRef key)
+  {
+    return const_cast<T *>(const_cast<const StringMap *>(this)->lookup_ptr(key));
+  }
+
   /**
    * Get a copy of the value corresponding to the key. If the key does not exist, return the
    * default value.