mirrors/Odin
1
0
Fork 0
mirror of https://github.com/odin-lang/Odin.git synced 2026-02-12 06:18:39 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add intrinsics.count_trailing_ones and intrinsics.count_leading_ones

Browse Source
This commit is contained in:
gingerBill
2026-02-02 14:28:16 +00:00
parent 80c9482774
commit bb7496a2fc
5 changed files with 47 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
base/intrinsics/intrinsics.odin
View File

@@ -38,6 +38,8 @@ count_ones :: proc(x: $T) -> T where type_is_integer(T) || type_is_sim
count_zeros :: proc(x: $T) -> T where type_is_integer(T) || type_is_simd_vector(T) ---
count_trailing_zeros :: proc(x: $T) -> T where type_is_integer(T) || type_is_simd_vector(T) ---
count_leading_zeros :: proc(x: $T) -> T where type_is_integer(T) || type_is_simd_vector(T) ---
count_trailing_ones :: proc(x: $T) -> T where type_is_integer(T) || type_is_simd_vector(T) ---
count_leading_ones :: proc(x: $T) -> T where type_is_integer(T) || type_is_simd_vector(T) ---
reverse_bits :: proc(x: $T) -> T where type_is_integer(T) || type_is_simd_vector(T) ---
byte_swap :: proc(x: $T) -> T where type_is_integer(T) || type_is_float(T) ---

23
src/check_builtin.cpp
View File

@@ -5204,6 +5204,8 @@ gb_internal bool check_builtin_procedure(CheckerContext *c, Operand *operand, As
case BuiltinProc_count_zeros:
case BuiltinProc_count_trailing_zeros:
case BuiltinProc_count_leading_zeros:
case BuiltinProc_count_trailing_ones:
case BuiltinProc_count_leading_ones:
case BuiltinProc_reverse_bits:
{
Operand x = {};
@@ -5301,6 +5303,27 @@ gb_internal bool check_builtin_procedure(CheckerContext *c, Operand *operand, As
v += 1;
}
break;
case BuiltinProc_count_trailing_ones:
for (u64 i = 0; i < bit_size; i++) {
u8 b = cast(u8)(i & 7);
u8 j = cast(u8)(i >> 3);
if ((rop[j] & (1 << b)) == 0) {
break;
}
v += 1;
}
break;
case BuiltinProc_count_leading_ones:
for (u64 i = bit_size-1; i < bit_size; i--) {
u8 b = cast(u8)(i & 7);
u8 j = cast(u8)(i >> 3);
if ((rop[j] & (1 << b)) == 0) {
break;
}
v += 1;
}
break;
}

4
src/checker_builtin_procs.hpp
View File

@@ -74,6 +74,8 @@ enum BuiltinProcId {
BuiltinProc_count_zeros,
BuiltinProc_count_trailing_zeros,
BuiltinProc_count_leading_zeros,
BuiltinProc_count_trailing_ones,
BuiltinProc_count_leading_ones,
BuiltinProc_reverse_bits,
BuiltinProc_byte_swap,
@@ -453,6 +455,8 @@ gb_global BuiltinProc builtin_procs[BuiltinProc_COUNT] = {
{STR_LIT("count_zeros"), 1, false, Expr_Expr, BuiltinProcPkg_intrinsics},
{STR_LIT("count_trailing_zeros"), 1, false, Expr_Expr, BuiltinProcPkg_intrinsics},
{STR_LIT("count_leading_zeros"), 1, false, Expr_Expr, BuiltinProcPkg_intrinsics},
{STR_LIT("count_trailing_ones"), 1, false, Expr_Expr, BuiltinProcPkg_intrinsics},
{STR_LIT("count_leading_ones"), 1, false, Expr_Expr, BuiltinProcPkg_intrinsics},
{STR_LIT("reverse_bits"), 1, false, Expr_Expr, BuiltinProcPkg_intrinsics},
{STR_LIT("byte_swap"), 1, false, Expr_Expr, BuiltinProcPkg_intrinsics},

6
src/llvm_backend_proc.cpp
View File

@@ -2850,6 +2850,12 @@ gb_internal lbValue lb_build_builtin_proc(lbProcedure *p, Ast *expr, TypeAndValu
case BuiltinProc_count_leading_zeros:
return lb_emit_count_leading_zeros(p, lb_build_expr(p, ce->args[0]), tv.type);
case BuiltinProc_count_trailing_ones:
return lb_emit_count_trailing_ones(p, lb_build_expr(p, ce->args[0]), tv.type);
case BuiltinProc_count_leading_ones:
return lb_emit_count_leading_ones(p, lb_build_expr(p, ce->args[0]), tv.type);
case BuiltinProc_count_ones:
return lb_emit_count_ones(p, lb_build_expr(p, ce->args[0]), tv.type);
case BuiltinProc_count_zeros:

12
src/llvm_backend_utility.cpp
View File

@@ -648,6 +648,18 @@ gb_internal lbValue lb_emit_count_leading_zeros(lbProcedure *p, lbValue x, Type
return res;
}
gb_internal lbValue lb_emit_unary_arith(lbProcedure *p, TokenKind op, lbValue x, Type *type);
gb_internal lbValue lb_emit_count_trailing_ones(lbProcedure *p, lbValue x, Type *type) {
lbValue z = lb_emit_unary_arith(p, Token_Xor, x, type);
return lb_emit_count_trailing_zeros(p, z, type);
}
gb_internal lbValue lb_emit_count_leading_ones(lbProcedure *p, lbValue x, Type *type) {
lbValue z = lb_emit_unary_arith(p, Token_Xor, x, type);
return lb_emit_count_leading_zeros(p, z, type);
}
gb_internal lbValue lb_emit_reverse_bits(lbProcedure *p, lbValue x, Type *type) {