duby 0.0.2-java

data/lib/duby/jvm/method_lookup.rb

@@ -0,0 +1,193 @@
+module Duby
+  module JVM
+    module MethodLookup
+      # dummy log; it's expected the inclusion target will have it
+      def log(msg); end
+      
+      # def jvm_type(type)
+      #   return type if type.kind_of? Java::JavaClass
+      #   return type.jvm_type
+      # end
+      # 
+      # def convert_params(params)
+      #   params.map {|param| jvm_type(param)}
+      # end
+
+      def find_method(mapped_type, name, mapped_params, meta)
+        # mapped_type = jvm_type(mapped_type)
+        # mapped_params = convert_params(mapped_params)
+        raise ArgumentError if mapped_params.any? {|p| p.nil?}
+        if name == 'new'
+          if meta
+            name = "<init>"
+            constructor = true
+          else
+            constructor = false
+          end
+        end
+
+        begin
+          if constructor
+            method = mapped_type.constructor(*mapped_params)
+          else
+            method = mapped_type.java_method(name, *mapped_params)
+          end
+        rescue NameError
+          # exact args failed, do a deeper search
+          log "No exact match for #{mapped_type.name}.#{name}(#{mapped_params.map(&:name)})"
+
+          method = find_jls(mapped_type, name, mapped_params, meta, constructor)
+          
+          unless method
+            log "Failed to locate method #{mapped_type.name}.#{name}(#{mapped_params.map(&:name)})"
+            return nil
+          end
+        end
+
+        log "Found method #{method.declaring_class.name}.#{name}(#{method.argument_types.map(&:name)}) from #{mapped_type.name}"
+        return method
+      end
+      
+      def find_jls(mapped_type, name, mapped_params, meta, constructor)
+        # mapped_type = jvm_type(mapped_type)
+        # mapped_params = convert_params(mapped_params)
+        if constructor
+          all_methods = mapped_type.unmeta.declared_constructors
+          by_name = all_methods
+        elsif meta
+          all_methods = mapped_type.declared_class_methods
+        else
+          all_methods = []
+          cls = mapped_type
+          while cls
+            all_methods += cls.declared_instance_methods
+            cls = cls.superclass
+          end
+        end
+        # filter by name, if we haven't already got a by_name list
+        by_name ||= all_methods.select {|m| m.name == name && mapped_params.size <= m.argument_types.size}
+        # filter by arity
+        by_name_and_arity = by_name.select {|m| m.argument_types.size == mapped_params.size}
+
+        phase1_methods = phase1(mapped_params, by_name_and_arity)
+
+        if phase1_methods.size > 1
+          raise "Ambiguous targets invoking #{mapped_type}.#{name}:\n#{phase1_methods}"
+        end
+
+        phase1_methods[0] ||
+          phase2(mapped_params, by_name) ||
+          phase3(mapped_params, by_name)
+      end
+        
+      def phase1(mapped_params, potentials)
+        log "Beginning JLS phase 1 search with params (#{mapped_params.map(&:name)})"
+        
+        # cycle through methods looking for more specific matches; gather matches of equal specificity
+        methods = potentials.inject([]) do |currents, potential|
+          method_params = potential.argument_types
+          
+          # exact match always wins; duplicates not possible
+          if each_is_exact(mapped_params, method_params)
+            return [potential]
+          end
+          
+          # otherwise, check for potential match and compare to current
+          # TODO: missing ambiguity check; picks last method of equal specificity
+          if each_is_exact_or_subtype_or_convertible(mapped_params, method_params)
+            if currents.size > 0
+              if is_more_specific?(potential.argument_types, currents[0].argument_types)
+                # potential is better, dump all currents
+                currents = [potential]
+              elsif is_more_specific?(currents[0].argument_types, potential.argument_types)
+                # currents are better, try next potential
+                #next
+              else
+                # equal specificity, append to currents
+                currents << potential
+              end
+            else
+              # no previous matches, use potential
+              currents = [potential]
+            end
+          end
+          
+          currents
+        end
+
+        methods
+      end
+      
+      def is_more_specific?(potential, current)
+        each_is_exact_or_subtype_or_convertible(potential, current)
+      end
+      
+      def phase2(mapped_params, potentials)
+        nil
+      end
+      
+      def phase3(mapped_params, potentials)
+        nil
+      end
+      
+      def each_is_exact(incoming, target)
+        incoming.each_with_index do |in_type, i|
+          target_type = target[i]
+          
+          # exact match
+          return false unless target_type == in_type
+        end
+        return true
+      end
+      
+      def each_is_exact_or_subtype_or_convertible(incoming, target)
+        incoming.each_with_index do |in_type, i|
+          target_type = target[i]
+          
+          # exact match
+          next if target_type == in_type
+          
+          # primitive is safely convertible
+          if target_type.primitive?
+            if in_type.primitive?
+              next if primitive_convertible? in_type, target_type
+            end
+            return false
+          end
+          
+          # object type is assignable
+          return false unless target_type.assignable_from? in_type
+        end
+        return true
+      end
+      
+      BOOLEAN = Java::boolean.java_class
+      BYTE = Java::byte.java_class
+      SHORT = Java::short.java_class
+      CHAR = Java::char.java_class
+      INT = Java::int.java_class
+      LONG = Java::long.java_class
+      FLOAT = Java::float.java_class
+      DOUBLE = Java::double.java_class
+      
+      PrimitiveConversions = {
+        BOOLEAN => [BOOLEAN],
+        BYTE => [BYTE, SHORT, CHAR, INT, LONG, FLOAT, DOUBLE],
+        SHORT => [SHORT, INT, LONG, FLOAT, DOUBLE],
+        CHAR => [CHAR, INT, LONG, FLOAT, DOUBLE],
+        INT => [INT, LONG, FLOAT, DOUBLE],
+        LONG => [LONG, DOUBLE],
+        FLOAT => [FLOAT, DOUBLE],
+        DOUBLE => [DOUBLE]
+      }
+      
+      def primitive_convertible?(in_type, target_type)
+        if PrimitiveConversions.include? in_type
+          PrimitiveConversions[in_type].include?(target_type)
+        else
+          in_type.convertible_to?(target_type)
+        end
+      end
+    end
+  end
+end

data/lib/duby/jvm/source_compiler.rb

@@ -0,0 +1,605 @@
+require 'duby'
+require 'duby/ast'
+require 'duby/jvm/types'
+require 'duby/jvm/compiler'
+require 'duby/jvm/source_generator/builder'
+require 'duby/jvm/source_generator/precompile'
+require 'duby/jvm/source_generator/loops'
+
+class String
+  def compile(compiler, expression)
+    compiler.method.print self if expression
+  end
+end
+
+module Duby
+  module Compiler
+    class JavaSource
+      JVMTypes = Duby::JVM::Types
+      include Duby::Compiler::JVM::JVMLogger
+      attr_accessor :filename, :method, :static, :class, :lvalue
+
+      Operators = [
+        '+', '-', '+@', '-@', '/', '%', '*', '<',
+        '<=', '==', '!=', '>=', '>',
+        '<<', '>>', '>>>', '|', '&', '^', '~'
+      ]
+      ArrayOps = [
+        '[]', '[]=', 'length'
+      ]
+      
+      ImplicitReturn = Struct.new(:value)
+
+      def initialize(filename)
+        @filename = File.basename(filename)
+        @static = true
+        parts = filename.split '/'
+        classname = parts.pop.sub /[.].+/, ''
+        package = parts.join('.') unless parts.empty?
+
+        @file = Duby::JavaSource::Builder.new(filename, self)
+        #@file.package = package
+        @type = AST::type(classname)
+      end
+
+      def toplevel_class
+        @class = @type.define(@file)
+      end
+
+      def generate(&block)
+        log "Generating source files..."
+        @file.generate do |filename, builder|
+          log "  #{builder.class_name}"
+          if block_given?
+            yield filename, builder
+          else
+            File.open(filename, 'w') {|f| f.write(builder.generate)}
+          end
+        end
+        log "...done!"
+      end
+
+      def define_main(body)
+        if body.class != AST::ClassDefinition
+          @class = @type.define(@file)
+          with :method => @class.main do
+            log "Starting main method"
+
+            @method.start
+
+            body.compile(self, false)
+
+            @method.stop
+          end
+          @class.stop
+        else
+          body.compile(self, false)
+        end
+
+        log "Main method complete!"
+      end
+
+      def define_method(node)
+        name, signature, args = node.name, node.signature, node.arguments.args
+        args ||= []
+        return_type = signature[:return]
+        exceptions = signature[:throws] || []
+        with :static => @static || node.static? do
+          if @static
+            method = @class.public_static_method(name.to_s, return_type, exceptions, *args)
+          else
+            method = @class.public_method(name.to_s, return_type, exceptions, *args)
+          end
+
+          with :method => method do
+            log "Starting new method #{name}"
+            @method.annotate(node.annotations)
+            @method.start
+
+            unless @method.type.nil? || @method.type.void?
+              self.return(ImplicitReturn.new(node.body))
+            else
+              node.body.compile(self, false) if node.body
+            end
+        
+            log "Method #{name} complete!"
+            @method.stop
+          end
+        end
+      end
+
+      def constructor(node)
+        args = node.arguments.args || []
+        exceptions = node.signature[:throws] || []
+        method = @class.public_constructor(exceptions, *args)
+        with :method => method do
+          @method.annotate(node.annotations)
+          @method.start
+          if node.delegate_args
+            delegate = if node.calls_super
+              "super"
+            else
+              "this"
+            end
+            method.print "#{delegate}("
+            node.delegate_args.each_with_index do |arg, index|
+              method.print ', ' unless index == 0
+              raise "Invalid constructor argument #{arg}" unless arg.expr?(self)
+              arg.compile(self, true)
+            end
+            method.puts ");"
+          end
+          
+          node.body.compile(self, false) if node.body
+          method.stop
+        end
+      end
+
+      def return(node)
+        if @method.type.nil? || @method.type.void?
+          @method.puts 'return;'
+          return
+        end
+        if node.value.expr?(self)
+          @method.print 'return '
+          node.value.compile(self, true)
+          @method.puts ';'
+        else
+          store_value('return ', node.value)
+        end
+      end
+
+      def _raise(node)
+        if node.expr?(self)
+          @method.print 'throw '
+          node.compile(self, true)
+          @method.puts ';'
+        else
+          store_value('throw ', node)
+        end
+      end
+      
+      def rescue(node, expression)
+        @method.block 'try' do
+          maybe_store(node.body, expression)
+        end
+        node.clauses.each do |clause|
+          clause.types.each do |type|
+            name = clause.name || 'tmp$ex'
+            @method.block "catch (#{type.to_source} #{name})" do
+              maybe_store(clause.body, expression)
+            end
+          end
+        end
+      end
+      
+      def ensure(node, expression)
+        @method.block 'try' do
+          maybe_store(node.body, expression)
+        end
+        @method.block 'finally' do
+          node.clause.compile(self, false)
+        end
+      end
+
+      def line(num)
+      end
+      
+      def declare_local(name, type)
+        @method.declare_local(type, name)
+      end
+      
+      def declare_field(name, type, annotations)
+        @class.declare_field(name, type, @static, annotations)
+      end
+
+      def local(name, type)
+        @method.print name
+      end
+      
+      def field(name, type, annotations)
+        name = name[1..-1]
+        declare_field(name, type, annotations)
+        @method.print "#{this}.#{name}"
+      end
+
+      def this
+        @static ? @class.class_name : 'this'
+      end
+
+      def local_assign(name, type, expression, value)
+        value = value.precompile(self)
+        if method.local?(name)
+          @method.print @lvalue if expression
+          @method.print "#{name} = "
+          value.compile(self, true)
+          @method.puts ';'
+        else
+          @method.declare_local(type, name) do
+            value.compile(self, true)
+          end
+          if expression
+            @method.puts "#{@lvalue}#{name};"
+          end
+        end
+      end
+
+      def field_declare(name, type, annotations)
+        name = name[1..-1]
+        declare_field(name, type, annotations)
+      end
+      
+      def local_declare(name, type)
+        declare_local(name, type)
+      end
+      
+      def field_assign(name, type, expression, value, annotations)
+        name = name[1..-1]
+        declare_field(name, type, annotations)
+        lvalue = "#{@lvalue if expression}#{this}.#{name} = "
+        store_value(lvalue, value)
+      end
+      
+      def store_value(lvalue, value)
+        if value.is_a? String
+          @method.puts "#{lvalue}#{value};"
+        elsif value.expr?(self)
+          @method.print lvalue
+          value.compile(self, true)
+          @method.puts ';'
+        else
+          with :lvalue => lvalue do
+            value.compile(self, true)
+          end
+        end
+      end
+      
+      def assign(name, value)
+        store_value("#{name} = ", value)
+        name
+      end
+
+      def maybe_store(value, expression)
+        if expression
+          store_value(@lvalue, value)
+        else
+          value.compile(self, false)
+        end
+      end
+
+      def body(body, expression)
+        # all except the last element in a body of code is treated as a statement
+        i, last = 0, body.children.size - 1
+        while i < last
+          body.children[i].compile(self, false)
+          i += 1
+        end
+        # last element is an expression only if the body is an expression
+        maybe_store(body.children[last], expression)
+      end
+      
+      def branch_expression(node)
+        node.condition.compile(self, true)
+        @method.print ' ? ('
+        if node.body
+          node.body.compile(self, true)
+        else
+          @method.print @method.init_value(node.inferred_type)
+        end
+        @method.print ') : ('
+        if node.else
+          node.else.compile(self, true)
+        else
+          @method.print @method.init_value(node.inferred_type)
+        end
+        @method.print ')'
+      end
+      
+      def branch(node, expression)
+        if expression && node.expr?(self)
+          return branch_expression(node)
+        end
+        predicate = node.condition.predicate.precompile(self)
+        @method.print 'if ('
+        predicate.compile(self, true)
+        @method.block ")" do
+          if node.body
+            maybe_store(node.body, expression)
+          elsif expression
+            store_value(@lvalue, @method.init_value(node.inferred_type))
+          end
+        end
+        if node.else || expression
+          @method.block 'else' do
+            if node.else
+              maybe_store(node.else, expression)
+            else
+              store_value(@lvalue, @method.init_value(node.inferred_type))
+            end
+          end
+        end
+      end
+      
+      def loop(loop, expression)
+        if loop.redo? || loop.post || !loop.condition.predicate.expr?(self)
+          loop = ComplexWhileLoop.new(loop, self)
+        else
+          loop = SimpleWhileLoop.new(loop, self)
+        end
+        with(:loop => loop) do
+          loop.compile(expression)
+        end
+      end
+      
+      def expr?(target, params)
+        !([target] + params).any? {|x| x.kind_of? Duby::AST::TempValue}
+      end
+
+      def operator(target, op, params, expression)
+        simple = expr?(target, params)
+        if expression && !simple
+          @method.print @lvalue
+        end
+        if params.size == 0
+          # unary operator
+          op = op[0,1]
+          @method.print op
+          target.compile(self, true)
+        else
+          @method.print '('
+          other = params[0]
+          target.compile(self, true)
+          @method.print " #{op} "
+          other.compile(self, true)
+          @method.print ')'
+        end
+        unless expression && simple
+          @method.puts ';'
+        end
+      end
+
+      def compile_args(call)
+        call.parameters.map do |param|
+          param.precompile(self)
+        end
+      end
+
+      def self_type
+        type = AST::type(@class.name.tr('/', '.'))
+        type = type.meta if @static
+        type
+      end
+
+      def super_call(call, expression)
+        super_method_call(this, call, compile_args(call), expression)
+      end
+
+      def self_call(call, expression)
+        if call.cast?
+          args = compile_args(call)
+          simple = call.expr?(self)
+          @method.print @lvalue if expression && !simple
+          @method.print "(#{call.inferred_type.name})("
+          args.each{|arg| arg.compile(self, true)}
+          @method.print ")"
+          @method.puts ';' unless simple && expression
+        else
+          method_call(this, call, compile_args(call), expression)
+        end
+      end
+
+      def call(call, expression)
+        if Duby::AST::Constant === call.target
+          target = call.target.inferred_type.name
+        else
+          target = call.target.precompile(self)
+        end
+        params = compile_args(call)
+        
+        if Operators.include? call.name
+          operator(target, call.name, params, expression)
+        elsif call.target.inferred_type.array? && ArrayOps.include?(call.name)
+          array_op(target, call.name, params, expression)
+        elsif call.name == 'nil?'
+          operator(target, '==', ['null'], expression)
+        else
+          method_call(target, call, params, expression)
+        end
+      end
+      
+      def array_op(target, name, args, expression)
+        simple = expr?(target, args)
+        index, value = args
+        if expression && !simple
+          @method.print @lvalue
+        end
+        target.compile(self, true)
+        if name == 'length'
+          @method.print '.length'
+        else
+          @method.print '['
+          index.compile(self, true)
+          @method.print ']'
+          if name == '[]='
+            @method.print " = "
+            value.compile(self, true)
+          end
+        end
+        unless simple && expression
+          @method.puts ';'
+        end
+      end
+      
+      def break(node)
+        @loop.break
+      end
+      
+      def next(node)
+        @loop.next
+      end
+      
+      def redo(node)
+        @loop.redo
+      end
+
+      # TODO: merge cleanly with method_call logic
+      def super_method_call(target, call, params, expression)
+        simple = call.expr?(self)
+        method = call.method(self)
+        unless simple || method.actual_return_type.void?
+          @method.print @lvalue if expression
+        end
+        if method.constructor?
+          @method.print "super("
+        else
+          @method.print "super.#{method.name}("
+        end
+        params.each_with_index do |param, index|
+          @method.print ', ' unless index == 0
+          param.compile(self, true)
+        end
+        if simple && expression
+          @method.print ')'
+        else
+          @method.puts ');'
+        end
+        if method.actual_return_type.void? && expression
+          @method.print @lvalue
+          if method.static?
+            @method.puts 'null;'
+          else
+            target.compile(self, true)
+            @method.puts ';'
+          end
+        end
+
+      end
+      
+      def method_call(target, call, params, expression)
+        simple = call.expr?(self)
+        method = call.method(self)
+        unless simple || method.actual_return_type.void?
+          @method.print @lvalue if expression
+        end
+        if method.constructor?
+          @method.print "new "
+          target.compile(self, true)
+          @method.print '('
+        else
+          target.compile(self, true)
+          @method.print ".#{method.name}("
+        end
+        params.each_with_index do |param, index|
+          @method.print ', ' unless index == 0
+          param.compile(self, true)
+        end
+        if simple && expression
+          @method.print ')'
+        else
+          @method.puts ');'
+        end
+        if method.actual_return_type.void? && expression
+          @method.print @lvalue
+          if method.static?
+            @method.puts 'null;'
+          else
+            target.compile(self, true)
+            @method.puts ';'
+          end
+        end
+        
+      end
+
+      def temp(expression, value=nil)
+        assign(@method.tmp(expression.inferred_type), value || expression)
+      end
+
+      def empty_array(type, size)
+        sizevar = size.precompile(self)
+        @method.print "#{@lvalue unless size.expr?(self)}new #{type.name}["
+        sizevar.compile(self, true)
+        @method.print ']'
+      end
+
+      def import(short, long)
+      end
+
+      def string(value)
+        @method.print value.inspect
+      end
+      
+      def boolean(value)
+        @method.print value ? 'true' : 'false'
+      end
+
+      def array(node, expression)
+        if expression
+          # create unmodifiable list from array (simplest way to do this in Java source)
+          @method.print "java.util.Collections.unmodifiableList(java.util.Arrays.asList("
+
+          # elements, as expressions
+          boolean comma = false
+          node.children.each do |node|
+            @method.print ", "# if comma
+            node.compile(self, true)
+            comma = true
+          end
+          
+          @method.print("))")
+        else
+          # elements, as non-expressions
+          # TODO: ensure they're all reference types!
+          node.children.each do |node|
+            node.compile(self, false)
+          end
+        end
+      end
+      
+      def null
+        @method.print 'null'
+      end
+      
+      def compile_self
+        @method.print 'this'
+      end
+      
+      def print(node)
+        value = node.parameters[0]
+        value = value && value.precompile(self)
+        if node.println
+          @method.print "System.out.println("
+        else
+          @method.print "System.out.print("
+        end
+        value.compile(self, true) if value
+        @method.puts ');'
+      end
+
+      def define_class(class_def, expression)
+        with(:class => class_def.inferred_type.define(@file),
+             :static => false) do
+          @class.annotate(class_def.annotations)
+          class_def.body.compile(self, false) if class_def.body
+        
+          @class.stop
+        end
+      end
+
+      def with(vars)
+        orig_values = {}
+        begin
+          vars.each do |name, new_value|
+            name = "@#{name}"
+            orig_values[name] = instance_variable_get name
+            instance_variable_set name, new_value
+          end
+          yield
+        ensure
+          orig_values.each do |name, value|
+            instance_variable_set name, value
+          end
+        end
+      end
+    end
+  end
+end