File: stream.scm

package info (click to toggle)
sdc 1.0.8beta-8
  • links: PTS
  • area: contrib
  • in suites: slink
  • size: 1,400 kB
  • ctags: 874
  • sloc: lisp: 8,120; ansic: 967; makefile: 671; perl: 136; sh: 50
file content (161 lines) | stat: -rw-r--r-- 4,181 bytes parent folder | download
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
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
;{{{ module stream

(module stream
	(include "stream.sch")
	(export
	 ; (cons-stream hd tl)
	 (inline stream-empty? stream)
	 empty-stream
	 (stream . elements)
	 (inline head stream)
	 (inline tail stream)
	 (inline stream-tail stream n)
	 (accumulate combiner null stream)
	 (stream->list s)
	 (stream-map proc stream) 
	 (stream-for-each proc stream)
	 (stream-through stream . procs)
	 (stream-filter predicate stream)
	 (stream-append s1 . so)
	 (stream2-append s1 . so)
	 (stream-memq element stream)
	 (substream stream after)
	 (stream-insert s . elements)
	 (stream-cute pred action stream)
	 (stream-finde pred action stream)

	 (stream-dbg-watch msg stream)
	 )
)

;}}}
(define-inline (stream-empty? s) (null? s))

(define empty-stream '())

(define (stream . elements)
  (if (null? elements)
      empty-stream
      (cons-stream (car elements) (apply stream (cdr elements)))))

(define-inline (head stream) (car stream))


(define-inline (tail stream)
  (let ((v (cdr	stream)))
     (if (procedure? v)
	 (let ((nv (v)))
	   (set-cdr! stream nv)
	   nv)
	 v)))


(define-inline (stream-tail stream n)
  (if (eqv? n 0) stream (stream-tail (tail stream) (- n 1))))

(define (accumulate combiner null stream)
  (if (stream-empty? stream)
      null
      (combiner (head stream) (accumulate combiner null (tail stream)))))


(define (stream->list s)
  (let loop ((r s))
    (if (stream-empty? r)
	s
	(loop (tail r)))))


(define (stream-map proc stream)
  (if (stream-empty? stream)
      stream
      (cons-stream (proc (head stream)) (stream-map proc (tail stream)))))

(define (stream-for-each proc stream)
  (if (stream-empty? stream)
      empty-stream
      (begin
	(proc (head stream))
	(stream-for-each proc (tail stream)))))

(define (stream-through stream . procs)
  (if (null? procs)
      stream
      ((car procs) (apply stream-through stream (cdr procs)))))

(define (stream-cond-map start? end? proc stream)
  (letrec
      ((mi
	(lambda (inside start? end? proc stream)
	  (if stream
	      (let ((rs ((if inside end? start?) stream)))
		(if rs
		    (mi (not inside) start? end? proc rs)
		    (cons-stream
		     (if inside (proc (head stream)) (head stream))
		     (mi inside start? end? proc (tail stream)))))))))
    (mi #f start? end? proc stream)))

(define (stream-filter predicate stream)
  (cond
   ((stream-empty? stream) stream)
   ((predicate (head stream))
    (cons-stream (head stream) (stream-filter predicate (tail stream))))
   (else (stream-filter predicate (tail stream)))))

(define (stream-append s1 . so)
  (if (stream-empty? s1)
      (if (null? so) empty-stream (apply stream-append so))
      (cons-stream (head s1) (apply stream-append (tail s1) so))))

(define (stream2-append s1 . so)
  (if (stream-empty? s1)
      (if (null? so)
	  empty-stream
	  (apply stream2-append (force (car so)) (cdr so)))
      (cons-stream (head s1) (apply stream2-append (tail s1) so))))

(define (substream stream after)
  (cond
   ((not stream) #f)
   ((eq? stream after) #f)
   (else (cons-stream (head stream) (substream (tail stream) after)))))

(define (stream-insert stream . elements)
  (if (null? elements)
      stream
      (cons-stream (car elements)
		   (apply stream-insert stream (cdr elements)))))

(define (stream-finde pred action stream)
  (cond
   ((stream-empty? stream) empty-stream)
   ((pred (head stream)) (action stream))
   (else (stream-finde pred action (tail stream)))))

(define (stream-cute pred action stream) ; the name: pun intended
  (cond
   ((stream-empty? stream) empty-stream)
   ((pred (head stream)) (action stream))
   (else (cons-stream (head stream)
		      (stream-cute pred action (tail stream))))))

(define (stream-memq element stream)
  (cond
   ((not stream) #f)
   ((eq? element (head stream)) stream)
   (else (stream-memq element (tail stream)))))

;{{{ Debugging Aid

(define (stream-dbg-watch msg stream)
  (display msg (current-error-port))
  (if (stream-empty? stream)
      (begin
	(display #" it's EMPTY now." (current-error-port))
	empty-stream)
   (begin
     (write (head stream) (current-error-port))
     (cons-stream (head stream) (stream-dbg-watch msg (tail stream))))))

;}}}