src/analysis/inline_get_and_set.nit

   1 # This file is part of NIT ( http://www.nitlanguage.org ).
   2 #
   3 # Copyright 2009 Jean-Sebastien Gelinas <calestar@gmail.com>
   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 # Inline every automatic 'getters' and 'setters'
  18 # This optimization is done BEFORE analysis
  19 package inline_get_and_set
  20
  21 import icode
  22 import syntax
  23 import program
  24
  25 redef class Program
  26         var _number_getter_setter_inlined : Int = 0
  27
  28         # This method will create a file and output this optimization's stats in it
  29         fun dump_inline_get_set(directory_name: String) do
  30                 var f = new OFStream.open("{directory_name}/{main_module.name}.inline_get_set.log")
  31
  32                 f.write ("Number of getters and setters inlined: {_number_getter_setter_inlined}\n")
  33
  34                 f.close
  35         end
  36
  37         fun inline_get_set do
  38                 with_each_iroutines !action(i, m) do
  39                         var v = new InlineGetSetVisitor(m, i)
  40                         v.visit_iroutine(i)
  41
  42                         _number_getter_setter_inlined += v.number_inlined
  43                 end
  44         end
  45 end
  46
  47 private class InlineGetSetVisitor
  48         super ICodeVisitor
  49         var _icb: ICodeBuilder
  50         readable var _number_inlined: Int = 0
  51
  52         redef fun visit_icode(ic)
  53         do
  54                 # Algo mostly from inline_methods.nit, by Jean Privat
  55                 if ic isa ICall then
  56                         var m = ic.property
  57                         var ir = m.iroutine
  58                         if ir != null and m isa MMAttrImplementationMethod then
  59                                 var icb = _icb
  60                                 var seq = new ISeq
  61                                 var old_seq = icb.seq
  62                                 icb.seq = seq
  63                                 current_icode.insert_before(seq)
  64                                 var e = icb.inline_routine(ir, ic.exprs, ic.closure_defs)
  65                                 var r = ic.result
  66                                 if r != null then
  67                                         assert e != null
  68                                         current_icode.insert_before(new IMove(r, e))
  69                                 end
  70                                 current_icode.delete
  71                                 icb.seq = old_seq
  72                                 _number_inlined += 1
  73                                 visit_icode(seq)
  74                         end
  75                 end
  76                 super
  77         end
  78
  79         init(m: MMModule, r: IRoutine)
  80         do
  81                 _icb = new ICodeBuilder(m, r)
  82         end
  83 end