589 lines
12 KiB
D
589 lines
12 KiB
D
/* This Source Code Form is subject to the terms of the Mozilla Public
|
|
* License, v. 2.0. If a copy of the MPL was not distributed with this
|
|
* file, You can obtain one at http://mozilla.org/MPL/2.0/. */
|
|
|
|
/**
|
|
* Range adapters.
|
|
*
|
|
* A range adapter wraps another range and modifies the way, how the original
|
|
* range is iterated, or the order in which its elements are accessed.
|
|
*
|
|
* All adapters are lazy algorithms, they request the next element of the
|
|
* adapted range on demand.
|
|
*
|
|
* Copyright: Eugene Wissner 2018.
|
|
* License: $(LINK2 https://www.mozilla.org/en-US/MPL/2.0/,
|
|
* Mozilla Public License, v. 2.0).
|
|
* Authors: $(LINK2 mailto:info@caraus.de, Eugene Wissner)
|
|
* Source: $(LINK2 https://github.com/caraus-ecms/tanya/blob/master/source/tanya/algorithm/iteration.d,
|
|
* tanya/algorithm/iteration.d)
|
|
*/
|
|
module tanya.algorithm.iteration;
|
|
|
|
import tanya.algorithm.comparison;
|
|
import tanya.algorithm.mutation;
|
|
import tanya.range;
|
|
|
|
private mixin template Take(R, bool exactly)
|
|
{
|
|
private R source;
|
|
size_t length_;
|
|
|
|
@disable this();
|
|
|
|
private this(R source, size_t length)
|
|
{
|
|
this.source = source;
|
|
static if (!exactly && hasLength!R)
|
|
{
|
|
this.length_ = min(source.length, length);
|
|
}
|
|
else
|
|
{
|
|
this.length_ = length;
|
|
}
|
|
}
|
|
|
|
@property auto ref front()
|
|
in
|
|
{
|
|
assert(!empty);
|
|
}
|
|
do
|
|
{
|
|
return this.source.front;
|
|
}
|
|
|
|
void popFront()
|
|
in
|
|
{
|
|
assert(!empty);
|
|
}
|
|
do
|
|
{
|
|
this.source.popFront();
|
|
--this.length_;
|
|
}
|
|
|
|
@property bool empty()
|
|
{
|
|
static if (exactly || isInfinite!R)
|
|
{
|
|
return length == 0;
|
|
}
|
|
else
|
|
{
|
|
return length == 0 || this.source.empty;
|
|
}
|
|
}
|
|
|
|
@property size_t length()
|
|
{
|
|
return this.length_;
|
|
}
|
|
|
|
static if (hasAssignableElements!R)
|
|
{
|
|
@property void front(ref ElementType!R value)
|
|
in
|
|
{
|
|
assert(!empty);
|
|
}
|
|
do
|
|
{
|
|
this.source.front = value;
|
|
}
|
|
|
|
@property void front(ElementType!R value)
|
|
in
|
|
{
|
|
assert(!empty);
|
|
}
|
|
do
|
|
{
|
|
this.source.front = move(value);
|
|
}
|
|
}
|
|
|
|
static if (isForwardRange!R)
|
|
{
|
|
typeof(this) save()
|
|
{
|
|
return typeof(this)(this.source.save(), length);
|
|
}
|
|
}
|
|
static if (isRandomAccessRange!R)
|
|
{
|
|
@property auto ref back()
|
|
in
|
|
{
|
|
assert(!empty);
|
|
}
|
|
do
|
|
{
|
|
return this.source[this.length - 1];
|
|
}
|
|
|
|
void popBack()
|
|
in
|
|
{
|
|
assert(!empty);
|
|
}
|
|
do
|
|
{
|
|
--this.length_;
|
|
}
|
|
|
|
auto ref opIndex(size_t i)
|
|
in
|
|
{
|
|
assert(i < length);
|
|
}
|
|
do
|
|
{
|
|
return this.source[i];
|
|
}
|
|
|
|
static if (hasAssignableElements!R)
|
|
{
|
|
@property void back(ref ElementType!R value)
|
|
in
|
|
{
|
|
assert(!empty);
|
|
}
|
|
do
|
|
{
|
|
this.source[length - 1] = value;
|
|
}
|
|
|
|
@property void back(ElementType!R value)
|
|
in
|
|
{
|
|
assert(!empty);
|
|
}
|
|
do
|
|
{
|
|
this.source[length - 1] = move(value);
|
|
}
|
|
|
|
void opIndexAssign(ref ElementType!R value, size_t i)
|
|
in
|
|
{
|
|
assert(i < length);
|
|
}
|
|
do
|
|
{
|
|
this.source[i] = value;
|
|
}
|
|
|
|
void opIndexAssign(ElementType!R value, size_t i)
|
|
in
|
|
{
|
|
assert(i < length);
|
|
}
|
|
do
|
|
{
|
|
this.source[i] = move(value);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Takes $(D_PARAM n) elements from $(D_PARAM range).
|
|
*
|
|
* If $(D_PARAM range) doesn't have $(D_PARAM n) elements, the resulting range
|
|
* spans all elements of $(D_PARAM range).
|
|
*
|
|
* $(D_PSYMBOL take) is particulary useful with infinite ranges. You can take
|
|
` $(B n) elements from such range and pass the result to an algorithm which
|
|
* expects a finit range.
|
|
*
|
|
* Params:
|
|
* R = Type of the adapted range.
|
|
* range = The range to take the elements from.
|
|
* n = The number of elements to take.
|
|
*
|
|
* Returns: A range containing maximum $(D_PARAM n) first elements of
|
|
* $(D_PARAM range).
|
|
*
|
|
* See_Also: $(D_PSYMBOL takeExactly).
|
|
*/
|
|
auto take(R)(R range, size_t n)
|
|
if (isInputRange!R)
|
|
{
|
|
static struct Take
|
|
{
|
|
mixin .Take!(R, false);
|
|
|
|
static if (hasSlicing!R)
|
|
{
|
|
auto opSlice(size_t i, size_t j)
|
|
in
|
|
{
|
|
assert(i <= j);
|
|
assert(j <= length);
|
|
}
|
|
do
|
|
{
|
|
return typeof(this)(this.source[i .. j], length);
|
|
}
|
|
}
|
|
}
|
|
return Take(range, n);
|
|
}
|
|
|
|
///
|
|
@nogc nothrow pure @safe unittest
|
|
{
|
|
static struct InfiniteRange
|
|
{
|
|
private size_t front_ = 1;
|
|
|
|
enum bool empty = false;
|
|
|
|
@property size_t front() @nogc nothrow pure @safe
|
|
{
|
|
return this.front_;
|
|
}
|
|
|
|
@property void front(size_t i) @nogc nothrow pure @safe
|
|
{
|
|
this.front_ = i;
|
|
}
|
|
|
|
void popFront() @nogc nothrow pure @safe
|
|
{
|
|
++this.front_;
|
|
}
|
|
|
|
size_t opIndex(size_t i) @nogc nothrow pure @safe
|
|
{
|
|
return this.front_ + i;
|
|
}
|
|
|
|
void opIndexAssign(size_t value, size_t i) @nogc nothrow pure @safe
|
|
{
|
|
this.front = i + value;
|
|
}
|
|
|
|
InfiniteRange save() @nogc nothrow pure @safe
|
|
{
|
|
return this;
|
|
}
|
|
}
|
|
|
|
auto t = InfiniteRange().take(3);
|
|
assert(t.length == 3);
|
|
assert(t.front == 1);
|
|
assert(t.back == 3);
|
|
|
|
t.popFront();
|
|
assert(t.front == 2);
|
|
assert(t.back == 3);
|
|
|
|
t.popBack();
|
|
assert(t.front == 2);
|
|
assert(t.back == 2);
|
|
|
|
t.popFront();
|
|
assert(t.empty);
|
|
}
|
|
|
|
/**
|
|
* Takes exactly $(D_PARAM n) elements from $(D_PARAM range).
|
|
*
|
|
* $(D_PARAM range) must have at least $(D_PARAM n) elements.
|
|
*
|
|
* $(D_PSYMBOL takeExactly) is particulary useful with infinite ranges. You can
|
|
` take $(B n) elements from such range and pass the result to an algorithm
|
|
* which expects a finit range.
|
|
*
|
|
* Params:
|
|
* R = Type of the adapted range.
|
|
* range = The range to take the elements from.
|
|
* n = The number of elements to take.
|
|
*
|
|
* Returns: A range containing $(D_PARAM n) first elements of $(D_PARAM range).
|
|
*
|
|
* See_Also: $(D_PSYMBOL take).
|
|
*/
|
|
auto takeExactly(R)(R range, size_t n)
|
|
if (isInputRange!R)
|
|
{
|
|
static if (hasSlicing!R)
|
|
{
|
|
return range[0 .. n];
|
|
}
|
|
else
|
|
{
|
|
static struct TakeExactly
|
|
{
|
|
mixin Take!(R, true);
|
|
}
|
|
return TakeExactly(range, n);
|
|
}
|
|
}
|
|
|
|
///
|
|
@nogc nothrow pure @safe unittest
|
|
{
|
|
static struct InfiniteRange
|
|
{
|
|
private size_t front_ = 1;
|
|
|
|
enum bool empty = false;
|
|
|
|
@property size_t front() @nogc nothrow pure @safe
|
|
{
|
|
return this.front_;
|
|
}
|
|
|
|
@property void front(size_t i) @nogc nothrow pure @safe
|
|
{
|
|
this.front_ = i;
|
|
}
|
|
|
|
void popFront() @nogc nothrow pure @safe
|
|
{
|
|
++this.front_;
|
|
}
|
|
|
|
size_t opIndex(size_t i) @nogc nothrow pure @safe
|
|
{
|
|
return this.front_ + i;
|
|
}
|
|
|
|
void opIndexAssign(size_t value, size_t i) @nogc nothrow pure @safe
|
|
{
|
|
this.front = i + value;
|
|
}
|
|
|
|
InfiniteRange save() @nogc nothrow pure @safe
|
|
{
|
|
return this;
|
|
}
|
|
}
|
|
|
|
auto t = InfiniteRange().takeExactly(3);
|
|
assert(t.length == 3);
|
|
assert(t.front == 1);
|
|
assert(t.back == 3);
|
|
|
|
t.popFront();
|
|
assert(t.front == 2);
|
|
assert(t.back == 3);
|
|
|
|
t.popBack();
|
|
assert(t.front == 2);
|
|
assert(t.back == 2);
|
|
|
|
t.popFront();
|
|
assert(t.empty);
|
|
}
|
|
|
|
// Takes minimum length if the range length > n
|
|
@nogc nothrow pure @safe unittest
|
|
{
|
|
auto range = take(cast(int[]) null, 8);
|
|
assert(range.length == 0);
|
|
}
|
|
|
|
@nogc nothrow pure @safe unittest
|
|
{
|
|
const int[9] range = [1, 2, 3, 4, 5, 6, 7, 8, 9];
|
|
{
|
|
auto slice = take(range[], 8)[1 .. 3];
|
|
|
|
assert(slice.length == 2);
|
|
assert(slice.front == 2);
|
|
assert(slice.back == 3);
|
|
}
|
|
{
|
|
auto slice = takeExactly(range[], 8)[1 .. 3];
|
|
|
|
assert(slice.length == 2);
|
|
assert(slice.front == 2);
|
|
assert(slice.back == 3);
|
|
}
|
|
}
|
|
|
|
private
|
|
{
|
|
// Reverse-access-order range returned by `retro`.
|
|
struct Retro(Range)
|
|
{
|
|
Range source;
|
|
|
|
@disable this();
|
|
|
|
private this(Range source)
|
|
{
|
|
this.source = source;
|
|
}
|
|
|
|
Retro save()
|
|
{
|
|
return this;
|
|
}
|
|
|
|
@property auto ref front()
|
|
in (!empty)
|
|
{
|
|
return this.source.back;
|
|
}
|
|
|
|
void popFront()
|
|
in (!empty)
|
|
{
|
|
this.source.popBack();
|
|
}
|
|
|
|
@property auto ref back()
|
|
in (!empty)
|
|
{
|
|
return this.source.front;
|
|
}
|
|
|
|
void popBack()
|
|
in (!empty)
|
|
{
|
|
this.source.popFront();
|
|
}
|
|
|
|
@property bool empty()
|
|
{
|
|
return this.source.empty;
|
|
}
|
|
|
|
static if (hasLength!Range)
|
|
{
|
|
@property size_t length()
|
|
{
|
|
return this.source.length;
|
|
}
|
|
}
|
|
|
|
static if (isRandomAccessRange!Range && hasLength!Range)
|
|
{
|
|
auto ref opIndex(size_t i)
|
|
in (i < length)
|
|
{
|
|
return this.source[$ - ++i];
|
|
}
|
|
}
|
|
|
|
static if (hasAssignableElements!Range)
|
|
{
|
|
@property void front(ref ElementType!Range value)
|
|
in (!empty)
|
|
{
|
|
this.source.back = value;
|
|
}
|
|
|
|
@property void front(ElementType!Range value)
|
|
in (!empty)
|
|
{
|
|
this.source.back = move(value);
|
|
}
|
|
|
|
@property void back(ref ElementType!Range value)
|
|
in (!empty)
|
|
{
|
|
this.source.front = value;
|
|
}
|
|
|
|
@property void back(ElementType!Range value)
|
|
in (!empty)
|
|
{
|
|
this.source.front = move(value);
|
|
}
|
|
|
|
static if (isRandomAccessRange!Range && hasLength!Range)
|
|
{
|
|
void opIndexAssign(ref ElementType!Range value, size_t i)
|
|
in (i < length)
|
|
{
|
|
this.source[$ - ++i] = value;
|
|
}
|
|
|
|
void opIndexAssign(ElementType!Range value, size_t i)
|
|
in (i < length)
|
|
{
|
|
this.source[$ - ++i] = move(value);
|
|
}
|
|
}
|
|
}
|
|
|
|
version (unittest) static assert(isBidirectionalRange!Retro);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Iterates a bidirectional range backwards.
|
|
*
|
|
* If $(D_PARAM Range) is a random-access range as well, the resulting range
|
|
* is a random-access range too.
|
|
*
|
|
* Params:
|
|
* Range = Bidirectional range type.
|
|
* range = Bidirectional range.
|
|
*
|
|
* Returns: Bidirectional range with the elements order reversed.
|
|
*/
|
|
auto retro(Range)(return Range range)
|
|
if (isBidirectionalRange!Range)
|
|
{
|
|
// Special case: retro(retro(range)) is range
|
|
static if (is(Range == Retro!RRange, RRange))
|
|
return range.source;
|
|
else
|
|
return Retro!Range(range);
|
|
}
|
|
|
|
///
|
|
@nogc nothrow pure @safe unittest
|
|
{
|
|
import tanya.algorithm.comparison : equal;
|
|
|
|
const int[3] given = [1, 2, 3];
|
|
const int[3] expected = [3, 2, 1];
|
|
|
|
auto actual = retro(given[]);
|
|
|
|
assert(actual.length == expected.length);
|
|
assert(!actual.empty);
|
|
assert(equal(actual, expected[]));
|
|
}
|
|
|
|
// Elements are accessible in reverse order
|
|
@nogc nothrow pure @safe unittest
|
|
{
|
|
const int[3] given = [1, 2, 3];
|
|
auto actual = retro(given[]);
|
|
|
|
assert(actual.back == given[].front);
|
|
assert(actual[0] == 3);
|
|
assert(actual[2] == 1);
|
|
|
|
actual.popBack();
|
|
assert(actual.back == 2);
|
|
assert(actual[1] == 2);
|
|
}
|
|
|
|
// Elements can be assigned
|
|
@nogc nothrow pure @safe unittest
|
|
{
|
|
int[4] given = [1, 2, 3, 4];
|
|
auto actual = retro(given[]);
|
|
|
|
actual.front = 5;
|
|
assert(given[].back == 5);
|
|
|
|
actual.back = 8;
|
|
assert(given[].front == 8);
|
|
|
|
actual[2] = 10;
|
|
assert(given[1] == 10);
|
|
}
|