separate generators into different library

1 files changed, 248 insertions, 0 deletions

diff --git a/srfi.252.generators.sld b/srfi.252.generators.sld
new file mode 100644
index 0000000..e19ecec
--- /dev/null
+++ b/srfi.252.generators.sld
@@ -0,0 +1,248 @@
+;; Property-based testing extension for SRFI 64.
+;; SPDX-License-Identifier: MIT
+;; Copyright 2024 Antero Mejr <mail@antr.me>
+
+;; Permission is hereby granted, free of charge, to any person
+;; obtaining a copy of this software and associated documentation
+;; files (the "Software"), to deal in the Software without
+;; restriction, including without limitation the rights to use, copy,
+;; modify, merge, publish, distribute, sublicense, and/or sell copies
+;; of the Software, and to permit persons to whom the Software is
+;; furnished to do so, subject to the following conditions:
+
+;; The above copyright notice and this permission notice shall be
+;; included in all copies or substantial portions of the Software.
+
+;; THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+;; EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+;; MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+;; NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS
+;; BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN
+;; ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
+;; CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+;; SOFTWARE.
+
+(define-library (srfi 252 generators)
+  (import (scheme base)
+          (scheme case-lambda)
+          (scheme complex)
+          (srfi 1)
+          (srfi 158)
+          (srfi 194)
+          (srfi 143)
+          (srfi 144))
+  (export
+   ;; Generator procedures
+   boolean-generator bytevector-generator
+   char-generator string-generator symbol-generator
+   ;; exact number generators
+   exact-complex-generator exact-integer-generator
+   exact-number-generator exact-rational-generator
+   exact-real-generator
+   exact-integer-complex-generator
+   ;; inexact number generators
+   inexact-complex-generator inexact-integer-generator
+   inexact-number-generator inexact-rational-generator
+   inexact-real-generator
+   ;; Unions of numerical generators
+   complex-generator integer-generator
+   number-generator rational-generator
+   real-generator
+   ;; Special generators
+   list-generator-of pair-generator-of procedure-generator-of
+   vector-generator-of)
+  (begin
+    ;; Constants
+    ;; These values may be implementation-dependent, but should be reasonably
+    ;; high numbers.
+    ;; Value range for exact random generators.
+    (define max-exact fx-greatest)
+    (define min-exact fx-least)
+    ;; Value range for inexact random generators.
+    (define max-inexact fl-greatest)
+    (define min-inexact fl-least)
+    ;; Maximum size for random bytevector/list/string/symbol/vector generators.
+    (define max-size 1001)
+    ;; Maximum character supported by integer->char.
+    (define max-char #x10FFFF)
+    ;; Omit values that are not distinguished in the implementation.
+    (define special-number
+      (append
+       ;; Exact integers
+       '(0 1 -1)
+       ;; Exact ratios
+       '(1/2 -1/2)
+       ;; Exact complex
+       '(0+i 0-i 1+i 1-i -1+i -1-i)
+       ;; Exact complex ratios
+       '(1/2+1/2i 1/2-1/2i -1/2+1/2i -1/2-1/2i)
+       ;; Inexact (integers and non-integers)
+       '(0.0 -0.0 0.5 -0.5 1.0 -1.0)
+       ;; Inexact-complex
+       '(0.0+1.0i 0.0-1.0i -0.0+1.0i -0.0-1.0i
+                  0.5+0.5i 0.5-0.5i -0.5+0.5i -0.5-0.5i
+                  1.0+1.0i 1.0-1.0i -1.0+1.0i -1.0-1.0i
+                  +inf.0+inf.0i +inf.0-inf.0i -inf.0+inf.0i -inf.0-inf.0i
+                  +nan.0+nan.0i)
+       ;; Other (-nan.0 not required, synonymous with +nan.0)
+       '(+inf.0 -inf.0 +nan.0)))
+    (define (special-number-generator)
+      (list->generator special-number))
+    ;; Generator procedures
+    (define (boolean-generator)
+      (gcons* #t #f (make-random-boolean-generator)))
+    (define (bytevector-generator)
+      (let ((gen (make-random-u8-generator)))
+        (gcons* (bytevector)
+                (gmap (lambda (len)
+                        (apply bytevector (generator->list gen len)))
+                      (make-random-integer-generator 0 max-size)))))
+    (define (char-generator)
+      (gcons* #\null
+              (gmap integer->char
+                    (gfilter (lambda (x)
+                               (or (< x #xD800) (> x #xDFFF)))
+                             (make-random-integer-generator 0 max-char)))))
+    (define (string-generator)
+      (gcons* ""
+              (gmap (lambda (n)
+                      (generator->string (gdrop (char-generator) 1) n))
+                    (make-random-integer-generator 1 max-size))))
+    (define (symbol-generator)
+      (gmap string->symbol (string-generator)))
+    ;; Exact number generators
+    (define (exact-complex-generator)
+      (gappend (gfilter (lambda (x)
+                          (and (complex? x)
+                               (exact? (real-part x))
+                               (exact? (imag-part x))))
+                        (special-number-generator))
+               (gmap make-rectangular
+                     (exact-real-generator)
+                     (exact-real-generator))))
+    (define (exact-integer-generator)
+      (gappend (gfilter (lambda (x)
+                          (and (exact? x) (integer? x)))
+                        (special-number-generator))
+               (make-random-integer-generator min-exact max-exact)))
+    (define (ratio-gen)
+      (gmap /
+            (make-random-integer-generator min-exact max-exact)
+            (gfilter (lambda (x) (not (zero? x)))
+                     (make-random-integer-generator min-exact max-exact))))
+    (define (exact-number-generator)
+      ;; Ensure there are no repeated special values, and a random sampling
+      ;; between exact ratios, complex, and integers.
+      (gappend
+       (gfilter exact? (special-number-generator))
+       (gsampling (gmap make-rectangular
+                        (exact-real-generator) (exact-real-generator))
+                  (ratio-gen)
+                  (make-random-integer-generator min-exact max-exact))))
+    (define (exact-rational-generator)
+      (gappend
+       (gfilter (lambda (x)
+                  (and (rational? x) (exact? x)))
+                (special-number-generator))
+       (gsampling (ratio-gen)
+                  (make-random-integer-generator min-exact max-exact))))
+    (define (exact-real-generator)
+      (gappend
+       (gfilter (lambda (x)
+                  (and (real? x) (exact? x)))
+                (special-number-generator))
+       (gsampling (ratio-gen)
+                  (make-random-integer-generator min-exact max-exact))))
+    (define (exact-integer-complex-generator)
+      (gappend (gfilter (lambda (x)
+                          (and (complex? x)
+                               (exact? (real-part x))
+                               (exact? (imag-part x))
+                               (integer? (real-part x))
+                               (integer? (imag-part x))))
+                        (special-number-generator))
+               (gmap make-rectangular
+                     (make-random-integer-generator min-exact max-exact)
+                     (make-random-integer-generator min-exact max-exact))))
+    ;; Inexact number generators
+    (define (inexact-complex-generator)
+      (gappend (gfilter (lambda (x)
+                          (and (complex? x)
+                               (inexact? (real-part x))
+                               (inexact? (imag-part x))))
+                        (special-number-generator))
+               (make-random-rectangular-generator min-inexact max-inexact
+                                                  min-inexact max-inexact)))
+    (define (inexact-integer-generator)
+      (gmap inexact (exact-integer-generator)))
+    (define (inexact-number-generator)
+      (gappend (gfilter inexact? (special-number-generator))
+               (gsampling (make-random-rectangular-generator
+                           min-inexact max-inexact min-inexact max-inexact)
+                          (make-random-real-generator min-inexact max-inexact))))
+    (define (inexact-rational-generator)
+      (gappend (gfilter (lambda (x)
+                          (and (rational? x)
+                               (inexact? x)))
+                        (special-number-generator))
+               (make-random-real-generator min-inexact max-inexact)))
+    (define (inexact-real-generator)
+      (gappend (gfilter (lambda (x)
+                          (and (real? x)
+                               (inexact? x)))
+                        (special-number-generator))
+               (make-random-real-generator min-inexact max-inexact)))
+    ;; Unions of number generators
+    (define (complex-generator)
+      (gsampling (exact-complex-generator)
+                 (inexact-complex-generator)))
+    (define (integer-generator)
+      (gsampling (exact-integer-generator)
+                 (inexact-integer-generator)))
+    (define (number-generator)
+      ;; TODO: May need to be modified for unusual implementation-specific
+      ;; number types, like Kawa's quaternion.
+      (gsampling (exact-number-generator)
+                 (inexact-number-generator)))
+    (define (rational-generator)
+      (gsampling (exact-rational-generator)
+                 (inexact-rational-generator)))
+    (define (real-generator)
+      (gsampling (exact-real-generator)
+                 (inexact-real-generator)))
+    ;; Special generators for collection types
+    (define list-generator-of
+      (case-lambda
+        ((gen)
+         (gcons* '()
+                 (gmap (lambda (len)
+                         (generator->list gen len))
+                       (make-random-integer-generator 1 max-size))))
+        ((gen max-length)
+         (gcons* '()
+                 (gmap (lambda (len)
+                         (generator->list gen len))
+                       (make-random-integer-generator 1 max-length))))))
+    (define pair-generator-of
+      (case-lambda
+        ((gen1) (gmap cons gen1 gen1))
+        ((gen1 gen2) (gmap cons gen2 gen2))))
+    (define (procedure-generator-of gen)
+      ;; Generate variadic procedures that returns a value from a generator.
+      ;; Useful for testing procedures that accept procedure arguments.
+      (gmap (lambda (x)
+              (lambda _
+                x))
+            gen))
+    (define vector-generator-of
+      (case-lambda
+        ((gen)
+         (gcons* (vector)
+                 (gmap (lambda (len)
+                         (generator->vector gen len))
+                       (make-random-integer-generator 0 max-size))))
+        ((gen max-length)
+         (gcons* (vector)
+                 (gmap (lambda (len)
+                         (generator->vector gen len))
+                       (make-random-integer-generator 0 max-length))))))))
\ No newline at end of file