blob: 2b67f78811e4156b7f407d15cdedb8de81873471 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
|
(define-module (starling ring-buffer)
#:use-module (srfi srfi-9)
#:export (make-ring-buffer
ring-buffer
ring-buffer-length
ring-buffer-put!
ring-buffer-get!
ring-buffer-ref))
(define-record-type <ring-buffer>
(%make-ring-buffer vector length head tail)
ring-buffer?
(vector ring-buffer-vector)
(length ring-buffer-length set-ring-buffer-length!)
(head ring-buffer-head set-ring-buffer-head!)
(tail ring-buffer-tail set-ring-buffer-tail!))
(define (make-ring-buffer size)
(%make-ring-buffer (make-vector size #f) 0 0 0))
(define (ring-buffer-empty? ring)
(zero? (ring-buffer-length ring)))
(define (ring-buffer-put! ring x)
(let* ((head (ring-buffer-head ring))
(tail (ring-buffer-tail ring))
(l (ring-buffer-length ring))
(v (ring-buffer-vector ring))
(vl (vector-length v)))
(vector-set! v tail x)
(set-ring-buffer-length! ring (min (+ l 1) vl))
(when (and (> l 0) (= head tail))
(set-ring-buffer-head! ring (modulo (+ head 1) vl)))
(set-ring-buffer-tail! ring (modulo (+ tail 1) vl))))
(define (ring-buffer-get! ring)
(if (ring-buffer-empty? ring)
(error "ring buffer empty" ring)
(let* ((head (ring-buffer-head ring))
(v (ring-buffer-vector ring))
(result (vector-ref v head)))
(vector-set! v head #f)
(set-ring-buffer-head! ring (modulo (+ head 1) (vector-length v)))
(set-ring-buffer-length! ring (- (ring-buffer-length ring) 1))
result)))
(define (ring-buffer-ref ring i)
(let ((l (ring-buffer-length ring))
(v (ring-buffer-vector ring)))
(if (>= i l)
(error "ring buffer index out of bounds" i)
(vector-ref v (modulo (+ (ring-buffer-head ring) i)
(vector-length v))))))
;; (define (test)
;; (define (do-test ring)
;; (ring-buffer-put! ring 'h)
;; (pk ring)
;; (ring-buffer-put! ring 'e)
;; (pk ring)
;; (ring-buffer-put! ring 'l)
;; (pk ring)
;; (ring-buffer-put! ring 'l)
;; (pk ring)
;; (ring-buffer-put! ring 'o)
;; (pk ring)
;; (ring-buffer-put! ring 'w)
;; (pk ring)
;; (ring-buffer-put! ring 'o)
;; (pk ring)
;; (ring-buffer-put! ring 'r)
;; (pk ring)
;; (ring-buffer-put! ring 'l)
;; (pk ring)
;; (ring-buffer-put! ring 'd)
;; (pk ring)
;; (pk (ring-buffer-get! ring) (ring-buffer-get! ring) (ring-buffer-get! ring) (ring-buffer-get! ring))
;; (pk ring))
;; (let ((ring (make-ring-buffer 4)))
;; (do-test ring)
;; (do-test ring)))
|