1 /////////////////////////////////////////////////////////////////////////////
2 //
3 // (C) Copyright Ion Gaztanaga 2006-2013
4 //
5 // Distributed under the Boost Software License, Version 1.0.
6 // (See accompanying file LICENSE_1_0.txt or copy at
7 // http://www.boost.org/LICENSE_1_0.txt)
8 //
9 // See http://www.boost.org/libs/intrusive for documentation.
10 //
11 /////////////////////////////////////////////////////////////////////////////
12 //[doc_splay_set_code
13 #include <boost/intrusive/splay_set.hpp>
14 #include <vector>
15 #include <functional>
16
17 using namespace boost::intrusive;
18
19 class mytag;
20
21 class MyClass
22 : public bs_set_base_hook<>
23 {
24 int int_;
25
26 public:
27 //This is a member hook
28 bs_set_member_hook<> member_hook_;
29
MyClass(int i)30 MyClass(int i)
31 : int_(i)
32 {}
operator <(const MyClass & a,const MyClass & b)33 friend bool operator< (const MyClass &a, const MyClass &b)
34 { return a.int_ < b.int_; }
operator >(const MyClass & a,const MyClass & b)35 friend bool operator> (const MyClass &a, const MyClass &b)
36 { return a.int_ > b.int_; }
operator ==(const MyClass & a,const MyClass & b)37 friend bool operator== (const MyClass &a, const MyClass &b)
38 { return a.int_ == b.int_; }
39 };
40
41 //Define a set using the base hook that will store values in reverse order
42 typedef splay_set< MyClass, compare<std::greater<MyClass> > > BaseSplaySet;
43
44 //Define an multiset using the member hook
45 typedef member_hook<MyClass, bs_set_member_hook<>, &MyClass::member_hook_> MemberOption;
46 typedef splay_multiset< MyClass, MemberOption> MemberSplayMultiset;
47
main()48 int main()
49 {
50 typedef std::vector<MyClass>::iterator VectIt;
51
52 //Create several MyClass objects, each one with a different value
53 std::vector<MyClass> values;
54 for(int i = 0; i < 100; ++i) values.push_back(MyClass(i));
55
56 BaseSplaySet baseset;
57 MemberSplayMultiset membermultiset;
58
59
60 //Insert values in the container
61 for(VectIt it(values.begin()), itend(values.end()); it != itend; ++it){
62 baseset.insert(*it);
63 membermultiset.insert(*it);
64 }
65
66 //Now test sets
67 {
68 BaseSplaySet::reverse_iterator rbit(baseset.rbegin());
69 MemberSplayMultiset::iterator mit(membermultiset.begin());
70 VectIt it(values.begin()), itend(values.end());
71
72 //Test the objects inserted in the base hook set
73 for(; it != itend; ++it, ++rbit){
74 if(&*rbit != &*it) return 1;
75 }
76
77 //Test the objects inserted in member and binary search hook sets
78 for(it = values.begin(); it != itend; ++it, ++mit){
79 if(&*mit != &*it) return 1;
80 }
81 }
82 return 0;
83 }
84 //]
85