X-Git-Url: http://repo.macrolet.net/gitweb/?a=blobdiff_plain;f=src%2Frandom.lisp;h=53f423b53da6a85b38702d318405354a43e9061b;hb=b37e3f37f173eea9c8149109f73f9b34c6a1515c;hp=f91ca4671535a10dbfd11c021ee75c5ff061ecf0;hpb=dd25ec9992371a9aa37e2dfdec64d9a149c54e79;p=fiveam.git

diff --git a/src/random.lisp b/src/random.lisp
index f91ca46..53f423b 100644
--- a/src/random.lisp
+++ b/src/random.lisp
@@ -235,6 +235,21 @@ producs objects."
        repeat (funcall length)
        collect (funcall elements))))
 
+(defun gen-tree (&key (size 20)
+                      (elements (gen-integer :min -10 :max 10)))
+  "Returns a generator which producs random trees. SIZE control
+the approximate size of the tree, but don't try anything above
+ 30, you have been warned. ELEMENTS must be a generator which
+will produce the elements."
+  (labels ((rec (&optional (current-depth 0))
+             (let ((key (random (+ 3 (- size current-depth)))))
+               (cond ((> key 2)
+                      (list (rec (+ current-depth 1))
+                            (rec (+ current-depth 1))))
+                     (t (funcall elements))))))
+    (lambda ()
+      (rec))))
+
 (defun gen-buffer (&key (length (gen-integer :min 0 :max 50))
                         (element-type '(unsigned-byte 8))
                         (elements (gen-integer :min 0 :max (1- (expt 2 8)))))