;;;; entropy-source.scm
;;;; Kon Lovett, Sep '23
;;;; Kon Lovett, Oct '09

(module entropy-source

(;export
  *make-entropy-source
  entropy-source? check-entropy-source error-entropy-source
  *entropy-source-name
  *entropy-source-documentation
  @entropy-source-constructor
  @entropy-source-u8
  @entropy-source-f64
  @entropy-source-u8vector
  @entropy-source-f64vector
  ;
  entropy-source-integer
  entropy-source-f64-integer
  ;
  registered-entropy-sources
  registered-entropy-source
  unregister-entropy-source
  register-entropy-source!)

(import scheme
  (chicken base)
  (chicken type)
  (only (srfi 1) alist-cons alist-delete!)
  (only type-checks-basic define-check+error-type)
  (only type-checks-structured check-procedure)
  (only type-checks-atoms check-symbol)
  source-registration)

;;;

(include-relative "srfi-27-common-types")

(: *make-entropy-source (procedure entropy-source-name string procedure procedure procedure procedure -> entropy-source))
(: entropy-source? (* -> boolean : entropy-source))
(: check-entropy-source ((or false symbol string) * #!optional (or symbol string) -> entropy-source))
(: error-entropy-source ((or false symbol string) * #!optional (or symbol string) -> void))
(: @entropy-source-constructor (entropy-source -> procedure))
(: *entropy-source-name (entropy-source -> entropy-source-name))
(: *entropy-source-documentation (entropy-source -> string))
(: @entropy-source-u8 (entropy-source -> procedure))
(: @entropy-source-f64 (entropy-source -> procedure))
(: @entropy-source-u8vector (entropy-source -> procedure))
(: @entropy-source-f64vector (entropy-source -> procedure))

(: entropy-source-integer (entropy-source -> number))
(: entropy-source-f64-integer (entropy-source -> number))

(: registered-entropy-sources (-> list))
(: registered-entropy-source (entropy-source-name -> (or false entropy-source)))
(: unregister-entropy-source (entropy-source-name -> void))
(: register-entropy-source! (entropy-source-name procedure -> void))

;;;

;;

(define-record-type entropy-source
  (*make-entropy-source ctor name docu u8 f64 u8vec f64vec)
  entropy-source?
  (ctor       @entropy-source-constructor)
  (name       *entropy-source-name)
  (docu       *entropy-source-documentation)
  (u8         @entropy-source-u8)
  (f64        @entropy-source-f64)
  (u8vec      @entropy-source-u8vector)
  (f64vec     @entropy-source-f64vector) )

(define-check+error-type entropy-source)

;;

(define (entropy-source-integer es)
  ;ugly but ...
  (let ((genf64 (@entropy-source-f64 es)))
    (let loop ((x (genf64)))
      (if (integer? x) x
          (loop (genf64)) ) ) ) )

(define entropy-source-f64-integer entropy-source-integer)

;; Entropy Source Constructor Registry

(define +reg+ (make-source-registration 'entropy-source '()))

(define (registered-entropy-sources)
  ((@source-registration-key +reg+)) )

(define (registered-entropy-source name)
  ((@source-registration-ref +reg+) name) )

(define (unregister-entropy-source name)
  ((@source-registration-deref! +reg+) name) )

(define (register-entropy-source! name ctor)
  ((@source-registration-register! +reg+) name ctor) )

) ;entropy-source