mescc: support -E.
[mes.git] / scripts / mescc.mes
index 33b89e5c2ae3d45570112955dbfcfc8ddef5ed80..b8c6e169abb65ee07590ee904a7579b3a5686d3e 100755 (executable)
@@ -39,6 +39,7 @@ exit $r
 
 (mes-use-module (mes guile))
 (mes-use-module (mes getopt-long))
+(mes-use-module (mes pretty-print))
 (mes-use-module (language c99 compiler))
 (mes-use-module (srfi srfi-1))
 (mes-use-module (srfi srfi-26))
@@ -54,24 +55,26 @@ exit $r
 (define (parse-opts args)
   (let* ((option-spec
           '((c (single-char #\c))
-           (D (single-char #\D) (value #t))
+            (D (single-char #\D) (value #t))
+            (E (single-char #\E))
             (help (single-char #\h))
-           (I (single-char #\I) (value #t))
-           (o (single-char #\o) (value #t))
-           (version (single-char #\V) (value #t))))
-        (options (getopt-long args option-spec))
-        (help? (option-ref options 'help #f))
-        (files (option-ref options '() '()))
-        (usage? (and (not help?) (null? files)))
+            (I (single-char #\I) (value #t))
+            (o (single-char #\o) (value #t))
+            (version (single-char #\V) (value #t))))
+         (options (getopt-long args option-spec))
+         (help? (option-ref options 'help #f))
+         (files (option-ref options '() '()))
+         (usage? (and (not help?) (null? files)))
          (version? (option-ref options 'version #f)))
     (or
      (and version?
-         (format (current-output-port) "mescc.scm (mes) ~a\n" %version))
+          (format (current-output-port) "mescc.scm (mes) ~a\n" %version))
      (and (or help? usage?)
           (format (or (and usage? (current-error-port)) (current-output-port)) "\
 Usage: mescc [OPTION]... FILE...
   -c                 compile and assemble, but do not link
   -D DEFINE          define DEFINE
+  -E                 preprocess only; do not compile, assemble or link
   -h, --help         display this help and exit
   -I DIR             append DIR to include path
   -o FILE            write output to FILE
@@ -80,47 +83,70 @@ Usage: mescc [OPTION]... FILE...
           (exit (or (and usage? 2) 0)))
      options)))
 
-(define (object->info file)
-  (let* ((string (with-input-from-file file read-string))
-         (module (resolve-module '(language c99 compiler))))
-    (eval-string string module)))
-
 (define (object->info file)
   (let* ((lst (with-input-from-file file read)))
     (primitive-eval lst)))
 
+(define (main:ast->info file)
+  (let ((ast (with-input-from-file file read)))
+    (with-input-from-file file
+      (lambda ()
+        (c99-ast->info ast)))))
+
+(define (source->ast defines includes)
+  (lambda (file)
+    (with-input-from-file file
+      (lambda ()
+        (pretty-print (c99-input->ast #:defines defines #:includes includes))))))
+
 (define (source->info defines includes)
   (lambda (file)
     (with-input-from-file file
       (lambda ()
         ((c99-input->info #:defines defines #:includes includes))))))
 
+(define (ast? o)
+  (or (string-suffix? ".E" o)
+      (string-suffix? ".mes-E" o)))
+
+(define (object? o)
+  (or (string-suffix? ".o" o)
+      (string-suffix? ".mes-o" o)))
+
 (define (main args)
   (let* ((args (cons* (car args) (cdr (member "--" args))))
          (options (parse-opts args))
          (files (option-ref options '() '()))
          (file (if (null? files) (string-append %docdir "examples/main.c")
                    (car files)))
+         (preprocess? (option-ref options 'E #f))
          (compile? (option-ref options 'c #f))
+         (asts (filter ast? files))
+         (objects (filter object? files))
          (sources (filter (cut string-suffix? ".c" <>) files))
-         (objects (filter (negate (cut string-suffix? ".c" <>)) files))
          (base (substring file (1+ (or (string-rindex file #\/) -1)) (- (string-length file) 2)))
-         (out (option-ref options 'o (if compile? (string-append base ".o") "a.out")))
+         (out (option-ref options 'o (cond (compile? (string-append base ".o"))
+                                           (preprocess? (string-append base ".E"))
+                                           (else "a.out"))))
          (multi-opt (lambda (option) (lambda (o) (and (eq? (car o) option) (cdr o)))))
          (defines (reverse (filter-map (multi-opt 'D) options)))
          (includes (reverse (filter-map (multi-opt 'I) options))))
     (when (getenv "MES_DEBUG") (format (current-error-port) "options=~s\n" options)
           (format (current-error-port) "output: ~a\n" out))
     (if (and (pair? sources) (pair? objects)) (error "cannot mix source and object files:" files))
-    (format (current-error-port) "inputs: ~a\n" files)
-    (with-output-to-port (open-output-file out (if compile? S_IRWXU))
+    (with-output-to-port (open-output-file out (if (and (not compile?)
+                                                        (not preprocess?)) S_IRWXU))
       (lambda ()
-        (if (pair? objects) (let ((infos (map object->info objects)))
+        (cond ((pair? objects) (let ((infos (map object->info objects)))
+                                 (if compile? (infos->object infos)
+                                     (infos->elf infos))))
+              ((pair? asts) (let ((infos (map main:ast->info asts)))
                               (if compile? (infos->object infos)
-                                  (infos->elf infos)))
-            (let ((infos (map (source->info defines includes) sources)))
-              (if compile? (infos->object infos)
-                  (infos->elf infos))))))))
+                                  (infos->elf infos))))
+              ((pair? sources) (if preprocess? (map (source->ast defines includes) sources)
+                                   (let ((infos (map (source->info defines includes) sources)))
+                                     (if compile? (infos->object infos)
+                                         (infos->elf infos))))))))))
 
 (main (command-line))
 ()