paoda
/
bit-string
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Compare commits

base: paoda:dbe6bd50cd8eb20a40df658a3319d77225604fd0
Branches Tags
paoda:main
..
compare: paoda:0af0295acf508dac297b99e6b95e607282b56c1b
Branches Tags
paoda:main

No commits in common. "dbe6bd50cd8eb20a40df658a3319d77225604fd0" and "0af0295acf508dac297b99e6b95e607282b56c1b" have entirely different histories.
dbe6bd50cd ... 0af0295acf

2 changed files with 33 additions and 73 deletions
Show Stats Expand all files Collapse all files

10
build.zig
View File

@ -46,14 +46,4 @@ pub fn build(b: *std.Build) void {
// This will evaluate the `test` step rather than the default, which is "install". // This will evaluate the `test` step rather than the default, which is "install".
const test_step = b.step("test", "Run library tests"); const test_step = b.step("test", "Run library tests");
test_step.dependOn(&run_main_tests.step); test_step.dependOn(&run_main_tests.step);
// Build Documentation
const doc = b.addInstallDirectory(.{
.source_dir = lib.getEmittedDocs(),
.install_dir = .prefix,
.install_subdir = "docs",
});
const doc_step = b.step("doc", "Generate documentation");
doc_step.dependOn(&doc.step);
} }

96
src/lib.zig
View File

@ -5,37 +5,36 @@
//! const std = @import("std"); //! const std = @import("std");
//! //!
//! test "doc test" { //! test "doc test" {
//! const value: u8 = 0b10001011; //! const value: u8 = 0b10001011;
//! //!
//! try std.testing.expectEqual(true, match("1000_1011", value)); //! try std.testing.expectEqual(true, match("10001011", value));
//! try std.testing.expectEqual(false, match("11111011", value)); //! try std.testing.expectEqual(false, match("11111011", value));
//! try std.testing.expectEqual(true, match("1---1011", value)); //! try std.testing.expectEqual(true, match("1---1011", value));
//! //!
//! { //! {
//! const ret = extract("1000aaaa", value); //! const ret = extract("1000aaaa", value);
//! try std.testing.expectEqual(@as(u4, 0b1011), ret.a); //! try std.testing.expectEqual(@as(u4, 0b1011), ret.a);
//! } //! }
//! { //! {
//! const ret = extract("1aaa1aaa", value); //! const ret = extract("1aaa1aaa", value);
//! try std.testing.expectEqual(@as(u6, 0b000011), ret.a); //! try std.testing.expectEqual(@as(u6, 0b000011), ret.a);
//! } //! }
//! { //! {
//! const ret = extract("1---abcd", value); //! const ret = extract("1---abcd", value);
//! try std.testing.expectEqual(@as(u3, 0b1), ret.a); //! try std.testing.expectEqual(@as(u3, 0b1), ret.a);
//! try std.testing.expectEqual(@as(u3, 0b0), ret.b); //! try std.testing.expectEqual(@as(u3, 0b0), ret.b);
//! try std.testing.expectEqual(@as(u3, 0b1), ret.c); //! try std.testing.expectEqual(@as(u3, 0b1), ret.c);
//! try std.testing.expectEqual(@as(u3, 0b1), ret.d); //! try std.testing.expectEqual(@as(u3, 0b1), ret.d);
//! } //! }
//! } //! }
//! ``` //! ```
//! ## Syntax //! ## Syntax
//! | Token | Meaning | Description //! | Token | Meaning | Description
//! | ------- | --------- | ----------- //! | ------- | --------- | -----------
//! | `0` | Clear bit | In the equivalent position, the value's bit must be cleared. //! | `0` | Unset bit | In the equivalent position, the value's bit must be set.
//! | `1` | Set bit | In the equivalent position, the value's bit must be set. //! | `1` | Set bit | In the equivalent position, the value's bit must be set.
//! | `a..=z` | Variable | Given the 4-bit bit string, `"1aa0"`, the value `0b1010` would produce the variable `a` with the value `0b01` //! | `a..=z` | Variable | Given the 4-bit bit string, `"1aa0"`, the value `0b1010` would produce the variable `a` with the value `0b01`
//! | `-` | Ignored | In the equivalent position, the value's bit does not matter. //! | `-` | Ignored | In the equivalent position, the value's bit does not matter.
//! | `_` | Ignored* | Underscores are completely ignored during parsing, use to make bit strings easier to read e.g. `1111_1111`
//! //!
//! ## Notes //! ## Notes
//! - This library does the majority of it's work at `comptime`. Due to this, you cannot create strings to match against at runtime. //! - This library does the majority of it's work at `comptime`. Due to this, you cannot create strings to match against at runtime.
@ -50,10 +49,8 @@ const Log2Int = std.math.Log2Int;
/// ```zig /// ```zig
/// match("1100", @as(u4, 0b1100)) // true /// match("1100", @as(u4, 0b1100)) // true
/// match("1100", @as(u4, 0b1110)) // false /// match("1100", @as(u4, 0b1110)) // false
///
/// match("1--0", @as(u4, 0b1010)) // true /// match("1--0", @as(u4, 0b1010)) // true
/// match("1ab0", @as(u4, 0b1010)) // true /// match("1ab0", @as(u4, 0b1010)) // true
/// match("11_00", @as(u4, 0b1100)) // true
/// ``` /// ```
pub fn match(comptime bit_string: []const u8, value: anytype) bool { pub fn match(comptime bit_string: []const u8, value: anytype) bool {
@setEvalBranchQuota(std.math.maxInt(u32)); // FIXME: bad practice @setEvalBranchQuota(std.math.maxInt(u32)); // FIXME: bad practice
@ -66,14 +63,12 @@ pub fn match(comptime bit_string: []const u8, value: anytype) bool {
var set: ValT = 0; var set: ValT = 0;
var clr: ValT = 0; var clr: ValT = 0;
var offset = 0;
// FIXME: I linear search like this 5 times across the entie lib. Consider structuring this like a regex lib (compiling a match) // FIXME: I linear search bit_string 4 separate times. Consider doing a single search and compromizing on memory + stateless API? (imagine a "regex compile"-like API)
for (bit_string, 0..) |char, i| { for (bit_string, 0..) |char, i| {
switch (char) { switch (char) {
'0' => clr |= @as(ValT, 1) << @intCast((bit_count - 1 - (i - offset))), '0' => clr |= @as(ValT, 1) << @intCast(bit_count - 1 - i),
'1' => set |= @as(ValT, 1) << @intCast((bit_count - 1 - (i - offset))), '1' => set |= @as(ValT, 1) << @intCast(bit_count - 1 - i),
'_' => offset += 1,
'a'...'z', '-' => continue, 'a'...'z', '-' => continue,
else => @compileError("'" ++ [_]u8{char} ++ "' was unexpected when parsing bitstring"), else => @compileError("'" ++ [_]u8{char} ++ "' was unexpected when parsing bitstring"),
} }
@ -90,12 +85,10 @@ pub fn match(comptime bit_string: []const u8, value: anytype) bool {
test "match" { test "match" {
// doc tests // doc tests
try std.testing.expectEqual(true, match("1100", @as(u4, 0b1100))); try std.testing.expectEqual(true, match("1100", @as(u4, 0b1100))); // true
try std.testing.expectEqual(false, match("1100", @as(u4, 0b1110))); try std.testing.expectEqual(false, match("1100", @as(u4, 0b1110))); // false
try std.testing.expectEqual(true, match("1--0", @as(u4, 0b1010))); // true
try std.testing.expectEqual(true, match("1--0", @as(u4, 0b1010))); try std.testing.expectEqual(true, match("1ab0", @as(u4, 0b1010))); // true
try std.testing.expectEqual(true, match("1ab0", @as(u4, 0b1010)));
try std.testing.expectEqual(true, match("11_00", @as(u4, 0b1100)));
// other tests // other tests
try std.testing.expectEqual(true, match("11111111", @as(u8, 0b11111111))); try std.testing.expectEqual(true, match("11111111", @as(u8, 0b11111111)));
@ -105,10 +98,6 @@ test "match" {
try std.testing.expectEqual(true, match("aaa---11", @as(u8, 0b01011111))); try std.testing.expectEqual(true, match("aaa---11", @as(u8, 0b01011111)));
try std.testing.expectEqual(true, match("1a0b1c0d", @as(u8, 0b10011101))); try std.testing.expectEqual(true, match("1a0b1c0d", @as(u8, 0b10011101)));
try std.testing.expectEqual(false, match("aaa---11", @as(u8, 0b01011110))); try std.testing.expectEqual(false, match("aaa---11", @as(u8, 0b01011110)));
try std.testing.expectEqual(true, match("1111_1111", @as(u8, 0b11111111)));
try std.testing.expectEqual(true, match("________11111111", @as(u8, 0b11111111)));
try std.testing.expectEqual(true, match("11111111________", @as(u8, 0b11111111)));
} }
/// Extracts the variables (defined in the bit string) from a value. /// Extracts the variables (defined in the bit string) from a value.
@ -119,7 +108,6 @@ test "match" {
/// const ret = extract("abcd", @as(u4, 0b1001)); // ret.a == 0b1, ret.b == 0b0, ret.c == 0b0, ret.d == 0b1 /// const ret = extract("abcd", @as(u4, 0b1001)); // ret.a == 0b1, ret.b == 0b0, ret.c == 0b0, ret.d == 0b1
/// const ret = extract("a0ab", @as(u4, 0b1001)); // ret.a == 0b10, ret.b == 0b1 /// const ret = extract("a0ab", @as(u4, 0b1001)); // ret.a == 0b10, ret.b == 0b1
/// const ret = extract("-a-a", @as(u4, 0b1001)); // ret.a == 0b01 /// const ret = extract("-a-a", @as(u4, 0b1001)); // ret.a == 0b01
/// const ret = extract("aa_aa", @as(u4, 0b1001)); // ret.a == 0b1001
/// ``` /// ```
/// ///
/// Note: In Debug and ReleaseSafe builds, there's a runtime assert that /// Note: In Debug and ReleaseSafe builds, there's a runtime assert that
@ -140,21 +128,13 @@ pub fn extract(comptime bit_string: []const u8, value: anytype) Bitfield(bit_str
inline for (@typeInfo(ReturnT).Struct.fields) |field| { inline for (@typeInfo(ReturnT).Struct.fields) |field| {
@field(ret, field.name) = blk: { @field(ret, field.name) = blk: {
var masked_val: ValT = 0; var masked_val: ValT = 0;
var offset: usize = 0; // FIXME(URGENT): this whole block should be happening at comptime...
for (bit_string, 0..) |char, i| { for (bit_string, 0..) |char, i| {
const rev = @typeInfo(ValT).Int.bits - 1 - (i - offset); const rev = @typeInfo(ValT).Int.bits - 1 - i;
if (char == field.name[0]) masked_val |= @as(ValT, 1) << @intCast(rev); // no penalty
switch (char) {
'_' => offset += 1,
else => if (char == field.name[0]) {
masked_val |= @as(ValT, 1) << @intCast(rev); // no penalty
},
}
} }
// TODO: decide at compile time if we're calling the 32-bit or 64-bit version of `PEXT` // TODO: decide at compile time if we're calling the 32-bit or 64-bit version of `PEXT`
// TODO: rewrite this invariant thing to account for underscores
// invariant: the bit count in the field we're writing to and the // invariant: the bit count in the field we're writing to and the
// # of bits we happened to find in this linear search are identical // # of bits we happened to find in this linear search are identical
@ -195,10 +175,6 @@ test "extract" {
const ret = extract("-a-a", @as(u4, 0b1001)); const ret = extract("-a-a", @as(u4, 0b1001));
try std.testing.expectEqual(@as(u2, 0b01), ret.a); try std.testing.expectEqual(@as(u2, 0b01), ret.a);
} }
{
const ret = extract("aa_aa", @as(u4, 0b1001));
try std.testing.expectEqual(@as(u4, 0b1001), ret.a);
}
// other tests // other tests
{ {
@ -276,8 +252,8 @@ pub fn Bitfield(comptime bit_string: []const u8) type {
things[pos].bits += 1; things[pos].bits += 1;
things[pos].char = c; things[pos].char = c;
}, },
'1', '0', '-', '_' => continue, '1', '0', '-' => continue,
else => @compileError("unexpected char '" ++ [_]u8{char} ++ "' when parsing bit string"), else => @compileError("error when parsing bitset string"),
} }
} }
@ -313,13 +289,7 @@ fn verify(comptime T: type, comptime bit_string: []const u8) void {
std.debug.assert(info != .ComptimeInt); std.debug.assert(info != .ComptimeInt);
std.debug.assert(info.Int.signedness == .unsigned); std.debug.assert(info.Int.signedness == .unsigned);
std.debug.assert(info.Int.bits <= 64); // x86 PEXT u32 and u64 operands only std.debug.assert(info.Int.bits <= 64); // x86 PEXT u32 and u64 operands only
std.debug.assert(bit_string.len == info.Int.bits); // TODO: Support Underscores?
var underscore_count = 0;
for (bit_string) |c| {
if (c == '_') underscore_count += 1;
}
std.debug.assert((bit_string.len - underscore_count) == info.Int.bits);
} }
const pext = struct { const pext = struct {
@ -404,7 +374,7 @@ const pext = struct {
test "doc test" { test "doc test" {
const value: u8 = 0b10001011; const value: u8 = 0b10001011;
try std.testing.expectEqual(true, match("1000_1011", value)); try std.testing.expectEqual(true, match("10001011", value));
try std.testing.expectEqual(false, match("11111011", value)); try std.testing.expectEqual(false, match("11111011", value));
try std.testing.expectEqual(true, match("1---1011", value)); try std.testing.expectEqual(true, match("1---1011", value));