factor/library/math/integer.factor

116 lines
2.4 KiB
Factor

! Copyright (C) 2004, 2005 Slava Pestov.
! See http://factor.sf.net/license.txt for BSD license.
IN: math
USING: errors generic kernel kernel-internals math sequences
sequences-internals ;
UNION: integer fixnum bignum ;
: even? ( n -- ? ) 1 bitand 0 = ;
: odd? ( n -- ? ) 1 bitand 1 = ;
: (gcd) ( b a y x -- a d )
dup 0 number= [
drop nip
] [
tuck /mod >r pick * swap >r swapd - r> r> (gcd)
] if ; inline
: gcd ( x y -- a d )
#! Compute the greatest common divisor d and multiplier a
#! such that a*x=d mod y.
swap 0 1 2swap (gcd) abs ; foldable
: (next-power-of-2) ( i n -- n )
2dup >= [
drop
] [
>r 1 shift r> (next-power-of-2)
] if ;
: next-power-of-2 ( n -- n )
1 swap (next-power-of-2) ;
: first-bignum ( -- n )
1 cell 8 * tag-bits - 1- shift ; inline
: most-positive-fixnum ( -- n )
first-bignum 1- >fixnum ; inline
: most-negative-fixnum ( -- n )
first-bignum neg >fixnum ; inline
IN: math-internals
: fraction> ( a b -- a/b )
dup 1 number= [ drop ] [ (fraction>) ] if ; inline
: division-by-zero ( x y -- ) "Division by zero" throw ;
M: integer / ( x y -- x/y )
dup 0 number= [
division-by-zero
] [
dup 0 < [ [ neg ] 2apply ] when
2dup gcd nip tuck /i >r /i r> fraction>
] if ;
M: fixnum number=
#! Fixnums are immediate values, so equality testing is
#! trivial.
eq? ;
M: fixnum < fixnum< ;
M: fixnum <= fixnum<= ;
M: fixnum > fixnum> ;
M: fixnum >= fixnum>= ;
M: fixnum + fixnum+ ;
M: fixnum - fixnum- ;
M: fixnum * fixnum* ;
M: fixnum /i fixnum/i ;
M: fixnum /f fixnum/f ;
M: fixnum mod fixnum-mod ;
M: fixnum /mod fixnum/mod ;
M: fixnum 1+ 1 fixnum+ ;
M: fixnum 1- 1 fixnum- ;
M: fixnum bitand fixnum-bitand ;
M: fixnum bitor fixnum-bitor ;
M: fixnum bitxor fixnum-bitxor ;
M: fixnum shift fixnum-shift ;
M: fixnum bitnot fixnum-bitnot ;
M: bignum number= bignum= ;
M: bignum < bignum< ;
M: bignum <= bignum<= ;
M: bignum > bignum> ;
M: bignum >= bignum>= ;
M: bignum + bignum+ ;
M: bignum - bignum- ;
M: bignum * bignum* ;
M: bignum /i bignum/i ;
M: bignum /f bignum/f ;
M: bignum mod bignum-mod ;
M: bignum /mod bignum/mod ;
M: bignum 1+ 1 >bignum bignum+ ;
M: bignum 1- 1 >bignum bignum- ;
M: bignum bitand bignum-bitand ;
M: bignum bitor bignum-bitor ;
M: bignum bitxor bignum-bitxor ;
M: bignum shift bignum-shift ;
M: bignum bitnot bignum-bitnot ;
M: integer truncate ;
M: integer floor ;
M: integer ceiling ;