summaryrefslogtreecommitdiff
path: root/libcutl/tests/container/multi-index/driver.cxx
diff options
context:
space:
mode:
Diffstat (limited to 'libcutl/tests/container/multi-index/driver.cxx')
-rw-r--r--libcutl/tests/container/multi-index/driver.cxx292
1 files changed, 292 insertions, 0 deletions
diff --git a/libcutl/tests/container/multi-index/driver.cxx b/libcutl/tests/container/multi-index/driver.cxx
new file mode 100644
index 0000000..289ec4a
--- /dev/null
+++ b/libcutl/tests/container/multi-index/driver.cxx
@@ -0,0 +1,292 @@
+// file : tests/container/multi-index/driver.cxx
+// copyright : Copyright (c) 2009-2013 Code Synthesis Tools CC
+// license : MIT; see accompanying LICENSE file
+
+#include <map>
+#include <list>
+#include <string>
+#include <cassert>
+#include <iostream>
+
+#include <cutl/container/multi-index.hxx>
+
+using namespace std;
+using namespace cutl::container;
+
+struct person
+{
+ person (const string& e, const string& f, const string& l, unsigned short a)
+ : email (e), first (f), last (l), age (a) {}
+
+ const string email;
+ const string first;
+ const string last;
+ unsigned short age;
+};
+
+struct person_email_set
+{
+ typedef map<key<string>, person> email_map;
+ typedef map_const_iterator<email_map> const_iterator;
+
+ pair<const_iterator, bool>
+ insert (const person& v)
+ {
+ pair<email_map::iterator, bool> r (
+ email_map_.insert (email_map::value_type (v.email, v)));
+
+ const_iterator i (r.first);
+
+ if (r.second)
+ r.first->first.assign (i->email);
+
+ return make_pair (i, r.second);
+ }
+
+ const_iterator
+ find (const string& email) const
+ {
+ return email_map_.find (email);
+ }
+
+ const_iterator begin () const {return email_map_.begin ();}
+ const_iterator end () const {return email_map_.end ();}
+
+private:
+ email_map email_map_;
+};
+
+struct person_name_set
+{
+ typedef key<string, string> name_key;
+ typedef map<name_key, person> name_map;
+ typedef map_const_iterator<name_map> const_iterator;
+
+ pair<const_iterator, bool>
+ insert (const person& v)
+ {
+ pair<name_map::iterator, bool> r (
+ name_map_.insert (
+ name_map::value_type (name_key (v.first, v.last), v)));
+
+ const_iterator i (r.first);
+
+ if (r.second)
+ r.first->first.assign (i->first, i->last);
+
+ return make_pair (i, r.second);
+ }
+
+ const_iterator
+ find (const string& first, const string& last) const
+ {
+ return name_map_.find (name_key (first, last));
+ }
+
+ const_iterator begin () const {return name_map_.begin ();}
+ const_iterator end () const {return name_map_.end ();}
+
+private:
+ name_map name_map_;
+};
+
+struct person_email_name_set
+{
+ typedef key<string, string> name_key;
+ typedef map<name_key, person> name_map;
+ typedef map_iterator<name_map> iterator;
+ typedef map_const_iterator<name_map> const_iterator;
+
+ typedef map<key<string>, iterator> email_map;
+
+ pair<iterator, bool>
+ insert (const person& v)
+ {
+ // First check that we don't have any collisions in the secondary
+ // indexes.
+ //
+ {
+ email_map::iterator i (email_map_.find (v.email));
+
+ if (i != email_map_.end ())
+ return make_pair (i->second, false);
+ }
+
+ pair<name_map::iterator, bool> r (
+ name_map_.insert (
+ name_map::value_type (name_key (v.first, v.last), v)));
+
+ iterator i (r.first);
+
+ if (r.second)
+ {
+ r.first->first.assign (i->first, i->last);
+ email_map_.insert (email_map::value_type (i->email, i));
+ }
+
+ return make_pair (i, r.second);
+ }
+
+ iterator
+ find (const string& first, const string& last)
+ {
+ return name_map_.find (name_key (first, last));
+ }
+
+ const_iterator
+ find (const string& first, const string& last) const
+ {
+ return name_map_.find (name_key (first, last));
+ }
+
+ iterator
+ find (const string& email)
+ {
+ email_map::iterator i (email_map_.find (email));
+ return i != email_map_.end () ? i->second : end ();
+ }
+
+ const_iterator
+ find (const string& email) const
+ {
+ email_map::const_iterator i (email_map_.find (email));
+ return i != email_map_.end () ? i->second : end ();
+ }
+
+ void
+ erase (iterator i )
+ {
+ email_map_.erase (i->email);
+ name_map_.erase (i);
+ }
+
+ iterator begin () {return name_map_.begin ();}
+ const_iterator begin () const {return name_map_.begin ();}
+
+ iterator end () {return name_map_.end ();}
+ const_iterator end () const {return name_map_.end ();}
+
+private:
+ name_map name_map_;
+ email_map email_map_;
+};
+
+struct person_list_email_set
+{
+ typedef list<person> person_list;
+ typedef person_list::iterator iterator;
+ typedef person_list::const_iterator const_iterator;
+
+ typedef map<key<string>, iterator> email_map;
+
+ pair<iterator, bool>
+ insert (const person& v)
+ {
+ // First check that we don't have any collisions in the secondary
+ // indexes.
+ //
+ {
+ email_map::iterator i (email_map_.find (v.email));
+
+ if (i != email_map_.end ())
+ return make_pair (i->second, false);
+ }
+
+ iterator i (person_list_.insert (end (), v));
+ email_map_.insert (email_map::value_type (i->email, i));
+ return make_pair (i, true);
+ }
+
+ iterator
+ find (const string& email)
+ {
+ email_map::iterator i (email_map_.find (email));
+ return i != email_map_.end () ? i->second : end ();
+ }
+
+ const_iterator
+ find (const string& email) const
+ {
+ email_map::const_iterator i (email_map_.find (email));
+ return i != email_map_.end () ? i->second : end ();
+ }
+
+ iterator begin () {return person_list_.begin ();}
+ const_iterator begin () const {return person_list_.begin ();}
+
+ iterator end () {return person_list_.end ();}
+ const_iterator end () const {return person_list_.end ();}
+
+private:
+ person_list person_list_;
+ email_map email_map_;
+};
+
+int
+main ()
+{
+ {
+ person_email_set s;
+
+ assert (s.insert (person ("john@doe.com", "John", "Doe", 20)).second);
+ assert (s.insert (person ("jane@doe.com", "Jane", "Doe", 21)).second);
+ assert (!s.insert (person ("john@doe.com", "Johnny", "Doe", 22)).second);
+
+ assert (s.find ("john@doe.com") != s.end ());
+ assert (s.find ("jane@doe.com") != s.end ());
+ assert (s.find ("john@doe.org") == s.end ());
+ }
+
+ {
+ person_name_set s;
+
+ assert (s.insert (person ("john@doe.com", "John", "Doe", 20)).second);
+ assert (s.insert (person ("jane@doe.com", "Jane", "Doe", 21)).second);
+ assert (!s.insert (person ("john@doe.org", "John", "Doe", 22)).second);
+
+ assert (s.find ("John", "Doe") != s.end ());
+ assert (s.find ("Jane", "Doe") != s.end ());
+ assert (s.find ("Johnny", "Doe") == s.end ());
+ }
+
+ {
+ person_email_name_set s;
+ person_email_name_set const& cs (s);
+
+ assert (s.insert (person ("john@doe.com", "John", "Doe", 20)).second);
+ assert (s.insert (person ("jane@doe.com", "Jane", "Doe", 21)).second);
+ assert (!s.insert (person ("john@doe.org", "John", "Doe", 22)).second);
+ assert (!s.insert (person ("john@doe.com", "Johnny", "Doe", 23)).second);
+
+ assert (s.find ("John", "Doe") != s.end ());
+ assert (cs.find ("Jane", "Doe") != cs.end ());
+ assert (s.find ("john@doe.com") != s.end ());
+ assert (cs.find ("jane@doe.com") != s.end ());
+ assert (s.find ("Johnny", "Doe") == s.end ());
+ assert (cs.find ("john@doe.org") == s.end ());
+
+ person_email_name_set::iterator i (s.find ("John", "Doe"));
+ i->age++;
+
+ s.erase (i);
+ assert (s.find ("John", "Doe") == s.end ());
+ assert (s.find ("john@doe.com") == s.end ());
+ }
+
+ {
+ person_list_email_set s;
+
+ assert (s.insert (person ("john@doe.com", "John", "Doe", 20)).second);
+ assert (s.insert (person ("jane@doe.com", "Jane", "Doe", 21)).second);
+ assert (!s.insert (person ("john@doe.com", "Johnny", "Doe", 22)).second);
+
+ assert (s.find ("john@doe.com") != s.end ());
+ assert (s.find ("jane@doe.com") != s.end ());
+ assert (s.find ("jane@doe.org") == s.end ());
+
+ person_list_email_set::iterator i (s.begin ());
+ assert (i != s.end () && i->email == "john@doe.com");
+ assert (++i != s.end () && i->email == "jane@doe.com");
+ assert (++i == s.end ());
+ }
+}