1 ! Copyright (C) 2008 John Benediktsson
2 ! See http://factorcode.org/license.txt for BSD license
3 USING: accessors arrays assocs calendar combinators
4 combinators.smart fry generalizations io io.streams.string
5 kernel macros math math.functions math.parser namespaces
6 peg.ebnf present prettyprint quotations sequences strings
7 unicode.case unicode.categories vectors ;
8 FROM: math.parser.private => format-float ;
13 : compose-all ( seq -- quot )
14 [ ] [ compose ] reduce ; inline
16 : fix-sign ( string -- string )
17 dup CHAR: 0 swap index 0 =
18 [ dup 0 swap [ [ CHAR: 0 = not ] keep digit? and ] find-from
19 [ dup 1 - rot dup [ nth ] dip swap
21 { CHAR: - [ [ 1 - ] dip remove-nth "-" prepend ] }
22 { CHAR: + [ [ 1 - ] dip remove-nth "+" prepend ] }
28 : >digits ( string -- digits )
29 [ 0 ] [ string>number ] if-empty ;
31 : format-simple ( x digits string -- string )
32 [ [ >float ] [ number>string ] bi* "%." ] dip
33 surround format-float ;
35 : format-scientific ( x digits -- string ) "e" format-simple ;
37 : format-decimal ( x digits -- string ) "f" format-simple ;
39 ERROR: unknown-printf-directive ;
43 zero = "0" => [[ CHAR: 0 ]]
44 char = "'" (.) => [[ second ]]
46 pad-char = (zero|char)? => [[ CHAR: \s or ]]
47 pad-align = ("-")? => [[ \ pad-tail \ pad-head ? ]]
48 pad-width = ([0-9])* => [[ >digits ]]
49 pad = pad-align pad-char pad-width => [[ <reversed> >quotation dup first 0 = [ drop [ ] ] when ]]
51 sign = ("+")? => [[ [ dup CHAR: - swap index [ "+" prepend ] unless ] [ ] ? ]]
53 width_ = "." ([0-9])* => [[ second >digits '[ _ short head ] ]]
54 width = (width_)? => [[ [ ] or ]]
56 digits_ = "." ([0-9])* => [[ second >digits ]]
57 digits = (digits_)? => [[ 6 or ]]
59 fmt-% = "%" => [[ "%" ]]
60 fmt-c = "c" => [[ [ 1string ] ]]
61 fmt-C = "C" => [[ [ 1string >upper ] ]]
62 fmt-s = "s" => [[ [ present ] ]]
63 fmt-S = "S" => [[ [ present >upper ] ]]
64 fmt-u = "u" => [[ [ unparse ] ]]
65 fmt-d = "d" => [[ [ >integer number>string ] ]]
66 fmt-o = "o" => [[ [ >integer >oct ] ]]
67 fmt-b = "b" => [[ [ >integer >bin ] ]]
68 fmt-e = digits "e" => [[ first '[ _ format-scientific ] ]]
69 fmt-E = digits "E" => [[ first '[ _ format-scientific >upper ] ]]
70 fmt-f = digits "f" => [[ first '[ _ format-decimal ] ]]
71 fmt-x = "x" => [[ [ >hex ] ]]
72 fmt-X = "X" => [[ [ >hex >upper ] ]]
73 unknown = (.)* => [[ unknown-printf-directive ]]
75 strings_ = fmt-c|fmt-C|fmt-s|fmt-S|fmt-u
76 strings = pad width strings_ => [[ <reversed> compose-all ]]
78 numbers_ = fmt-d|fmt-o|fmt-b|fmt-e|fmt-E|fmt-f|fmt-x|fmt-X
79 numbers = sign pad numbers_ => [[ unclip-last prefix compose-all [ fix-sign ] append ]]
81 types = strings|numbers
83 lists = "[%" types ", %]" => [[ second '[ _ map ", " join "{ " prepend " }" append ] ]]
85 assocs = "[%" types ": %" types " %]" => [[ [ second ] [ fourth ] bi '[ unzip [ _ map ] dip _ map zip [ ":" join ] map ", " join "{ " prepend " }" append ] ]]
87 formats = "%" (types|fmt-%|lists|assocs|unknown) => [[ second ]]
89 plain-text = (!("%").)+ => [[ >string ]]
91 text = (formats|plain-text)* => [[ ]]
97 MACRO: printf ( format-string -- )
98 parse-printf [ [ callable? ] count ] keep [
99 dup string? [ 1quotation ] [ [ 1 - ] dip ] if
101 ] map nip [ compose-all ] [ length ] bi '[
102 @ output-stream get [ stream-write ] curry _ napply
105 : sprintf ( format-string -- result )
106 [ printf ] with-string-writer ; inline
108 : vprintf ( seq format-string -- )
109 parse-printf output-stream get '[
111 [ unclip-slice ] dip call( x -- y )
112 ] unless _ stream-write
115 : vsprintf ( seq format-string -- result )
116 [ vprintf ] with-string-writer ; inline
120 : pad-00 ( n -- string ) number>string 2 CHAR: 0 pad-head ; inline
122 : pad-000 ( n -- string ) number>string 3 CHAR: 0 pad-head ; inline
124 : >time ( timestamp -- string )
125 [ hour>> ] [ minute>> ] [ second>> floor ] tri 3array
126 [ pad-00 ] map ":" join ; inline
128 : >date ( timestamp -- string )
129 [ month>> ] [ day>> ] [ year>> ] tri 3array
130 [ pad-00 ] map "/" join ; inline
132 : >datetime ( timestamp -- string )
135 [ day-of-week day-abbreviation3 ]
136 [ month>> month-abbreviation ]
139 [ year>> number>string ]
141 ] output>array " " join ; inline
143 : (week-of-year) ( timestamp day -- n )
144 [ dup clone 1 >>month 1 >>day day-of-week dup ] dip > [ 7 swap - ] when
145 [ day-of-year ] dip 2dup < [ 0 2nip ] [ - 7 / 1 + >fixnum ] if ;
147 : week-of-year-sunday ( timestamp -- n ) 0 (week-of-year) ; inline
149 : week-of-year-monday ( timestamp -- n ) 1 (week-of-year) ; inline
153 fmt-% = "%" => [[ "%" ]]
154 fmt-a = "a" => [[ [ day-of-week day-abbreviation3 ] ]]
155 fmt-A = "A" => [[ [ day-of-week day-name ] ]]
156 fmt-b = "b" => [[ [ month>> month-abbreviation ] ]]
157 fmt-B = "B" => [[ [ month>> month-name ] ]]
158 fmt-c = "c" => [[ [ >datetime ] ]]
159 fmt-d = "d" => [[ [ day>> pad-00 ] ]]
160 fmt-H = "H" => [[ [ hour>> pad-00 ] ]]
161 fmt-I = "I" => [[ [ hour>> dup 12 > [ 12 - ] when pad-00 ] ]]
162 fmt-j = "j" => [[ [ day-of-year pad-000 ] ]]
163 fmt-m = "m" => [[ [ month>> pad-00 ] ]]
164 fmt-M = "M" => [[ [ minute>> pad-00 ] ]]
165 fmt-p = "p" => [[ [ hour>> 12 < "AM" "PM" ? ] ]]
166 fmt-S = "S" => [[ [ second>> floor pad-00 ] ]]
167 fmt-U = "U" => [[ [ week-of-year-sunday pad-00 ] ]]
168 fmt-w = "w" => [[ [ day-of-week number>string ] ]]
169 fmt-W = "W" => [[ [ week-of-year-monday pad-00 ] ]]
170 fmt-x = "x" => [[ [ >date ] ]]
171 fmt-X = "X" => [[ [ >time ] ]]
172 fmt-y = "y" => [[ [ year>> 100 mod pad-00 ] ]]
173 fmt-Y = "Y" => [[ [ year>> number>string ] ]]
174 fmt-Z = "Z" => [[ [ "Not yet implemented" throw ] ]]
175 unknown = (.)* => [[ "Unknown directive" throw ]]
177 formats_ = fmt-%|fmt-a|fmt-A|fmt-b|fmt-B|fmt-c|fmt-d|fmt-H|fmt-I|
178 fmt-j|fmt-m|fmt-M|fmt-p|fmt-S|fmt-U|fmt-w|fmt-W|fmt-x|
179 fmt-X|fmt-y|fmt-Y|fmt-Z|unknown
181 formats = "%" (formats_) => [[ second ]]
183 plain-text = (!("%").)+ => [[ >string ]]
185 text = (formats|plain-text)* => [[ ]]
191 MACRO: strftime ( format-string -- )
196 '[ over @ swap push-all ]
199 SBUF" " clone [ _ cleave drop ] keep "" like