/* Ranges for OctaSTD. * * This file is part of OctaSTD. See COPYING.md for futher information. */ #ifndef OSTD_RANGE_HH #define OSTD_RANGE_HH #include #include #include #include #include #include #include "ostd/types.hh" #include "ostd/utility.hh" #include "ostd/type_traits.hh" namespace ostd { struct InputRangeTag {}; struct OutputRangeTag {}; struct ForwardRangeTag: InputRangeTag {}; struct BidirectionalRangeTag: ForwardRangeTag {}; struct RandomAccessRangeTag: BidirectionalRangeTag {}; struct FiniteRandomAccessRangeTag: RandomAccessRangeTag {}; struct ContiguousRangeTag: FiniteRandomAccessRangeTag {}; template struct RangeHalf; #define OSTD_RANGE_TRAIT(Name) \ namespace detail { \ template \ struct Range##Name##Test { \ template \ static char test(RemoveReference *); \ template \ static int test(...); \ static constexpr bool value = (sizeof(test(0)) == sizeof(char)); \ }; \ template::value> \ struct Range##Name##Base {}; \ template \ struct Range##Name##Base { \ using Type = typename T::Name; \ }; \ } \ template \ using Range##Name = typename detail::Range##Name##Base::Type; OSTD_RANGE_TRAIT(Category) OSTD_RANGE_TRAIT(Size) OSTD_RANGE_TRAIT(Value) OSTD_RANGE_TRAIT(Reference) OSTD_RANGE_TRAIT(Difference) #undef OSTD_RANGE_TRAIT namespace detail { template static char is_range_test( typename U::Category *, typename U::Size *, typename U::Difference *, typename U::Value *, RemoveReference * ); template static int is_range_test(...); template constexpr bool IsRangeTest = (sizeof(is_range_test(0, 0, 0, 0, 0)) == sizeof(char)); } // is input range namespace detail { template constexpr bool IsInputRangeCore = IsConvertible, InputRangeTag>; template> constexpr bool IsInputRangeBase = false; template constexpr bool IsInputRangeBase = detail::IsInputRangeCore; } template constexpr bool IsInputRange = detail::IsInputRangeBase; // is forward range namespace detail { template constexpr bool IsForwardRangeCore = IsConvertible, ForwardRangeTag>; template> constexpr bool IsForwardRangeBase = false; template constexpr bool IsForwardRangeBase = detail::IsForwardRangeCore; } template constexpr bool IsForwardRange = detail::IsForwardRangeBase; // is bidirectional range namespace detail { template constexpr bool IsBidirectionalRangeCore = IsConvertible, BidirectionalRangeTag>; template> constexpr bool IsBidirectionalRangeBase = false; template constexpr bool IsBidirectionalRangeBase = detail::IsBidirectionalRangeCore; } template constexpr bool IsBidirectionalRange = detail::IsBidirectionalRangeBase; // is random access range namespace detail { template constexpr bool IsRandomAccessRangeCore = IsConvertible, RandomAccessRangeTag>; template> constexpr bool IsRandomAccessRangeBase = false; template constexpr bool IsRandomAccessRangeBase = detail::IsRandomAccessRangeCore; } template constexpr bool IsRandomAccessRange = detail::IsRandomAccessRangeBase; // is finite random access range namespace detail { template constexpr bool IsFiniteRandomAccessRangeCore = IsConvertible, FiniteRandomAccessRangeTag>; template> constexpr bool IsFiniteRandomAccessRangeBase = false; template constexpr bool IsFiniteRandomAccessRangeBase = detail::IsFiniteRandomAccessRangeCore; } template constexpr bool IsFiniteRandomAccessRange = detail::IsFiniteRandomAccessRangeBase; // is infinite random access range template constexpr bool IsInfiniteRandomAccessRange = IsRandomAccessRange && !IsFiniteRandomAccessRange; // is contiguous range namespace detail { template constexpr bool IsContiguousRangeCore = IsConvertible, ContiguousRangeTag>; template> constexpr bool IsContiguousRangeBase = false; template constexpr bool IsContiguousRangeBase = detail::IsContiguousRangeCore; } template constexpr bool IsContiguousRange = detail::IsContiguousRangeBase; // is output range namespace detail { template struct OutputRangeTest { template struct Test {}; template static char test(Test *); template static int test(...); static constexpr bool value = (sizeof(test(0)) == sizeof(char)); }; template constexpr bool IsOutputRangeCore = IsConvertible, OutputRangeTag> || ( IsInputRange && ( detail::OutputRangeTest const &>::value || detail::OutputRangeTest &&>::value || detail::OutputRangeTest >::value ) ); template> constexpr bool IsOutputRangeBase = false; template constexpr bool IsOutputRangeBase = detail::IsOutputRangeCore; } template constexpr bool IsOutputRange = detail::IsOutputRangeBase; namespace detail { // range iterator template struct RangeIterator { RangeIterator(): p_range(), p_init(false) {} explicit RangeIterator(T const &range): p_range(), p_init(true) { ::new(&get_ref()) T(range); } explicit RangeIterator(T &&range): p_range(), p_init(true) { ::new(&get_ref()) T(std::move(range)); } RangeIterator(const RangeIterator &v): p_range(), p_init(true) { ::new(&get_ref()) T(v.get_ref()); } RangeIterator(RangeIterator &&v): p_range(), p_init(true) { ::new(&get_ref()) T(std::move(v.get_ref())); } RangeIterator &operator=(const RangeIterator &v) { destroy(); ::new(&get_ref()) T(v.get_ref()); p_init = true; return *this; } RangeIterator &operator=(RangeIterator &&v) { destroy(); swap(v); return *this; } ~RangeIterator() { destroy(); } RangeIterator &operator++() { get_ref().pop_front(); return *this; } RangeReference operator*() const { return get_ref().front(); } bool operator!=(RangeIterator) const { return !get_ref().empty(); } void swap(RangeIterator &v) { using std::swap; swap(get_ref(). v.get_ref()); swap(p_init, v.p_init); } private: T &get_ref() { return *reinterpret_cast(&p_range); } T const &get_ref() const { return *reinterpret_cast(&p_range); } void destroy() { if (p_init) { get_ref().~T(); p_init = false; } } AlignedStorage p_range; bool p_init; }; } // range half template struct HalfRange; namespace detail { template> struct RangeAdd; template struct RangeAdd { using Diff = RangeDifference; static Diff add_n(R &half, Diff n) { if (n < 0) { return -half.prev_n(n); } return half.next_n(n); } static Diff sub_n(R &half, Diff n) { if (n < 0) { return -half.next_n(n); } return half.prev_n(n); } }; template struct RangeAdd { using Diff = RangeDifference; static Diff add_n(R &half, Diff n) { if (n < 0) { return 0; } return half.next_n(n); } static Diff sub_n(R &half, Diff n) { if (n < 0) { return 0; } return half.prev_n(n); } }; } namespace detail { template struct RangeIteratorTag { /* better range types all become random access iterators */ using Type = std::random_access_iterator_tag; }; template<> struct RangeIteratorTag { using Type = std::input_iterator_tag; }; template<> struct RangeIteratorTag { using Type = std::output_iterator_tag; }; template<> struct RangeIteratorTag { using Type = std::forward_iterator_tag; }; template<> struct RangeIteratorTag { using Type = std::bidirectional_iterator_tag; }; } template struct RangeHalf { private: T p_range; public: using Range = T; using iterator_category = typename detail::RangeIteratorTag::Type; using value_type = RangeValue; using difference_type = RangeDifference; using pointer = RangeValue *; using reference = RangeReference; RangeHalf() = delete; RangeHalf(T const &range): p_range(range) {} template>> RangeHalf(RangeHalf const &half): p_range(half.p_range) {} RangeHalf(RangeHalf const &half): p_range(half.p_range) {} RangeHalf(RangeHalf &&half): p_range(std::move(half.p_range)) {} RangeHalf &operator=(RangeHalf const &half) { p_range = half.p_range; return *this; } RangeHalf &operator=(RangeHalf &&half) { p_range = std::move(half.p_range); return *this; } bool next() { return p_range.pop_front(); } bool prev() { return p_range.push_front(); } RangeSize next_n(RangeSize n) { return p_range.pop_front_n(n); } RangeSize prev_n(RangeSize n) { return p_range.push_front_n(n); } RangeDifference add_n(RangeDifference n) { return detail::RangeAdd>::add_n(*this, n); } RangeDifference sub_n(RangeDifference n) { return detail::RangeAdd>::sub_n(*this, n); } RangeReference get() const { return p_range.front(); } RangeDifference distance(RangeHalf const &half) const { return p_range.distance_front(half.p_range); } bool equals(RangeHalf const &half) const { return p_range.equals_front(half.p_range); } bool operator==(RangeHalf const &half) const { return equals(half); } bool operator!=(RangeHalf const &half) const { return !equals(half); } /* iterator like interface */ RangeReference operator*() const { return p_range.front(); } RangeReference operator[](RangeSize idx) const { return p_range[idx]; } RangeHalf &operator++() { next(); return *this; } RangeHalf operator++(int) { RangeHalf tmp(*this); next(); return tmp; } RangeHalf &operator--() { prev(); return *this; } RangeHalf operator--(int) { RangeHalf tmp(*this); prev(); return tmp; } RangeHalf operator+(RangeDifference n) const { RangeHalf tmp(*this); tmp.add_n(n); return tmp; } RangeHalf operator-(RangeDifference n) const { RangeHalf tmp(*this); tmp.sub_n(n); return tmp; } RangeHalf &operator+=(RangeDifference n) { add_n(n); return *this; } RangeHalf &operator-=(RangeDifference n) { sub_n(n); return *this; } T iter() const { return p_range; } HalfRange iter(RangeHalf const &other) const { return HalfRange(*this, other); } RangeValue *data() { return p_range.data(); } RangeValue const *data() const { return p_range.data(); } }; template inline RangeDifference operator-( RangeHalf const &lhs, RangeHalf const &rhs ) { return rhs.distance(lhs); } namespace detail { template RangeSize pop_front_n(R &range, RangeSize n) { for (RangeSize i = 0; i < n; ++i) { if (!range.pop_front()) { return i; } } return n; } template RangeSize pop_back_n(R &range, RangeSize n) { for (RangeSize i = 0; i < n; ++i) { if (!range.pop_back()) { return i; } } return n; } template RangeSize push_front_n(R &range, RangeSize n) { for (RangeSize i = 0; i < n; ++i) { if (!range.push_front()) { return i; } } return n; } template RangeSize push_back_n(R &range, RangeSize n) { for (RangeSize i = 0; i < n; ++i) { if (!range.push_back()) { return i; } } return n; } } template struct ReverseRange; template struct MoveRange; template struct EnumeratedRange; template struct TakeRange; template struct ChunksRange; template struct JoinRange; template struct ZipRange; template< typename B, typename C, typename V, typename R = V &, typename S = size_t, typename D = ptrdiff_t > struct InputRange { using Category = C; using Size = S; using Difference = D; using Value = V; using Reference = R; detail::RangeIterator begin() const { return detail::RangeIterator(*static_cast(this)); } detail::RangeIterator end() const { return detail::RangeIterator(); } Size pop_front_n(Size n) { return detail::pop_front_n(*static_cast(this), n); } Size pop_back_n(Size n) { return detail::pop_back_n(*static_cast(this), n); } Size push_front_n(Size n) { return detail::push_front_n(*static_cast(this), n); } Size push_back_n(Size n) { return detail::push_back_n(*static_cast(this), n); } B iter() const { return B(*static_cast(this)); } ReverseRange reverse() const { return ReverseRange(iter()); } MoveRange movable() const { return MoveRange(iter()); } EnumeratedRange enumerate() const { return EnumeratedRange(iter()); } TakeRange take(Size n) const { return TakeRange(iter(), n); } ChunksRange chunks(Size n) const { return ChunksRange(iter(), n); } template JoinRange join(R1 r1, RR ...rr) const { return JoinRange(iter(), std::move(r1), std::move(rr)...); } template ZipRange zip(R1 r1, RR ...rr) const { return ZipRange(iter(), std::move(r1), std::move(rr)...); } RangeHalf half() const { return RangeHalf(iter()); } Size put_n(Value const *p, Size n) { B &r = *static_cast(this); Size on = n; for (; n && r.put(*p++); --n); return (on - n); } template EnableIf, Size> copy(OR &&orange, Size n = -1) { B r(*static_cast(this)); Size on = n; for (; n && !r.empty(); --n) { if (!orange.put(r.front())) { break; } r.pop_front(); } return (on - n); } Size copy(RemoveCv *p, Size n = -1) { B r(*static_cast(this)); Size on = n; for (; n && !r.empty(); --n) { *p++ = r.front(); r.pop_front(); } return (on - n); } /* iterator like interface operating on the front part of the range * this is sometimes convenient as it can be used within expressions */ Reference operator*() const { return std::forward(static_cast(this)->front()); } B &operator++() { static_cast(this)->pop_front(); return *static_cast(this); } B operator++(int) { B tmp(*static_cast(this)); static_cast(this)->pop_front(); return tmp; } B &operator--() { static_cast(this)->push_front(); return *static_cast(this); } B operator--(int) { B tmp(*static_cast(this)); static_cast(this)->push_front(); return tmp; } B operator+(Difference n) const { B tmp(*static_cast(this)); tmp.pop_front_n(n); return tmp; } B operator-(Difference n) const { B tmp(*static_cast(this)); tmp.push_front_n(n); return tmp; } B &operator+=(Difference n) { static_cast(this)->pop_front_n(n); return *static_cast(this); } B &operator-=(Difference n) { static_cast(this)->push_front_n(n); return *static_cast(this); } /* universal bool operator */ explicit operator bool() const { return !(static_cast(this)->empty()); } }; template>> inline auto operator|(R &&range, F &&func) { return func(std::forward(range)); } inline auto reverse() { return [](auto &&obj) { return obj.reverse(); }; } inline auto movable() { return [](auto &&obj) { return obj.movable(); }; } inline auto enumerate() { return [](auto &&obj) { return obj.enumerate(); }; } template inline auto take(T n) { return [n](auto &&obj) { return obj.take(n); }; } template inline auto chunks(T n) { return [n](auto &&obj) { return obj.chunks(n); }; } namespace detail { template inline auto join_proxy( T &&obj, std::tuple &&tup, std::index_sequence ) { return obj.join(std::forward( std::get(std::forward>(tup)) )...); } template inline auto zip_proxy( T &&obj, std::tuple &&tup, std::index_sequence ) { return obj.zip(std::forward( std::get(std::forward>(tup)) )...); } } template inline auto join(R &&range) { return [range = std::forward(range)](auto &&obj) mutable { return obj.join(std::forward(range)); }; } template inline auto join(R1 &&r1, R &&...rr) { return [ ranges = std::forward_as_tuple( std::forward(r1), std::forward(rr)... ) ] (auto &&obj) mutable { return detail::join_proxy( std::forward(obj), std::forward(ranges), std::make_index_sequence() ); }; } template inline auto zip(R &&range) { return [range = std::forward(range)](auto &&obj) mutable { return obj.zip(std::forward(range)); }; } template inline auto zip(R1 &&r1, R &&...rr) { return [ ranges = std::forward_as_tuple( std::forward(r1), std::forward(rr)... ) ] (auto &&obj) mutable { return detail::zip_proxy( std::forward(obj), std::forward(ranges), std::make_index_sequence() ); }; } template struct ranged_traits; namespace detail { template static True test_direct_iter(decltype(std::declval().iter()) *); template static False test_direct_iter(...); template constexpr bool direct_iter_test = decltype(test_direct_iter(0))::value; } template struct ranged_traits>> { static auto iter(C &r) -> decltype(r.iter()) { return r.iter(); } }; template inline auto iter(T &r) -> decltype(ranged_traits::iter(r)) { return ranged_traits::iter(r); } template inline auto iter(T const &r) -> decltype(ranged_traits::iter(r)) { return ranged_traits::iter(r); } template inline auto citer(T const &r) -> decltype(ranged_traits::iter(r)) { return ranged_traits::iter(r); } template< typename B, typename V, typename R = V &, typename S = size_t, typename D = ptrdiff_t > struct OutputRange { using Category = OutputRangeTag; using Size = S; using Difference = D; using Value = V; using Reference = R; Size put_n(Value const *p, Size n) { B &r = *static_cast(this); Size on = n; for (; n && r.put(*p++); --n); return (on - n); } }; template struct HalfRange: InputRange, RangeCategory, RangeValue, RangeReference, RangeSize, RangeDifference > { private: using Rtype = typename T::Range; T p_beg; T p_end; public: HalfRange() = delete; HalfRange(HalfRange const &range): p_beg(range.p_beg), p_end(range.p_end) {} HalfRange(HalfRange &&range): p_beg(std::move(range.p_beg)), p_end(std::move(range.p_end)) {} HalfRange(T const &beg, T const &end): p_beg(beg),p_end(end) {} HalfRange(T &&beg, T &&end): p_beg(std::move(beg)), p_end(std::move(end)) {} HalfRange &operator=(HalfRange const &range) { p_beg = range.p_beg; p_end = range.p_end; return *this; } HalfRange &operator=(HalfRange &&range) { p_beg = std::move(range.p_beg); p_end = std::move(range.p_end); return *this; } bool empty() const { return p_beg == p_end; } bool pop_front() { if (empty()) { return false; } return p_beg.next(); } bool push_front() { return p_beg.prev(); } bool pop_back() { if (empty()) { return false; } return p_end.prev(); } bool push_back() { return p_end.next(); } RangeReference front() const { return *p_beg; } RangeReference back() const { return *(p_end - 1); } bool equals_front(HalfRange const &range) const { return p_beg == range.p_beg; } bool equals_back(HalfRange const &range) const { return p_end == range.p_end; } RangeDifference distance_front(HalfRange const &range) const { return range.p_beg - p_beg; } RangeDifference distance_back(HalfRange const &range) const { return range.p_end - p_end; } RangeSize size() const { return p_end - p_beg; } HalfRange slice(RangeSize start, RangeSize end) const { return HalfRange(p_beg + start, p_beg + end); } RangeReference operator[](RangeSize idx) const { return p_beg[idx]; } bool put(RangeValue const &v) { return p_beg.range().put(v); } bool put(RangeValue &&v) { return p_beg.range().put(std::move(v)); } RangeValue *data() { return p_beg.data(); } RangeValue const *data() const { return p_beg.data(); } }; template struct ReverseRange: InputRange, CommonType, FiniteRandomAccessRangeTag>, RangeValue, RangeReference, RangeSize, RangeDifference > { private: using Rref = RangeReference; using Rsize = RangeSize; T p_range; public: ReverseRange() = delete; ReverseRange(T const &range): p_range(range) {} ReverseRange(ReverseRange const &it): p_range(it.p_range) {} ReverseRange(ReverseRange &&it): p_range(std::move(it.p_range)) {} ReverseRange &operator=(ReverseRange const &v) { p_range = v.p_range; return *this; } ReverseRange &operator=(ReverseRange &&v) { p_range = std::move(v.p_range); return *this; } ReverseRange &operator=(T const &v) { p_range = v; return *this; } ReverseRange &operator=(T &&v) { p_range = std::move(v); return *this; } bool empty() const { return p_range.empty(); } Rsize size() const { return p_range.size(); } bool equals_front(ReverseRange const &r) const { return p_range.equals_back(r.p_range); } bool equals_back(ReverseRange const &r) const { return p_range.equals_front(r.p_range); } RangeDifference distance_front(ReverseRange const &r) const { return -p_range.distance_back(r.p_range); } RangeDifference distance_back(ReverseRange const &r) const { return -p_range.distance_front(r.p_range); } bool pop_front() { return p_range.pop_back(); } bool pop_back() { return p_range.pop_front(); } bool push_front() { return p_range.push_back(); } bool push_back() { return p_range.push_front(); } Rsize pop_front_n(Rsize n) { return p_range.pop_front_n(n); } Rsize pop_back_n(Rsize n) { return p_range.pop_back_n(n); } Rsize push_front_n(Rsize n) { return p_range.push_front_n(n); } Rsize push_back_n(Rsize n) { return p_range.push_back_n(n); } Rref front() const { return p_range.back(); } Rref back() const { return p_range.front(); } Rref operator[](Rsize i) const { return p_range[size() - i - 1]; } ReverseRange slice(Rsize start, Rsize end) const { Rsize len = p_range.size(); return ReverseRange(p_range.slice(len - end, len - start)); } }; template struct MoveRange: InputRange, CommonType, FiniteRandomAccessRangeTag>, RangeValue, RangeValue &&, RangeSize, RangeDifference > { private: using Rval = RangeValue; using Rref = RangeValue &&; using Rsize = RangeSize; T p_range; public: MoveRange() = delete; MoveRange(T const &range): p_range(range) {} MoveRange(MoveRange const &it): p_range(it.p_range) {} MoveRange(MoveRange &&it): p_range(std::move(it.p_range)) {} MoveRange &operator=(MoveRange const &v) { p_range = v.p_range; return *this; } MoveRange &operator=(MoveRange &&v) { p_range = std::move(v.p_range); return *this; } MoveRange &operator=(T const &v) { p_range = v; return *this; } MoveRange &operator=(T &&v) { p_range = std::move(v); return *this; } bool empty() const { return p_range.empty(); } Rsize size() const { return p_range.size(); } bool equals_front(MoveRange const &r) const { return p_range.equals_front(r.p_range); } bool equals_back(MoveRange const &r) const { return p_range.equals_back(r.p_range); } RangeDifference distance_front(MoveRange const &r) const { return p_range.distance_front(r.p_range); } RangeDifference distance_back(MoveRange const &r) const { return p_range.distance_back(r.p_range); } bool pop_front() { return p_range.pop_front(); } bool pop_back() { return p_range.pop_back(); } bool push_front() { return p_range.push_front(); } bool push_back() { return p_range.push_back(); } Rsize pop_front_n(Rsize n) { return p_range.pop_front_n(n); } Rsize pop_back_n(Rsize n) { return p_range.pop_back_n(n); } Rsize push_front_n(Rsize n) { return p_range.push_front_n(n); } Rsize push_back_n(Rsize n) { return p_range.push_back_n(n); } Rref front() const { return std::move(p_range.front()); } Rref back() const { return std::move(p_range.back()); } Rref operator[](Rsize i) const { return std::move(p_range[i]); } MoveRange slice(Rsize start, Rsize end) const { return MoveRange(p_range.slice(start, end)); } bool put(Rval const &v) { return p_range.put(v); } bool put(Rval &&v) { return p_range.put(std::move(v)); } }; template struct NumberRange: InputRange, ForwardRangeTag, T, T> { NumberRange() = delete; NumberRange(T a, T b, T step = T(1)): p_a(a), p_b(b), p_step(step) {} NumberRange(T v): p_a(0), p_b(v), p_step(1) {} bool empty() const { return p_a * p_step >= p_b * p_step; } bool equals_front(NumberRange const &range) const { return p_a == range.p_a; } bool pop_front() { p_a += p_step; return true; } T front() const { return p_a; } private: T p_a, p_b, p_step; }; template inline NumberRange range(T a, T b, T step = T(1)) { return NumberRange(a, b, step); } template inline NumberRange range(T v) { return NumberRange(v); } template struct PointerRange: InputRange, ContiguousRangeTag, T> { private: struct Nat {}; public: PointerRange(): p_beg(nullptr), p_end(nullptr) {} template PointerRange(T *beg, U end, EnableIf< (IsPointer || IsNullPointer) && IsConvertible, Nat > = Nat()): p_beg(beg), p_end(end) {} PointerRange(T *beg, size_t n): p_beg(beg), p_end(beg + n) {} template>> PointerRange(PointerRange const &v): p_beg(&v[0]), p_end(&v[v.size()]) {} PointerRange &operator=(PointerRange const &v) { p_beg = v.p_beg; p_end = v.p_end; return *this; } /* satisfy InputRange / ForwardRange */ bool empty() const { return p_beg == p_end; } bool pop_front() { if (p_beg == p_end) { return false; } ++p_beg; return true; } bool push_front() { --p_beg; return true; } size_t pop_front_n(size_t n) { size_t olen = p_end - p_beg; p_beg += n; if (p_beg > p_end) { p_beg = p_end; return olen; } return n; } size_t push_front_n(size_t n) { p_beg -= n; return true; } T &front() const { return *p_beg; } bool equals_front(PointerRange const &range) const { return p_beg == range.p_beg; } ptrdiff_t distance_front(PointerRange const &range) const { return range.p_beg - p_beg; } /* satisfy BidirectionalRange */ bool pop_back() { if (p_end == p_beg) { return false; } --p_end; return true; } bool push_back() { ++p_end; return true; } size_t pop_back_n(size_t n) { size_t olen = p_end - p_beg; p_end -= n; if (p_end < p_beg) { p_end = p_beg; return olen; } return n; } size_t push_back_n(size_t n) { p_end += n; return true; } T &back() const { return *(p_end - 1); } bool equals_back(PointerRange const &range) const { return p_end == range.p_end; } ptrdiff_t distance_back(PointerRange const &range) const { return range.p_end - p_end; } /* satisfy FiniteRandomAccessRange */ size_t size() const { return p_end - p_beg; } PointerRange slice(size_t start, size_t end) const { return PointerRange(p_beg + start, p_beg + end); } T &operator[](size_t i) const { return p_beg[i]; } /* satisfy OutputRange */ bool put(T const &v) { if (empty()) { return false; } *(p_beg++) = v; return true; } bool put(T &&v) { if (empty()) { return false; } *(p_beg++) = std::move(v); return true; } size_t put_n(T const *p, size_t n) { size_t ret = size(); if (n < ret) { ret = n; } if (IsPod) { memcpy(p_beg, p, ret * sizeof(T)); p_beg += ret; return ret; } for (size_t i = ret; i; --i) { *p_beg++ = *p++; } return ret; } template EnableIf, size_t> copy(R &&orange, size_t n = -1) { size_t c = size(); if (n < c) { c = n; } return orange.put_n(p_beg, c); } size_t copy(RemoveCv *p, size_t n = -1) { size_t c = size(); if (n < c) { c = n; } if (IsPod) { memcpy(p_beg, data(), c * sizeof(T)); return c; } return copy(PointerRange(p, c), c); } T *data() { return p_beg; } T const *data() const { return p_beg; } private: T *p_beg, *p_end; }; template inline PointerRange iter(T (&array)[N]) { return PointerRange(array, N); } template inline PointerRange iter(T const (&array)[N]) { return PointerRange(array, N); } template inline PointerRange citer(T const (&array)[N]) { return PointerRange(array, N); } namespace detail { struct PtrNat {}; } template inline PointerRange iter(T *a, U b, EnableIf< (IsPointer || IsNullPointer) && IsConvertible, detail::PtrNat > = detail::PtrNat()) { return PointerRange(a, b); } template inline PointerRange iter(T *a, size_t b) { return PointerRange(a, b); } template struct EnumeratedValue { S index; T value; }; template struct EnumeratedRange: InputRange, CommonType, ForwardRangeTag>, RangeValue, EnumeratedValue, RangeSize>, RangeSize > { private: using Rref = RangeReference; using Rsize = RangeSize; T p_range; Rsize p_index; public: EnumeratedRange() = delete; EnumeratedRange(T const &range): p_range(range), p_index(0) {} EnumeratedRange(EnumeratedRange const &it): p_range(it.p_range), p_index(it.p_index) {} EnumeratedRange(EnumeratedRange &&it): p_range(std::move(it.p_range)), p_index(it.p_index) {} EnumeratedRange &operator=(EnumeratedRange const &v) { p_range = v.p_range; p_index = v.p_index; return *this; } EnumeratedRange &operator=(EnumeratedRange &&v) { p_range = std::move(v.p_range); p_index = v.p_index; return *this; } EnumeratedRange &operator=(T const &v) { p_range = v; p_index = 0; return *this; } EnumeratedRange &operator=(T &&v) { p_range = std::move(v); p_index = 0; return *this; } bool empty() const { return p_range.empty(); } bool equals_front(EnumeratedRange const &r) const { return p_range.equals_front(r.p_range); } bool pop_front() { if (p_range.pop_front()) { ++p_index; return true; } return false; } Rsize pop_front_n(Rsize n) { Rsize ret = p_range.pop_front_n(n); p_index += ret; return ret; } EnumeratedValue front() const { return EnumeratedValue { p_index, p_range.front() }; } }; template struct TakeRange: InputRange, CommonType, ForwardRangeTag>, RangeValue, RangeReference, RangeSize > { private: T p_range; RangeSize p_remaining; public: TakeRange() = delete; TakeRange(T const &range, RangeSize rem): p_range(range), p_remaining(rem) {} TakeRange(TakeRange const &it): p_range(it.p_range), p_remaining(it.p_remaining) {} TakeRange(TakeRange &&it): p_range(std::move(it.p_range)), p_remaining(it.p_remaining) {} TakeRange &operator=(TakeRange const &v) { p_range = v.p_range; p_remaining = v.p_remaining; return *this; } TakeRange &operator=(TakeRange &&v) { p_range = std::move(v.p_range); p_remaining = v.p_remaining; return *this; } bool empty() const { return (p_remaining <= 0) || p_range.empty(); } bool pop_front() { if (p_range.pop_front()) { --p_remaining; return true; } return false; } RangeSize pop_front_n(RangeSize n) { RangeSize ret = p_range.pop_front_n(n); p_remaining -= ret; return ret; } RangeReference front() const { return p_range.front(); } bool equals_front(TakeRange const &r) const { return p_range.equals_front(r.p_range); } }; template struct ChunksRange: InputRange, CommonType, ForwardRangeTag>, TakeRange, TakeRange, RangeSize > { private: T p_range; RangeSize p_chunksize; public: ChunksRange() = delete; ChunksRange(T const &range, RangeSize chs): p_range(range), p_chunksize(chs) {} ChunksRange(ChunksRange const &it): p_range(it.p_range), p_chunksize(it.p_chunksize) {} ChunksRange(ChunksRange &&it): p_range(std::move(it.p_range)), p_chunksize(it.p_chunksize) {} ChunksRange &operator=(ChunksRange const &v) { p_range = v.p_range; p_chunksize = v.p_chunksize; return *this; } ChunksRange &operator=(ChunksRange &&v) { p_range = std::move(v.p_range); p_chunksize = v.p_chunksize; return *this; } bool empty() const { return p_range.empty(); } bool equals_front(ChunksRange const &r) const { return p_range.equals_front(r.p_range); } bool pop_front() { return p_range.pop_front_n(p_chunksize) > 0; } RangeSize pop_front_n(RangeSize n) { return p_range.pop_front_n(p_chunksize * n) / p_chunksize; } TakeRange front() const { return p_range.take(p_chunksize); } }; namespace detail { template struct JoinRangeEmpty { template static bool empty(T const &tup) { if (!std::get(tup).empty()) { return false; } return JoinRangeEmpty::empty(tup); } }; template struct JoinRangeEmpty { template static bool empty(T const &) { return true; } }; template struct TupleRangeEqual { template static bool equal(T const &tup1, T const &tup2) { if (!std::get(tup1).equals_front(std::get(tup2))) { return false; } return TupleRangeEqual::equal(tup1, tup2); } }; template struct TupleRangeEqual { template static bool equal(T const &, T const &) { return true; } }; template struct JoinRangePop { template static bool pop(T &tup) { if (!std::get(tup).empty()) { return std::get(tup).pop_front(); } return JoinRangePop::pop(tup); } }; template struct JoinRangePop { template static bool pop(T &) { return false; } }; template struct JoinRangeFront { template static T front(U const &tup) { if (!std::get(tup).empty()) { return std::get(tup).front(); } return JoinRangeFront::front(tup); } }; template struct JoinRangeFront { template static T front(U const &tup) { return std::get<0>(tup).front(); } }; } template struct JoinRange: InputRange, CommonType...>, CommonType...>, CommonType...>, CommonType...>, CommonType...>> { private: std::tuple p_ranges; public: JoinRange() = delete; JoinRange(R const &...ranges): p_ranges(ranges...) {} JoinRange(R &&...ranges): p_ranges(std::forward(ranges)...) {} JoinRange(JoinRange const &v): p_ranges(v.p_ranges) {} JoinRange(JoinRange &&v): p_ranges(std::move(v.p_ranges)) {} JoinRange &operator=(JoinRange const &v) { p_ranges = v.p_ranges; return *this; } JoinRange &operator=(JoinRange &&v) { p_ranges = std::move(v.p_ranges); return *this; } bool empty() const { return detail::JoinRangeEmpty<0, sizeof...(R)>::empty(p_ranges); } bool equals_front(JoinRange const &r) const { return detail::TupleRangeEqual<0, sizeof...(R)>::equal( p_ranges, r.p_ranges ); } bool pop_front() { return detail::JoinRangePop<0, sizeof...(R)>::pop(p_ranges); } CommonType...> front() const { return detail::JoinRangeFront< 0, sizeof...(R), CommonType...> >::front(p_ranges); } }; namespace detail { template struct ZipValueType { using Type = std::tuple; }; template struct ZipValueType { using Type = std::pair; }; template using ZipValue = typename detail::ZipValueType::Type; template struct ZipRangeEmpty { template static bool empty(T const &tup) { if (std::get(tup).empty()) { return true; } return ZipRangeEmpty::empty(tup); } }; template struct ZipRangeEmpty { template static bool empty(T const &) { return false; } }; template struct ZipRangePop { template static bool pop(T &tup) { return ( std::get(tup).pop_front() && ZipRangePop::pop(tup) ); } }; template struct ZipRangePop { template static bool pop(T &) { return true; } }; template struct ZipRangeFront { template static ZipValue tup_get(U const &tup, std::index_sequence) { return ZipValue(std::get(tup).front()...); } template static ZipValue front(U const &tup) { return ZipRangeFront::tup_get( tup, std::make_index_sequence() ); } }; } template struct ZipRange: InputRange, CommonType...>, detail::ZipValue...>, detail::ZipValue...>, CommonType...>, CommonType...>> { private: std::tuple p_ranges; public: ZipRange() = delete; ZipRange(R const &...ranges): p_ranges(ranges...) {} ZipRange(R &&...ranges): p_ranges(std::forward(ranges)...) {} ZipRange(ZipRange const &v): p_ranges(v.p_ranges) {} ZipRange(ZipRange &&v): p_ranges(std::move(v.p_ranges)) {} ZipRange &operator=(ZipRange const &v) { p_ranges = v.p_ranges; return *this; } ZipRange &operator=(ZipRange &&v) { p_ranges = std::move(v.p_ranges); return *this; } bool empty() const { return detail::ZipRangeEmpty<0, sizeof...(R)>::empty(p_ranges); } bool equals_front(ZipRange const &r) const { return detail::TupleRangeEqual<0, sizeof...(R)>::equal( p_ranges, r.p_ranges ); } bool pop_front() { return detail::ZipRangePop<0, sizeof...(R)>::pop(p_ranges); } detail::ZipValue...> front() const { return detail::ZipRangeFront...>::front(p_ranges); } }; template struct AppenderRange: OutputRange, typename T::value_type, typename T::reference, typename T::size_type, typename T::difference_type> { AppenderRange(): p_data() {} AppenderRange(T const &v): p_data(v) {} AppenderRange(T &&v): p_data(std::move(v)) {} AppenderRange(AppenderRange const &v): p_data(v.p_data) {} AppenderRange(AppenderRange &&v): p_data(std::move(v.p_data)) {} AppenderRange &operator=(AppenderRange const &v) { p_data = v.p_data; return *this; } AppenderRange &operator=(AppenderRange &&v) { p_data = std::move(v.p_data); return *this; } AppenderRange &operator=(T const &v) { p_data = v; return *this; } AppenderRange &operator=(T &&v) { p_data = std::move(v); return *this; } void clear() { p_data.clear(); } void reserve(typename T::size_type cap) { p_data.reserve(cap); } void resize(typename T::size_type len) { p_data.resize(len); } typename T::size_type size() const { return p_data.size(); } typename T::size_type capacity() const { return p_data.capacity(); } bool put(typename T::const_reference v) { p_data.push_back(v); return true; } bool put(typename T::value_type &&v) { p_data.push_back(std::move(v)); return true; } T &get() { return p_data; } private: T p_data; }; template inline AppenderRange appender() { return AppenderRange(); } template inline AppenderRange appender(T &&v) { return AppenderRange(std::forward(v)); } namespace detail { template struct IteratorRangeTagBase { /* fallback, the most basic range */ using Type = InputRangeTag; }; template<> struct IteratorRangeTagBase { using Type = OutputRangeTag; }; template<> struct IteratorRangeTagBase { using Type = ForwardRangeTag; }; template<> struct IteratorRangeTagBase { using Type = BidirectionalRangeTag; }; template<> struct IteratorRangeTagBase { using Type = FiniteRandomAccessRangeTag; }; } template using IteratorRangeTag = typename detail::IteratorRangeTagBase::Type; template struct IteratorRange: InputRange< IteratorRange, IteratorRangeTag::iterator_category>, typename std::iterator_traits::value_type, typename std::iterator_traits::reference, size_t, typename std::iterator_traits::difference_type > { private: struct Nat {}; using RefT = typename std::iterator_traits::reference; using DiffT = typename std::iterator_traits::difference_type; public: IteratorRange(T beg = T{}, T end = T{}): p_beg(beg), p_end(end) {} IteratorRange(T beg, size_t n): p_beg(beg), p_end(beg + n) {} IteratorRange(IteratorRange const &v): p_beg(v.p_beg), p_end(v.p_end) {} IteratorRange(IteratorRange &&v): p_beg(std::move(v.p_beg)), p_end(std::move(v.p_end)) {} IteratorRange &operator=(IteratorRange const &v) { p_beg = v.p_beg; p_end = v.p_end; return *this; } IteratorRange &operator=(IteratorRange &&v) { p_beg = std::move(v.p_beg); p_end = std::move(v.p_end); return *this; } /* satisfy InputRange / ForwardRange */ bool empty() const { return p_beg == p_end; } bool pop_front() { if (p_beg == p_end) { return false; } ++p_beg; return true; } bool push_front() { --p_beg; return true; } RefT front() const { return *p_beg; } bool equals_front(IteratorRange const &range) const { return p_beg == range.p_beg; } DiffT distance_front(IteratorRange const &range) const { return range.p_beg - p_beg; } /* satisfy BidirectionalRange */ bool pop_back() { if (p_end == p_beg) { return false; } --p_end; return true; } bool push_back() { ++p_end; return true; } RefT back() const { return *(p_end - 1); } bool equals_back(IteratorRange const &range) const { return p_end == range.p_end; } ptrdiff_t distance_back(IteratorRange const &range) const { return range.p_end - p_end; } /* satisfy FiniteRandomAccessRange */ size_t size() const { return p_end - p_beg; } IteratorRange slice(size_t start, size_t end) const { return IteratorRange(p_beg + start, p_beg + end); } RefT operator[](size_t i) const { return p_beg[i]; } /* satisfy OutputRange */ bool put(T const &v) { if (empty()) { return false; } *(p_beg++) = v; return true; } bool put(T &&v) { if (empty()) { return false; } *(p_beg++) = std::move(v); return true; } private: T p_beg, p_end; }; template IteratorRange make_range(T beg, T end) { return IteratorRange{beg, end}; } template IteratorRange make_range(T beg, size_t n) { return IteratorRange{beg, beg + n}; } } /* namespace ostd */ #endif