src/analysis/inline_methods.nit

   1 # This file is part of NIT ( http://www.nitlanguage.org ).
   2 #
   3 # Copyright 2009 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 # Detect inlinable methods and inline them
  18 package inline_methods
  19
  20 import icode
  21
  22 private class InlineMethodVisitor
  23 special ICodeVisitor
  24         var _pass: Int = 0
  25         var _icb: ICodeBuilder
  26
  27         var _current_inlining: Array[IRoutine]
  28
  29         redef fun visit_icode(ic)
  30         do
  31                 if ic isa ICall then
  32                         var m = ic.property
  33                         var ir = m.iroutine
  34                         if ir != null and ic.is_inlinable then
  35                                 if _current_inlining.has(ir) then
  36                                         # We cannot inline ir
  37                                         # So, at least transform the call to a static one
  38                                         current_icode.delete
  39                                         var icall = new IStaticCall(ic.property, ic.exprs)
  40                                         icall.closure_defs = ic.closure_defs
  41                                         icall.result = ic.result
  42                                         current_icode.insert_before(icall)
  43                                         current_icode.delete
  44                                 else
  45                                         var icb = _icb
  46                                         _current_inlining.push(ir)
  47                                         var seq = new ISeq
  48                                         var old_seq = icb.seq
  49                                         icb.seq = seq
  50                                         current_icode.insert_before(seq)
  51                                         var e = icb.inline_routine(ir, ic.exprs, ic.closure_defs)
  52                                         var r = ic.result
  53                                         if r != null then
  54                                                 assert e != null
  55                                                 current_icode.insert_before(new IMove(r, e))
  56                                         end
  57                                         current_icode.delete
  58                                         icb.seq = old_seq
  59                                         visit_icode(seq)
  60                                         _current_inlining.pop
  61                                 end
  62                         end
  63                 end
  64                 super
  65         end
  66
  67         init(m: MMModule, r: IRoutine)
  68         do
  69                 _current_inlining = [r]
  70                 _icb = new ICodeBuilder(m, r)
  71         end
  72 end
  73
  74 redef class ICall
  75         fun is_inlinable: Bool
  76         do
  77                 var m = property
  78                 var mn = m.name
  79                 var cn = m.local_class.name
  80                 return (m.is_intern and cn != once ("Object".to_symbol)) or
  81                         (cn == (once ("Array".to_symbol)) and (mn == (once ("length".to_symbol)) or mn == (once ("[]".to_symbol)) or mn == (once ("iterate".to_symbol)))) or
  82                         (cn == (once ("AbstractArrayRead".to_symbol)) and (mn == (once ("length".to_symbol)) or mn == (once ("[]".to_symbol)))) or
  83                         (m.global.intro.local_class.name == (once ("Inline__".to_symbol)))
  84         end
  85 end
  86
  87 redef class IRoutine
  88         fun inline_methods(m: MMModule)
  89         do
  90                 var v = new InlineMethodVisitor(m, self)
  91                 v.visit_iroutine(self)
  92         end
  93 end