1 | // (C) Copyright Joel de Guzman 2003. |
---|
2 | // Distributed under the Boost Software License, Version 1.0. (See |
---|
3 | // accompanying file LICENSE_1_0.txt or copy at |
---|
4 | // http://www.boost.org/LICENSE_1_0.txt) |
---|
5 | |
---|
6 | #ifndef MAP_INDEXING_SUITE_JDG20038_HPP |
---|
7 | # define MAP_INDEXING_SUITE_JDG20038_HPP |
---|
8 | |
---|
9 | # include "indexing_suite.hpp" |
---|
10 | # include <boost/python/iterator.hpp> |
---|
11 | # include <boost/python/call_method.hpp> |
---|
12 | # include <boost/python/tuple.hpp> |
---|
13 | |
---|
14 | namespace boost { namespace python { |
---|
15 | |
---|
16 | // Forward declaration |
---|
17 | template <class Container, bool NoProxy, class DerivedPolicies> |
---|
18 | class map_indexing_suite; |
---|
19 | |
---|
20 | namespace detail |
---|
21 | { |
---|
22 | template <class Container, bool NoProxy> |
---|
23 | class final_map_derived_policies |
---|
24 | : public map_indexing_suite<Container, |
---|
25 | NoProxy, final_map_derived_policies<Container, NoProxy> > {}; |
---|
26 | } |
---|
27 | |
---|
28 | // The map_indexing_suite class is a predefined indexing_suite derived |
---|
29 | // class for wrapping std::vector (and std::vector like) classes. It provides |
---|
30 | // all the policies required by the indexing_suite (see indexing_suite). |
---|
31 | // Example usage: |
---|
32 | // |
---|
33 | // class X {...}; |
---|
34 | // |
---|
35 | // ... |
---|
36 | // |
---|
37 | // class_<std::map<std::string, X> >("XMap") |
---|
38 | // .def(map_indexing_suite<std::map<std::string, X> >()) |
---|
39 | // ; |
---|
40 | // |
---|
41 | // By default indexed elements are returned by proxy. This can be |
---|
42 | // disabled by supplying *true* in the NoProxy template parameter. |
---|
43 | // |
---|
44 | template < |
---|
45 | class Container, |
---|
46 | bool NoProxy = false, |
---|
47 | class DerivedPolicies |
---|
48 | = detail::final_map_derived_policies<Container, NoProxy> > |
---|
49 | class map_indexing_suite |
---|
50 | : public indexing_suite< |
---|
51 | Container |
---|
52 | , DerivedPolicies |
---|
53 | , NoProxy |
---|
54 | , true |
---|
55 | , typename Container::value_type::second_type |
---|
56 | , typename Container::key_type |
---|
57 | , typename Container::key_type |
---|
58 | > |
---|
59 | { |
---|
60 | public: |
---|
61 | |
---|
62 | typedef typename Container::value_type value_type; |
---|
63 | typedef typename Container::value_type::second_type data_type; |
---|
64 | typedef typename Container::key_type key_type; |
---|
65 | typedef typename Container::key_type index_type; |
---|
66 | typedef typename Container::size_type size_type; |
---|
67 | typedef typename Container::difference_type difference_type; |
---|
68 | |
---|
69 | template <class Class> |
---|
70 | static void |
---|
71 | extension_def(Class& cl) |
---|
72 | { |
---|
73 | // Wrap the map's element (value_type) |
---|
74 | std::string elem_name = "map_indexing_suite_"; |
---|
75 | object class_name(cl.attr("__name__")); |
---|
76 | extract<std::string> class_name_extractor(class_name); |
---|
77 | elem_name += class_name_extractor(); |
---|
78 | elem_name += "_entry"; |
---|
79 | |
---|
80 | typedef typename mpl::if_< |
---|
81 | is_class<data_type> |
---|
82 | , return_internal_reference<> |
---|
83 | , default_call_policies |
---|
84 | >::type get_data_return_policy; |
---|
85 | |
---|
86 | class_<value_type>(elem_name.c_str()) |
---|
87 | .def("__repr__", &DerivedPolicies::print_elem) |
---|
88 | .def("data", &DerivedPolicies::get_data, get_data_return_policy()) |
---|
89 | .def("key", &DerivedPolicies::get_key) |
---|
90 | ; |
---|
91 | } |
---|
92 | |
---|
93 | static object |
---|
94 | print_elem(typename Container::value_type const& e) |
---|
95 | { |
---|
96 | return "(%s, %s)" % python::make_tuple(e.first, e.second); |
---|
97 | } |
---|
98 | |
---|
99 | static |
---|
100 | typename mpl::if_< |
---|
101 | is_class<data_type> |
---|
102 | , data_type& |
---|
103 | , data_type |
---|
104 | >::type |
---|
105 | get_data(typename Container::value_type& e) |
---|
106 | { |
---|
107 | return e.second; |
---|
108 | } |
---|
109 | |
---|
110 | static typename Container::key_type |
---|
111 | get_key(typename Container::value_type& e) |
---|
112 | { |
---|
113 | return e.first; |
---|
114 | } |
---|
115 | |
---|
116 | static data_type& |
---|
117 | get_item(Container& container, index_type i_) |
---|
118 | { |
---|
119 | typename Container::iterator i = container.find(i_); |
---|
120 | if (i == container.end()) |
---|
121 | { |
---|
122 | PyErr_SetString(PyExc_KeyError, "Invalid key"); |
---|
123 | throw_error_already_set(); |
---|
124 | } |
---|
125 | return i->second; |
---|
126 | } |
---|
127 | |
---|
128 | static void |
---|
129 | set_item(Container& container, index_type i, data_type const& v) |
---|
130 | { |
---|
131 | container[i] = v; |
---|
132 | } |
---|
133 | |
---|
134 | static void |
---|
135 | delete_item(Container& container, index_type i) |
---|
136 | { |
---|
137 | container.erase(i); |
---|
138 | } |
---|
139 | |
---|
140 | static size_t |
---|
141 | size(Container& container) |
---|
142 | { |
---|
143 | return container.size(); |
---|
144 | } |
---|
145 | |
---|
146 | static bool |
---|
147 | contains(Container& container, key_type const& key) |
---|
148 | { |
---|
149 | return container.find(key) != container.end(); |
---|
150 | } |
---|
151 | |
---|
152 | static bool |
---|
153 | compare_index(Container& container, index_type a, index_type b) |
---|
154 | { |
---|
155 | return container.key_comp()(a, b); |
---|
156 | } |
---|
157 | |
---|
158 | static index_type |
---|
159 | convert_index(Container& container, PyObject* i_) |
---|
160 | { |
---|
161 | extract<key_type const&> i(i_); |
---|
162 | if (i.check()) |
---|
163 | { |
---|
164 | return i(); |
---|
165 | } |
---|
166 | else |
---|
167 | { |
---|
168 | extract<key_type> i(i_); |
---|
169 | if (i.check()) |
---|
170 | return i(); |
---|
171 | } |
---|
172 | |
---|
173 | PyErr_SetString(PyExc_TypeError, "Invalid index type"); |
---|
174 | throw_error_already_set(); |
---|
175 | return index_type(); |
---|
176 | } |
---|
177 | }; |
---|
178 | |
---|
179 | }} // namespace boost::python |
---|
180 | |
---|
181 | #endif // MAP_INDEXING_SUITE_JDG20038_HPP |
---|