factor/library/math/parse-numbers.factor

76 lines
1.9 KiB
Factor

! Copyright (C) 2004, 2005 Slava Pestov.
! See http://factor.sf.net/license.txt for BSD license.
IN: math
USING: errors generic kernel math-internals namespaces sequences
strings ;
! Number parsing
: not-a-number "Not a number" throw ;
GENERIC: digit> ( ch -- n )
M: digit digit> CHAR: 0 - ;
M: letter digit> CHAR: a - 10 + ;
M: LETTER digit> CHAR: A - 10 + ;
M: object digit> not-a-number ;
: digit+ ( num digit base -- num )
2dup < [ rot * + ] [ not-a-number ] if ;
: (base>) ( base str -- num )
dup empty? [
not-a-number
] [
0 [ digit> pick digit+ ] reduce nip
] if ;
: base> ( str base -- num )
#! Convert a string to an integer. Throw an error if
#! conversion fails.
swap "-" ?head >r (base>) r> [ neg ] when ;
: string>ratio ( "a/b" -- a/b )
"/" split1 [ 10 base> ] 2apply / ;
: string>number ( string -- n )
{
{ [ CHAR: / over member? ] [ string>ratio ] }
{ [ CHAR: . over member? ] [ string>float ] }
{ [ t ] [ 10 base> ] }
} cond ;
: bin> 2 base> ;
: oct> 8 base> ;
: hex> 16 base> ;
: >digit ( n -- ch )
dup 10 < [ CHAR: 0 + ] [ 10 - CHAR: a + ] if ;
: integer, ( num radix -- )
dup >r /mod >digit , dup 0 >
[ r> integer, ] [ r> 2drop ] if ;
: >base ( num radix -- string )
#! Convert a number to a string in a certain base.
[
over 0 < [
swap neg swap integer, CHAR: - ,
] [
integer,
] if
] "" make reverse ;
: >bin ( num -- string ) 2 >base ;
: >oct ( num -- string ) 8 >base ;
: >hex ( num -- string ) 16 >base ;
M: integer number>string ( obj -- str ) 10 >base ;
M: ratio number>string ( num -- str )
[ dup numerator # CHAR: / , denominator # ] "" make ;
M: float number>string ( float -- str )
#! This is terrible. Will go away when we do our own float
#! output.
float>string CHAR: . over member? [ ".0" append ] unless ;