;;;
;;; isomorph.scm - check isomorphism
;;;
;;;   Copyright (c) 2000-2022  Shiro Kawai  <shiro@acm.org>
;;;
;;;   Redistribution and use in source and binary forms, with or without
;;;   modification, are permitted provided that the following conditions
;;;   are met:
;;;
;;;   1. Redistributions of source code must retain the above copyright
;;;      notice, this list of conditions and the following disclaimer.
;;;
;;;   2. Redistributions in binary form must reproduce the above copyright
;;;      notice, this list of conditions and the following disclaimer in the
;;;      documentation and/or other materials provided with the distribution.
;;;
;;;   3. Neither the name of the authors nor the names of its contributors
;;;      may be used to endorse or promote products derived from this
;;;      software without specific prior written permission.
;;;
;;;   THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
;;;   "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
;;;   LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
;;;   A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
;;;   OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
;;;   SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED
;;;   TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
;;;   PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
;;;   LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
;;;   NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
;;;   SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
;;;

(define-module util.isomorph
  (use scheme.list)
  (use scheme.vector)
  (export isomorphic? object-isomorphic?)
  )
(select-module util.isomorph)

(define (isomorphic? a b :optional (ctx (make-hash-table)))
  (define (iso? a b)
    (cond [(or (boolean? a) (null? a) (char? a) (number? a) (symbol? a)
               (keyword? a) (undefined? a) (eof-object? a))
           (eqv? a b)]
          [(hash-table-get ctx a #f)   ;node has been appeared
           => (^[bb] (eq? bb b))]
          [else
           (hash-table-put! ctx a b)
           (cond [(pair? a)   (and (pair? b)
                                   (iso? (car a) (car b))
                                   (iso? (cdr a) (cdr b)))]
                 [(string? a) (and (string? b) (string=? a b))]
                 [(vector? a) (vector-iso? a b)]
                 [else (object-isomorphic? a b ctx)])]))

  (define (vector-iso? a b)
    (and (vector? b)
         (= (vector-length a) (vector-length b))
         (vector-every iso? a b)))

  (define (hash-iso? a b)
    (and (hash-table? b)
         (let loop ([la (hash-table-map a cons)]
                    [lb (hash-table-map b cons)])
           (cond [(null? a) (null? b)]
                 [(null? b) #f]
                 [(assq (caar la) lb)
                  => (^p (and (iso? (cdar la) (cdr p))
                              (loop (cdr la) (alist-delete (caar la) lb))))]
                 [else #f]))))

  (unless (hash-table? ctx)
    (error "hash table required, but got" ctx))
  (iso? a b))

(define-method object-isomorphic? (a b context)
  (equal? a b))                         ;default
