Add optional result for insertion in std::set

Author: sethrj

Examples/test-suite/fortran/li_std_set_runme.F90

@@ -52,17 +52,20 @@ subroutine test_set_string
   use li_std_set, only : set_string
   implicit none
   type(set_string) :: s
+  logical :: inserted = .false.
 
   s = set_string()
   ASSERT(s%empty())
 
   call s%insert("yoohoo")
-  call s%insert("howdy")
+  call s%insert("howdy", inserted)
+  ASSERT(inserted .eqv. .true.)
   ASSERT(s%size() == 2)
   ASSERT(s%count("yoohoo") == 1)
   ASSERT(s%count("hiya") == 0)
 
-  call s%insert("howdy")
+  call s%insert("howdy", inserted)
+  ASSERT(inserted .eqv. .false.)
   ASSERT(s%count("howdy") == 1)
   ASSERT(s%size() == 2)
 

Lib/fortran/std_multiset.i

@@ -9,6 +9,8 @@
 
 %define %swig_std_multiset(_Key, _Compare, _Alloc)
   %swig_std_setcommon(multiset, _Key, _Compare, _Alloc)
+
+  void insert(const_reference x);
 %enddef
 
 namespace std {

Lib/fortran/std_set.i

@@ -37,11 +37,19 @@ public:
 
   size_type erase(const key_type& x);
   size_type count(const key_type& x) const;
-  void insert(const_reference x);
 %enddef
 
 %define %swig_std_set(_Key, _Compare, _Alloc)
   %swig_std_setcommon(set, _Key, _Compare, _Alloc)
+
+  %fortransubroutine insert;
+
+  %extend {
+    bool insert(const_reference x) {
+      std::pair<std::set<_Key, _Compare, _Alloc >::iterator, bool> result = $self->insert(x);
+      return result.second;
+    }
+  }
 %enddef
 
 namespace std {