Make 'self' a standard parameter (ParamVariable)
[nit.git] / src / syntax / typing.nit
index cae16f3..5d01a75 100644 (file)
@@ -44,8 +44,8 @@ special AbsSyntaxVisitor
        # Current knowledge about variables names and types
        readable writable attr _variable_ctx: VariableContext
 
-       # Type of the receiver
-       readable writable attr _self_type: MMType
+       # The current reciever
+       readable writable attr _self_var: ParamVariable
 
        # Block of the current method
        readable writable attr _top_block: PExpr
@@ -58,7 +58,7 @@ special AbsSyntaxVisitor
 
        init(tc, module) do super
 
-       private meth get_default_constructor_for(n: PNode, c: MMLocalClass, prop: MMMethod): MMMethod
+       private meth get_default_constructor_for(n: PNode, c: MMLocalClass, prop: MMSrcMethod): MMMethod
        do
                var v = self
                #var prop = v.local_property
@@ -70,15 +70,15 @@ special AbsSyntaxVisitor
                        if not g.is_init then continue
                        if g.intro.local_class != c then continue
                        var gp = c[g]
-                       assert gp isa MMMethod
+                       assert gp isa MMSrcMethod
                        var garity = gp.signature.arity
-                       if prop != null and g.intro.name == prop.name then
-                               if garity == 0 or prop.signature < gp.signature then
+                       if prop != null and gp.name == prop.name then
+                               if garity == 0 or (parity == garity and prop.signature < gp.signature) then
                                        return gp
                                else
                                        false_candidates.add(gp)
                                end
-                       else if garity == 0 then
+                       else if garity == 0 and gp.name == once ("init".to_symbol) then
                                candidates.add(gp)
                                false_candidates.add(gp)
                        else
@@ -91,7 +91,7 @@ special AbsSyntaxVisitor
                        v.error(n, "Error: Conflicting default constructor to call for {c}: {candidates.join(", ")}.")
                        return null
                else if false_candidates.length > 0 then
-                       v.error(n, "Error: there is no available compatible constrctor in {c}.")
+                       v.error(n, "Error: there is no available compatible constrctor in {c}. discarded candidates are {false_candidates.join(", ")}.")
                        return null
                else
                        v.warning(n, "Error: there is no available compatible constrctor in {c}.")
@@ -132,15 +132,15 @@ private class VariableContext
        attr _dico: Map[Symbol, Variable]
 
        # Build a new VariableContext
-       meth sub: VariableContext
+       meth sub: SubVariableContext
        do
-               return new SubVariableContext.with(self, null, null)
+               return new SubVariableContext.with_prev(self, null, null)
        end
 
        # Build a nested VariableContext with new variable information
-       meth sub_with(v: Variable, t: MMType): VariableContext
+       meth sub_with(v: Variable, t: MMType): SubVariableContext
        do
-               return new SubVariableContext.with(self, v, t)
+               return new SubVariableContext.with_prev(self, v, t)
        end
 
        init
@@ -172,7 +172,7 @@ special VariableContext
                return prev.stype(v)
        end
 
-       init with(p: VariableContext, v: Variable, t: MMType)
+       init with_prev(p: VariableContext, v: Variable, t: MMType)
        do
                init
                _prev = p
@@ -196,7 +196,8 @@ end
 redef class PClassdef
        redef meth accept_typing(v)
        do
-               v.self_type = local_class.get_type
+               v.self_var = new ParamVariable("self".to_symbol, self)
+               v.self_var.stype = local_class.get_type
                super
        end
 end
@@ -206,15 +207,17 @@ redef class AAttrPropdef
        do
                super
                if n_expr != null then
-                       v.check_conform(n_expr, n_expr.stype, prop.signature.return_type)
+                       v.check_conform_expr(n_expr, prop.signature.return_type)
                end
        end
 end
 
 redef class AMethPropdef
+       redef readable attr _self_var: ParamVariable
        redef meth accept_typing(v)
        do
                v.variable_ctx = new VariableContext
+               _self_var = v.self_var
                super
        end
 end
@@ -233,8 +236,8 @@ redef class AConcreteInitPropdef
                else 
                        var i = 0
                        var l = explicit_super_init_calls.length
-                       var cur_m: MMMethod
-                       var cur_c: MMLocalClass
+                       var cur_m: MMMethod = null
+                       var cur_c: MMLocalClass = null
                        if i < l then
                                cur_m = explicit_super_init_calls[i]
                                cur_c = cur_m.global.intro.local_class
@@ -293,29 +296,27 @@ redef class PExpr
        # Is the expression the current receiver (implicit or explicit)
        meth is_self: Bool do return false
 
-       # Is the expression a variable access
-       meth is_variable: Bool do return false
+       # The variable accessed is any
+       meth its_variable: Variable do return null
 
        # The variable type information if current boolean expression is true
        readable private attr _if_true_variable_ctx: VariableContext
 end
 
 redef class AVardeclExpr
-       # Assiociated local variable
-        readable attr _variable: Variable
-
        redef meth after_typing(v)
        do
-               var va = new Variable(n_id.to_symbol, self)
-               _variable = va
+               var va = new VarVariable(n_id.to_symbol, self)
+               variable = va
                v.variable_ctx.add(va)
 
                if n_type != null then
                        va.stype = n_type.stype
                        if n_expr != null then
-                               v.check_conform(self, n_expr.stype, va.stype)
+                               v.check_conform_expr(n_expr, va.stype)
                        end
                else
+                       v.check_expr(n_expr)
                        va.stype = n_expr.stype
                end
        end
@@ -342,7 +343,7 @@ redef class AReturnExpr
                else if n_expr != null and t == null then
                        v.error(self, "Error: Return with value in a procedure.")
                else if n_expr != null and t != null then
-                       v.check_conform(self, n_expr.stype, t)
+                       v.check_conform_expr(n_expr, t)
                end
        end
 end
@@ -352,7 +353,7 @@ redef class AIfExpr
        do
                var old_var_ctx = v.variable_ctx
                v.visit(n_expr)
-               v.check_conform(self, n_expr.stype, v.type_bool)
+               v.check_conform_expr(n_expr, v.type_bool)
 
                if n_expr.if_true_variable_ctx != null then
                        v.variable_ctx = n_expr.if_true_variable_ctx
@@ -372,7 +373,7 @@ end
 redef class AWhileExpr
        redef meth after_typing(v)
        do
-               v.check_conform(self, n_expr.stype, v.type_bool)
+               v.check_conform_expr(n_expr, v.type_bool)
        end
 end
 
@@ -387,52 +388,44 @@ redef class AForExpr
 end
 
 redef class AForVardeclExpr
-       # Associated automatic local variable
-       readable attr _variable: Variable
-
        redef meth after_typing(v)
        do
                v.variable_ctx = v.variable_ctx.sub
-               var variable = new Variable(n_id.to_symbol, self)
-               _variable = variable
-               v.variable_ctx.add(variable)
+               var va = new AutoVariable(n_id.to_symbol, self)
+               variable = va
+               v.variable_ctx.add(va)
 
                var expr_type = n_expr.stype
-               if not v.check_conform(self, expr_type, v.type_collection) then
+               if not v.check_conform_expr(n_expr, v.type_collection) then
                        return
                end
-               var prop = expr_type.select_method(once ("iterator".to_symbol))
+               var prop = expr_type.local_class.select_method(once ("iterator".to_symbol))
                if prop == null then
                        v.error(self, "Error: Collection MUST have an iterate method")
                        return
                end
-               var iter_type = prop.signature.return_type
-               var prop2 = iter_type.select_method(once ("item".to_symbol))
+               var iter_type = prop.signature_for(expr_type).return_type
+               var prop2 = iter_type.local_class.select_method(once ("item".to_symbol))
                if prop2 == null then
                        v.error(self, "Error: {iter_type} MUST have an item method")
                        return
                end
-               var t = prop2.signature.return_type
+               var t = prop2.signature_for(iter_type).return_type
                if not n_expr.is_self then t = t.not_for_self
-               variable.stype = t
+               va.stype = t
        end
 end
 
 redef class AAssertExpr
        redef meth after_typing(v)
        do
-               v.check_conform(self, n_expr.stype, v.type_bool)
+               v.check_conform_expr(n_expr, v.type_bool)
                if n_expr.if_true_variable_ctx != null then v.variable_ctx = n_expr.if_true_variable_ctx
        end
 end
 
-redef class AVarFormExpr
-       # Associated local variable
-        readable writable attr _variable: Variable 
-end
-
 redef class AVarExpr
-       redef meth is_variable do return true
+       redef meth its_variable do return variable
 
        redef meth after_typing(v)
        do
@@ -444,7 +437,7 @@ redef class AVarAssignExpr
        redef meth after_typing(v)
        do
                var t = v.variable_ctx.stype(variable)
-               v.check_conform(self, n_value.stype, t)
+               v.check_conform_expr(n_value, t)
        end
 end
 
@@ -456,15 +449,15 @@ redef class AReassignFormExpr
                        return
                end
                var name = n_assign_op.method_name
-               var prop = type_lvalue.select_method(name)
+               var prop = type_lvalue.local_class.select_method(name)
                if prop == null then
                        v.error(self, "Error: Method '{name}' doesn't exists in {type_lvalue}.")
                        return
                end
                prop.global.check_visibility(v, self, v.module, false)
-               var psig = prop.signature
+               var psig = prop.signature_for(type_lvalue)
                _assign_method = prop
-               v.check_conform(self, n_value.stype, psig[0].not_for_self)
+               v.check_conform_expr(n_value, psig[0].not_for_self)
                v.check_conform(self, psig.return_type.not_for_self, n_value.stype)
        end
 
@@ -472,6 +465,14 @@ redef class AReassignFormExpr
        readable attr _assign_method: MMMethod
 end
 
+redef class AVarReassignExpr
+       redef meth after_typing(v)
+       do
+               var t = v.variable_ctx.stype(variable)
+               do_lvalue_typing(v, t)
+       end
+end
+
 redef class PAssignOp
        meth method_name: Symbol is abstract
 end
@@ -482,19 +483,13 @@ redef class AMinusAssignOp
        redef meth method_name do return once "-".to_symbol
 end
 
-redef class AVarReassignExpr
-       redef meth after_typing(v)
-       do
-               var t = v.variable_ctx.stype(variable)
-               do_lvalue_typing(v, t)
-       end
-end
-
 redef class ASelfExpr
+       redef meth its_variable do return variable
+
        redef meth after_typing(v)
        do
-               assert v.self_type != null
-               _stype = v.self_type
+               variable = v.self_var
+               _stype = v.variable_ctx.stype(variable)
        end
 
         redef meth is_self do return true
@@ -515,7 +510,9 @@ redef class AIfexprExpr
                v.variable_ctx = old_var_ctx
                v.visit(n_else)
 
-               v.check_conform(self, n_expr.stype, v.type_bool)
+               v.check_conform_expr(n_expr, v.type_bool)
+
+               if not v.check_expr(n_then) or not v.check_expr(n_else) then return
 
                var t = n_then.stype
                var te = n_else.stype
@@ -540,8 +537,8 @@ end
 redef class AOrExpr
        redef meth after_typing(v)
        do
-               v.check_conform(self, n_expr.stype, v.type_bool)
-               v.check_conform(self, n_expr2.stype, v.type_bool)
+               v.check_conform_expr(n_expr, v.type_bool)
+               v.check_conform_expr(n_expr2, v.type_bool)
                _stype = v.type_bool
        end
 end
@@ -563,8 +560,8 @@ redef class AAndExpr
 
                v.variable_ctx = old_var_ctx
 
-               v.check_conform(self, n_expr.stype, v.type_bool)
-               v.check_conform(self, n_expr2.stype, v.type_bool)
+               v.check_conform_expr(n_expr, v.type_bool)
+               v.check_conform_expr(n_expr2, v.type_bool)
                _stype = v.type_bool
        end
 end
@@ -572,7 +569,7 @@ end
 redef class ANotExpr
        redef meth after_typing(v)
        do
-               v.check_conform(self, n_expr.stype, v.type_bool)
+               v.check_conform_expr(n_expr, v.type_bool)
                _stype = v.type_bool
        end
 end
@@ -625,7 +622,7 @@ redef class AArrayExpr
 
        redef meth after_typing(v)
        do
-               var stype: MMType
+               var stype: MMType = null
                for n in n_exprs do
                        var ntype = n.stype
                        if stype == null or (ntype != null and stype < ntype) then
@@ -633,7 +630,7 @@ redef class AArrayExpr
                        end
                end
                for n in n_exprs do
-                       v.check_conform(self, n.stype, stype)
+                       v.check_conform_expr(n, stype)
                end
                _stype = v.type_array(stype)
        end
@@ -654,7 +651,8 @@ redef class ARangeExpr
                        return
                end
                var dtype = v.type_discrete
-               v.check_conform(self, ntype, dtype)
+               v.check_conform_expr(n_expr, dtype)
+               v.check_conform_expr(n_expr2, dtype)
                _stype = v.type_range(ntype)
        end
 end
@@ -665,7 +663,7 @@ special ASuperInitCall
        readable attr _init_in_superclass: MMMethod
        redef meth after_typing(v)
        do
-               var precs: Array[MMLocalProperty] = v.local_property.cprhe.direct_greaters
+               var precs: Array[MMLocalProperty] = v.local_property.prhe.direct_greaters
                if not precs.is_empty then
                        v.local_property.need_super = true
                else if v.local_property.global.is_init then
@@ -674,7 +672,7 @@ special ASuperInitCall
                                if not p.global.is_init then
                                        v.error(self, "Error: {p.local_class}::{p} is not a constructor.")
                                else
-                                       precs.add(v.self_type.select_property(p.global))
+                                       precs.add(v.local_class[p.global])
                                end
                        end
                        if precs.is_empty then
@@ -689,19 +687,20 @@ special ASuperInitCall
                        _init_in_superclass = p
                        register_super_init_call(v, p)
                        if n_args.length > 0 then
-                               _arguments = process_signature(v, p, true, n_args.to_a)
+                               var signature = get_signature(v, v.self_var.stype, p, true)
+                               _arguments = process_signature(v, signature, p, n_args.to_a)
                        end
                else
                        v.error(self, "Error: No super method to call for {v.local_property}.")
                        return
                end
 
-               if precs.first.signature.return_type != null then
+               if precs.first.signature_for(v.self_var.stype).return_type != null then
                        var stypes = new Array[MMType]
-                       var stype: MMType
+                       var stype: MMType = null
                        for prop in precs do
                                assert prop isa MMMethod
-                               var t = prop.signature.return_type.for_module(v.module).adapt_to(v.local_property.signature.recv)
+                               var t = prop.signature_for(v.self_var.stype).return_type.for_module(v.module).adapt_to(v.local_property.signature.recv)
                                stypes.add(t)
                                if stype == null or stype < t then
                                        stype = t
@@ -722,15 +721,16 @@ redef class AAttrFormExpr
        # Attribute accessed
        readable attr _prop: MMAttribute
 
+       # Attribute type of the acceded attribute
+       readable attr _attr_type: MMType
+
        # Compute the attribute accessed
        private meth do_typing(v: TypingVisitor)
        do
+               if not v.check_expr(n_expr) then return
                var type_recv = n_expr.stype
-               if type_recv == null then
-                       return
-               end
                var name = n_id.to_symbol
-               var prop = type_recv.select_attribute(name)
+               var prop = type_recv.local_class.select_attribute(name)
                if prop == null then
                        v.error(self, "Error: Attribute {name} doesn't exists in {type_recv}.")
                        return
@@ -738,6 +738,9 @@ redef class AAttrFormExpr
                        v.error(self, "Error: Attribute {name} from {prop.global.local_class.module} is invisible in {v.module}")
                end
                _prop = prop
+               var at = prop.signature_for(type_recv).return_type 
+               if not n_expr.is_self then at = at.not_for_self
+               _attr_type = at
        end
 end
 
@@ -748,8 +751,6 @@ redef class AAttrExpr
                if prop == null then
                        return
                end
-               var attr_type = prop.signature.return_type
-               if not n_expr.is_self then attr_type = attr_type.not_for_self
                _stype = attr_type
        end
 end
@@ -761,9 +762,7 @@ redef class AAttrAssignExpr
                if prop == null then
                        return
                end
-               var attr_type = prop.signature.return_type
-               if not n_expr.is_self then attr_type = attr_type.not_for_self
-               v.check_conform(self, n_value.stype, attr_type)
+               v.check_conform_expr(n_value, attr_type)
        end
 end
 
@@ -774,8 +773,6 @@ redef class AAttrReassignExpr
                if prop == null then
                        return
                end
-               var attr_type = prop.signature.return_type
-               if not n_expr.is_self then attr_type = attr_type.not_for_self
                do_lvalue_typing(v, attr_type)
        end
 end
@@ -787,7 +784,9 @@ special PExpr
        do
                var prop = get_property(v, type_recv, is_implicit_self, name)
                if prop == null then return
-               var args = process_signature(v, prop, recv_is_self, raw_args)
+               var sig = get_signature(v, type_recv, prop, recv_is_self)
+               if sig == null then return
+               var args = process_signature(v, sig, prop, raw_args)
                if args == null then return
                _prop = prop
                _arguments = args
@@ -796,14 +795,14 @@ special PExpr
        private meth get_property(v: TypingVisitor, type_recv: MMType, is_implicit_self: Bool, name: Symbol): MMMethod
        do
                if type_recv == null then return null
-               var prop = type_recv.select_method(name)
+               var prop = type_recv.local_class.select_method(name)
                if prop == null and v.local_property.global.is_init then
                        var props = type_recv.local_class.super_methods_named(name)
                        if props.length > 1 then
                                v.error(self, "Error: Ambigous method name '{name}' for {props.join(", ")}. Use explicit designation.")
                                return null
                        else if props.length == 1 then 
-                               var p = type_recv.select_property(props.first.global)
+                               var p = type_recv.local_class[props.first.global]
                                assert p isa MMMethod
                                prop = p
                        end
@@ -820,10 +819,17 @@ special PExpr
                return prop
        end
 
-       private meth process_signature(v: TypingVisitor, prop: MMMethod, recv_is_self: Bool, raw_args: Array[PExpr]): Array[PExpr]
+       private meth get_signature(v: TypingVisitor, type_recv: MMType, prop: MMMethod, recv_is_self: Bool): MMSignature
        do
                prop.global.check_visibility(v, self, v.module, recv_is_self)
-               var psig = prop.signature
+               var psig = prop.signature_for(type_recv)
+               if not recv_is_self then psig = psig.not_for_self
+               return psig
+       end
+       
+       # Check the conformity of a set of arguments `raw_args' to a signature.
+       private meth process_signature(v: TypingVisitor, psig: MMSignature, prop: MMMethod, raw_args: Array[PExpr]): Array[PExpr]
+       do
                var par_vararg = psig.vararg_rank
                var par_arity = psig.arity
                var raw_arity: Int
@@ -837,12 +843,11 @@ special PExpr
                for par_idx in [0..par_arity[ do
                        var a: PExpr
                        var par_type = psig[par_idx]
-                       if not recv_is_self then par_type = par_type.not_for_self
                        if par_idx == par_vararg then
                                var star = new Array[PExpr]
                                for i in [0..(raw_arity-par_arity)] do
                                        a = raw_args[arg_idx]
-                                       v.check_conform(self, a.stype, par_type)
+                                       v.check_conform_expr(a, par_type)
                                        star.add(a)
                                        arg_idx = arg_idx + 1
                                end
@@ -851,7 +856,7 @@ special PExpr
                                a = aa
                        else
                                a = raw_args[arg_idx]
-                               v.check_conform(self, a.stype, par_type)
+                               v.check_conform_expr(a, par_type)
                                arg_idx = arg_idx + 1
                        end
                        args.add(a)
@@ -876,7 +881,7 @@ special AAbsSendExpr
                        v.error(self, "Error: Constructor invocation {property} must not be in nested block.")
                end
                var cla = v.module[property.global.intro.local_class.global]
-               var prev_class: MMLocalClass
+               var prev_class: MMLocalClass = null
                if not v.explicit_super_init_calls.is_empty then
                        prev_class = v.explicit_super_init_calls.last.global.intro.local_class
                end
@@ -923,6 +928,8 @@ special AAbsSendExpr
                end
 
                do_typing(v, t, false, false, name, n_args.to_a)
+               if prop == null then return
+
                if not prop.global.is_init then
                        v.error(self, "Error: {prop} is not a constructor.")
                end
@@ -946,6 +953,7 @@ special ASuperInitCall
 
        private meth do_all_typing(v: TypingVisitor)
        do
+               if not v.check_expr(n_expr) then return
                do_typing(v, n_expr.stype, n_expr.is_implicit_self, n_expr.is_self, name, raw_arguments)
                if prop == null then return
                if prop.global.is_init then
@@ -957,7 +965,7 @@ special ASuperInitCall
                                register_super_init_call(v, prop)
                        end
                end
-               var t = prop.signature.return_type
+               var t = prop.signature_for(n_expr.stype).return_type
                if t != null and not n_expr.is_self then t = t.not_for_self
                _stype = t
        end
@@ -969,6 +977,7 @@ special AReassignFormExpr
        readable attr _read_prop: MMMethod
        redef meth do_all_typing(v)
        do
+               if not v.check_expr(n_expr) then return
                var raw_args = raw_arguments
                do_typing(v, n_expr.stype, n_expr.is_implicit_self, n_expr.is_self, name, raw_args)
                if prop == null then return
@@ -979,7 +988,7 @@ special AReassignFormExpr
                                v.error(self, "Error: constructor {prop} is not invoken on 'self'.")
                        end
                end
-               var t = prop.signature.return_type
+               var t = prop.signature_for(n_expr.stype).return_type
                if not n_expr.is_self then t = t.not_for_self
 
                do_lvalue_typing(v, t)
@@ -1134,15 +1143,22 @@ end
 redef class AIsaExpr
        redef meth after_typing(v)
        do
-               if n_expr.is_variable then
-                       var n = n_expr
-                       assert n isa AVarExpr
-                       _if_true_variable_ctx = v.variable_ctx.sub_with(n.variable, n_type.stype)
+               var variable = n_expr.its_variable
+               if variable != null then
+                       _if_true_variable_ctx = v.variable_ctx.sub_with(variable, n_type.stype)
                end
                _stype = v.type_bool
        end
 end
 
+redef class AAsCastExpr
+       redef meth after_typing(v)
+       do
+               v.check_expr(n_expr)
+               _stype = n_type.stype
+       end
+end
+
 redef class AProxyExpr
        redef meth after_typing(v)
        do