Add pprint versions of codegen macros for readability. (#1437)

* Add pretty-printing macros for codegen macros.

* Define UFix type to print nicer in the output of codegen.

* `UFix` -> `ufixnum`.

* Add the macro names instead of generating for allowing to grep.

* `print` -> `pprint`.

Author: fukamachi

library/primitive-types.lisp

@@ -76,6 +76,6 @@
   (define-type IFix
     "Non-allocating tagged integer; range is platform-dependent. Does not error on overflow. Uses `fixnum`.")
 
-  (repr :native (cl:and cl:fixnum cl:unsigned-byte))
-  (define-type UFix 
+  (repr :native coalton-impl/util:ufixnum)
+  (define-type UFix
     "Non-allocating tagged non-negative integer; range is platform-dependent. Uses `(and fixnum unsigned-byte)`."))

src/package.lisp

@@ -19,6 +19,9 @@
    #:coalton-codegen
    #:coalton-codegen-ast
    #:coalton-codegen-types
+   #:pprint-coalton-codegen
+   #:pprint-coalton-codegen-ast
+   #:pprint-coalton-codegen-types
    #:coalton
    #:declare
    #:define

src/reader.lisp

@@ -184,5 +184,28 @@ It ensures the presence of source metadata for STREAM and then calls MAYBE-READ-
   "Dump the AST for toplevel definitions occurring in FORMS to *standard-out* and return NIL."
   (compile-forms 'coalton:coalton-codegen-ast forms))
 
+(defmacro define-pprint-codegen-macro (name codegen)
+  "Generate a pretty-printing macro for codegen macros."
+  `(defmacro ,name (&body forms)
+     ,(format nil "Pretty print code generated by ~S." codegen)
+     (let ((package (gensym "PACKAGE")))
+       `(let ((,package (make-package '#:pprint-codegen-package
+                                      :use '(#:cl))))
+          (unwind-protect (progn
+                            ;; Import all symbols in the current package
+                            ;; not to print its package prefix.
+                            (do-symbols (symb *package*)
+                              (unless (nth-value 1
+                                                 (find-symbol (symbol-name symb) ,package))
+                                (import symb ,package)))
+                            (let ((*package* ,package))
+                              (pprint (,',codegen ,@forms))
+                              (values)))
+            (delete-package ,package))))))
+
+(define-pprint-codegen-macro coalton:pprint-coalton-codegen coalton:coalton-codegen)
+(define-pprint-codegen-macro coalton:pprint-coalton-codegen-types coalton:coalton-codegen-types)
+(define-pprint-codegen-macro coalton:pprint-coalton-codegen-ast coalton:coalton-codegen-ast)
+
 (defmacro coalton:coalton (&rest forms)
   (compile-forms 'coalton:coalton forms))

src/utilities.lisp

@@ -32,6 +32,7 @@
    #:project-indices                    ; FUNCTION
    #:project-elements                   ; FUNCTION
    #:maybe-read-form                    ; FUNCTION
+   #:ufixnum                            ; TYPE
    ))
 
 (in-package #:coalton-impl/util)
@@ -216,3 +217,6 @@ Examples:
   (mapcar (lambda (idx)
             (nth (position idx list) data))
           elements))
+
+(deftype ufixnum ()
+  '(and fixnum unsigned-byte))