factor/basis/persistent/hashtables/nodes/leaf/leaf.factor

29 lines
1015 B
Factor

! Based on Clojure's PersistentHashMap by Rich Hickey.
USING: kernel accessors locals math arrays namespaces make
persistent.hashtables.config
persistent.hashtables.nodes ;
IN: persistent.hashtables.nodes.leaf
: matching-key? ( key hashcode leaf-node -- ? )
tuck hashcode>> eq? [ key>> = ] [ 2drop f ] if ; inline
M: leaf-node (entry-at) [ matching-key? ] keep and ;
M: leaf-node (pluck-at) [ matching-key? not ] keep and ;
M:: leaf-node (new-at) ( shift value key hashcode leaf-node -- node' added-leaf )
hashcode leaf-node hashcode>> eq? [
key leaf-node key>> = [
value leaf-node value>> =
[ leaf-node f ] [ value key hashcode <leaf-node> f ] if
] [
[let | new-leaf [ value key hashcode <leaf-node> ] |
hashcode leaf-node new-leaf 2array <collision-node>
new-leaf
]
] if
] [ shift leaf-node value key hashcode make-bitmap-node ] if ;
M: leaf-node >alist% [ key>> ] [ value>> ] bi 2array , ;