1 # This file is part of NIT ( http://www.nitlanguage.org ).
3 # Copyright 2012 Jean Privat <jean@pryen.org>
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
9 # http://www.apache.org/licenses/LICENSE-2.0
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.
17 # Intraprocedural resolution of static types and OO-services
18 # By OO-services we mean message sending, attribute access, instantiation, etc.
22 import modelize_property
26 redef class ToolContext
27 var typing_phase
: Phase = new TypingPhase(self, [flow_phase
, modelize_property_phase
, local_var_init_phase
])
31 # Does the MPropDef contains a call to super or a call of a super-constructor?
32 var has_supercall
: Bool = false
35 private class TypingPhase
37 redef fun process_npropdef
(npropdef
) do npropdef
.do_typing
(toolcontext
.modelbuilder
)
40 private class TypeVisitor
41 var modelbuilder
: ModelBuilder
42 var nclassdef
: AClassdef
43 var mpropdef
: MPropDef
45 var selfvariable
: Variable = new Variable("self")
47 init(modelbuilder
: ModelBuilder, nclassdef
: AClassdef, mpropdef
: MPropDef)
49 self.modelbuilder
= modelbuilder
50 self.nclassdef
= nclassdef
51 self.mpropdef
= mpropdef
53 var mclass
= nclassdef
.mclassdef
.mclass
55 var selfvariable
= new Variable("self")
56 self.selfvariable
= selfvariable
57 selfvariable
.declared_type
= mclass
.mclass_type
60 fun mmodule
: MModule do return self.nclassdef
.mclassdef
.mmodule
62 fun anchor
: MClassType do return self.nclassdef
.mclassdef
.bound_mtype
64 fun anchor_to
(mtype
: MType): MType
66 var mmodule
= self.nclassdef
.mclassdef
.mmodule
67 var anchor
= self.nclassdef
.mclassdef
.bound_mtype
68 return mtype
.anchor_to
(mmodule
, anchor
)
71 fun is_subtype
(sub
, sup
: MType): Bool
73 var mmodule
= self.nclassdef
.mclassdef
.mmodule
74 var anchor
= self.nclassdef
.mclassdef
.bound_mtype
75 return sub
.is_subtype
(mmodule
, anchor
, sup
)
78 fun resolve_for
(mtype
, subtype
: MType, for_self
: Bool): MType
80 var mmodule
= self.nclassdef
.mclassdef
.mmodule
81 var anchor
= self.nclassdef
.mclassdef
.bound_mtype
82 #print "resolve_for {mtype} sub={subtype} forself={for_self} mmodule={mmodule} anchor={anchor}"
83 var res
= mtype
.resolve_for
(subtype
, anchor
, mmodule
, not for_self
)
87 # Retrieve the signature of a `MMethodDef` resolved for a specific call.
88 # This method is an helper to symplify the query on the model.
90 # Note: `for_self` indicates if the reciever is self or not.
91 # If yes, virtual types are not resolved.
92 fun resolve_signature_for
(mmethoddef
: MMethodDef, recv
: MType, for_self
: Bool): MSignature
94 return self.resolve_for
(mmethoddef
.msignature
.as(not null), recv
, for_self
).as(MSignature)
97 # Check that `sub` is a subtype of `sup`.
98 # If `sub` is not a valud suptype, then display an error on `node` an return null.
99 # If `sub` is a safe subtype of `sup` then return `sub`.
100 # If `sub` is an insafe subtype (ie an imlicit cast is required), then return `sup`.
102 # The point of the return type is to determinate the usable type on an expression:
103 # If the suptype is safe, then the return type is the one on the expression typed by `sub`.
104 # Is the subtype is unsafe, then the return type is the one of an implicit cast on `sup`.
105 fun check_subtype
(node
: ANode, sub
, sup
: MType): nullable MType
107 if self.is_subtype
(sub
, sup
) then return sub
108 if self.is_subtype
(sub
, self.anchor_to
(sup
)) then
109 # FIXME workarround to the current unsafe typing policy. To remove once fixed virtual types exists.
110 #node.debug("Unsafe typing: expected {sup}, got {sub}")
113 self.modelbuilder
.error
(node
, "Type error: expected {sup}, got {sub}")
117 # Visit an expression and do not care about the return value
118 fun visit_stmt
(nexpr
: nullable AExpr)
120 if nexpr
== null then return
121 nexpr
.accept_typing
(self)
124 # Visit an expression and expects that it is not a statement
125 # Return the type of the expression
126 # Display an error and return null if:
127 # * the type cannot be determined or
128 # * `nexpr` is a statement
129 fun visit_expr
(nexpr
: AExpr): nullable MType
131 nexpr
.accept_typing
(self)
132 var mtype
= nexpr
.mtype
133 if mtype
!= null then return mtype
134 if not nexpr
.is_typed
then
135 if not self.modelbuilder
.toolcontext
.error_count
> 0 then # check that there is really an error
136 if self.modelbuilder
.toolcontext
.verbose_level
> 1 then
137 nexpr
.debug
("No return type but no error.")
140 return null # forward error
142 self.error
(nexpr
, "Type error: expected expression.")
146 # Visit an expression and expect its static type is a least a `sup`
147 # Return the type of the expression or null if
148 # * the type cannot be determined or
149 # * `nexpr` is a statement or
150 # * `nexpr` is not a `sup`
151 fun visit_expr_subtype
(nexpr
: AExpr, sup
: nullable MType): nullable MType
153 var sub
= visit_expr
(nexpr
)
154 if sub
== null then return null # Forward error
156 if sup
== null then return null # Forward error
158 var res
= check_subtype
(nexpr
, sub
, sup
)
160 nexpr
.implicit_cast_to
= res
165 # Visit an expression and expect its static type is a `Bool`
166 # Return the type of the expression or null if
167 # * the type cannot be determined or
168 # * `nexpr` is a statement or
169 # * `nexpr` is not a `Bool`
170 fun visit_expr_bool
(nexpr
: AExpr): nullable MType
172 return self.visit_expr_subtype
(nexpr
, self.type_bool
(nexpr
))
176 private fun visit_expr_cast
(node
: ANode, nexpr
: AExpr, ntype
: AType): nullable MType
178 var sub
= visit_expr
(nexpr
)
179 if sub
== null then return null # Forward error
181 var sup
= self.resolve_mtype
(ntype
)
182 if sup
== null then return null # Forward error
184 var mmodule
= self.nclassdef
.mclassdef
.mmodule
185 var anchor
= self.nclassdef
.mclassdef
.bound_mtype
187 self.modelbuilder
.warning
(node
, "Warning: Expression is already a {sup}.")
188 else if self.is_subtype
(sub
, sup
) and not sup
.need_anchor
then
189 self.modelbuilder
.warning
(node
, "Warning: Expression is already a {sup} since it is a {sub}.")
194 fun try_get_mproperty_by_name2
(anode
: ANode, mtype
: MType, name
: String): nullable MProperty
196 return self.modelbuilder
.try_get_mproperty_by_name2
(anode
, self.nclassdef
.mclassdef
.mmodule
, mtype
, name
)
199 fun resolve_mtype
(node
: AType): nullable MType
201 return self.modelbuilder
.resolve_mtype
(self.nclassdef
, node
)
204 fun try_get_mclass
(node
: ANode, name
: String): nullable MClass
206 var mmodule
= self.nclassdef
.mclassdef
.mmodule
207 var mclass
= modelbuilder
.try_get_mclass_by_name
(node
, mmodule
, name
)
211 fun get_mclass
(node
: ANode, name
: String): nullable MClass
213 var mmodule
= self.nclassdef
.mclassdef
.mmodule
214 var mclass
= modelbuilder
.try_get_mclass_by_name
(node
, mmodule
, name
)
215 if mclass
== null then
216 self.modelbuilder
.error
(node
, "Type Error: missing primitive class `{name}'.")
221 fun type_bool
(node
: ANode): nullable MType
223 var mclass
= self.get_mclass
(node
, "Bool")
224 if mclass
== null then return null
225 return mclass
.mclass_type
228 fun get_method
(node
: ANode, recvtype
: MType, name
: String, recv_is_self
: Bool): nullable CallSite
230 var unsafe_type
= self.anchor_to
(recvtype
)
232 #debug("recv: {recvtype} (aka {unsafe_type})")
233 if recvtype
isa MNullType then
234 self.error
(node
, "Error: Method '{name}' call on 'null'.")
238 var mproperty
= self.try_get_mproperty_by_name2
(node
, unsafe_type
, name
)
239 if mproperty
== null then
240 #self.modelbuilder.error(node, "Type error: property {name} not found in {unsafe_type} (ie {recvtype})")
242 self.modelbuilder
.error
(node
, "Error: Method or variable '{name}' unknown in {recvtype}.")
244 self.modelbuilder
.error
(node
, "Error: Method '{name}' doesn't exists in {recvtype}.")
249 assert mproperty
isa MMethod
250 if mproperty
.visibility
== protected_visibility
and not recv_is_self
and self.mmodule
.visibility_for
(mproperty
.intro_mclassdef
.mmodule
) < intrude_visibility
then
251 self.modelbuilder
.error
(node
, "Error: Method '{name}' is protected and can only acceded by self. {mproperty.intro_mclassdef.mmodule.visibility_for(self.mmodule)}")
255 var propdefs
= mproperty
.lookup_definitions
(self.mmodule
, unsafe_type
)
257 if propdefs
.length
== 0 then
258 self.modelbuilder
.error
(node
, "Type error: no definition found for property {name} in {unsafe_type}")
260 else if propdefs
.length
== 1 then
261 mpropdef
= propdefs
.first
263 self.modelbuilder
.warning
(node
, "Warning: confliting property definitions for property {name} in {unsafe_type}: {propdefs.join(" ")}")
264 mpropdef
= mproperty
.intro
268 var msignature
= self.resolve_signature_for
(mpropdef
, recvtype
, recv_is_self
)
270 var erasure_cast
= false
271 var rettype
= mpropdef
.msignature
.return_mtype
272 if not recv_is_self
and rettype
!= null then
273 if rettype
isa MNullableType then rettype
= rettype
.mtype
274 if rettype
isa MParameterType then
275 var erased_rettype
= msignature
.return_mtype
276 assert erased_rettype
!= null
277 #node.debug("Erasure cast: Really a {rettype} but unsafely a {erased_rettype}")
282 var callsite
= new CallSite(node
, recvtype
, recv_is_self
, mproperty
, mpropdef
, msignature
, erasure_cast
)
286 # Visit the expressions of args and cheik their conformity with the corresponding typi in signature
287 # The point of this method is to handle varargs correctly
288 # Note: The signature must be correctly adapted
289 fun check_signature
(node
: ANode, args
: Array[AExpr], name
: String, msignature
: MSignature): Bool
291 var vararg_rank
= msignature
.vararg_rank
292 if vararg_rank
>= 0 then
293 if args
.length
< msignature
.arity
then
294 #self.modelbuilder.error(node, "Error: Incorrect number of parameters. Got {args.length}, expected at least {msignature.arity}. Signature is {msignature}")
295 self.modelbuilder
.error
(node
, "Error: arity mismatch; prototype is '{name}{msignature}'")
298 else if args
.length
!= msignature
.arity
then
299 self.modelbuilder
.error
(node
, "Error: Incorrect number of parameters. Got {args.length}, expected {msignature.arity}. Signature is {msignature}")
303 #debug("CALL {unsafe_type}.{msignature}")
305 var vararg_decl
= args
.length
- msignature
.arity
306 for i
in [0..msignature
.arity
[ do
308 if i
== vararg_rank
then continue # skip the vararg
309 if i
> vararg_rank
then
312 var paramtype
= msignature
.mparameters
[i
].mtype
313 self.visit_expr_subtype
(args
[j
], paramtype
)
315 if vararg_rank
>= 0 then
316 var varargs
= new Array[AExpr]
317 var paramtype
= msignature
.mparameters
[vararg_rank
].mtype
318 for j
in [vararg_rank
..vararg_rank
+vararg_decl
] do
320 self.visit_expr_subtype
(args
[j
], paramtype
)
326 fun error
(node
: ANode, message
: String)
328 self.modelbuilder
.toolcontext
.error
(node
.hot_location
, message
)
331 fun get_variable
(node
: AExpr, variable
: Variable): nullable MType
333 var flow
= node
.after_flow_context
335 self.error
(node
, "No context!")
339 if flow
.vars
.has_key
(variable
) then
340 return flow
.vars
[variable
]
342 #node.debug("*** START Collected for {variable}")
343 var mtypes
= flow
.collect_types
(variable
)
344 #node.debug("**** END Collected for {variable}")
345 if mtypes
== null or mtypes
.length
== 0 then
346 return variable
.declared_type
347 else if mtypes
.length
== 1 then
350 var res
= merge_types
(node
,mtypes
)
351 if res
== null then res
= variable
.declared_type
357 fun set_variable
(node
: AExpr, variable
: Variable, mtype
: nullable MType)
359 var flow
= node
.after_flow_context
362 flow
.set_var
(variable
, mtype
)
365 fun merge_types
(node
: ANode, col
: Array[nullable MType]): nullable MType
367 if col
.length
== 1 then return col
.first
368 var res
= new Array[nullable MType]
370 if t1
== null then continue # return null
373 if t2
== null then continue # return null
374 if t2
isa MNullableType or t2
isa MNullType then
377 if not is_subtype
(t2
, t1
) then found
= false
380 #print "merge {col.join(" ")} -> {t1}"
384 #self.modelbuilder.warning(node, "Type Error: {col.length} conflicting types: <{col.join(", ")}>")
389 # A specific method call site with its associated informations.
391 # The assiciated node for location
394 # The statis type of the receiver
397 # Is the receiver self?
398 # If "for_self", virtual types of the signature are keeped
399 # If "not_for_self", virtual type are erased
400 var recv_is_self
: Bool
402 # The designated method
403 var mproperty
: MMethod
405 # The statically designated method definition
406 # The most specif one, it is.
407 var mpropdef
: MMethodDef
409 # The resolved signature for the receiver
410 var msignature
: MSignature
412 # Is a implicit cast required on erasure typing policy?
413 var erasure_cast
: Bool
415 private fun check_signature
(v
: TypeVisitor, args
: Array[AExpr]): Bool
417 return v
.check_signature
(self.node
, args
, self.mproperty
.name
, self.msignature
)
422 # The declared type of the variable
423 var declared_type
: nullable MType
426 redef class FlowContext
427 # Store changes of types because of type evolution
428 private var vars
: HashMap[Variable, nullable MType] = new HashMap[Variable, nullable MType]
429 private var cache
: HashMap[Variable, nullable Array[nullable MType]] = new HashMap[Variable, nullable Array[nullable MType]]
431 # Adapt the variable to a static type
432 # Warning1: do not modify vars directly.
433 # Warning2: sub-flow may have cached a unadapted variabial
434 private fun set_var
(variable
: Variable, mtype
: nullable MType)
436 self.vars
[variable
] = mtype
437 self.cache
.keys
.remove
(variable
)
440 private fun collect_types
(variable
: Variable): nullable Array[nullable MType]
442 if cache
.has_key
(variable
) then
443 return cache
[variable
]
445 var res
: nullable Array[nullable MType] = null
446 if vars
.has_key
(variable
) then
447 var mtype
= vars
[variable
]
449 else if self.previous
.is_empty
then
451 res
= [variable
.declared_type
]
453 for flow
in self.previous
do
454 if flow
.is_unreachable
then continue
455 var r2
= flow
.collect_types
(variable
)
456 if r2
== null then continue
461 if not res
.has
(t
) then res
.add
(t
)
466 cache
[variable
] = res
472 # The entry point of the whole typing analysis
473 fun do_typing
(modelbuilder
: ModelBuilder)
477 # The variable associated to the reciever (if any)
478 var selfvariable
: nullable Variable
481 redef class AConcreteMethPropdef
482 redef fun do_typing
(modelbuilder
: ModelBuilder)
484 var nclassdef
= self.parent
.as(AClassdef)
485 var mpropdef
= self.mpropdef
.as(not null)
486 var v
= new TypeVisitor(modelbuilder
, nclassdef
, mpropdef
)
487 self.selfvariable
= v
.selfvariable
489 var nblock
= self.n_block
490 if nblock
== null then return
492 var mmethoddef
= self.mpropdef
.as(not null)
493 for i
in [0..mmethoddef
.msignature
.arity
[ do
494 var mtype
= mmethoddef
.msignature
.mparameters
[i
].mtype
495 if mmethoddef
.msignature
.vararg_rank
== i
then
496 var arrayclass
= v
.get_mclass
(self.n_signature
.n_params
[i
], "Array")
497 if arrayclass
== null then return # Skip error
498 mtype
= arrayclass
.get_mtype
([mtype
])
500 var variable
= self.n_signature
.n_params
[i
].variable
501 assert variable
!= null
502 variable
.declared_type
= mtype
506 if not nblock
.after_flow_context
.is_unreachable
and mmethoddef
.msignature
.return_mtype
!= null then
507 # We reach the end of the function without having a return, it is bad
508 v
.error
(self, "Control error: Reached end of function (a 'return' with a value was expected).")
513 redef class AAttrPropdef
514 redef fun do_typing
(modelbuilder
: ModelBuilder)
516 var nclassdef
= self.parent
.as(AClassdef)
517 var v
= new TypeVisitor(modelbuilder
, nclassdef
, self.mpropdef
.as(not null))
518 self.selfvariable
= v
.selfvariable
520 var nexpr
= self.n_expr
521 if nexpr
!= null then
522 var mtype
= self.mpropdef
.static_mtype
523 v
.visit_expr_subtype
(nexpr
, mtype
)
531 # The static type of the expression.
532 # null if self is a statement of in case of error
533 var mtype
: nullable MType = null
535 # Is the statement correctly typed?
536 # Used to distinguish errors and statements when `mtype == null`
537 var is_typed
: Bool = false
539 # If required, the following implicit cast `.as(XXX)`
540 # Such a cast may by required after evaluating the expression when
541 # a unsafe operation is detected (silently accepted by the Nit language).
542 # The attribute is computed by `check_subtype`
543 var implicit_cast_to
: nullable MType = null
545 # Return the variable read (if any)
546 # Used to perform adaptive typing
547 fun its_variable
: nullable Variable do return null
549 private fun accept_typing
(v
: TypeVisitor)
551 v
.error
(self, "no implemented accept_typing for {self.class_name}")
555 redef class ABlockExpr
556 redef fun accept_typing
(v
)
558 for e
in self.n_expr
do v
.visit_stmt
(e
)
562 # The type of a blockexpr is the one of the last expression (or null if empty)
565 if self.n_expr
.is_empty
then return null
566 return self.n_expr
.last
.mtype
570 redef class AVardeclExpr
571 redef fun accept_typing
(v
)
573 var variable
= self.variable
574 if variable
== null then return # Skip error
576 var ntype
= self.n_type
577 var mtype
: nullable MType
578 if ntype
== null then
581 mtype
= v
.resolve_mtype
(ntype
)
582 if mtype
== null then return # Skip error
585 var nexpr
= self.n_expr
586 if nexpr
!= null then
587 if mtype
!= null then
588 v
.visit_expr_subtype
(nexpr
, mtype
)
590 mtype
= v
.visit_expr
(nexpr
)
591 if mtype
== null then return # Skip error
596 if mtype
== null or mtype
isa MNullType then
597 decltype
= v
.get_mclass
(self, "Object").mclass_type
.as_nullable
598 if mtype
== null then mtype
= decltype
601 variable
.declared_type
= decltype
602 v
.set_variable
(self, variable
, mtype
)
604 #debug("var {variable}: {mtype}")
611 redef fun its_variable
do return self.variable
612 redef fun accept_typing
(v
)
614 var variable
= self.variable
615 if variable
== null then return # Skip error
617 var mtype
= v
.get_variable
(self, variable
)
618 if mtype
!= null then
619 #debug("{variable} is {mtype}")
621 #debug("{variable} is untyped")
628 redef class AVarAssignExpr
629 redef fun accept_typing
(v
)
631 var variable
= self.variable
632 assert variable
!= null
634 var mtype
= v
.visit_expr_subtype
(n_value
, variable
.declared_type
)
636 v
.set_variable
(self, variable
, mtype
)
642 redef class AReassignFormExpr
643 # @depreciated use `reassign_callsite`
644 fun reassign_property
: nullable MMethodDef do return self.reassign_callsite
.mpropdef
646 # The method designed by the reassign operator.
647 var reassign_callsite
: nullable CallSite
649 var read_type
: nullable MType = null
651 # Determine the `reassign_property`
652 # `readtype` is the type of the reading of the left value.
653 # `writetype` is the type of the writing of the left value.
654 # (Because of `ACallReassignExpr`, both can be different.
655 # Return the static type of the value to store.
656 private fun resolve_reassignment
(v
: TypeVisitor, readtype
, writetype
: MType): nullable MType
658 var reassign_name
: String
659 if self.n_assign_op
isa APlusAssignOp then
661 else if self.n_assign_op
isa AMinusAssignOp then
667 self.read_type
= readtype
669 if readtype
isa MNullType then
670 v
.error
(self, "Error: Method '{reassign_name}' call on 'null'.")
674 var callsite
= v
.get_method
(self, readtype
, reassign_name
, false)
675 if callsite
== null then return null # Skip error
676 self.reassign_callsite
= callsite
678 var msignature
= callsite
.msignature
679 var rettype
= msignature
.return_mtype
680 assert msignature
.arity
== 1 and rettype
!= null
682 var value_type
= v
.visit_expr_subtype
(self.n_value
, msignature
.mparameters
.first
.mtype
)
683 if value_type
== null then return null # Skip error
685 v
.check_subtype
(self, rettype
, writetype
)
690 redef class AVarReassignExpr
691 redef fun accept_typing
(v
)
693 var variable
= self.variable
694 assert variable
!= null
696 var readtype
= v
.get_variable
(self, variable
)
697 if readtype
== null then return
701 var writetype
= variable
.declared_type
702 if writetype
== null then return
704 var rettype
= self.resolve_reassignment
(v
, readtype
, writetype
)
706 v
.set_variable
(self, variable
, rettype
)
713 redef class AContinueExpr
714 redef fun accept_typing
(v
)
716 var nexpr
= self.n_expr
717 if nexpr
!= null then
718 var mtype
= v
.visit_expr
(nexpr
)
724 redef class ABreakExpr
725 redef fun accept_typing
(v
)
727 var nexpr
= self.n_expr
728 if nexpr
!= null then
729 var mtype
= v
.visit_expr
(nexpr
)
735 redef class AReturnExpr
736 redef fun accept_typing
(v
)
738 var nexpr
= self.n_expr
739 var ret_type
= v
.mpropdef
.as(MMethodDef).msignature
.return_mtype
740 if nexpr
!= null then
741 if ret_type
!= null then
742 var mtype
= v
.visit_expr_subtype
(nexpr
, ret_type
)
744 var mtype
= v
.visit_expr
(nexpr
)
745 v
.error
(self, "Error: Return with value in a procedure.")
747 else if ret_type
!= null then
748 v
.error
(self, "Error: Return without value in a function.")
754 redef class AAbortExpr
755 redef fun accept_typing
(v
)
762 redef fun accept_typing
(v
)
764 v
.visit_expr_bool
(n_expr
)
772 redef class AIfexprExpr
773 redef fun accept_typing
(v
)
775 v
.visit_expr_bool
(n_expr
)
777 var t1
= v
.visit_expr
(n_then
)
778 var t2
= v
.visit_expr
(n_else
)
780 if t1
== null or t2
== null then
784 var t
= v
.merge_types
(self, [t1
, t2
])
786 v
.error
(self, "Type Error: ambiguous type {t1} vs {t2}")
793 redef fun accept_typing
(v
)
795 v
.visit_stmt
(n_block
)
800 redef class AWhileExpr
801 redef fun accept_typing
(v
)
803 v
.visit_expr_bool
(n_expr
)
805 v
.visit_stmt
(n_block
)
810 redef class ALoopExpr
811 redef fun accept_typing
(v
)
813 v
.visit_stmt
(n_block
)
819 var coltype
: nullable MClassType
821 var method_iterator
: nullable MMethod
822 var method_is_ok
: nullable MMethod
823 var method_item
: nullable MMethod
824 var method_next
: nullable MMethod
825 var method_key
: nullable MMethod
827 private fun do_type_iterator
(v
: TypeVisitor, mtype
: MType)
830 var objcla
= v
.get_mclass
(self, "Object")
831 if objcla
== null then return
833 # check iterator method
834 var unsafe_type
= v
.anchor_to
(mtype
)
835 if v
.try_get_mproperty_by_name2
(self, unsafe_type
, "iterator") == null then
836 if v
.try_get_mproperty_by_name2
(self, unsafe_type
, "iterate") == null then
837 v
.error
(self, "Type Error: Expected method 'iterator' in type {mtype}")
839 v
.modelbuilder
.error
(self, "NOT YET IMPLEMENTED: Do 'for' on {mtype}")
844 var itdef
= v
.get_method
(self, mtype
, "iterator", true)
845 if itdef
== null then
846 v
.error
(self, "Type Error: Expected method 'iterator' in type {mtype}")
849 self.method_iterator
= itdef
.mproperty
851 # check that iterator return something
852 var ittype
= itdef
.msignature
.return_mtype
853 if ittype
== null then
854 v
.error
(self, "Type Error: Expected method 'iterator' to return an Iterator or MapIterator type")
859 var colit_cla
= v
.try_get_mclass
(self, "Iterator")
860 var mapit_cla
= v
.try_get_mclass
(self, "MapIterator")
864 if colit_cla
!= null and v
.is_subtype
(ittype
, colit_cla
.get_mtype
([objcla
.mclass_type
.as_nullable
])) then
866 var coltype
= ittype
.supertype_to
(v
.mmodule
, v
.anchor
, colit_cla
)
867 var variables
= self.variables
868 if variables
.length
!= 1 then
869 v
.error
(self, "Type Error: Expected one variable")
871 variables
.first
.declared_type
= coltype
.arguments
.first
876 if mapit_cla
!= null and v
.is_subtype
(ittype
, mapit_cla
.get_mtype
([objcla
.mclass_type
, objcla
.mclass_type
.as_nullable
])) then
878 var coltype
= ittype
.supertype_to
(v
.mmodule
, v
.anchor
, mapit_cla
)
879 var variables
= self.variables
880 if variables
.length
!= 2 then
881 v
.error
(self, "Type Error: Expected two variables")
883 variables
[0].declared_type
= coltype
.arguments
[0]
884 variables
[1].declared_type
= coltype
.arguments
[1]
889 if not is_col
and not is_map
then
890 v
.error
(self, "Type Error: Expected method 'iterator' to return an Iterator of MapIterator type")
894 # anchor formal and virtual types
895 if mtype
.need_anchor
then mtype
= v
.anchor_to
(mtype
)
897 self.coltype
= mtype
.as(MClassType)
899 # get methods is_ok, next, item
900 var ikdef
= v
.get_method
(self, ittype
, "is_ok", false)
901 if ikdef
== null then
902 v
.error
(self, "Type Error: Expected method 'is_ok' in Iterator type {ittype}")
905 self.method_is_ok
= ikdef
.mproperty
907 var itemdef
= v
.get_method
(self, ittype
, "item", false)
908 if itemdef
== null then
909 v
.error
(self, "Type Error: Expected method 'item' in Iterator type {ittype}")
912 self.method_item
= itemdef
.mproperty
914 var nextdef
= v
.get_method
(self, ittype
, "next", false)
915 if nextdef
== null then
916 v
.error
(self, "Type Error: Expected method 'next' in Iterator type {ittype}")
919 self.method_next
= nextdef
.mproperty
922 var keydef
= v
.get_method
(self, ittype
, "key", false)
923 if keydef
== null then
924 v
.error
(self, "Type Error: Expected method 'key' in Iterator type {ittype}")
927 self.method_key
= keydef
.mproperty
931 redef fun accept_typing
(v
)
933 var mtype
= v
.visit_expr
(n_expr
)
934 if mtype
== null then return
936 self.do_type_iterator
(v
, mtype
)
938 v
.visit_stmt
(n_block
)
943 redef class AAssertExpr
944 redef fun accept_typing
(v
)
946 v
.visit_expr_bool
(n_expr
)
954 redef fun accept_typing
(v
)
956 v
.visit_expr_bool
(n_expr
)
957 v
.visit_expr_bool
(n_expr2
)
958 self.mtype
= v
.type_bool
(self)
962 redef class AImpliesExpr
963 redef fun accept_typing
(v
)
965 v
.visit_expr_bool
(n_expr
)
966 v
.visit_expr_bool
(n_expr2
)
967 self.mtype
= v
.type_bool
(self)
972 redef fun accept_typing
(v
)
974 v
.visit_expr_bool
(n_expr
)
975 v
.visit_expr_bool
(n_expr2
)
976 self.mtype
= v
.type_bool
(self)
982 redef fun accept_typing
(v
)
984 v
.visit_expr_bool
(n_expr
)
985 self.mtype
= v
.type_bool
(self)
989 redef class AOrElseExpr
990 redef fun accept_typing
(v
)
992 var t1
= v
.visit_expr
(n_expr
)
993 var t2
= v
.visit_expr
(n_expr2
)
995 if t1
== null or t2
== null then
999 if t1
isa MNullableType then
1003 var t
= v
.merge_types
(self, [t1
, t2
])
1005 t
= v
.mmodule
.object_type
1006 if t2
isa MNullableType then
1009 #v.error(self, "Type Error: ambiguous type {t1} vs {t2}")
1015 redef class ATrueExpr
1016 redef fun accept_typing
(v
)
1018 self.mtype
= v
.type_bool
(self)
1022 redef class AFalseExpr
1023 redef fun accept_typing
(v
)
1025 self.mtype
= v
.type_bool
(self)
1029 redef class AIntExpr
1030 redef fun accept_typing
(v
)
1032 var mclass
= v
.get_mclass
(self, "Int")
1033 if mclass
== null then return # Forward error
1034 self.mtype
= mclass
.mclass_type
1038 redef class AFloatExpr
1039 redef fun accept_typing
(v
)
1041 var mclass
= v
.get_mclass
(self, "Float")
1042 if mclass
== null then return # Forward error
1043 self.mtype
= mclass
.mclass_type
1047 redef class ACharExpr
1048 redef fun accept_typing
(v
)
1050 var mclass
= v
.get_mclass
(self, "Char")
1051 if mclass
== null then return # Forward error
1052 self.mtype
= mclass
.mclass_type
1056 redef class AStringFormExpr
1057 redef fun accept_typing
(v
)
1059 var mclass
= v
.get_mclass
(self, "String")
1060 if mclass
== null then return # Forward error
1061 self.mtype
= mclass
.mclass_type
1065 redef class ASuperstringExpr
1066 redef fun accept_typing
(v
)
1068 var mclass
= v
.get_mclass
(self, "String")
1069 if mclass
== null then return # Forward error
1070 self.mtype
= mclass
.mclass_type
1071 for nexpr
in self.n_exprs
do
1072 var t
= v
.visit_expr
(nexpr
)
1077 redef class AArrayExpr
1078 redef fun accept_typing
(v
)
1080 var mtypes
= new Array[nullable MType]
1081 for e
in self.n_exprs
.n_exprs
do
1082 var t
= v
.visit_expr
(e
)
1088 var mtype
= v
.merge_types
(self, mtypes
)
1089 if mtype
== null then
1090 v
.error
(self, "Type Error: ambiguous array type {mtypes.join(" ")}")
1093 var mclass
= v
.get_mclass
(self, "Array")
1094 if mclass
== null then return # Forward error
1095 self.mtype
= mclass
.get_mtype
([mtype
])
1099 redef class ARangeExpr
1100 redef fun accept_typing
(v
)
1102 var discrete_class
= v
.get_mclass
(self, "Discrete")
1103 if discrete_class
== null then return # Forward error
1104 var discrete_type
= discrete_class
.intro
.bound_mtype
1105 var t1
= v
.visit_expr_subtype
(self.n_expr
, discrete_type
)
1106 var t2
= v
.visit_expr_subtype
(self.n_expr2
, discrete_type
)
1107 if t1
== null or t2
== null then return
1108 var mclass
= v
.get_mclass
(self, "Range")
1109 if mclass
== null then return # Forward error
1110 if v
.is_subtype
(t1
, t2
) then
1111 self.mtype
= mclass
.get_mtype
([t2
])
1112 else if v
.is_subtype
(t2
, t1
) then
1113 self.mtype
= mclass
.get_mtype
([t1
])
1115 v
.error
(self, "Type Error: Cannot create range: {t1} vs {t2}")
1120 redef class ANullExpr
1121 redef fun accept_typing
(v
)
1123 self.mtype
= v
.mmodule
.model
.null_type
1127 redef class AIsaExpr
1128 # The static type to cast to.
1129 # (different from the static type of the expression that is `Bool`).
1130 var cast_type
: nullable MType
1131 redef fun accept_typing
(v
)
1133 var mtype
= v
.visit_expr_cast
(self, self.n_expr
, self.n_type
)
1134 self.cast_type
= mtype
1136 var variable
= self.n_expr
.its_variable
1137 if variable
!= null then
1138 var orig
= self.n_expr
.mtype
1139 var from
= if orig
!= null then orig
.to_s
else "invalid"
1140 var to
= if mtype
!= null then mtype
.to_s
else "invalid"
1141 #debug("adapt {variable}: {from} -> {to}")
1142 self.after_flow_context
.when_true
.set_var
(variable
, mtype
)
1145 self.mtype
= v
.type_bool
(self)
1149 redef class AAsCastExpr
1150 redef fun accept_typing
(v
)
1152 self.mtype
= v
.visit_expr_cast
(self, self.n_expr
, self.n_type
)
1156 redef class AAsNotnullExpr
1157 redef fun accept_typing
(v
)
1159 var mtype
= v
.visit_expr
(self.n_expr
)
1160 if mtype
isa MNullType then
1161 v
.error
(self, "Type error: as(not null) on null")
1164 if mtype
isa MNullableType then
1165 self.mtype
= mtype
.mtype
1168 # TODO: warn on useless as not null
1173 redef class AProxyExpr
1174 redef fun accept_typing
(v
)
1176 self.mtype
= v
.visit_expr
(self.n_expr
)
1180 redef class ASelfExpr
1181 redef var its_variable
: nullable Variable
1182 redef fun accept_typing
(v
)
1184 var variable
= v
.selfvariable
1185 self.its_variable
= variable
1186 self.mtype
= v
.get_variable
(self, variable
)
1190 ## MESSAGE SENDING AND PROPERTY
1192 redef class ASendExpr
1193 # @depreciated: use `callsite`
1194 fun mproperty
: nullable MMethod do return callsite
.mproperty
1196 # The property invoked by the send.
1197 var callsite
: nullable CallSite
1199 redef fun accept_typing
(v
)
1201 var recvtype
= v
.visit_expr
(self.n_expr
)
1202 var name
= self.property_name
1204 if recvtype
== null then return # Forward error
1205 if recvtype
isa MNullType then
1206 v
.error
(self, "Error: Method '{name}' call on 'null'.")
1210 var callsite
= v
.get_method
(self, recvtype
, name
, self.n_expr
isa ASelfExpr)
1211 if callsite
== null then return
1212 self.callsite
= callsite
1213 var msignature
= callsite
.msignature
1215 var args
= compute_raw_arguments
1216 self.raw_arguments
= args
1218 callsite
.check_signature
(v
, args
)
1220 if callsite
.mproperty
.is_init
then
1221 var vmpropdef
= v
.mpropdef
1222 if not (vmpropdef
isa MMethodDef and vmpropdef
.mproperty
.is_init
) then
1223 v
.error
(self, "Can call a init only in another init")
1227 var ret
= msignature
.return_mtype
1231 self.is_typed
= true
1235 # The name of the property
1236 # Each subclass simply provide the correct name.
1237 private fun property_name
: String is abstract
1239 # An array of all arguments (excluding self)
1240 var raw_arguments
: nullable Array[AExpr]
1242 private fun compute_raw_arguments
: Array[AExpr] is abstract
1245 redef class ABinopExpr
1246 redef fun compute_raw_arguments
do return [n_expr2
]
1249 redef fun property_name
do return "=="
1250 redef fun accept_typing
(v
)
1254 var variable
= self.n_expr
.its_variable
1255 if variable
== null then return
1256 var mtype
= self.n_expr2
.mtype
1257 if not mtype
isa MNullType then return
1258 var vartype
= v
.get_variable
(self, variable
)
1259 if not vartype
isa MNullableType then return
1260 self.after_flow_context
.when_true
.set_var
(variable
, mtype
)
1261 self.after_flow_context
.when_false
.set_var
(variable
, vartype
.mtype
)
1262 #debug("adapt {variable}:{vartype} ; true->{mtype} false->{vartype.mtype}")
1266 redef fun property_name
do return "!="
1267 redef fun accept_typing
(v
)
1271 var variable
= self.n_expr
.its_variable
1272 if variable
== null then return
1273 var mtype
= self.n_expr2
.mtype
1274 if not mtype
isa MNullType then return
1275 var vartype
= v
.get_variable
(self, variable
)
1276 if not vartype
isa MNullableType then return
1277 self.after_flow_context
.when_false
.set_var
(variable
, mtype
)
1278 self.after_flow_context
.when_true
.set_var
(variable
, vartype
.mtype
)
1279 #debug("adapt {variable}:{vartype} ; true->{vartype.mtype} false->{mtype}")
1283 redef fun property_name
do return "<"
1286 redef fun property_name
do return "<="
1289 redef fun property_name
do return "<<"
1292 redef fun property_name
do return ">"
1295 redef fun property_name
do return ">="
1298 redef fun property_name
do return ">>"
1300 redef class APlusExpr
1301 redef fun property_name
do return "+"
1303 redef class AMinusExpr
1304 redef fun property_name
do return "-"
1306 redef class AStarshipExpr
1307 redef fun property_name
do return "<=>"
1309 redef class AStarExpr
1310 redef fun property_name
do return "*"
1312 redef class ASlashExpr
1313 redef fun property_name
do return "/"
1315 redef class APercentExpr
1316 redef fun property_name
do return "%"
1319 redef class AUminusExpr
1320 redef fun property_name
do return "unary -"
1321 redef fun compute_raw_arguments
do return new Array[AExpr]
1325 redef class ACallExpr
1326 redef fun property_name
do return n_id
.text
1327 redef fun compute_raw_arguments
do return n_args
.to_a
1330 redef class ACallAssignExpr
1331 redef fun property_name
do return n_id
.text
+ "="
1332 redef fun compute_raw_arguments
1334 var res
= n_args
.to_a
1340 redef class ABraExpr
1341 redef fun property_name
do return "[]"
1342 redef fun compute_raw_arguments
do return n_args
.to_a
1345 redef class ABraAssignExpr
1346 redef fun property_name
do return "[]="
1347 redef fun compute_raw_arguments
1349 var res
= n_args
.to_a
1355 redef class ASendReassignFormExpr
1356 # @depreciated use `write_callsite`
1357 fun write_mproperty
: nullable MMethod do return write_callsite
.mproperty
1359 # The property invoked for the writing
1360 var write_callsite
: nullable CallSite
1362 redef fun accept_typing
(v
)
1364 var recvtype
= v
.visit_expr
(self.n_expr
)
1365 var name
= self.property_name
1367 if recvtype
== null then return # Forward error
1368 if recvtype
isa MNullType then
1369 v
.error
(self, "Error: Method '{name}' call on 'null'.")
1373 var for_self
= self.n_expr
isa ASelfExpr
1374 var callsite
= v
.get_method
(self, recvtype
, name
, for_self
)
1376 if callsite
== null then return
1377 self.callsite
= callsite
1379 var args
= compute_raw_arguments
1380 self.raw_arguments
= args
1382 callsite
.check_signature
(v
, args
)
1384 var readtype
= callsite
.msignature
.return_mtype
1385 if readtype
== null then
1386 v
.error
(self, "Error: {name} is not a function")
1390 var wcallsite
= v
.get_method
(self, recvtype
, name
+ "=", self.n_expr
isa ASelfExpr)
1391 if wcallsite
== null then return
1392 self.write_callsite
= wcallsite
1394 var wtype
= self.resolve_reassignment
(v
, readtype
, wcallsite
.msignature
.mparameters
.last
.mtype
)
1395 if wtype
== null then return
1397 args
= args
.to_a
# duplicate so raw_arguments keeps only the getter args
1398 args
.add
(self.n_value
)
1399 wcallsite
.check_signature
(v
, args
)
1401 self.is_typed
= true
1405 redef class ACallReassignExpr
1406 redef fun property_name
do return n_id
.text
1407 redef fun compute_raw_arguments
do return n_args
.to_a
1410 redef class ABraReassignExpr
1411 redef fun property_name
do return "[]"
1412 redef fun compute_raw_arguments
do return n_args
.to_a
1415 redef class AInitExpr
1416 redef fun property_name
do return "init"
1417 redef fun compute_raw_arguments
do return n_args
.to_a
1421 fun to_a
: Array[AExpr] do return self.n_exprs
.to_a
1426 redef class ASuperExpr
1427 # The method to call if the super is in fact a 'super init call'
1428 # Note: if the super is a normal call-next-method, then this attribute is null
1429 var mproperty
: nullable MMethod
1431 redef fun accept_typing
(v
)
1433 var recvtype
= v
.nclassdef
.mclassdef
.bound_mtype
1434 var mproperty
= v
.mpropdef
.mproperty
1435 if not mproperty
isa MMethod then
1436 v
.error
(self, "Error: super only usable in a method")
1439 var superprops
= mproperty
.lookup_super_definitions
(v
.mmodule
, recvtype
)
1440 if superprops
.length
== 0 then
1441 if mproperty
.is_init
and v
.mpropdef
.is_intro
then
1442 process_superinit
(v
)
1445 v
.error
(self, "Error: No super method to call for {mproperty}.")
1448 # FIXME: covariance of return type in linear extension?
1449 var superprop
= superprops
.first
1450 assert superprop
isa MMethodDef
1452 var msignature
= v
.resolve_signature_for
(superprop
, recvtype
, true)
1453 var args
= self.n_args
.to_a
1454 if args
.length
> 0 then
1455 v
.check_signature
(self, args
, mproperty
.name
, msignature
)
1457 self.mtype
= msignature
.return_mtype
1458 self.is_typed
= true
1459 v
.mpropdef
.has_supercall
= true
1462 private fun process_superinit
(v
: TypeVisitor)
1464 var recvtype
= v
.nclassdef
.mclassdef
.bound_mtype
1465 var mproperty
= v
.mpropdef
.mproperty
1466 var superprop
: nullable MMethodDef = null
1467 for msupertype
in v
.nclassdef
.mclassdef
.supertypes
do
1468 msupertype
= msupertype
.anchor_to
(v
.mmodule
, recvtype
)
1469 var errcount
= v
.modelbuilder
.toolcontext
.error_count
1470 var candidate
= v
.try_get_mproperty_by_name2
(self, msupertype
, mproperty
.name
).as(nullable MMethod)
1471 if candidate
== null then
1472 if v
.modelbuilder
.toolcontext
.error_count
> errcount
then return # Forard error
1473 continue # Try next super-class
1475 if superprop
!= null and superprop
.mproperty
!= candidate
then
1476 v
.error
(self, "Error: conflicting super constructor to call for {mproperty}: {candidate.full_name}, {superprop.mproperty.full_name}")
1479 var candidatedefs
= candidate
.lookup_definitions
(v
.mmodule
, recvtype
)
1480 if superprop
!= null then
1481 if superprop
== candidatedefs
.first
then continue
1482 candidatedefs
.add
(superprop
)
1484 if candidatedefs
.length
> 1 then
1485 v
.error
(self, "Error: confliting property definitions for property {mproperty} in {recvtype}: {candidatedefs.join(", ")}")
1488 superprop
= candidatedefs
.first
1490 if superprop
== null then
1491 v
.error
(self, "Error: No super method to call for {mproperty}.")
1494 self.mproperty
= superprop
.mproperty
1496 var args
= self.n_args
.to_a
1497 var msignature
= v
.resolve_signature_for
(superprop
, recvtype
, true)
1498 if args
.length
> 0 then
1499 v
.check_signature
(self, args
, mproperty
.name
, msignature
)
1501 # TODO: Check signature
1504 self.is_typed
= true
1510 redef class ANewExpr
1511 # @depreciated use `callsite`
1512 fun mproperty
: nullable MMethod do return self.callsite
.mproperty
1514 # The constructor invoked by the new.
1515 var callsite
: nullable CallSite
1517 redef fun accept_typing
(v
)
1519 var recvtype
= v
.resolve_mtype
(self.n_type
)
1520 if recvtype
== null then return
1521 self.mtype
= recvtype
1523 if not recvtype
isa MClassType then
1524 if recvtype
isa MNullableType then
1525 v
.error
(self, "Type error: cannot instantiate the nullable type {recvtype}.")
1528 v
.error
(self, "Type error: cannot instantiate the formal type {recvtype}.")
1532 if recvtype
.mclass
.kind
== abstract_kind
then
1533 v
.error
(self, "Cannot instantiate abstract class {recvtype}.")
1535 else if recvtype
.mclass
.kind
== interface_kind
then
1536 v
.error
(self, "Cannot instantiate interface {recvtype}.")
1548 var callsite
= v
.get_method
(self, recvtype
, name
, false)
1549 if callsite
== null then return
1551 self.callsite
= callsite
1553 if not callsite
.mproperty
.is_init_for
(recvtype
.mclass
) then
1554 v
.error
(self, "Error: {name} is not a constructor.")
1558 var args
= n_args
.to_a
1559 callsite
.check_signature
(v
, args
)
1565 redef class AAttrFormExpr
1566 # The attribute acceded.
1567 var mproperty
: nullable MAttribute
1569 # The static type of the attribute.
1570 var attr_type
: nullable MType
1572 # Resolve the attribute acceded.
1573 private fun resolve_property
(v
: TypeVisitor)
1575 var recvtype
= v
.visit_expr
(self.n_expr
)
1576 if recvtype
== null then return # Skip error
1577 var name
= self.n_id
.text
1578 if recvtype
isa MNullType then
1579 v
.error
(self, "Error: Attribute '{name}' access on 'null'.")
1583 var unsafe_type
= v
.anchor_to
(recvtype
)
1584 var mproperty
= v
.try_get_mproperty_by_name2
(self, unsafe_type
, name
)
1585 if mproperty
== null then
1586 v
.modelbuilder
.error
(self, "Error: Attribute {name} doesn't exists in {recvtype}.")
1589 assert mproperty
isa MAttribute
1590 self.mproperty
= mproperty
1592 var mpropdefs
= mproperty
.lookup_definitions
(v
.mmodule
, unsafe_type
)
1593 assert mpropdefs
.length
== 1
1594 var mpropdef
= mpropdefs
.first
1595 var attr_type
= mpropdef
.static_mtype
.as(not null)
1596 attr_type
= v
.resolve_for
(attr_type
, recvtype
, self.n_expr
isa ASelfExpr)
1597 self.attr_type
= attr_type
1601 redef class AAttrExpr
1602 redef fun accept_typing
(v
)
1604 self.resolve_property
(v
)
1605 self.mtype
= self.attr_type
1610 redef class AAttrAssignExpr
1611 redef fun accept_typing
(v
)
1613 self.resolve_property
(v
)
1614 var mtype
= self.attr_type
1616 v
.visit_expr_subtype
(self.n_value
, mtype
)
1617 self.is_typed
= true
1621 redef class AAttrReassignExpr
1622 redef fun accept_typing
(v
)
1624 self.resolve_property
(v
)
1625 var mtype
= self.attr_type
1626 if mtype
== null then return # Skip error
1628 self.resolve_reassignment
(v
, mtype
, mtype
)
1630 self.is_typed
= true
1634 redef class AIssetAttrExpr
1635 redef fun accept_typing
(v
)
1637 self.resolve_property
(v
)
1638 var mtype
= self.attr_type
1639 if mtype
== null then return # Skip error
1641 var recvtype
= self.n_expr
.mtype
.as(not null)
1642 var bound
= v
.resolve_for
(mtype
, recvtype
, false)
1643 if bound
isa MNullableType then
1644 v
.error
(self, "Error: isset on a nullable attribute.")
1646 self.mtype
= v
.type_bool
(self)
1652 redef class ADebugTypeExpr
1653 redef fun accept_typing
(v
)
1655 var expr
= v
.visit_expr
(self.n_expr
)
1656 if expr
== null then return
1657 var unsafe
= v
.anchor_to
(expr
)
1658 var ntype
= self.n_type
1659 var mtype
= v
.resolve_mtype
(ntype
)
1660 if mtype
!= null and mtype
!= expr
then
1661 var umtype
= v
.anchor_to
(mtype
)
1662 v
.modelbuilder
.warning
(self, "Found type {expr} (-> {unsafe}), expected {mtype} (-> {umtype})")
1664 self.is_typed
= true