src: Compiler, interpreter and parser updates for UTF-8
[nit.git] / src / compiler / abstract_compiler.nit
1 # This file is part of NIT ( http://www.nitlanguage.org ).
2 #
3 # Copyright 2012 Jean Privat <jean@pryen.org>
4 #
5 # Licensed under the Apache License, Version 2.0 (the "License");
6 # you may not use this file except in compliance with the License.
7 # You may obtain a copy of the License at
8 #
9 # http://www.apache.org/licenses/LICENSE-2.0
10 #
11 # Unless required by applicable law or agreed to in writing, software
12 # distributed under the License is distributed on an "AS IS" BASIS,
13 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14 # See the License for the specific language governing permissions and
15 # limitations under the License.
16
17 # Abstract compiler
18 module abstract_compiler
19
20 import literal
21 import semantize
22 import platform
23 import c_tools
24 private import annotation
25 import mixin
26 import counter
27
28 # Add compiling options
29 redef class ToolContext
30 # --output
31 var opt_output = new OptionString("Output file", "-o", "--output")
32 # --dir
33 var opt_dir = new OptionString("Output directory", "--dir")
34 # --no-cc
35 var opt_no_cc = new OptionBool("Do not invoke C compiler", "--no-cc")
36 # --no-main
37 var opt_no_main = new OptionBool("Do not generate main entry point", "--no-main")
38 # --make-flags
39 var opt_make_flags = new OptionString("Additional options to make", "--make-flags")
40 # --max-c-lines
41 var opt_max_c_lines = new OptionInt("Maximum number of lines in generated C files. Use 0 for unlimited", 10000, "--max-c-lines")
42 # --group-c-files
43 var opt_group_c_files = new OptionBool("Group all generated code in the same series of files", "--group-c-files")
44 # --compile-dir
45 var opt_compile_dir = new OptionString("Directory used to generate temporary files", "--compile-dir")
46 # --hardening
47 var opt_hardening = new OptionBool("Generate contracts in the C code against bugs in the compiler", "--hardening")
48 # --no-check-covariance
49 var opt_no_check_covariance = new OptionBool("Disable type tests of covariant parameters (dangerous)", "--no-check-covariance")
50 # --no-check-attr-isset
51 var opt_no_check_attr_isset = new OptionBool("Disable isset tests before each attribute access (dangerous)", "--no-check-attr-isset")
52 # --no-check-assert
53 var opt_no_check_assert = new OptionBool("Disable the evaluation of explicit 'assert' and 'as' (dangerous)", "--no-check-assert")
54 # --no-check-autocast
55 var opt_no_check_autocast = new OptionBool("Disable implicit casts on unsafe expression usage (dangerous)", "--no-check-autocast")
56 # --no-check-null
57 var opt_no_check_null = new OptionBool("Disable tests of null receiver (dangerous)", "--no-check-null")
58 # --no-check-all
59 var opt_no_check_all = new OptionBool("Disable all tests (dangerous)", "--no-check-all")
60 # --typing-test-metrics
61 var opt_typing_test_metrics = new OptionBool("Enable static and dynamic count of all type tests", "--typing-test-metrics")
62 # --invocation-metrics
63 var opt_invocation_metrics = new OptionBool("Enable static and dynamic count of all method invocations", "--invocation-metrics")
64 # --isset-checks-metrics
65 var opt_isset_checks_metrics = new OptionBool("Enable static and dynamic count of isset checks before attributes access", "--isset-checks-metrics")
66 # --no-stacktrace
67 var opt_no_stacktrace = new OptionBool("Disable the generation of stack traces", "--no-stacktrace")
68 # --no-gcc-directives
69 var opt_no_gcc_directive = new OptionArray("Disable a advanced gcc directives for optimization", "--no-gcc-directive")
70 # --release
71 var opt_release = new OptionBool("Compile in release mode and finalize application", "--release")
72
73 redef init
74 do
75 super
76 self.option_context.add_option(self.opt_output, self.opt_dir, self.opt_no_cc, self.opt_no_main, self.opt_make_flags, self.opt_compile_dir, self.opt_hardening)
77 self.option_context.add_option(self.opt_no_check_covariance, self.opt_no_check_attr_isset, self.opt_no_check_assert, self.opt_no_check_autocast, self.opt_no_check_null, self.opt_no_check_all)
78 self.option_context.add_option(self.opt_typing_test_metrics, self.opt_invocation_metrics, self.opt_isset_checks_metrics)
79 self.option_context.add_option(self.opt_no_stacktrace)
80 self.option_context.add_option(self.opt_no_gcc_directive)
81 self.option_context.add_option(self.opt_release)
82 self.option_context.add_option(self.opt_max_c_lines, self.opt_group_c_files)
83
84 opt_no_main.hidden = true
85 end
86
87 redef fun process_options(args)
88 do
89 super
90
91 if opt_output.value != null and opt_dir.value != null then
92 print "Option Error: cannot use both --dir and --output"
93 exit(1)
94 end
95
96 if opt_no_check_all.value then
97 opt_no_check_covariance.value = true
98 opt_no_check_attr_isset.value = true
99 opt_no_check_assert.value = true
100 opt_no_check_autocast.value = true
101 opt_no_check_null.value = true
102 end
103 end
104 end
105
106 redef class ModelBuilder
107 # Simple indirection to `Toolchain::write_and_make`
108 protected fun write_and_make(compiler: AbstractCompiler)
109 do
110 var platform = compiler.target_platform
111 var toolchain = platform.toolchain(toolcontext, compiler)
112 compiler.toolchain = toolchain
113 toolchain.write_and_make
114 end
115 end
116
117 redef class Platform
118 # The specific tool-chain associated to the platform
119 fun toolchain(toolcontext: ToolContext, compiler: AbstractCompiler): Toolchain
120 do
121 return new MakefileToolchain(toolcontext, compiler)
122 end
123 end
124
125 # Build toolchain for a specific target program, varies per `Platform`
126 class Toolchain
127
128 # Toolcontext
129 var toolcontext: ToolContext
130
131 # Compiler of the target program
132 var compiler: AbstractCompiler
133
134 # Directory where to generate all files
135 #
136 # The option `--compile_dir` change this directory.
137 fun root_compile_dir: String
138 do
139 var compile_dir = toolcontext.opt_compile_dir.value
140 if compile_dir == null then compile_dir = "nit_compile"
141 return compile_dir
142 end
143
144 # Directory where to generate all C files
145 #
146 # By default it is `root_compile_dir` but some platform may require that it is a subdirectory.
147 fun compile_dir: String do return root_compile_dir
148
149 # Write all C files and compile them
150 fun write_and_make is abstract
151 end
152
153 # Default toolchain using a Makefile
154 class MakefileToolchain
155 super Toolchain
156
157 redef fun write_and_make
158 do
159 var compile_dir = compile_dir
160
161 # Remove the compilation directory unless explicitly set
162 var auto_remove = toolcontext.opt_compile_dir.value == null
163
164 # Generate the .h and .c files
165 # A single C file regroups many compiled rumtime functions
166 # Note that we do not try to be clever an a small change in a Nit source file may change the content of all the generated .c files
167 var time0 = get_time
168 self.toolcontext.info("*** WRITING C ***", 1)
169
170 root_compile_dir.mkdir
171 compile_dir.mkdir
172
173 var cfiles = new Array[String]
174 write_files(compile_dir, cfiles)
175
176 # Generate the Makefile
177
178 write_makefile(compile_dir, cfiles)
179
180 var time1 = get_time
181 self.toolcontext.info("*** END WRITING C: {time1-time0} ***", 2)
182
183 # Execute the Makefile
184
185 if self.toolcontext.opt_no_cc.value then return
186
187 time0 = time1
188 self.toolcontext.info("*** COMPILING C ***", 1)
189
190 compile_c_code(compile_dir)
191
192 if auto_remove then
193 sys.system("rm -r -- '{root_compile_dir.escape_to_sh}/'")
194 end
195
196 time1 = get_time
197 self.toolcontext.info("*** END COMPILING C: {time1-time0} ***", 2)
198 end
199
200 # Write all source files to the `compile_dir`
201 fun write_files(compile_dir: String, cfiles: Array[String])
202 do
203 var platform = compiler.target_platform
204 if platform.supports_libunwind then compiler.build_c_to_nit_bindings
205 var cc_opt_with_libgc = "-DWITH_LIBGC"
206 if not platform.supports_libgc then cc_opt_with_libgc = ""
207
208 # Add gc_choser.h to aditionnal bodies
209 var gc_chooser = new ExternCFile("gc_chooser.c", cc_opt_with_libgc)
210 if cc_opt_with_libgc != "" then gc_chooser.pkgconfigs.add "bdw-gc"
211 compiler.extern_bodies.add(gc_chooser)
212 var clib = toolcontext.nit_dir / "clib"
213 compiler.files_to_copy.add "{clib}/gc_chooser.c"
214 compiler.files_to_copy.add "{clib}/gc_chooser.h"
215
216 # FFI
217 for m in compiler.mainmodule.in_importation.greaters do
218 compiler.finalize_ffi_for_module(m)
219 end
220
221 # Copy original .[ch] files to compile_dir
222 for src in compiler.files_to_copy do
223 var basename = src.basename
224 var dst = "{compile_dir}/{basename}"
225 src.file_copy_to dst
226 end
227
228 var hfilename = compiler.header.file.name + ".h"
229 var hfilepath = "{compile_dir}/{hfilename}"
230 var h = new FileWriter.open(hfilepath)
231 for l in compiler.header.decl_lines do
232 h.write l
233 h.write "\n"
234 end
235 for l in compiler.header.lines do
236 h.write l
237 h.write "\n"
238 end
239 h.close
240
241 var max_c_lines = toolcontext.opt_max_c_lines.value
242 for f in compiler.files do
243 var i = 0
244 var count = 0
245 var file: nullable FileWriter = null
246 for vis in f.writers do
247 if vis == compiler.header then continue
248 var total_lines = vis.lines.length + vis.decl_lines.length
249 if total_lines == 0 then continue
250 count += total_lines
251 if file == null or (count > max_c_lines and max_c_lines > 0) then
252 i += 1
253 if file != null then file.close
254 var cfilename = "{f.name}.{i}.c"
255 var cfilepath = "{compile_dir}/{cfilename}"
256 self.toolcontext.info("new C source files to compile: {cfilepath}", 3)
257 cfiles.add(cfilename)
258 file = new FileWriter.open(cfilepath)
259 file.write "#include \"{f.name}.0.h\"\n"
260 count = total_lines
261 end
262 for l in vis.decl_lines do
263 file.write l
264 file.write "\n"
265 end
266 for l in vis.lines do
267 file.write l
268 file.write "\n"
269 end
270 end
271 if file == null then continue
272 file.close
273
274 var cfilename = "{f.name}.0.h"
275 var cfilepath = "{compile_dir}/{cfilename}"
276 var hfile: nullable FileWriter = null
277 hfile = new FileWriter.open(cfilepath)
278 hfile.write "#include \"{hfilename}\"\n"
279 for key in f.required_declarations do
280 if not compiler.provided_declarations.has_key(key) then
281 var node = compiler.requirers_of_declarations.get_or_null(key)
282 if node != null then
283 node.debug "No provided declaration for {key}"
284 else
285 print "No provided declaration for {key}"
286 end
287 abort
288 end
289 hfile.write compiler.provided_declarations[key]
290 hfile.write "\n"
291 end
292 hfile.close
293 end
294
295 self.toolcontext.info("Total C source files to compile: {cfiles.length}", 2)
296 end
297
298 # Get the name of the Makefile to use
299 fun makefile_name: String do return "{compiler.mainmodule.c_name}.mk"
300
301 # Get the default name of the executable to produce
302 fun default_outname: String
303 do
304 var mainmodule = compiler.mainmodule.first_real_mmodule
305 return mainmodule.name
306 end
307
308 # Combine options and platform informations to get the final path of the outfile
309 fun outfile(mainmodule: MModule): String
310 do
311 var res = self.toolcontext.opt_output.value
312 if res != null then return res
313 res = default_outname
314 var dir = self.toolcontext.opt_dir.value
315 if dir != null then return dir.join_path(res)
316 return res
317 end
318
319 # Write the Makefile
320 fun write_makefile(compile_dir: String, cfiles: Array[String])
321 do
322 var mainmodule = compiler.mainmodule
323 var platform = compiler.target_platform
324
325 var outname = outfile(mainmodule)
326
327 var real_outpath = compile_dir.relpath(outname)
328 var outpath = real_outpath.escape_to_mk
329 if outpath != real_outpath then
330 # If the name is crazy and need escaping, we will do an indirection
331 # 1. generate the binary in the nit_compile dir under an escaped name
332 # 2. copy the binary at the right place in the `all` goal.
333 outpath = mainmodule.c_name
334 end
335 var makename = makefile_name
336 var makepath = "{compile_dir}/{makename}"
337 var makefile = new FileWriter.open(makepath)
338
339 var linker_options = new HashSet[String]
340 for m in mainmodule.in_importation.greaters do
341 var libs = m.collect_linker_libs
342 if libs != null then linker_options.add_all(libs)
343 end
344
345 makefile.write("CC = ccache cc\nCXX = ccache c++\nCFLAGS = -g -O2 -Wno-unused-value -Wno-switch -Wno-attributes\nCINCL =\nLDFLAGS ?= \nLDLIBS ?= -lm {linker_options.join(" ")}\n\n")
346
347 makefile.write "\n# SPECIAL CONFIGURATION FLAGS\n"
348 if platform.supports_libunwind then
349 if toolcontext.opt_no_stacktrace.value then
350 makefile.write "NO_STACKTRACE=True"
351 else
352 makefile.write "NO_STACKTRACE= # Set to `True` to enable"
353 end
354 end
355
356 # Dynamic adaptations
357 # While `platform` enable complex toolchains, they are statically applied
358 # For a dynamic adaptsation of the compilation, the generated Makefile should check and adapt things itself
359 makefile.write "\n\n"
360
361 # Check and adapt the targeted system
362 makefile.write("uname_S := $(shell sh -c 'uname -s 2>/dev/null || echo not')\n")
363
364 # Check and adapt for the compiler used
365 # clang need an additionnal `-Qunused-arguments`
366 makefile.write("clang_check := $(shell sh -c '$(CC) -v 2>&1 | grep -q clang; echo $$?')\nifeq ($(clang_check), 0)\n\tCFLAGS += -Qunused-arguments\nendif\n")
367
368 if platform.supports_libunwind then
369 makefile.write """
370 ifneq ($(NO_STACKTRACE), True)
371 # Check and include lib-unwind in a portable way
372 ifneq ($(uname_S),Darwin)
373 # already included on macosx, but need to get the correct flags in other supported platforms.
374 ifeq ($(shell pkg-config --exists 'libunwind'; echo $$?), 0)
375 LDLIBS += `pkg-config --libs libunwind`
376 CFLAGS += `pkg-config --cflags libunwind`
377 else
378 $(warning "[_] stack-traces disabled. Please install libunwind-dev.")
379 CFLAGS += -D NO_STACKTRACE
380 endif
381 endif
382 else
383 # Stacktraces disabled
384 CFLAGS += -D NO_STACKTRACE
385 endif
386
387 """
388 else
389 makefile.write("CFLAGS += -D NO_STACKTRACE\n\n")
390 end
391
392 makefile.write("all: {outpath}\n")
393 if outpath != real_outpath then
394 makefile.write("\tcp -- {outpath.escape_to_sh} {real_outpath.escape_to_sh.replace("$","$$")}")
395 end
396 makefile.write("\n")
397
398 var ofiles = new Array[String]
399 var dep_rules = new Array[String]
400 # Compile each generated file
401 for f in cfiles do
402 var o = f.strip_extension(".c") + ".o"
403 makefile.write("{o}: {f}\n\t$(CC) $(CFLAGS) $(CINCL) -c -o {o} {f}\n\n")
404 ofiles.add(o)
405 dep_rules.add(o)
406 end
407
408 # Generate linker script, if any
409 if not compiler.linker_script.is_empty then
410 var linker_script_path = "{compile_dir}/linker_script"
411 ofiles.add "linker_script"
412 var f = new FileWriter.open(linker_script_path)
413 for l in compiler.linker_script do
414 f.write l
415 f.write "\n"
416 end
417 f.close
418 end
419
420 var java_files = new Array[ExternFile]
421
422 var pkgconfigs = new Array[String]
423 for f in compiler.extern_bodies do
424 pkgconfigs.add_all f.pkgconfigs
425 end
426 # Protect pkg-config
427 if not pkgconfigs.is_empty then
428 makefile.write """
429 # does pkg-config exists?
430 ifneq ($(shell which pkg-config >/dev/null; echo $$?), 0)
431 $(error "Command `pkg-config` not found. Please install it")
432 endif
433 """
434 for p in pkgconfigs do
435 makefile.write """
436 # Check for library {{{p}}}
437 ifneq ($(shell pkg-config --exists '{{{p}}}'; echo $$?), 0)
438 $(error "pkg-config: package {{{p}}} is not found.")
439 endif
440 """
441 end
442 end
443
444 # Compile each required extern body into a specific .o
445 for f in compiler.extern_bodies do
446 var o = f.makefile_rule_name
447 var ff = f.filename.basename
448 makefile.write("{o}: {ff}\n")
449 makefile.write("\t{f.makefile_rule_content}\n\n")
450 dep_rules.add(f.makefile_rule_name)
451
452 if f.compiles_to_o_file then ofiles.add(o)
453 if f.add_to_jar then java_files.add(f)
454 end
455
456 if not java_files.is_empty then
457 var jar_file = "{outpath}.jar"
458
459 var class_files_array = new Array[String]
460 for f in java_files do class_files_array.add(f.makefile_rule_name)
461 var class_files = class_files_array.join(" ")
462
463 makefile.write("{jar_file}: {class_files}\n")
464 makefile.write("\tjar cf {jar_file} {class_files}\n\n")
465 dep_rules.add jar_file
466 end
467
468 # Link edition
469 var pkg = ""
470 if not pkgconfigs.is_empty then
471 pkg = "`pkg-config --libs {pkgconfigs.join(" ")}`"
472 end
473 makefile.write("{outpath}: {dep_rules.join(" ")}\n\t$(CC) $(LDFLAGS) -o {outpath.escape_to_sh} {ofiles.join(" ")} $(LDLIBS) {pkg}\n\n")
474 # Clean
475 makefile.write("clean:\n\trm {ofiles.join(" ")} 2>/dev/null\n")
476 if outpath != real_outpath then
477 makefile.write("\trm -- {outpath.escape_to_sh} 2>/dev/null\n")
478 end
479 makefile.close
480 self.toolcontext.info("Generated makefile: {makepath}", 2)
481
482 makepath.file_copy_to "{compile_dir}/Makefile"
483 end
484
485 # The C code is generated, compile it to an executable
486 fun compile_c_code(compile_dir: String)
487 do
488 var makename = makefile_name
489
490 var makeflags = self.toolcontext.opt_make_flags.value
491 if makeflags == null then makeflags = ""
492
493 var command = "make -B -C {compile_dir} -f {makename} -j 4 {makeflags}"
494 self.toolcontext.info(command, 2)
495
496 var res
497 if self.toolcontext.verbose_level >= 3 then
498 res = sys.system("{command} 2>&1")
499 else
500 res = sys.system("{command} 2>&1 >/dev/null")
501 end
502 if res != 0 then
503 toolcontext.error(null, "Compilation Error: `make` failed with error code: {res}. The command was `{command}`.")
504 end
505 end
506 end
507
508 # Singleton that store the knowledge about the compilation process
509 abstract class AbstractCompiler
510 type VISITOR: AbstractCompilerVisitor
511
512 # Table corresponding c_names to nit names (methods)
513 var names = new HashMap[String, String]
514
515 # The main module of the program currently compiled
516 # Is assigned during the separate compilation
517 var mainmodule: MModule is writable
518
519 # The real main module of the program
520 var realmainmodule: MModule is noinit
521
522 # The modelbuilder used to know the model and the AST
523 var modelbuilder: ModelBuilder is protected writable
524
525 # The associated toolchain
526 #
527 # Set by `modelbuilder.write_and_make` and permit sub-routines to access the current toolchain if required.
528 var toolchain: Toolchain is noinit
529
530 # Is hardening asked? (see --hardening)
531 fun hardening: Bool do return self.modelbuilder.toolcontext.opt_hardening.value
532
533 # The targeted specific platform
534 var target_platform: Platform is noinit
535
536 init
537 do
538 self.realmainmodule = mainmodule
539 target_platform = mainmodule.target_platform or else new Platform
540 end
541
542 # Do the full code generation of the program `mainmodule`
543 # It is the main method usually called after the instantiation
544 fun do_compilation is abstract
545
546 # Force the creation of a new file
547 # The point is to avoid contamination between must-be-compiled-separately files
548 fun new_file(name: String): CodeFile
549 do
550 if modelbuilder.toolcontext.opt_group_c_files.value then
551 if self.files.is_empty then
552 var f = new CodeFile(mainmodule.c_name)
553 self.files.add(f)
554 end
555 return self.files.first
556 end
557 var f = new CodeFile(name)
558 self.files.add(f)
559 return f
560 end
561
562 # The list of all associated files
563 # Used to generate .c files
564 var files = new List[CodeFile]
565
566 # Initialize a visitor specific for a compiler engine
567 fun new_visitor: VISITOR is abstract
568
569 # Where global declaration are stored (the main .h)
570 var header: CodeWriter is writable, noinit
571
572 # Additionnal linker script for `ld`.
573 # Mainly used to do specific link-time symbol resolution
574 var linker_script = new Array[String]
575
576 # Provide a declaration that can be requested (before or latter) by a visitor
577 fun provide_declaration(key: String, s: String)
578 do
579 if self.provided_declarations.has_key(key) then
580 assert self.provided_declarations[key] == s
581 end
582 self.provided_declarations[key] = s
583 end
584
585 private var provided_declarations = new HashMap[String, String]
586
587 private var requirers_of_declarations = new HashMap[String, ANode]
588
589 # Builds the .c and .h files to be used when generating a Stack Trace
590 # Binds the generated C function names to Nit function names
591 fun build_c_to_nit_bindings
592 do
593 var compile_dir = toolchain.compile_dir
594
595 var stream = new FileWriter.open("{compile_dir}/c_functions_hash.c")
596 stream.write("#include <string.h>\n")
597 stream.write("#include <stdlib.h>\n")
598 stream.write("#include \"c_functions_hash.h\"\n")
599 stream.write("typedef struct C_Nit_Names\{char* name; char* nit_name;\}C_Nit_Names;\n")
600 stream.write("const char* get_nit_name(register const char* procproc, register unsigned int len)\{\n")
601 stream.write("char* procname = malloc(len+1);")
602 stream.write("memcpy(procname, procproc, len);")
603 stream.write("procname[len] = '\\0';")
604 stream.write("static const C_Nit_Names map[{names.length}] = \{\n")
605 for i in names.keys do
606 stream.write("\{\"")
607 stream.write(i.escape_to_c)
608 stream.write("\",\"")
609 stream.write(names[i].escape_to_c)
610 stream.write("\"\},\n")
611 end
612 stream.write("\};\n")
613 stream.write("int i;")
614 stream.write("for(i = 0; i < {names.length}; i++)\{")
615 stream.write("if(strcmp(procname,map[i].name) == 0)\{")
616 stream.write("free(procname);")
617 stream.write("return map[i].nit_name;")
618 stream.write("\}")
619 stream.write("\}")
620 stream.write("free(procname);")
621 stream.write("return NULL;")
622 stream.write("\}\n")
623 stream.close
624
625 stream = new FileWriter.open("{compile_dir}/c_functions_hash.h")
626 stream.write("const char* get_nit_name(register const char* procname, register unsigned int len);\n")
627 stream.close
628
629 extern_bodies.add(new ExternCFile("{compile_dir}/c_functions_hash.c", ""))
630 end
631
632 # Compile C headers
633 # This method call compile_header_strucs method that has to be refined
634 fun compile_header do
635 self.header.add_decl("#include <stdlib.h>")
636 self.header.add_decl("#include <stdio.h>")
637 self.header.add_decl("#include <string.h>")
638 self.header.add_decl("#include <sys/types.h>\n")
639 self.header.add_decl("#include <unistd.h>\n")
640 self.header.add_decl("#include <stdint.h>\n")
641 self.header.add_decl("#include \"gc_chooser.h\"")
642 self.header.add_decl("#ifdef ANDROID")
643 self.header.add_decl(" #include <android/log.h>")
644 self.header.add_decl(" #define PRINT_ERROR(...) (void)__android_log_print(ANDROID_LOG_WARN, \"Nit\", __VA_ARGS__)")
645 self.header.add_decl("#else")
646 self.header.add_decl(" #define PRINT_ERROR(...) fprintf(stderr, __VA_ARGS__)")
647 self.header.add_decl("#endif")
648
649 compile_header_structs
650 compile_nitni_structs
651
652 var gccd_disable = modelbuilder.toolcontext.opt_no_gcc_directive.value
653 if gccd_disable.has("noreturn") or gccd_disable.has("all") then
654 # Signal handler function prototype
655 self.header.add_decl("void fatal_exit(int);")
656 else
657 self.header.add_decl("void fatal_exit(int) __attribute__ ((noreturn));")
658 end
659
660 if gccd_disable.has("likely") or gccd_disable.has("all") then
661 self.header.add_decl("#define likely(x) (x)")
662 self.header.add_decl("#define unlikely(x) (x)")
663 else if gccd_disable.has("correct-likely") then
664 # invert the `likely` definition
665 # Used by masochists to bench the worst case
666 self.header.add_decl("#define likely(x) __builtin_expect((x),0)")
667 self.header.add_decl("#define unlikely(x) __builtin_expect((x),1)")
668 else
669 self.header.add_decl("#define likely(x) __builtin_expect((x),1)")
670 self.header.add_decl("#define unlikely(x) __builtin_expect((x),0)")
671 end
672
673 # Global variable used by intern methods
674 self.header.add_decl("extern int glob_argc;")
675 self.header.add_decl("extern char **glob_argv;")
676 self.header.add_decl("extern val *glob_sys;")
677 end
678
679 # Declaration of structures for live Nit types
680 protected fun compile_header_structs is abstract
681
682 # Declaration of structures for nitni undelying the FFI
683 protected fun compile_nitni_structs
684 do
685 self.header.add_decl """
686 /* Native reference to Nit objects */
687 /* This structure is used to represent every Nit type in extern methods and custom C code. */
688 struct nitni_ref {
689 struct nitni_ref *next,
690 *prev; /* adjacent global references in global list */
691 int count; /* number of time this global reference has been marked */
692 };
693
694 /* List of global references from C code to Nit objects */
695 /* Instanciated empty at init of Nit system and filled explicitly by user in C code */
696 struct nitni_global_ref_list_t {
697 struct nitni_ref *head, *tail;
698 };
699 extern struct nitni_global_ref_list_t *nitni_global_ref_list;
700
701 /* Initializer of global reference list */
702 extern void nitni_global_ref_list_init();
703
704 /* Intern function to add a global reference to the list */
705 extern void nitni_global_ref_add( struct nitni_ref *ref );
706
707 /* Intern function to remove a global reference from the list */
708 extern void nitni_global_ref_remove( struct nitni_ref *ref );
709
710 /* Increase count on an existing global reference */
711 extern void nitni_global_ref_incr( struct nitni_ref *ref );
712
713 /* Decrease count on an existing global reference */
714 extern void nitni_global_ref_decr( struct nitni_ref *ref );
715 """
716 end
717
718 fun compile_finalizer_function
719 do
720 var finalizable_type = mainmodule.finalizable_type
721 if finalizable_type == null then return
722
723 var finalize_meth = mainmodule.try_get_primitive_method("finalize", finalizable_type.mclass)
724
725 if finalize_meth == null then
726 modelbuilder.toolcontext.error(null, "Error: the `Finalizable` class does not declare the `finalize` method.")
727 return
728 end
729
730 var v = self.new_visitor
731 v.add_decl "void gc_finalize (void *obj, void *client_data) \{"
732 var recv = v.new_expr("obj", finalizable_type)
733 v.send(finalize_meth, [recv])
734 v.add "\}"
735 end
736
737 # Generate the main C function.
738 #
739 # This function:
740 #
741 # * allocate the Sys object if it exists
742 # * call init if is exists
743 # * call main if it exists
744 fun compile_main_function
745 do
746 var v = self.new_visitor
747 v.add_decl("#include <signal.h>")
748 var platform = target_platform
749
750 var no_main = platform.no_main or modelbuilder.toolcontext.opt_no_main.value
751
752 if platform.supports_libunwind then
753 v.add_decl("#ifndef NO_STACKTRACE")
754 v.add_decl("#define UNW_LOCAL_ONLY")
755 v.add_decl("#include <libunwind.h>")
756 v.add_decl("#include \"c_functions_hash.h\"")
757 v.add_decl("#endif")
758 end
759 v.add_decl("int glob_argc;")
760 v.add_decl("char **glob_argv;")
761 v.add_decl("val *glob_sys;")
762
763 if self.modelbuilder.toolcontext.opt_typing_test_metrics.value then
764 for tag in count_type_test_tags do
765 v.add_decl("long count_type_test_resolved_{tag};")
766 v.add_decl("long count_type_test_unresolved_{tag};")
767 v.add_decl("long count_type_test_skipped_{tag};")
768 v.compiler.header.add_decl("extern long count_type_test_resolved_{tag};")
769 v.compiler.header.add_decl("extern long count_type_test_unresolved_{tag};")
770 v.compiler.header.add_decl("extern long count_type_test_skipped_{tag};")
771 end
772 end
773
774 if self.modelbuilder.toolcontext.opt_invocation_metrics.value then
775 v.add_decl("long count_invoke_by_tables;")
776 v.add_decl("long count_invoke_by_direct;")
777 v.add_decl("long count_invoke_by_inline;")
778 v.compiler.header.add_decl("extern long count_invoke_by_tables;")
779 v.compiler.header.add_decl("extern long count_invoke_by_direct;")
780 v.compiler.header.add_decl("extern long count_invoke_by_inline;")
781 end
782
783 if self.modelbuilder.toolcontext.opt_isset_checks_metrics.value then
784 v.add_decl("long count_attr_reads = 0;")
785 v.add_decl("long count_isset_checks = 0;")
786 v.compiler.header.add_decl("extern long count_attr_reads;")
787 v.compiler.header.add_decl("extern long count_isset_checks;")
788 end
789
790 v.add_decl("static void show_backtrace(void) \{")
791 if platform.supports_libunwind then
792 v.add_decl("#ifndef NO_STACKTRACE")
793 v.add_decl("char* opt = getenv(\"NIT_NO_STACK\");")
794 v.add_decl("unw_cursor_t cursor;")
795 v.add_decl("if(opt==NULL)\{")
796 v.add_decl("unw_context_t uc;")
797 v.add_decl("unw_word_t ip;")
798 v.add_decl("char* procname = malloc(sizeof(char) * 100);")
799 v.add_decl("unw_getcontext(&uc);")
800 v.add_decl("unw_init_local(&cursor, &uc);")
801 v.add_decl("PRINT_ERROR(\"-------------------------------------------------\\n\");")
802 v.add_decl("PRINT_ERROR(\"-- Stack Trace ------------------------------\\n\");")
803 v.add_decl("PRINT_ERROR(\"-------------------------------------------------\\n\");")
804 v.add_decl("while (unw_step(&cursor) > 0) \{")
805 v.add_decl(" unw_get_proc_name(&cursor, procname, 100, &ip);")
806 v.add_decl(" const char* recv = get_nit_name(procname, strlen(procname));")
807 v.add_decl(" if (recv != NULL)\{")
808 v.add_decl(" PRINT_ERROR(\"` %s\\n\", recv);")
809 v.add_decl(" \}else\{")
810 v.add_decl(" PRINT_ERROR(\"` %s\\n\", procname);")
811 v.add_decl(" \}")
812 v.add_decl("\}")
813 v.add_decl("PRINT_ERROR(\"-------------------------------------------------\\n\");")
814 v.add_decl("free(procname);")
815 v.add_decl("\}")
816 v.add_decl("#endif /* NO_STACKTRACE */")
817 end
818 v.add_decl("\}")
819
820 v.add_decl("void sig_handler(int signo)\{")
821 v.add_decl("PRINT_ERROR(\"Caught signal : %s\\n\", strsignal(signo));")
822 v.add_decl("show_backtrace();")
823 # rethrows
824 v.add_decl("signal(signo, SIG_DFL);")
825 v.add_decl("kill(getpid(), signo);")
826 v.add_decl("\}")
827
828 v.add_decl("void fatal_exit(int status) \{")
829 v.add_decl("show_backtrace();")
830 v.add_decl("exit(status);")
831 v.add_decl("\}")
832
833 if no_main then
834 v.add_decl("int nit_main(int argc, char** argv) \{")
835 else
836 v.add_decl("int main(int argc, char** argv) \{")
837 end
838
839 v.add("signal(SIGABRT, sig_handler);")
840 v.add("signal(SIGFPE, sig_handler);")
841 v.add("signal(SIGILL, sig_handler);")
842 v.add("signal(SIGINT, sig_handler);")
843 v.add("signal(SIGTERM, sig_handler);")
844 v.add("signal(SIGSEGV, sig_handler);")
845 v.add("signal(SIGPIPE, SIG_IGN);")
846
847 v.add("glob_argc = argc; glob_argv = argv;")
848 v.add("initialize_gc_option();")
849
850 v.add "initialize_nitni_global_refs();"
851
852 var main_type = mainmodule.sys_type
853 if main_type != null then
854 var mainmodule = v.compiler.mainmodule
855 var glob_sys = v.init_instance(main_type)
856 v.add("glob_sys = {glob_sys};")
857 var main_init = mainmodule.try_get_primitive_method("init", main_type.mclass)
858 if main_init != null then
859 v.send(main_init, [glob_sys])
860 end
861 var main_method = mainmodule.try_get_primitive_method("run", main_type.mclass) or else
862 mainmodule.try_get_primitive_method("main", main_type.mclass)
863 if main_method != null then
864 v.send(main_method, [glob_sys])
865 end
866 end
867
868 if self.modelbuilder.toolcontext.opt_typing_test_metrics.value then
869 v.add_decl("long count_type_test_resolved_total = 0;")
870 v.add_decl("long count_type_test_unresolved_total = 0;")
871 v.add_decl("long count_type_test_skipped_total = 0;")
872 v.add_decl("long count_type_test_total_total = 0;")
873 for tag in count_type_test_tags do
874 v.add_decl("long count_type_test_total_{tag};")
875 v.add("count_type_test_total_{tag} = count_type_test_resolved_{tag} + count_type_test_unresolved_{tag} + count_type_test_skipped_{tag};")
876 v.add("count_type_test_resolved_total += count_type_test_resolved_{tag};")
877 v.add("count_type_test_unresolved_total += count_type_test_unresolved_{tag};")
878 v.add("count_type_test_skipped_total += count_type_test_skipped_{tag};")
879 v.add("count_type_test_total_total += count_type_test_total_{tag};")
880 end
881 v.add("printf(\"# dynamic count_type_test: total %l\\n\");")
882 v.add("printf(\"\\tresolved\\tunresolved\\tskipped\\ttotal\\n\");")
883 var tags = count_type_test_tags.to_a
884 tags.add("total")
885 for tag in tags do
886 v.add("printf(\"{tag}\");")
887 v.add("printf(\"\\t%ld (%.2f%%)\", count_type_test_resolved_{tag}, 100.0*count_type_test_resolved_{tag}/count_type_test_total_total);")
888 v.add("printf(\"\\t%ld (%.2f%%)\", count_type_test_unresolved_{tag}, 100.0*count_type_test_unresolved_{tag}/count_type_test_total_total);")
889 v.add("printf(\"\\t%ld (%.2f%%)\", count_type_test_skipped_{tag}, 100.0*count_type_test_skipped_{tag}/count_type_test_total_total);")
890 v.add("printf(\"\\t%ld (%.2f%%)\\n\", count_type_test_total_{tag}, 100.0*count_type_test_total_{tag}/count_type_test_total_total);")
891 end
892 end
893
894 if self.modelbuilder.toolcontext.opt_invocation_metrics.value then
895 v.add_decl("long count_invoke_total;")
896 v.add("count_invoke_total = count_invoke_by_tables + count_invoke_by_direct + count_invoke_by_inline;")
897 v.add("printf(\"# dynamic count_invocation: total %ld\\n\", count_invoke_total);")
898 v.add("printf(\"by table: %ld (%.2f%%)\\n\", count_invoke_by_tables, 100.0*count_invoke_by_tables/count_invoke_total);")
899 v.add("printf(\"direct: %ld (%.2f%%)\\n\", count_invoke_by_direct, 100.0*count_invoke_by_direct/count_invoke_total);")
900 v.add("printf(\"inlined: %ld (%.2f%%)\\n\", count_invoke_by_inline, 100.0*count_invoke_by_inline/count_invoke_total);")
901 end
902
903 if self.modelbuilder.toolcontext.opt_isset_checks_metrics.value then
904 v.add("printf(\"# dynamic attribute reads: %ld\\n\", count_attr_reads);")
905 v.add("printf(\"# dynamic isset checks: %ld\\n\", count_isset_checks);")
906 end
907
908 v.add("return 0;")
909 v.add("\}")
910
911 for m in mainmodule.in_importation.greaters do
912 var f = "FILE_"+m.c_name
913 v.add "const char {f}[] = \"{m.location.file.filename.escape_to_c}\";"
914 provide_declaration(f, "extern const char {f}[];")
915 end
916 end
917
918 # Copile all C functions related to the [incr|decr]_ref features of the FFI
919 fun compile_nitni_global_ref_functions
920 do
921 var v = self.new_visitor
922 v.add """
923 struct nitni_global_ref_list_t *nitni_global_ref_list;
924 void initialize_nitni_global_refs() {
925 nitni_global_ref_list = (struct nitni_global_ref_list_t*)nit_alloc(sizeof(struct nitni_global_ref_list_t));
926 nitni_global_ref_list->head = NULL;
927 nitni_global_ref_list->tail = NULL;
928 }
929
930 void nitni_global_ref_add( struct nitni_ref *ref ) {
931 if ( nitni_global_ref_list->head == NULL ) {
932 nitni_global_ref_list->head = ref;
933 ref->prev = NULL;
934 } else {
935 nitni_global_ref_list->tail->next = ref;
936 ref->prev = nitni_global_ref_list->tail;
937 }
938 nitni_global_ref_list->tail = ref;
939
940 ref->next = NULL;
941 }
942
943 void nitni_global_ref_remove( struct nitni_ref *ref ) {
944 if ( ref->prev == NULL ) {
945 nitni_global_ref_list->head = ref->next;
946 } else {
947 ref->prev->next = ref->next;
948 }
949
950 if ( ref->next == NULL ) {
951 nitni_global_ref_list->tail = ref->prev;
952 } else {
953 ref->next->prev = ref->prev;
954 }
955 }
956
957 extern void nitni_global_ref_incr( struct nitni_ref *ref ) {
958 if ( ref->count == 0 ) /* not registered */
959 {
960 /* add to list */
961 nitni_global_ref_add( ref );
962 }
963
964 ref->count ++;
965 }
966
967 extern void nitni_global_ref_decr( struct nitni_ref *ref ) {
968 if ( ref->count == 1 ) /* was last reference */
969 {
970 /* remove from list */
971 nitni_global_ref_remove( ref );
972 }
973
974 ref->count --;
975 }
976 """
977 end
978
979 # List of additional files required to compile (FFI)
980 var extern_bodies = new Array[ExternFile]
981
982 # List of source files to copy over to the compile dir
983 var files_to_copy = new Array[String]
984
985 # This is used to avoid adding an extern file more than once
986 private var seen_extern = new ArraySet[String]
987
988 # Generate code that initialize the attributes on a new instance
989 fun generate_init_attr(v: VISITOR, recv: RuntimeVariable, mtype: MClassType)
990 do
991 var cds = mtype.collect_mclassdefs(self.mainmodule).to_a
992 self.mainmodule.linearize_mclassdefs(cds)
993 for cd in cds do
994 for npropdef in modelbuilder.collect_attr_propdef(cd) do
995 npropdef.init_expr(v, recv)
996 end
997 end
998 end
999
1000 # Generate code that check if an attribute is correctly initialized
1001 fun generate_check_attr(v: VISITOR, recv: RuntimeVariable, mtype: MClassType)
1002 do
1003 var cds = mtype.collect_mclassdefs(self.mainmodule).to_a
1004 self.mainmodule.linearize_mclassdefs(cds)
1005 for cd in cds do
1006 for npropdef in modelbuilder.collect_attr_propdef(cd) do
1007 npropdef.check_expr(v, recv)
1008 end
1009 end
1010 end
1011
1012 # stats
1013
1014 var count_type_test_tags: Array[String] = ["isa", "as", "auto", "covariance", "erasure"]
1015 var count_type_test_resolved: HashMap[String, Int] = init_count_type_test_tags
1016 var count_type_test_unresolved: HashMap[String, Int] = init_count_type_test_tags
1017 var count_type_test_skipped: HashMap[String, Int] = init_count_type_test_tags
1018
1019 protected fun init_count_type_test_tags: HashMap[String, Int]
1020 do
1021 var res = new HashMap[String, Int]
1022 for tag in count_type_test_tags do
1023 res[tag] = 0
1024 end
1025 return res
1026 end
1027
1028 # Display stats about compilation process
1029 #
1030 # Metrics used:
1031 #
1032 # * type tests against resolved types (`x isa Collection[Animal]`)
1033 # * type tests against unresolved types (`x isa Collection[E]`)
1034 # * type tests skipped
1035 # * type tests total
1036 fun display_stats
1037 do
1038 if self.modelbuilder.toolcontext.opt_typing_test_metrics.value then
1039 print "# static count_type_test"
1040 print "\tresolved:\tunresolved\tskipped\ttotal"
1041 var count_type_test_total = init_count_type_test_tags
1042 count_type_test_resolved["total"] = 0
1043 count_type_test_unresolved["total"] = 0
1044 count_type_test_skipped["total"] = 0
1045 count_type_test_total["total"] = 0
1046 for tag in count_type_test_tags do
1047 count_type_test_total[tag] = count_type_test_resolved[tag] + count_type_test_unresolved[tag] + count_type_test_skipped[tag]
1048 count_type_test_resolved["total"] += count_type_test_resolved[tag]
1049 count_type_test_unresolved["total"] += count_type_test_unresolved[tag]
1050 count_type_test_skipped["total"] += count_type_test_skipped[tag]
1051 count_type_test_total["total"] += count_type_test_total[tag]
1052 end
1053 var count_type_test = count_type_test_total["total"]
1054 var tags = count_type_test_tags.to_a
1055 tags.add("total")
1056 for tag in tags do
1057 printn tag
1058 printn "\t{count_type_test_resolved[tag]} ({div(count_type_test_resolved[tag],count_type_test)}%)"
1059 printn "\t{count_type_test_unresolved[tag]} ({div(count_type_test_unresolved[tag],count_type_test)}%)"
1060 printn "\t{count_type_test_skipped[tag]} ({div(count_type_test_skipped[tag],count_type_test)}%)"
1061 printn "\t{count_type_test_total[tag]} ({div(count_type_test_total[tag],count_type_test)}%)"
1062 print ""
1063 end
1064 end
1065 end
1066
1067 fun finalize_ffi_for_module(mmodule: MModule) do mmodule.finalize_ffi(self)
1068 end
1069
1070 # A file unit (may be more than one file if
1071 # A file unit aim to be autonomous and is made or one or more `CodeWriter`s
1072 class CodeFile
1073 var name: String
1074 var writers = new Array[CodeWriter]
1075 var required_declarations = new HashSet[String]
1076 end
1077
1078 # Where to store generated lines
1079 class CodeWriter
1080 var file: CodeFile
1081 var lines: List[String] = new List[String]
1082 var decl_lines: List[String] = new List[String]
1083
1084 # Add a line in the main part of the generated C
1085 fun add(s: String) do self.lines.add(s)
1086
1087 # Add a line in the
1088 # (used for local or global declaration)
1089 fun add_decl(s: String) do self.decl_lines.add(s)
1090
1091 init
1092 do
1093 file.writers.add(self)
1094 end
1095 end
1096
1097 # A visitor on the AST of property definition that generate the C code.
1098 abstract class AbstractCompilerVisitor
1099
1100 type COMPILER: AbstractCompiler
1101
1102 # The associated compiler
1103 var compiler: COMPILER
1104
1105 # The current visited AST node
1106 var current_node: nullable ANode = null is writable
1107
1108 # The current `StaticFrame`
1109 var frame: nullable StaticFrame = null is writable
1110
1111 # Alias for self.compiler.mainmodule.object_type
1112 fun object_type: MClassType do return self.compiler.mainmodule.object_type
1113
1114 # Alias for self.compiler.mainmodule.bool_type
1115 fun bool_type: MClassType do return self.compiler.mainmodule.bool_type
1116
1117 var writer: CodeWriter is noinit
1118
1119 init
1120 do
1121 self.writer = new CodeWriter(compiler.files.last)
1122 end
1123
1124 # Force to get the primitive property named `name` in the instance `recv` or abort
1125 fun get_property(name: String, recv: MType): MMethod
1126 do
1127 assert recv isa MClassType
1128 return self.compiler.modelbuilder.force_get_primitive_method(self.current_node, name, recv.mclass, self.compiler.mainmodule)
1129 end
1130
1131 fun compile_callsite(callsite: CallSite, arguments: Array[RuntimeVariable]): nullable RuntimeVariable
1132 do
1133 var initializers = callsite.mpropdef.initializers
1134 if not initializers.is_empty then
1135 var recv = arguments.first
1136
1137 var i = 1
1138 for p in initializers do
1139 if p isa MMethod then
1140 var args = [recv]
1141 for x in p.intro.msignature.mparameters do
1142 args.add arguments[i]
1143 i += 1
1144 end
1145 self.send(p, args)
1146 else if p isa MAttribute then
1147 self.write_attribute(p, recv, arguments[i])
1148 i += 1
1149 else abort
1150 end
1151 assert i == arguments.length
1152
1153 return self.send(callsite.mproperty, [recv])
1154 end
1155
1156 return self.send(callsite.mproperty, arguments)
1157 end
1158
1159 fun native_array_instance(elttype: MType, length: RuntimeVariable): RuntimeVariable is abstract
1160
1161 fun calloc_array(ret_type: MType, arguments: Array[RuntimeVariable]) is abstract
1162
1163 fun native_array_def(pname: String, ret_type: nullable MType, arguments: Array[RuntimeVariable]) is abstract
1164
1165 # Return an element of a native array.
1166 # The method is unsafe and is just a direct wrapper for the specific implementation of native arrays
1167 fun native_array_get(native_array: RuntimeVariable, index: Int): RuntimeVariable is abstract
1168
1169 # Store an element in a native array.
1170 # The method is unsafe and is just a direct wrapper for the specific implementation of native arrays
1171 fun native_array_set(native_array: RuntimeVariable, index: Int, value: RuntimeVariable) is abstract
1172
1173 # Evaluate `args` as expressions in the call of `mpropdef` on `recv`.
1174 # This method is used to manage varargs in signatures and returns the real array
1175 # of runtime variables to use in the call.
1176 fun varargize(mpropdef: MMethodDef, map: nullable SignatureMap, recv: RuntimeVariable, args: SequenceRead[AExpr]): Array[RuntimeVariable]
1177 do
1178 var msignature = mpropdef.new_msignature or else mpropdef.msignature.as(not null)
1179 var res = new Array[RuntimeVariable]
1180 res.add(recv)
1181
1182 if msignature.arity == 0 then return res
1183
1184 if map == null then
1185 assert args.length == msignature.arity
1186 for ne in args do
1187 res.add self.expr(ne, null)
1188 end
1189 return res
1190 end
1191
1192 # Eval in order of arguments, not parameters
1193 var exprs = new Array[RuntimeVariable].with_capacity(args.length)
1194 for ne in args do
1195 exprs.add self.expr(ne, null)
1196 end
1197
1198 # Fill `res` with the result of the evaluation according to the mapping
1199 for i in [0..msignature.arity[ do
1200 var param = msignature.mparameters[i]
1201 var j = map.map.get_or_null(i)
1202 if j == null then
1203 # default value
1204 res.add(null_instance)
1205 continue
1206 end
1207 if param.is_vararg and map.vararg_decl > 0 then
1208 var vararg = exprs.sub(j, map.vararg_decl)
1209 var elttype = param.mtype
1210 var arg = self.vararg_instance(mpropdef, recv, vararg, elttype)
1211 res.add(arg)
1212 continue
1213 end
1214 res.add exprs[j]
1215 end
1216 return res
1217 end
1218
1219 # Type handling
1220
1221 # Anchor a type to the main module and the current receiver
1222 fun anchor(mtype: MType): MType
1223 do
1224 if not mtype.need_anchor then return mtype
1225 return mtype.anchor_to(self.compiler.mainmodule, self.frame.receiver)
1226 end
1227
1228 fun resolve_for(mtype: MType, recv: RuntimeVariable): MType
1229 do
1230 if not mtype.need_anchor then return mtype
1231 return mtype.resolve_for(recv.mcasttype, self.frame.receiver, self.compiler.mainmodule, true)
1232 end
1233
1234 # Unsafely cast a value to a new type
1235 # ie the result share the same C variable but my have a different mcasttype
1236 # NOTE: if the adaptation is useless then `value` is returned as it.
1237 # ENSURE: `result.name == value.name`
1238 fun autoadapt(value: RuntimeVariable, mtype: MType): RuntimeVariable
1239 do
1240 mtype = self.anchor(mtype)
1241 var valmtype = value.mcasttype
1242 if valmtype.is_subtype(self.compiler.mainmodule, null, mtype) then
1243 return value
1244 end
1245
1246 if valmtype isa MNullableType and valmtype.mtype.is_subtype(self.compiler.mainmodule, null, mtype) then
1247 var res = new RuntimeVariable(value.name, valmtype, valmtype.mtype)
1248 return res
1249 else
1250 var res = new RuntimeVariable(value.name, valmtype, mtype)
1251 return res
1252 end
1253 end
1254
1255 # Generate a super call from a method definition
1256 fun supercall(m: MMethodDef, recvtype: MClassType, args: Array[RuntimeVariable]): nullable RuntimeVariable is abstract
1257
1258 # Adapt the arguments of a method according to targetted `MMethodDef`
1259 fun adapt_signature(m: MMethodDef, args: Array[RuntimeVariable]) is abstract
1260
1261 # Unbox all the arguments of a method when implemented `extern` or `intern`
1262 fun unbox_signature_extern(m: MMethodDef, args: Array[RuntimeVariable]) is abstract
1263
1264 # Box or unbox a value to another type iff a C type conversion is needed
1265 # ENSURE: `result.mtype.ctype == mtype.ctype`
1266 fun autobox(value: RuntimeVariable, mtype: MType): RuntimeVariable is abstract
1267
1268 # Box extern classes to be used in the generated code
1269 fun box_extern(value: RuntimeVariable, mtype: MType): RuntimeVariable is abstract
1270
1271 # Unbox extern classes to be used in extern code (legacy NI and FFI)
1272 fun unbox_extern(value: RuntimeVariable, mtype: MType): RuntimeVariable is abstract
1273
1274 # Generate a polymorphic subtype test
1275 fun type_test(value: RuntimeVariable, mtype: MType, tag: String): RuntimeVariable is abstract
1276
1277 # Generate the code required to dynamically check if 2 objects share the same runtime type
1278 fun is_same_type_test(value1, value2: RuntimeVariable): RuntimeVariable is abstract
1279
1280 # Generate a Nit "is" for two runtime_variables
1281 fun equal_test(value1, value2: RuntimeVariable): RuntimeVariable is abstract
1282
1283 # Sends
1284
1285 # Generate a static call on a method definition
1286 fun call(m: MMethodDef, recvtype: MClassType, args: Array[RuntimeVariable]): nullable RuntimeVariable is abstract
1287
1288 # Generate a polymorphic send for the method `m` and the arguments `args`
1289 fun send(m: MMethod, args: Array[RuntimeVariable]): nullable RuntimeVariable is abstract
1290
1291 # Generate a monomorphic send for the method `m`, the type `t` and the arguments `args`
1292 fun monomorphic_send(m: MMethod, t: MType, args: Array[RuntimeVariable]): nullable RuntimeVariable
1293 do
1294 assert t isa MClassType
1295 var propdef = m.lookup_first_definition(self.compiler.mainmodule, t)
1296 return self.call(propdef, t, args)
1297 end
1298
1299 # Generate a monomorphic super send from the method `m`, the type `t` and the arguments `args`
1300 fun monomorphic_super_send(m: MMethodDef, t: MType, args: Array[RuntimeVariable]): nullable RuntimeVariable
1301 do
1302 assert t isa MClassType
1303 m = m.lookup_next_definition(self.compiler.mainmodule, t)
1304 return self.call(m, t, args)
1305 end
1306
1307 # Attributes handling
1308
1309 # Generate a polymorphic attribute is_set test
1310 fun isset_attribute(a: MAttribute, recv: RuntimeVariable): RuntimeVariable is abstract
1311
1312 # Generate a polymorphic attribute read
1313 fun read_attribute(a: MAttribute, recv: RuntimeVariable): RuntimeVariable is abstract
1314
1315 # Generate a polymorphic attribute write
1316 fun write_attribute(a: MAttribute, recv: RuntimeVariable, value: RuntimeVariable) is abstract
1317
1318 # Checks
1319
1320 # Add a check and an abort for a null receiver if needed
1321 fun check_recv_notnull(recv: RuntimeVariable)
1322 do
1323 if self.compiler.modelbuilder.toolcontext.opt_no_check_null.value then return
1324
1325 var maybenull = recv.mcasttype isa MNullableType or recv.mcasttype isa MNullType
1326 if maybenull then
1327 self.add("if (unlikely({recv} == NULL)) \{")
1328 self.add_abort("Receiver is null")
1329 self.add("\}")
1330 end
1331 end
1332
1333 # Names handling
1334
1335 private var names = new HashSet[String]
1336 private var last: Int = 0
1337
1338 # Return a new name based on `s` and unique in the visitor
1339 fun get_name(s: String): String
1340 do
1341 if not self.names.has(s) then
1342 self.names.add(s)
1343 return s
1344 end
1345 var i = self.last + 1
1346 loop
1347 var s2 = s + i.to_s
1348 if not self.names.has(s2) then
1349 self.last = i
1350 self.names.add(s2)
1351 return s2
1352 end
1353 i = i + 1
1354 end
1355 end
1356
1357 # Return an unique and stable identifier associated with an escapemark
1358 fun escapemark_name(e: nullable EscapeMark): String
1359 do
1360 assert e != null
1361 if frame.escapemark_names.has_key(e) then return frame.escapemark_names[e]
1362 var name = e.name
1363 if name == null then name = "label"
1364 name = get_name(name)
1365 frame.escapemark_names[e] = name
1366 return name
1367 end
1368
1369 # Insert a C label for associated with an escapemark
1370 fun add_escape_label(e: nullable EscapeMark)
1371 do
1372 if e == null then return
1373 if e.escapes.is_empty then return
1374 add("BREAK_{escapemark_name(e)}: (void)0;")
1375 end
1376
1377 # Return a "const char*" variable associated to the classname of the dynamic type of an object
1378 # NOTE: we do not return a `RuntimeVariable` "NativeString" as the class may not exist in the module/program
1379 fun class_name_string(value: RuntimeVariable): String is abstract
1380
1381 # Variables handling
1382
1383 protected var variables = new HashMap[Variable, RuntimeVariable]
1384
1385 # Return the local runtime_variable associated to a Nit local variable
1386 fun variable(variable: Variable): RuntimeVariable
1387 do
1388 if self.variables.has_key(variable) then
1389 return self.variables[variable]
1390 else
1391 var name = self.get_name("var_{variable.name}")
1392 var mtype = variable.declared_type.as(not null)
1393 mtype = self.anchor(mtype)
1394 var res = new RuntimeVariable(name, mtype, mtype)
1395 self.add_decl("{mtype.ctype} {name} /* var {variable}: {mtype} */;")
1396 self.variables[variable] = res
1397 return res
1398 end
1399 end
1400
1401 # Return a new uninitialized local runtime_variable
1402 fun new_var(mtype: MType): RuntimeVariable
1403 do
1404 mtype = self.anchor(mtype)
1405 var name = self.get_name("var")
1406 var res = new RuntimeVariable(name, mtype, mtype)
1407 self.add_decl("{mtype.ctype} {name} /* : {mtype} */;")
1408 return res
1409 end
1410
1411 # The difference with `new_var` is the C static type of the local variable
1412 fun new_var_extern(mtype: MType): RuntimeVariable
1413 do
1414 mtype = self.anchor(mtype)
1415 var name = self.get_name("var")
1416 var res = new RuntimeVariable(name, mtype, mtype)
1417 self.add_decl("{mtype.ctype_extern} {name} /* : {mtype} for extern */;")
1418 return res
1419 end
1420
1421 # Return a new uninitialized named runtime_variable
1422 fun new_named_var(mtype: MType, name: String): RuntimeVariable
1423 do
1424 mtype = self.anchor(mtype)
1425 var res = new RuntimeVariable(name, mtype, mtype)
1426 self.add_decl("{mtype.ctype} {name} /* : {mtype} */;")
1427 return res
1428 end
1429
1430 # Correctly assign a left and a right value
1431 # Boxing and unboxing is performed if required
1432 fun assign(left, right: RuntimeVariable)
1433 do
1434 right = self.autobox(right, left.mtype)
1435 self.add("{left} = {right};")
1436 end
1437
1438 # Generate instances
1439
1440 # Generate a alloc-instance + init-attributes
1441 fun init_instance(mtype: MClassType): RuntimeVariable is abstract
1442
1443 # Allocate and init attributes of an instance of a standard or extern class
1444 #
1445 # Does not support universals and the pseudo-internal `NativeArray` class.
1446 fun init_instance_or_extern(mtype: MClassType): RuntimeVariable
1447 do
1448 var recv
1449 var ctype = mtype.ctype
1450 assert mtype.mclass.name != "NativeArray"
1451 if not mtype.is_c_primitive then
1452 recv = init_instance(mtype)
1453 else if ctype == "char*" then
1454 recv = new_expr("NULL/*special!*/", mtype)
1455 else
1456 recv = new_expr("({ctype})0/*special!*/", mtype)
1457 end
1458 return recv
1459 end
1460
1461 # Set a GC finalizer on `recv`, only if `recv` isa Finalizable
1462 fun set_finalizer(recv: RuntimeVariable)
1463 do
1464 var mtype = recv.mtype
1465 var finalizable_type = compiler.mainmodule.finalizable_type
1466 if finalizable_type != null and not mtype.need_anchor and
1467 mtype.is_subtype(compiler.mainmodule, null, finalizable_type) then
1468 add "gc_register_finalizer({recv});"
1469 end
1470 end
1471
1472 # The currently processed module
1473 #
1474 # alias for `compiler.mainmodule`
1475 fun mmodule: MModule do return compiler.mainmodule
1476
1477 # Generate an integer value
1478 fun int_instance(value: Int): RuntimeVariable
1479 do
1480 var t = mmodule.int_type
1481 var res = new RuntimeVariable("{value.to_s}l", t, t)
1482 return res
1483 end
1484
1485 # Generate a byte value
1486 fun byte_instance(value: Byte): RuntimeVariable
1487 do
1488 var t = mmodule.byte_type
1489 var res = new RuntimeVariable("((unsigned char){value.to_s})", t, t)
1490 return res
1491 end
1492
1493 # Generate a char value
1494 fun char_instance(value: Char): RuntimeVariable
1495 do
1496 var t = mmodule.char_type
1497
1498 if value.ascii < 128 then
1499 return new RuntimeVariable("'{value.to_s.escape_to_c}'", t, t)
1500 else
1501 return new RuntimeVariable("{value.ascii}", t, t)
1502 end
1503 end
1504
1505 # Generate a float value
1506 #
1507 # FIXME pass a Float, not a string
1508 fun float_instance(value: String): RuntimeVariable
1509 do
1510 var t = mmodule.float_type
1511 var res = new RuntimeVariable("{value}", t, t)
1512 return res
1513 end
1514
1515 # Generate an integer value
1516 fun bool_instance(value: Bool): RuntimeVariable
1517 do
1518 var s = if value then "1" else "0"
1519 var res = new RuntimeVariable(s, bool_type, bool_type)
1520 return res
1521 end
1522
1523 # Generate the `null` value
1524 fun null_instance: RuntimeVariable
1525 do
1526 var t = compiler.mainmodule.model.null_type
1527 var res = new RuntimeVariable("((val*)NULL)", t, t)
1528 return res
1529 end
1530
1531 # Generate a string value
1532 fun string_instance(string: String): RuntimeVariable
1533 do
1534 var mtype = mmodule.string_type
1535 var name = self.get_name("varonce")
1536 self.add_decl("static {mtype.ctype} {name};")
1537 var res = self.new_var(mtype)
1538 self.add("if (likely({name}!=NULL)) \{")
1539 self.add("{res} = {name};")
1540 self.add("\} else \{")
1541 var native_mtype = mmodule.native_string_type
1542 var nat = self.new_var(native_mtype)
1543 self.add("{nat} = \"{string.escape_to_c}\";")
1544 var length = self.int_instance(string.bytelen)
1545 self.add("{res} = {self.send(self.get_property("to_s_with_length", native_mtype), [nat, length]).as(not null)};")
1546 self.add("{name} = {res};")
1547 self.add("\}")
1548 return res
1549 end
1550
1551 fun value_instance(object: Object): RuntimeVariable
1552 do
1553 if object isa Int then
1554 return int_instance(object)
1555 else if object isa Bool then
1556 return bool_instance(object)
1557 else if object isa String then
1558 return string_instance(object)
1559 else
1560 abort
1561 end
1562 end
1563
1564 # Generate an array value
1565 fun array_instance(array: Array[RuntimeVariable], elttype: MType): RuntimeVariable is abstract
1566
1567 # Get an instance of a array for a vararg
1568 fun vararg_instance(mpropdef: MPropDef, recv: RuntimeVariable, varargs: Array[RuntimeVariable], elttype: MType): RuntimeVariable is abstract
1569
1570 # Code generation
1571
1572 # Add a line in the main part of the generated C
1573 fun add(s: String) do self.writer.lines.add(s)
1574
1575 # Add a line in the
1576 # (used for local or global declaration)
1577 fun add_decl(s: String) do self.writer.decl_lines.add(s)
1578
1579 # Request the presence of a global declaration
1580 fun require_declaration(key: String)
1581 do
1582 var reqs = self.writer.file.required_declarations
1583 if reqs.has(key) then return
1584 reqs.add(key)
1585 var node = current_node
1586 if node != null then compiler.requirers_of_declarations[key] = node
1587 end
1588
1589 # Add a declaration in the local-header
1590 # The declaration is ensured to be present once
1591 fun declare_once(s: String)
1592 do
1593 self.compiler.provide_declaration(s, s)
1594 self.require_declaration(s)
1595 end
1596
1597 # Look for a needed .h and .c file for a given module
1598 # This is used for the legacy FFI
1599 fun add_extern(mmodule: MModule)
1600 do
1601 var file = mmodule.location.file.filename
1602 file = file.strip_extension(".nit")
1603 var tryfile = file + ".nit.h"
1604 if tryfile.file_exists then
1605 self.declare_once("#include \"{tryfile.basename}\"")
1606 self.compiler.files_to_copy.add(tryfile)
1607 end
1608 tryfile = file + "_nit.h"
1609 if tryfile.file_exists then
1610 self.declare_once("#include \"{tryfile.basename}\"")
1611 self.compiler.files_to_copy.add(tryfile)
1612 end
1613
1614 if self.compiler.seen_extern.has(file) then return
1615 self.compiler.seen_extern.add(file)
1616 tryfile = file + ".nit.c"
1617 if not tryfile.file_exists then
1618 tryfile = file + "_nit.c"
1619 if not tryfile.file_exists then return
1620 end
1621 var f = new ExternCFile(tryfile.basename, "")
1622 self.compiler.extern_bodies.add(f)
1623 self.compiler.files_to_copy.add(tryfile)
1624 end
1625
1626 # Return a new local runtime_variable initialized with the C expression `cexpr`.
1627 fun new_expr(cexpr: String, mtype: MType): RuntimeVariable
1628 do
1629 var res = new_var(mtype)
1630 self.add("{res} = {cexpr};")
1631 return res
1632 end
1633
1634 # Generate generic abort
1635 # used by aborts, asserts, casts, etc.
1636 fun add_abort(message: String)
1637 do
1638 self.add("PRINT_ERROR(\"Runtime error: %s\", \"{message.escape_to_c}\");")
1639 add_raw_abort
1640 end
1641
1642 fun add_raw_abort
1643 do
1644 if self.current_node != null and self.current_node.location.file != null and
1645 self.current_node.location.file.mmodule != null then
1646 var f = "FILE_{self.current_node.location.file.mmodule.c_name}"
1647 self.require_declaration(f)
1648 self.add("PRINT_ERROR(\" (%s:%d)\\n\", {f}, {current_node.location.line_start});")
1649 else
1650 self.add("PRINT_ERROR(\"\\n\");")
1651 end
1652 self.add("fatal_exit(1);")
1653 end
1654
1655 # Add a dynamic cast
1656 fun add_cast(value: RuntimeVariable, mtype: MType, tag: String)
1657 do
1658 var res = self.type_test(value, mtype, tag)
1659 self.add("if (unlikely(!{res})) \{")
1660 var cn = self.class_name_string(value)
1661 self.add("PRINT_ERROR(\"Runtime error: Cast failed. Expected `%s`, got `%s`\", \"{mtype.to_s.escape_to_c}\", {cn});")
1662 self.add_raw_abort
1663 self.add("\}")
1664 end
1665
1666 # Generate a return with the value `s`
1667 fun ret(s: RuntimeVariable)
1668 do
1669 self.assign(self.frame.returnvar.as(not null), s)
1670 self.add("goto {self.frame.returnlabel.as(not null)};")
1671 end
1672
1673 # Compile a statement (if any)
1674 fun stmt(nexpr: nullable AExpr)
1675 do
1676 if nexpr == null then return
1677 if nexpr.mtype == null and not nexpr.is_typed then
1678 # Untyped expression.
1679 # Might mean dead code or invalid code
1680 # so aborts
1681 add_abort("FATAL: bad statement executed.")
1682 return
1683 end
1684
1685 var narray = nexpr.comprehension
1686 if narray != null then
1687 var recv = frame.comprehension.as(not null)
1688 var val = expr(nexpr, narray.element_mtype)
1689 compile_callsite(narray.push_callsite.as(not null), [recv, val])
1690 return
1691 end
1692
1693 var old = self.current_node
1694 self.current_node = nexpr
1695 nexpr.stmt(self)
1696 self.current_node = old
1697 end
1698
1699 # Compile an expression an return its result
1700 # `mtype` is the expected return type, pass null if no specific type is expected.
1701 fun expr(nexpr: AExpr, mtype: nullable MType): RuntimeVariable
1702 do
1703 var old = self.current_node
1704 self.current_node = nexpr
1705
1706 var res = null
1707 if nexpr.mtype != null then
1708 res = nexpr.expr(self)
1709 end
1710
1711 if res == null then
1712 # Untyped expression.
1713 # Might mean dead code or invalid code.
1714 # so aborts
1715 add_abort("FATAL: bad expression executed.")
1716 # and return a placebo result to please the C compiler
1717 if mtype == null then mtype = compiler.mainmodule.object_type
1718 res = new_var(mtype)
1719
1720 self.current_node = old
1721 return res
1722 end
1723
1724 if mtype != null then
1725 mtype = self.anchor(mtype)
1726 res = self.autobox(res, mtype)
1727 end
1728 res = autoadapt(res, nexpr.mtype.as(not null))
1729 var implicit_cast_to = nexpr.implicit_cast_to
1730 if implicit_cast_to != null and not self.compiler.modelbuilder.toolcontext.opt_no_check_autocast.value then
1731 add_cast(res, implicit_cast_to, "auto")
1732 res = autoadapt(res, implicit_cast_to)
1733 end
1734 self.current_node = old
1735 return res
1736 end
1737
1738 # Alias for `self.expr(nexpr, self.bool_type)`
1739 fun expr_bool(nexpr: AExpr): RuntimeVariable do return expr(nexpr, bool_type)
1740
1741 # Safely show a debug message on the current node and repeat the message in the C code as a comment
1742 fun debug(message: String)
1743 do
1744 var node = self.current_node
1745 if node == null then
1746 print "?: {message}"
1747 else
1748 node.debug(message)
1749 end
1750 self.add("/* DEBUG: {message} */")
1751 end
1752 end
1753
1754 # A C function associated to a Nit method
1755 # Because of customization, a given Nit method can be compiler more that once
1756 abstract class AbstractRuntimeFunction
1757
1758 type COMPILER: AbstractCompiler
1759 type VISITOR: AbstractCompilerVisitor
1760
1761 # The associated Nit method
1762 var mmethoddef: MMethodDef
1763
1764 # The mangled c name of the runtime_function
1765 # Subclasses should redefine `build_c_name` instead
1766 fun c_name: String
1767 do
1768 var res = self.c_name_cache
1769 if res != null then return res
1770 res = self.build_c_name
1771 self.c_name_cache = res
1772 return res
1773 end
1774
1775 # Non cached version of `c_name`
1776 protected fun build_c_name: String is abstract
1777
1778 protected var c_name_cache: nullable String = null is writable
1779
1780 # Implements a call of the runtime_function
1781 # May inline the body or generate a C function call
1782 fun call(v: VISITOR, arguments: Array[RuntimeVariable]): nullable RuntimeVariable is abstract
1783
1784 # Generate the code for the `AbstractRuntimeFunction`
1785 # Warning: compile more than once compilation makes CC unhappy
1786 fun compile_to_c(compiler: COMPILER) is abstract
1787 end
1788
1789 # A runtime variable hold a runtime value in C.
1790 # Runtime variables are associated to Nit local variables and intermediate results in Nit expressions.
1791 #
1792 # The tricky point is that a single C variable can be associated to more than one `RuntimeVariable` because the static knowledge of the type of an expression can vary in the C code.
1793 class RuntimeVariable
1794 # The name of the variable in the C code
1795 var name: String
1796
1797 # The static type of the variable (as declard in C)
1798 var mtype: MType
1799
1800 # The current casted type of the variable (as known in Nit)
1801 var mcasttype: MType is writable
1802
1803 # If the variable exaclty a mcasttype?
1804 # false (usual value) means that the variable is a mcasttype or a subtype.
1805 var is_exact: Bool = false is writable
1806
1807 init
1808 do
1809 assert not mtype.need_anchor
1810 assert not mcasttype.need_anchor
1811 end
1812
1813 redef fun to_s do return name
1814
1815 redef fun inspect
1816 do
1817 var exact_str
1818 if self.is_exact then
1819 exact_str = " exact"
1820 else
1821 exact_str = ""
1822 end
1823 var type_str
1824 if self.mtype == self.mcasttype then
1825 type_str = "{mtype}{exact_str}"
1826 else
1827 type_str = "{mtype}({mcasttype}{exact_str})"
1828 end
1829 return "<{name}:{type_str}>"
1830 end
1831 end
1832
1833 # The static context of a visited property in a `AbstractCompilerVisitor`
1834 class StaticFrame
1835
1836 type VISITOR: AbstractCompilerVisitor
1837
1838 # The associated visitor
1839 var visitor: VISITOR
1840
1841 # The executed property.
1842 # A Method in case of a call, an attribute in case of a default initialization.
1843 var mpropdef: MPropDef
1844
1845 # The static type of the receiver
1846 var receiver: MClassType
1847
1848 # Arguments of the method (the first is the receiver)
1849 var arguments: Array[RuntimeVariable]
1850
1851 # The runtime_variable associated to the return (in a function)
1852 var returnvar: nullable RuntimeVariable = null is writable
1853
1854 # The label at the end of the property
1855 var returnlabel: nullable String = null is writable
1856
1857 # Labels associated to a each escapemarks.
1858 # Because of inlinings, escape-marks must be associated to their context (the frame)
1859 private var escapemark_names = new HashMap[EscapeMark, String]
1860
1861 # The array comprehension currently filled, if any
1862 private var comprehension: nullable RuntimeVariable = null
1863 end
1864
1865 redef class MType
1866 # Return the C type associated to a given Nit static type
1867 fun ctype: String do return "val*"
1868
1869 # C type outside of the compiler code and in boxes
1870 fun ctype_extern: String do return "val*"
1871
1872 # Short name of the `ctype` to use in unions
1873 fun ctypename: String do return "val"
1874
1875 # Is the associated C type a primitive one?
1876 #
1877 # ENSURE `result == (ctype != "val*")`
1878 fun is_c_primitive: Bool do return false
1879 end
1880
1881 redef class MClassType
1882
1883 redef var ctype is lazy do
1884 if mclass.name == "Int" then
1885 return "long"
1886 else if mclass.name == "Bool" then
1887 return "short int"
1888 else if mclass.name == "Char" then
1889 return "uint32_t"
1890 else if mclass.name == "Float" then
1891 return "double"
1892 else if mclass.name == "Byte" then
1893 return "unsigned char"
1894 else if mclass.name == "NativeString" then
1895 return "char*"
1896 else if mclass.name == "NativeArray" then
1897 return "val*"
1898 else
1899 return "val*"
1900 end
1901 end
1902
1903 redef var is_c_primitive is lazy do return ctype != "val*"
1904
1905 redef fun ctype_extern: String
1906 do
1907 if mclass.kind == extern_kind then
1908 return "void*"
1909 else
1910 return ctype
1911 end
1912 end
1913
1914 redef fun ctypename: String
1915 do
1916 if mclass.name == "Int" then
1917 return "l"
1918 else if mclass.name == "Bool" then
1919 return "s"
1920 else if mclass.name == "Char" then
1921 return "c"
1922 else if mclass.name == "Float" then
1923 return "d"
1924 else if mclass.name == "Byte" then
1925 return "b"
1926 else if mclass.name == "NativeString" then
1927 return "str"
1928 else if mclass.name == "NativeArray" then
1929 #return "{self.arguments.first.ctype}*"
1930 return "val"
1931 else
1932 return "val"
1933 end
1934 end
1935 end
1936
1937 redef class MPropDef
1938 type VISITOR: AbstractCompilerVisitor
1939 end
1940
1941 redef class MMethodDef
1942 # Can the body be inlined?
1943 fun can_inline(v: VISITOR): Bool
1944 do
1945 if is_abstract then return true
1946 var modelbuilder = v.compiler.modelbuilder
1947 var node = modelbuilder.mpropdef2node(self)
1948 if node isa APropdef then
1949 return node.can_inline
1950 else if node isa AClassdef then
1951 # Automatic free init is always inlined since it is empty or contains only attribtes assigments
1952 return true
1953 else
1954 abort
1955 end
1956 end
1957
1958 # Inline the body in another visitor
1959 fun compile_inside_to_c(v: VISITOR, arguments: Array[RuntimeVariable]): nullable RuntimeVariable
1960 do
1961 var modelbuilder = v.compiler.modelbuilder
1962 var val = constant_value
1963 var node = modelbuilder.mpropdef2node(self)
1964
1965 if is_abstract then
1966 var cn = v.class_name_string(arguments.first)
1967 v.current_node = node
1968 v.add("PRINT_ERROR(\"Runtime error: Abstract method `%s` called on `%s`\", \"{mproperty.name.escape_to_c}\", {cn});")
1969 v.add_raw_abort
1970 return null
1971 end
1972
1973 if node isa APropdef then
1974 var oldnode = v.current_node
1975 v.current_node = node
1976 self.compile_parameter_check(v, arguments)
1977 node.compile_to_c(v, self, arguments)
1978 v.current_node = oldnode
1979 else if node isa AClassdef then
1980 var oldnode = v.current_node
1981 v.current_node = node
1982 self.compile_parameter_check(v, arguments)
1983 node.compile_to_c(v, self, arguments)
1984 v.current_node = oldnode
1985 else if val != null then
1986 v.ret(v.value_instance(val))
1987 else
1988 abort
1989 end
1990 return null
1991 end
1992
1993 # Generate type checks in the C code to check covariant parameters
1994 fun compile_parameter_check(v: VISITOR, arguments: Array[RuntimeVariable])
1995 do
1996 if v.compiler.modelbuilder.toolcontext.opt_no_check_covariance.value then return
1997
1998 for i in [0..msignature.arity[ do
1999 # skip test for vararg since the array is instantiated with the correct polymorphic type
2000 if msignature.vararg_rank == i then continue
2001
2002 # skip if the cast is not required
2003 var origmtype = self.mproperty.intro.msignature.mparameters[i].mtype
2004 if not origmtype.need_anchor then continue
2005
2006 # get the parameter type
2007 var mtype = self.msignature.mparameters[i].mtype
2008
2009 # generate the cast
2010 # note that v decides if and how to implements the cast
2011 v.add("/* Covariant cast for argument {i} ({self.msignature.mparameters[i].name}) {arguments[i+1].inspect} isa {mtype} */")
2012 v.add_cast(arguments[i+1], mtype, "covariance")
2013 end
2014 end
2015 end
2016
2017 # Node visit
2018
2019 redef class APropdef
2020 fun compile_to_c(v: AbstractCompilerVisitor, mpropdef: MMethodDef, arguments: Array[RuntimeVariable])
2021 do
2022 v.add("PRINT_ERROR(\"NOT YET IMPLEMENTED {class_name} {mpropdef} at {location.to_s}\\n\");")
2023 debug("Not yet implemented")
2024 end
2025
2026 fun can_inline: Bool do return true
2027 end
2028
2029 redef class AMethPropdef
2030 redef fun compile_to_c(v, mpropdef, arguments)
2031 do
2032 # Call the implicit super-init
2033 var auto_super_inits = self.auto_super_inits
2034 if auto_super_inits != null then
2035 var args = [arguments.first]
2036 for auto_super_init in auto_super_inits do
2037 assert auto_super_init.mproperty != mpropdef.mproperty
2038 args.clear
2039 for i in [0..auto_super_init.msignature.arity+1[ do
2040 args.add(arguments[i])
2041 end
2042 assert auto_super_init.mproperty != mpropdef.mproperty
2043 v.compile_callsite(auto_super_init, args)
2044 end
2045 end
2046 if auto_super_call then
2047 v.supercall(mpropdef, arguments.first.mtype.as(MClassType), arguments)
2048 end
2049
2050 # Try special compilation
2051 if mpropdef.is_intern then
2052 if compile_intern_to_c(v, mpropdef, arguments) then return
2053 else if mpropdef.is_extern then
2054 if mpropdef.mproperty.is_init then
2055 if compile_externinit_to_c(v, mpropdef, arguments) then return
2056 else
2057 if compile_externmeth_to_c(v, mpropdef, arguments) then return
2058 end
2059 end
2060
2061 # Compile block if any
2062 var n_block = n_block
2063 if n_block != null then
2064 for i in [0..mpropdef.msignature.arity[ do
2065 var variable = self.n_signature.n_params[i].variable.as(not null)
2066 v.assign(v.variable(variable), arguments[i+1])
2067 end
2068 v.stmt(n_block)
2069 return
2070 end
2071
2072 # We have a problem
2073 var cn = v.class_name_string(arguments.first)
2074 v.add("PRINT_ERROR(\"Runtime error: uncompiled method `%s` called on `%s`. NOT YET IMPLEMENTED\", \"{mpropdef.mproperty.name.escape_to_c}\", {cn});")
2075 v.add_raw_abort
2076 end
2077
2078 redef fun can_inline
2079 do
2080 if self.auto_super_inits != null then return false
2081 var nblock = self.n_block
2082 if nblock == null then return true
2083 if (mpropdef.mproperty.name == "==" or mpropdef.mproperty.name == "!=") and mpropdef.mclassdef.mclass.name == "Object" then return true
2084 if nblock isa ABlockExpr and nblock.n_expr.length == 0 then return true
2085 return false
2086 end
2087
2088 fun compile_intern_to_c(v: AbstractCompilerVisitor, mpropdef: MMethodDef, arguments: Array[RuntimeVariable]): Bool
2089 do
2090 var pname = mpropdef.mproperty.name
2091 var cname = mpropdef.mclassdef.mclass.name
2092 var ret = mpropdef.msignature.return_mtype
2093 if ret != null then
2094 ret = v.resolve_for(ret, arguments.first)
2095 end
2096 if pname != "==" and pname != "!=" then
2097 v.adapt_signature(mpropdef, arguments)
2098 v.unbox_signature_extern(mpropdef, arguments)
2099 end
2100 if cname == "Int" then
2101 if pname == "output" then
2102 v.add("printf(\"%ld\\n\", {arguments.first});")
2103 return true
2104 else if pname == "object_id" then
2105 v.ret(arguments.first)
2106 return true
2107 else if pname == "+" then
2108 v.ret(v.new_expr("{arguments[0]} + {arguments[1]}", ret.as(not null)))
2109 return true
2110 else if pname == "-" then
2111 v.ret(v.new_expr("{arguments[0]} - {arguments[1]}", ret.as(not null)))
2112 return true
2113 else if pname == "unary -" then
2114 v.ret(v.new_expr("-{arguments[0]}", ret.as(not null)))
2115 return true
2116 else if pname == "unary +" then
2117 v.ret(arguments[0])
2118 return true
2119 else if pname == "*" then
2120 v.ret(v.new_expr("{arguments[0]} * {arguments[1]}", ret.as(not null)))
2121 return true
2122 else if pname == "/" then
2123 v.ret(v.new_expr("{arguments[0]} / {arguments[1]}", ret.as(not null)))
2124 return true
2125 else if pname == "%" then
2126 v.ret(v.new_expr("{arguments[0]} % {arguments[1]}", ret.as(not null)))
2127 return true
2128 else if pname == "lshift" then
2129 v.ret(v.new_expr("{arguments[0]} << {arguments[1]}", ret.as(not null)))
2130 return true
2131 else if pname == "rshift" then
2132 v.ret(v.new_expr("{arguments[0]} >> {arguments[1]}", ret.as(not null)))
2133 return true
2134 else if pname == "==" then
2135 v.ret(v.equal_test(arguments[0], arguments[1]))
2136 return true
2137 else if pname == "!=" then
2138 var res = v.equal_test(arguments[0], arguments[1])
2139 v.ret(v.new_expr("!{res}", ret.as(not null)))
2140 return true
2141 else if pname == "<" then
2142 v.ret(v.new_expr("{arguments[0]} < {arguments[1]}", ret.as(not null)))
2143 return true
2144 else if pname == ">" then
2145 v.ret(v.new_expr("{arguments[0]} > {arguments[1]}", ret.as(not null)))
2146 return true
2147 else if pname == "<=" then
2148 v.ret(v.new_expr("{arguments[0]} <= {arguments[1]}", ret.as(not null)))
2149 return true
2150 else if pname == ">=" then
2151 v.ret(v.new_expr("{arguments[0]} >= {arguments[1]}", ret.as(not null)))
2152 return true
2153 else if pname == "to_f" then
2154 v.ret(v.new_expr("(double){arguments[0]}", ret.as(not null)))
2155 return true
2156 else if pname == "to_b" then
2157 v.ret(v.new_expr("(unsigned char){arguments[0]}", ret.as(not null)))
2158 return true
2159 else if pname == "ascii" then
2160 v.ret(v.new_expr("(uint32_t){arguments[0]}", ret.as(not null)))
2161 return true
2162 end
2163 else if cname == "Char" then
2164 if pname == "object_id" then
2165 v.ret(v.new_expr("(long){arguments.first}", ret.as(not null)))
2166 return true
2167 else if pname == "successor" then
2168 v.ret(v.new_expr("{arguments[0]} + {arguments[1]}", ret.as(not null)))
2169 return true
2170 else if pname == "predecessor" then
2171 v.ret(v.new_expr("{arguments[0]} - {arguments[1]}", ret.as(not null)))
2172 return true
2173 else if pname == "==" then
2174 v.ret(v.equal_test(arguments[0], arguments[1]))
2175 return true
2176 else if pname == "!=" then
2177 var res = v.equal_test(arguments[0], arguments[1])
2178 v.ret(v.new_expr("!{res}", ret.as(not null)))
2179 return true
2180 else if pname == "<" then
2181 v.ret(v.new_expr("{arguments[0]} < {arguments[1]}", ret.as(not null)))
2182 return true
2183 else if pname == ">" then
2184 v.ret(v.new_expr("{arguments[0]} > {arguments[1]}", ret.as(not null)))
2185 return true
2186 else if pname == "<=" then
2187 v.ret(v.new_expr("{arguments[0]} <= {arguments[1]}", ret.as(not null)))
2188 return true
2189 else if pname == ">=" then
2190 v.ret(v.new_expr("{arguments[0]} >= {arguments[1]}", ret.as(not null)))
2191 return true
2192 else if pname == "to_i" then
2193 v.ret(v.new_expr("{arguments[0]}-'0'", ret.as(not null)))
2194 return true
2195 else if pname == "ascii" then
2196 v.ret(v.new_expr("(long){arguments[0]}", ret.as(not null)))
2197 return true
2198 end
2199 else if cname == "Byte" then
2200 if pname == "output" then
2201 v.add("printf(\"%x\\n\", {arguments.first});")
2202 return true
2203 else if pname == "object_id" then
2204 v.ret(v.new_expr("(long){arguments.first}", ret.as(not null)))
2205 return true
2206 else if pname == "+" then
2207 v.ret(v.new_expr("{arguments[0]} + {arguments[1]}", ret.as(not null)))
2208 return true
2209 else if pname == "-" then
2210 v.ret(v.new_expr("{arguments[0]} - {arguments[1]}", ret.as(not null)))
2211 return true
2212 else if pname == "unary -" then
2213 v.ret(v.new_expr("-{arguments[0]}", ret.as(not null)))
2214 return true
2215 else if pname == "unary +" then
2216 v.ret(arguments[0])
2217 return true
2218 else if pname == "*" then
2219 v.ret(v.new_expr("{arguments[0]} * {arguments[1]}", ret.as(not null)))
2220 return true
2221 else if pname == "/" then
2222 v.ret(v.new_expr("{arguments[0]} / {arguments[1]}", ret.as(not null)))
2223 return true
2224 else if pname == "%" then
2225 v.ret(v.new_expr("{arguments[0]} % {arguments[1]}", ret.as(not null)))
2226 return true
2227 else if pname == "lshift" then
2228 v.ret(v.new_expr("{arguments[0]} << {arguments[1]}", ret.as(not null)))
2229 return true
2230 else if pname == "rshift" then
2231 v.ret(v.new_expr("{arguments[0]} >> {arguments[1]}", ret.as(not null)))
2232 return true
2233 else if pname == "==" then
2234 v.ret(v.equal_test(arguments[0], arguments[1]))
2235 return true
2236 else if pname == "!=" then
2237 var res = v.equal_test(arguments[0], arguments[1])
2238 v.ret(v.new_expr("!{res}", ret.as(not null)))
2239 return true
2240 else if pname == "<" then
2241 v.ret(v.new_expr("{arguments[0]} < {arguments[1]}", ret.as(not null)))
2242 return true
2243 else if pname == ">" then
2244 v.ret(v.new_expr("{arguments[0]} > {arguments[1]}", ret.as(not null)))
2245 return true
2246 else if pname == "<=" then
2247 v.ret(v.new_expr("{arguments[0]} <= {arguments[1]}", ret.as(not null)))
2248 return true
2249 else if pname == ">=" then
2250 v.ret(v.new_expr("{arguments[0]} >= {arguments[1]}", ret.as(not null)))
2251 return true
2252 else if pname == "to_i" then
2253 v.ret(v.new_expr("(long){arguments[0]}", ret.as(not null)))
2254 return true
2255 else if pname == "to_f" then
2256 v.ret(v.new_expr("(double){arguments[0]}", ret.as(not null)))
2257 return true
2258 else if pname == "ascii" then
2259 v.ret(v.new_expr("{arguments[0]}", ret.as(not null)))
2260 return true
2261 end
2262 else if cname == "Bool" then
2263 if pname == "output" then
2264 v.add("printf({arguments.first}?\"true\\n\":\"false\\n\");")
2265 return true
2266 else if pname == "object_id" then
2267 v.ret(v.new_expr("(long){arguments.first}", ret.as(not null)))
2268 return true
2269 else if pname == "==" then
2270 v.ret(v.equal_test(arguments[0], arguments[1]))
2271 return true
2272 else if pname == "!=" then
2273 var res = v.equal_test(arguments[0], arguments[1])
2274 v.ret(v.new_expr("!{res}", ret.as(not null)))
2275 return true
2276 end
2277 else if cname == "Float" then
2278 if pname == "output" then
2279 v.add("printf(\"%f\\n\", {arguments.first});")
2280 return true
2281 else if pname == "object_id" then
2282 v.ret(v.new_expr("(double){arguments.first}", ret.as(not null)))
2283 return true
2284 else if pname == "+" then
2285 v.ret(v.new_expr("{arguments[0]} + {arguments[1]}", ret.as(not null)))
2286 return true
2287 else if pname == "-" then
2288 v.ret(v.new_expr("{arguments[0]} - {arguments[1]}", ret.as(not null)))
2289 return true
2290 else if pname == "unary -" then
2291 v.ret(v.new_expr("-{arguments[0]}", ret.as(not null)))
2292 return true
2293 else if pname == "unary +" then
2294 v.ret(arguments[0])
2295 return true
2296 else if pname == "succ" then
2297 v.ret(v.new_expr("{arguments[0]}+1", ret.as(not null)))
2298 return true
2299 else if pname == "prec" then
2300 v.ret(v.new_expr("{arguments[0]}-1", ret.as(not null)))
2301 return true
2302 else if pname == "*" then
2303 v.ret(v.new_expr("{arguments[0]} * {arguments[1]}", ret.as(not null)))
2304 return true
2305 else if pname == "/" then
2306 v.ret(v.new_expr("{arguments[0]} / {arguments[1]}", ret.as(not null)))
2307 return true
2308 else if pname == "==" then
2309 v.ret(v.equal_test(arguments[0], arguments[1]))
2310 return true
2311 else if pname == "!=" then
2312 var res = v.equal_test(arguments[0], arguments[1])
2313 v.ret(v.new_expr("!{res}", ret.as(not null)))
2314 return true
2315 else if pname == "<" then
2316 v.ret(v.new_expr("{arguments[0]} < {arguments[1]}", ret.as(not null)))
2317 return true
2318 else if pname == ">" then
2319 v.ret(v.new_expr("{arguments[0]} > {arguments[1]}", ret.as(not null)))
2320 return true
2321 else if pname == "<=" then
2322 v.ret(v.new_expr("{arguments[0]} <= {arguments[1]}", ret.as(not null)))
2323 return true
2324 else if pname == ">=" then
2325 v.ret(v.new_expr("{arguments[0]} >= {arguments[1]}", ret.as(not null)))
2326 return true
2327 else if pname == "to_i" then
2328 v.ret(v.new_expr("(long){arguments[0]}", ret.as(not null)))
2329 return true
2330 else if pname == "to_b" then
2331 v.ret(v.new_expr("(unsigned char){arguments[0]}", ret.as(not null)))
2332 return true
2333 end
2334 else if cname == "NativeString" then
2335 if pname == "[]" then
2336 v.ret(v.new_expr("(unsigned char)((int){arguments[0]}[{arguments[1]}])", ret.as(not null)))
2337 return true
2338 else if pname == "[]=" then
2339 v.add("{arguments[0]}[{arguments[1]}]=(unsigned char){arguments[2]};")
2340 return true
2341 else if pname == "copy_to" then
2342 v.add("memmove({arguments[1]}+{arguments[4]},{arguments[0]}+{arguments[3]},{arguments[2]});")
2343 return true
2344 else if pname == "atoi" then
2345 v.ret(v.new_expr("atoi({arguments[0]});", ret.as(not null)))
2346 return true
2347 else if pname == "fast_cstring" then
2348 v.ret(v.new_expr("{arguments[0]} + {arguments[1]}", ret.as(not null)))
2349 return true
2350 else if pname == "new" then
2351 v.ret(v.new_expr("(char*)nit_alloc({arguments[1]})", ret.as(not null)))
2352 return true
2353 end
2354 else if cname == "NativeArray" then
2355 v.native_array_def(pname, ret, arguments)
2356 return true
2357 end
2358 if pname == "exit" then
2359 v.add("exit({arguments[1]});")
2360 return true
2361 else if pname == "sys" then
2362 v.ret(v.new_expr("glob_sys", ret.as(not null)))
2363 return true
2364 else if pname == "calloc_string" then
2365 v.ret(v.new_expr("(char*)nit_alloc({arguments[1]})", ret.as(not null)))
2366 return true
2367 else if pname == "calloc_array" then
2368 v.calloc_array(ret.as(not null), arguments)
2369 return true
2370 else if pname == "object_id" then
2371 v.ret(v.new_expr("(long){arguments.first}", ret.as(not null)))
2372 return true
2373 else if pname == "is_same_type" then
2374 v.ret(v.is_same_type_test(arguments[0], arguments[1]))
2375 return true
2376 else if pname == "is_same_instance" then
2377 v.ret(v.equal_test(arguments[0], arguments[1]))
2378 return true
2379 else if pname == "output_class_name" then
2380 var nat = v.class_name_string(arguments.first)
2381 v.add("printf(\"%s\\n\", {nat});")
2382 return true
2383 else if pname == "native_class_name" then
2384 var nat = v.class_name_string(arguments.first)
2385 v.ret(v.new_expr("(char*){nat}", ret.as(not null)))
2386 return true
2387 else if pname == "force_garbage_collection" then
2388 v.add("nit_gcollect();")
2389 return true
2390 else if pname == "native_argc" then
2391 v.ret(v.new_expr("glob_argc", ret.as(not null)))
2392 return true
2393 else if pname == "native_argv" then
2394 v.ret(v.new_expr("glob_argv[{arguments[1]}]", ret.as(not null)))
2395 return true
2396 end
2397 return false
2398 end
2399
2400 # Compile an extern method
2401 # Return `true` if the compilation was successful, `false` if a fall-back is needed
2402 fun compile_externmeth_to_c(v: AbstractCompilerVisitor, mpropdef: MMethodDef, arguments: Array[RuntimeVariable]): Bool
2403 do
2404 var externname
2405 var at = self.get_single_annotation("extern", v.compiler.modelbuilder)
2406 if at != null and at.n_args.length == 1 then
2407 externname = at.arg_as_string(v.compiler.modelbuilder)
2408 if externname == null then return false
2409 else
2410 return false
2411 end
2412 v.add_extern(mpropdef.mclassdef.mmodule)
2413 var res: nullable RuntimeVariable = null
2414 var ret = mpropdef.msignature.return_mtype
2415 if ret != null then
2416 ret = v.resolve_for(ret, arguments.first)
2417 res = v.new_var_extern(ret)
2418 end
2419 v.adapt_signature(mpropdef, arguments)
2420 v.unbox_signature_extern(mpropdef, arguments)
2421
2422 if res == null then
2423 v.add("{externname}({arguments.join(", ")});")
2424 else
2425 v.add("{res} = {externname}({arguments.join(", ")});")
2426 res = v.box_extern(res, ret.as(not null))
2427 v.ret(res)
2428 end
2429 return true
2430 end
2431
2432 # Compile an extern factory
2433 # Return `true` if the compilation was successful, `false` if a fall-back is needed
2434 fun compile_externinit_to_c(v: AbstractCompilerVisitor, mpropdef: MMethodDef, arguments: Array[RuntimeVariable]): Bool
2435 do
2436 var externname
2437 var at = self.get_single_annotation("extern", v.compiler.modelbuilder)
2438 if at != null then
2439 externname = at.arg_as_string(v.compiler.modelbuilder)
2440 if externname == null then return false
2441 else
2442 return false
2443 end
2444 v.add_extern(mpropdef.mclassdef.mmodule)
2445 v.adapt_signature(mpropdef, arguments)
2446 v.unbox_signature_extern(mpropdef, arguments)
2447 var ret = arguments.first.mtype
2448 var res = v.new_var_extern(ret)
2449
2450 arguments.shift
2451
2452 v.add("{res} = {externname}({arguments.join(", ")});")
2453 res = v.box_extern(res, ret)
2454 v.ret(res)
2455 return true
2456 end
2457 end
2458
2459 redef class AAttrPropdef
2460 redef fun can_inline: Bool do return not is_lazy
2461
2462 redef fun compile_to_c(v, mpropdef, arguments)
2463 do
2464 if mpropdef == mreadpropdef then
2465 assert arguments.length == 1
2466 var recv = arguments.first
2467 var res
2468 if is_lazy then
2469 var set
2470 var ret = self.mtype
2471 var useiset = not ret.is_c_primitive and not ret isa MNullableType
2472 var guard = self.mlazypropdef.mproperty
2473 if useiset then
2474 set = v.isset_attribute(self.mpropdef.mproperty, recv)
2475 else
2476 set = v.read_attribute(guard, recv)
2477 end
2478 v.add("if(likely({set})) \{")
2479 res = v.read_attribute(self.mpropdef.mproperty, recv)
2480 v.add("\} else \{")
2481
2482 var value = evaluate_expr(v, recv)
2483
2484 v.assign(res, value)
2485 if not useiset then
2486 var true_v = v.bool_instance(true)
2487 v.write_attribute(guard, arguments.first, true_v)
2488 end
2489 v.add("\}")
2490 else
2491 res = v.read_attribute(self.mpropdef.mproperty, arguments.first)
2492 end
2493 v.assign(v.frame.returnvar.as(not null), res)
2494 else if mpropdef == mwritepropdef then
2495 assert arguments.length == 2
2496 v.write_attribute(self.mpropdef.mproperty, arguments.first, arguments[1])
2497 if is_lazy then
2498 var ret = self.mtype
2499 var useiset = not ret.is_c_primitive and not ret isa MNullableType
2500 if not useiset then
2501 v.write_attribute(self.mlazypropdef.mproperty, arguments.first, v.bool_instance(true))
2502 end
2503 end
2504 else
2505 abort
2506 end
2507 end
2508
2509 fun init_expr(v: AbstractCompilerVisitor, recv: RuntimeVariable)
2510 do
2511 if has_value and not is_lazy and not n_expr isa ANullExpr then evaluate_expr(v, recv)
2512 end
2513
2514 # Evaluate, store and return the default value of the attribute
2515 private fun evaluate_expr(v: AbstractCompilerVisitor, recv: RuntimeVariable): RuntimeVariable
2516 do
2517 var oldnode = v.current_node
2518 v.current_node = self
2519 var old_frame = v.frame
2520 var frame = new StaticFrame(v, self.mreadpropdef.as(not null), recv.mcasttype.undecorate.as(MClassType), [recv])
2521 v.frame = frame
2522
2523 var value
2524 var mtype = self.mtype
2525 assert mtype != null
2526
2527 var nexpr = self.n_expr
2528 var nblock = self.n_block
2529 if nexpr != null then
2530 value = v.expr(nexpr, mtype)
2531 else if nblock != null then
2532 value = v.new_var(mtype)
2533 frame.returnvar = value
2534 frame.returnlabel = v.get_name("RET_LABEL")
2535 v.add("\{")
2536 v.stmt(nblock)
2537 v.add("{frame.returnlabel.as(not null)}:(void)0;")
2538 v.add("\}")
2539 else
2540 abort
2541 end
2542
2543 v.write_attribute(self.mpropdef.mproperty, recv, value)
2544
2545 v.frame = old_frame
2546 v.current_node = oldnode
2547
2548 return value
2549 end
2550
2551 fun check_expr(v: AbstractCompilerVisitor, recv: RuntimeVariable)
2552 do
2553 var nexpr = self.n_expr
2554 if nexpr != null then return
2555
2556 var oldnode = v.current_node
2557 v.current_node = self
2558 var old_frame = v.frame
2559 var frame = new StaticFrame(v, self.mpropdef.as(not null), recv.mtype.as(MClassType), [recv])
2560 v.frame = frame
2561 # Force read to check the initialization
2562 v.read_attribute(self.mpropdef.mproperty, recv)
2563 v.frame = old_frame
2564 v.current_node = oldnode
2565 end
2566 end
2567
2568 redef class AClassdef
2569 private fun compile_to_c(v: AbstractCompilerVisitor, mpropdef: MMethodDef, arguments: Array[RuntimeVariable])
2570 do
2571 if mpropdef == self.mfree_init then
2572 assert mpropdef.mproperty.is_root_init
2573 assert arguments.length == 1
2574 if not mpropdef.is_intro then
2575 v.supercall(mpropdef, arguments.first.mtype.as(MClassType), arguments)
2576 end
2577 return
2578 else
2579 abort
2580 end
2581 end
2582 end
2583
2584 redef class AExpr
2585 # Try to compile self as an expression
2586 # Do not call this method directly, use `v.expr` instead
2587 private fun expr(v: AbstractCompilerVisitor): nullable RuntimeVariable
2588 do
2589 v.add("PRINT_ERROR(\"NOT YET IMPLEMENTED {class_name}:{location.to_s}\\n\");")
2590 var mtype = self.mtype
2591 if mtype == null then
2592 return null
2593 else
2594 var res = v.new_var(mtype)
2595 v.add("/* {res} = NOT YET {class_name} */")
2596 return res
2597 end
2598 end
2599
2600 # Try to compile self as a statement
2601 # Do not call this method directly, use `v.stmt` instead
2602 private fun stmt(v: AbstractCompilerVisitor)
2603 do
2604 expr(v)
2605 end
2606 end
2607
2608 redef class ABlockExpr
2609 redef fun stmt(v)
2610 do
2611 for e in self.n_expr do v.stmt(e)
2612 end
2613 redef fun expr(v)
2614 do
2615 var last = self.n_expr.last
2616 for e in self.n_expr do
2617 if e == last then break
2618 v.stmt(e)
2619 end
2620 return v.expr(last, null)
2621 end
2622 end
2623
2624 redef class AVardeclExpr
2625 redef fun stmt(v)
2626 do
2627 var variable = self.variable.as(not null)
2628 var ne = self.n_expr
2629 if ne != null then
2630 var i = v.expr(ne, variable.declared_type)
2631 v.assign(v.variable(variable), i)
2632 end
2633 end
2634 end
2635
2636 redef class AVarExpr
2637 redef fun expr(v)
2638 do
2639 var res = v.variable(self.variable.as(not null))
2640 var mtype = self.mtype.as(not null)
2641 return v.autoadapt(res, mtype)
2642 end
2643 end
2644
2645 redef class AVarAssignExpr
2646 redef fun expr(v)
2647 do
2648 var variable = self.variable.as(not null)
2649 var i = v.expr(self.n_value, variable.declared_type)
2650 v.assign(v.variable(variable), i)
2651 return i
2652 end
2653 end
2654
2655 redef class AVarReassignExpr
2656 redef fun stmt(v)
2657 do
2658 var variable = self.variable.as(not null)
2659 var vari = v.variable(variable)
2660 var value = v.expr(self.n_value, variable.declared_type)
2661 var res = v.compile_callsite(self.reassign_callsite.as(not null), [vari, value])
2662 assert res != null
2663 v.assign(v.variable(variable), res)
2664 end
2665 end
2666
2667 redef class ASelfExpr
2668 redef fun expr(v) do return v.frame.arguments.first
2669 end
2670
2671 redef class AImplicitSelfExpr
2672 redef fun expr(v) do
2673 if not is_sys then return super
2674 return v.new_expr("glob_sys", mtype.as(not null))
2675 end
2676 end
2677
2678 redef class AEscapeExpr
2679 redef fun stmt(v) do v.add("goto BREAK_{v.escapemark_name(self.escapemark)};")
2680 end
2681
2682 redef class AReturnExpr
2683 redef fun stmt(v)
2684 do
2685 var nexpr = self.n_expr
2686 if nexpr != null then
2687 var returnvar = v.frame.returnvar.as(not null)
2688 var i = v.expr(nexpr, returnvar.mtype)
2689 v.assign(returnvar, i)
2690 end
2691 v.add("goto {v.frame.returnlabel.as(not null)};")
2692 end
2693 end
2694
2695 redef class AAbortExpr
2696 redef fun stmt(v) do v.add_abort("Aborted")
2697 end
2698
2699 redef class AIfExpr
2700 redef fun stmt(v)
2701 do
2702 var cond = v.expr_bool(self.n_expr)
2703 v.add("if ({cond})\{")
2704 v.stmt(self.n_then)
2705 v.add("\} else \{")
2706 v.stmt(self.n_else)
2707 v.add("\}")
2708 end
2709
2710 redef fun expr(v)
2711 do
2712 var res = v.new_var(self.mtype.as(not null))
2713 var cond = v.expr_bool(self.n_expr)
2714 v.add("if ({cond})\{")
2715 v.assign(res, v.expr(self.n_then.as(not null), null))
2716 v.add("\} else \{")
2717 v.assign(res, v.expr(self.n_else.as(not null), null))
2718 v.add("\}")
2719 return res
2720 end
2721 end
2722
2723 redef class AIfexprExpr
2724 redef fun expr(v)
2725 do
2726 var res = v.new_var(self.mtype.as(not null))
2727 var cond = v.expr_bool(self.n_expr)
2728 v.add("if ({cond})\{")
2729 v.assign(res, v.expr(self.n_then, null))
2730 v.add("\} else \{")
2731 v.assign(res, v.expr(self.n_else, null))
2732 v.add("\}")
2733 return res
2734 end
2735 end
2736
2737 redef class ADoExpr
2738 redef fun stmt(v)
2739 do
2740 v.stmt(self.n_block)
2741 v.add_escape_label(break_mark)
2742 end
2743 end
2744
2745 redef class AWhileExpr
2746 redef fun stmt(v)
2747 do
2748 v.add("for(;;) \{")
2749 var cond = v.expr_bool(self.n_expr)
2750 v.add("if (!{cond}) break;")
2751 v.stmt(self.n_block)
2752 v.add_escape_label(continue_mark)
2753 v.add("\}")
2754 v.add_escape_label(break_mark)
2755 end
2756 end
2757
2758 redef class ALoopExpr
2759 redef fun stmt(v)
2760 do
2761 v.add("for(;;) \{")
2762 v.stmt(self.n_block)
2763 v.add_escape_label(continue_mark)
2764 v.add("\}")
2765 v.add_escape_label(break_mark)
2766 end
2767 end
2768
2769 redef class AForExpr
2770 redef fun stmt(v)
2771 do
2772 var cl = v.expr(self.n_expr, null)
2773 var it_meth = self.method_iterator
2774 assert it_meth != null
2775 var it = v.compile_callsite(it_meth, [cl])
2776 assert it != null
2777 v.add("for(;;) \{")
2778 var isok_meth = self.method_is_ok
2779 assert isok_meth != null
2780 var ok = v.compile_callsite(isok_meth, [it])
2781 assert ok != null
2782 v.add("if(!{ok}) break;")
2783 if self.variables.length == 1 then
2784 var item_meth = self.method_item
2785 assert item_meth != null
2786 var i = v.compile_callsite(item_meth, [it])
2787 assert i != null
2788 v.assign(v.variable(variables.first), i)
2789 else if self.variables.length == 2 then
2790 var key_meth = self.method_key
2791 assert key_meth != null
2792 var i = v.compile_callsite(key_meth, [it])
2793 assert i != null
2794 v.assign(v.variable(variables[0]), i)
2795 var item_meth = self.method_item
2796 assert item_meth != null
2797 i = v.compile_callsite(item_meth, [it])
2798 assert i != null
2799 v.assign(v.variable(variables[1]), i)
2800 else
2801 abort
2802 end
2803 v.stmt(self.n_block)
2804 v.add_escape_label(continue_mark)
2805 var next_meth = self.method_next
2806 assert next_meth != null
2807 v.compile_callsite(next_meth, [it])
2808 v.add("\}")
2809 v.add_escape_label(break_mark)
2810
2811 var method_finish = self.method_finish
2812 if method_finish != null then
2813 # TODO: Find a way to call this also in long escape (e.g. return)
2814 v.compile_callsite(method_finish, [it])
2815 end
2816 end
2817 end
2818
2819 redef class AAssertExpr
2820 redef fun stmt(v)
2821 do
2822 if v.compiler.modelbuilder.toolcontext.opt_no_check_assert.value then return
2823
2824 var cond = v.expr_bool(self.n_expr)
2825 v.add("if (unlikely(!{cond})) \{")
2826 v.stmt(self.n_else)
2827 var nid = self.n_id
2828 if nid != null then
2829 v.add_abort("Assert '{nid.text}' failed")
2830 else
2831 v.add_abort("Assert failed")
2832 end
2833 v.add("\}")
2834 end
2835 end
2836
2837 redef class AOrExpr
2838 redef fun expr(v)
2839 do
2840 var res = v.new_var(self.mtype.as(not null))
2841 var i1 = v.expr_bool(self.n_expr)
2842 v.add("if ({i1}) \{")
2843 v.add("{res} = 1;")
2844 v.add("\} else \{")
2845 var i2 = v.expr_bool(self.n_expr2)
2846 v.add("{res} = {i2};")
2847 v.add("\}")
2848 return res
2849 end
2850 end
2851
2852 redef class AImpliesExpr
2853 redef fun expr(v)
2854 do
2855 var res = v.new_var(self.mtype.as(not null))
2856 var i1 = v.expr_bool(self.n_expr)
2857 v.add("if (!{i1}) \{")
2858 v.add("{res} = 1;")
2859 v.add("\} else \{")
2860 var i2 = v.expr_bool(self.n_expr2)
2861 v.add("{res} = {i2};")
2862 v.add("\}")
2863 return res
2864 end
2865 end
2866
2867 redef class AAndExpr
2868 redef fun expr(v)
2869 do
2870 var res = v.new_var(self.mtype.as(not null))
2871 var i1 = v.expr_bool(self.n_expr)
2872 v.add("if (!{i1}) \{")
2873 v.add("{res} = 0;")
2874 v.add("\} else \{")
2875 var i2 = v.expr_bool(self.n_expr2)
2876 v.add("{res} = {i2};")
2877 v.add("\}")
2878 return res
2879 end
2880 end
2881
2882 redef class ANotExpr
2883 redef fun expr(v)
2884 do
2885 var cond = v.expr_bool(self.n_expr)
2886 return v.new_expr("!{cond}", self.mtype.as(not null))
2887 end
2888 end
2889
2890 redef class AOrElseExpr
2891 redef fun expr(v)
2892 do
2893 var res = v.new_var(self.mtype.as(not null))
2894 var i1 = v.expr(self.n_expr, null)
2895 v.add("if ({i1}!=NULL) \{")
2896 v.assign(res, i1)
2897 v.add("\} else \{")
2898 var i2 = v.expr(self.n_expr2, null)
2899 v.assign(res, i2)
2900 v.add("\}")
2901 return res
2902 end
2903 end
2904
2905 redef class AIntExpr
2906 redef fun expr(v) do return v.int_instance(self.value.as(not null))
2907 end
2908
2909 redef class AByteExpr
2910 redef fun expr(v) do return v.byte_instance(self.value.as(not null))
2911 end
2912
2913 redef class AFloatExpr
2914 redef fun expr(v) do return v.float_instance("{self.n_float.text}") # FIXME use value, not n_float
2915 end
2916
2917 redef class ACharExpr
2918 redef fun expr(v) do return v.char_instance(self.value.as(not null))
2919 end
2920
2921 redef class AArrayExpr
2922 redef fun expr(v)
2923 do
2924 var mtype = self.element_mtype.as(not null)
2925 var array = new Array[RuntimeVariable]
2926 var res = v.array_instance(array, mtype)
2927
2928 var old_comprehension = v.frame.comprehension
2929 v.frame.comprehension = res
2930 for nexpr in self.n_exprs do
2931 v.stmt(nexpr)
2932 end
2933 v.frame.comprehension = old_comprehension
2934
2935 return res
2936 end
2937 end
2938
2939 redef class AStringFormExpr
2940 redef fun expr(v) do return v.string_instance(self.value.as(not null))
2941 end
2942
2943 redef class ASuperstringExpr
2944 redef fun expr(v)
2945 do
2946 var type_string = mtype.as(not null)
2947
2948 # Collect elements of the superstring
2949 var array = new Array[AExpr]
2950 for ne in self.n_exprs do
2951 # Drop literal empty string.
2952 # They appears in things like "{a}" that is ["", a, ""]
2953 if ne isa AStringFormExpr and ne.value == "" then continue # skip empty sub-strings
2954 array.add(ne)
2955 end
2956
2957 # Store the allocated native array in a static variable
2958 # For reusing later
2959 var varonce = v.get_name("varonce")
2960 v.add("if (unlikely({varonce}==NULL)) \{")
2961
2962 # The native array that will contains the elements to_s-ized.
2963 # For fast concatenation.
2964 var a = v.native_array_instance(type_string, v.int_instance(array.length))
2965
2966 v.add_decl("static {a.mtype.ctype} {varonce};")
2967
2968 # Pre-fill the array with the literal string parts.
2969 # So they do not need to be filled again when reused
2970 for i in [0..array.length[ do
2971 var ne = array[i]
2972 if not ne isa AStringFormExpr then continue
2973 var e = v.expr(ne, null)
2974 v.native_array_set(a, i, e)
2975 end
2976
2977 v.add("\} else \{")
2978 # Take the native-array from the store.
2979 # The point is to prevent that some recursive execution use (and corrupt) the same native array
2980 # WARNING: not thread safe! (FIXME?)
2981 v.add("{a} = {varonce};")
2982 v.add("{varonce} = NULL;")
2983 v.add("\}")
2984
2985 # Stringify the elements and put them in the native array
2986 var to_s_method = v.get_property("to_s", v.object_type)
2987 for i in [0..array.length[ do
2988 var ne = array[i]
2989 if ne isa AStringFormExpr then continue
2990 var e = v.expr(ne, null)
2991 # Skip the `to_s` if the element is already a String
2992 if not e.mcasttype.is_subtype(v.compiler.mainmodule, null, type_string) then
2993 e = v.send(to_s_method, [e]).as(not null)
2994 end
2995 v.native_array_set(a, i, e)
2996 end
2997
2998 # Fast join the native string to get the result
2999 var res = v.send(v.get_property("native_to_s", a.mtype), [a])
3000
3001 # We finish to work with the native array,
3002 # so store it so that it can be reused
3003 v.add("{varonce} = {a};")
3004 return res
3005 end
3006 end
3007
3008 redef class ACrangeExpr
3009 redef fun expr(v)
3010 do
3011 var i1 = v.expr(self.n_expr, null)
3012 var i2 = v.expr(self.n_expr2, null)
3013 var mtype = self.mtype.as(MClassType)
3014 var res = v.init_instance(mtype)
3015 v.compile_callsite(init_callsite.as(not null), [res, i1, i2])
3016 return res
3017 end
3018 end
3019
3020 redef class AOrangeExpr
3021 redef fun expr(v)
3022 do
3023 var i1 = v.expr(self.n_expr, null)
3024 var i2 = v.expr(self.n_expr2, null)
3025 var mtype = self.mtype.as(MClassType)
3026 var res = v.init_instance(mtype)
3027 v.compile_callsite(init_callsite.as(not null), [res, i1, i2])
3028 return res
3029 end
3030 end
3031
3032 redef class ATrueExpr
3033 redef fun expr(v) do return v.bool_instance(true)
3034 end
3035
3036 redef class AFalseExpr
3037 redef fun expr(v) do return v.bool_instance(false)
3038 end
3039
3040 redef class ANullExpr
3041 redef fun expr(v) do return v.null_instance
3042 end
3043
3044 redef class AIsaExpr
3045 redef fun expr(v)
3046 do
3047 var i = v.expr(self.n_expr, null)
3048 var cast_type = self.cast_type
3049 if cast_type == null then return null # no-no on broken node
3050 return v.type_test(i, cast_type, "isa")
3051 end
3052 end
3053
3054 redef class AAsCastExpr
3055 redef fun expr(v)
3056 do
3057 var i = v.expr(self.n_expr, null)
3058 if v.compiler.modelbuilder.toolcontext.opt_no_check_assert.value then return i
3059
3060 v.add_cast(i, self.mtype.as(not null), "as")
3061 return i
3062 end
3063 end
3064
3065 redef class AAsNotnullExpr
3066 redef fun expr(v)
3067 do
3068 var i = v.expr(self.n_expr, null)
3069 if v.compiler.modelbuilder.toolcontext.opt_no_check_assert.value then return i
3070
3071 if i.mtype.is_c_primitive then return i
3072
3073 v.add("if (unlikely({i} == NULL)) \{")
3074 v.add_abort("Cast failed")
3075 v.add("\}")
3076 return i
3077 end
3078 end
3079
3080 redef class AParExpr
3081 redef fun expr(v) do return v.expr(self.n_expr, null)
3082 end
3083
3084 redef class AOnceExpr
3085 redef fun expr(v)
3086 do
3087 var mtype = self.mtype.as(not null)
3088 var name = v.get_name("varonce")
3089 var guard = v.get_name(name + "_guard")
3090 v.add_decl("static {mtype.ctype} {name};")
3091 v.add_decl("static int {guard};")
3092 var res = v.new_var(mtype)
3093 v.add("if (likely({guard})) \{")
3094 v.add("{res} = {name};")
3095 v.add("\} else \{")
3096 var i = v.expr(self.n_expr, mtype)
3097 v.add("{res} = {i};")
3098 v.add("{name} = {res};")
3099 v.add("{guard} = 1;")
3100 v.add("\}")
3101 return res
3102 end
3103 end
3104
3105 redef class ASendExpr
3106 redef fun expr(v)
3107 do
3108 var recv = v.expr(self.n_expr, null)
3109 var callsite = self.callsite.as(not null)
3110 var args = v.varargize(callsite.mpropdef, callsite.signaturemap, recv, self.raw_arguments)
3111 return v.compile_callsite(callsite, args)
3112 end
3113 end
3114
3115 redef class ASendReassignFormExpr
3116 redef fun stmt(v)
3117 do
3118 var recv = v.expr(self.n_expr, null)
3119 var callsite = self.callsite.as(not null)
3120 var args = v.varargize(callsite.mpropdef, callsite.signaturemap, recv, self.raw_arguments)
3121
3122 var value = v.expr(self.n_value, null)
3123
3124 var left = v.compile_callsite(callsite, args)
3125 assert left != null
3126
3127 var res = v.compile_callsite(self.reassign_callsite.as(not null), [left, value])
3128 assert res != null
3129
3130 args.add(res)
3131 v.compile_callsite(self.write_callsite.as(not null), args)
3132 end
3133 end
3134
3135 redef class ASuperExpr
3136 redef fun expr(v)
3137 do
3138 var recv = v.frame.arguments.first
3139
3140 var callsite = self.callsite
3141 if callsite != null then
3142 var args
3143
3144 if self.n_args.n_exprs.is_empty then
3145 # Add automatic arguments for the super init call
3146 args = [recv]
3147 for i in [0..callsite.msignature.arity[ do
3148 args.add(v.frame.arguments[i+1])
3149 end
3150 else
3151 args = v.varargize(callsite.mpropdef, callsite.signaturemap, recv, self.n_args.n_exprs)
3152 end
3153
3154 # Super init call
3155 var res = v.compile_callsite(callsite, args)
3156 return res
3157 end
3158
3159 var mpropdef = self.mpropdef.as(not null)
3160
3161 var args
3162 if self.n_args.n_exprs.is_empty then
3163 args = v.frame.arguments
3164 else
3165 args = v.varargize(mpropdef, signaturemap, recv, self.n_args.n_exprs)
3166 end
3167
3168 # Standard call-next-method
3169 return v.supercall(mpropdef, recv.mtype.as(MClassType), args)
3170 end
3171 end
3172
3173 redef class ANewExpr
3174 redef fun expr(v)
3175 do
3176 var mtype = self.recvtype
3177 assert mtype != null
3178
3179 if mtype.mclass.name == "NativeArray" then
3180 assert self.n_args.n_exprs.length == 1
3181 var l = v.expr(self.n_args.n_exprs.first, null)
3182 assert mtype isa MGenericType
3183 var elttype = mtype.arguments.first
3184 return v.native_array_instance(elttype, l)
3185 end
3186
3187 var recv = v.init_instance_or_extern(mtype)
3188
3189 var callsite = self.callsite
3190 if callsite == null then return recv
3191
3192 var args = v.varargize(callsite.mpropdef, callsite.signaturemap, recv, self.n_args.n_exprs)
3193 var res2 = v.compile_callsite(callsite, args)
3194 if res2 != null then
3195 #self.debug("got {res2} from {mproperty}. drop {recv}")
3196 return res2
3197 end
3198 return recv
3199 end
3200 end
3201
3202 redef class AAttrExpr
3203 redef fun expr(v)
3204 do
3205 var recv = v.expr(self.n_expr, null)
3206 var mproperty = self.mproperty.as(not null)
3207 return v.read_attribute(mproperty, recv)
3208 end
3209 end
3210
3211 redef class AAttrAssignExpr
3212 redef fun expr(v)
3213 do
3214 var recv = v.expr(self.n_expr, null)
3215 var i = v.expr(self.n_value, null)
3216 var mproperty = self.mproperty.as(not null)
3217 v.write_attribute(mproperty, recv, i)
3218 return i
3219 end
3220 end
3221
3222 redef class AAttrReassignExpr
3223 redef fun stmt(v)
3224 do
3225 var recv = v.expr(self.n_expr, null)
3226 var value = v.expr(self.n_value, null)
3227 var mproperty = self.mproperty.as(not null)
3228 var attr = v.read_attribute(mproperty, recv)
3229 var res = v.compile_callsite(self.reassign_callsite.as(not null), [attr, value])
3230 assert res != null
3231 v.write_attribute(mproperty, recv, res)
3232 end
3233 end
3234
3235 redef class AIssetAttrExpr
3236 redef fun expr(v)
3237 do
3238 var recv = v.expr(self.n_expr, null)
3239 var mproperty = self.mproperty.as(not null)
3240 return v.isset_attribute(mproperty, recv)
3241 end
3242 end
3243
3244 redef class AVarargExpr
3245 redef fun expr(v)
3246 do
3247 return v.expr(self.n_expr, null)
3248 end
3249 end
3250
3251 redef class ANamedargExpr
3252 redef fun expr(v)
3253 do
3254 return v.expr(self.n_expr, null)
3255 end
3256 end
3257
3258 redef class ADebugTypeExpr
3259 redef fun stmt(v)
3260 do
3261 # do nothing
3262 end
3263 end
3264
3265 # Utils
3266
3267 redef class Array[E]
3268 # Return a new `Array` with the elements only contened in self and not in `o`
3269 fun -(o: Array[E]): Array[E] do
3270 var res = new Array[E]
3271 for e in self do if not o.has(e) then res.add(e)
3272 return res
3273 end
3274 end
3275
3276 redef class MModule
3277 # All `MProperty` associated to all `MClassDef` of `mclass`
3278 fun properties(mclass: MClass): Set[MProperty] do
3279 if not self.properties_cache.has_key(mclass) then
3280 var properties = new HashSet[MProperty]
3281 var parents = new Array[MClass]
3282 if self.flatten_mclass_hierarchy.has(mclass) then
3283 parents.add_all(mclass.in_hierarchy(self).direct_greaters)
3284 end
3285 for parent in parents do
3286 properties.add_all(self.properties(parent))
3287 end
3288 for mclassdef in mclass.mclassdefs do
3289 if not self.in_importation <= mclassdef.mmodule then continue
3290 for mprop in mclassdef.intro_mproperties do
3291 properties.add(mprop)
3292 end
3293 end
3294 self.properties_cache[mclass] = properties
3295 end
3296 return properties_cache[mclass]
3297 end
3298 private var properties_cache: Map[MClass, Set[MProperty]] = new HashMap[MClass, Set[MProperty]]
3299
3300 # Write FFI and nitni results to file
3301 fun finalize_ffi(c: AbstractCompiler) do end
3302
3303 # Give requided addinional system libraries (as given to LD_LIBS)
3304 # Note: can return null instead of an empty set
3305 fun collect_linker_libs: nullable Array[String] do return null
3306 end
3307
3308 # Create a tool context to handle options and paths
3309 var toolcontext = new ToolContext
3310
3311 toolcontext.tooldescription = "Usage: nitc [OPTION]... file.nit...\nCompiles Nit programs."
3312
3313 # We do not add other options, so process them now!
3314 toolcontext.process_options(args)
3315
3316 # We need a model to collect stufs
3317 var model = new Model
3318 # An a model builder to parse files
3319 var modelbuilder = new ModelBuilder(model, toolcontext)
3320
3321 var arguments = toolcontext.option_context.rest
3322 if arguments.length > 1 and toolcontext.opt_output.value != null then
3323 print "Option Error: --output needs a single source file. Do you prefer --dir?"
3324 exit 1
3325 end
3326
3327 # Here we load an process all modules passed on the command line
3328 var mmodules = modelbuilder.parse(arguments)
3329
3330 if mmodules.is_empty then return
3331 modelbuilder.run_phases
3332
3333 for mmodule in mmodules do
3334 toolcontext.info("*** PROCESS {mmodule} ***", 1)
3335 var ms = [mmodule]
3336 toolcontext.run_global_phases(ms)
3337 end