diff --git a/include/boost/python/suite/indexing/map_indexing_suite.hpp b/include/boost/python/suite/indexing/map_indexing_suite.hpp new file mode 100644 index 00000000..d8cb85ec --- /dev/null +++ b/include/boost/python/suite/indexing/map_indexing_suite.hpp @@ -0,0 +1,152 @@ +// (C) Copyright Joel de Guzman 2003. +// Permission to copy, use, modify, sell and distribute this software +// is granted provided this copyright notice appears in all copies. This +// software is provided "as is" without express or implied warranty, and +// with no claim as to its suitability for any purpose. + +#ifndef MAP_INDEXING_SUITE_JDG20038_HPP +# define MAP_INDEXING_SUITE_JDG20038_HPP + +# include +# include + +namespace boost { namespace python { + + // Forward declaration + template + class map_indexing_suite; + + namespace detail + { + template + class final_map_derived_policies + : public map_indexing_suite > {}; + } + + // The map_indexing_suite class is a predefined indexing_suite derived + // class for wrapping std::vector (and std::vector like) classes. It provides + // all the policies required by the indexing_suite (see indexing_suite). + // Example usage: + // + // class X {...}; + // + // ... + // + // class_ >("XMap") + // .def(map_indexing_suite >()) + // ; + // + // 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 > + class map_indexing_suite + : public indexing_suite< + Container + , DerivedPolicies + , NoProxy + , true + , typename Container::mapped_type + , typename Container::key_type + , typename Container::key_type + > + { + public: + + typedef typename Container::mapped_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 +// static void +// extension_def(Class& cl) +// { +// cl +// .def("append", &base_append) +// .def("extend", &base_extend) +// ; +// } + + static data_type& + get_item(Container& container, index_type i_) + { + typename Container::iterator i = container.find(i_); + key_check(container, i); + return i->second; + } + + static void + set_item(Container& container, index_type i_, data_type const& v) + { + typename Container::iterator i = container.find(i_); + key_check(container, i); + i->second = 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 i(i_); + if (i.check()) + { + return i(); + } + else + { + extract i(i_); + if (i.check()) + return i(); + } + + PyErr_SetString(PyExc_TypeError, "Invalid index type"); + throw_error_already_set(); + return index_type(); + } + + private: + + static void + key_check(Container& container, typename Container::iterator i) + { + if (i == container.end()) + { + PyErr_SetString(PyExc_KeyError, "Invalid key"); + throw_error_already_set(); + } + } + }; + +}} // namespace boost::python + +#endif // MAP_INDEXING_SUITE_JDG20038_HPP