1 # This file is part of NIT ( http://www.nitlanguage.org ).
3 # Copyright 2012 Jean Privat <jean@pryen.org>
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
9 # http://www.apache.org/licenses/LICENSE-2.0
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.
18 module abstract_compiler
24 private import annotation
28 # Add compiling options
29 redef class ToolContext
31 var opt_output
= new OptionString("Filename of the generated executable", "-o", "--output")
33 var opt_dir
= new OptionString("Output directory", "--dir")
35 var opt_no_cc
= new OptionBool("Do not invoke the C compiler", "--no-cc")
37 var opt_no_main
= new OptionBool("Do not generate main entry point", "--no-main")
39 var opt_make_flags
= new OptionString("Additional options to the `make` command", "--make-flags")
41 var opt_max_c_lines
= new OptionInt("Maximum number of lines in generated C files. Use 0 for unlimited", 10000, "--max-c-lines")
43 var opt_group_c_files
= new OptionBool("Group all generated code in the same series of files", "--group-c-files")
45 var opt_compile_dir
= new OptionString("Directory used to generate temporary files", "--compile-dir")
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")
53 var opt_no_check_assert
= new OptionBool("Disable the evaluation of explicit `assert` and `as` (dangerous)", "--no-check-assert")
55 var opt_no_check_autocast
= new OptionBool("Disable implicit casts on unsafe expression usage (dangerous)", "--no-check-autocast")
57 var opt_no_check_null
= new OptionBool("Disable tests of null receiver (dangerous)", "--no-check-null")
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")
67 var opt_no_stacktrace
= new OptionBool("Disable the generation of stack traces", "--no-stacktrace")
69 var opt_no_gcc_directive
= new OptionArray("Disable advanced gcc directives for optimization", "--no-gcc-directive")
71 var opt_release
= new OptionBool("Compile in release mode and finalize application", "--release")
73 var opt_debug
= new OptionBool("Compile in debug mode (no C-side optimization)", "-g", "--debug")
78 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
)
79 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
)
80 self.option_context
.add_option
(self.opt_typing_test_metrics
, self.opt_invocation_metrics
, self.opt_isset_checks_metrics
)
81 self.option_context
.add_option
(self.opt_no_stacktrace
)
82 self.option_context
.add_option
(self.opt_no_gcc_directive
)
83 self.option_context
.add_option
(self.opt_release
)
84 self.option_context
.add_option
(self.opt_max_c_lines
, self.opt_group_c_files
)
85 self.option_context
.add_option
(self.opt_debug
)
87 opt_no_main
.hidden
= true
90 redef fun process_options
(args
)
94 if opt_output
.value
!= null and opt_dir
.value
!= null then
95 print
"Option Error: cannot use both --dir and --output"
99 if opt_no_check_all
.value
then
100 opt_no_check_covariance
.value
= true
101 opt_no_check_attr_isset
.value
= true
102 opt_no_check_assert
.value
= true
103 opt_no_check_autocast
.value
= true
104 opt_no_check_null
.value
= true
109 redef class ModelBuilder
110 # Simple indirection to `Toolchain::write_and_make`
111 protected fun write_and_make
(compiler
: AbstractCompiler)
113 var platform
= compiler
.target_platform
114 var toolchain
= platform
.toolchain
(toolcontext
, compiler
)
115 compiler
.toolchain
= toolchain
116 toolchain
.write_and_make
121 # The specific tool-chain associated to the platform
122 fun toolchain
(toolcontext
: ToolContext, compiler
: AbstractCompiler): Toolchain
124 return new MakefileToolchain(toolcontext
, compiler
)
128 # Build toolchain for a specific target program, varies per `Platform`
132 var toolcontext
: ToolContext
134 # Compiler of the target program
135 var compiler
: AbstractCompiler
137 # Directory where to generate all files
139 # The option `--compile_dir` change this directory.
140 fun root_compile_dir
: String
142 var compile_dir
= toolcontext
.opt_compile_dir
.value
143 if compile_dir
== null then compile_dir
= "nit_compile"
147 # Directory where to generate all C files
149 # By default it is `root_compile_dir` but some platform may require that it is a subdirectory.
150 fun compile_dir
: String do return root_compile_dir
152 # Write all C files and compile them
153 fun write_and_make
is abstract
156 # Default toolchain using a Makefile
157 class MakefileToolchain
160 redef fun write_and_make
162 var debug
= toolcontext
.opt_debug
.value
163 var compile_dir
= compile_dir
165 # Remove the compilation directory unless explicitly set
166 var auto_remove
= toolcontext
.opt_compile_dir
.value
== null
167 # If debug flag is set, do not remove sources
168 if debug
then auto_remove
= false
170 # Generate the .h and .c files
171 # A single C file regroups many compiled rumtime functions
172 # 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
174 self.toolcontext
.info
("*** WRITING C ***", 1)
176 root_compile_dir
.mkdir
179 var cfiles
= new Array[String]
180 write_files
(compile_dir
, cfiles
)
182 # Generate the Makefile
184 write_makefile
(compile_dir
, cfiles
)
187 self.toolcontext
.info
("*** END WRITING C: {time1-time0} ***", 2)
189 # Execute the Makefile
191 if self.toolcontext
.opt_no_cc
.value
then return
194 self.toolcontext
.info
("*** COMPILING C ***", 1)
196 compile_c_code
(compile_dir
)
199 sys
.system
("rm -r -- '{root_compile_dir.escape_to_sh}/'")
203 self.toolcontext
.info
("*** END COMPILING C: {time1-time0} ***", 2)
206 # Write all source files to the `compile_dir`
207 fun write_files
(compile_dir
: String, cfiles
: Array[String])
209 var platform
= compiler
.target_platform
210 if platform
.supports_libunwind
then compiler
.build_c_to_nit_bindings
211 var cc_opt_with_libgc
= "-DWITH_LIBGC"
212 if not platform
.supports_libgc
then cc_opt_with_libgc
= ""
214 # Add gc_choser.h to aditionnal bodies
215 var gc_chooser
= new ExternCFile("gc_chooser.c", cc_opt_with_libgc
)
216 if cc_opt_with_libgc
!= "" then gc_chooser
.pkgconfigs
.add
"bdw-gc"
217 compiler
.extern_bodies
.add
(gc_chooser
)
218 var clib
= toolcontext
.nit_dir
/ "clib"
219 compiler
.files_to_copy
.add
"{clib}/gc_chooser.c"
220 compiler
.files_to_copy
.add
"{clib}/gc_chooser.h"
223 for m
in compiler
.mainmodule
.in_importation
.greaters
do
224 compiler
.finalize_ffi_for_module
(m
)
227 # Copy original .[ch] files to compile_dir
228 for src
in compiler
.files_to_copy
do
229 var basename
= src
.basename
230 var dst
= "{compile_dir}/{basename}"
234 var hfilename
= compiler
.header
.file
.name
+ ".h"
235 var hfilepath
= "{compile_dir}/{hfilename}"
236 var h
= new FileWriter.open
(hfilepath
)
237 for l
in compiler
.header
.decl_lines
do
241 for l
in compiler
.header
.lines
do
247 var max_c_lines
= toolcontext
.opt_max_c_lines
.value
248 for f
in compiler
.files
do
251 var file
: nullable FileWriter = null
252 for vis
in f
.writers
do
253 if vis
== compiler
.header
then continue
254 var total_lines
= vis
.lines
.length
+ vis
.decl_lines
.length
255 if total_lines
== 0 then continue
257 if file
== null or (count
> max_c_lines
and max_c_lines
> 0) then
259 if file
!= null then file
.close
260 var cfilename
= "{f.name}.{i}.c"
261 var cfilepath
= "{compile_dir}/{cfilename}"
262 self.toolcontext
.info
("new C source files to compile: {cfilepath}", 3)
263 cfiles
.add
(cfilename
)
264 file
= new FileWriter.open
(cfilepath
)
265 file
.write
"#include \"{f.name}.0.h\
"\n"
268 for l
in vis
.decl_lines
do
272 for l
in vis
.lines
do
277 if file
== null then continue
280 var cfilename
= "{f.name}.0.h"
281 var cfilepath
= "{compile_dir}/{cfilename}"
282 var hfile
: nullable FileWriter = null
283 hfile
= new FileWriter.open
(cfilepath
)
284 hfile
.write
"#include \"{hfilename}\
"\n"
285 for key
in f
.required_declarations
do
286 if not compiler
.provided_declarations
.has_key
(key
) then
287 var node
= compiler
.requirers_of_declarations
.get_or_null
(key
)
289 node
.debug
"No provided declaration for {key}"
291 print
"No provided declaration for {key}"
295 hfile
.write compiler
.provided_declarations
[key
]
301 self.toolcontext
.info
("Total C source files to compile: {cfiles.length}", 2)
304 # Get the name of the Makefile to use
305 fun makefile_name
: String do return "{compiler.mainmodule.c_name}.mk"
307 # Get the default name of the executable to produce
308 fun default_outname
: String
310 var mainmodule
= compiler
.mainmodule
.first_real_mmodule
311 return mainmodule
.name
314 # Combine options and platform informations to get the final path of the outfile
315 fun outfile
(mainmodule
: MModule): String
317 var res
= self.toolcontext
.opt_output
.value
318 if res
!= null then return res
319 res
= default_outname
320 var dir
= self.toolcontext
.opt_dir
.value
321 if dir
!= null then return dir
.join_path
(res
)
326 fun write_makefile
(compile_dir
: String, cfiles
: Array[String])
328 var mainmodule
= compiler
.mainmodule
329 var platform
= compiler
.target_platform
331 var outname
= outfile
(mainmodule
)
333 var real_outpath
= compile_dir
.relpath
(outname
)
334 var outpath
= real_outpath
.escape_to_mk
335 if outpath
!= real_outpath
then
336 # If the name is crazy and need escaping, we will do an indirection
337 # 1. generate the binary in the nit_compile dir under an escaped name
338 # 2. copy the binary at the right place in the `all` goal.
339 outpath
= mainmodule
.c_name
341 var makename
= makefile_name
342 var makepath
= "{compile_dir}/{makename}"
343 var makefile
= new FileWriter.open
(makepath
)
345 var linker_options
= new HashSet[String]
346 for m
in mainmodule
.in_importation
.greaters
do
347 var libs
= m
.collect_linker_libs
348 if libs
!= null then linker_options
.add_all
(libs
)
350 var debug
= toolcontext
.opt_debug
.value
352 makefile
.write
("CC = ccache cc\nCXX = ccache c++\nCFLAGS = -g{ if not debug then " -O2 " else " "}-Wno-unused-value -Wno-switch -Wno-attributes -Wno-trigraphs\nCINCL =\nLDFLAGS ?= \nLDLIBS ?= -lm {linker_options.join(" ")}\n\n")
354 makefile
.write
"\n# SPECIAL CONFIGURATION FLAGS\n"
355 if platform
.supports_libunwind
then
356 if toolcontext
.opt_no_stacktrace
.value
then
357 makefile
.write
"NO_STACKTRACE=True"
359 makefile
.write
"NO_STACKTRACE= # Set to `True` to enable"
363 # Dynamic adaptations
364 # While `platform` enable complex toolchains, they are statically applied
365 # For a dynamic adaptsation of the compilation, the generated Makefile should check and adapt things itself
366 makefile
.write
"\n\n"
368 # Check and adapt the targeted system
369 makefile
.write
("uname_S := $(shell sh -c 'uname -s 2>/dev/null || echo not')\n")
371 # Check and adapt for the compiler used
372 # clang need an additionnal `-Qunused-arguments`
373 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")
375 if platform
.supports_libunwind
then
377 ifneq ($(NO_STACKTRACE), True)
378 # Check and include lib-unwind in a portable way
379 ifneq ($(uname_S),Darwin)
380 # already included on macosx, but need to get the correct flags in other supported platforms.
381 ifeq ($(shell pkg-config --exists 'libunwind'; echo $$?), 0)
382 LDLIBS += `pkg-config --libs libunwind`
383 CFLAGS += `pkg-config --cflags libunwind`
385 $(warning "[_] stack-traces disabled. Please install libunwind-dev.")
386 CFLAGS += -D NO_STACKTRACE
390 # Stacktraces disabled
391 CFLAGS += -D NO_STACKTRACE
396 makefile
.write
("CFLAGS += -D NO_STACKTRACE\n\n")
400 # Special configuration for Darwin
401 ifeq ($(uname_S),Darwin)
402 # Remove POSIX flag -lrt
403 LDLIBS := $(filter-out -lrt,$(LDLIBS))
408 makefile
.write
("all: {outpath}\n")
409 if outpath
!= real_outpath
then
410 makefile
.write
("\tcp -- {outpath.escape_to_sh} {real_outpath.escape_to_sh.replace("$","$$")}")
414 var ofiles
= new Array[String]
415 var dep_rules
= new Array[String]
416 # Compile each generated file
418 var o
= f
.strip_extension
(".c") + ".o"
419 makefile
.write
("{o}: {f}\n\t$(CC) $(CFLAGS) $(CINCL) -c -o {o} {f}\n\n")
424 # Generate linker script, if any
425 if not compiler
.linker_script
.is_empty
then
426 var linker_script_path
= "{compile_dir}/linker_script"
427 ofiles
.add
"linker_script"
428 var f
= new FileWriter.open
(linker_script_path
)
429 for l
in compiler
.linker_script
do
436 var java_files
= new Array[ExternFile]
438 var pkgconfigs
= new Array[String]
439 for f
in compiler
.extern_bodies
do
440 pkgconfigs
.add_all f
.pkgconfigs
443 if not pkgconfigs
.is_empty
then
445 # does pkg-config exists?
446 ifneq ($(shell which pkg-config >/dev/null; echo $$?), 0)
447 $(error "Command `pkg-config` not found. Please install it")
450 for p
in pkgconfigs
do
452 # Check for library {{{p}}}
453 ifneq ($(shell pkg-config --exists '{{{p}}}'; echo $$?), 0)
454 $(error "pkg-config: package {{{p}}} is not found.")
460 # Compile each required extern body into a specific .o
461 for f
in compiler
.extern_bodies
do
462 var o
= f
.makefile_rule_name
463 var ff
= f
.filename
.basename
464 makefile
.write
("{o}: {ff}\n")
465 makefile
.write
("\t{f.makefile_rule_content}\n\n")
466 dep_rules
.add
(f
.makefile_rule_name
)
468 if f
.compiles_to_o_file
then ofiles
.add
(o
)
469 if f
.add_to_jar
then java_files
.add
(f
)
472 if not java_files
.is_empty
then
473 var jar_file
= "{outpath}.jar"
475 var class_files_array
= new Array[String]
476 for f
in java_files
do class_files_array
.add
(f
.makefile_rule_name
)
477 var class_files
= class_files_array
.join
(" ")
479 makefile
.write
("{jar_file}: {class_files}\n")
480 makefile
.write
("\tjar cf {jar_file} {class_files}\n\n")
481 dep_rules
.add jar_file
486 if not pkgconfigs
.is_empty
then
487 pkg
= "`pkg-config --libs {pkgconfigs.join(" ")}`"
489 makefile
.write
("{outpath}: {dep_rules.join(" ")}\n\t$(CC) $(LDFLAGS) -o {outpath.escape_to_sh} {ofiles.join(" ")} $(LDLIBS) {pkg}\n\n")
491 makefile
.write
("clean:\n\trm {ofiles.join(" ")} 2>/dev/null\n")
492 if outpath
!= real_outpath
then
493 makefile
.write
("\trm -- {outpath.escape_to_sh} 2>/dev/null\n")
496 self.toolcontext
.info
("Generated makefile: {makepath}", 2)
498 makepath
.file_copy_to
"{compile_dir}/Makefile"
501 # The C code is generated, compile it to an executable
502 fun compile_c_code
(compile_dir
: String)
504 var makename
= makefile_name
506 var makeflags
= self.toolcontext
.opt_make_flags
.value
507 if makeflags
== null then makeflags
= ""
509 var command
= "make -B -C {compile_dir} -f {makename} -j 4 {makeflags}"
510 self.toolcontext
.info
(command
, 2)
513 if self.toolcontext
.verbose_level
>= 3 then
514 res
= sys
.system
("{command} 2>&1")
516 res
= sys
.system
("{command} 2>&1 >/dev/null")
519 toolcontext
.error
(null, "Compilation Error: `make` failed with error code: {res}. The command was `{command}`.")
524 # Singleton that store the knowledge about the compilation process
525 abstract class AbstractCompiler
526 type VISITOR: AbstractCompilerVisitor
528 # Table corresponding c_names to nit names (methods)
529 var names
= new HashMap[String, String]
531 # The main module of the program currently compiled
532 # Is assigned during the separate compilation
533 var mainmodule
: MModule is writable
535 # The real main module of the program
536 var realmainmodule
: MModule is noinit
538 # The modelbuilder used to know the model and the AST
539 var modelbuilder
: ModelBuilder is protected writable
541 # The associated toolchain
543 # Set by `modelbuilder.write_and_make` and permit sub-routines to access the current toolchain if required.
544 var toolchain
: Toolchain is noinit
546 # Is hardening asked? (see --hardening)
547 fun hardening
: Bool do return self.modelbuilder
.toolcontext
.opt_hardening
.value
549 # The targeted specific platform
550 var target_platform
: Platform is noinit
554 self.realmainmodule
= mainmodule
555 target_platform
= mainmodule
.target_platform
or else new Platform
558 # Do the full code generation of the program `mainmodule`
559 # It is the main method usually called after the instantiation
560 fun do_compilation
is abstract
562 # Force the creation of a new file
563 # The point is to avoid contamination between must-be-compiled-separately files
564 fun new_file
(name
: String): CodeFile
566 if modelbuilder
.toolcontext
.opt_group_c_files
.value
then
567 if self.files
.is_empty
then
568 var f
= new CodeFile(mainmodule
.c_name
)
571 return self.files
.first
573 var f
= new CodeFile(name
)
578 # The list of all associated files
579 # Used to generate .c files
580 var files
= new Array[CodeFile]
582 # Initialize a visitor specific for a compiler engine
583 fun new_visitor
: VISITOR is abstract
585 # Where global declaration are stored (the main .h)
586 var header
: CodeWriter is writable, noinit
588 # Additionnal linker script for `ld`.
589 # Mainly used to do specific link-time symbol resolution
590 var linker_script
= new Array[String]
592 # Provide a declaration that can be requested (before or latter) by a visitor
593 fun provide_declaration
(key
: String, s
: String)
595 if self.provided_declarations
.has_key
(key
) then
596 assert self.provided_declarations
[key
] == s
598 self.provided_declarations
[key
] = s
601 private var provided_declarations
= new HashMap[String, String]
603 private var requirers_of_declarations
= new HashMap[String, ANode]
605 # Builds the .c and .h files to be used when generating a Stack Trace
606 # Binds the generated C function names to Nit function names
607 fun build_c_to_nit_bindings
609 var compile_dir
= toolchain
.compile_dir
611 var stream
= new FileWriter.open
("{compile_dir}/c_functions_hash.c")
612 stream
.write
("#include <string.h>\n")
613 stream
.write
("#include <stdlib.h>\n")
614 stream
.write
("#include \"c_functions_hash
.h\
"\n")
615 stream
.write
("typedef struct C_Nit_Names\{char* name; char* nit_name;\}C_Nit_Names;\n")
616 stream
.write
("const char* get_nit_name(register const char* procproc, register unsigned int len)\{\n")
617 stream
.write
("char* procname = malloc(len+1);")
618 stream
.write
("memcpy(procname, procproc, len);")
619 stream
.write
("procname[len] = '\\0';")
620 stream
.write
("static const C_Nit_Names map[{names.length}] = \{\n")
621 for i
in names
.keys
do
623 stream.write(i.escape_to_c)
624 stream.write("\
",\"")
625 stream.write(names[i].escape_to_c)
626 stream.write("\
"\},\n")
628 stream
.write
("\};\n")
629 stream
.write
("int i;")
630 stream
.write
("for(i = 0; i < {names.length}; i++)\{")
631 stream
.write
("if(strcmp(procname,map[i].name) == 0)\{")
632 stream
.write
("free(procname);")
633 stream
.write
("return map[i].nit_name;")
636 stream
.write
("free(procname);")
637 stream
.write
("return NULL;")
641 stream
= new FileWriter.open
("{compile_dir}/c_functions_hash.h")
642 stream
.write
("const char* get_nit_name(register const char* procname, register unsigned int len);\n")
645 extern_bodies
.add
(new ExternCFile("{compile_dir}/c_functions_hash.c", ""))
649 # This method call compile_header_strucs method that has to be refined
650 fun compile_header
do
651 self.header
.add_decl
("#include <stdlib.h>")
652 self.header
.add_decl
("#include <stdio.h>")
653 self.header
.add_decl
("#include <string.h>")
655 self.header
.add_decl
("#include <setjmp.h>\n")
656 self.header
.add_decl
("#include <sys/types.h>\n")
657 self.header
.add_decl
("#include <unistd.h>\n")
658 self.header
.add_decl
("#include <stdint.h>\n")
659 self.header
.add_decl
("#ifdef __linux__")
660 self.header
.add_decl
(" #include <endian.h>")
661 self.header
.add_decl
("#endif")
662 self.header
.add_decl
("#include <inttypes.h>\n")
663 self.header
.add_decl
("#include \"gc_chooser
.h\
"")
664 self.header
.add_decl
("#ifdef __APPLE__")
665 self.header
.add_decl
(" #include <libkern/OSByteOrder.h>")
666 self.header
.add_decl
(" #define be32toh(x) OSSwapBigToHostInt32(x)")
667 self.header
.add_decl
("#endif")
668 self.header
.add_decl
("#ifdef __pnacl__")
669 self.header
.add_decl
(" #define be16toh(val) (((val) >> 8) | ((val) << 8))")
670 self.header
.add_decl
(" #define be32toh(val) ((be16toh((val) << 16) | (be16toh((val) >> 16))))")
671 self.header
.add_decl
("#endif")
672 self.header
.add_decl
("#ifdef ANDROID")
673 self.header
.add_decl
(" #ifndef be32toh")
674 self.header
.add_decl
(" #define be32toh(val) betoh32(val)")
675 self.header
.add_decl
(" #endif")
676 self.header
.add_decl
(" #include <android/log.h>")
677 self.header
.add_decl
(" #define PRINT_ERROR(...) (void)__android_log_print(ANDROID_LOG_WARN, \"Nit\
", __VA_ARGS__)")
678 self.header
.add_decl
("#else")
679 self.header
.add_decl
(" #define PRINT_ERROR(...) fprintf(stderr, __VA_ARGS__)")
680 self.header
.add_decl
("#endif")
682 compile_header_structs
683 compile_nitni_structs
686 var gccd_disable
= modelbuilder
.toolcontext
.opt_no_gcc_directive
.value
687 if gccd_disable
.has
("noreturn") or gccd_disable
.has
("all") then
688 # Signal handler function prototype
689 self.header
.add_decl
("void fatal_exit(int);")
691 self.header
.add_decl
("void fatal_exit(int) __attribute__ ((noreturn));")
694 if gccd_disable
.has
("likely") or gccd_disable
.has
("all") then
695 self.header
.add_decl
("#define likely(x) (x)")
696 self.header
.add_decl
("#define unlikely(x) (x)")
697 else if gccd_disable
.has
("correct-likely") then
698 # invert the `likely` definition
699 # Used by masochists to bench the worst case
700 self.header
.add_decl
("#define likely(x) __builtin_expect((x),0)")
701 self.header
.add_decl
("#define unlikely(x) __builtin_expect((x),1)")
703 self.header
.add_decl
("#define likely(x) __builtin_expect((x),1)")
704 self.header
.add_decl
("#define unlikely(x) __builtin_expect((x),0)")
707 # Global variable used by intern methods
708 self.header
.add_decl
("extern int glob_argc;")
709 self.header
.add_decl
("extern char **glob_argv;")
710 self.header
.add_decl
("extern val *glob_sys;")
713 # Stack stocking environment for longjumps
714 protected fun compile_catch_stack
do
715 self.header
.add_decl
"""
716 struct catch_stack_t {
720 extern struct catch_stack_t catchStack;
724 # Declaration of structures for live Nit types
725 protected fun compile_header_structs
is abstract
727 # Declaration of structures for nitni undelying the FFI
728 protected fun compile_nitni_structs
730 self.header
.add_decl
"""
731 /* Native reference to Nit objects */
732 /* This structure is used to represent every Nit type in extern methods and custom C code. */
734 struct nitni_ref *next,
735 *prev; /* adjacent global references in global list */
736 int count; /* number of time this global reference has been marked */
739 /* List of global references from C code to Nit objects */
740 /* Instanciated empty at init of Nit system and filled explicitly by user in C code */
741 struct nitni_global_ref_list_t {
742 struct nitni_ref *head, *tail;
744 extern struct nitni_global_ref_list_t *nitni_global_ref_list;
746 /* Initializer of global reference list */
747 extern void nitni_global_ref_list_init();
749 /* Intern function to add a global reference to the list */
750 extern void nitni_global_ref_add( struct nitni_ref *ref );
752 /* Intern function to remove a global reference from the list */
753 extern void nitni_global_ref_remove( struct nitni_ref *ref );
755 /* Increase count on an existing global reference */
756 extern void nitni_global_ref_incr( struct nitni_ref *ref );
758 /* Decrease count on an existing global reference */
759 extern void nitni_global_ref_decr( struct nitni_ref *ref );
763 fun compile_finalizer_function
765 var finalizable_type
= mainmodule
.finalizable_type
766 if finalizable_type
== null then return
768 var finalize_meth
= mainmodule
.try_get_primitive_method
("finalize", finalizable_type
.mclass
)
770 if finalize_meth
== null then
771 modelbuilder
.toolcontext
.error
(null, "Error: the `Finalizable` class does not declare the `finalize` method.")
775 var v
= self.new_visitor
776 v
.add_decl
"void gc_finalize (void *obj, void *client_data) \{"
777 var recv
= v
.new_expr
("obj", finalizable_type
)
778 v
.send
(finalize_meth
, [recv
])
782 # Generate the main C function.
786 # * allocate the Sys object if it exists
787 # * call init if is exists
788 # * call main if it exists
789 fun compile_main_function
791 var v
= self.new_visitor
792 v
.add_decl
("#include <signal.h>")
793 var platform
= target_platform
795 var no_main
= platform
.no_main
or modelbuilder
.toolcontext
.opt_no_main
.value
797 if platform
.supports_libunwind
then
798 v
.add_decl
("#ifndef NO_STACKTRACE")
799 v
.add_decl
("#define UNW_LOCAL_ONLY")
800 v
.add_decl
("#include <libunwind.h>")
801 v
.add_decl
("#include \"c_functions_hash
.h\
"")
804 v
.add_decl
("int glob_argc;")
805 v
.add_decl
("char **glob_argv;")
806 v
.add_decl
("val *glob_sys;")
807 v
.add_decl
("struct catch_stack_t catchStack;")
809 if self.modelbuilder
.toolcontext
.opt_typing_test_metrics
.value
then
810 for tag
in count_type_test_tags
do
811 v
.add_decl
("long count_type_test_resolved_{tag};")
812 v
.add_decl
("long count_type_test_unresolved_{tag};")
813 v
.add_decl
("long count_type_test_skipped_{tag};")
814 v
.compiler
.header
.add_decl
("extern long count_type_test_resolved_{tag};")
815 v
.compiler
.header
.add_decl
("extern long count_type_test_unresolved_{tag};")
816 v
.compiler
.header
.add_decl
("extern long count_type_test_skipped_{tag};")
820 if self.modelbuilder
.toolcontext
.opt_invocation_metrics
.value
then
821 v
.add_decl
("long count_invoke_by_tables;")
822 v
.add_decl
("long count_invoke_by_direct;")
823 v
.add_decl
("long count_invoke_by_inline;")
824 v
.compiler
.header
.add_decl
("extern long count_invoke_by_tables;")
825 v
.compiler
.header
.add_decl
("extern long count_invoke_by_direct;")
826 v
.compiler
.header
.add_decl
("extern long count_invoke_by_inline;")
829 if self.modelbuilder
.toolcontext
.opt_isset_checks_metrics
.value
then
830 v
.add_decl
("long count_attr_reads = 0;")
831 v
.add_decl
("long count_isset_checks = 0;")
832 v
.compiler
.header
.add_decl
("extern long count_attr_reads;")
833 v
.compiler
.header
.add_decl
("extern long count_isset_checks;")
836 v
.add_decl
("static void show_backtrace(void) \{")
837 if platform
.supports_libunwind
then
838 v
.add_decl
("#ifndef NO_STACKTRACE")
839 v
.add_decl
("char* opt = getenv(\"NIT_NO_STACK\
");")
840 v
.add_decl
("unw_cursor_t cursor;")
841 v
.add_decl
("if(opt==NULL)\{")
842 v
.add_decl
("unw_context_t uc;")
843 v
.add_decl
("unw_word_t ip;")
844 v
.add_decl
("char* procname = malloc(sizeof(char) * 100);")
845 v
.add_decl
("unw_getcontext(&uc);")
846 v
.add_decl
("unw_init_local(&cursor, &uc);")
847 v
.add_decl
("PRINT_ERROR(\"-------------------------------------------------\\n\
");")
848 v
.add_decl
("PRINT_ERROR(\"-- Stack Trace ------------------------------\\n\
");")
849 v
.add_decl
("PRINT_ERROR(\"-------------------------------------------------\\n\
");")
850 v
.add_decl
("while (unw_step(&cursor) > 0) \{")
851 v
.add_decl
(" unw_get_proc_name(&cursor, procname, 100, &ip);")
852 v
.add_decl
(" const char* recv = get_nit_name(procname, strlen(procname));")
853 v
.add_decl
(" if (recv != NULL)\{")
854 v
.add_decl
(" PRINT_ERROR(\"` %s\\n\", recv);")
855 v.add_decl(" \}else\{")
856 v.add_decl(" PRINT_ERROR(\"` %s\\n\
", procname);")
859 v
.add_decl
("PRINT_ERROR(\"-------------------------------------------------\\n\
");")
860 v
.add_decl
("free(procname);")
862 v
.add_decl
("#endif /* NO_STACKTRACE */")
866 v
.add_decl
("void sig_handler(int signo)\{")
867 v
.add_decl
("PRINT_ERROR(\"Caught signal
: %s\\n\
", strsignal(signo));")
868 v
.add_decl
("show_backtrace();")
870 v
.add_decl
("signal(signo, SIG_DFL);")
871 v
.add_decl
("kill(getpid(), signo);")
874 v
.add_decl
("void fatal_exit(int status) \{")
875 v
.add_decl
("show_backtrace();")
876 v
.add_decl
("exit(status);")
880 v
.add_decl
("int nit_main(int argc, char** argv) \{")
882 v
.add_decl
("int main(int argc, char** argv) \{")
885 v
.add
"#if !defined(__ANDROID__) && !defined(TARGET_OS_IPHONE)"
886 v
.add
("signal(SIGABRT, sig_handler);")
887 v
.add
("signal(SIGFPE, sig_handler);")
888 v
.add
("signal(SIGILL, sig_handler);")
889 v
.add
("signal(SIGINT, sig_handler);")
890 v
.add
("signal(SIGTERM, sig_handler);")
891 v
.add
("signal(SIGSEGV, sig_handler);")
893 v
.add
("signal(SIGPIPE, SIG_IGN);")
895 v
.add
("glob_argc = argc; glob_argv = argv;")
896 v
.add
("catchStack.cursor = -1;")
897 v
.add
("initialize_gc_option();")
899 v
.add
"initialize_nitni_global_refs();"
901 var main_type
= mainmodule
.sys_type
902 if main_type
!= null then
903 var mainmodule
= v
.compiler
.mainmodule
904 var glob_sys
= v
.init_instance
(main_type
)
905 v
.add
("glob_sys = {glob_sys};")
906 var main_init
= mainmodule
.try_get_primitive_method
("init", main_type
.mclass
)
907 if main_init
!= null then
908 v
.send
(main_init
, [glob_sys
])
910 var main_method
= mainmodule
.try_get_primitive_method
("run", main_type
.mclass
) or else
911 mainmodule
.try_get_primitive_method
("main", main_type
.mclass
)
912 if main_method
!= null then
913 v
.send
(main_method
, [glob_sys
])
917 if self.modelbuilder
.toolcontext
.opt_typing_test_metrics
.value
then
918 v
.add_decl
("long count_type_test_resolved_total = 0;")
919 v
.add_decl
("long count_type_test_unresolved_total = 0;")
920 v
.add_decl
("long count_type_test_skipped_total = 0;")
921 v
.add_decl
("long count_type_test_total_total = 0;")
922 for tag
in count_type_test_tags
do
923 v
.add_decl
("long count_type_test_total_{tag};")
924 v
.add
("count_type_test_total_{tag} = count_type_test_resolved_{tag} + count_type_test_unresolved_{tag} + count_type_test_skipped_{tag};")
925 v
.add
("count_type_test_resolved_total += count_type_test_resolved_{tag};")
926 v
.add
("count_type_test_unresolved_total += count_type_test_unresolved_{tag};")
927 v
.add
("count_type_test_skipped_total += count_type_test_skipped_{tag};")
928 v
.add
("count_type_test_total_total += count_type_test_total_{tag};")
930 v
.add
("printf(\"# dynamic count_type_test: total %l\\n\");")
931 v
.add
("printf(\"\\tresolved\\tunresolved\\tskipped\\ttotal\\n\
");")
932 var tags
= count_type_test_tags
.to_a
935 v
.add
("printf(\"{tag}\
");")
936 v
.add
("printf(\"\\t
%ld
(%.2f
%%)\
", count_type_test_resolved_{tag}, 100.0*count_type_test_resolved_{tag}/count_type_test_total_total);")
937 v
.add
("printf(\"\\t
%ld
(%.2f
%%)\
", count_type_test_unresolved_{tag}, 100.0*count_type_test_unresolved_{tag}/count_type_test_total_total);")
938 v
.add
("printf(\"\\t
%ld
(%.2f
%%)\
", count_type_test_skipped_{tag}, 100.0*count_type_test_skipped_{tag}/count_type_test_total_total);")
939 v
.add
("printf(\"\\t
%ld
(%.2f
%%)\\n\
", count_type_test_total_{tag}, 100.0*count_type_test_total_{tag}/count_type_test_total_total);")
943 if self.modelbuilder
.toolcontext
.opt_invocation_metrics
.value
then
944 v
.add_decl
("long count_invoke_total;")
945 v
.add
("count_invoke_total = count_invoke_by_tables + count_invoke_by_direct + count_invoke_by_inline;")
946 v
.add
("printf(\"# dynamic count_invocation: total %ld\\n\", count_invoke_total);")
947 v
.add
("printf(\"by table
: %ld
(%.2f
%%)\\n\
", count_invoke_by_tables, 100.0*count_invoke_by_tables/count_invoke_total);")
948 v
.add
("printf(\"direct
: %ld
(%.2f
%%)\\n\
", count_invoke_by_direct, 100.0*count_invoke_by_direct/count_invoke_total);")
949 v
.add
("printf(\"inlined
: %ld
(%.2f
%%)\\n\
", count_invoke_by_inline, 100.0*count_invoke_by_inline/count_invoke_total);")
952 if self.modelbuilder
.toolcontext
.opt_isset_checks_metrics
.value
then
953 v
.add
("printf(\"# dynamic attribute reads: %ld\\n\", count_attr_reads);")
954 v
.add
("printf(\"# dynamic isset checks: %ld\\n\", count_isset_checks);")
960 for m
in mainmodule
.in_importation
.greaters
do
961 var f
= "FILE_"+m
.c_name
962 v
.add
"const char {f}[] = \"{m.location.file.filename.escape_to_c}\
";"
963 provide_declaration
(f
, "extern const char {f}[];")
967 # Copile all C functions related to the [incr|decr]_ref features of the FFI
968 fun compile_nitni_global_ref_functions
970 var v
= self.new_visitor
972 struct nitni_global_ref_list_t *nitni_global_ref_list;
973 void initialize_nitni_global_refs() {
974 nitni_global_ref_list = (struct nitni_global_ref_list_t*)nit_alloc(sizeof(struct nitni_global_ref_list_t));
975 nitni_global_ref_list->head = NULL;
976 nitni_global_ref_list->tail = NULL;
979 void nitni_global_ref_add( struct nitni_ref *ref ) {
980 if ( nitni_global_ref_list->head == NULL ) {
981 nitni_global_ref_list->head = ref;
984 nitni_global_ref_list->tail->next = ref;
985 ref->prev = nitni_global_ref_list->tail;
987 nitni_global_ref_list->tail = ref;
992 void nitni_global_ref_remove( struct nitni_ref *ref ) {
993 if ( ref->prev == NULL ) {
994 nitni_global_ref_list->head = ref->next;
996 ref->prev->next = ref->next;
999 if ( ref->next == NULL ) {
1000 nitni_global_ref_list->tail = ref->prev;
1002 ref->next->prev = ref->prev;
1006 extern void nitni_global_ref_incr( struct nitni_ref *ref ) {
1007 if ( ref->count == 0 ) /* not registered */
1010 nitni_global_ref_add( ref );
1016 extern void nitni_global_ref_decr( struct nitni_ref *ref ) {
1017 if ( ref->count == 1 ) /* was last reference */
1019 /* remove from list */
1020 nitni_global_ref_remove( ref );
1028 # List of additional files required to compile (FFI)
1029 var extern_bodies
= new Array[ExternFile]
1031 # List of source files to copy over to the compile dir
1032 var files_to_copy
= new Array[String]
1034 # This is used to avoid adding an extern file more than once
1035 private var seen_extern
= new ArraySet[String]
1037 # Generate code that initialize the attributes on a new instance
1038 fun generate_init_attr
(v
: VISITOR, recv
: RuntimeVariable, mtype
: MClassType)
1040 var cds
= mtype
.collect_mclassdefs
(self.mainmodule
).to_a
1041 self.mainmodule
.linearize_mclassdefs
(cds
)
1043 for npropdef
in modelbuilder
.collect_attr_propdef
(cd
) do
1044 npropdef
.init_expr
(v
, recv
)
1049 # Generate code that check if an attribute is correctly initialized
1050 fun generate_check_attr
(v
: VISITOR, recv
: RuntimeVariable, mtype
: MClassType)
1052 var cds
= mtype
.collect_mclassdefs
(self.mainmodule
).to_a
1053 self.mainmodule
.linearize_mclassdefs
(cds
)
1055 for npropdef
in modelbuilder
.collect_attr_propdef
(cd
) do
1056 npropdef
.check_expr
(v
, recv
)
1063 var count_type_test_tags
: Array[String] = ["isa", "as", "auto", "covariance", "erasure"]
1064 var count_type_test_resolved
: HashMap[String, Int] = init_count_type_test_tags
1065 var count_type_test_unresolved
: HashMap[String, Int] = init_count_type_test_tags
1066 var count_type_test_skipped
: HashMap[String, Int] = init_count_type_test_tags
1068 protected fun init_count_type_test_tags
: HashMap[String, Int]
1070 var res
= new HashMap[String, Int]
1071 for tag
in count_type_test_tags
do
1077 # Display stats about compilation process
1081 # * type tests against resolved types (`x isa Collection[Animal]`)
1082 # * type tests against unresolved types (`x isa Collection[E]`)
1083 # * type tests skipped
1084 # * type tests total
1087 if self.modelbuilder
.toolcontext
.opt_typing_test_metrics
.value
then
1088 print
"# static count_type_test"
1089 print
"\tresolved:\tunresolved\tskipped\ttotal"
1090 var count_type_test_total
= init_count_type_test_tags
1091 count_type_test_resolved
["total"] = 0
1092 count_type_test_unresolved
["total"] = 0
1093 count_type_test_skipped
["total"] = 0
1094 count_type_test_total
["total"] = 0
1095 for tag
in count_type_test_tags
do
1096 count_type_test_total
[tag
] = count_type_test_resolved
[tag
] + count_type_test_unresolved
[tag
] + count_type_test_skipped
[tag
]
1097 count_type_test_resolved
["total"] += count_type_test_resolved
[tag
]
1098 count_type_test_unresolved
["total"] += count_type_test_unresolved
[tag
]
1099 count_type_test_skipped
["total"] += count_type_test_skipped
[tag
]
1100 count_type_test_total
["total"] += count_type_test_total
[tag
]
1102 var count_type_test
= count_type_test_total
["total"]
1103 var tags
= count_type_test_tags
.to_a
1107 printn
"\t{count_type_test_resolved[tag]} ({div(count_type_test_resolved[tag],count_type_test)}%)"
1108 printn
"\t{count_type_test_unresolved[tag]} ({div(count_type_test_unresolved[tag],count_type_test)}%)"
1109 printn
"\t{count_type_test_skipped[tag]} ({div(count_type_test_skipped[tag],count_type_test)}%)"
1110 printn
"\t{count_type_test_total[tag]} ({div(count_type_test_total[tag],count_type_test)}%)"
1116 fun finalize_ffi_for_module
(mmodule
: MModule) do mmodule
.finalize_ffi
(self)
1119 # A file unit (may be more than one file if
1120 # A file unit aim to be autonomous and is made or one or more `CodeWriter`s
1123 var writers
= new Array[CodeWriter]
1124 var required_declarations
= new HashSet[String]
1127 # Where to store generated lines
1130 var lines
= new Array[String]
1131 var decl_lines
= new Array[String]
1133 # Add a line in the main part of the generated C
1134 fun add
(s
: String) do self.lines
.add
(s
)
1137 # (used for local or global declaration)
1138 fun add_decl
(s
: String) do self.decl_lines
.add
(s
)
1142 file
.writers
.add
(self)
1146 # A visitor on the AST of property definition that generate the C code.
1147 abstract class AbstractCompilerVisitor
1149 type COMPILER: AbstractCompiler
1151 # The associated compiler
1152 var compiler
: COMPILER
1154 # The current visited AST node
1155 var current_node
: nullable ANode = null is writable
1157 # The current `StaticFrame`
1158 var frame
: nullable StaticFrame = null is writable
1160 # Alias for self.compiler.mainmodule.object_type
1161 fun object_type
: MClassType do return self.compiler
.mainmodule
.object_type
1163 # Alias for self.compiler.mainmodule.bool_type
1164 fun bool_type
: MClassType do return self.compiler
.mainmodule
.bool_type
1166 var writer
: CodeWriter is noinit
1170 self.writer
= new CodeWriter(compiler
.files
.last
)
1173 # Force to get the primitive property named `name` in the instance `recv` or abort
1174 fun get_property
(name
: String, recv
: MType): MMethod
1176 assert recv
isa MClassType
1177 return self.compiler
.modelbuilder
.force_get_primitive_method
(self.current_node
, name
, recv
.mclass
, self.compiler
.mainmodule
)
1180 fun compile_callsite
(callsite
: CallSite, arguments
: Array[RuntimeVariable]): nullable RuntimeVariable
1182 if callsite
.is_broken
then return null
1183 var initializers
= callsite
.mpropdef
.initializers
1184 if not initializers
.is_empty
then
1185 var recv
= arguments
.first
1188 for p
in initializers
do
1189 if p
isa MMethod then
1191 for x
in p
.intro
.msignature
.mparameters
do
1192 args
.add arguments
[i
]
1196 else if p
isa MAttribute then
1197 self.write_attribute
(p
, recv
, arguments
[i
])
1201 assert i
== arguments
.length
1203 return self.send
(callsite
.mproperty
, [recv
])
1206 return self.send
(callsite
.mproperty
, arguments
)
1209 fun native_array_instance
(elttype
: MType, length
: RuntimeVariable): RuntimeVariable is abstract
1211 fun calloc_array
(ret_type
: MType, arguments
: Array[RuntimeVariable]) is abstract
1213 fun native_array_def
(pname
: String, ret_type
: nullable MType, arguments
: Array[RuntimeVariable]): Bool do return false
1215 # Return an element of a native array.
1216 # The method is unsafe and is just a direct wrapper for the specific implementation of native arrays
1217 fun native_array_get
(native_array
: RuntimeVariable, index
: Int): RuntimeVariable is abstract
1219 # Store an element in a native array.
1220 # The method is unsafe and is just a direct wrapper for the specific implementation of native arrays
1221 fun native_array_set
(native_array
: RuntimeVariable, index
: Int, value
: RuntimeVariable) is abstract
1223 # Evaluate `args` as expressions in the call of `mpropdef` on `recv`.
1224 # This method is used to manage varargs in signatures and returns the real array
1225 # of runtime variables to use in the call.
1226 fun varargize
(mpropdef
: MMethodDef, map
: nullable SignatureMap, recv
: RuntimeVariable, args
: SequenceRead[AExpr]): Array[RuntimeVariable]
1228 var msignature
= mpropdef
.new_msignature
or else mpropdef
.msignature
.as(not null)
1229 var res
= new Array[RuntimeVariable]
1232 if msignature
.arity
== 0 then return res
1235 assert args
.length
== msignature
.arity
1237 res
.add
self.expr
(ne
, null)
1242 # Eval in order of arguments, not parameters
1243 var exprs
= new Array[RuntimeVariable].with_capacity
(args
.length
)
1245 exprs
.add
self.expr
(ne
, null)
1248 # Fill `res` with the result of the evaluation according to the mapping
1249 for i
in [0..msignature
.arity
[ do
1250 var param
= msignature
.mparameters
[i
]
1251 var j
= map
.map
.get_or_null
(i
)
1254 res
.add
(null_instance
)
1257 if param
.is_vararg
and args
[i
].vararg_decl
> 0 then
1258 var vararg
= exprs
.sub
(j
, args
[i
].vararg_decl
)
1259 var elttype
= param
.mtype
1260 var arg
= self.vararg_instance
(mpropdef
, recv
, vararg
, elttype
)
1271 # Anchor a type to the main module and the current receiver
1272 fun anchor
(mtype
: MType): MType
1274 if not mtype
.need_anchor
then return mtype
1275 return mtype
.anchor_to
(self.compiler
.mainmodule
, self.frame
.receiver
)
1278 fun resolve_for
(mtype
: MType, recv
: RuntimeVariable): MType
1280 if not mtype
.need_anchor
then return mtype
1281 return mtype
.resolve_for
(recv
.mcasttype
, self.frame
.receiver
, self.compiler
.mainmodule
, true)
1284 # Unsafely cast a value to a new type
1285 # ie the result share the same C variable but my have a different mcasttype
1286 # NOTE: if the adaptation is useless then `value` is returned as it.
1287 # ENSURE: `result.name == value.name`
1288 fun autoadapt
(value
: RuntimeVariable, mtype
: MType): RuntimeVariable
1290 mtype
= self.anchor
(mtype
)
1291 var valmtype
= value
.mcasttype
1293 # Do nothing if useless autocast
1294 if valmtype
.is_subtype
(self.compiler
.mainmodule
, null, mtype
) then
1298 # Just as_not_null if the target is not nullable.
1300 # eg `nullable PreciseType` adapted to `Object` gives precisetype.
1301 if valmtype
isa MNullableType and valmtype
.mtype
.is_subtype
(self.compiler
.mainmodule
, null, mtype
) then
1302 mtype
= valmtype
.mtype
1305 var res
= new RuntimeVariable(value
.name
, value
.mtype
, mtype
)
1309 # Generate a super call from a method definition
1310 fun supercall
(m
: MMethodDef, recvtype
: MClassType, args
: Array[RuntimeVariable]): nullable RuntimeVariable is abstract
1312 # Adapt the arguments of a method according to targetted `MMethodDef`
1313 fun adapt_signature
(m
: MMethodDef, args
: Array[RuntimeVariable]) is abstract
1315 # Unbox all the arguments of a method when implemented `extern` or `intern`
1316 fun unbox_signature_extern
(m
: MMethodDef, args
: Array[RuntimeVariable]) is abstract
1318 # Box or unbox a value to another type iff a C type conversion is needed
1319 # ENSURE: `result.mtype.ctype == mtype.ctype`
1320 fun autobox
(value
: RuntimeVariable, mtype
: MType): RuntimeVariable is abstract
1322 # Box extern classes to be used in the generated code
1323 fun box_extern
(value
: RuntimeVariable, mtype
: MType): RuntimeVariable is abstract
1325 # Unbox extern classes to be used in extern code (legacy NI and FFI)
1326 fun unbox_extern
(value
: RuntimeVariable, mtype
: MType): RuntimeVariable is abstract
1328 # Generate a polymorphic subtype test
1329 fun type_test
(value
: RuntimeVariable, mtype
: MType, tag
: String): RuntimeVariable is abstract
1331 # Generate the code required to dynamically check if 2 objects share the same runtime type
1332 fun is_same_type_test
(value1
, value2
: RuntimeVariable): RuntimeVariable is abstract
1334 # Generate a Nit "is" for two runtime_variables
1335 fun equal_test
(value1
, value2
: RuntimeVariable): RuntimeVariable is abstract
1339 # Generate a static call on a method definition
1340 fun call
(m
: MMethodDef, recvtype
: MClassType, args
: Array[RuntimeVariable]): nullable RuntimeVariable is abstract
1342 # Generate a polymorphic send for the method `m` and the arguments `args`
1343 fun send
(m
: MMethod, args
: Array[RuntimeVariable]): nullable RuntimeVariable is abstract
1345 # Generate a monomorphic send for the method `m`, the type `t` and the arguments `args`
1346 fun monomorphic_send
(m
: MMethod, t
: MType, args
: Array[RuntimeVariable]): nullable RuntimeVariable
1348 assert t
isa MClassType
1349 var propdef
= m
.lookup_first_definition
(self.compiler
.mainmodule
, t
)
1350 return self.call
(propdef
, t
, args
)
1353 # Generate a monomorphic super send from the method `m`, the type `t` and the arguments `args`
1354 fun monomorphic_super_send
(m
: MMethodDef, t
: MType, args
: Array[RuntimeVariable]): nullable RuntimeVariable
1356 assert t
isa MClassType
1357 m
= m
.lookup_next_definition
(self.compiler
.mainmodule
, t
)
1358 return self.call
(m
, t
, args
)
1361 # Attributes handling
1363 # Generate a polymorphic attribute is_set test
1364 fun isset_attribute
(a
: MAttribute, recv
: RuntimeVariable): RuntimeVariable is abstract
1366 # Generate a polymorphic attribute read
1367 fun read_attribute
(a
: MAttribute, recv
: RuntimeVariable): RuntimeVariable is abstract
1369 # Generate a polymorphic attribute write
1370 fun write_attribute
(a
: MAttribute, recv
: RuntimeVariable, value
: RuntimeVariable) is abstract
1374 # Can value be null? (according to current knowledge)
1375 fun maybenull
(value
: RuntimeVariable): Bool
1377 return value
.mcasttype
isa MNullableType or value
.mcasttype
isa MNullType
1380 # Add a check and an abort for a null receiver if needed
1381 fun check_recv_notnull
(recv
: RuntimeVariable)
1383 if self.compiler
.modelbuilder
.toolcontext
.opt_no_check_null
.value
then return
1385 if maybenull
(recv
) then
1386 self.add
("if (unlikely({recv} == NULL)) \{")
1387 self.add_abort
("Receiver is null")
1394 private var names
= new HashSet[String]
1395 private var last
: Int = 0
1397 # Return a new name based on `s` and unique in the visitor
1398 fun get_name
(s
: String): String
1400 if not self.names
.has
(s
) then
1404 var i
= self.last
+ 1
1407 if not self.names
.has
(s2
) then
1416 # Return an unique and stable identifier associated with an escapemark
1417 fun escapemark_name
(e
: nullable EscapeMark): String
1420 if frame
.escapemark_names
.has_key
(e
) then return frame
.escapemark_names
[e
]
1422 if name
== null then name
= "label"
1423 name
= get_name
(name
)
1424 frame
.escapemark_names
[e
] = name
1428 # Insert a C label for associated with an escapemark
1429 fun add_escape_label
(e
: nullable EscapeMark)
1431 if e
== null then return
1432 if e
.escapes
.is_empty
then return
1433 add
("BREAK_{escapemark_name(e)}: (void)0;")
1436 # Return a "const char*" variable associated to the classname of the dynamic type of an object
1437 # NOTE: we do not return a `RuntimeVariable` "NativeString" as the class may not exist in the module/program
1438 fun class_name_string
(value
: RuntimeVariable): String is abstract
1440 # Variables handling
1442 protected var variables
= new HashMap[Variable, RuntimeVariable]
1444 # Return the local runtime_variable associated to a Nit local variable
1445 fun variable
(variable
: Variable): RuntimeVariable
1447 if self.variables
.has_key
(variable
) then
1448 return self.variables
[variable
]
1450 var name
= self.get_name
("var_{variable.name}")
1451 var mtype
= variable
.declared_type
.as(not null)
1452 mtype
= self.anchor
(mtype
)
1453 var res
= new RuntimeVariable(name
, mtype
, mtype
)
1454 self.add_decl
("{mtype.ctype} {name} /* var {variable}: {mtype} */;")
1455 self.variables
[variable
] = res
1460 # Return a new uninitialized local runtime_variable
1461 fun new_var
(mtype
: MType): RuntimeVariable
1463 mtype
= self.anchor
(mtype
)
1464 var name
= self.get_name
("var")
1465 var res
= new RuntimeVariable(name
, mtype
, mtype
)
1466 self.add_decl
("{mtype.ctype} {name} /* : {mtype} */;")
1470 # The difference with `new_var` is the C static type of the local variable
1471 fun new_var_extern
(mtype
: MType): RuntimeVariable
1473 mtype
= self.anchor
(mtype
)
1474 var name
= self.get_name
("var")
1475 var res
= new RuntimeVariable(name
, mtype
, mtype
)
1476 self.add_decl
("{mtype.ctype_extern} {name} /* : {mtype} for extern */;")
1480 # Return a new uninitialized named runtime_variable
1481 fun new_named_var
(mtype
: MType, name
: String): RuntimeVariable
1483 mtype
= self.anchor
(mtype
)
1484 var res
= new RuntimeVariable(name
, mtype
, mtype
)
1485 self.add_decl
("{mtype.ctype} {name} /* : {mtype} */;")
1489 # Correctly assign a left and a right value
1490 # Boxing and unboxing is performed if required
1491 fun assign
(left
, right
: RuntimeVariable)
1493 right
= self.autobox
(right
, left
.mtype
)
1494 self.add
("{left} = {right};")
1497 # Generate instances
1499 # Generate a alloc-instance + init-attributes
1500 fun init_instance
(mtype
: MClassType): RuntimeVariable is abstract
1502 # Allocate and init attributes of an instance of a standard or extern class
1504 # Does not support universals and the pseudo-internal `NativeArray` class.
1505 fun init_instance_or_extern
(mtype
: MClassType): RuntimeVariable
1508 var ctype
= mtype
.ctype
1509 assert mtype
.mclass
.name
!= "NativeArray"
1510 if not mtype
.is_c_primitive
then
1511 recv
= init_instance
(mtype
)
1512 else if ctype
== "char*" then
1513 recv
= new_expr
("NULL/*special!*/", mtype
)
1515 recv
= new_expr
("({ctype})0/*special!*/", mtype
)
1520 # Set a GC finalizer on `recv`, only if `recv` isa Finalizable
1521 fun set_finalizer
(recv
: RuntimeVariable)
1523 var mtype
= recv
.mtype
1524 var finalizable_type
= compiler
.mainmodule
.finalizable_type
1525 if finalizable_type
!= null and not mtype
.need_anchor
and
1526 mtype
.is_subtype
(compiler
.mainmodule
, null, finalizable_type
) then
1527 add
"gc_register_finalizer({recv});"
1531 # The currently processed module
1533 # alias for `compiler.mainmodule`
1534 fun mmodule
: MModule do return compiler
.mainmodule
1536 # Generate an integer value
1537 fun int_instance
(value
: Int): RuntimeVariable
1539 var t
= mmodule
.int_type
1540 var res
= new RuntimeVariable("{value.to_s}l", t
, t
)
1544 # Generate a byte value
1545 fun byte_instance
(value
: Byte): RuntimeVariable
1547 var t
= mmodule
.byte_type
1548 var res
= new RuntimeVariable("((unsigned char){value.to_s})", t
, t
)
1552 # Generate an int8 value
1553 fun int8_instance
(value
: Int8): RuntimeVariable
1555 var t
= mmodule
.int8_type
1556 var res
= new RuntimeVariable("((int8_t){value.to_s})", t
, t
)
1560 # Generate an int16 value
1561 fun int16_instance
(value
: Int16): RuntimeVariable
1563 var t
= mmodule
.int16_type
1564 var res
= new RuntimeVariable("((int16_t){value.to_s})", t
, t
)
1568 # Generate a uint16 value
1569 fun uint16_instance
(value
: UInt16): RuntimeVariable
1571 var t
= mmodule
.uint16_type
1572 var res
= new RuntimeVariable("((uint16_t){value.to_s})", t
, t
)
1576 # Generate an int32 value
1577 fun int32_instance
(value
: Int32): RuntimeVariable
1579 var t
= mmodule
.int32_type
1580 var res
= new RuntimeVariable("((int32_t){value.to_s})", t
, t
)
1584 # Generate a uint32 value
1585 fun uint32_instance
(value
: UInt32): RuntimeVariable
1587 var t
= mmodule
.uint32_type
1588 var res
= new RuntimeVariable("((uint32_t){value.to_s})", t
, t
)
1592 # Generate a char value
1593 fun char_instance
(value
: Char): RuntimeVariable
1595 var t
= mmodule
.char_type
1597 if value
.code_point
< 128 then
1598 return new RuntimeVariable("'{value.to_s.escape_to_c}'", t
, t
)
1600 return new RuntimeVariable("{value.code_point}", t
, t
)
1604 # Generate a float value
1606 # FIXME pass a Float, not a string
1607 fun float_instance
(value
: String): RuntimeVariable
1609 var t
= mmodule
.float_type
1610 var res
= new RuntimeVariable("{value}", t
, t
)
1614 # Generate an integer value
1615 fun bool_instance
(value
: Bool): RuntimeVariable
1617 var s
= if value
then "1" else "0"
1618 var res
= new RuntimeVariable(s
, bool_type
, bool_type
)
1622 # Generate the `null` value
1623 fun null_instance
: RuntimeVariable
1625 var t
= compiler
.mainmodule
.model
.null_type
1626 var res
= new RuntimeVariable("((val*)NULL)", t
, t
)
1630 # Generates a NativeString instance fully escaped in C-style \xHH fashion
1631 fun native_string_instance
(ns
: NativeString, len
: Int): RuntimeVariable do
1632 var mtype
= mmodule
.native_string_type
1633 var nat
= new_var
(mtype
)
1634 var byte_esc
= new Buffer.with_cap
(len
* 4)
1635 for i
in [0 .. len
[ do
1636 byte_esc
.append
("\\x{ns[i].to_s.substring_from(2)}")
1638 self.add
("{nat} = \"{byte_esc}\
";")
1642 # Generate a string value
1643 fun string_instance
(string
: String): RuntimeVariable
1645 var mtype
= mmodule
.string_type
1646 var name
= self.get_name
("varonce")
1647 self.add_decl
("static {mtype.ctype} {name};")
1648 var res
= self.new_var
(mtype
)
1649 self.add
("if (likely({name}!=NULL)) \{")
1650 self.add
("{res} = {name};")
1651 self.add
("\} else \{")
1652 var native_mtype
= mmodule
.native_string_type
1653 var nat
= self.new_var
(native_mtype
)
1654 self.add
("{nat} = \"{string.escape_to_c}\
";")
1655 var bytelen
= self.int_instance
(string
.bytelen
)
1656 var unilen
= self.int_instance
(string
.length
)
1657 self.add
("{res} = {self.send(self.get_property("to_s_full", native_mtype), [nat, bytelen, unilen]).as(not null)};")
1658 self.add
("{name} = {res};")
1663 fun value_instance
(object
: Object): RuntimeVariable
1665 if object
isa Int then
1666 return int_instance
(object
)
1667 else if object
isa Bool then
1668 return bool_instance
(object
)
1669 else if object
isa String then
1670 return string_instance
(object
)
1676 # Generate an array value
1677 fun array_instance
(array
: Array[RuntimeVariable], elttype
: MType): RuntimeVariable is abstract
1679 # Get an instance of a array for a vararg
1680 fun vararg_instance
(mpropdef
: MPropDef, recv
: RuntimeVariable, varargs
: Array[RuntimeVariable], elttype
: MType): RuntimeVariable is abstract
1684 # Add a line in the main part of the generated C
1685 fun add
(s
: String) do self.writer
.lines
.add
(s
)
1688 # (used for local or global declaration)
1689 fun add_decl
(s
: String) do self.writer
.decl_lines
.add
(s
)
1691 # Request the presence of a global declaration
1692 fun require_declaration
(key
: String)
1694 var reqs
= self.writer
.file
.required_declarations
1695 if reqs
.has
(key
) then return
1697 var node
= current_node
1698 if node
!= null then compiler
.requirers_of_declarations
[key
] = node
1701 # Add a declaration in the local-header
1702 # The declaration is ensured to be present once
1703 fun declare_once
(s
: String)
1705 self.compiler
.provide_declaration
(s
, s
)
1706 self.require_declaration
(s
)
1709 # Look for a needed .h and .c file for a given module
1710 # This is used for the legacy FFI
1711 fun add_extern
(mmodule
: MModule)
1713 var file
= mmodule
.filepath
1714 file
= file
.strip_extension
(".nit")
1715 var tryfile
= file
+ ".nit.h"
1716 if tryfile
.file_exists
then
1717 self.declare_once
("#include \"{tryfile.basename}\
"")
1718 self.compiler
.files_to_copy
.add
(tryfile
)
1720 tryfile
= file
+ "_nit.h"
1721 if tryfile
.file_exists
then
1722 self.declare_once
("#include \"{tryfile.basename}\
"")
1723 self.compiler
.files_to_copy
.add
(tryfile
)
1726 if self.compiler
.seen_extern
.has
(file
) then return
1727 self.compiler
.seen_extern
.add
(file
)
1728 tryfile
= file
+ ".nit.c"
1729 if not tryfile
.file_exists
then
1730 tryfile
= file
+ "_nit.c"
1731 if not tryfile
.file_exists
then return
1733 var f
= new ExternCFile(tryfile
.basename
, "")
1734 self.compiler
.extern_bodies
.add
(f
)
1735 self.compiler
.files_to_copy
.add
(tryfile
)
1738 # Return a new local runtime_variable initialized with the C expression `cexpr`.
1739 fun new_expr
(cexpr
: String, mtype
: MType): RuntimeVariable
1741 var res
= new_var
(mtype
)
1742 self.add
("{res} = {cexpr};")
1746 # Generate generic abort
1747 # used by aborts, asserts, casts, etc.
1748 fun add_abort
(message
: String)
1750 self.add
("if(catchStack.cursor >= 0)\{")
1751 self.add
("longjmp(catchStack.envs[catchStack.cursor], 1);")
1753 self.add
("PRINT_ERROR(\"Runtime error
: %s\
", \"{message.escape_to_c}\
");")
1759 if self.current_node
!= null and self.current_node
.location
.file
!= null and
1760 self.current_node
.location
.file
.mmodule
!= null then
1761 var f
= "FILE_{self.current_node.location.file.mmodule.c_name}"
1762 self.require_declaration
(f
)
1763 self.add
("PRINT_ERROR(\" (%s
:%d
)\\n\
", {f}, {current_node.location.line_start});")
1765 self.add
("PRINT_ERROR(\"\\n\
");")
1767 self.add
("fatal_exit(1);")
1770 # Add a dynamic cast
1771 fun add_cast
(value
: RuntimeVariable, mtype
: MType, tag
: String)
1773 var res
= self.type_test
(value
, mtype
, tag
)
1774 self.add
("if (unlikely(!{res})) \{")
1775 var cn
= self.class_name_string
(value
)
1776 self.add
("PRINT_ERROR(\"Runtime error
: Cast failed
. Expected `%s`, got
`%s`\
", \"{mtype.to_s.escape_to_c}\
", {cn});")
1781 # Generate a return with the value `s`
1782 fun ret
(s
: RuntimeVariable)
1784 self.assign
(self.frame
.returnvar
.as(not null), s
)
1785 self.add
("goto {self.frame.returnlabel.as(not null)};")
1788 # Compile a statement (if any)
1789 fun stmt
(nexpr
: nullable AExpr)
1791 if nexpr
== null then return
1792 if nexpr
.is_broken
then
1793 # Untyped expression.
1794 # Might mean dead code or invalid code
1796 add_abort
("FATAL: bad statement executed.")
1800 var narray
= nexpr
.comprehension
1801 if narray
!= null then
1802 var recv
= frame
.comprehension
.as(not null)
1803 var val
= expr
(nexpr
, narray
.element_mtype
)
1804 compile_callsite
(narray
.push_callsite
.as(not null), [recv
, val
])
1808 var old
= self.current_node
1809 self.current_node
= nexpr
1811 self.current_node
= old
1814 # Compile an expression an return its result
1815 # `mtype` is the expected return type, pass null if no specific type is expected.
1816 fun expr
(nexpr
: AExpr, mtype
: nullable MType): RuntimeVariable
1818 var old
= self.current_node
1819 self.current_node
= nexpr
1822 if nexpr
.mtype
!= null then
1823 res
= nexpr
.expr
(self)
1827 # Untyped expression.
1828 # Might mean dead code or invalid code.
1830 add_abort
("FATAL: bad expression executed.")
1831 # and return a placebo result to please the C compiler
1832 if mtype
== null then mtype
= compiler
.mainmodule
.object_type
1833 res
= new_var
(mtype
)
1835 self.current_node
= old
1839 if mtype
!= null then
1840 mtype
= self.anchor
(mtype
)
1841 res
= self.autobox
(res
, mtype
)
1843 res
= autoadapt
(res
, nexpr
.mtype
.as(not null))
1844 var implicit_cast_to
= nexpr
.implicit_cast_to
1845 if implicit_cast_to
!= null and not self.compiler
.modelbuilder
.toolcontext
.opt_no_check_autocast
.value
then
1846 add_cast
(res
, implicit_cast_to
, "auto")
1847 res
= autoadapt
(res
, implicit_cast_to
)
1849 self.current_node
= old
1853 # Alias for `self.expr(nexpr, self.bool_type)`
1854 fun expr_bool
(nexpr
: AExpr): RuntimeVariable do return expr
(nexpr
, bool_type
)
1856 # Safely show a debug message on the current node and repeat the message in the C code as a comment
1857 fun debug
(message
: String)
1859 var node
= self.current_node
1860 if node
== null then
1861 print
"?: {message}"
1865 self.add
("/* DEBUG: {message} */")
1869 # A C function associated to a Nit method
1870 # Because of customization, a given Nit method can be compiler more that once
1871 abstract class AbstractRuntimeFunction
1873 type COMPILER: AbstractCompiler
1874 type VISITOR: AbstractCompilerVisitor
1876 # The associated Nit method
1877 var mmethoddef
: MMethodDef
1879 # The mangled c name of the runtime_function
1880 # Subclasses should redefine `build_c_name` instead
1883 var res
= self.c_name_cache
1884 if res
!= null then return res
1885 res
= self.build_c_name
1886 self.c_name_cache
= res
1890 # Non cached version of `c_name`
1891 protected fun build_c_name
: String is abstract
1893 protected var c_name_cache
: nullable String = null is writable
1895 # Implements a call of the runtime_function
1896 # May inline the body or generate a C function call
1897 fun call
(v
: VISITOR, arguments
: Array[RuntimeVariable]): nullable RuntimeVariable is abstract
1899 # Generate the code for the `AbstractRuntimeFunction`
1900 # Warning: compile more than once compilation makes CC unhappy
1901 fun compile_to_c
(compiler
: COMPILER) is abstract
1904 # A runtime variable hold a runtime value in C.
1905 # Runtime variables are associated to Nit local variables and intermediate results in Nit expressions.
1907 # 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.
1908 class RuntimeVariable
1909 # The name of the variable in the C code
1912 # The static type of the variable (as declard in C)
1915 # The current casted type of the variable (as known in Nit)
1916 var mcasttype
: MType is writable
1918 # If the variable exaclty a mcasttype?
1919 # false (usual value) means that the variable is a mcasttype or a subtype.
1920 var is_exact
: Bool = false is writable
1924 assert not mtype
.need_anchor
1925 assert not mcasttype
.need_anchor
1928 redef fun to_s
do return name
1933 if self.is_exact
then
1934 exact_str
= " exact"
1939 if self.mtype
== self.mcasttype
then
1940 type_str
= "{mtype}{exact_str}"
1942 type_str
= "{mtype}({mcasttype}{exact_str})"
1944 return "<{name}:{type_str}>"
1948 # The static context of a visited property in a `AbstractCompilerVisitor`
1951 type VISITOR: AbstractCompilerVisitor
1953 # The associated visitor
1954 var visitor
: VISITOR
1956 # The executed property.
1957 # A Method in case of a call, an attribute in case of a default initialization.
1958 var mpropdef
: MPropDef
1960 # The static type of the receiver
1961 var receiver
: MClassType
1963 # Arguments of the method (the first is the receiver)
1964 var arguments
: Array[RuntimeVariable]
1966 # The runtime_variable associated to the return (in a function)
1967 var returnvar
: nullable RuntimeVariable = null is writable
1969 # The label at the end of the property
1970 var returnlabel
: nullable String = null is writable
1972 # Labels associated to a each escapemarks.
1973 # Because of inlinings, escape-marks must be associated to their context (the frame)
1974 private var escapemark_names
= new HashMap[EscapeMark, String]
1976 # The array comprehension currently filled, if any
1977 private var comprehension
: nullable RuntimeVariable = null
1981 # Return the C type associated to a given Nit static type
1982 fun ctype
: String do return "val*"
1984 # C type outside of the compiler code and in boxes
1985 fun ctype_extern
: String do return "val*"
1987 # Short name of the `ctype` to use in unions
1988 fun ctypename
: String do return "val"
1990 # Is the associated C type a primitive one?
1992 # ENSURE `result == (ctype != "val*")`
1993 fun is_c_primitive
: Bool do return false
1996 redef class MClassType
1998 redef var ctype
is lazy
do
1999 if mclass
.name
== "Int" then
2001 else if mclass
.name
== "Bool" then
2003 else if mclass
.name
== "Char" then
2005 else if mclass
.name
== "Float" then
2007 else if mclass
.name
== "Int8" then
2009 else if mclass
.name
== "Byte" then
2010 return "unsigned char"
2011 else if mclass
.name
== "Int16" then
2013 else if mclass
.name
== "UInt16" then
2015 else if mclass
.name
== "Int32" then
2017 else if mclass
.name
== "UInt32" then
2019 else if mclass
.name
== "NativeString" then
2021 else if mclass
.name
== "NativeArray" then
2028 redef var is_c_primitive
is lazy
do return ctype
!= "val*"
2030 redef fun ctype_extern
: String
2032 if mclass
.kind
== extern_kind
then
2039 redef fun ctypename
: String
2041 if mclass
.name
== "Int" then
2043 else if mclass
.name
== "Bool" then
2045 else if mclass
.name
== "Char" then
2047 else if mclass
.name
== "Float" then
2049 else if mclass
.name
== "Int8" then
2051 else if mclass
.name
== "Byte" then
2053 else if mclass
.name
== "Int16" then
2055 else if mclass
.name
== "UInt16" then
2057 else if mclass
.name
== "Int32" then
2059 else if mclass
.name
== "UInt32" then
2061 else if mclass
.name
== "NativeString" then
2063 else if mclass
.name
== "NativeArray" then
2064 #return "{self.arguments.first.ctype}*"
2072 redef class MPropDef
2073 type VISITOR: AbstractCompilerVisitor
2076 redef class MMethodDef
2077 # Can the body be inlined?
2078 fun can_inline
(v
: VISITOR): Bool
2080 if is_abstract
then return true
2081 if constant_value
!= null then return true
2082 var modelbuilder
= v
.compiler
.modelbuilder
2083 var node
= modelbuilder
.mpropdef2node
(self)
2084 if node
isa APropdef then
2085 return node
.can_inline
2086 else if node
isa AClassdef then
2087 # Automatic free init is always inlined since it is empty or contains only attribtes assigments
2089 else if node
== null then
2096 # Inline the body in another visitor
2097 fun compile_inside_to_c
(v
: VISITOR, arguments
: Array[RuntimeVariable]): nullable RuntimeVariable
2099 var modelbuilder
= v
.compiler
.modelbuilder
2100 var val
= constant_value
2101 var node
= modelbuilder
.mpropdef2node
(self)
2104 var cn
= v
.class_name_string
(arguments
.first
)
2105 v
.current_node
= node
2106 v
.add
("PRINT_ERROR(\"Runtime error
: Abstract method
`%s` called on
`%s`\
", \"{mproperty.name.escape_to_c}\
", {cn});")
2111 if node
isa APropdef then
2112 var oldnode
= v
.current_node
2113 v
.current_node
= node
2114 self.compile_parameter_check
(v
, arguments
)
2115 node
.compile_to_c
(v
, self, arguments
)
2116 v
.current_node
= oldnode
2117 else if node
isa AClassdef then
2118 var oldnode
= v
.current_node
2119 v
.current_node
= node
2120 self.compile_parameter_check
(v
, arguments
)
2121 node
.compile_to_c
(v
, self, arguments
)
2122 v
.current_node
= oldnode
2123 else if val
!= null then
2124 v
.ret
(v
.value_instance
(val
))
2131 # Generate type checks in the C code to check covariant parameters
2132 fun compile_parameter_check
(v
: VISITOR, arguments
: Array[RuntimeVariable])
2134 if v
.compiler
.modelbuilder
.toolcontext
.opt_no_check_covariance
.value
then return
2136 var msignature
= self.msignature
.as(not null)
2138 for i
in [0..msignature
.arity
[ do
2139 var mp
= msignature
.mparameters
[i
]
2140 # skip test for vararg since the array is instantiated with the correct polymorphic type
2141 if mp
.is_vararg
then continue
2143 # skip if the cast is not required
2144 var origmtype
= self.mproperty
.intro
.msignature
.mparameters
[i
].mtype
2145 if not origmtype
.need_anchor
then continue
2147 # get the parameter type
2148 var mtype
= mp
.mtype
2151 # note that v decides if and how to implements the cast
2152 v
.add
("/* Covariant cast for argument {i} ({mp.name}) {arguments[i+1].inspect} isa {mtype} */")
2153 v
.add_cast
(arguments
[i
+1], mtype
, "covariance")
2160 redef class APropdef
2161 fun compile_to_c
(v
: AbstractCompilerVisitor, mpropdef
: MMethodDef, arguments
: Array[RuntimeVariable])
2163 v
.add
("PRINT_ERROR(\"NOT YET IMPLEMENTED {class_name} {mpropdef} at
{location.to_s}\\n\
");")
2164 debug
("Not yet implemented")
2167 fun can_inline
: Bool do return true
2170 redef class AMethPropdef
2171 redef fun compile_to_c
(v
, mpropdef
, arguments
)
2173 # Call the implicit super-init
2174 var auto_super_inits
= self.auto_super_inits
2175 if auto_super_inits
!= null then
2176 var args
= [arguments
.first
]
2177 for auto_super_init
in auto_super_inits
do
2178 assert auto_super_init
.mproperty
!= mpropdef
.mproperty
2180 for i
in [0..auto_super_init
.msignature
.arity
+1[ do
2181 args
.add
(arguments
[i
])
2183 assert auto_super_init
.mproperty
!= mpropdef
.mproperty
2184 v
.compile_callsite
(auto_super_init
, args
)
2187 if auto_super_call
then
2188 v
.supercall
(mpropdef
, arguments
.first
.mtype
.as(MClassType), arguments
)
2191 # Try special compilation
2192 if mpropdef
.is_intern
then
2193 if compile_intern_to_c
(v
, mpropdef
, arguments
) then return
2195 if mpropdef
.is_extern
then
2196 if mpropdef
.mproperty
.is_init
then
2197 if compile_externinit_to_c
(v
, mpropdef
, arguments
) then return
2199 if compile_externmeth_to_c
(v
, mpropdef
, arguments
) then return
2203 # Compile block if any
2204 var n_block
= n_block
2205 if n_block
!= null then
2206 for i
in [0..mpropdef
.msignature
.arity
[ do
2207 var variable
= self.n_signature
.n_params
[i
].variable
.as(not null)
2208 v
.assign
(v
.variable
(variable
), arguments
[i
+1])
2215 var cn
= v
.class_name_string
(arguments
.first
)
2216 v
.add
("PRINT_ERROR(\"Runtime error
: uncompiled method
`%s` called on
`%s`. NOT YET IMPLEMENTED\
", \"{mpropdef.mproperty.name.escape_to_c}\
", {cn});")
2220 redef fun can_inline
2222 if self.auto_super_inits
!= null then return false
2223 var nblock
= self.n_block
2224 if nblock
== null then return true
2225 if (mpropdef
.mproperty
.name
== "==" or mpropdef
.mproperty
.name
== "!=") and mpropdef
.mclassdef
.mclass
.name
== "Object" then return true
2226 if nblock
isa ABlockExpr and nblock
.n_expr
.length
== 0 then return true
2230 fun compile_intern_to_c
(v
: AbstractCompilerVisitor, mpropdef
: MMethodDef, arguments
: Array[RuntimeVariable]): Bool
2232 var pname
= mpropdef
.mproperty
.name
2233 var cname
= mpropdef
.mclassdef
.mclass
.name
2234 var ret
= mpropdef
.msignature
.return_mtype
2236 ret
= v
.resolve_for
(ret
, arguments
.first
)
2238 if pname
!= "==" and pname
!= "!=" then
2239 v
.adapt_signature
(mpropdef
, arguments
)
2240 v
.unbox_signature_extern
(mpropdef
, arguments
)
2242 if cname
== "Int" then
2243 if pname
== "output" then
2244 v
.add
("printf(\"%ld\\n\
", {arguments.first});")
2246 else if pname
== "object_id" then
2247 v
.ret
(arguments
.first
)
2249 else if pname
== "+" then
2250 v
.ret
(v
.new_expr
("{arguments[0]} + {arguments[1]}", ret
.as(not null)))
2252 else if pname
== "-" then
2253 v
.ret
(v
.new_expr
("{arguments[0]} - {arguments[1]}", ret
.as(not null)))
2255 else if pname
== "unary -" then
2256 v
.ret
(v
.new_expr
("-{arguments[0]}", ret
.as(not null)))
2258 else if pname
== "unary +" then
2261 else if pname
== "*" then
2262 v
.ret
(v
.new_expr
("{arguments[0]} * {arguments[1]}", ret
.as(not null)))
2264 else if pname
== "/" then
2265 v
.ret
(v
.new_expr
("{arguments[0]} / {arguments[1]}", ret
.as(not null)))
2267 else if pname
== "%" then
2268 v
.ret
(v
.new_expr
("{arguments[0]} % {arguments[1]}", ret
.as(not null)))
2270 else if pname
== "==" then
2271 v
.ret
(v
.equal_test
(arguments
[0], arguments
[1]))
2273 else if pname
== "!=" then
2274 var res
= v
.equal_test
(arguments
[0], arguments
[1])
2275 v
.ret
(v
.new_expr
("!{res}", ret
.as(not null)))
2277 else if pname
== "<" then
2278 v
.ret
(v
.new_expr
("{arguments[0]} < {arguments[1]}", ret
.as(not null)))
2280 else if pname
== ">" then
2281 v
.ret
(v
.new_expr
("{arguments[0]} > {arguments[1]}", ret
.as(not null)))
2283 else if pname
== "<=" then
2284 v
.ret
(v
.new_expr
("{arguments[0]} <= {arguments[1]}", ret
.as(not null)))
2286 else if pname
== ">=" then
2287 v
.ret
(v
.new_expr
("{arguments[0]} >= {arguments[1]}", ret
.as(not null)))
2289 else if pname
== "to_i8" then
2290 v
.ret
(v
.new_expr
("(int8_t){arguments[0]}", ret
.as(not null)))
2292 else if pname
== "to_i16" then
2293 v
.ret
(v
.new_expr
("(int16_t){arguments[0]}", ret
.as(not null)))
2295 else if pname
== "to_u16" then
2296 v
.ret
(v
.new_expr
("(uint16_t){arguments[0]}", ret
.as(not null)))
2298 else if pname
== "to_i32" then
2299 v
.ret
(v
.new_expr
("(int32_t){arguments[0]}", ret
.as(not null)))
2301 else if pname
== "to_u32" then
2302 v
.ret
(v
.new_expr
("(uint32_t){arguments[0]}", ret
.as(not null)))
2304 else if pname
== "to_f" then
2305 v
.ret
(v
.new_expr
("(double){arguments[0]}", ret
.as(not null)))
2307 else if pname
== "to_b" then
2308 v
.ret
(v
.new_expr
("(unsigned char){arguments[0]}", ret
.as(not null)))
2310 else if pname
== "code_point" then
2311 v
.ret
(v
.new_expr
("(uint32_t){arguments[0]}", ret
.as(not null)))
2313 else if pname
== "&" then
2314 v
.ret
(v
.new_expr
("{arguments[0]} & {arguments[1]}", ret
.as(not null)))
2316 else if pname
== "|" then
2317 v
.ret
(v
.new_expr
("{arguments[0]} | {arguments[1]}", ret
.as(not null)))
2319 else if pname
== ">>" then
2320 v
.ret
(v
.new_expr
("{arguments[0]} >> {arguments[1]}", ret
.as(not null)))
2322 else if pname
== "<<" then
2323 v
.ret
(v
.new_expr
("{arguments[0]} << {arguments[1]}", ret
.as(not null)))
2326 else if cname
== "Char" then
2327 if pname
== "object_id" then
2328 v
.ret
(v
.new_expr
("(long){arguments.first}", ret
.as(not null)))
2330 else if pname
== "successor" then
2331 v
.ret
(v
.new_expr
("{arguments[0]} + {arguments[1]}", ret
.as(not null)))
2333 else if pname
== "predecessor" then
2334 v
.ret
(v
.new_expr
("{arguments[0]} - {arguments[1]}", ret
.as(not null)))
2336 else if pname
== "==" then
2337 v
.ret
(v
.equal_test
(arguments
[0], arguments
[1]))
2339 else if pname
== "!=" then
2340 var res
= v
.equal_test
(arguments
[0], arguments
[1])
2341 v
.ret
(v
.new_expr
("!{res}", ret
.as(not null)))
2343 else if pname
== "<" then
2344 v
.ret
(v
.new_expr
("{arguments[0]} < {arguments[1]}", ret
.as(not null)))
2346 else if pname
== ">" then
2347 v
.ret
(v
.new_expr
("{arguments[0]} > {arguments[1]}", ret
.as(not null)))
2349 else if pname
== "<=" then
2350 v
.ret
(v
.new_expr
("{arguments[0]} <= {arguments[1]}", ret
.as(not null)))
2352 else if pname
== ">=" then
2353 v
.ret
(v
.new_expr
("{arguments[0]} >= {arguments[1]}", ret
.as(not null)))
2355 else if pname
== "to_i" then
2356 v
.ret
(v
.new_expr
("{arguments[0]}-'0'", ret
.as(not null)))
2358 else if pname
== "code_point" then
2359 v
.ret
(v
.new_expr
("(long){arguments[0]}", ret
.as(not null)))
2362 else if cname
== "Byte" then
2363 if pname
== "output" then
2364 v
.add
("printf(\"%x\\n\
", {arguments.first});")
2366 else if pname
== "object_id" then
2367 v
.ret
(v
.new_expr
("(long){arguments.first}", ret
.as(not null)))
2369 else if pname
== "+" then
2370 v
.ret
(v
.new_expr
("{arguments[0]} + {arguments[1]}", ret
.as(not null)))
2372 else if pname
== "-" then
2373 v
.ret
(v
.new_expr
("{arguments[0]} - {arguments[1]}", ret
.as(not null)))
2375 else if pname
== "unary -" then
2376 v
.ret
(v
.new_expr
("-{arguments[0]}", ret
.as(not null)))
2378 else if pname
== "unary +" then
2381 else if pname
== "*" then
2382 v
.ret
(v
.new_expr
("{arguments[0]} * {arguments[1]}", ret
.as(not null)))
2384 else if pname
== "/" then
2385 v
.ret
(v
.new_expr
("{arguments[0]} / {arguments[1]}", ret
.as(not null)))
2387 else if pname
== "%" then
2388 v
.ret
(v
.new_expr
("{arguments[0]} % {arguments[1]}", ret
.as(not null)))
2390 else if pname
== "==" then
2391 v
.ret
(v
.equal_test
(arguments
[0], arguments
[1]))
2393 else if pname
== "!=" then
2394 var res
= v
.equal_test
(arguments
[0], arguments
[1])
2395 v
.ret
(v
.new_expr
("!{res}", ret
.as(not null)))
2397 else if pname
== "<" then
2398 v
.ret
(v
.new_expr
("{arguments[0]} < {arguments[1]}", ret
.as(not null)))
2400 else if pname
== ">" then
2401 v
.ret
(v
.new_expr
("{arguments[0]} > {arguments[1]}", ret
.as(not null)))
2403 else if pname
== "<=" then
2404 v
.ret
(v
.new_expr
("{arguments[0]} <= {arguments[1]}", ret
.as(not null)))
2406 else if pname
== ">=" then
2407 v
.ret
(v
.new_expr
("{arguments[0]} >= {arguments[1]}", ret
.as(not null)))
2409 else if pname
== ">>" then
2410 v
.ret
(v
.new_expr
("{arguments[0]} >> {arguments[1]}", ret
.as(not null)))
2412 else if pname
== "<<" then
2413 v
.ret
(v
.new_expr
("{arguments[0]} << {arguments[1]}", ret
.as(not null)))
2415 else if pname
== "&" then
2416 v
.ret
(v
.new_expr
("{arguments[0]} & {arguments[1]}", ret
.as(not null)))
2418 else if pname
== "to_i" then
2419 v
.ret
(v
.new_expr
("(long){arguments[0]}", ret
.as(not null)))
2421 else if pname
== "to_f" then
2422 v
.ret
(v
.new_expr
("(double){arguments[0]}", ret
.as(not null)))
2424 else if pname
== "to_i8" then
2425 v
.ret
(v
.new_expr
("(int8_t){arguments[0]}", ret
.as(not null)))
2427 else if pname
== "to_i16" then
2428 v
.ret
(v
.new_expr
("(int16_t){arguments[0]}", ret
.as(not null)))
2430 else if pname
== "to_u16" then
2431 v
.ret
(v
.new_expr
("(uint16_t){arguments[0]}", ret
.as(not null)))
2433 else if pname
== "to_i32" then
2434 v
.ret
(v
.new_expr
("(int32_t){arguments[0]}", ret
.as(not null)))
2436 else if pname
== "to_u32" then
2437 v
.ret
(v
.new_expr
("(uint32_t){arguments[0]}", ret
.as(not null)))
2439 else if pname
== "ascii" then
2440 v
.ret
(v
.new_expr
("(uint32_t){arguments[0]}", ret
.as(not null)))
2443 else if cname
== "Bool" then
2444 if pname
== "output" then
2445 v
.add
("printf({arguments.first}?\"true\\n\
":\"false\\n\
");")
2447 else if pname
== "object_id" then
2448 v
.ret
(v
.new_expr
("(long){arguments.first}", ret
.as(not null)))
2450 else if pname
== "==" then
2451 v
.ret
(v
.equal_test
(arguments
[0], arguments
[1]))
2453 else if pname
== "!=" then
2454 var res
= v
.equal_test
(arguments
[0], arguments
[1])
2455 v
.ret
(v
.new_expr
("!{res}", ret
.as(not null)))
2458 else if cname
== "Float" then
2459 if pname
== "output" then
2460 v
.add
("printf(\"%f\\n\
", {arguments.first});")
2462 else if pname
== "object_id" then
2463 v
.ret
(v
.new_expr
("(double){arguments.first}", ret
.as(not null)))
2465 else if pname
== "+" then
2466 v
.ret
(v
.new_expr
("{arguments[0]} + {arguments[1]}", ret
.as(not null)))
2468 else if pname
== "-" then
2469 v
.ret
(v
.new_expr
("{arguments[0]} - {arguments[1]}", ret
.as(not null)))
2471 else if pname
== "unary -" then
2472 v
.ret
(v
.new_expr
("-{arguments[0]}", ret
.as(not null)))
2474 else if pname
== "unary +" then
2477 else if pname
== "succ" then
2478 v
.ret
(v
.new_expr
("{arguments[0]}+1", ret
.as(not null)))
2480 else if pname
== "prec" then
2481 v
.ret
(v
.new_expr
("{arguments[0]}-1", ret
.as(not null)))
2483 else if pname
== "*" then
2484 v
.ret
(v
.new_expr
("{arguments[0]} * {arguments[1]}", ret
.as(not null)))
2486 else if pname
== "/" then
2487 v
.ret
(v
.new_expr
("{arguments[0]} / {arguments[1]}", ret
.as(not null)))
2489 else if pname
== "==" then
2490 v
.ret
(v
.equal_test
(arguments
[0], arguments
[1]))
2492 else if pname
== "!=" then
2493 var res
= v
.equal_test
(arguments
[0], arguments
[1])
2494 v
.ret
(v
.new_expr
("!{res}", ret
.as(not null)))
2496 else if pname
== "<" then
2497 v
.ret
(v
.new_expr
("{arguments[0]} < {arguments[1]}", ret
.as(not null)))
2499 else if pname
== ">" then
2500 v
.ret
(v
.new_expr
("{arguments[0]} > {arguments[1]}", ret
.as(not null)))
2502 else if pname
== "<=" then
2503 v
.ret
(v
.new_expr
("{arguments[0]} <= {arguments[1]}", ret
.as(not null)))
2505 else if pname
== ">=" then
2506 v
.ret
(v
.new_expr
("{arguments[0]} >= {arguments[1]}", ret
.as(not null)))
2508 else if pname
== "to_i" then
2509 v
.ret
(v
.new_expr
("(long){arguments[0]}", ret
.as(not null)))
2511 else if pname
== "to_b" then
2512 v
.ret
(v
.new_expr
("(unsigned char){arguments[0]}", ret
.as(not null)))
2514 else if pname
== "to_i8" then
2515 v
.ret
(v
.new_expr
("(int8_t){arguments[0]}", ret
.as(not null)))
2517 else if pname
== "to_i16" then
2518 v
.ret
(v
.new_expr
("(int16_t){arguments[0]}", ret
.as(not null)))
2520 else if pname
== "to_u16" then
2521 v
.ret
(v
.new_expr
("(uint16_t){arguments[0]}", ret
.as(not null)))
2523 else if pname
== "to_i32" then
2524 v
.ret
(v
.new_expr
("(int32_t){arguments[0]}", ret
.as(not null)))
2526 else if pname
== "to_u32" then
2527 v
.ret
(v
.new_expr
("(uint32_t){arguments[0]}", ret
.as(not null)))
2530 else if cname
== "NativeString" then
2531 if pname
== "[]" then
2532 v
.ret
(v
.new_expr
("(unsigned char)((int){arguments[0]}[{arguments[1]}])", ret
.as(not null)))
2534 else if pname
== "[]=" then
2535 v
.add
("{arguments[0]}[{arguments[1]}]=(unsigned char){arguments[2]};")
2537 else if pname
== "copy_to" then
2538 v
.add
("memmove({arguments[1]}+{arguments[4]},{arguments[0]}+{arguments[3]},{arguments[2]});")
2540 else if pname
== "atoi" then
2541 v
.ret
(v
.new_expr
("atoi({arguments[0]});", ret
.as(not null)))
2543 else if pname
== "fast_cstring" then
2544 v
.ret
(v
.new_expr
("{arguments[0]} + {arguments[1]}", ret
.as(not null)))
2546 else if pname
== "==" then
2547 v
.ret
(v
.equal_test
(arguments
[0], arguments
[1]))
2549 else if pname
== "!=" then
2550 var res
= v
.equal_test
(arguments
[0], arguments
[1])
2551 v
.ret
(v
.new_expr
("!{res}", ret
.as(not null)))
2553 else if pname
== "new" then
2554 v
.ret
(v
.new_expr
("(char*)nit_alloc({arguments[1]})", ret
.as(not null)))
2556 else if pname
== "fetch_4_chars" then
2557 v
.ret
(v
.new_expr
("(long)*((uint32_t*)({arguments[0]} + {arguments[1]}))", ret
.as(not null)))
2559 else if pname
== "fetch_4_hchars" then
2560 v
.ret
(v
.new_expr
("(long)be32toh(*((uint32_t*)({arguments[0]} + {arguments[1]})))", ret
.as(not null)))
2563 else if cname
== "NativeArray" then
2564 return v
.native_array_def
(pname
, ret
, arguments
)
2565 else if cname
== "Int8" then
2566 if pname
== "output" then
2567 v
.add
("printf(\"%\
"PRIi8 \"\\n\
", {arguments.first});")
2569 else if pname
== "object_id" then
2570 v
.ret
(v
.new_expr
("(long){arguments.first}", ret
.as(not null)))
2572 else if pname
== "+" then
2573 v
.ret
(v
.new_expr
("{arguments[0]} + {arguments[1]}", ret
.as(not null)))
2575 else if pname
== "-" then
2576 v
.ret
(v
.new_expr
("{arguments[0]} - {arguments[1]}", ret
.as(not null)))
2578 else if pname
== "unary -" then
2579 v
.ret
(v
.new_expr
("-{arguments[0]}", ret
.as(not null)))
2581 else if pname
== "unary +" then
2584 else if pname
== "*" then
2585 v
.ret
(v
.new_expr
("{arguments[0]} * {arguments[1]}", ret
.as(not null)))
2587 else if pname
== "/" then
2588 v
.ret
(v
.new_expr
("{arguments[0]} / {arguments[1]}", ret
.as(not null)))
2590 else if pname
== "%" then
2591 v
.ret
(v
.new_expr
("{arguments[0]} % {arguments[1]}", ret
.as(not null)))
2593 else if pname
== "<<" then
2594 v
.ret
(v
.new_expr
("{arguments[0]} << {arguments[1]}", ret
.as(not null)))
2596 else if pname
== ">>" then
2597 v
.ret
(v
.new_expr
("{arguments[0]} >> {arguments[1]}", ret
.as(not null)))
2599 else if pname
== "==" then
2600 v
.ret
(v
.equal_test
(arguments
[0], arguments
[1]))
2602 else if pname
== "!=" then
2603 var res
= v
.equal_test
(arguments
[0], arguments
[1])
2604 v
.ret
(v
.new_expr
("!{res}", ret
.as(not null)))
2606 else if pname
== "<" then
2607 v
.ret
(v
.new_expr
("{arguments[0]} < {arguments[1]}", ret
.as(not null)))
2609 else if pname
== ">" then
2610 v
.ret
(v
.new_expr
("{arguments[0]} > {arguments[1]}", ret
.as(not null)))
2612 else if pname
== "<=" then
2613 v
.ret
(v
.new_expr
("{arguments[0]} <= {arguments[1]}", ret
.as(not null)))
2615 else if pname
== ">=" then
2616 v
.ret
(v
.new_expr
("{arguments[0]} >= {arguments[1]}", ret
.as(not null)))
2618 else if pname
== "to_i" then
2619 v
.ret
(v
.new_expr
("(long){arguments[0]}", ret
.as(not null)))
2621 else if pname
== "to_b" then
2622 v
.ret
(v
.new_expr
("(unsigned char){arguments[0]}", ret
.as(not null)))
2624 else if pname
== "to_i16" then
2625 v
.ret
(v
.new_expr
("(int16_t){arguments[0]}", ret
.as(not null)))
2627 else if pname
== "to_u16" then
2628 v
.ret
(v
.new_expr
("(uint16_t){arguments[0]}", ret
.as(not null)))
2630 else if pname
== "to_i32" then
2631 v
.ret
(v
.new_expr
("(int32_t){arguments[0]}", ret
.as(not null)))
2633 else if pname
== "to_u32" then
2634 v
.ret
(v
.new_expr
("(uint32_t){arguments[0]}", ret
.as(not null)))
2636 else if pname
== "to_f" then
2637 v
.ret
(v
.new_expr
("(double){arguments[0]}", ret
.as(not null)))
2639 else if pname
== "&" then
2640 v
.ret
(v
.new_expr
("{arguments[0]} & {arguments[1]}", ret
.as(not null)))
2642 else if pname
== "|" then
2643 v
.ret
(v
.new_expr
("{arguments[0]} | {arguments[1]}", ret
.as(not null)))
2645 else if pname
== "^" then
2646 v
.ret
(v
.new_expr
("{arguments[0]} ^ {arguments[1]}", ret
.as(not null)))
2648 else if pname
== "unary ~" then
2649 v
.ret
(v
.new_expr
("~{arguments[0]}", ret
.as(not null)))
2652 else if cname
== "Int16" then
2653 if pname
== "output" then
2654 v
.add
("printf(\"%\
"PRIi16 \"\\n\
", {arguments.first});")
2656 else if pname
== "object_id" then
2657 v
.ret
(v
.new_expr
("(long){arguments.first}", ret
.as(not null)))
2659 else if pname
== "+" then
2660 v
.ret
(v
.new_expr
("{arguments[0]} + {arguments[1]}", ret
.as(not null)))
2662 else if pname
== "-" then
2663 v
.ret
(v
.new_expr
("{arguments[0]} - {arguments[1]}", ret
.as(not null)))
2665 else if pname
== "unary -" then
2666 v
.ret
(v
.new_expr
("-{arguments[0]}", ret
.as(not null)))
2668 else if pname
== "unary +" then
2671 else if pname
== "*" then
2672 v
.ret
(v
.new_expr
("{arguments[0]} * {arguments[1]}", ret
.as(not null)))
2674 else if pname
== "/" then
2675 v
.ret
(v
.new_expr
("{arguments[0]} / {arguments[1]}", ret
.as(not null)))
2677 else if pname
== "%" then
2678 v
.ret
(v
.new_expr
("{arguments[0]} % {arguments[1]}", ret
.as(not null)))
2680 else if pname
== "<<" then
2681 v
.ret
(v
.new_expr
("{arguments[0]} << {arguments[1]}", ret
.as(not null)))
2683 else if pname
== ">>" then
2684 v
.ret
(v
.new_expr
("{arguments[0]} >> {arguments[1]}", ret
.as(not null)))
2686 else if pname
== "==" then
2687 v
.ret
(v
.equal_test
(arguments
[0], arguments
[1]))
2689 else if pname
== "!=" then
2690 var res
= v
.equal_test
(arguments
[0], arguments
[1])
2691 v
.ret
(v
.new_expr
("!{res}", ret
.as(not null)))
2693 else if pname
== "<" then
2694 v
.ret
(v
.new_expr
("{arguments[0]} < {arguments[1]}", ret
.as(not null)))
2696 else if pname
== ">" then
2697 v
.ret
(v
.new_expr
("{arguments[0]} > {arguments[1]}", ret
.as(not null)))
2699 else if pname
== "<=" then
2700 v
.ret
(v
.new_expr
("{arguments[0]} <= {arguments[1]}", ret
.as(not null)))
2702 else if pname
== ">=" then
2703 v
.ret
(v
.new_expr
("{arguments[0]} >= {arguments[1]}", ret
.as(not null)))
2705 else if pname
== "to_i" then
2706 v
.ret
(v
.new_expr
("(long){arguments[0]}", ret
.as(not null)))
2708 else if pname
== "to_b" then
2709 v
.ret
(v
.new_expr
("(unsigned char){arguments[0]}", ret
.as(not null)))
2711 else if pname
== "to_i8" then
2712 v
.ret
(v
.new_expr
("(int8_t){arguments[0]}", ret
.as(not null)))
2714 else if pname
== "to_u16" then
2715 v
.ret
(v
.new_expr
("(uint16_t){arguments[0]}", ret
.as(not null)))
2717 else if pname
== "to_i32" then
2718 v
.ret
(v
.new_expr
("(int32_t){arguments[0]}", ret
.as(not null)))
2720 else if pname
== "to_u32" then
2721 v
.ret
(v
.new_expr
("(uint32_t){arguments[0]}", ret
.as(not null)))
2723 else if pname
== "to_f" then
2724 v
.ret
(v
.new_expr
("(double){arguments[0]}", ret
.as(not null)))
2726 else if pname
== "&" then
2727 v
.ret
(v
.new_expr
("{arguments[0]} & {arguments[1]}", ret
.as(not null)))
2729 else if pname
== "|" then
2730 v
.ret
(v
.new_expr
("{arguments[0]} | {arguments[1]}", ret
.as(not null)))
2732 else if pname
== "^" then
2733 v
.ret
(v
.new_expr
("{arguments[0]} ^ {arguments[1]}", ret
.as(not null)))
2735 else if pname
== "unary ~" then
2736 v
.ret
(v
.new_expr
("~{arguments[0]}", ret
.as(not null)))
2739 else if cname
== "UInt16" then
2740 if pname
== "output" then
2741 v
.add
("printf(\"%\
"PRIu16 \"\\n\
", {arguments.first});")
2743 else if pname
== "object_id" then
2744 v
.ret
(v
.new_expr
("(long){arguments.first}", ret
.as(not null)))
2746 else if pname
== "+" then
2747 v
.ret
(v
.new_expr
("{arguments[0]} + {arguments[1]}", ret
.as(not null)))
2749 else if pname
== "-" then
2750 v
.ret
(v
.new_expr
("{arguments[0]} - {arguments[1]}", ret
.as(not null)))
2752 else if pname
== "unary -" then
2753 v
.ret
(v
.new_expr
("-{arguments[0]}", ret
.as(not null)))
2755 else if pname
== "unary +" then
2758 else if pname
== "*" then
2759 v
.ret
(v
.new_expr
("{arguments[0]} * {arguments[1]}", ret
.as(not null)))
2761 else if pname
== "/" then
2762 v
.ret
(v
.new_expr
("{arguments[0]} / {arguments[1]}", ret
.as(not null)))
2764 else if pname
== "%" then
2765 v
.ret
(v
.new_expr
("{arguments[0]} % {arguments[1]}", ret
.as(not null)))
2767 else if pname
== "<<" then
2768 v
.ret
(v
.new_expr
("{arguments[0]} << {arguments[1]}", ret
.as(not null)))
2770 else if pname
== ">>" then
2771 v
.ret
(v
.new_expr
("{arguments[0]} >> {arguments[1]}", ret
.as(not null)))
2773 else if pname
== "==" then
2774 v
.ret
(v
.equal_test
(arguments
[0], arguments
[1]))
2776 else if pname
== "!=" then
2777 var res
= v
.equal_test
(arguments
[0], arguments
[1])
2778 v
.ret
(v
.new_expr
("!{res}", ret
.as(not null)))
2780 else if pname
== "<" then
2781 v
.ret
(v
.new_expr
("{arguments[0]} < {arguments[1]}", ret
.as(not null)))
2783 else if pname
== ">" then
2784 v
.ret
(v
.new_expr
("{arguments[0]} > {arguments[1]}", ret
.as(not null)))
2786 else if pname
== "<=" then
2787 v
.ret
(v
.new_expr
("{arguments[0]} <= {arguments[1]}", ret
.as(not null)))
2789 else if pname
== ">=" then
2790 v
.ret
(v
.new_expr
("{arguments[0]} >= {arguments[1]}", ret
.as(not null)))
2792 else if pname
== "to_i" then
2793 v
.ret
(v
.new_expr
("(long){arguments[0]}", ret
.as(not null)))
2795 else if pname
== "to_b" then
2796 v
.ret
(v
.new_expr
("(unsigned char){arguments[0]}", ret
.as(not null)))
2798 else if pname
== "to_i8" then
2799 v
.ret
(v
.new_expr
("(int8_t){arguments[0]}", ret
.as(not null)))
2801 else if pname
== "to_i16" then
2802 v
.ret
(v
.new_expr
("(int16_t){arguments[0]}", ret
.as(not null)))
2804 else if pname
== "to_i32" then
2805 v
.ret
(v
.new_expr
("(int32_t){arguments[0]}", ret
.as(not null)))
2807 else if pname
== "to_u32" then
2808 v
.ret
(v
.new_expr
("(uint32_t){arguments[0]}", ret
.as(not null)))
2810 else if pname
== "to_f" then
2811 v
.ret
(v
.new_expr
("(double){arguments[0]}", ret
.as(not null)))
2813 else if pname
== "&" then
2814 v
.ret
(v
.new_expr
("{arguments[0]} & {arguments[1]}", ret
.as(not null)))
2816 else if pname
== "|" then
2817 v
.ret
(v
.new_expr
("{arguments[0]} | {arguments[1]}", ret
.as(not null)))
2819 else if pname
== "^" then
2820 v
.ret
(v
.new_expr
("{arguments[0]} ^ {arguments[1]}", ret
.as(not null)))
2822 else if pname
== "unary ~" then
2823 v
.ret
(v
.new_expr
("~{arguments[0]}", ret
.as(not null)))
2826 else if cname
== "Int32" then
2827 if pname
== "output" then
2828 v
.add
("printf(\"%\
"PRIi32 \"\\n\
", {arguments.first});")
2830 else if pname
== "object_id" then
2831 v
.ret
(v
.new_expr
("(long){arguments.first}", ret
.as(not null)))
2833 else if pname
== "+" then
2834 v
.ret
(v
.new_expr
("{arguments[0]} + {arguments[1]}", ret
.as(not null)))
2836 else if pname
== "-" then
2837 v
.ret
(v
.new_expr
("{arguments[0]} - {arguments[1]}", ret
.as(not null)))
2839 else if pname
== "unary -" then
2840 v
.ret
(v
.new_expr
("-{arguments[0]}", ret
.as(not null)))
2842 else if pname
== "unary +" then
2845 else if pname
== "*" then
2846 v
.ret
(v
.new_expr
("{arguments[0]} * {arguments[1]}", ret
.as(not null)))
2848 else if pname
== "/" then
2849 v
.ret
(v
.new_expr
("{arguments[0]} / {arguments[1]}", ret
.as(not null)))
2851 else if pname
== "%" then
2852 v
.ret
(v
.new_expr
("{arguments[0]} % {arguments[1]}", ret
.as(not null)))
2854 else if pname
== "<<" then
2855 v
.ret
(v
.new_expr
("{arguments[0]} << {arguments[1]}", ret
.as(not null)))
2857 else if pname
== ">>" then
2858 v
.ret
(v
.new_expr
("{arguments[0]} >> {arguments[1]}", ret
.as(not null)))
2860 else if pname
== "==" then
2861 v
.ret
(v
.equal_test
(arguments
[0], arguments
[1]))
2863 else if pname
== "!=" then
2864 var res
= v
.equal_test
(arguments
[0], arguments
[1])
2865 v
.ret
(v
.new_expr
("!{res}", ret
.as(not null)))
2867 else if pname
== "<" then
2868 v
.ret
(v
.new_expr
("{arguments[0]} < {arguments[1]}", ret
.as(not null)))
2870 else if pname
== ">" then
2871 v
.ret
(v
.new_expr
("{arguments[0]} > {arguments[1]}", ret
.as(not null)))
2873 else if pname
== "<=" then
2874 v
.ret
(v
.new_expr
("{arguments[0]} <= {arguments[1]}", ret
.as(not null)))
2876 else if pname
== ">=" then
2877 v
.ret
(v
.new_expr
("{arguments[0]} >= {arguments[1]}", ret
.as(not null)))
2879 else if pname
== "to_i" then
2880 v
.ret
(v
.new_expr
("(long){arguments[0]}", ret
.as(not null)))
2882 else if pname
== "to_b" then
2883 v
.ret
(v
.new_expr
("(unsigned char){arguments[0]}", ret
.as(not null)))
2885 else if pname
== "to_i8" then
2886 v
.ret
(v
.new_expr
("(int8_t){arguments[0]}", ret
.as(not null)))
2888 else if pname
== "to_i16" then
2889 v
.ret
(v
.new_expr
("(int16_t){arguments[0]}", ret
.as(not null)))
2891 else if pname
== "to_u16" then
2892 v
.ret
(v
.new_expr
("(uint16_t){arguments[0]}", ret
.as(not null)))
2894 else if pname
== "to_u32" then
2895 v
.ret
(v
.new_expr
("(uint32_t){arguments[0]}", ret
.as(not null)))
2897 else if pname
== "to_f" then
2898 v
.ret
(v
.new_expr
("(double){arguments[0]}", ret
.as(not null)))
2900 else if pname
== "&" then
2901 v
.ret
(v
.new_expr
("{arguments[0]} & {arguments[1]}", ret
.as(not null)))
2903 else if pname
== "|" then
2904 v
.ret
(v
.new_expr
("{arguments[0]} | {arguments[1]}", ret
.as(not null)))
2906 else if pname
== "^" then
2907 v
.ret
(v
.new_expr
("{arguments[0]} ^ {arguments[1]}", ret
.as(not null)))
2909 else if pname
== "unary ~" then
2910 v
.ret
(v
.new_expr
("~{arguments[0]}", ret
.as(not null)))
2913 else if cname
== "UInt32" then
2914 if pname
== "output" then
2915 v
.add
("printf(\"%\
"PRIu32 \"\\n\
", {arguments.first});")
2917 else if pname
== "object_id" then
2918 v
.ret
(v
.new_expr
("(long){arguments.first}", ret
.as(not null)))
2920 else if pname
== "+" then
2921 v
.ret
(v
.new_expr
("{arguments[0]} + {arguments[1]}", ret
.as(not null)))
2923 else if pname
== "-" then
2924 v
.ret
(v
.new_expr
("{arguments[0]} - {arguments[1]}", ret
.as(not null)))
2926 else if pname
== "unary -" then
2927 v
.ret
(v
.new_expr
("-{arguments[0]}", ret
.as(not null)))
2929 else if pname
== "unary +" then
2932 else if pname
== "*" then
2933 v
.ret
(v
.new_expr
("{arguments[0]} * {arguments[1]}", ret
.as(not null)))
2935 else if pname
== "/" then
2936 v
.ret
(v
.new_expr
("{arguments[0]} / {arguments[1]}", ret
.as(not null)))
2938 else if pname
== "%" then
2939 v
.ret
(v
.new_expr
("{arguments[0]} % {arguments[1]}", ret
.as(not null)))
2941 else if pname
== "<<" then
2942 v
.ret
(v
.new_expr
("{arguments[0]} << {arguments[1]}", ret
.as(not null)))
2944 else if pname
== ">>" then
2945 v
.ret
(v
.new_expr
("{arguments[0]} >> {arguments[1]}", ret
.as(not null)))
2947 else if pname
== "==" then
2948 v
.ret
(v
.equal_test
(arguments
[0], arguments
[1]))
2950 else if pname
== "!=" then
2951 var res
= v
.equal_test
(arguments
[0], arguments
[1])
2952 v
.ret
(v
.new_expr
("!{res}", ret
.as(not null)))
2954 else if pname
== "<" then
2955 v
.ret
(v
.new_expr
("{arguments[0]} < {arguments[1]}", ret
.as(not null)))
2957 else if pname
== ">" then
2958 v
.ret
(v
.new_expr
("{arguments[0]} > {arguments[1]}", ret
.as(not null)))
2960 else if pname
== "<=" then
2961 v
.ret
(v
.new_expr
("{arguments[0]} <= {arguments[1]}", ret
.as(not null)))
2963 else if pname
== ">=" then
2964 v
.ret
(v
.new_expr
("{arguments[0]} >= {arguments[1]}", ret
.as(not null)))
2966 else if pname
== "to_i" then
2967 v
.ret
(v
.new_expr
("(long){arguments[0]}", ret
.as(not null)))
2969 else if pname
== "to_b" then
2970 v
.ret
(v
.new_expr
("(unsigned char){arguments[0]}", ret
.as(not null)))
2972 else if pname
== "to_i8" then
2973 v
.ret
(v
.new_expr
("(int8_t){arguments[0]}", ret
.as(not null)))
2975 else if pname
== "to_i16" then
2976 v
.ret
(v
.new_expr
("(int16_t){arguments[0]}", ret
.as(not null)))
2978 else if pname
== "to_u16" then
2979 v
.ret
(v
.new_expr
("(uint16_t){arguments[0]}", ret
.as(not null)))
2981 else if pname
== "to_i32" then
2982 v
.ret
(v
.new_expr
("(int32_t){arguments[0]}", ret
.as(not null)))
2984 else if pname
== "to_f" then
2985 v
.ret
(v
.new_expr
("(double){arguments[0]}", ret
.as(not null)))
2987 else if pname
== "&" then
2988 v
.ret
(v
.new_expr
("{arguments[0]} & {arguments[1]}", ret
.as(not null)))
2990 else if pname
== "|" then
2991 v
.ret
(v
.new_expr
("{arguments[0]} | {arguments[1]}", ret
.as(not null)))
2993 else if pname
== "^" then
2994 v
.ret
(v
.new_expr
("{arguments[0]} ^ {arguments[1]}", ret
.as(not null)))
2996 else if pname
== "unary ~" then
2997 v
.ret
(v
.new_expr
("~{arguments[0]}", ret
.as(not null)))
3001 if pname
== "exit" then
3002 v
.add
("exit({arguments[1]});")
3004 else if pname
== "sys" then
3005 v
.ret
(v
.new_expr
("glob_sys", ret
.as(not null)))
3007 else if pname
== "calloc_string" then
3008 v
.ret
(v
.new_expr
("(char*)nit_alloc({arguments[1]})", ret
.as(not null)))
3010 else if pname
== "calloc_array" then
3011 v
.calloc_array
(ret
.as(not null), arguments
)
3013 else if pname
== "object_id" then
3014 v
.ret
(v
.new_expr
("(long){arguments.first}", ret
.as(not null)))
3016 else if pname
== "is_same_type" then
3017 v
.ret
(v
.is_same_type_test
(arguments
[0], arguments
[1]))
3019 else if pname
== "is_same_instance" then
3020 v
.ret
(v
.equal_test
(arguments
[0], arguments
[1]))
3022 else if pname
== "output_class_name" then
3023 var nat
= v
.class_name_string
(arguments
.first
)
3024 v
.add
("printf(\"%s\\n\
", {nat});")
3026 else if pname
== "native_class_name" then
3027 var nat
= v
.class_name_string
(arguments
.first
)
3028 v
.ret
(v
.new_expr
("(char*){nat}", ret
.as(not null)))
3030 else if pname
== "force_garbage_collection" then
3031 v
.add
("nit_gcollect();")
3033 else if pname
== "native_argc" then
3034 v
.ret
(v
.new_expr
("glob_argc", ret
.as(not null)))
3036 else if pname
== "native_argv" then
3037 v
.ret
(v
.new_expr
("glob_argv[{arguments[1]}]", ret
.as(not null)))
3043 # Compile an extern method
3044 # Return `true` if the compilation was successful, `false` if a fall-back is needed
3045 fun compile_externmeth_to_c
(v
: AbstractCompilerVisitor, mpropdef
: MMethodDef, arguments
: Array[RuntimeVariable]): Bool
3048 var at
= self.get_single_annotation
("extern", v
.compiler
.modelbuilder
)
3049 if at
!= null and at
.n_args
.length
== 1 then
3050 externname
= at
.arg_as_string
(v
.compiler
.modelbuilder
)
3051 if externname
== null then return false
3055 v
.add_extern
(mpropdef
.mclassdef
.mmodule
)
3056 var res
: nullable RuntimeVariable = null
3057 var ret
= mpropdef
.msignature
.return_mtype
3059 ret
= v
.resolve_for
(ret
, arguments
.first
)
3060 res
= v
.new_var_extern
(ret
)
3062 v
.adapt_signature
(mpropdef
, arguments
)
3063 v
.unbox_signature_extern
(mpropdef
, arguments
)
3066 v
.add
("{externname}({arguments.join(", ")});")
3068 v
.add
("{res} = {externname}({arguments.join(", ")});")
3069 res
= v
.box_extern
(res
, ret
.as(not null))
3075 # Compile an extern factory
3076 # Return `true` if the compilation was successful, `false` if a fall-back is needed
3077 fun compile_externinit_to_c
(v
: AbstractCompilerVisitor, mpropdef
: MMethodDef, arguments
: Array[RuntimeVariable]): Bool
3080 var at
= self.get_single_annotation
("extern", v
.compiler
.modelbuilder
)
3082 externname
= at
.arg_as_string
(v
.compiler
.modelbuilder
)
3083 if externname
== null then return false
3087 v
.add_extern
(mpropdef
.mclassdef
.mmodule
)
3088 v
.adapt_signature
(mpropdef
, arguments
)
3089 v
.unbox_signature_extern
(mpropdef
, arguments
)
3090 var ret
= arguments
.first
.mtype
3091 var res
= v
.new_var_extern
(ret
)
3095 v
.add
("{res} = {externname}({arguments.join(", ")});")
3096 res
= v
.box_extern
(res
, ret
)
3102 redef class AAttrPropdef
3103 redef fun can_inline
: Bool do return not is_lazy
3105 redef fun compile_to_c
(v
, mpropdef
, arguments
)
3107 if mpropdef
== mreadpropdef
then
3108 assert arguments
.length
== 1
3109 var recv
= arguments
.first
3113 var ret
= self.mtype
3114 var useiset
= not ret
.is_c_primitive
and not ret
isa MNullableType
3115 var guard
= self.mlazypropdef
.mproperty
3117 set
= v
.isset_attribute
(self.mpropdef
.mproperty
, recv
)
3119 set
= v
.read_attribute
(guard
, recv
)
3121 v
.add
("if(likely({set})) \{")
3122 res
= v
.read_attribute
(self.mpropdef
.mproperty
, recv
)
3125 var value
= evaluate_expr
(v
, recv
)
3127 v
.assign
(res
, value
)
3129 var true_v
= v
.bool_instance
(true)
3130 v
.write_attribute
(guard
, arguments
.first
, true_v
)
3134 res
= v
.read_attribute
(self.mpropdef
.mproperty
, arguments
.first
)
3136 v
.assign
(v
.frame
.returnvar
.as(not null), res
)
3137 else if mpropdef
== mwritepropdef
then
3138 assert arguments
.length
== 2
3139 var recv
= arguments
.first
3140 var arg
= arguments
[1]
3141 if is_optional
and v
.maybenull
(arg
) then
3142 var value
= v
.new_var
(self.mpropdef
.static_mtype
.as(not null))
3143 v
.add
("if ({arg} == NULL) \{")
3144 v
.assign
(value
, evaluate_expr
(v
, recv
))
3146 v
.assign
(value
, arg
)
3150 v
.write_attribute
(self.mpropdef
.mproperty
, arguments
.first
, arg
)
3152 var ret
= self.mtype
3153 var useiset
= not ret
.is_c_primitive
and not ret
isa MNullableType
3155 v
.write_attribute
(self.mlazypropdef
.mproperty
, arguments
.first
, v
.bool_instance
(true))
3163 fun init_expr
(v
: AbstractCompilerVisitor, recv
: RuntimeVariable)
3165 if has_value
and not is_lazy
and not n_expr
isa ANullExpr then evaluate_expr
(v
, recv
)
3168 # Evaluate, store and return the default value of the attribute
3169 private fun evaluate_expr
(v
: AbstractCompilerVisitor, recv
: RuntimeVariable): RuntimeVariable
3171 var oldnode
= v
.current_node
3172 v
.current_node
= self
3173 var old_frame
= v
.frame
3174 var frame
= new StaticFrame(v
, self.mreadpropdef
.as(not null), recv
.mcasttype
.undecorate
.as(MClassType), [recv
])
3178 var mtype
= self.mtype
3179 assert mtype
!= null
3181 var nexpr
= self.n_expr
3182 var nblock
= self.n_block
3183 if nexpr
!= null then
3184 value
= v
.expr
(nexpr
, mtype
)
3185 else if nblock
!= null then
3186 value
= v
.new_var
(mtype
)
3187 frame
.returnvar
= value
3188 frame
.returnlabel
= v
.get_name
("RET_LABEL")
3191 v
.add
("{frame.returnlabel.as(not null)}:(void)0;")
3197 v
.write_attribute
(self.mpropdef
.mproperty
, recv
, value
)
3200 v
.current_node
= oldnode
3205 fun check_expr
(v
: AbstractCompilerVisitor, recv
: RuntimeVariable)
3207 var nexpr
= self.n_expr
3208 if nexpr
!= null then return
3210 var oldnode
= v
.current_node
3211 v
.current_node
= self
3212 var old_frame
= v
.frame
3213 var frame
= new StaticFrame(v
, self.mpropdef
.as(not null), recv
.mtype
.as(MClassType), [recv
])
3215 # Force read to check the initialization
3216 v
.read_attribute
(self.mpropdef
.mproperty
, recv
)
3218 v
.current_node
= oldnode
3222 redef class AClassdef
3223 private fun compile_to_c
(v
: AbstractCompilerVisitor, mpropdef
: MMethodDef, arguments
: Array[RuntimeVariable])
3225 if mpropdef
.mproperty
.is_root_init
then
3226 assert arguments
.length
== 1
3227 if not mpropdef
.is_intro
then
3228 v
.supercall
(mpropdef
, arguments
.first
.mtype
.as(MClassType), arguments
)
3238 # Try to compile self as an expression
3239 # Do not call this method directly, use `v.expr` instead
3240 private fun expr
(v
: AbstractCompilerVisitor): nullable RuntimeVariable
3242 v
.add
("PRINT_ERROR(\"NOT YET IMPLEMENTED {class_name}:{location.to_s}\\n\
");")
3243 var mtype
= self.mtype
3244 if mtype
== null then
3247 var res
= v
.new_var
(mtype
)
3248 v
.add
("/* {res} = NOT YET {class_name} */")
3253 # Try to compile self as a statement
3254 # Do not call this method directly, use `v.stmt` instead
3255 private fun stmt
(v
: AbstractCompilerVisitor)
3261 redef class ABlockExpr
3264 for e
in self.n_expr
do v
.stmt
(e
)
3268 var last
= self.n_expr
.last
3269 for e
in self.n_expr
do
3270 if e
== last
then break
3273 return v
.expr
(last
, null)
3277 redef class AVardeclExpr
3280 var variable
= self.variable
.as(not null)
3281 var ne
= self.n_expr
3283 var i
= v
.expr
(ne
, variable
.declared_type
)
3284 v
.assign
(v
.variable
(variable
), i
)
3289 redef class AVarExpr
3292 var res
= v
.variable
(self.variable
.as(not null))
3293 var mtype
= self.mtype
.as(not null)
3294 return v
.autoadapt
(res
, mtype
)
3298 redef class AVarAssignExpr
3301 var variable
= self.variable
.as(not null)
3302 var i
= v
.expr
(self.n_value
, variable
.declared_type
)
3303 v
.assign
(v
.variable
(variable
), i
)
3308 redef class AVarReassignExpr
3311 var variable
= self.variable
.as(not null)
3312 var vari
= v
.variable
(variable
)
3313 var value
= v
.expr
(self.n_value
, variable
.declared_type
)
3314 var res
= v
.compile_callsite
(self.reassign_callsite
.as(not null), [vari
, value
])
3316 v
.assign
(v
.variable
(variable
), res
)
3320 redef class ASelfExpr
3321 redef fun expr
(v
) do return v
.frame
.arguments
.first
3324 redef class AImplicitSelfExpr
3325 redef fun expr
(v
) do
3326 if not is_sys
then return super
3327 return v
.new_expr
("glob_sys", mtype
.as(not null))
3331 redef class AEscapeExpr
3332 redef fun stmt
(v
) do v
.add
("goto BREAK_{v.escapemark_name(self.escapemark)};")
3335 redef class AReturnExpr
3338 var nexpr
= self.n_expr
3339 if nexpr
!= null then
3340 var returnvar
= v
.frame
.returnvar
.as(not null)
3341 var i
= v
.expr
(nexpr
, returnvar
.mtype
)
3342 v
.assign
(returnvar
, i
)
3344 v
.add
("goto {v.frame.returnlabel.as(not null)};")
3348 redef class AAbortExpr
3349 redef fun stmt
(v
) do v
.add_abort
("Aborted")
3355 var cond
= v
.expr_bool
(self.n_expr
)
3356 v
.add
("if ({cond})\{")
3365 var res
= v
.new_var
(self.mtype
.as(not null))
3366 var cond
= v
.expr_bool
(self.n_expr
)
3367 v
.add
("if ({cond})\{")
3368 v
.assign
(res
, v
.expr
(self.n_then
.as(not null), null))
3370 v
.assign
(res
, v
.expr
(self.n_else
.as(not null), null))
3376 redef class AIfexprExpr
3379 var res
= v
.new_var
(self.mtype
.as(not null))
3380 var cond
= v
.expr_bool
(self.n_expr
)
3381 v
.add
("if ({cond})\{")
3382 v
.assign
(res
, v
.expr
(self.n_then
, null))
3384 v
.assign
(res
, v
.expr
(self.n_else
, null))
3393 if self.n_catch
!= null then
3394 v
.add
("catchStack.cursor += 1;")
3395 v
.add
("if(!setjmp(catchStack.envs[catchStack.cursor]))\{")
3396 v
.stmt
(self.n_block
)
3397 v
.add
("catchStack.cursor -= 1;")
3399 v
.add
("catchStack.cursor -= 1;")
3400 v
.stmt
(self.n_catch
)
3403 v
.stmt
(self.n_block
)
3405 v
.add_escape_label
(break_mark
)
3409 redef class AWhileExpr
3413 var cond
= v
.expr_bool
(self.n_expr
)
3414 v
.add
("if (!{cond}) break;")
3415 v
.stmt
(self.n_block
)
3416 v
.add_escape_label
(continue_mark
)
3418 v
.add_escape_label
(break_mark
)
3422 redef class ALoopExpr
3426 v
.stmt
(self.n_block
)
3427 v
.add_escape_label
(continue_mark
)
3429 v
.add_escape_label
(break_mark
)
3433 redef class AForExpr
3436 for g
in n_groups
do
3437 var cl
= v
.expr
(g
.n_expr
, null)
3438 var it_meth
= g
.method_iterator
3439 assert it_meth
!= null
3440 var it
= v
.compile_callsite
(it_meth
, [cl
])
3445 for g
in n_groups
do
3447 var isok_meth
= g
.method_is_ok
3448 assert isok_meth
!= null
3449 var ok
= v
.compile_callsite
(isok_meth
, [it
])
3451 v
.add
("if(!{ok}) break;")
3452 if g
.variables
.length
== 1 then
3453 var item_meth
= g
.method_item
3454 assert item_meth
!= null
3455 var i
= v
.compile_callsite
(item_meth
, [it
])
3457 v
.assign
(v
.variable
(g
.variables
.first
), i
)
3458 else if g
.variables
.length
== 2 then
3459 var key_meth
= g
.method_key
3460 assert key_meth
!= null
3461 var i
= v
.compile_callsite
(key_meth
, [it
])
3463 v
.assign
(v
.variable
(g
.variables
[0]), i
)
3464 var item_meth
= g
.method_item
3465 assert item_meth
!= null
3466 i
= v
.compile_callsite
(item_meth
, [it
])
3468 v
.assign
(v
.variable
(g
.variables
[1]), i
)
3473 v
.stmt
(self.n_block
)
3474 v
.add_escape_label
(continue_mark
)
3475 for g
in n_groups
do
3476 var next_meth
= g
.method_next
3477 assert next_meth
!= null
3478 v
.compile_callsite
(next_meth
, [g
.it
])
3481 v
.add_escape_label
(break_mark
)
3483 for g
in n_groups
do
3484 var method_finish
= g
.method_finish
3485 if method_finish
!= null then
3486 # TODO: Find a way to call this also in long escape (e.g. return)
3487 v
.compile_callsite
(method_finish
, [g
.it
])
3493 redef class AForGroup
3494 # C variable representing the iterator
3495 private var it
: RuntimeVariable is noinit
3498 redef class AAssertExpr
3501 if v
.compiler
.modelbuilder
.toolcontext
.opt_no_check_assert
.value
then return
3503 var cond
= v
.expr_bool
(self.n_expr
)
3504 v
.add
("if (unlikely(!{cond})) \{")
3508 v
.add_abort
("Assert '{nid.text}' failed")
3510 v
.add_abort
("Assert failed")
3519 var res
= v
.new_var
(self.mtype
.as(not null))
3520 var i1
= v
.expr_bool
(self.n_expr
)
3521 v
.add
("if ({i1}) \{")
3524 var i2
= v
.expr_bool
(self.n_expr2
)
3525 v
.add
("{res} = {i2};")
3531 redef class AImpliesExpr
3534 var res
= v
.new_var
(self.mtype
.as(not null))
3535 var i1
= v
.expr_bool
(self.n_expr
)
3536 v
.add
("if (!{i1}) \{")
3539 var i2
= v
.expr_bool
(self.n_expr2
)
3540 v
.add
("{res} = {i2};")
3546 redef class AAndExpr
3549 var res
= v
.new_var
(self.mtype
.as(not null))
3550 var i1
= v
.expr_bool
(self.n_expr
)
3551 v
.add
("if (!{i1}) \{")
3554 var i2
= v
.expr_bool
(self.n_expr2
)
3555 v
.add
("{res} = {i2};")
3561 redef class ANotExpr
3564 var cond
= v
.expr_bool
(self.n_expr
)
3565 return v
.new_expr
("!{cond}", self.mtype
.as(not null))
3569 redef class AOrElseExpr
3572 var res
= v
.new_var
(self.mtype
.as(not null))
3573 var i1
= v
.expr
(self.n_expr
, null)
3575 if not v
.maybenull
(i1
) then return i1
3577 v
.add
("if ({i1}!=NULL) \{")
3580 var i2
= v
.expr
(self.n_expr2
, null)
3587 redef class AIntegerExpr
3588 redef fun expr
(v
) do
3589 if value
isa Int then return v
.int_instance
(value
.as(Int))
3590 if value
isa Byte then return v
.byte_instance
(value
.as(Byte))
3591 if value
isa Int8 then return v
.int8_instance
(value
.as(Int8))
3592 if value
isa Int16 then return v
.int16_instance
(value
.as(Int16))
3593 if value
isa UInt16 then return v
.uint16_instance
(value
.as(UInt16))
3594 if value
isa Int32 then return v
.int32_instance
(value
.as(Int32))
3595 if value
isa UInt32 then return v
.uint32_instance
(value
.as(UInt32))
3596 # Should never happen
3601 redef class AFloatExpr
3602 redef fun expr
(v
) do return v
.float_instance
("{self.n_float.text}") # FIXME use value, not n_float
3605 redef class ACharExpr
3606 redef fun expr
(v
) do
3607 if is_ascii
then return v
.byte_instance
(value
.as(not null).ascii
)
3608 if is_code_point
then return v
.int_instance
(value
.as(not null).code_point
)
3609 return v
.char_instance
(self.value
.as(not null))
3613 redef class AArrayExpr
3616 var mtype
= self.element_mtype
.as(not null)
3617 var array
= new Array[RuntimeVariable]
3618 var res
= v
.array_instance
(array
, mtype
)
3620 var old_comprehension
= v
.frame
.comprehension
3621 v
.frame
.comprehension
= res
3622 for nexpr
in self.n_exprs
do
3625 v
.frame
.comprehension
= old_comprehension
3631 redef class AugmentedStringFormExpr
3632 # Factorize the making of a `Regex` object from a literal prefixed string
3633 protected fun make_re
(v
: AbstractCompilerVisitor, rs
: RuntimeVariable): nullable RuntimeVariable do
3636 var res
= v
.compile_callsite
(re
, [rs
])
3638 print
"Cannot call property `to_re` on {self}"
3641 for i
in suffix
.chars
do
3643 var ign
= ignore_case
3645 v
.compile_callsite
(ign
, [res
, v
.bool_instance
(true)])
3651 v
.compile_callsite
(nl
, [res
, v
.bool_instance
(true)])
3657 v
.compile_callsite
(ext
, [res
, v
.bool_instance
(false)])
3660 # Should not happen, this needs to be updated
3661 # along with the addition of new suffixes
3668 redef class AStringFormExpr
3669 redef fun expr
(v
) do return v
.string_instance
(value
)
3672 redef class AStringExpr
3673 redef fun expr
(v
) do
3674 var s
= v
.string_instance
(value
)
3675 if is_string
then return s
3676 if is_bytestring
then
3677 var ns
= v
.native_string_instance
(bytes
.items
, bytes
.length
)
3678 var ln
= v
.int_instance
(bytes
.length
)
3679 var cs
= to_bytes_with_copy
3681 var res
= v
.compile_callsite
(cs
, [ns
, ln
])
3685 var res
= make_re
(v
, s
)
3689 print
"Unimplemented prefix or suffix for {self}"
3696 redef class ASuperstringExpr
3699 var type_string
= v
.mmodule
.string_type
3701 # Collect elements of the superstring
3702 var array
= new Array[AExpr]
3703 for ne
in self.n_exprs
do
3704 # Drop literal empty string.
3705 # They appears in things like "{a}" that is ["", a, ""]
3706 if ne
isa AStringFormExpr and ne
.value
== "" then continue # skip empty sub-strings
3710 # Store the allocated native array in a static variable
3712 var varonce
= v
.get_name
("varonce")
3713 v
.add
("if (unlikely({varonce}==NULL)) \{")
3715 # The native array that will contains the elements to_s-ized.
3716 # For fast concatenation.
3717 var a
= v
.native_array_instance
(type_string
, v
.int_instance
(array
.length
))
3719 v
.add_decl
("static {a.mtype.ctype} {varonce};")
3721 # Pre-fill the array with the literal string parts.
3722 # So they do not need to be filled again when reused
3723 for i
in [0..array
.length
[ do
3725 if not ne
isa AStringFormExpr then continue
3726 var e
= v
.expr
(ne
, null)
3727 v
.native_array_set
(a
, i
, e
)
3731 # Take the native-array from the store.
3732 # The point is to prevent that some recursive execution use (and corrupt) the same native array
3733 # WARNING: not thread safe! (FIXME?)
3734 v
.add
("{a} = {varonce};")
3735 v
.add
("{varonce} = NULL;")
3738 # Stringify the elements and put them in the native array
3739 var to_s_method
= v
.get_property
("to_s", v
.object_type
)
3740 for i
in [0..array
.length
[ do
3742 if ne
isa AStringFormExpr then continue
3743 var e
= v
.expr
(ne
, null)
3744 # Skip the `to_s` if the element is already a String
3745 if not e
.mcasttype
.is_subtype
(v
.compiler
.mainmodule
, null, type_string
) then
3746 e
= v
.send
(to_s_method
, [e
]).as(not null)
3748 v
.native_array_set
(a
, i
, e
)
3751 # Fast join the native string to get the result
3752 var res
= v
.send
(v
.get_property
("native_to_s", a
.mtype
), [a
])
3755 if is_re
then res
= make_re
(v
, res
)
3757 # We finish to work with the native array,
3758 # so store it so that it can be reused
3759 v
.add
("{varonce} = {a};")
3765 redef class ACrangeExpr
3768 var i1
= v
.expr
(self.n_expr
, null)
3769 var i2
= v
.expr
(self.n_expr2
, null)
3770 var mtype
= self.mtype
.as(MClassType)
3771 var res
= v
.init_instance
(mtype
)
3772 v
.compile_callsite
(init_callsite
.as(not null), [res
, i1
, i2
])
3777 redef class AOrangeExpr
3780 var i1
= v
.expr
(self.n_expr
, null)
3781 var i2
= v
.expr
(self.n_expr2
, null)
3782 var mtype
= self.mtype
.as(MClassType)
3783 var res
= v
.init_instance
(mtype
)
3784 v
.compile_callsite
(init_callsite
.as(not null), [res
, i1
, i2
])
3789 redef class ATrueExpr
3790 redef fun expr
(v
) do return v
.bool_instance
(true)
3793 redef class AFalseExpr
3794 redef fun expr
(v
) do return v
.bool_instance
(false)
3797 redef class ANullExpr
3798 redef fun expr
(v
) do return v
.null_instance
3801 redef class AIsaExpr
3804 var i
= v
.expr
(self.n_expr
, null)
3805 var cast_type
= self.cast_type
3806 if cast_type
== null then return null # no-no on broken node
3807 return v
.type_test
(i
, cast_type
, "isa")
3811 redef class AAsCastExpr
3814 var i
= v
.expr
(self.n_expr
, null)
3815 if v
.compiler
.modelbuilder
.toolcontext
.opt_no_check_assert
.value
then return i
3817 v
.add_cast
(i
, self.mtype
.as(not null), "as")
3822 redef class AAsNotnullExpr
3825 var i
= v
.expr
(self.n_expr
, null)
3826 if v
.compiler
.modelbuilder
.toolcontext
.opt_no_check_assert
.value
then return i
3828 if not v
.maybenull
(i
) then return i
3830 v
.add
("if (unlikely({i} == NULL)) \{")
3831 v
.add_abort
("Cast failed")
3837 redef class AParExpr
3838 redef fun expr
(v
) do return v
.expr
(self.n_expr
, null)
3841 redef class AOnceExpr
3844 var mtype
= self.mtype
.as(not null)
3845 var name
= v
.get_name
("varonce")
3846 var guard
= v
.get_name
(name
+ "_guard")
3847 v
.add_decl
("static {mtype.ctype} {name};")
3848 v
.add_decl
("static int {guard};")
3849 var res
= v
.new_var
(mtype
)
3850 v
.add
("if (likely({guard})) \{")
3851 v
.add
("{res} = {name};")
3853 var i
= v
.expr
(self.n_expr
, mtype
)
3854 v
.add
("{res} = {i};")
3855 v
.add
("{name} = {res};")
3856 v
.add
("{guard} = 1;")
3862 redef class ASendExpr
3865 var recv
= v
.expr
(self.n_expr
, null)
3866 var callsite
= self.callsite
.as(not null)
3867 if callsite
.is_broken
then return null
3868 var args
= v
.varargize
(callsite
.mpropdef
, callsite
.signaturemap
, recv
, self.raw_arguments
)
3869 return v
.compile_callsite
(callsite
, args
)
3873 redef class ASendReassignFormExpr
3876 var recv
= v
.expr
(self.n_expr
, null)
3877 var callsite
= self.callsite
.as(not null)
3878 if callsite
.is_broken
then return
3879 var args
= v
.varargize
(callsite
.mpropdef
, callsite
.signaturemap
, recv
, self.raw_arguments
)
3881 var value
= v
.expr
(self.n_value
, null)
3883 var left
= v
.compile_callsite
(callsite
, args
)
3886 var res
= v
.compile_callsite
(self.reassign_callsite
.as(not null), [left
, value
])
3890 v
.compile_callsite
(self.write_callsite
.as(not null), args
)
3894 redef class ASuperExpr
3897 var frame
= v
.frame
.as(not null)
3898 var recv
= frame
.arguments
.first
3900 var callsite
= self.callsite
3901 if callsite
!= null then
3902 if callsite
.is_broken
then return null
3905 if self.n_args
.n_exprs
.is_empty
then
3906 # Add automatic arguments for the super init call
3908 for i
in [0..callsite
.msignature
.arity
[ do
3909 args
.add
(frame
.arguments
[i
+1])
3912 args
= v
.varargize
(callsite
.mpropdef
, callsite
.signaturemap
, recv
, self.n_args
.n_exprs
)
3916 var res
= v
.compile_callsite
(callsite
, args
)
3920 var mpropdef
= self.mpropdef
.as(not null)
3923 if self.n_args
.n_exprs
.is_empty
then
3924 args
= frame
.arguments
3926 args
= v
.varargize
(mpropdef
, signaturemap
, recv
, self.n_args
.n_exprs
)
3929 # Standard call-next-method
3930 return v
.supercall
(mpropdef
, recv
.mtype
.as(MClassType), args
)
3934 redef class ANewExpr
3937 var mtype
= self.recvtype
3938 assert mtype
!= null
3940 if mtype
.mclass
.name
== "NativeArray" then
3941 assert self.n_args
.n_exprs
.length
== 1
3942 var l
= v
.expr
(self.n_args
.n_exprs
.first
, null)
3943 assert mtype
isa MGenericType
3944 var elttype
= mtype
.arguments
.first
3945 return v
.native_array_instance
(elttype
, l
)
3948 var callsite
= self.callsite
3949 if callsite
== null then return v
.init_instance_or_extern
(mtype
)
3950 if callsite
.is_broken
then return null
3953 # new factories are badly implemented.
3954 # They assume a stub temporary receiver exists.
3955 # This temporary receiver is required because it
3956 # currently holds the method and the formal types.
3958 # However, this object could be reused if the formal types are the same.
3959 # Therefore, the following code will `once` it in these case
3960 if callsite
.mproperty
.is_new
and not mtype
.need_anchor
then
3961 var name
= v
.get_name
("varoncenew")
3962 var guard
= v
.get_name
(name
+ "_guard")
3963 v
.add_decl
("static {mtype.ctype} {name};")
3964 v
.add_decl
("static int {guard};")
3965 recv
= v
.new_var
(mtype
)
3966 v
.add
("if (likely({guard})) \{")
3967 v
.add
("{recv} = {name};")
3969 var i
= v
.init_instance_or_extern
(mtype
)
3970 v
.add
("{recv} = {i};")
3971 v
.add
("{name} = {recv};")
3972 v
.add
("{guard} = 1;")
3975 recv
= v
.init_instance_or_extern
(mtype
)
3978 var args
= v
.varargize
(callsite
.mpropdef
, callsite
.signaturemap
, recv
, self.n_args
.n_exprs
)
3979 var res2
= v
.compile_callsite
(callsite
, args
)
3980 if res2
!= null then
3981 #self.debug("got {res2} from {mproperty}. drop {recv}")
3988 redef class AAttrExpr
3991 var recv
= v
.expr
(self.n_expr
, null)
3992 var mproperty
= self.mproperty
.as(not null)
3993 return v
.read_attribute
(mproperty
, recv
)
3997 redef class AAttrAssignExpr
4000 var recv
= v
.expr
(self.n_expr
, null)
4001 var i
= v
.expr
(self.n_value
, null)
4002 var mproperty
= self.mproperty
.as(not null)
4003 v
.write_attribute
(mproperty
, recv
, i
)
4008 redef class AAttrReassignExpr
4011 var recv
= v
.expr
(self.n_expr
, null)
4012 var value
= v
.expr
(self.n_value
, null)
4013 var mproperty
= self.mproperty
.as(not null)
4014 var attr
= v
.read_attribute
(mproperty
, recv
)
4015 var res
= v
.compile_callsite
(self.reassign_callsite
.as(not null), [attr
, value
])
4017 v
.write_attribute
(mproperty
, recv
, res
)
4021 redef class AIssetAttrExpr
4024 var recv
= v
.expr
(self.n_expr
, null)
4025 var mproperty
= self.mproperty
.as(not null)
4026 return v
.isset_attribute
(mproperty
, recv
)
4030 redef class AVarargExpr
4033 return v
.expr
(self.n_expr
, null)
4037 redef class ANamedargExpr
4040 return v
.expr
(self.n_expr
, null)
4044 redef class ADebugTypeExpr
4053 redef class Array[E
]
4054 # Return a new `Array` with the elements only contened in self and not in `o`
4055 fun -(o
: Array[E
]): Array[E
] do
4056 var res
= new Array[E
]
4057 for e
in self do if not o
.has
(e
) then res
.add
(e
)
4063 # All `MProperty` associated to all `MClassDef` of `mclass`
4064 fun properties
(mclass
: MClass): Set[MProperty] do
4065 if not self.properties_cache
.has_key
(mclass
) then
4066 var properties
= new HashSet[MProperty]
4067 var parents
= new Array[MClass]
4068 if self.flatten_mclass_hierarchy
.has
(mclass
) then
4069 parents
.add_all
(mclass
.in_hierarchy
(self).direct_greaters
)
4071 for parent
in parents
do
4072 properties
.add_all
(self.properties
(parent
))
4074 for mclassdef
in mclass
.mclassdefs
do
4075 if not self.in_importation
<= mclassdef
.mmodule
then continue
4076 for mprop
in mclassdef
.intro_mproperties
do
4077 properties
.add
(mprop
)
4080 self.properties_cache
[mclass
] = properties
4082 return properties_cache
[mclass
]
4084 private var properties_cache
: Map[MClass, Set[MProperty]] = new HashMap[MClass, Set[MProperty]]
4086 # Write FFI and nitni results to file
4087 fun finalize_ffi
(c
: AbstractCompiler) do end
4089 # Give requided addinional system libraries (as given to LD_LIBS)
4090 # Note: can return null instead of an empty set
4091 fun collect_linker_libs
: nullable Array[String] do return null
4094 # Create a tool context to handle options and paths
4095 var toolcontext
= new ToolContext
4097 toolcontext
.tooldescription
= "Usage: nitc [OPTION]... file.nit...\nCompiles Nit programs."
4099 # We do not add other options, so process them now!
4100 toolcontext
.process_options
(args
)
4102 # We need a model to collect stufs
4103 var model
= new Model
4104 # An a model builder to parse files
4105 var modelbuilder
= new ModelBuilder(model
, toolcontext
)
4107 var arguments
= toolcontext
.option_context
.rest
4108 if arguments
.length
> 1 and toolcontext
.opt_output
.value
!= null then
4109 print
"Option Error: --output needs a single source file. Do you prefer --dir?"
4113 # Here we load an process all modules passed on the command line
4114 var mmodules
= modelbuilder
.parse
(arguments
)
4116 if mmodules
.is_empty
then toolcontext
.quit
4118 modelbuilder
.run_phases
4120 for mmodule
in mmodules
do
4121 toolcontext
.info
("*** PROCESS {mmodule} ***", 1)
4123 toolcontext
.run_global_phases
(ms
)