Scheme continuation 学习之 tree->generator

Lispor

最近在学习 scheme 中的 continuation，终于把 tree-generator(Teach Yourself Scheme in Fixnum Days) 看懂了，
根据自己的理解又写了一个改良版本，比较容易理解。

1. ;;;    Function: tree->generator
   
2. ;;;    Example:
   
3. ;;;    > (define t '(1 (2) 3))
   
4. ;;;    > (define gen (tree->generator t))
   
5. ;;;    > (gen)
   
6. ;;;    1
   
7. ;;;    > (gen)
   
8. ;;;    2
   
9. ;;;    > (gen)
   
10. ;;;    3
    
11. ;;;    > (gen)
    
12. ;;;    '()
    
13. ;;;    > (gen)
    
14. ;;;    '()
    
15. 
    
16. (define (tree->generator tree)
    
17.   (let ((leaf #f) (walk-tree #f))
    
18.     (define (walk-tree)
    
19.       (let loop ((tree tree))
    
20.         (cond ((null? tree) 'skip)
    
21.               ((pair? tree) (loop (car tree))
    
22.                             (loop (cdr tree)))
    
23.               (else (call/cc
    
24.                      (lambda (rest-of-tree)
    
25.                        (set! walk-tree
    
26.                              (lambda ()
    
27.                                (rest-of-tree 'resume)))
    
28.                        (leaf tree))))))
    
29.       (leaf '()))
    
30.     (define (next)
    
31.       (call/cc
    
32.        (lambda (k)
    
33.          (set! leaf k)
    
34.          (walk-tree))))
    
35.     next))

复制代码