2015-04-15 23:18:00 +00:00
|
|
|
/* Ranges for OctaSTD.
|
|
|
|
*
|
|
|
|
* This file is part of OctaSTD. See COPYING.md for futher information.
|
|
|
|
*/
|
|
|
|
|
|
|
|
#ifndef OCTA_RANGE_H
|
|
|
|
#define OCTA_RANGE_H
|
|
|
|
|
|
|
|
#include <stddef.h>
|
|
|
|
|
|
|
|
#include "octa/types.h"
|
2015-04-17 23:11:16 +00:00
|
|
|
#include "octa/utility.h"
|
2015-04-15 23:18:00 +00:00
|
|
|
|
|
|
|
namespace octa {
|
2015-04-16 20:50:53 +00:00
|
|
|
struct InputRange {};
|
2015-04-17 20:33:01 +00:00
|
|
|
struct OutputRange {};
|
2015-04-16 23:15:20 +00:00
|
|
|
struct ForwardRange {};
|
2015-04-16 20:50:53 +00:00
|
|
|
struct BidirectionalRange {};
|
|
|
|
struct RandomAccessRange {};
|
|
|
|
|
2015-04-15 23:18:00 +00:00
|
|
|
template<typename T>
|
|
|
|
struct RangeTraits {
|
2015-04-16 20:50:53 +00:00
|
|
|
typedef typename T::type::category category;
|
|
|
|
typedef typename T::type::value value;
|
|
|
|
typedef typename T::type::reference reference;
|
2015-04-15 23:18:00 +00:00
|
|
|
};
|
|
|
|
|
2015-04-17 17:33:41 +00:00
|
|
|
namespace internal {
|
|
|
|
template<typename T>
|
|
|
|
struct RangeIterator {
|
|
|
|
RangeIterator(): p_range() {}
|
|
|
|
explicit RangeIterator(const T &range): p_range(range) {}
|
|
|
|
RangeIterator &operator++() {
|
|
|
|
p_range.pop_first();
|
|
|
|
return *this;
|
|
|
|
}
|
|
|
|
typename RangeTraits<T>::reference operator*() {
|
|
|
|
return p_range.first();
|
|
|
|
}
|
|
|
|
typename RangeTraits<T>::reference operator*() const {
|
|
|
|
return p_range.first();
|
|
|
|
}
|
|
|
|
bool operator!=(RangeIterator) const { return !p_range.empty(); }
|
|
|
|
private:
|
|
|
|
T p_range;
|
|
|
|
};
|
|
|
|
}
|
2015-04-15 23:18:00 +00:00
|
|
|
|
2015-04-17 19:29:38 +00:00
|
|
|
template<typename B, typename C, typename V, typename R = V &>
|
2015-04-17 20:33:01 +00:00
|
|
|
struct InputRangeBase {
|
2015-04-16 01:07:32 +00:00
|
|
|
struct type {
|
2015-04-17 19:14:43 +00:00
|
|
|
typedef C category;
|
|
|
|
typedef V value;
|
|
|
|
typedef R reference;
|
2015-04-16 01:07:32 +00:00
|
|
|
};
|
|
|
|
|
2015-04-17 19:14:43 +00:00
|
|
|
internal::RangeIterator<B> begin() {
|
|
|
|
return internal::RangeIterator<B>((const B &)*this);
|
|
|
|
}
|
|
|
|
internal::RangeIterator<B> end() {
|
|
|
|
return internal::RangeIterator<B>();
|
|
|
|
}
|
|
|
|
};
|
|
|
|
|
2015-04-17 20:33:01 +00:00
|
|
|
template<typename V, typename R = V &>
|
|
|
|
struct OutputRangeBase {
|
|
|
|
struct type {
|
|
|
|
typedef OutputRange category;
|
|
|
|
typedef V value;
|
|
|
|
typedef R reference;
|
|
|
|
};
|
|
|
|
};
|
|
|
|
|
2015-04-17 19:14:43 +00:00
|
|
|
template<typename T>
|
2015-04-17 20:33:01 +00:00
|
|
|
struct ReverseRange: InputRangeBase<ReverseRange<T>,
|
2015-04-17 19:14:43 +00:00
|
|
|
typename RangeTraits<T>::category,
|
|
|
|
typename RangeTraits<T>::value,
|
|
|
|
typename RangeTraits<T>::reference
|
|
|
|
> {
|
2015-04-16 01:07:32 +00:00
|
|
|
ReverseRange(): p_range() {}
|
|
|
|
ReverseRange(const T &range): p_range(range) {}
|
|
|
|
ReverseRange(const ReverseRange &it): p_range(it.p_range) {}
|
|
|
|
ReverseRange(ReverseRange &&it): p_range(move(it.p_range)) {}
|
|
|
|
|
2015-04-18 19:31:07 +00:00
|
|
|
ReverseRange &operator=(const ReverseRange &v) {
|
|
|
|
p_range = v.p_range;
|
|
|
|
return *this;
|
|
|
|
}
|
|
|
|
ReverseRange &operator=(ReverseRange &&v) {
|
|
|
|
p_range = move(v.p_range);
|
|
|
|
return *this;
|
|
|
|
}
|
|
|
|
ReverseRange &operator=(const T &v) {
|
|
|
|
p_range = v;
|
|
|
|
return *this;
|
|
|
|
}
|
|
|
|
ReverseRange &operator=(T &&v) {
|
|
|
|
p_range = forward<T>(v);
|
|
|
|
return *this;
|
|
|
|
}
|
|
|
|
|
2015-04-16 19:35:10 +00:00
|
|
|
bool empty () const { return p_range.empty (); }
|
2015-04-16 01:07:32 +00:00
|
|
|
size_t length() const { return p_range.length(); }
|
|
|
|
|
|
|
|
void pop_first() { p_range.pop_last (); }
|
|
|
|
void pop_last () { p_range.pop_first(); }
|
|
|
|
|
2015-04-16 19:35:10 +00:00
|
|
|
bool operator==(const ReverseRange &v) const {
|
2015-04-16 01:07:32 +00:00
|
|
|
return p_range == v.p_range;
|
|
|
|
}
|
2015-04-16 19:35:10 +00:00
|
|
|
bool operator!=(const ReverseRange &v) const {
|
2015-04-16 01:07:32 +00:00
|
|
|
return p_range != v.p_range;
|
|
|
|
}
|
|
|
|
|
2015-04-17 19:14:43 +00:00
|
|
|
typename RangeTraits<T>::reference first() { return p_range.last(); }
|
|
|
|
typename RangeTraits<T>::reference first() const { return p_range.last(); }
|
2015-04-16 01:07:32 +00:00
|
|
|
|
2015-04-17 19:14:43 +00:00
|
|
|
typename RangeTraits<T>::reference last() { return p_range.first(); }
|
|
|
|
typename RangeTraits<T>::reference last() const { return p_range.first(); }
|
2015-04-16 01:07:32 +00:00
|
|
|
|
2015-04-17 19:14:43 +00:00
|
|
|
typename RangeTraits<T>::reference operator[](size_t i) {
|
2015-04-16 01:13:53 +00:00
|
|
|
return p_range[length() - i - 1];
|
|
|
|
}
|
2015-04-17 19:14:43 +00:00
|
|
|
typename RangeTraits<T>::reference operator[](size_t i) const {
|
2015-04-16 01:13:53 +00:00
|
|
|
return p_range[length() - i - 1];
|
|
|
|
}
|
2015-04-16 01:07:32 +00:00
|
|
|
|
2015-04-18 21:05:24 +00:00
|
|
|
ReverseRange<T> slice(size_t start, size_t end) {
|
|
|
|
size_t len = p_range.length();
|
|
|
|
return ReverseRange<T>(p_range.slice(len - end, len - start));
|
|
|
|
}
|
|
|
|
|
2015-04-16 01:07:32 +00:00
|
|
|
private:
|
|
|
|
T p_range;
|
|
|
|
};
|
|
|
|
|
|
|
|
template<typename T>
|
2015-04-16 22:47:31 +00:00
|
|
|
ReverseRange<T> make_reverse_range(const T &it) {
|
2015-04-16 01:07:32 +00:00
|
|
|
return ReverseRange<T>(it);
|
|
|
|
}
|
2015-04-16 22:06:39 +00:00
|
|
|
|
|
|
|
template<typename T>
|
2015-04-17 20:33:01 +00:00
|
|
|
struct MoveRange: InputRangeBase<MoveRange<T>,
|
2015-04-17 19:14:43 +00:00
|
|
|
typename RangeTraits<T>::category,
|
|
|
|
typename RangeTraits<T>::value,
|
|
|
|
typename RangeTraits<T>::value &&
|
|
|
|
> {
|
2015-04-16 22:06:39 +00:00
|
|
|
MoveRange(): p_range() {}
|
|
|
|
MoveRange(const T &range): p_range(range) {}
|
|
|
|
MoveRange(const MoveRange &it): p_range(it.p_range) {}
|
|
|
|
MoveRange(MoveRange &&it): p_range(move(it.p_range)) {}
|
|
|
|
|
2015-04-18 19:31:07 +00:00
|
|
|
MoveRange &operator=(const MoveRange &v) {
|
|
|
|
p_range = v.p_range;
|
|
|
|
return *this;
|
|
|
|
}
|
|
|
|
MoveRange &operator=(MoveRange &&v) {
|
|
|
|
p_range = move(v.p_range);
|
|
|
|
return *this;
|
|
|
|
}
|
|
|
|
MoveRange &operator=(const T &v) {
|
|
|
|
p_range = v;
|
|
|
|
return *this;
|
|
|
|
}
|
|
|
|
MoveRange &operator=(T &&v) {
|
|
|
|
p_range = forward<T>(v);
|
|
|
|
return *this;
|
|
|
|
}
|
|
|
|
|
2015-04-16 22:06:39 +00:00
|
|
|
bool empty () const { return p_range.empty (); }
|
|
|
|
size_t length() const { return p_range.length(); }
|
|
|
|
|
|
|
|
void pop_first() { p_range.pop_first(); }
|
|
|
|
void pop_last () { p_range.pop_last (); }
|
|
|
|
|
|
|
|
bool operator==(const MoveRange &v) const {
|
|
|
|
return p_range == v.p_range;
|
|
|
|
}
|
|
|
|
bool operator!=(const MoveRange &v) const {
|
|
|
|
return p_range != v.p_range;
|
|
|
|
}
|
|
|
|
|
2015-04-17 19:14:43 +00:00
|
|
|
typename RangeTraits<T>::value &&first() { return move(p_range.first()); }
|
|
|
|
typename RangeTraits<T>::value &&last () { return move(p_range.last ()); }
|
2015-04-16 22:06:39 +00:00
|
|
|
|
2015-04-17 19:14:43 +00:00
|
|
|
typename RangeTraits<T>::value &&operator[](size_t i) {
|
2015-04-16 22:06:39 +00:00
|
|
|
return move(p_range[i]);
|
|
|
|
}
|
|
|
|
|
2015-04-18 21:05:24 +00:00
|
|
|
MoveRange<T> slice(size_t start, size_t end) {
|
|
|
|
return MoveRange<T>(p_range.slice(start, end));
|
|
|
|
}
|
|
|
|
|
2015-04-16 22:06:39 +00:00
|
|
|
private:
|
|
|
|
T p_range;
|
|
|
|
};
|
2015-04-16 22:26:10 +00:00
|
|
|
|
|
|
|
template<typename T>
|
2015-04-16 22:47:31 +00:00
|
|
|
MoveRange<T> make_move_range(const T &it) {
|
2015-04-16 22:26:10 +00:00
|
|
|
return MoveRange<T>(it);
|
|
|
|
}
|
2015-04-16 23:55:50 +00:00
|
|
|
|
|
|
|
template<typename T>
|
2015-04-17 20:33:01 +00:00
|
|
|
struct NumberRange: InputRangeBase<NumberRange<T>, ForwardRange, T> {
|
2015-04-16 23:55:50 +00:00
|
|
|
NumberRange(): p_a(0), p_b(0), p_step(0) {}
|
|
|
|
NumberRange(const NumberRange &it): p_a(it.p_a), p_b(it.p_b),
|
|
|
|
p_step(it.p_step) {}
|
|
|
|
NumberRange(T a, T b, T step = 1): p_a(a), p_b(b), p_step(step) {}
|
2015-04-17 19:33:36 +00:00
|
|
|
NumberRange(T v): p_a(0), p_b(v), p_step(1) {}
|
2015-04-16 23:55:50 +00:00
|
|
|
|
2015-04-17 19:20:38 +00:00
|
|
|
bool operator==(const NumberRange &v) const {
|
|
|
|
return p_a == v.p_a && p_b == v.p_b && p_step == v.p_step;
|
|
|
|
}
|
|
|
|
bool operator!=(const NumberRange &v) const {
|
|
|
|
return p_a != v.p_a || p_b != v.p_b || p_step != v.p_step;
|
|
|
|
}
|
|
|
|
|
2015-04-16 23:55:50 +00:00
|
|
|
bool empty() const { return p_a * p_step >= p_b * p_step; }
|
|
|
|
void pop_first() { p_a += p_step; }
|
|
|
|
T &first() { return p_a; }
|
|
|
|
|
|
|
|
private:
|
|
|
|
T p_a, p_b, p_step;
|
|
|
|
};
|
|
|
|
|
|
|
|
template<typename T>
|
|
|
|
NumberRange<T> range(T a, T b, T step = 1) {
|
|
|
|
return NumberRange<T>(a, b, step);
|
|
|
|
}
|
2015-04-17 19:33:36 +00:00
|
|
|
|
|
|
|
template<typename T>
|
|
|
|
NumberRange<T> range(T v) {
|
|
|
|
return NumberRange<T>(v);
|
|
|
|
}
|
2015-04-18 23:02:38 +00:00
|
|
|
|
|
|
|
template<typename T>
|
|
|
|
struct PointerRange: InputRangeBase<PointerRange<T>, RandomAccessRange, T> {
|
|
|
|
PointerRange(): p_beg(nullptr), p_end(nullptr) {}
|
|
|
|
PointerRange(const PointerRange &v): p_beg(v.p_beg), p_end(v.p_end) {}
|
|
|
|
PointerRange(T *beg, T *end): p_beg(beg), p_end(end) {}
|
|
|
|
|
|
|
|
bool operator==(const PointerRange &v) const {
|
|
|
|
return p_beg == v.p_beg && p_end == v.p_end;
|
|
|
|
}
|
|
|
|
bool operator!=(const PointerRange &v) const {
|
|
|
|
return p_beg != v.p_beg || p_end != v.p_end;
|
|
|
|
}
|
|
|
|
|
|
|
|
/* satisfy InputRange / ForwardRange */
|
|
|
|
bool empty() const { return p_beg == nullptr; }
|
|
|
|
|
|
|
|
void pop_first() {
|
|
|
|
if (p_beg == nullptr) return;
|
|
|
|
if (++p_beg == p_end) p_beg = p_end = nullptr;
|
|
|
|
}
|
|
|
|
|
|
|
|
T &first() { return *p_beg; }
|
|
|
|
const T &first() const { return *p_beg; }
|
|
|
|
|
|
|
|
/* satisfy BidirectionalRange */
|
|
|
|
void pop_last() {
|
|
|
|
if (p_end-- == p_beg) { p_end = nullptr; return; }
|
|
|
|
if (p_end == p_beg) p_beg = p_end = nullptr;
|
|
|
|
}
|
|
|
|
|
|
|
|
T &last() { return *(p_end - 1); }
|
|
|
|
const T &last() const { return *(p_end - 1); }
|
|
|
|
|
|
|
|
/* satisfy RandomAccessRange */
|
|
|
|
size_t length() const { return p_end - p_beg; }
|
|
|
|
|
|
|
|
PointerRange slice(size_t start, size_t end) {
|
|
|
|
return PointerRange(p_beg + start, p_beg + end);
|
|
|
|
}
|
|
|
|
|
|
|
|
T &operator[](size_t i) { return p_beg[i]; }
|
|
|
|
const T &operator[](size_t i) const { return p_beg[i]; }
|
|
|
|
|
|
|
|
private:
|
|
|
|
T *p_beg, *p_end;
|
|
|
|
};
|
2015-04-15 23:18:00 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
#endif
|