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