package nitc
import abstracttool
+import analysis
+import program
private import compiling
+private import syntax
# The main class of the nitcompiler program
class NitCompiler
readable var _opt_bindir: OptionString = new OptionString("NIT tools directory", "--bindir")
readable var _opt_compdir: OptionString = new OptionString("Intermediate compilation directory", "--compdir")
readable var _opt_extension_prefix: OptionString = new OptionString("Append prefix to file extension", "-p", "--extension-prefix")
+ readable var _opt_dump: OptionBool = new OptionBool("Dump intermediate code", "--dump")
init
do
super("nitc")
- option_context.add_option(opt_output, opt_boost, opt_no_cc, opt_global, opt_clibdir, opt_bindir, opt_compdir, opt_extension_prefix)
+ option_context.add_option(opt_output, opt_boost, opt_no_cc, opt_global, opt_clibdir, opt_bindir, opt_compdir, opt_extension_prefix, opt_dump)
end
redef fun process_options
if clibdir == null then
var dir = once ("NIT_DIR".to_symbol).environ
if dir.is_empty then
- var dir = "{sys.program_name.dirname}/../lib"
+ dir = "{sys.program_name.dirname}/../clib"
if dir.file_exists then clibdir = dir
else
- dir = "{dir}/lib"
+ dir = "{dir}/clib"
if dir.file_exists then clibdir = dir
end
if clibdir == null then
- error("Error: Cannot locate NIT C library directory. Uses --clibdir or envvar NIT_DIR.")
- exit(1)
+ fatal_error(null, "Error: Cannot locate NIT C library directory. Uses --clibdir or envvar NIT_DIR.")
end
end
bindir = opt_bindir.value
if bindir == null then
var dir = once ("NIT_DIR".to_symbol).environ
if dir.is_empty then
- var dir = "{sys.program_name.dirname}/../bin"
+ dir = "{sys.program_name.dirname}/../bin"
if dir.file_exists then bindir = dir
else
dir = "{dir}/bin"
if dir.file_exists then bindir = dir
end
if bindir == null then
- error("Error: Cannot locate NIT tools directory. Uses --bindir or envvar NIT_DIR.")
- exit(1)
+ fatal_error(null, "Error: Cannot locate NIT tools directory. Uses --bindir or envvar NIT_DIR.")
end
end
end
- redef fun perform_work(mods)
+ fun dump_intermediate_code(mods: Collection[MMModule])
do
for mod in mods do
- mod.compile_prog_to_c(self)
+ for c in mod.local_classes do
+ if not c isa MMConcreteClass then continue
+ for p in c.local_local_properties do
+ var routine: nullable IRoutine = null
+ if p isa MMAttribute then
+ routine = p.iroutine
+ else if p isa MMMethod then
+ routine = p.iroutine
+ end
+ if routine == null then continue
+ print "**** Property {p.full_name} ****"
+ var icd = new ICodeDumper
+ routine.dump(icd)
+ print "**** OPTIMIZE {p.full_name} ****"
+ routine.optimize(mod)
+ icd = new ICodeDumper
+ routine.dump(icd)
+ end
+ end
end
+ end
+ redef fun perform_work(mods)
+ do
+ if opt_dump.value then
+ dump_intermediate_code(mods)
+ end
+ for mod in mods do
+ var p = new Program(mod)
+ p.do_table_computation(self)
+ p.compile_prog_to_c(self)
+ end
end
end