mescc: support -E.
[mes.git] / guile / mescc.scm
1 #! /bin/sh
2 # -*-scheme-*-
3 GODIR=${GODIR-@GODIR@}
4 GUILEDIR=${GUILEDIR-@GUILEDIR@}
5 [ "$GODIR" = @"GODIR"@ ] && GODIR=$(dirname $0)
6 [ "$GUILEDIR" = @"GUILEDIR"@ ] && GUILEDIR=$(dirname $0)
7 export GUILE_AUTO_COMPILE=${GUILE_AUTO_COMPILE-0}
8 exec ${GUILE-guile} -L $GUILEDIR -C $GODIR -e '(mescc)' -s "$0" "$@"
9 !#
10
11 ;;; Mes --- The Maxwell Equations of Software
12 ;;; Copyright © 2016,2017 Jan Nieuwenhuizen <janneke@gnu.org>
13 ;;;
14 ;;; This file is part of GNU Guix.
15 ;;;
16 ;;; Mes is free software; you can redistribute it and/or modify it
17 ;;; under the terms of the GNU General Public License as published by
18 ;;; the Free Software Foundation; either version 3 of the License, or (at
19 ;;; your option) any later version.
20 ;;;
21 ;;; Mes is distributed in the hope that it will be useful, but
22 ;;; WITHOUT ANY WARRANTY; without even the implied warranty of
23 ;;; MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
24 ;;; GNU General Public License for more details.
25 ;;;
26 ;;; You should have received a copy of the GNU General Public License
27 ;;; along with Mes.  If not, see <http://www.gnu.org/licenses/>.
28
29 ;; The Maxwell Equations of Software -- John McCarthy page 13
30 ;; http://www.softwarepreservation.org/projects/LISP/book/LISP%201.5%20Programmers%20Manual.pdf
31
32 #!
33 Run with Guile-1.8:
34 GUILE='~/src/guile-1.8/build/pre-inst-guile --debug -q' guile/mescc.scm
35 !#
36
37 (define-module (mescc)
38   #:use-module (language c99 compiler)
39   #:use-module (ice-9 getopt-long)
40   #:use-module (ice-9 pretty-print)
41   #:use-module (srfi srfi-1)
42   #:use-module (srfi srfi-26)
43   #:export (main))
44
45 (define %prefix (if (string-prefix? "@PREFIX" "@PREFIX@") "" "@PREFIX@"))
46 (define %datadir (if (string-prefix? "@DATADIR" "@DATADIR@") "" "@DATADIR@"))
47 (define %docdir (if (string-prefix? "@DOCDIR" "@DOCDIR@") "doc/" "@DOCDIR@"))
48 (define %moduledir "module/")
49 (define %version (if (string-prefix? "@VERSION" "@VERSION@") "git" "@VERSION@"))
50 (module-define! (resolve-module '(language c99 compiler)) '%datadir %datadir)
51 (module-define! (resolve-module '(language c99 compiler)) '%docdir %docdir)
52 (module-define! (resolve-module '(language c99 compiler)) '%moduledir %moduledir)
53 (module-define! (resolve-module '(language c99 compiler)) '%prefix %prefix)
54 (module-define! (resolve-module '(language c99 compiler)) '%version %version)
55
56 (define (parse-opts args)
57   (let* ((option-spec
58           '((c (single-char #\c))
59             (D (single-char #\D) (value #t))
60             (E (single-char #\E))
61             (help (single-char #\h))
62             (I (single-char #\I) (value #t))
63             (o (single-char #\o) (value #t))
64             (version (single-char #\V) (value #t))))
65          (options (getopt-long args option-spec))
66          (help? (option-ref options 'help #f))
67          (files (option-ref options '() '()))
68          (usage? (and (not help?) (null? files)))
69          (version? (option-ref options 'version #f)))
70     (or
71      (and version?
72           (format (current-output-port) "mescc.scm (mes) ~a\n" %version))
73      (and (or help? usage?)
74           (format (or (and usage? (current-error-port)) (current-output-port)) "\
75 Usage: mescc [OPTION]... FILE...
76   -c                 compile and assemble, but do not link
77   -D DEFINE          define DEFINE
78   -E                 preprocess only; do not compile, assemble or link
79   -h, --help         display this help and exit
80   -I DIR             append DIR to include path
81   -o FILE            write output to FILE
82   -v, --version      display version and exit
83 ")
84           (exit (or (and usage? 2) 0)))
85      options)))
86
87 (define (object->info file)
88   (let* ((lst (with-input-from-file file read))
89          (module (resolve-module '(language c99 compiler))))
90     (eval lst module)))
91
92 (define (main:ast->info file)
93   (let ((ast (with-input-from-file file read)))
94     (with-input-from-file file
95       (lambda ()
96         (c99-ast->info ast)))))
97
98 (define (source->ast defines includes)
99   (lambda (file)
100     (with-input-from-file file
101       (lambda ()
102         (pretty-print (c99-input->ast #:defines defines #:includes includes))))))
103
104 (define (source->info defines includes)
105   (lambda (file)
106     (with-input-from-file file
107       (lambda ()
108         ((c99-input->info #:defines defines #:includes includes))))))
109
110 (define (ast? o)
111   (or (string-suffix? ".E" o)
112       (string-suffix? ".guile-E" o)))
113
114 (define (object? o)
115   (or (string-suffix? ".o" o)
116       (string-suffix? ".guile-o" o)))
117
118 (define (main args)
119   (let* ((options (parse-opts args))
120          (files (option-ref options '() '()))
121          (file (if (null? files) (string-append %docdir "examples/main.c")
122                    (car files)))
123          (preprocess? (option-ref options 'E #f))
124          (compile? (option-ref options 'c #f))
125          (asts (filter ast? files))
126          (objects (filter object? files))
127          (sources (filter (cut string-suffix? ".c" <>) files))
128          (base (substring file (1+ (or (string-rindex file #\/) -1)) (- (string-length file) 2)))
129          (out (option-ref options 'o (cond (compile? (string-append base ".o"))
130                                            (preprocess? (string-append base ".E"))
131                                            (else "a.out"))))
132          (multi-opt (lambda (option) (lambda (o) (and (eq? (car o) option) (cdr o)))))
133          (defines (reverse (filter-map (multi-opt 'D) options)))
134          (includes (reverse (filter-map (multi-opt 'I) options))))
135     (when (getenv "MES_DEBUG") (format (current-error-port) "options=~s\n" options)
136           (format (current-error-port) "output: ~a\n" out))
137     (if (and (pair? sources) (pair? objects)) (error "cannot mix source and object files:" files))
138     (with-output-to-file out
139       (lambda ()
140         (if (and (not compile?)
141                  (not preprocess?)) (set-port-encoding! (current-output-port) "ISO-8859-1"))
142         (cond ((pair? objects) (let ((infos (map object->info objects)))
143                                  (if compile? (infos->object infos)
144                                      (infos->elf infos))))
145               ((pair? asts) (let ((infos (map main:ast->info asts)))
146                               (if compile? (infos->object infos)
147                                   (infos->elf infos))))
148               ((pair? sources) (if preprocess? (map (source->ast defines includes) sources)
149                                    (let ((infos (map (source->info defines includes) sources)))
150                                      (if compile? (infos->object infos)
151                                          (infos->elf infos))))))))
152     (if (and (not compile?)
153              (not preprocess?))
154         (chmod out #o755))))