mirror/zig
1
0
Fork 0
mirror of https://github.com/ziglang/zig.git synced 2026-01-20 14:25:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

added mem.containsAtLeast

Browse Source
This commit is contained in:
Sobeston 2021-01-06 23:56:02 +00:00 committed by Andrew Kelley
parent 7e0e27e738
commit 95e66a915b
1 changed files with 34 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

34
lib/std/mem.zig
View File

@ -991,6 +991,40 @@ test "mem.count" {
testing.expect(count(u8, "owowowu", "owowu") == 1);
}
/// Returns true if the haystack contains expected_count or more needles
/// needle.len must be > 0
/// does not count overlapping needles
pub fn containsAtLeast(comptime T: type, haystack: []const T, expected_count: usize, needle: []const T) bool {
assert(needle.len > 0);
if (expected_count == 0) return true;
var i: usize = 0;
var found: usize = 0;
while (indexOfPos(T, haystack, i, needle)) |idx| {
i = idx + needle.len;
found += 1;
if (found == expected_count) return true;
}
return false;
}
test "mem.containsAtLeast" {
testing.expect(containsAtLeast(u8, "aa", 0, "a"));
testing.expect(containsAtLeast(u8, "aa", 1, "a"));
testing.expect(containsAtLeast(u8, "aa", 2, "a"));
testing.expect(!containsAtLeast(u8, "aa", 3, "a"));
testing.expect(containsAtLeast(u8, "radaradar", 1, "radar"));
testing.expect(!containsAtLeast(u8, "radaradar", 2, "radar"));
testing.expect(containsAtLeast(u8, "radarradaradarradar", 3, "radar"));
testing.expect(!containsAtLeast(u8, "radarradaradarradar", 4, "radar"));
testing.expect(containsAtLeast(u8, " radar radar ", 2, "radar"));
testing.expect(!containsAtLeast(u8, " radar radar ", 3, "radar"));
}
/// Reads an integer from memory with size equal to bytes.len.
/// T specifies the return type, which must be large enough to store
/// the result.