1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798 |
- #include <boost/python.hpp>
- #include <boost/python/slice.hpp>
- #include <boost/python/str.hpp>
- #include <vector>
- // Copyright (c) 2004 Jonathan Brandmeyer
- // Use, modification and distribution are subject to 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)
- using namespace boost::python;
- #if BOOST_WORKAROUND(__SUNPRO_CC, BOOST_TESTED_AT(0x580)) || BOOST_WORKAROUND(BOOST_MSVC, BOOST_TESTED_AT(1500))
- # define make_tuple boost::python::make_tuple
- #endif
- // These checks are only valid under Python 2.3
- // (rich slicing wasn't supported for builtins under Python 2.2)
- bool check_string_rich_slice()
- {
- object s("hello, world");
- // default slice
- if (s[slice()] != "hello, world")
- return false;
- // simple reverse
- if (s[slice(_,_,-1)] != "dlrow ,olleh")
- return false;
- // reverse with mixed-sign offsets
- if (s[slice(-6,1,-1)] != " ,oll")
- return false;
- // all of the object.cpp check_string_slice() checks should work
- // with the form that omits the step argument.
- if (s[slice(_,-3)] != "hello, wo")
- return false;
- if (s[slice(-3,_)] != "rld")
- return false;
- if (", " != s[slice(5,7)])
- return false;
- return s[slice(2,-1)][slice(1,-1)] == "lo, wor";
- }
- // Tried to get more info into the error message (actual array
- // contents) but Numeric complains that treating an array as a boolean
- // value doesn't make any sense.
- #define ASSERT_EQUAL( e1, e2 ) \
- if (!all((e1) == (e2))) \
- return "assertion failed: " #e1 " == " #e2 "\nLHS:\n%s\nRHS:\n%s" % make_tuple(e1,e2); \
- else
- // Verify functions accepting a slice argument can be called
- bool accept_slice( slice) { return true; }
- #if BOOST_WORKAROUND( BOOST_MSVC, BOOST_TESTED_AT(1400)) \
- || BOOST_WORKAROUND( BOOST_INTEL_WIN, == 710)
- int check_slice_get_indices(slice index);
- #endif
- int check_slice_get_indices(
- #if !BOOST_WORKAROUND(__SUNPRO_CC, BOOST_TESTED_AT(0x590))
- const
- #endif
- slice index)
- {
- // A vector of integers from [-5, 5].
- std::vector<int> coll(11);
- typedef std::vector<int>::iterator coll_iterator;
-
- for (coll_iterator i = coll.begin(); i != coll.end(); ++i) {
- *i = i - coll.begin() - 5;
- }
-
- slice::range<std::vector<int>::iterator> bounds;
- try {
- bounds = index.get_indices(coll.begin(), coll.end());
- }
- catch (std::invalid_argument) {
- return 0;
- }
- int sum = 0;
- while (bounds.start != bounds.stop) {
- sum += *bounds.start;
- std::advance( bounds.start, bounds.step);
- }
- sum += *bounds.start;
- return sum;
- }
- BOOST_PYTHON_MODULE(slice_ext)
- {
- def( "accept_slice", accept_slice);
- def( "check_string_rich_slice", check_string_rich_slice);
- def( "check_slice_get_indices", check_slice_get_indices);
- }
|