adding and testing opIndex usage

dlang · May 25, 2024 · b14fee7 · b14fee7
1 parent 254e5cc
commit b14fee7
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 0 deletions.
diff --git a/extras/AssociativeContainers/map.hpp b/extras/AssociativeContainers/map.hpp
@@ -23,4 +23,6 @@ class Map
     static Map<K,V>* make()         { return new Map<K, V>(); }
 
     size_t size()                   { return this->map.size(); }
+
+    V& operator[] (K const& key)    { return this->map[key]; }
 };
diff --git a/source/stdcpp/map.d b/source/stdcpp/map.d
@@ -19,6 +19,15 @@ module stdcpp.map;
     }
 
     static Map* make();
+
     private void insertOrAssign(const ref Key, const ref Value);
+
     size_t size();
+
+    ref Value opIndex( ref const Key key);
+
+    ref Value opIndex(const Key key)
+    {
+        return this.opIndex(key); // handle rvalue-ref
+    }
  }
diff --git a/source/stdcpp/test/map.d b/source/stdcpp/test/map.d
@@ -13,4 +13,8 @@ unittest
     auto mymap = Map!(int, char).make();
     mymap.opIndexAssign(1, 'a');
     assert(mymap.size == 1);
+    assert(mymap.opIndex(1) == 'a');
+    mymap.opIndex(2) = 'b';
+    mymap.opIndex(3) = 'c';
+    assert(mymap.size == 3);
 }