138 lines
5.0 KiB
Scheme
138 lines
5.0 KiB
Scheme
;;; b-tree.scm - a B-tree for Xanadu
|
|
;;;
|
|
;;; Copyright (c) 2012 Johan Ceuppens
|
|
;;;
|
|
;;; All rights reserved.
|
|
;;;
|
|
;;; Redistribution and use in source and binary forms, with or without
|
|
;;; modification, are permitted provided that the following conditions
|
|
;;; are met:
|
|
;;; 1. Redistributions of source code must retain the above copyright
|
|
;;; notice, this list of conditions and the following disclaimer.
|
|
;;; 2. Redistributions in binary form must reproduce the above copyright
|
|
;;; notice, this list of conditions and the following disclaimer in the
|
|
;;; documentation and/or other materials provided with the distribution.
|
|
;;; 3. The name of the authors may not be used to endorse or promote products
|
|
;;; derived from this software without specific prior written permission.
|
|
;;;
|
|
;;; THIS SOFTWARE IS PROVIDED BY THE AUTHORS ``AS IS'' AND ANY EXPRESS OR
|
|
;;; IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
|
|
;;; OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
|
|
;;; IN NO EVENT SHALL THE AUTHORS BE LIABLE FOR ANY DIRECT, INDIRECT,
|
|
;;; INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
|
|
;;; NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
|
;;; DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
|
;;; THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
|
;;; (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
|
|
;;; THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
|
|
(define (make-b-tree-node l r)
|
|
(let ((numitems 0)
|
|
(numnodes 0)
|
|
(root 'foo)
|
|
(data #f)
|
|
(left l)
|
|
(right r))
|
|
|
|
(define (get-data)
|
|
data)
|
|
|
|
(define (set-data! value)
|
|
(set! data value))
|
|
|
|
(define (get-left)
|
|
left)
|
|
|
|
(define (get-right)
|
|
right)
|
|
|
|
(define (get-root)
|
|
root)
|
|
|
|
(define (get-numitems)
|
|
numitems)
|
|
|
|
(define (get-numnodes)
|
|
numnodes)
|
|
|
|
(define (dispatch msg)
|
|
(lambda (msg)
|
|
(cond ((eq? msg 'get-root)
|
|
get-root)
|
|
((eq? msg 'get-numitems)
|
|
get-numitems)
|
|
((eq? msg 'get-numnodes)
|
|
get-numnodes)
|
|
((eq? msg 'get-left)
|
|
get-left)
|
|
((eq? msg 'get-data)
|
|
get-data)
|
|
((eq? msg 'set-data!)
|
|
set-data!)
|
|
((eq? msg 'get-right)
|
|
get-right)
|
|
(else (display "b-tree-node : message not understood")))))
|
|
dispatch))
|
|
|
|
(define (make-b-tree n-ary);;NOTE FIXME n-ary and vector-length
|
|
(let ((*tree (make-vector n-ary (make-b-tree-node (make-vector n-ary) (make-vector n-ary)))))
|
|
|
|
(define (vector-median v)
|
|
(let ((len (ceiling (/ (vector-length v) 2))))
|
|
(let ((retl (make-vector len))
|
|
(retr (if (odd? len) (make-vector (- len 1)) (make-vector len))))
|
|
(do ((i 0 (+ i 1)))
|
|
((= i len) (list retl retr))
|
|
(vector-set! retl i (vector-ref v i))
|
|
(vector-set! retr (- len (- i 1)) (vector-ref v (- len (- i 1))))
|
|
))))
|
|
|
|
|
|
(define (search-rec str tree) ;; root param in b-treenode
|
|
(let ((len (vector-length tree)))
|
|
(do ((i 0 (+ i 1)))
|
|
((cond ((>= i (- len 1));;last node
|
|
(do ((j 0 (+ j 1)))
|
|
((= j len) (display "node not found.") 0)
|
|
(add-rec str (vector-ref j tree))))
|
|
((and (string<? str (vector-ref tree i))
|
|
(string>? str (vector-ref tree (+ i 1))))
|
|
(display "node not found in tree.") 0)
|
|
((string=? str (vector-ref tree i))
|
|
(display "string found in tree.") str)
|
|
(else (display "never reached."))))
|
|
)))
|
|
|
|
(define (search str)
|
|
(search-rec *tree))
|
|
|
|
(define (add-rec str tree) ;; root param in b-treenode
|
|
(let ((len (vector-length tree)))
|
|
(do ((i 0 (+ i 1)))
|
|
((cond ((>= i (- len 1));;last node
|
|
(do ((j 0 (+ j 1)))
|
|
((= j len) (display "node not added.") 0)
|
|
(add-rec str (vector-ref j tree)))
|
|
)))
|
|
((and (string<? str (((vector-ref tree i) 'get-data)))
|
|
(string>? str (((vector-ref tree (+ i 1)) 'get-data))))
|
|
(let ((lr (vector-median tree)))
|
|
(let ((new-node (make-b-tree-node (car lr) (cadr lr))))
|
|
((new-node 'set-data!) str)
|
|
(vector-set! tree i new-node)
|
|
)))
|
|
((string=? str (vector-ref tree i))
|
|
(set! i (vector-length tree)))
|
|
(else (display "never reached."))))
|
|
)))
|
|
|
|
|
|
(define (add str)
|
|
(add-rec *tree))
|
|
|
|
(define (dispatch msg)
|
|
(cond ((eq? msg 'add) add)
|
|
((eq? msg 'search) search)
|
|
(else (display "b-tree : message not understood."))))
|
|
dispatch))
|