72 lines
		
	
	
		
			1.8 KiB
		
	
	
	
		
			Factor
		
	
	
			
		
		
	
	
			72 lines
		
	
	
		
			1.8 KiB
		
	
	
	
		
			Factor
		
	
	
! Copyright (C) 2008 Slava Pestov.
 | 
						|
! See http://factorcode.org/license.txt for BSD license.
 | 
						|
USING: math kernel sequences sequences.private byte-arrays
 | 
						|
alien.c-types prettyprint.custom parser accessors ;
 | 
						|
IN: nibble-arrays
 | 
						|
 | 
						|
TUPLE: nibble-array
 | 
						|
{ length array-capacity read-only }
 | 
						|
{ underlying byte-array read-only } ;
 | 
						|
 | 
						|
<PRIVATE
 | 
						|
 | 
						|
: nibble BIN: 1111 ; inline
 | 
						|
 | 
						|
: nibbles>bytes 1 + 2/ ; inline
 | 
						|
 | 
						|
: byte/nibble ( n -- shift n' )
 | 
						|
    [ 1 bitand 2 shift ] [ -1 shift ] bi ; inline
 | 
						|
 | 
						|
: get-nibble ( n byte -- nibble )
 | 
						|
    swap neg shift nibble bitand ; inline
 | 
						|
 | 
						|
: set-nibble ( value n byte -- byte' )
 | 
						|
    nibble pick shift bitnot bitand -rot shift bitor ; inline
 | 
						|
 | 
						|
: nibble@ ( n nibble-array -- shift n' byte-array )
 | 
						|
    [ >fixnum byte/nibble ] [ underlying>> ] bi* ; inline
 | 
						|
 | 
						|
PRIVATE>
 | 
						|
 | 
						|
: <nibble-array> ( n -- nibble-array )
 | 
						|
    dup nibbles>bytes <byte-array> nibble-array boa ; inline
 | 
						|
 | 
						|
M: nibble-array length length>> ;
 | 
						|
 | 
						|
M: nibble-array nth-unsafe
 | 
						|
    nibble@ nth-unsafe get-nibble ;
 | 
						|
 | 
						|
M: nibble-array set-nth-unsafe
 | 
						|
    nibble@ [ nth-unsafe set-nibble ] 2keep set-nth-unsafe ;
 | 
						|
 | 
						|
M: nibble-array clone
 | 
						|
    [ length>> ] [ underlying>> clone ] bi nibble-array boa ;
 | 
						|
 | 
						|
: >nibble-array ( seq -- nibble-array )
 | 
						|
    T{ nibble-array } clone-like ; inline
 | 
						|
 | 
						|
M: nibble-array like
 | 
						|
    drop dup nibble-array? [ >nibble-array ] unless ;
 | 
						|
 | 
						|
M: nibble-array new-sequence drop <nibble-array> ;
 | 
						|
 | 
						|
M: nibble-array equal?
 | 
						|
    over nibble-array? [ sequence= ] [ 2drop f ] if ;
 | 
						|
 | 
						|
M: nibble-array resize
 | 
						|
    [ drop ] [
 | 
						|
        [ nibbles>bytes ] [ underlying>> ] bi*
 | 
						|
        resize-byte-array
 | 
						|
    ] 2bi
 | 
						|
    nibble-array boa ;
 | 
						|
 | 
						|
M: nibble-array byte-length length nibbles>bytes ;
 | 
						|
 | 
						|
: N{ \ } [ >nibble-array ] parse-literal ; parsing
 | 
						|
 | 
						|
INSTANCE: nibble-array sequence
 | 
						|
 | 
						|
M: nibble-array pprint-delims drop \ N{ \ } ;
 | 
						|
M: nibble-array >pprint-sequence ;
 | 
						|
M: nibble-array pprint* pprint-object ;
 |