123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181 |
- // (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
|