1 ! Copyright (C) 2007, 2011 Slava Pestov.
2 ! See http://factorcode.org/license.txt for BSD license.
3 USING: alien alien.data accessors io.binary math math.bitwise
4 alien.accessors kernel kernel.private sequences
5 sequences.private byte-arrays parser prettyprint.custom fry
10 { length array-capacity read-only }
11 { underlying byte-array read-only } ;
15 : n>byte ( m -- n ) -3 shift ; inline
17 : bit/byte ( n -- bit byte ) [ 7 bitand ] [ n>byte ] bi ; inline
19 : bit-index ( n bit-array -- bit# byte# byte-array )
20 [ >fixnum bit/byte ] [ underlying>> ] bi* ; inline
22 : bits>cells ( m -- n ) 31 + -5 shift ; inline
24 : bits>bytes ( m -- n ) 7 + n>byte ; inline
26 : (set-bits) ( bit-array n -- )
27 [ [ length bits>cells ] keep ] dip swap underlying>>
28 '[ [ _ _ ] dip 4 * set-alien-unsigned-4 ] each-integer ; inline
30 : clean-up ( bit-array -- )
31 ! Zero bits after the end.
32 dup underlying>> empty? [ drop ] [
34 [ underlying>> length 8 * ] [ length ] bi -
35 8 swap - -1 swap shift bitnot
37 [ underlying>> last bitand ]
38 [ underlying>> set-last ]
44 ERROR: bad-array-length n ;
46 : <bit-array> ( n -- bit-array )
47 dup 0 < [ bad-array-length ] when
48 dup bits>bytes <byte-array>
49 bit-array boa ; inline
51 M: bit-array length length>> ; inline
53 M: bit-array nth-unsafe
54 bit-index nth-unsafe swap bit? ; inline
56 :: toggle-bit ( ? n x -- y )
57 x n ? [ set-bit ] [ clear-bit ] if ; inline
59 M: bit-array set-nth-unsafe
60 bit-index [ toggle-bit ] change-nth-unsafe ; inline
62 GENERIC: clear-bits ( bit-array -- )
64 M: bit-array clear-bits 0 (set-bits) ; inline
66 GENERIC: set-bits ( bit-array -- )
68 M: bit-array set-bits -1 (set-bits) ; inline
71 [ length>> ] [ underlying>> clone ] bi bit-array boa ; inline
73 : >bit-array ( seq -- bit-array )
74 T{ bit-array f 0 B{ } } clone-like ; inline
76 M: bit-array like drop dup bit-array? [ >bit-array ] unless ; inline
78 M: bit-array new-sequence drop <bit-array> ; inline
81 over bit-array? [ [ underlying>> ] bi@ sequence= ] [ 2drop f ] if ;
85 [ bits>bytes ] [ underlying>> ] bi*
91 M: bit-array byte-length length bits>bytes ; inline
93 SYNTAX: ?{ \ } [ >bit-array ] parse-literal ;
95 : integer>bit-array ( n -- bit-array )
98 [ dup log2 1 + [ nip ] [ bits>bytes >le ] 2bi bit-array boa ] if ;
100 : bit-array>integer ( bit-array -- n )
103 INSTANCE: bit-array sequence
105 M: bit-array pprint-delims drop \ ?{ \ } ;
106 M: bit-array >pprint-sequence ;
107 M: bit-array pprint* pprint-object ;