diff options
Diffstat (limited to 'lib/std/special/compiler_rt/fixuint.zig')
| -rw-r--r-- | lib/std/special/compiler_rt/fixuint.zig | 51 |
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); + } +} |
