blob: c3840d220e82501ca90af554af201841abe449f5 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
|
;;; Copyright © 2023 Jakob L. Kreuze <zerodaysfordays@sdf.org>
;;;
;;; This program is free software; you can redistribute it and/or
;;; modify it under the terms of the GNU General Public License as
;;; published by the Free Software Foundation; either version 3 of the
;;; License, or (at your option) any later version.
;;;
;;; This program is distributed in the hope that it will be useful,
;;; but WITHOUT ANY WARRANTY; without even the implied warranty of
;;; MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
;;; General Public License for more details.
;;;
;;; You should have received a copy of the GNU General Public License
;;; along with this program. If not, see
;;; <http://www.gnu.org/licenses/>.
(define-module (compiler passes lift-comprehensions)
#:use-module (compiler lean)
#:use-module (compiler types)
#:use-module (compiler utilities)
#:use-module (ice-9 match)
#:use-module (ice-9 pretty-print)
#:use-module (oop goops)
#:use-module (srfi srfi-1)
#:use-module (srfi srfi-11)
#:export (compile-comprehensions))
(define (compile-comprehensions policy)
(define new-definitions (list))
;; For generating anonymous rule names.
(define top-level-names (map (compose inner first inner left) (rules policy)))
(define (new-comprehension-definition breadcrumb comprehension)
(match comprehension
(($ <constant> ($ <comprehension> 'object key value body))
(let* ((name (fresh-identifier top-level-names "objectcomp"))
(definition (make <definition>
#:kind 'field-eqv
#:left key
#:right value
#:default #f
#:rules (if (not (list? body))
`(,body)
body)))
(local (make <local>
#:identifier (make <variable-ref> #:name name)
#:constraints `((== ,(canonicalize name breadcrumb) object))
#:lean-name (make <variable-ref> #:name (format #f "~a_eqv" name)))))
(values local definition)))))
(let ((policy
(policy-map
policy
(lambda (scope breadcrumb form)
(match form
(($ <constant> ($ <comprehension> 'object key value body))
(let-values (((identifier definition) (new-comprehension-definition breadcrumb form)))
;; Add the resulting rule to the policy.
(set! new-definitions `(,definition ,@new-definitions))
;; Also add it to the scope.
(add-to-scope scope breadcrumb identifier)
;; Make like Indiana Jones and swap this out with a reference to the
;; new lifted object.
(make <reference> #:parts `(,(inner identifier)))))
(_ form))))))
(match policy
(($ <policy> _package _imports _scope rules _comments)
(with-modified-slot policy 'rules
(append rules new-definitions))))))
|
© 2015 - 2026 Jakob L. Kreuze
