244 lines
		
	
	
		
			7.8 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
		
		
			
		
	
	
			244 lines
		
	
	
		
			7.8 KiB
		
	
	
	
		
			C++
		
	
	
	
	
	
|   | // Boost.Range library
 | ||
|  | //
 | ||
|  | //  Copyright Neil Groves 2010. Use, modification and
 | ||
|  | //  distribution is 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)
 | ||
|  | //
 | ||
|  | //
 | ||
|  | // For more information, see http://www.boost.org/libs/range/
 | ||
|  | //
 | ||
|  | #ifndef BOOST_RANGE_IRANGE_HPP_INCLUDED
 | ||
|  | #define BOOST_RANGE_IRANGE_HPP_INCLUDED
 | ||
|  | 
 | ||
|  | #include <boost/assert.hpp>
 | ||
|  | #include <boost/iterator/iterator_facade.hpp>
 | ||
|  | #include <boost/range/iterator_range.hpp>
 | ||
|  | 
 | ||
|  | namespace boost | ||
|  | { | ||
|  |     namespace range_detail | ||
|  |     { | ||
|  |         // integer_iterator is an iterator over an integer sequence that
 | ||
|  |         // is bounded only by the limits of the underlying integer
 | ||
|  |         // representation.
 | ||
|  |         //
 | ||
|  |         // This is useful for implementing the irange(first, last)
 | ||
|  |         // function.
 | ||
|  |         //
 | ||
|  |         // Note:
 | ||
|  |         // This use of this iterator and irange is appreciably less
 | ||
|  |         // performant than the corresponding hand-written integer
 | ||
|  |         // loop on many compilers.
 | ||
|  |         template<typename Integer> | ||
|  |         class integer_iterator | ||
|  |             : public boost::iterator_facade< | ||
|  |                         integer_iterator<Integer>, | ||
|  |                         Integer, | ||
|  |                         boost::random_access_traversal_tag, | ||
|  |                         Integer, | ||
|  |                         std::ptrdiff_t | ||
|  |                     > | ||
|  |         { | ||
|  |             typedef boost::iterator_facade< | ||
|  |                         integer_iterator<Integer>, | ||
|  |                         Integer, | ||
|  |                         boost::random_access_traversal_tag, | ||
|  |                         Integer, | ||
|  |                         std::ptrdiff_t | ||
|  |                     > base_t; | ||
|  |         public: | ||
|  |             typedef typename base_t::value_type value_type; | ||
|  |             typedef typename base_t::difference_type difference_type; | ||
|  |             typedef typename base_t::reference reference; | ||
|  |             typedef std::random_access_iterator_tag iterator_category; | ||
|  | 
 | ||
|  |             integer_iterator() : m_value() {} | ||
|  |             explicit integer_iterator(value_type x) : m_value(x) {} | ||
|  | 
 | ||
|  |         private: | ||
|  |             void increment() | ||
|  |             { | ||
|  |                 ++m_value; | ||
|  |             } | ||
|  | 
 | ||
|  |             void decrement() | ||
|  |             { | ||
|  |                 --m_value; | ||
|  |             } | ||
|  | 
 | ||
|  |             void advance(difference_type offset) | ||
|  |             { | ||
|  |                 m_value += offset; | ||
|  |             } | ||
|  | 
 | ||
|  |             difference_type distance_to(const integer_iterator& other) const | ||
|  |             { | ||
|  |                 return is_signed<value_type>::value | ||
|  |                     ? (other.m_value - m_value) | ||
|  |                     : (other.m_value >= m_value) | ||
|  |                         ? static_cast<difference_type>(other.m_value - m_value) | ||
|  |                         : -static_cast<difference_type>(m_value - other.m_value); | ||
|  |             } | ||
|  | 
 | ||
|  |             bool equal(const integer_iterator& other) const | ||
|  |             { | ||
|  |                 return m_value == other.m_value; | ||
|  |             } | ||
|  | 
 | ||
|  |             reference dereference() const | ||
|  |             { | ||
|  |                 return m_value; | ||
|  |             } | ||
|  | 
 | ||
|  |             friend class ::boost::iterator_core_access; | ||
|  |             value_type m_value; | ||
|  |         }; | ||
|  | 
 | ||
|  |         // integer_iterator_with_step is similar in nature to the
 | ||
|  |         // integer_iterator but provides the ability to 'move' in
 | ||
|  |         // a number of steps specified at construction time.
 | ||
|  |         //
 | ||
|  |         // The three variable implementation provides the best guarantees
 | ||
|  |         // of loop termination upon various combinations of input.
 | ||
|  |         //
 | ||
|  |         // While this design is less performant than some less
 | ||
|  |         // safe alternatives, the use of ranges and iterators to
 | ||
|  |         // perform counting will never be optimal anyhow, hence
 | ||
|  |         // if optimal performance is desired a hand-coded loop
 | ||
|  |         // is the solution.
 | ||
|  |         template<typename Integer> | ||
|  |         class integer_iterator_with_step | ||
|  |             : public boost::iterator_facade< | ||
|  |                         integer_iterator_with_step<Integer>, | ||
|  |                         Integer, | ||
|  |                         boost::random_access_traversal_tag, | ||
|  |                         Integer, | ||
|  |                         std::ptrdiff_t | ||
|  |                     > | ||
|  |         { | ||
|  |             typedef boost::iterator_facade< | ||
|  |                         integer_iterator_with_step<Integer>, | ||
|  |                         Integer, | ||
|  |                         boost::random_access_traversal_tag, | ||
|  |                         Integer, | ||
|  |                         std::ptrdiff_t | ||
|  |                     > base_t; | ||
|  |         public: | ||
|  |             typedef typename base_t::value_type value_type; | ||
|  |             typedef typename base_t::difference_type difference_type; | ||
|  |             typedef typename base_t::reference reference; | ||
|  |             typedef std::random_access_iterator_tag iterator_category; | ||
|  | 
 | ||
|  |             integer_iterator_with_step(value_type first, difference_type step, value_type step_size) | ||
|  |                 : m_first(first) | ||
|  |                 , m_step(step) | ||
|  |                 , m_step_size(step_size) | ||
|  |             { | ||
|  |             } | ||
|  | 
 | ||
|  |         private: | ||
|  |             void increment() | ||
|  |             { | ||
|  |                 ++m_step; | ||
|  |             } | ||
|  | 
 | ||
|  |             void decrement() | ||
|  |             { | ||
|  |                 --m_step; | ||
|  |             } | ||
|  | 
 | ||
|  |             void advance(difference_type offset) | ||
|  |             { | ||
|  |                 m_step += offset; | ||
|  |             } | ||
|  | 
 | ||
|  |             difference_type distance_to(const integer_iterator_with_step& other) const | ||
|  |             { | ||
|  |                 return other.m_step - m_step; | ||
|  |             } | ||
|  | 
 | ||
|  |             bool equal(const integer_iterator_with_step& other) const | ||
|  |             { | ||
|  |                 return m_step == other.m_step; | ||
|  |             } | ||
|  | 
 | ||
|  |             reference dereference() const | ||
|  |             { | ||
|  |                 return m_first + (m_step * m_step_size); | ||
|  |             } | ||
|  | 
 | ||
|  |             friend class ::boost::iterator_core_access; | ||
|  |             value_type m_first; | ||
|  |             difference_type m_step; | ||
|  |             difference_type m_step_size; | ||
|  |         }; | ||
|  | 
 | ||
|  |     } // namespace range_detail
 | ||
|  | 
 | ||
|  |     template<typename Integer> | ||
|  |     class integer_range | ||
|  |         : public iterator_range< range_detail::integer_iterator<Integer> > | ||
|  |     { | ||
|  |         typedef range_detail::integer_iterator<Integer> iterator_t; | ||
|  |         typedef iterator_range<iterator_t> base_t; | ||
|  |     public: | ||
|  |         integer_range(Integer first, Integer last) | ||
|  |             : base_t(iterator_t(first), iterator_t(last)) | ||
|  |         { | ||
|  |         } | ||
|  |     }; | ||
|  | 
 | ||
|  |     template<typename Integer> | ||
|  |     class strided_integer_range | ||
|  |     : public iterator_range< range_detail::integer_iterator_with_step<Integer> > | ||
|  |     { | ||
|  |         typedef range_detail::integer_iterator_with_step<Integer> iterator_t; | ||
|  |         typedef iterator_range<iterator_t> base_t; | ||
|  |     public: | ||
|  |         template<typename Iterator> | ||
|  |         strided_integer_range(Iterator first, Iterator last) | ||
|  |             : base_t(first, last) | ||
|  |         { | ||
|  |         } | ||
|  |     }; | ||
|  | 
 | ||
|  |     template<typename Integer> | ||
|  |     integer_range<Integer> | ||
|  |     irange(Integer first, Integer last) | ||
|  |     { | ||
|  |         BOOST_ASSERT( first <= last ); | ||
|  |         return integer_range<Integer>(first, last); | ||
|  |     } | ||
|  | 
 | ||
|  |     template<typename Integer, typename StepSize> | ||
|  |     strided_integer_range<Integer> | ||
|  |         irange(Integer first, Integer last, StepSize step_size) | ||
|  |     { | ||
|  |         BOOST_ASSERT( step_size != 0 ); | ||
|  |         BOOST_ASSERT( (step_size > 0) ? (last >= first) : (last <= first) ); | ||
|  | 
 | ||
|  |         typedef typename range_detail::integer_iterator_with_step<Integer> iterator_t; | ||
|  | 
 | ||
|  |         const std::ptrdiff_t sz = static_cast<std::ptrdiff_t>(step_size >= 0 ? step_size : -step_size); | ||
|  |         const Integer l = step_size >= 0 ? last : first; | ||
|  |         const Integer f = step_size >= 0 ? first : last; | ||
|  |         const std::ptrdiff_t num_steps = (l - f) / sz + ((l - f) % sz ? 1 : 0); | ||
|  |         BOOST_ASSERT(num_steps >= 0); | ||
|  | 
 | ||
|  |         return strided_integer_range<Integer>( | ||
|  |             iterator_t(first, 0, step_size), | ||
|  |             iterator_t(first, num_steps, step_size)); | ||
|  |     } | ||
|  | 
 | ||
|  |     template<typename Integer> | ||
|  |     integer_range<Integer> | ||
|  |     irange(Integer last) | ||
|  |     { | ||
|  |         return integer_range<Integer>(static_cast<Integer>(0), last); | ||
|  |     } | ||
|  | 
 | ||
|  | } // namespace boost
 | ||
|  | 
 | ||
|  | #endif // include guard
 |