aboutsummaryrefslogtreecommitdiff
path: root/lib/std/special/compiler_rt/fixuint.zig
diff options
context:
space:
mode:
Diffstat (limited to 'lib/std/special/compiler_rt/fixuint.zig')
-rw-r--r--lib/std/special/compiler_rt/fixuint.zig51
1 files changed, 51 insertions, 0 deletions
diff --git a/lib/std/special/compiler_rt/fixuint.zig b/lib/std/special/compiler_rt/fixuint.zig
new file mode 100644
index 0000000000..55a113b368
--- /dev/null
+++ b/lib/std/special/compiler_rt/fixuint.zig
@@ -0,0 +1,51 @@
+const is_test = @import("builtin").is_test;
+const Log2Int = @import("std").math.Log2Int;
+
+pub fn fixuint(comptime fp_t: type, comptime fixuint_t: type, a: fp_t) fixuint_t {
+ @setRuntimeSafety(is_test);
+
+ const rep_t = switch (fp_t) {
+ f32 => u32,
+ f64 => u64,
+ f128 => u128,
+ else => unreachable,
+ };
+ const srep_t = @IntType(true, rep_t.bit_count);
+ const significandBits = switch (fp_t) {
+ f32 => 23,
+ f64 => 52,
+ f128 => 112,
+ else => unreachable,
+ };
+ const typeWidth = rep_t.bit_count;
+ const exponentBits = (typeWidth - significandBits - 1);
+ const signBit = (rep_t(1) << (significandBits + exponentBits));
+ const maxExponent = ((1 << exponentBits) - 1);
+ const exponentBias = (maxExponent >> 1);
+
+ const implicitBit = (rep_t(1) << significandBits);
+ const significandMask = (implicitBit - 1);
+
+ // Break a into sign, exponent, significand
+ const aRep: rep_t = @bitCast(rep_t, a);
+ const absMask = signBit - 1;
+ const aAbs: rep_t = aRep & absMask;
+
+ const sign = if ((aRep & signBit) != 0) i32(-1) else i32(1);
+ const exponent = @intCast(i32, aAbs >> significandBits) - exponentBias;
+ const significand: rep_t = (aAbs & significandMask) | implicitBit;
+
+ // If either the value or the exponent is negative, the result is zero.
+ if (sign == -1 or exponent < 0) return 0;
+
+ // If the value is too large for the integer type, saturate.
+ if (@intCast(c_uint, exponent) >= fixuint_t.bit_count) return ~fixuint_t(0);
+
+ // If 0 <= exponent < significandBits, right shift to get the result.
+ // Otherwise, shift left.
+ if (exponent < significandBits) {
+ return @intCast(fixuint_t, significand >> @intCast(Log2Int(rep_t), significandBits - exponent));
+ } else {
+ return @intCast(fixuint_t, significand) << @intCast(Log2Int(fixuint_t), exponent - significandBits);
+ }
+}