| // (C) Copyright Joel de Guzman 2003. |
| // Distributed under the Boost Software License, Version 1.0. (See |
| // accompanying file LICENSE_1_0.txt or copy at |
| // http://www.boost.org/LICENSE_1_0.txt) |
| |
| #ifndef MAP_INDEXING_SUITE_JDG20038_HPP |
| # define MAP_INDEXING_SUITE_JDG20038_HPP |
| |
| # include <boost/python/suite/indexing/indexing_suite.hpp> |
| # include <boost/python/iterator.hpp> |
| # include <boost/python/call_method.hpp> |
| # include <boost/python/tuple.hpp> |
| |
| namespace boost { namespace python { |
| |
| // Forward declaration |
| template <class Container, bool NoProxy, class DerivedPolicies> |
| class map_indexing_suite; |
| |
| namespace detail |
| { |
| template <class Container, bool NoProxy> |
| class final_map_derived_policies |
| : public map_indexing_suite<Container, |
| NoProxy, final_map_derived_policies<Container, NoProxy> > {}; |
| } |
| |
| // The map_indexing_suite class is a predefined indexing_suite derived |
| // class for wrapping std::map (and std::map like) classes. It provides |
| // all the policies required by the indexing_suite (see indexing_suite). |
| // Example usage: |
| // |
| // class X {...}; |
| // |
| // ... |
| // |
| // class_<std::map<std::string, X> >("XMap") |
| // .def(map_indexing_suite<std::map<std::string, X> >()) |
| // ; |
| // |
| // By default indexed elements are returned by proxy. This can be |
| // disabled by supplying *true* in the NoProxy template parameter. |
| // |
| template < |
| class Container, |
| bool NoProxy = false, |
| class DerivedPolicies |
| = detail::final_map_derived_policies<Container, NoProxy> > |
| class map_indexing_suite |
| : public indexing_suite< |
| Container |
| , DerivedPolicies |
| , NoProxy |
| , true |
| , typename Container::value_type::second_type |
| , typename Container::key_type |
| , typename Container::key_type |
| > |
| { |
| public: |
| |
| typedef typename Container::value_type value_type; |
| typedef typename Container::value_type::second_type data_type; |
| typedef typename Container::key_type key_type; |
| typedef typename Container::key_type index_type; |
| typedef typename Container::size_type size_type; |
| typedef typename Container::difference_type difference_type; |
| |
| template <class Class> |
| static void |
| extension_def(Class& cl) |
| { |
| // Wrap the map's element (value_type) |
| std::string elem_name = "map_indexing_suite_"; |
| object class_name(cl.attr("__name__")); |
| extract<std::string> class_name_extractor(class_name); |
| elem_name += class_name_extractor(); |
| elem_name += "_entry"; |
| |
| typedef typename mpl::if_< |
| mpl::and_<is_class<data_type>, mpl::bool_<!NoProxy> > |
| , return_internal_reference<> |
| , default_call_policies |
| >::type get_data_return_policy; |
| |
| class_<value_type>(elem_name.c_str()) |
| .def("__repr__", &DerivedPolicies::print_elem) |
| .def("data", &DerivedPolicies::get_data, get_data_return_policy()) |
| .def("key", &DerivedPolicies::get_key) |
| ; |
| } |
| |
| static object |
| print_elem(typename Container::value_type const& e) |
| { |
| return "(%s, %s)" % python::make_tuple(e.first, e.second); |
| } |
| |
| static |
| typename mpl::if_< |
| mpl::and_<is_class<data_type>, mpl::bool_<!NoProxy> > |
| , data_type& |
| , data_type |
| >::type |
| get_data(typename Container::value_type& e) |
| { |
| return e.second; |
| } |
| |
| static typename Container::key_type |
| get_key(typename Container::value_type& e) |
| { |
| return e.first; |
| } |
| |
| static data_type& |
| get_item(Container& container, index_type i_) |
| { |
| typename Container::iterator i = container.find(i_); |
| if (i == container.end()) |
| { |
| PyErr_SetString(PyExc_KeyError, "Invalid key"); |
| throw_error_already_set(); |
| } |
| return i->second; |
| } |
| |
| static void |
| set_item(Container& container, index_type i, data_type const& v) |
| { |
| container[i] = v; |
| } |
| |
| static void |
| delete_item(Container& container, index_type i) |
| { |
| container.erase(i); |
| } |
| |
| static size_t |
| size(Container& container) |
| { |
| return container.size(); |
| } |
| |
| static bool |
| contains(Container& container, key_type const& key) |
| { |
| return container.find(key) != container.end(); |
| } |
| |
| static bool |
| compare_index(Container& container, index_type a, index_type b) |
| { |
| return container.key_comp()(a, b); |
| } |
| |
| static index_type |
| convert_index(Container& /*container*/, PyObject* i_) |
| { |
| extract<key_type const&> i(i_); |
| if (i.check()) |
| { |
| return i(); |
| } |
| else |
| { |
| extract<key_type> i(i_); |
| if (i.check()) |
| return i(); |
| } |
| |
| PyErr_SetString(PyExc_TypeError, "Invalid index type"); |
| throw_error_already_set(); |
| return index_type(); |
| } |
| }; |
| |
| }} // namespace boost::python |
| |
| #endif // MAP_INDEXING_SUITE_JDG20038_HPP |