[iter] Add hb_range()
hb_range() is like Python range. hb_iota() has slightly different API. Ie. it takes a start, instead of end.
This commit is contained in:
parent
05867d9f53
commit
7675d0d3a6
|
@ -572,13 +572,13 @@ struct
|
||||||
}
|
}
|
||||||
HB_FUNCOBJ (hb_apply);
|
HB_FUNCOBJ (hb_apply);
|
||||||
|
|
||||||
/* hb_iota() */
|
/* hb_iota()/hb_range() */
|
||||||
|
|
||||||
template <typename T, typename S>
|
template <typename T, typename S>
|
||||||
struct hb_iota_iter_t :
|
struct hb_counter_iter_t :
|
||||||
hb_iter_t<hb_iota_iter_t<T, S>, T>
|
hb_iter_t<hb_counter_iter_t<T, S>, T>
|
||||||
{
|
{
|
||||||
hb_iota_iter_t (T start, T end_, S step) : v (start), end_ (end__for (start, end_, step)), step (step) {}
|
hb_counter_iter_t (T start, T end_, S step) : v (start), end_ (end__for (start, end_, step)), step (step) {}
|
||||||
|
|
||||||
typedef T __item_t__;
|
typedef T __item_t__;
|
||||||
static constexpr bool is_random_access_iterator = true;
|
static constexpr bool is_random_access_iterator = true;
|
||||||
|
@ -591,8 +591,8 @@ struct hb_iota_iter_t :
|
||||||
void __forward__ (unsigned n) { v += n * step; }
|
void __forward__ (unsigned n) { v += n * step; }
|
||||||
void __prev__ () { v -= step; }
|
void __prev__ () { v -= step; }
|
||||||
void __rewind__ (unsigned n) { v -= n * step; }
|
void __rewind__ (unsigned n) { v -= n * step; }
|
||||||
hb_iota_iter_t __end___ () const { hb_iota_iter_t (end_, end_, step); }
|
hb_counter_iter_t __end___ () const { hb_counter_iter_t (end_, end_, step); }
|
||||||
bool operator != (const hb_iota_iter_t& o) const
|
bool operator != (const hb_counter_iter_t& o) const
|
||||||
{ return v != o.v || end_ != o.end_ || step != o.step; }
|
{ return v != o.v || end_ != o.end_ || step != o.step; }
|
||||||
|
|
||||||
private:
|
private:
|
||||||
|
@ -612,15 +612,22 @@ struct hb_iota_iter_t :
|
||||||
};
|
};
|
||||||
struct
|
struct
|
||||||
{
|
{
|
||||||
template <typename T = unsigned> hb_iota_iter_t<T, unsigned>
|
template <typename T = unsigned, typename S = unsigned> hb_counter_iter_t<T, S>
|
||||||
operator () (T end = (unsigned) -1) const
|
operator () (T start = 0u, S&& step = 1u) const
|
||||||
{ return hb_iota_iter_t<T, unsigned> (0, end, 1u); }
|
{ return hb_counter_iter_t<T, S> (start, hb_int_max (T), step); }
|
||||||
|
|
||||||
template <typename T, typename S = unsigned> hb_iota_iter_t<T, S>
|
|
||||||
operator () (T start, T end, S&& step = 1u) const
|
|
||||||
{ return hb_iota_iter_t<T, S> (start, end, step); }
|
|
||||||
}
|
}
|
||||||
HB_FUNCOBJ (hb_iota);
|
HB_FUNCOBJ (hb_iota);
|
||||||
|
struct
|
||||||
|
{
|
||||||
|
template <typename T = unsigned> hb_counter_iter_t<T, unsigned>
|
||||||
|
operator () (T end = (unsigned) -1) const
|
||||||
|
{ return hb_counter_iter_t<T, unsigned> (0, end, 1u); }
|
||||||
|
|
||||||
|
template <typename T, typename S = unsigned> hb_counter_iter_t<T, S>
|
||||||
|
operator () (T start, T end, S&& step = 1u) const
|
||||||
|
{ return hb_counter_iter_t<T, S> (start, end, step); }
|
||||||
|
}
|
||||||
|
HB_FUNCOBJ (hb_range);
|
||||||
|
|
||||||
|
|
||||||
/* hb_sink() */
|
/* hb_sink() */
|
||||||
|
|
|
@ -264,14 +264,17 @@ main (int argc, char **argv)
|
||||||
s >> vl;
|
s >> vl;
|
||||||
|
|
||||||
hb_iota ();
|
hb_iota ();
|
||||||
assert (hb_iota (9).len () == 9);
|
hb_iota (3);
|
||||||
assert (hb_iota (2, 9).len () == 7);
|
hb_iota (3, 2);
|
||||||
assert (hb_iota (2, 9, 3).len () == 3);
|
hb_range ();
|
||||||
assert (hb_iota (2, 8, 3).len () == 2);
|
assert (hb_range (9).len () == 9);
|
||||||
assert (hb_iota (2, 7, 3).len () == 2);
|
assert (hb_range (2, 9).len () == 7);
|
||||||
assert (hb_iota (-2, -9, -3).len () == 3);
|
assert (hb_range (2, 9, 3).len () == 3);
|
||||||
assert (hb_iota (-2, -8, -3).len () == 2);
|
assert (hb_range (2, 8, 3).len () == 2);
|
||||||
assert (hb_iota (-2, -7, -3).len () == 2);
|
assert (hb_range (2, 7, 3).len () == 2);
|
||||||
|
assert (hb_range (-2, -9, -3).len () == 3);
|
||||||
|
assert (hb_range (-2, -8, -3).len () == 2);
|
||||||
|
assert (hb_range (-2, -7, -3).len () == 2);
|
||||||
|
|
||||||
return 0;
|
return 0;
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue