src/nitc.nit

   1 # This file is part of NIT ( http://www.nitlanguage.org ).
   2 #
   3 # Copyright 2008 Jean Privat <jean@pryen.org>
   4 #
   5 # Licensed under the Apache License, Version 2.0 (the "License");
   6 # you may not use this file except in compliance with the License.
   7 # You may obtain a copy of the License at
   8 #
   9 #     http://www.apache.org/licenses/LICENSE-2.0
  10 #
  11 # Unless required by applicable law or agreed to in writing, software
  12 # distributed under the License is distributed on an "AS IS" BASIS,
  13 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  14 # See the License for the specific language governing permissions and
  15 # limitations under the License.
  16
  17 # Nit compiler main module
  18 package nitc
  19
  20 import abstracttool
  21 import analysis
  22 private import compiling
  23 private import syntax
  24
  25 # The main class of the nitcompiler program
  26 class NitCompiler
  27 special AbstractCompiler
  28         readable var _opt_output: OptionString = new OptionString("Output file", "-o", "--output")
  29         readable var _opt_boost: OptionBool = new OptionBool("Optimize compilation", "-O", "--boost")
  30         readable var _opt_no_cc: OptionBool = new OptionBool("Do not invoke C compiler", "--no_cc")
  31         readable var _opt_global: OptionBool = new OptionBool("Use global compilation", "--global")
  32         readable var _opt_clibdir: OptionString = new OptionString("NIT C library directory", "--clibdir")
  33         readable var _opt_bindir: OptionString = new OptionString("NIT tools directory", "--bindir")
  34         readable var _opt_compdir: OptionString = new OptionString("Intermediate compilation directory", "--compdir")
  35         readable var _opt_extension_prefix: OptionString = new OptionString("Append prefix to file extension", "-p", "--extension-prefix")
  36         readable var _opt_dump: OptionBool = new OptionBool("Dump intermediate code", "--dump")
  37
  38         init
  39         do
  40                 super("nitc")
  41                 option_context.add_option(opt_output, opt_boost, opt_no_cc, opt_global, opt_clibdir, opt_bindir, opt_compdir, opt_extension_prefix, opt_dump)
  42         end
  43
  44         redef fun process_options
  45         do
  46                 super
  47                 output_file = opt_output.value
  48                 boost = opt_boost.value
  49                 no_cc = opt_no_cc.value
  50                 var ext = opt_extension_prefix.value
  51                 if ext != null then ext_prefix = ext else ext_prefix = ""
  52                 global = opt_global.value
  53                 compdir = opt_compdir.value
  54                 if compdir == null then
  55                         var dir = once ("NIT_COMPDIR".to_symbol).environ
  56                         if not dir.is_empty then
  57                                 compdir = dir
  58                         end
  59                         if compdir == null then
  60                                 compdir = ".nit_compile"
  61                         end
  62                 end
  63                 compdir += ext_prefix
  64
  65                 clibdir = opt_clibdir.value
  66                 if clibdir == null then
  67                         var dir = once ("NIT_DIR".to_symbol).environ
  68                         if dir.is_empty then
  69                                 var dir = "{sys.program_name.dirname}/../lib"
  70                                 if dir.file_exists then clibdir = dir
  71                         else
  72                                 dir = "{dir}/lib"
  73                                 if dir.file_exists then clibdir = dir
  74                         end
  75                         if clibdir == null then
  76                                 fatal_error(null, "Error: Cannot locate NIT C library directory. Uses --clibdir or envvar NIT_DIR.")
  77                         end
  78                 end
  79                 bindir = opt_bindir.value
  80
  81                 if bindir == null then
  82                         var dir = once ("NIT_DIR".to_symbol).environ
  83                         if dir.is_empty then
  84                                 var dir = "{sys.program_name.dirname}/../bin"
  85                                 if dir.file_exists then bindir = dir
  86                         else
  87                                 dir = "{dir}/bin"
  88                                 if dir.file_exists then bindir = dir
  89                         end
  90                         if bindir == null then
  91                                 fatal_error(null, "Error: Cannot locate NIT tools directory. Uses --bindir or envvar NIT_DIR.")
  92                         end
  93                 end
  94         end
  95
  96         fun dump_intermediate_code(mods: Collection[MMModule])
  97         do
  98                 for mod in mods do
  99                         for c in mod.local_classes do
 100                                 if not c isa MMConcreteClass then continue
 101                                 for p in c.local_local_properties do
 102                                         var routine: nullable IRoutine = null
 103                                         if p isa MMAttribute then
 104                                                 routine = p.iroutine
 105                                         else if p isa MMMethod then
 106                                                 routine = p.iroutine
 107                                         end
 108                                         if routine == null then continue
 109                                         print "**** Property {p.full_name} ****"
 110                                         var icd = new ICodeDumper
 111                                         routine.dump(icd)
 112                                         print "**** OPTIMIZE {p.full_name} ****"
 113                                         routine.optimize
 114                                         icd = new ICodeDumper
 115                                         routine.dump(icd)
 116                                 end
 117                         end
 118                 end
 119         end
 120
 121         redef fun perform_work(mods)
 122         do
 123                 if opt_dump.value then
 124                         dump_intermediate_code(mods)
 125                 end
 126                 for mod in mods do
 127                         mod.compile_prog_to_c(self)
 128                 end
 129
 130         end
 131 end
 132
 133 var c = new NitCompiler
 134 c.exec_cmd_line