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