63 lines
		
	
	
		
			1.5 KiB
		
	
	
	
		
			Factor
		
	
	
			
		
		
	
	
			63 lines
		
	
	
		
			1.5 KiB
		
	
	
	
		
			Factor
		
	
	
| ! Copyright (C) 2008 Slava Pestov, James Cash.
 | |
| ! See http://factorcode.org/license.txt for BSD license.
 | |
| USING: accessors arrays assocs classes deques dlists fry kernel
 | |
| sequences sequences.private ;
 | |
| IN: linked-assocs
 | |
| 
 | |
| TUPLE: linked-assoc { assoc read-only } { dlist dlist read-only } ;
 | |
| 
 | |
| : <linked-assoc> ( exemplar -- assoc )
 | |
|     0 swap new-assoc <dlist> linked-assoc boa ;
 | |
| 
 | |
| : <linked-hash> ( -- assoc )
 | |
|     H{ } <linked-assoc> ;
 | |
| 
 | |
| M: linked-assoc assoc-size assoc>> assoc-size ;
 | |
| 
 | |
| M: linked-assoc at*
 | |
|     assoc>> at* [ [ obj>> second-unsafe ] when ] keep ;
 | |
| 
 | |
| <PRIVATE
 | |
| 
 | |
| : (delete-at) ( key assoc dlist -- )
 | |
|     '[ at [ _ delete-node ] when* ] [ delete-at ] 2bi ; inline
 | |
| 
 | |
| PRIVATE>
 | |
| 
 | |
| M: linked-assoc delete-at
 | |
|     [ assoc>> ] [ dlist>> ] bi (delete-at) ;
 | |
| 
 | |
| <PRIVATE
 | |
| 
 | |
| : add-to-dlist ( value key dlist -- node )
 | |
|     [ swap 2array ] dip push-back* ; inline
 | |
| 
 | |
| PRIVATE>
 | |
| 
 | |
| M: linked-assoc set-at
 | |
|     [ assoc>> ] [ dlist>> ] bi
 | |
|     '[ _ 2over key? [ 3dup (delete-at) ] when nip add-to-dlist ]
 | |
|     [ set-at ] 2bi ;
 | |
| 
 | |
| M: linked-assoc >alist
 | |
|     dlist>> dlist>sequence ;
 | |
| 
 | |
| M: linked-assoc clear-assoc
 | |
|     [ assoc>> clear-assoc ] [ dlist>> clear-deque ] bi ;
 | |
| 
 | |
| M: linked-assoc clone
 | |
|     [ assoc>> clone ] [ dlist>> clone ] bi linked-assoc boa ;
 | |
| 
 | |
| INSTANCE: linked-assoc assoc
 | |
| 
 | |
| : >linked-hash ( assoc -- assoc )
 | |
|     [ <linked-hash> ] dip assoc-union! ;
 | |
| 
 | |
| M: linked-assoc assoc-like
 | |
|     over linked-assoc?
 | |
|     [ 2dup [ assoc>> ] bi@ class-of instance? ] [ f ] if
 | |
|     [ drop ] [ assoc>> <linked-assoc> swap assoc-union! ] if ;
 | |
| 
 | |
| M: linked-assoc equal?
 | |
|     over linked-assoc? [ [ dlist>> ] bi@ = ] [ 2drop f ] if ;
 |