1 ! Copyright (C) 2007, 2008 Slava Pestov.
2 ! See http://factorcode.org/license.txt for BSD license.
3 USING: alien.c-types alien.data accessors math alien.accessors kernel
4 kernel.private sequences sequences.private byte-arrays
5 parser prettyprint.custom fry ;
9 { length array-capacity read-only }
10 { underlying byte-array read-only } ;
14 : n>byte ( m -- n ) -3 shift ; inline
16 : byte/bit ( n alien -- byte bit )
17 over n>byte alien-unsigned-1 swap 7 bitand ; inline
19 : set-bit ( ? byte bit -- byte )
20 2^ rot [ bitor ] [ bitnot bitand ] if ; 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 '[ 2 shift [ _ _ ] dip set-alien-unsigned-4 ] each ; 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 : <bit-array> ( n -- bit-array )
45 dup bits>bytes <byte-array> bit-array boa ; inline
47 M: bit-array length length>> ; inline
49 M: bit-array nth-unsafe
50 [ >fixnum ] [ underlying>> ] bi* byte/bit bit? ; inline
52 M: bit-array set-nth-unsafe
53 [ >fixnum ] [ underlying>> ] bi*
54 [ byte/bit set-bit ] 2keep
55 swap n>byte set-alien-unsigned-1 ; inline
57 GENERIC: clear-bits ( bit-array -- )
59 M: bit-array clear-bits 0 (set-bits) ; inline
61 GENERIC: set-bits ( bit-array -- )
63 M: bit-array set-bits -1 (set-bits) ; inline
66 [ length>> ] [ underlying>> clone ] bi bit-array boa ; inline
68 : >bit-array ( seq -- bit-array )
69 T{ bit-array f 0 B{ } } clone-like ; inline
71 M: bit-array like drop dup bit-array? [ >bit-array ] unless ; inline
73 M: bit-array new-sequence drop <bit-array> ; inline
76 over bit-array? [ [ underlying>> ] bi@ sequence= ] [ 2drop f ] if ;
80 [ bits>bytes ] [ underlying>> ] bi*
86 M: bit-array byte-length length 7 + -3 shift ; inline
88 SYNTAX: ?{ \ } [ >bit-array ] parse-literal ;
90 : integer>bit-array ( n -- bit-array )
94 [ log2 1 + <bit-array> 0 ] keep
96 [ pick underlying>> pick set-alien-unsigned-1 ] keep
97 [ 1 + ] [ -8 shift ] bi*
101 : bit-array>integer ( bit-array -- n )
102 0 swap underlying>> dup length <reversed> [
103 alien-unsigned-1 swap 8 shift bitor
106 INSTANCE: bit-array sequence
108 M: bit-array pprint-delims drop \ ?{ \ } ;
109 M: bit-array >pprint-sequence ;
110 M: bit-array pprint* pprint-object ;