mirror of
https://github.com/dlang/phobos.git
synced 2025-05-04 00:54:05 +03:00
Moved count from std.string to std.algorithm, which works toward fixing 4883
This commit is contained in:
parent
9d30071ea7
commit
b2bf0bc48c
2 changed files with 149 additions and 16 deletions
151
std/algorithm.d
151
std/algorithm.d
|
@ -2831,6 +2831,116 @@ unittest
|
||||||
assert(find!(pred)(a) == a);
|
assert(find!(pred)(a) == a);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// findSkip
|
||||||
|
/**
|
||||||
|
* Similar to $(D find), except returns the balance of $(D haystack)
|
||||||
|
* just after the found range. If $(D needle) is not found, returns a
|
||||||
|
* tuple with an empty range and $(D false). Otherwise, returns the
|
||||||
|
* portion of $(D haystack) starting right after the first occurrence
|
||||||
|
* of $(D needle).
|
||||||
|
*
|
||||||
|
* Example:
|
||||||
|
----
|
||||||
|
assert(findSkip("abcdef", "cd").balance == "ef");
|
||||||
|
assert(findSkip("abcdef", "cxd").balance == null);
|
||||||
|
assert(findSkip("abcdef", "cxd") == tuple("", false));
|
||||||
|
----
|
||||||
|
*/
|
||||||
|
Tuple!(R1, "balance", bool, "found")
|
||||||
|
findSkip(alias pred = "a == b", R1, R2)(R1 haystack, R2 needle)
|
||||||
|
if (isForwardRange!R1 && isForwardRange!R2
|
||||||
|
&& is(typeof(binaryFun!pred(haystack.front, needle.front))))
|
||||||
|
{
|
||||||
|
static if (isSomeString!R1 && isSomeString!R2 || isRandomAccessRange!R1 && hasLength!R2)
|
||||||
|
{
|
||||||
|
haystack = find!pred(haystack, needle);
|
||||||
|
if (haystack.empty) return typeof(return)(haystack, false);
|
||||||
|
return typeof(return)(haystack[needle.length .. haystack.length], true);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
if (needle.empty)
|
||||||
|
{
|
||||||
|
// The empty range is always found
|
||||||
|
return typeof(return)(haystack, true);
|
||||||
|
}
|
||||||
|
|
||||||
|
enum estimateNeedleLength = hasLength!R1 && !hasLength!R2;
|
||||||
|
|
||||||
|
static if (hasLength!R1)
|
||||||
|
{
|
||||||
|
static if (hasLength!R2)
|
||||||
|
size_t estimatedNeedleLength = 0;
|
||||||
|
else
|
||||||
|
immutable size_t estimatedNeedleLength = needle.length;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool haystackTooShort()
|
||||||
|
{
|
||||||
|
static if (hasLength!R1)
|
||||||
|
{
|
||||||
|
return haystack.length < estimatedNeedleLength;
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
return haystack.empty;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
searching:
|
||||||
|
for (;; haystack.popFront())
|
||||||
|
{
|
||||||
|
if (haystackTooShort())
|
||||||
|
{
|
||||||
|
// Failed search
|
||||||
|
static if (hasLength!R1)
|
||||||
|
{
|
||||||
|
static if (is(typeof(haystack[haystack.length ..
|
||||||
|
haystack.length]) : R1))
|
||||||
|
return tuple(haystack[haystack.length .. haystack.length], false);
|
||||||
|
else
|
||||||
|
return tuple(R1.init, false);
|
||||||
|
}
|
||||||
|
else
|
||||||
|
{
|
||||||
|
assert(haystack.empty);
|
||||||
|
return typeof(return)(haystack, false);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
static if (estimateNeedleLength)
|
||||||
|
size_t matchLength = 0;
|
||||||
|
auto h = haystack.save;
|
||||||
|
for (auto n = needle.save;
|
||||||
|
!n.empty;
|
||||||
|
h.popFront(), n.popFront())
|
||||||
|
{
|
||||||
|
if (h.empty || !binaryFun!pred(h.front, n.front))
|
||||||
|
{
|
||||||
|
// Failed searching n in h
|
||||||
|
static if (estimateNeedleLength)
|
||||||
|
{
|
||||||
|
if (estimatedNeedleLength < matchLength)
|
||||||
|
estimatedNeedleLength = matchLength;
|
||||||
|
}
|
||||||
|
continue searching;
|
||||||
|
}
|
||||||
|
static if (estimateNeedleLength)
|
||||||
|
++matchLength;
|
||||||
|
}
|
||||||
|
// Found
|
||||||
|
return typeof(return)(h, true);
|
||||||
|
}
|
||||||
|
return typeof(return)(haystack, false);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
unittest
|
||||||
|
{
|
||||||
|
assert(findSkip("abcdef", "cd").balance == "ef");
|
||||||
|
assert(findSkip("abcdef", "cxd").balance == null);
|
||||||
|
assert(findSkip("abcdef", "cxd") == tuple("", false));
|
||||||
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
If $(D haystack) supports slicing, returns the smallest number $(D n)
|
If $(D haystack) supports slicing, returns the smallest number $(D n)
|
||||||
such that $(D haystack[n .. $].startsWith!pred(needle)). Oherwise,
|
such that $(D haystack[n .. $].startsWith!pred(needle)). Oherwise,
|
||||||
|
@ -3281,7 +3391,7 @@ if (isInputRange!(Range) && Ranges.length > 0
|
||||||
alias doesThisEnd lhs;
|
alias doesThisEnd lhs;
|
||||||
alias withOneOfThese rhs;
|
alias withOneOfThese rhs;
|
||||||
// Special case for two arrays
|
// Special case for two arrays
|
||||||
static if (Ranges.length == 1 && isArray!(Range) && isArray!(Ranges[0])
|
static if (Ranges.length == 1 && isArray!Range && isArray!(Ranges[0])
|
||||||
&& is(typeof(binaryFun!(pred)(lhs[0], rhs[0][0]))))
|
&& is(typeof(binaryFun!(pred)(lhs[0], rhs[0][0]))))
|
||||||
{
|
{
|
||||||
if (lhs.length < rhs[0].length) return 0;
|
if (lhs.length < rhs[0].length) return 0;
|
||||||
|
@ -3480,9 +3590,9 @@ assert(count!("a > b")(a, 2) == 5);
|
||||||
*/
|
*/
|
||||||
|
|
||||||
size_t count(alias pred = "a == b", Range, E)(Range r, E value)
|
size_t count(alias pred = "a == b", Range, E)(Range r, E value)
|
||||||
if (isInputRange!(Range))
|
if (isInputRange!Range && is(typeof(binaryFun!pred(r.front, value)) == bool))
|
||||||
{
|
{
|
||||||
bool pred2(ElementType!(Range) a) { return binaryFun!(pred)(a, value); }
|
bool pred2(ElementType!(Range) a) { return binaryFun!pred(a, value); }
|
||||||
return count!(pred2)(r);
|
return count!(pred2)(r);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@ -3504,6 +3614,41 @@ unittest
|
||||||
assert(count!("a == '語'")("日本語"d) == 1);
|
assert(count!("a == '語'")("日本語"d) == 1);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Returns the number of times $(D needle) occurs in $(D
|
||||||
|
* haystack). Throws an exception if $(D needle.empty), as the _count
|
||||||
|
* of the empty range in any range would be infinite. Overlapped
|
||||||
|
* counts are not considered, for example $(D count("aaa", "aa")) is
|
||||||
|
* $(D 1), not $(D 2).
|
||||||
|
*
|
||||||
|
* Example:
|
||||||
|
----
|
||||||
|
assert(count("abcadfabf", "ab") == 2);
|
||||||
|
assert(count("ababab", "abab") == 1);
|
||||||
|
assert(count("ababab", "abx") == 0);
|
||||||
|
----
|
||||||
|
*/
|
||||||
|
size_t count(alias pred = "a == b", R1, R2)(R1 haystack, R2 needle)
|
||||||
|
if (isInputRange!R1 && isForwardRange!R2 && is(typeof(binaryFun!pred(haystack, needle)) == bool))
|
||||||
|
{
|
||||||
|
enforce(!needle.empty, "Cannot count occurrences of an empty range");
|
||||||
|
size_t result;
|
||||||
|
for (;; ++result)
|
||||||
|
{
|
||||||
|
auto r = findSkip!pred(haystack, needle);
|
||||||
|
if (!r.found) break;
|
||||||
|
haystack = r.balance;
|
||||||
|
}
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
|
||||||
|
unittest
|
||||||
|
{
|
||||||
|
assert(count("abcadfabf", "ab") == 2);
|
||||||
|
assert(count("ababab", "abab") == 1);
|
||||||
|
assert(count("ababab", "abx") == 0);
|
||||||
|
}
|
||||||
|
|
||||||
/**
|
/**
|
||||||
Counts the number of elements $(D x) in $(D r) for which $(D pred(x))
|
Counts the number of elements $(D x) in $(D r) for which $(D pred(x))
|
||||||
is $(D true). Performs $(BIGOH r.length) evaluations of $(D pred).
|
is $(D true). Performs $(BIGOH r.length) evaluations of $(D pred).
|
||||||
|
|
14
std/string.d
14
std/string.d
|
@ -2068,19 +2068,7 @@ unittest
|
||||||
* Count up all instances of sub[] in s[].
|
* Count up all instances of sub[] in s[].
|
||||||
*/
|
*/
|
||||||
|
|
||||||
size_t count(S)(S s, S sub) if (isSomeString!S)
|
alias std.algorithm.count count;
|
||||||
{
|
|
||||||
size_t count = 0;
|
|
||||||
for (size_t i = 0; i < s.length; )
|
|
||||||
{
|
|
||||||
immutable j = indexOf(s[i .. s.length], sub);
|
|
||||||
if (j == -1)
|
|
||||||
break;
|
|
||||||
count++;
|
|
||||||
i += j + sub.length;
|
|
||||||
}
|
|
||||||
return count;
|
|
||||||
}
|
|
||||||
|
|
||||||
unittest
|
unittest
|
||||||
{
|
{
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue