mirror of
https://github.com/status-im/nim-stint.git
synced 2025-02-22 20:08:25 +00:00
184e22b659
* Start on allowing compile-time stint. Works for Uint256 * remove and reorder initUintImpl, zero and one procs * Fix assignation of uint64 to 2xuint32, remove unsafeConv and use const for base/radix * assign_least_significant_words shouldn't implicitly capture result * Simplify least significant words macros
114 lines
3.0 KiB
Nim
114 lines
3.0 KiB
Nim
# Stint
|
|
# Copyright 2018 Status Research & Development GmbH
|
|
# Licensed under either of
|
|
#
|
|
# * Apache License, version 2.0, ([LICENSE-APACHE](LICENSE-APACHE) or http://www.apache.org/licenses/LICENSE-2.0)
|
|
# * MIT license ([LICENSE-MIT](LICENSE-MIT) or http://opensource.org/licenses/MIT)
|
|
#
|
|
# at your option. This file may not be copied, modified, or distributed except according to those terms.
|
|
|
|
import ./private/datatypes, macros
|
|
export StUint, UintImpl, uintImpl # TODO remove the need to export UintImpl and this macro
|
|
|
|
template make_unary(op, ResultTy): untyped =
|
|
func `op`*(x: StUint): ResultTy {.inline.} =
|
|
when ResultTy is StUint:
|
|
result.data = op(x.data)
|
|
else:
|
|
op(x.data)
|
|
export op
|
|
|
|
template make_binary(op, ResultTy): untyped =
|
|
func `op`*(x, y: StUint): ResultTy {.inline.} =
|
|
when ResultTy is StUint:
|
|
result.data = op(x.data, y.data)
|
|
else:
|
|
op(x.data, y.data)
|
|
export `op`
|
|
|
|
template make_binary_inplace(op): untyped =
|
|
func `op`*(x: var StUint, y: StUint) {.inline.} =
|
|
op(x.data, y.data)
|
|
export op
|
|
|
|
import ./private/uint_addsub
|
|
|
|
make_binary(`+`, StUint)
|
|
make_binary_inplace(`+=`)
|
|
make_binary(`-`, StUint)
|
|
make_binary_inplace(`-=`)
|
|
|
|
import ./private/uint_mul
|
|
make_binary(`*`, StUint)
|
|
|
|
import ./private/uint_div
|
|
|
|
make_binary(`div`, StUint)
|
|
make_binary(`mod`, StUint)
|
|
func divmod*(x, y: StUint): tuple[quot, rem: StUint] {.inline.} =
|
|
(result.quot.data, result.rem.data) = divmod(x.data, y.data)
|
|
|
|
import ./private/uint_comparison
|
|
|
|
make_binary(`<`, bool)
|
|
make_binary(`<=`, bool)
|
|
make_binary(`==`, bool)
|
|
make_unary(isZero, bool)
|
|
|
|
func isOdd(x: SomeUnsignedInt): bool {.inline.}=
|
|
# internal
|
|
bool(x and 1)
|
|
|
|
func isEven(x: SomeUnsignedInt): bool {.inline.}=
|
|
# internal
|
|
not x.isOdd
|
|
|
|
make_unary(isOdd, bool)
|
|
make_unary(isEven, bool)
|
|
|
|
import ./private/uint_bitwise_ops
|
|
|
|
make_unary(`not`, StUint)
|
|
make_binary(`or`, StUint)
|
|
make_binary(`and`, StUint)
|
|
make_binary(`xor`, StUint)
|
|
func `shr`*(x: StUint, y: SomeInteger): StUint {.inline.} =
|
|
result.data = x.data shr y
|
|
func `shl`*(x: StUint, y: SomeInteger): StUint {.inline.} =
|
|
result.data = x.data shl y
|
|
|
|
import ./private/uint_highlow
|
|
|
|
func high*[bits: static[int]](_: typedesc[Stuint[bits]]): Stuint[bits] {.inline.} =
|
|
result.data = high(type result.data)
|
|
|
|
func low*[bits: static[int]](_: typedesc[Stuint[bits]]): Stuint[bits] {.inline.} =
|
|
result.data = low(type result.data)
|
|
|
|
import ./private/bithacks
|
|
|
|
func countLeadingZeroBits*(x: StUint): int {.inline.} =
|
|
x.data.countLeadingZeroBits
|
|
|
|
import ./private/initialization
|
|
|
|
func zero*[bits: static[int]](T: typedesc[Stuint[bits] or Stint[bits]]): T {.inline.} =
|
|
discard
|
|
|
|
func one*[bits: static[int]](T: typedesc[Stuint[bits]]): T {.inline.} =
|
|
result.data = one(type result.data)
|
|
|
|
import ./private/uint_exp, math
|
|
|
|
func pow*(x: StUint, y: Natural): StUint {.inline.} =
|
|
when x.data is UintImpl:
|
|
result.data = x.data.pow(y)
|
|
else:
|
|
result.data = x.data ^ y
|
|
|
|
func pow*(x: StUint, y: StUint): StUint {.inline.} =
|
|
when x.data is UintImpl:
|
|
result.data = x.data.pow(y.data)
|
|
else:
|
|
result.data = x.data ^ y.data
|