From 3332d4a1e9f03caa748395ee200b1efcb05d535f Mon Sep 17 00:00:00 2001
From: Kaz Kylheku <kaz@kylheku.com>
Date: Sun, 17 Jan 2021 00:29:26 -0800
Subject: matcher: support loose mode for structures.

* share/txr/stdlib/match.tl (compile-struct-match):
Allow a pattern instead of a struct type name, in which
case the object can be of any struct type which has
the slots required by the pattern.

* txr.1: Documented.
---
 share/txr/stdlib/match.tl | 68 +++++++++++++++++++++++++++++++++--------------
 1 file changed, 48 insertions(+), 20 deletions(-)

(limited to 'share')

diff --git a/share/txr/stdlib/match.tl b/share/txr/stdlib/match.tl
index 9648901c..0a975134 100644
--- a/share/txr/stdlib/match.tl
+++ b/share/txr/stdlib/match.tl
@@ -59,26 +59,54 @@
 (defvar *match-form*)
 
 (defun compile-struct-match (struct-pat obj-var)
-  (let* ((required-type (cadr struct-pat))
-         (slot-pairs (plist-to-alist (cddr struct-pat)))
-         (required-slots [mapcar car slot-pairs])
-         (slot-gensyms [mapcar gensym required-slots])
-         (slot-patterns [mapcar cdr slot-pairs])
-         (slot-matches [mapcar compile-match slot-patterns slot-gensyms])
-         (slot-val-exprs [mapcar (ret ^(slot ,obj-var ',@1))
-                                 required-slots])
-         (guard (new match-guard
-                     vars slot-gensyms
-                     var-exprs slot-val-exprs
-                     guard-expr  ^(subtypep (typeof ,obj-var)
-                                            ',required-type))))
-    (new compiled-match
-         pattern struct-pat
-         obj-var obj-var
-         guard-chain (cons guard (mappend .guard-chain slot-matches))
-         test-expr ^(and ,*(mapcar .test-expr slot-matches))
-         vars [mappend .vars slot-matches]
-         var-exprs [mappend .var-exprs slot-matches])))
+  (tree-bind (op required-type . pairs) struct-pat
+    (let* ((loose-p (not (bindable required-type)))
+           (slot-pairs (plist-to-alist pairs))
+           (required-slots [mapcar car slot-pairs])
+           (slot-gensyms [mapcar gensym required-slots])
+           (type-gensym (if loose-p
+                          (gensym "type-")))
+           (slot-patterns [mapcar cdr slot-pairs])
+           (slot-matches [mapcar compile-match slot-patterns slot-gensyms])
+           (type-match (if loose-p
+                         (compile-match required-type type-gensym)))
+           (all-matches (if loose-p
+                          (cons type-match slot-matches)
+                          slot-matches))
+           (slot-val-exprs [mapcar (ret ^(slot ,obj-var ',@1)) required-slots])
+           (guard0 (if loose-p
+                     (new match-guard
+                          vars (list type-gensym)
+                          var-exprs (list ^(struct-type ,obj-var))
+                          guard-expr ^(structp ,obj-var))))
+           (guard1 (new match-guard
+                        vars slot-gensyms
+                        var-exprs slot-val-exprs
+                        guard-expr (if loose-p
+                                     ^(and ,*(mapcar
+                                               (ret ^(slotp ,type-gensym ',@1))
+                                               required-slots))
+                                     ^(subtypep (typeof ,obj-var)
+                                                ',required-type)))))
+      (unless loose-p
+        (let ((type (find-struct-type required-type)))
+          (if type
+            (each ((slot required-slots))
+              (unless (slotp type slot)
+                (compile-defr-warning *match-form* ^(slot . ,slot)
+                                      "~s has no slot ~s"
+                                      required-type slot)))
+            (compile-defr-warning *match-form* ^(struct-type . ,required-type)
+                                  "no such struct type: ~s"
+                                  required-type))))
+      (new compiled-match
+           pattern struct-pat
+           obj-var obj-var
+           guard-chain ^(,*(if guard0 (list guard0)) ,guard1
+                         ,*(mappend .guard-chain all-matches))
+           test-expr ^(and ,*(mapcar .test-expr all-matches))
+           vars [mappend .vars all-matches]
+           var-exprs [mappend .var-exprs all-matches]))))
 
 (defun compile-var-match (sym obj-var)
   (or (null sym) (bindable sym)
-- 
cgit v1.2.3