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
  26         redef fun visit_icode(ic)
  27         do
  28                 if ic isa ICall then
  29                         var m = ic.property
  30                         if m.iroutine != null and ic.is_inlinable then
  31                                 var ir = m.iroutine.as(not null)
  32                                 var seq = new ISeq
  33                                 current_icode.insert_before(seq)
  34                                 var e = ir.inline_in_seq(seq, ic.exprs)
  35                                 var r = ic.result
  36                                 if r != null then
  37                                         assert e != null
  38                                         current_icode.insert_before(new IMove(r, e))
  39                                 end
  40                                 current_icode.delete
  41                                 visit_icode(seq)
  42                         end
  43                 end
  44                 super
  45         end
  46 end
  47
  48 redef class ICall
  49         fun is_inlinable: Bool
  50         do
  51                 var m = property
  52                 var mn = m.name
  53                 var cn = m.local_class.name
  54                 return (m.is_intern and cn != once ("Object".to_symbol)) or
  55                         (cn == (once ("Array".to_symbol)) and (mn == (once ("length".to_symbol)) or mn == (once ("[]".to_symbol)))) or
  56                         (cn == (once ("AbstractArrayRead".to_symbol)) and (mn == (once ("length".to_symbol)) or mn == (once ("[]".to_symbol))))
  57         end
  58 end
  59
  60 redef class IRoutine
  61         fun inline_methods
  62         do
  63                 var v = new InlineMethodVisitor
  64                 v.visit_iroutine(self)
  65         end
  66 end