ed/Odin
1
0
Fork 0
mirror of https://github.com/Ed94/Odin.git synced 2026-08-06 15:48:51 +00:00
Code Issues Packages Projects Releases Wiki Activity

math/rand: support non-contiguous enums in choice_enum

Browse Source
This commit is contained in:
Laytan Laats
2024-10-25 20:41:57 +02:00
parent bcf2b93c6e
commit bb4fc9979a
1 changed files with 15 additions and 15 deletions
Download Patch File Download Diff File Expand all files Collapse all files
core/math/rand/rand.odin
+6 -6
View File
@@ -670,12 +670,8 @@ choice :: proc(array: $T/[]$E, gen := context.random_generator) -> (res: E) {
@(require_results) @(require_results)
choice_enum :: proc($T: typeid, gen := context.random_generator) -> T choice_enum :: proc($T: typeid, gen := context.random_generator) -> T where intrinsics.type_is_enum(T) {
where when size_of(T) <= 8 && len(T) == cap(T) {
intrinsics.type_is_enum(T),
size_of(T) <= 8,
len(T) == cap(T) /* Only allow contiguous enum types */ \
{
when intrinsics.type_is_unsigned(intrinsics.type_core_type(T)) && when intrinsics.type_is_unsigned(intrinsics.type_core_type(T)) &&
u64(max(T)) > u64(max(i64)) { u64(max(T)) > u64(max(i64)) {
i := uint64(gen) % u64(len(T)) i := uint64(gen) % u64(len(T))
@@ -686,4 +682,8 @@ choice_enum :: proc($T: typeid, gen := context.random_generator) -> T
i += i64(min(T)) i += i64(min(T))
return T(i) return T(i)
} }
} else {
values := runtime.type_info_base(type_info_of(T)).variant.(runtime.Type_Info_Enum).values
return T(choice(values))
}
} }