split_adaptor.hpp 3.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102
  1. /*
  2. [auto_generated]
  3. boost/numeric/odeint/util/split_adaptor.hpp
  4. [begin_description]
  5. A range adaptor which returns even-sized slices.
  6. [end_description]
  7. Copyright 2013 Karsten Ahnert
  8. Copyright 2013 Mario Mulansky
  9. Copyright 2013 Pascal Germroth
  10. Distributed under the Boost Software License, Version 1.0.
  11. (See accompanying file LICENSE_1_0.txt or
  12. copy at http://www.boost.org/LICENSE_1_0.txt)
  13. */
  14. #ifndef BOOST_NUMERIC_ODEINT_UTIL_SPLIT_ADAPTOR_INCLUDED
  15. #define BOOST_NUMERIC_ODEINT_UTIL_SPLIT_ADAPTOR_INCLUDED
  16. #include <boost/range/adaptor/argument_fwd.hpp>
  17. #include <boost/range/size_type.hpp>
  18. #include <boost/range/iterator_range.hpp>
  19. #include <algorithm>
  20. namespace boost {
  21. namespace numeric {
  22. namespace odeint {
  23. namespace detail {
  24. /** \brief Returns the begin and end offset for a sub-range */
  25. inline std::pair<std::size_t, std::size_t>
  26. split_offsets( std::size_t total_length, std::size_t index, std::size_t parts )
  27. {
  28. BOOST_ASSERT( parts > 0 );
  29. BOOST_ASSERT( index < parts );
  30. const std::size_t
  31. slice = total_length / parts,
  32. partial = total_length % parts,
  33. lo = (std::min)(index, partial),
  34. hi = (std::max<std::ptrdiff_t>)(0, index - partial),
  35. begin_offset = lo * (slice + 1) + hi * slice,
  36. length = slice + (index < partial ? 1 : 0),
  37. end_offset = begin_offset + length;
  38. return std::make_pair( begin_offset, end_offset );
  39. }
  40. /** \brief Return the sub-range `index` from a range which is split into `parts`.
  41. *
  42. * For example, splitting a range into three about equal-sized sub-ranges:
  43. * \code
  44. * sub0 = make_split_range(rng, 0, 3);
  45. * sub1 = rng | split(1, 3);
  46. * sub2 = rng | split(2, 3);
  47. * \endcode
  48. */
  49. template< class RandomAccessRange >
  50. inline iterator_range< typename range_iterator<RandomAccessRange>::type >
  51. make_split_range( RandomAccessRange& rng, std::size_t index, std::size_t parts )
  52. {
  53. const std::pair<std::size_t, std::size_t> off = split_offsets(boost::size(rng), index, parts);
  54. return make_iterator_range( boost::begin(rng) + off.first, boost::begin(rng) + off.second );
  55. }
  56. template< class RandomAccessRange >
  57. inline iterator_range< typename range_iterator<const RandomAccessRange>::type >
  58. make_split_range( const RandomAccessRange& rng, std::size_t index, std::size_t parts )
  59. {
  60. const std::pair<std::size_t, std::size_t> off = split_offsets(boost::size(rng), index, parts);
  61. return make_iterator_range( boost::begin(rng) + off.first, boost::begin(rng) + off.second );
  62. }
  63. struct split
  64. {
  65. split(std::size_t index, std::size_t parts)
  66. : index(index), parts(parts) {}
  67. std::size_t index, parts;
  68. };
  69. template< class RandomAccessRange >
  70. inline iterator_range< typename range_iterator<RandomAccessRange>::type >
  71. operator|( RandomAccessRange& rng, const split& f )
  72. {
  73. return make_split_range( rng, f.index, f.parts );
  74. }
  75. template< class RandomAccessRange >
  76. inline iterator_range< typename range_iterator<const RandomAccessRange>::type >
  77. operator|( const RandomAccessRange& rng, const split& f )
  78. {
  79. return make_split_range( rng, f.index, f.parts );
  80. }
  81. }
  82. }
  83. }
  84. }
  85. #endif