ed/Odin
1
0
Fork 0
mirror of https://github.com/Ed94/Odin.git synced 2026-08-05 15:18:49 +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
+14 -14
View File
@@ -670,20 +670,20 @@ 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), when intrinsics.type_is_unsigned(intrinsics.type_core_type(T)) &&
size_of(T) <= 8, u64(max(T)) > u64(max(i64)) {
len(T) == cap(T) /* Only allow contiguous enum types */ \ i := uint64(gen) % u64(len(T))
{ i += u64(min(T))
when intrinsics.type_is_unsigned(intrinsics.type_core_type(T)) && return T(i)
u64(max(T)) > u64(max(i64)) { } else {
i := uint64(gen) % u64(len(T)) i := int63_max(i64(len(T)), gen)
i += u64(min(T)) i += i64(min(T))
return T(i) return T(i)
}
} else { } else {
i := int63_max(i64(len(T)), gen) values := runtime.type_info_base(type_info_of(T)).variant.(runtime.Type_Info_Enum).values
i += i64(min(T)) return T(choice(values))
return T(i)
} }
} }