array reflection
This commit is contained in:
+6
-4
@@ -1,6 +1,8 @@
|
||||
Layout :: enum {
|
||||
auto
|
||||
c
|
||||
Layout :: enum { auto c }
|
||||
|
||||
ArrayInfo :: struct {
|
||||
child type
|
||||
len usize
|
||||
}
|
||||
|
||||
FieldInfo :: struct {
|
||||
@@ -28,7 +30,7 @@ TypeInfo :: union(enum) {
|
||||
bool void
|
||||
integer void
|
||||
float void
|
||||
array void
|
||||
array ArrayInfo
|
||||
pointer void
|
||||
slice void
|
||||
range void
|
||||
|
||||
@@ -7,6 +7,22 @@ TestTokenKind :: enum(u8) {
|
||||
}
|
||||
|
||||
TestNames :: alias EnumFieldStruct(TestTokenKind, ?[]u8, some!(null))
|
||||
TestArrayAlias :: alias [3]u16
|
||||
|
||||
hide array_info_matches func($Array, $Child type, $len usize) bool {
|
||||
match typeinfo!(Array) {
|
||||
.array |info|: return info.child == Child and info.len == len
|
||||
else: return false
|
||||
}
|
||||
}
|
||||
|
||||
array_reflection_exposes_child_and_logical_length test {
|
||||
try testing.expect($(array_info_matches([4]i32, i32, 4)))
|
||||
try testing.expect($(array_info_matches([0]bool, bool, 0)))
|
||||
try testing.expect($(array_info_matches(TestArrayAlias, u16, 3)))
|
||||
try testing.expect($(array_info_matches([2]mut i64, i64, 2)))
|
||||
try testing.expect($(array_info_matches([2;0]u8, u8, 2)))
|
||||
}
|
||||
|
||||
enum_field_struct_defaults test {
|
||||
names TestNames = {
|
||||
|
||||
@@ -0,0 +1,109 @@
|
||||
import "@std/mem"
|
||||
|
||||
StaticStringMap func($V type) type {
|
||||
return struct {
|
||||
keys [][]u8
|
||||
values []V
|
||||
len_indexes []u32
|
||||
min_len u32
|
||||
max_len u32
|
||||
}
|
||||
}
|
||||
|
||||
hide Pair func($V type) type {
|
||||
return struct { []u8, V }
|
||||
}
|
||||
|
||||
init func($V type, $N usize, $entries [N]Pair(V)) StaticStringMap(V) {
|
||||
return ${
|
||||
if N > usize(maxval!(u32)) {
|
||||
compile_error!("static string map has too many entries")
|
||||
}
|
||||
|
||||
keys [N]mut []u8 = undefined
|
||||
values [N]mut V = undefined
|
||||
for entries |entry, i| {
|
||||
if entry.0.len > usize(maxval!(u32)) {
|
||||
compile_error!("static string map key is too long")
|
||||
}
|
||||
for (usize(0))..i |prior| {
|
||||
other :: entries[prior].0
|
||||
equal bool = entry.0.len == other.len
|
||||
byte_index usize = 0
|
||||
while equal and byte_index < entry.0.len : byte_index += 1 {
|
||||
equal = entry.0[byte_index] == other[byte_index]
|
||||
}
|
||||
if equal {
|
||||
compile_error!("duplicate static string map key")
|
||||
}
|
||||
}
|
||||
keys[i] = entry.0
|
||||
values[i] = entry.1
|
||||
}
|
||||
|
||||
if N == 0 {
|
||||
len_indexes [0]mut u32 = undefined
|
||||
yield StaticStringMap(V) {
|
||||
keys = keys[..],
|
||||
values = values[..],
|
||||
len_indexes = len_indexes[..],
|
||||
min_len = 0,
|
||||
max_len = 0,
|
||||
}
|
||||
}
|
||||
|
||||
# ponytail: insertion sort is compile-time O(N²); replace if large maps affect builds.
|
||||
i usize = 1
|
||||
while i < N : i += 1 {
|
||||
key :: keys[i]
|
||||
value :: values[i]
|
||||
j usize = i
|
||||
while j > 0 and keys[j - 1].len > key.len : j -= 1 {
|
||||
keys[j] = keys[j - 1]
|
||||
values[j] = values[j - 1]
|
||||
}
|
||||
keys[j] = key
|
||||
values[j] = value
|
||||
}
|
||||
|
||||
min_len u32 :: u32(keys[0].len)
|
||||
max_len u32 :: u32(keys[N - 1].len)
|
||||
len_indexes [usize(max_len) + 1]mut u32 = undefined
|
||||
entry_index usize = 0
|
||||
length usize = 0
|
||||
while length <= usize(max_len) : length += 1 {
|
||||
while entry_index < N and keys[entry_index].len < length : entry_index += 1 {}
|
||||
len_indexes[length] = u32(entry_index)
|
||||
}
|
||||
|
||||
yield StaticStringMap(V) {
|
||||
keys = keys[..],
|
||||
values = values[..],
|
||||
len_indexes = len_indexes[..],
|
||||
min_len = min_len,
|
||||
max_len = max_len,
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
get func($V type, map @StaticStringMap(V), key []u8) ?V {
|
||||
if map.keys.len == 0 or key.len > usize(maxval!(u32)) {
|
||||
return null
|
||||
}
|
||||
length u32 :: u32(key.len)
|
||||
if length < map.min_len or length > map.max_len {
|
||||
return null
|
||||
}
|
||||
index usize = usize(map.len_indexes[usize(length)])
|
||||
while index < map.keys.len {
|
||||
candidate :: map.keys[index]
|
||||
if candidate.len != key.len {
|
||||
return null
|
||||
}
|
||||
if mem.eql(u8, candidate, key) {
|
||||
return map.values[index]
|
||||
}
|
||||
index += 1
|
||||
}
|
||||
return null
|
||||
}
|
||||
@@ -0,0 +1,36 @@
|
||||
import "@std/testing"
|
||||
|
||||
TokenKind :: enum {
|
||||
keyword_if
|
||||
keyword_else
|
||||
keyword_for
|
||||
keyword_return
|
||||
}
|
||||
|
||||
keywords StaticStringMap(TokenKind) = init([
|
||||
{"return", .keyword_return},
|
||||
{"if", .keyword_if},
|
||||
{"for", .keyword_for},
|
||||
{"else", .keyword_else},
|
||||
])
|
||||
|
||||
handles_length_bucket_lookups test {
|
||||
try testing.expect(keywords.keys.len == 4)
|
||||
try testing.expect(keywords.values.len == keywords.keys.len)
|
||||
try testing.expect(keywords.len_indexes.len == 7)
|
||||
try testing.expect_equal(some!(TokenKind.keyword_if), get(&keywords, "if"))
|
||||
try testing.expect_equal(some!(TokenKind.keyword_else), get(&keywords, "else"))
|
||||
try testing.expect_equal(some!(TokenKind.keyword_for), get(&keywords, "for"))
|
||||
try testing.expect_equal(some!(TokenKind.keyword_return), get(&keywords, "return"))
|
||||
try testing.expect_equal(null, get(&keywords, "no"))
|
||||
try testing.expect_equal(null, get(&keywords, "four"))
|
||||
try testing.expect_equal(null, get(&keywords, "longer-than-any-key"))
|
||||
}
|
||||
|
||||
handles_empty_maps test {
|
||||
empty StaticStringMap(TokenKind) = init([])
|
||||
try testing.expect(empty.keys.len == 0)
|
||||
try testing.expect(empty.values.len == 0)
|
||||
try testing.expect(empty.len_indexes.len == 0)
|
||||
try testing.expect_equal(null, get(&empty, "if"))
|
||||
}
|
||||
@@ -1,7 +1,9 @@
|
||||
import "io"
|
||||
import "enums"
|
||||
import "arraylist"
|
||||
import "static_string_map"
|
||||
|
||||
Io :: alias io.Io
|
||||
ArrayList :: alias arraylist.ArrayList
|
||||
EnumMap :: alias enums.EnumMap
|
||||
StaticStringMap :: alias static_string_map.StaticStringMap
|
||||
|
||||
Reference in New Issue
Block a user