Merge: Detach CallSite from the AST
[nit.git] / src / semantize / typing.nit
1 # This file is part of NIT ( http://www.nitlanguage.org ).
2 #
3 # Copyright 2012 Jean Privat <jean@pryen.org>
4 #
5 # Licensed under the Apache License, Version 2.0 (the "License");
6 # you may not use this file except in compliance with the License.
7 # You may obtain a copy of the License at
8 #
9 # http://www.apache.org/licenses/LICENSE-2.0
10 #
11 # Unless required by applicable law or agreed to in writing, software
12 # distributed under the License is distributed on an "AS IS" BASIS,
13 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14 # See the License for the specific language governing permissions and
15 # limitations under the License.
16
17 # Intraprocedural resolution of static types and OO-services
18 # By OO-services we mean message sending, attribute access, instantiation, etc.
19 module typing
20
21 import modelize
22 import local_var_init
23 import literal
24
25 redef class ToolContext
26 var typing_phase: Phase = new TypingPhase(self, [flow_phase, modelize_property_phase, local_var_init_phase])
27 end
28
29 private class TypingPhase
30 super Phase
31 redef fun process_npropdef(npropdef) do npropdef.do_typing(toolcontext.modelbuilder)
32 end
33
34 private class TypeVisitor
35 var modelbuilder: ModelBuilder
36
37 # The module of the analysis
38 # Used to correctly query the model
39 var mmodule: MModule
40
41 # The static type of the receiver
42 # Mainly used for type tests and type resolutions
43 var anchor: nullable MClassType = null
44
45 # The analyzed mclassdef
46 var mclassdef: nullable MClassDef = null
47
48 # The analyzed property
49 var mpropdef: nullable MPropDef
50
51 var selfvariable = new Variable("self")
52
53 # Is `self` use restricted?
54 # * no explicit `self`
55 # * method called on the implicit self must be top-level
56 # Currently only used for `new` factory since there is no valid receiver inside
57 var is_toplevel_context = false
58
59 init
60 do
61 var mpropdef = self.mpropdef
62
63 if mpropdef != null then
64 self.mpropdef = mpropdef
65 var mclassdef = mpropdef.mclassdef
66 self.mclassdef = mclassdef
67 self.anchor = mclassdef.bound_mtype
68
69 var mclass = mclassdef.mclass
70
71 var selfvariable = new Variable("self")
72 self.selfvariable = selfvariable
73 selfvariable.declared_type = mclass.mclass_type
74
75 var mprop = mpropdef.mproperty
76 if mprop isa MMethod and mprop.is_new then
77 is_toplevel_context = true
78 end
79 end
80 end
81
82 fun anchor_to(mtype: MType): MType
83 do
84 var anchor = anchor
85 if anchor == null then
86 assert not mtype.need_anchor
87 return mtype
88 end
89 return mtype.anchor_to(mmodule, anchor)
90 end
91
92 fun is_subtype(sub, sup: MType): Bool
93 do
94 return sub.is_subtype(mmodule, anchor, sup)
95 end
96
97 fun resolve_for(mtype, subtype: MType, for_self: Bool): MType
98 do
99 #print "resolve_for {mtype} sub={subtype} forself={for_self} mmodule={mmodule} anchor={anchor}"
100 var res = mtype.resolve_for(subtype, anchor, mmodule, not for_self)
101 return res
102 end
103
104 # Check that `sub` is a subtype of `sup`.
105 # If `sub` is not a valid suptype, then display an error on `node` an return null.
106 # If `sub` is a safe subtype of `sup` then return `sub`.
107 # If `sub` is an unsafe subtype (ie an implicit cast is required), then return `sup`.
108 #
109 # The point of the return type is to determinate the usable type on an expression when `autocast` is true:
110 # If the suptype is safe, then the return type is the one on the expression typed by `sub`.
111 # Is the subtype is unsafe, then the return type is the one of an implicit cast on `sup`.
112 fun check_subtype(node: ANode, sub, sup: MType, autocast: Bool): nullable MType
113 do
114 if self.is_subtype(sub, sup) then return sub
115 if autocast and self.is_subtype(sub, self.anchor_to(sup)) then
116 # FIXME workaround to the current unsafe typing policy. To remove once fixed virtual types exists.
117 #node.debug("Unsafe typing: expected {sup}, got {sub}")
118 return sup
119 end
120 if sub.need_anchor then
121 var u = anchor_to(sub)
122 self.modelbuilder.error(node, "Type Error: expected `{sup}`, got `{sub}: {u}`.")
123 else
124 self.modelbuilder.error(node, "Type Error: expected `{sup}`, got `{sub}`.")
125 end
126 return null
127 end
128
129 # Visit an expression and do not care about the return value
130 fun visit_stmt(nexpr: nullable AExpr)
131 do
132 if nexpr == null then return
133 nexpr.accept_typing(self)
134 end
135
136 # Visit an expression and expects that it is not a statement
137 # Return the type of the expression
138 # Display an error and return null if:
139 # * the type cannot be determined or
140 # * `nexpr` is a statement
141 fun visit_expr(nexpr: AExpr): nullable MType
142 do
143 nexpr.accept_typing(self)
144 var mtype = nexpr.mtype
145 if mtype != null then return mtype
146 if not nexpr.is_typed then
147 if not self.modelbuilder.toolcontext.error_count > 0 then # check that there is really an error
148 if self.modelbuilder.toolcontext.verbose_level > 1 then
149 nexpr.debug("No return type but no error.")
150 end
151 end
152 return null # forward error
153 end
154 self.error(nexpr, "Error: expected an expression.")
155 return null
156 end
157
158 # Visit an expression and expect its static type is a least a `sup`
159 # Return the type of the expression or null if
160 # * the type cannot be determined or
161 # * `nexpr` is a statement or
162 # * `nexpr` is not a `sup`
163 fun visit_expr_subtype(nexpr: AExpr, sup: nullable MType): nullable MType
164 do
165 var sub = visit_expr(nexpr)
166 if sub == null then return null # Forward error
167
168 if sup == null then return null # Forward error
169
170 var res = check_subtype(nexpr, sub, sup, true)
171 if res != sub then
172 nexpr.implicit_cast_to = res
173 end
174 return res
175 end
176
177 # Visit an expression and expect its static type is a `Bool`
178 # Return the type of the expression or null if
179 # * the type cannot be determined or
180 # * `nexpr` is a statement or
181 # * `nexpr` is not a `Bool`
182 fun visit_expr_bool(nexpr: AExpr): nullable MType
183 do
184 return self.visit_expr_subtype(nexpr, self.type_bool(nexpr))
185 end
186
187
188 fun check_expr_cast(node: ANode, nexpr: AExpr, ntype: AType): nullable MType
189 do
190 var sub = nexpr.mtype
191 if sub == null then return null # Forward error
192
193 var sup = ntype.mtype
194 if sup == null then return null # Forward error
195
196 if sup == sub then
197 self.modelbuilder.warning(node, "useless-type-test", "Warning: expression is already a `{sup}`.")
198 else if self.is_subtype(sub, sup) then
199 self.modelbuilder.warning(node, "useless-type-test", "Warning: expression is already a `{sup}` since it is a `{sub}`.")
200 end
201 return sup
202 end
203
204 # Can `mtype` be null (up to the current knowledge)?
205 fun can_be_null(mtype: MType): Bool
206 do
207 if mtype isa MNullableType or mtype isa MNullType then return true
208 if mtype isa MFormalType then
209 var x = anchor_to(mtype)
210 if x isa MNullableType or x isa MNullType then return true
211 end
212 return false
213 end
214
215 # Check that `mtype` can be null (up to the current knowledge).
216 #
217 # If not then display a `useless-null-test` warning on node and return false.
218 # Else return true.
219 fun check_can_be_null(anode: ANode, mtype: MType): Bool
220 do
221 if mtype isa MNullType then
222 modelbuilder.warning(anode, "useless-null-test", "Warning: expression is always `null`.")
223 return true
224 end
225 if can_be_null(mtype) then return true
226
227 if mtype isa MFormalType then
228 var res = anchor_to(mtype)
229 modelbuilder.warning(anode, "useless-null-test", "Warning: expression is not null, since it is a `{mtype}: {res}`.")
230 else
231 modelbuilder.warning(anode, "useless-null-test", "Warning: expression is not null, since it is a `{mtype}`.")
232 end
233 return false
234 end
235
236 # Special verification on != and == for null
237 # Return true
238 fun null_test(anode: ABinopExpr)
239 do
240 var mtype = anode.n_expr.mtype
241 var mtype2 = anode.n_expr2.mtype
242
243 if mtype == null or mtype2 == null then return
244
245 if not mtype2 isa MNullType then return
246
247 # Check of useless null
248 if not can_be_null(mtype) then return
249
250 if mtype isa MNullType then
251 # Because of type adaptation, we cannot just stop here
252 # so return use `null` as a bottom type that will be merged easily (cf) `merge_types`
253 mtype = null
254 else
255 mtype = mtype.as_notnull
256 end
257
258 # Check for type adaptation
259 var variable = anode.n_expr.its_variable
260 if variable == null then return
261
262 # One is null (mtype2 see above) the other is not null
263 if anode isa AEqExpr then
264 anode.after_flow_context.when_true.set_var(self, variable, mtype2)
265 anode.after_flow_context.when_false.set_var(self, variable, mtype)
266 else if anode isa ANeExpr then
267 anode.after_flow_context.when_false.set_var(self, variable, mtype2)
268 anode.after_flow_context.when_true.set_var(self, variable, mtype)
269 else
270 abort
271 end
272 end
273
274 fun try_get_mproperty_by_name2(anode: ANode, mtype: MType, name: String): nullable MProperty
275 do
276 return self.modelbuilder.try_get_mproperty_by_name2(anode, mmodule, mtype, name)
277 end
278
279 fun resolve_mtype(node: AType): nullable MType
280 do
281 return self.modelbuilder.resolve_mtype(mmodule, mclassdef, node)
282 end
283
284 fun try_get_mclass(node: ANode, name: String): nullable MClass
285 do
286 var mclass = modelbuilder.try_get_mclass_by_name(node, mmodule, name)
287 return mclass
288 end
289
290 fun get_mclass(node: ANode, name: String): nullable MClass
291 do
292 var mclass = modelbuilder.get_mclass_by_name(node, mmodule, name)
293 return mclass
294 end
295
296 fun type_bool(node: ANode): nullable MType
297 do
298 var mclass = self.get_mclass(node, "Bool")
299 if mclass == null then return null
300 return mclass.mclass_type
301 end
302
303 fun get_method(node: ANode, recvtype: MType, name: String, recv_is_self: Bool): nullable CallSite
304 do
305 var unsafe_type = self.anchor_to(recvtype)
306
307 #debug("recv: {recvtype} (aka {unsafe_type})")
308 if recvtype isa MNullType then
309 var objclass = get_mclass(node, "Object")
310 if objclass == null then return null # Forward error
311 unsafe_type = objclass.mclass_type
312 end
313
314 var mproperty = self.try_get_mproperty_by_name2(node, unsafe_type, name)
315 if name == "new" and mproperty == null then
316 name = "init"
317 mproperty = self.try_get_mproperty_by_name2(node, unsafe_type, name)
318 end
319
320 if mproperty == null then
321 if recv_is_self then
322 self.modelbuilder.error(node, "Error: method or variable `{name}` unknown in `{recvtype}`.")
323 else if recvtype.need_anchor then
324 self.modelbuilder.error(node, "Error: method `{name}` does not exists in `{recvtype}: {unsafe_type}`.")
325 else
326 self.modelbuilder.error(node, "Error: method `{name}` does not exists in `{recvtype}`.")
327 end
328 return null
329 end
330
331 assert mproperty isa MMethod
332
333 # `null` only accepts some methods of object.
334 if recvtype isa MNullType and not mproperty.is_null_safe then
335 self.error(node, "Error: method `{name}` called on `null`.")
336 return null
337 else if unsafe_type isa MNullableType and not mproperty.is_null_safe then
338 modelbuilder.advice(node, "call-on-nullable", "Warning: method call on a nullable receiver `{recvtype}`.")
339 end
340
341 if is_toplevel_context and recv_is_self and not mproperty.is_toplevel then
342 error(node, "Error: `{name}` is not a top-level method, thus need a receiver.")
343 end
344 if not recv_is_self and mproperty.is_toplevel then
345 error(node, "Error: cannot call `{name}`, a top-level method, with a receiver.")
346 end
347
348 if mproperty.visibility == protected_visibility and not recv_is_self and self.mmodule.visibility_for(mproperty.intro_mclassdef.mmodule) < intrude_visibility and not modelbuilder.toolcontext.opt_ignore_visibility.value then
349 self.modelbuilder.error(node, "Error: method `{name}` is protected and can only accessed by `self`.")
350 return null
351 end
352
353 var info = mproperty.deprecation
354 if info != null and self.mpropdef.mproperty.deprecation == null then
355 var mdoc = info.mdoc
356 if mdoc != null then
357 self.modelbuilder.warning(node, "deprecated-method", "Deprecation Warning: method `{name}` is deprecated: {mdoc.content.first}")
358 else
359 self.modelbuilder.warning(node, "deprecated-method", "Deprecation Warning: method `{name}` is deprecated.")
360 end
361 end
362
363 var propdefs = mproperty.lookup_definitions(self.mmodule, unsafe_type)
364 var mpropdef
365 if propdefs.length == 0 then
366 self.modelbuilder.error(node, "Type Error: no definition found for property `{name}` in `{unsafe_type}`.")
367 return null
368 else if propdefs.length == 1 then
369 mpropdef = propdefs.first
370 else
371 self.modelbuilder.warning(node, "property-conflict", "Warning: conflicting property definitions for property `{name}` in `{unsafe_type}`: {propdefs.join(" ")}")
372 mpropdef = mproperty.intro
373 end
374
375
376 var msignature = mpropdef.new_msignature or else mpropdef.msignature
377 if msignature == null then return null # skip error
378 msignature = resolve_for(msignature, recvtype, recv_is_self).as(MSignature)
379
380 var erasure_cast = false
381 var rettype = mpropdef.msignature.return_mtype
382 if not recv_is_self and rettype != null then
383 rettype = rettype.undecorate
384 if rettype isa MParameterType then
385 var erased_rettype = msignature.return_mtype
386 assert erased_rettype != null
387 #node.debug("Erasure cast: Really a {rettype} but unsafely a {erased_rettype}")
388 erasure_cast = true
389 end
390 end
391
392 var callsite = new CallSite(node.hot_location, recvtype, mmodule, anchor, recv_is_self, mproperty, mpropdef, msignature, erasure_cast)
393 return callsite
394 end
395
396 fun try_get_method(node: ANode, recvtype: MType, name: String, recv_is_self: Bool): nullable CallSite
397 do
398 var unsafe_type = self.anchor_to(recvtype)
399 var mproperty = self.try_get_mproperty_by_name2(node, unsafe_type, name)
400 if mproperty == null then return null
401 return get_method(node, recvtype, name, recv_is_self)
402 end
403
404
405 # Visit the expressions of args and check their conformity with the corresponding type in signature
406 # The point of this method is to handle varargs correctly
407 # Note: The signature must be correctly adapted
408 fun check_signature(node: ANode, args: Array[AExpr], mproperty: MProperty, msignature: MSignature): nullable SignatureMap
409 do
410 var vararg_rank = msignature.vararg_rank
411 if vararg_rank >= 0 then
412 if args.length < msignature.arity then
413 modelbuilder.error(node, "Error: expected at least {msignature.arity} argument(s) for `{mproperty}{msignature}`; got {args.length}. See introduction at `{mproperty.full_name}`.")
414 return null
415 end
416 else if args.length != msignature.arity then
417 # Too much argument
418 if args.length > msignature.arity then
419 modelbuilder.error(node, "Error: expected {msignature.arity} argument(s) for `{mproperty}{msignature}`; got {args.length}. See introduction at `{mproperty.full_name}`.")
420 return null
421 end
422 # Other cases are managed later
423 end
424
425
426 #debug("CALL {unsafe_type}.{msignature}")
427
428 # Associate each parameter to a position in the arguments
429 var map = new SignatureMap
430
431 # Special case for the isolated last argument
432 # TODO: reify this method characteristics (where? the param, the signature, the method?)
433 var last_is_padded = mproperty.name.chars.last == '='
434 var nbargs = args.length
435 if last_is_padded then
436 nbargs -= 1
437 assert not args.last isa ANamedargExpr
438 map.map[msignature.arity - 1] = args.length - 1
439 self.visit_expr_subtype(args.last, msignature.mparameters.last.mtype)
440 end
441
442 # First, handle named arguments
443 for i in [0..args.length[ do
444 var e = args[i]
445 if not e isa ANamedargExpr then continue
446 var name = e.n_id.text
447 var param = msignature.mparameter_by_name(name)
448 if param == null then
449 modelbuilder.error(e.n_id, "Error: no parameter `{name}` for `{mproperty}{msignature}`.")
450 return null
451 end
452 var idx = msignature.mparameters.index_of(param)
453 var prev = map.map.get_or_null(idx)
454 if prev != null then
455 modelbuilder.error(e, "Error: parameter `{name}` already associated with argument #{prev} for `{mproperty}{msignature}`.")
456 return null
457 end
458 map.map[idx] = i
459 e.mtype = self.visit_expr_subtype(e.n_expr, param.mtype)
460 end
461
462 # Number of minimum mandatory remaining parameters
463 var min_arity = 0
464
465 # Second, associate remaining parameters
466 var vararg_decl = args.length - msignature.arity
467 var j = 0
468 for i in [0..msignature.arity[ do
469 # Skip parameters associated by name
470 if map.map.has_key(i) then continue
471
472 var param = msignature.mparameters[i]
473
474 # Search the next free argument: skip named arguments since they are already associated
475 while j < nbargs and args[j] isa ANamedargExpr do j += 1
476 if j >= nbargs then
477 if not param.mtype isa MNullableType then
478 min_arity = j + 1
479 end
480 j += 1
481 continue
482 end
483 var arg = args[j]
484 map.map[i] = j
485 j += 1
486
487 if i == vararg_rank then
488 j += vararg_decl
489 continue # skip the vararg
490 end
491
492 var paramtype = param.mtype
493 self.visit_expr_subtype(arg, paramtype)
494 end
495
496 if min_arity > 0 then
497 if last_is_padded then min_arity += 1
498 if min_arity < msignature.arity then
499 modelbuilder.error(node, "Error: expected at least {min_arity} argument(s) for `{mproperty}{msignature}`; got {args.length}. See introduction at `{mproperty.full_name}`.")
500 else
501 modelbuilder.error(node, "Error: expected {min_arity} argument(s) for `{mproperty}{msignature}`; got {args.length}. See introduction at `{mproperty.full_name}`.")
502 end
503 return null
504 end
505
506 # Third, check varargs
507 if vararg_rank >= 0 then
508 var paramtype = msignature.mparameters[vararg_rank].mtype
509 var first = args[vararg_rank]
510 if vararg_decl == 0 then
511 var mclass = get_mclass(node, "Array")
512 if mclass == null then return null # Forward error
513 var array_mtype = mclass.get_mtype([paramtype])
514 if first isa AVarargExpr then
515 self.visit_expr_subtype(first.n_expr, array_mtype)
516 first.mtype = first.n_expr.mtype
517 else
518 # only one vararg, maybe `...` was forgot, so be gentle!
519 var t = visit_expr(first)
520 if t == null then return null # Forward error
521 if not is_subtype(t, paramtype) and is_subtype(t, array_mtype) then
522 # Not acceptable but could be a `...`
523 error(first, "Type Error: expected `{paramtype}`, got `{t}`. Is an ellipsis `...` missing on the argument?")
524 return null
525 end
526 # Standard valid vararg, finish the job
527 map.vararg_decl = 1
528 self.visit_expr_subtype(first, paramtype)
529 end
530 else
531 map.vararg_decl = vararg_decl + 1
532 for i in [vararg_rank..vararg_rank+vararg_decl] do
533 self.visit_expr_subtype(args[i], paramtype)
534 end
535 end
536 end
537
538 return map
539 end
540
541 fun error(node: ANode, message: String)
542 do
543 self.modelbuilder.error(node, message)
544 end
545
546 fun get_variable(node: AExpr, variable: Variable): nullable MType
547 do
548 if not variable.is_adapted then return variable.declared_type
549
550 var flow = node.after_flow_context
551 if flow == null then return null # skip error
552
553 if flow.vars.has_key(variable) then
554 return flow.vars[variable]
555 else
556 #node.debug("*** START Collected for {variable}")
557 var mtypes = flow.collect_types(variable)
558 #node.debug("**** END Collected for {variable}")
559 if mtypes.length == 0 then
560 return variable.declared_type
561 else if mtypes.length == 1 then
562 return mtypes.first
563 else
564 var res = merge_types(node,mtypes)
565 if res == null then res = variable.declared_type
566 return res
567 end
568 end
569 end
570
571 # Some variables where type-adapted during the visit
572 var dirty = false
573
574 # Some loops had been visited during the visit
575 var has_loop = false
576
577 fun set_variable(node: AExpr, variable: Variable, mtype: nullable MType)
578 do
579 var flow = node.after_flow_context
580 assert flow != null
581
582 flow.set_var(self, variable, mtype)
583 end
584
585 fun merge_types(node: ANode, col: Array[nullable MType]): nullable MType
586 do
587 if col.length == 1 then return col.first
588 for t1 in col do
589 if t1 == null then continue # return null
590 var found = true
591 for t2 in col do
592 if t2 == null then continue # return null
593 if can_be_null(t2) and not can_be_null(t1) then
594 t1 = t1.as_nullable
595 end
596 if not is_subtype(t2, t1) then found = false
597 end
598 if found then
599 #print "merge {col.join(" ")} -> {t1}"
600 return t1
601 end
602 end
603 #self.modelbuilder.warning(node, "Type Error: {col.length} conflicting types: <{col.join(", ")}>")
604 return null
605 end
606 end
607
608 # Mapping between parameters and arguments in a call.
609 #
610 # Parameters and arguments are not stored in the class but referenced by their position (starting from 0)
611 #
612 # The point of this class is to help engine and other things to map arguments in the AST to parameters of the model.
613 class SignatureMap
614 # Associate a parameter to an argument
615 var map = new ArrayMap[Int, Int]
616
617 # The length of the vararg sequence
618 # 0 if no vararg or if reverse vararg (cf `AVarargExpr`)
619 var vararg_decl: Int = 0
620 end
621
622 # A specific method call site with its associated informations.
623 class CallSite
624 super MEntity
625
626 # The associated location of the callsite
627 var location: Location
628
629 # The static type of the receiver (possibly unresolved)
630 var recv: MType
631
632 # The module where the callsite is present
633 var mmodule: MModule
634
635 # The anchor to use with `recv` or `msignature`
636 var anchor: nullable MClassType
637
638 # Is the receiver self?
639 # If "for_self", virtual types of the signature are kept
640 # If "not_for_self", virtual type are erased
641 var recv_is_self: Bool
642
643 # The designated method
644 var mproperty: MMethod
645
646 # The statically designated method definition
647 # The most specif one, it is.
648 var mpropdef: MMethodDef
649
650 # The resolved signature for the receiver
651 var msignature: MSignature
652
653 # Is a implicit cast required on erasure typing policy?
654 var erasure_cast: Bool
655
656 # The mapping used on the call to associate arguments to parameters
657 # If null then no specific association is required.
658 var signaturemap: nullable SignatureMap = null
659
660 private fun check_signature(v: TypeVisitor, node: ANode, args: Array[AExpr]): Bool
661 do
662 var map = v.check_signature(node, args, self.mproperty, self.msignature)
663 signaturemap = map
664 if map == null then is_broken = true
665 return map == null
666 end
667 end
668
669 redef class Variable
670 # The declared type of the variable
671 var declared_type: nullable MType is writable
672
673 # Was the variable type-adapted?
674 # This is used to speedup type retrieval while it remains `false`
675 private var is_adapted = false
676 end
677
678 redef class FlowContext
679 # Store changes of types because of type evolution
680 private var vars = new HashMap[Variable, nullable MType]
681
682 # Adapt the variable to a static type
683 # Warning1: do not modify vars directly.
684 # Warning2: sub-flow may have cached a unadapted variable
685 private fun set_var(v: TypeVisitor, variable: Variable, mtype: nullable MType)
686 do
687 if variable.declared_type == mtype and not variable.is_adapted then return
688 if vars.has_key(variable) and vars[variable] == mtype then return
689 self.vars[variable] = mtype
690 v.dirty = true
691 variable.is_adapted = true
692 #node.debug "set {variable} to {mtype or else "X"}"
693 end
694
695 # Look in the flow and previous flow and collect all first reachable type adaptation of a local variable
696 private fun collect_types(variable: Variable): Array[nullable MType]
697 do
698 #node.debug "flow for {variable}"
699 var res = new Array[nullable MType]
700
701 var todo = [self]
702 var seen = new HashSet[FlowContext]
703 while not todo.is_empty do
704 var f = todo.pop
705 if f.is_unreachable then continue
706 if seen.has(f) then continue
707 seen.add f
708
709 if f.vars.has_key(variable) then
710 # Found something. Collect it and do not process further on this path
711 res.add f.vars[variable]
712 #f.node.debug "process {variable}: got {f.vars[variable] or else "X"}"
713 else
714 todo.add_all f.previous
715 todo.add_all f.loops
716 if f.previous.is_empty then
717 # Root flowcontext mean a parameter or something related
718 res.add variable.declared_type
719 #f.node.debug "root process {variable}: got {variable.declared_type or else "X"}"
720 end
721 end
722 end
723 #self.node.debug "##### end flow for {variable}: {res.join(" ")}"
724 return res
725 end
726 end
727
728 redef class APropdef
729 # The entry point of the whole typing analysis
730 fun do_typing(modelbuilder: ModelBuilder)
731 do
732 end
733
734 # The variable associated to the receiver (if any)
735 var selfvariable: nullable Variable
736 end
737
738 redef class AMethPropdef
739 redef fun do_typing(modelbuilder: ModelBuilder)
740 do
741 var mpropdef = self.mpropdef
742 if mpropdef == null then return # skip error
743
744 var v = new TypeVisitor(modelbuilder, mpropdef.mclassdef.mmodule, mpropdef)
745 self.selfvariable = v.selfvariable
746
747 var mmethoddef = self.mpropdef.as(not null)
748 var msignature = mmethoddef.msignature
749 if msignature == null then return # skip error
750 for i in [0..msignature.arity[ do
751 var mtype = msignature.mparameters[i].mtype
752 if msignature.vararg_rank == i then
753 var arrayclass = v.get_mclass(self.n_signature.n_params[i], "Array")
754 if arrayclass == null then return # Skip error
755 mtype = arrayclass.get_mtype([mtype])
756 end
757 var variable = self.n_signature.n_params[i].variable
758 assert variable != null
759 variable.declared_type = mtype
760 end
761
762 var nblock = self.n_block
763 if nblock == null then return
764
765 loop
766 v.dirty = false
767 v.visit_stmt(nblock)
768 if not v.has_loop or not v.dirty then break
769 end
770
771 var post_visitor = new PostTypingVisitor(v)
772 post_visitor.enter_visit(self)
773
774 if not nblock.after_flow_context.is_unreachable and msignature.return_mtype != null then
775 # We reach the end of the function without having a return, it is bad
776 v.error(self, "Error: reached end of function; expected `return` with a value.")
777 end
778 end
779 end
780
781 private class PostTypingVisitor
782 super Visitor
783 var type_visitor: TypeVisitor
784 redef fun visit(n) do
785 n.visit_all(self)
786 n.accept_post_typing(type_visitor)
787 if n isa AExpr and n.mtype == null and not n.is_typed then
788 n.is_broken = true
789 end
790 end
791 end
792
793 redef class ANode
794 private fun accept_post_typing(v: TypeVisitor) do end
795 end
796
797 redef class AAttrPropdef
798 redef fun do_typing(modelbuilder: ModelBuilder)
799 do
800 if not has_value then return
801
802 var mpropdef = self.mreadpropdef
803 if mpropdef == null or mpropdef.msignature == null then return # skip error
804
805 var v = new TypeVisitor(modelbuilder, mpropdef.mclassdef.mmodule, mpropdef)
806 self.selfvariable = v.selfvariable
807
808 var nexpr = self.n_expr
809 if nexpr != null then
810 var mtype = self.mtype
811 v.visit_expr_subtype(nexpr, mtype)
812 end
813 var nblock = self.n_block
814 if nblock != null then
815 v.visit_stmt(nblock)
816 if not nblock.after_flow_context.is_unreachable then
817 # We reach the end of the init without having a return, it is bad
818 v.error(self, "Error: reached end of block; expected `return`.")
819 end
820 end
821 end
822 end
823
824 ###
825
826 redef class AExpr
827 # The static type of the expression.
828 # null if self is a statement or in case of error
829 var mtype: nullable MType = null
830
831 # Is the statement correctly typed?
832 # Used to distinguish errors and statements when `mtype == null`
833 var is_typed: Bool = false
834
835 # If required, the following implicit cast `.as(XXX)`
836 # Such a cast may by required after evaluating the expression when
837 # a unsafe operation is detected (silently accepted by the Nit language).
838 # The attribute is computed by `check_subtype`
839 var implicit_cast_to: nullable MType = null
840
841 # Return the variable read (if any)
842 # Used to perform adaptive typing
843 fun its_variable: nullable Variable do return null
844
845 private fun accept_typing(v: TypeVisitor)
846 do
847 v.error(self, "no implemented accept_typing for {self.class_name}")
848 end
849
850 # Is non-null if `self` is a leaf of a comprehension array construction.
851 # In this case, the enclosing literal array node is designated.
852 # The result of the evaluation of `self` must be
853 # stored inside the designated array (there is an implicit `push`)
854 var comprehension: nullable AArrayExpr = null
855 end
856
857 redef class ABlockExpr
858 redef fun accept_typing(v)
859 do
860 for e in self.n_expr do v.visit_stmt(e)
861 self.is_typed = true
862 end
863
864 # The type of a blockexpr is the one of the last expression (or null if empty)
865 redef fun mtype
866 do
867 if self.n_expr.is_empty then return null
868 return self.n_expr.last.mtype
869 end
870 end
871
872 redef class AVardeclExpr
873 redef fun accept_typing(v)
874 do
875 var variable = self.variable
876 if variable == null then return # Skip error
877
878 var ntype = self.n_type
879 var mtype: nullable MType
880 if ntype == null then
881 mtype = null
882 else
883 mtype = v.resolve_mtype(ntype)
884 if mtype == null then return # Skip error
885 end
886
887 var nexpr = self.n_expr
888 if nexpr != null then
889 if mtype != null then
890 var etype = v.visit_expr_subtype(nexpr, mtype)
891 if etype == mtype then
892 assert ntype != null
893 v.modelbuilder.advice(ntype, "useless-type", "Warning: useless type definition for variable `{variable.name}`")
894 end
895 else
896 mtype = v.visit_expr(nexpr)
897 if mtype == null then return # Skip error
898 end
899 end
900
901 var decltype = mtype
902 if mtype == null or mtype isa MNullType then
903 var objclass = v.get_mclass(self, "Object")
904 if objclass == null then return # skip error
905 decltype = objclass.mclass_type.as_nullable
906 if mtype == null then mtype = decltype
907 end
908
909 variable.declared_type = decltype
910 v.set_variable(self, variable, mtype)
911
912 #debug("var {variable}: {mtype}")
913
914 self.mtype = mtype
915 self.is_typed = true
916 end
917 end
918
919 redef class AVarExpr
920 redef fun its_variable do return self.variable
921 redef fun accept_typing(v)
922 do
923 var variable = self.variable
924 if variable == null then return # Skip error
925
926 var mtype = v.get_variable(self, variable)
927 if mtype != null then
928 #debug("{variable} is {mtype}")
929 else
930 #debug("{variable} is untyped")
931 end
932
933 self.mtype = mtype
934 end
935 end
936
937 redef class AVarAssignExpr
938 redef fun accept_typing(v)
939 do
940 var variable = self.variable
941 assert variable != null
942
943 var mtype = v.visit_expr_subtype(n_value, variable.declared_type)
944
945 v.set_variable(self, variable, mtype)
946
947 self.is_typed = true
948 end
949 end
950
951 redef class AReassignFormExpr
952 # The method designed by the reassign operator.
953 var reassign_callsite: nullable CallSite
954
955 var read_type: nullable MType = null
956
957 # Determine the `reassign_property`
958 # `readtype` is the type of the reading of the left value.
959 # `writetype` is the type of the writing of the left value.
960 # (Because of `ACallReassignExpr`, both can be different.
961 # Return the static type of the value to store.
962 private fun resolve_reassignment(v: TypeVisitor, readtype, writetype: MType): nullable MType
963 do
964 var reassign_name = self.n_assign_op.operator
965
966 self.read_type = readtype
967
968 var callsite = v.get_method(self.n_assign_op, readtype, reassign_name, false)
969 if callsite == null then return null # Skip error
970 self.reassign_callsite = callsite
971
972 var msignature = callsite.msignature
973 var rettype = msignature.return_mtype
974 assert msignature.arity == 1 and rettype != null
975
976 var value_type = v.visit_expr_subtype(self.n_value, msignature.mparameters.first.mtype)
977 if value_type == null then return null # Skip error
978
979 v.check_subtype(self, rettype, writetype, false)
980 return rettype
981 end
982 end
983
984 redef class AVarReassignExpr
985 redef fun accept_typing(v)
986 do
987 var variable = self.variable
988 assert variable != null
989
990 var readtype = v.get_variable(self, variable)
991 if readtype == null then return
992
993 read_type = readtype
994
995 var writetype = variable.declared_type
996 if writetype == null then return
997
998 var rettype = self.resolve_reassignment(v, readtype, writetype)
999
1000 v.set_variable(self, variable, rettype)
1001
1002 self.is_typed = rettype != null
1003 end
1004 end
1005
1006
1007 redef class AContinueExpr
1008 redef fun accept_typing(v)
1009 do
1010 var nexpr = self.n_expr
1011 if nexpr != null then
1012 v.visit_expr(nexpr)
1013 end
1014 self.is_typed = true
1015 end
1016 end
1017
1018 redef class ABreakExpr
1019 redef fun accept_typing(v)
1020 do
1021 var nexpr = self.n_expr
1022 if nexpr != null then
1023 v.visit_expr(nexpr)
1024 end
1025 self.is_typed = true
1026 end
1027 end
1028
1029 redef class AReturnExpr
1030 redef fun accept_typing(v)
1031 do
1032 var nexpr = self.n_expr
1033 var ret_type
1034 var mpropdef = v.mpropdef
1035 if mpropdef isa MMethodDef then
1036 ret_type = mpropdef.msignature.return_mtype
1037 else if mpropdef isa MAttributeDef then
1038 ret_type = mpropdef.static_mtype
1039 else
1040 abort
1041 end
1042 if nexpr != null then
1043 if ret_type != null then
1044 v.visit_expr_subtype(nexpr, ret_type)
1045 else
1046 v.visit_expr(nexpr)
1047 v.error(nexpr, "Error: `return` with value in a procedure.")
1048 return
1049 end
1050 else if ret_type != null then
1051 v.error(self, "Error: `return` without value in a function.")
1052 return
1053 end
1054 self.is_typed = true
1055 end
1056 end
1057
1058 redef class AAbortExpr
1059 redef fun accept_typing(v)
1060 do
1061 self.is_typed = true
1062 end
1063 end
1064
1065 redef class AIfExpr
1066 redef fun accept_typing(v)
1067 do
1068 v.visit_expr_bool(n_expr)
1069
1070 v.visit_stmt(n_then)
1071 v.visit_stmt(n_else)
1072
1073 self.is_typed = true
1074
1075 if n_then != null and n_else == null then
1076 self.mtype = n_then.mtype
1077 end
1078 end
1079 end
1080
1081 redef class AIfexprExpr
1082 redef fun accept_typing(v)
1083 do
1084 v.visit_expr_bool(n_expr)
1085
1086 var t1 = v.visit_expr(n_then)
1087 var t2 = v.visit_expr(n_else)
1088
1089 if t1 == null or t2 == null then
1090 return # Skip error
1091 end
1092
1093 var t = v.merge_types(self, [t1, t2])
1094 if t == null then
1095 v.error(self, "Type Error: ambiguous type `{t1}` vs `{t2}`.")
1096 end
1097 self.mtype = t
1098 end
1099 end
1100
1101 redef class ADoExpr
1102 redef fun accept_typing(v)
1103 do
1104 v.visit_stmt(n_block)
1105 self.is_typed = true
1106 end
1107 end
1108
1109 redef class AWhileExpr
1110 redef fun accept_typing(v)
1111 do
1112 v.has_loop = true
1113 v.visit_expr_bool(n_expr)
1114 v.visit_stmt(n_block)
1115 self.is_typed = true
1116 end
1117 end
1118
1119 redef class ALoopExpr
1120 redef fun accept_typing(v)
1121 do
1122 v.has_loop = true
1123 v.visit_stmt(n_block)
1124 self.is_typed = true
1125 end
1126 end
1127
1128 redef class AForExpr
1129 var coltype: nullable MClassType
1130
1131 var method_iterator: nullable CallSite
1132 var method_is_ok: nullable CallSite
1133 var method_item: nullable CallSite
1134 var method_next: nullable CallSite
1135 var method_key: nullable CallSite
1136 var method_finish: nullable CallSite
1137
1138 var method_lt: nullable CallSite
1139 var method_successor: nullable CallSite
1140
1141 private fun do_type_iterator(v: TypeVisitor, mtype: MType)
1142 do
1143 if mtype isa MNullType then
1144 v.error(self, "Type Error: `for` cannot iterate over `null`.")
1145 return
1146 end
1147
1148 # get obj class
1149 var objcla = v.get_mclass(self, "Object")
1150 if objcla == null then return
1151
1152 # check iterator method
1153 var itdef = v.get_method(self, mtype, "iterator", n_expr isa ASelfExpr)
1154 if itdef == null then
1155 v.error(self, "Type Error: `for` expects a type providing an `iterator` method, got `{mtype}`.")
1156 return
1157 end
1158 self.method_iterator = itdef
1159
1160 # check that iterator return something
1161 var ittype = itdef.msignature.return_mtype
1162 if ittype == null then
1163 v.error(self, "Type Error: `for` expects the method `iterator` to return an `Iterator` or `MapIterator` type.")
1164 return
1165 end
1166
1167 # get iterator type
1168 var colit_cla = v.try_get_mclass(self, "Iterator")
1169 var mapit_cla = v.try_get_mclass(self, "MapIterator")
1170 var is_col = false
1171 var is_map = false
1172
1173 if colit_cla != null and v.is_subtype(ittype, colit_cla.get_mtype([objcla.mclass_type.as_nullable])) then
1174 # Iterator
1175 var coltype = ittype.supertype_to(v.mmodule, v.anchor, colit_cla)
1176 var variables = self.variables
1177 if variables.length != 1 then
1178 v.error(self, "Type Error: `for` expects only one variable when using `Iterator`.")
1179 else
1180 variables.first.declared_type = coltype.arguments.first
1181 end
1182 is_col = true
1183 end
1184
1185 if mapit_cla != null and v.is_subtype(ittype, mapit_cla.get_mtype([objcla.mclass_type.as_nullable, objcla.mclass_type.as_nullable])) then
1186 # Map Iterator
1187 var coltype = ittype.supertype_to(v.mmodule, v.anchor, mapit_cla)
1188 var variables = self.variables
1189 if variables.length != 2 then
1190 v.error(self, "Type Error: `for` expects two variables when using `MapIterator`.")
1191 else
1192 variables[0].declared_type = coltype.arguments[0]
1193 variables[1].declared_type = coltype.arguments[1]
1194 end
1195 is_map = true
1196 end
1197
1198 if not is_col and not is_map then
1199 v.error(self, "Type Error: `for` expects the method `iterator` to return an `Iterator` or `MapIterator` type.")
1200 return
1201 end
1202
1203 # anchor formal and virtual types
1204 if mtype.need_anchor then mtype = v.anchor_to(mtype)
1205
1206 mtype = mtype.undecorate
1207 self.coltype = mtype.as(MClassType)
1208
1209 # get methods is_ok, next, item
1210 var ikdef = v.get_method(self, ittype, "is_ok", false)
1211 if ikdef == null then
1212 v.error(self, "Type Error: `for` expects a method `is_ok` in type `{ittype}`.")
1213 return
1214 end
1215 self.method_is_ok = ikdef
1216
1217 var itemdef = v.get_method(self, ittype, "item", false)
1218 if itemdef == null then
1219 v.error(self, "Type Error: `for` expects a method `item` in type `{ittype}`.")
1220 return
1221 end
1222 self.method_item = itemdef
1223
1224 var nextdef = v.get_method(self, ittype, "next", false)
1225 if nextdef == null then
1226 v.error(self, "Type Error: `for` expects a method `next` in type {ittype}.")
1227 return
1228 end
1229 self.method_next = nextdef
1230
1231 self.method_finish = v.try_get_method(self, ittype, "finish", false)
1232
1233 if is_map then
1234 var keydef = v.get_method(self, ittype, "key", false)
1235 if keydef == null then
1236 v.error(self, "Type Error: `for` expects a method `key` in type `{ittype}`.")
1237 return
1238 end
1239 self.method_key = keydef
1240 end
1241
1242 if self.variables.length == 1 and n_expr isa ARangeExpr then
1243 var variable = variables.first
1244 var vtype = variable.declared_type.as(not null)
1245
1246 if n_expr isa AOrangeExpr then
1247 self.method_lt = v.get_method(self, vtype, "<", false)
1248 else
1249 self.method_lt = v.get_method(self, vtype, "<=", false)
1250 end
1251
1252 self.method_successor = v.get_method(self, vtype, "successor", false)
1253 end
1254 end
1255
1256 redef fun accept_typing(v)
1257 do
1258 v.has_loop = true
1259 var mtype = v.visit_expr(n_expr)
1260 if mtype == null then return
1261
1262 self.do_type_iterator(v, mtype)
1263
1264 v.visit_stmt(n_block)
1265
1266 self.mtype = n_block.mtype
1267 self.is_typed = true
1268 end
1269 end
1270
1271 redef class AWithExpr
1272 var method_start: nullable CallSite
1273 var method_finish: nullable CallSite
1274
1275 redef fun accept_typing(v: TypeVisitor)
1276 do
1277 var mtype = v.visit_expr(n_expr)
1278 if mtype == null then return
1279
1280 method_start = v.get_method(self, mtype, "start", n_expr isa ASelfExpr)
1281 method_finish = v.get_method(self, mtype, "finish", n_expr isa ASelfExpr)
1282
1283 v.visit_stmt(n_block)
1284 self.mtype = n_block.mtype
1285 self.is_typed = true
1286 end
1287 end
1288
1289 redef class AAssertExpr
1290 redef fun accept_typing(v)
1291 do
1292 v.visit_expr_bool(n_expr)
1293
1294 v.visit_stmt(n_else)
1295 self.is_typed = true
1296 end
1297 end
1298
1299 redef class AOrExpr
1300 redef fun accept_typing(v)
1301 do
1302 v.visit_expr_bool(n_expr)
1303 v.visit_expr_bool(n_expr2)
1304 self.mtype = v.type_bool(self)
1305 end
1306 end
1307
1308 redef class AImpliesExpr
1309 redef fun accept_typing(v)
1310 do
1311 v.visit_expr_bool(n_expr)
1312 v.visit_expr_bool(n_expr2)
1313 self.mtype = v.type_bool(self)
1314 end
1315 end
1316
1317 redef class AAndExpr
1318 redef fun accept_typing(v)
1319 do
1320 v.visit_expr_bool(n_expr)
1321 v.visit_expr_bool(n_expr2)
1322 self.mtype = v.type_bool(self)
1323 end
1324 end
1325
1326
1327 redef class ANotExpr
1328 redef fun accept_typing(v)
1329 do
1330 v.visit_expr_bool(n_expr)
1331 self.mtype = v.type_bool(self)
1332 end
1333 end
1334
1335 redef class AOrElseExpr
1336 redef fun accept_typing(v)
1337 do
1338 var t1 = v.visit_expr(n_expr)
1339 var t2 = v.visit_expr(n_expr2)
1340
1341 if t1 == null or t2 == null then
1342 return # Skip error
1343 end
1344
1345 if t1 isa MNullType then
1346 self.mtype = t2
1347 return
1348 else if v.can_be_null(t1) then
1349 t1 = t1.as_notnull
1350 end
1351
1352 var t = v.merge_types(self, [t1, t2])
1353 if t == null then
1354 var c = v.get_mclass(self, "Object")
1355 if c == null then return # forward error
1356 t = c.mclass_type
1357 if v.can_be_null(t2) then
1358 t = t.as_nullable
1359 end
1360 #v.error(self, "Type Error: ambiguous type {t1} vs {t2}")
1361 end
1362 self.mtype = t
1363 end
1364
1365 redef fun accept_post_typing(v)
1366 do
1367 var t1 = n_expr.mtype
1368 if t1 == null then
1369 return
1370 else
1371 v.check_can_be_null(n_expr, t1)
1372 end
1373 end
1374 end
1375
1376 redef class ATrueExpr
1377 redef fun accept_typing(v)
1378 do
1379 self.mtype = v.type_bool(self)
1380 end
1381 end
1382
1383 redef class AFalseExpr
1384 redef fun accept_typing(v)
1385 do
1386 self.mtype = v.type_bool(self)
1387 end
1388 end
1389
1390 redef class AIntegerExpr
1391 redef fun accept_typing(v)
1392 do
1393 var mclass: nullable MClass = null
1394 if value isa Byte then
1395 mclass = v.get_mclass(self, "Byte")
1396 else if value isa Int then
1397 mclass = v.get_mclass(self, "Int")
1398 else if value isa Int8 then
1399 mclass = v.get_mclass(self, "Int8")
1400 else if value isa Int16 then
1401 mclass = v.get_mclass(self, "Int16")
1402 else if value isa UInt16 then
1403 mclass = v.get_mclass(self, "UInt16")
1404 else if value isa Int32 then
1405 mclass = v.get_mclass(self, "Int32")
1406 else if value isa UInt32 then
1407 mclass = v.get_mclass(self, "UInt32")
1408 end
1409 if mclass == null then return # Forward error
1410 self.mtype = mclass.mclass_type
1411 end
1412 end
1413
1414 redef class AFloatExpr
1415 redef fun accept_typing(v)
1416 do
1417 var mclass = v.get_mclass(self, "Float")
1418 if mclass == null then return # Forward error
1419 self.mtype = mclass.mclass_type
1420 end
1421 end
1422
1423 redef class ACharExpr
1424 redef fun accept_typing(v)
1425 do
1426 var mclass = v.get_mclass(self, "Char")
1427 if mclass == null then return # Forward error
1428 self.mtype = mclass.mclass_type
1429 end
1430 end
1431
1432 redef class AStringFormExpr
1433 redef fun accept_typing(v)
1434 do
1435 var mclass = v.get_mclass(self, "String")
1436 if mclass == null then return # Forward error
1437 self.mtype = mclass.mclass_type
1438 end
1439 end
1440
1441 redef class ASuperstringExpr
1442 redef fun accept_typing(v)
1443 do
1444 var mclass = v.get_mclass(self, "String")
1445 if mclass == null then return # Forward error
1446 self.mtype = mclass.mclass_type
1447 var objclass = v.get_mclass(self, "Object")
1448 if objclass == null then return # Forward error
1449 var objtype = objclass.mclass_type
1450 for nexpr in self.n_exprs do
1451 v.visit_expr_subtype(nexpr, objtype)
1452 end
1453 end
1454 end
1455
1456 redef class AArrayExpr
1457 # The `with_capacity` method on Array
1458 var with_capacity_callsite: nullable CallSite
1459
1460 # The `push` method on arrays
1461 var push_callsite: nullable CallSite
1462
1463 # The element of each type
1464 var element_mtype: nullable MType
1465
1466 # Set that `self` is a part of comprehension array `na`
1467 # If `self` is a `for`, or a `if`, then `set_comprehension` is recursively applied.
1468 private fun set_comprehension(n: nullable AExpr)
1469 do
1470 if n == null then
1471 return
1472 else if n isa AForExpr then
1473 set_comprehension(n.n_block)
1474 else if n isa AIfExpr then
1475 set_comprehension(n.n_then)
1476 set_comprehension(n.n_else)
1477 else
1478 # is a leave
1479 n.comprehension = self
1480 end
1481 end
1482 redef fun accept_typing(v)
1483 do
1484 var mtype: nullable MType = null
1485 var ntype = self.n_type
1486 if ntype != null then
1487 mtype = v.resolve_mtype(ntype)
1488 if mtype == null then return # Skip error
1489 end
1490 var mtypes = new Array[nullable MType]
1491 var useless = false
1492 for e in self.n_exprs do
1493 var t = v.visit_expr(e)
1494 if t == null then
1495 return # Skip error
1496 end
1497 set_comprehension(e)
1498 if mtype != null then
1499 if v.check_subtype(e, t, mtype, false) == null then return # Forward error
1500 if t == mtype then useless = true
1501 else
1502 mtypes.add(t)
1503 end
1504 end
1505 if mtype == null then
1506 # Ensure monotony for type adaptation on loops
1507 if self.element_mtype != null then mtypes.add self.element_mtype
1508 mtype = v.merge_types(self, mtypes)
1509 end
1510 if mtype == null or mtype isa MNullType then
1511 v.error(self, "Type Error: ambiguous array type {mtypes.join(" ")}")
1512 return
1513 end
1514 if useless then
1515 assert ntype != null
1516 v.modelbuilder.warning(ntype, "useless-type", "Warning: useless type declaration `{mtype}` in literal Array since it can be inferred from the elements type.")
1517 end
1518
1519 self.element_mtype = mtype
1520
1521 var mclass = v.get_mclass(self, "Array")
1522 if mclass == null then return # Forward error
1523 var array_mtype = mclass.get_mtype([mtype])
1524
1525 with_capacity_callsite = v.get_method(self, array_mtype, "with_capacity", false)
1526 push_callsite = v.get_method(self, array_mtype, "push", false)
1527
1528 self.mtype = array_mtype
1529 end
1530 end
1531
1532 redef class ARangeExpr
1533 var init_callsite: nullable CallSite
1534
1535 redef fun accept_typing(v)
1536 do
1537 var discrete_class = v.get_mclass(self, "Discrete")
1538 if discrete_class == null then return # Forward error
1539 var discrete_type = discrete_class.intro.bound_mtype
1540 var t1 = v.visit_expr_subtype(self.n_expr, discrete_type)
1541 var t2 = v.visit_expr_subtype(self.n_expr2, discrete_type)
1542 if t1 == null or t2 == null then return
1543 var mclass = v.get_mclass(self, "Range")
1544 if mclass == null then return # Forward error
1545 var mtype
1546 if v.is_subtype(t1, t2) then
1547 mtype = mclass.get_mtype([t2])
1548 else if v.is_subtype(t2, t1) then
1549 mtype = mclass.get_mtype([t1])
1550 else
1551 v.error(self, "Type Error: cannot create range: `{t1}` vs `{t2}`.")
1552 return
1553 end
1554
1555 self.mtype = mtype
1556
1557 # get the constructor
1558 var callsite
1559 if self isa ACrangeExpr then
1560 callsite = v.get_method(self, mtype, "init", false)
1561 else if self isa AOrangeExpr then
1562 callsite = v.get_method(self, mtype, "without_last", false)
1563 else
1564 abort
1565 end
1566 init_callsite = callsite
1567 end
1568 end
1569
1570 redef class ANullExpr
1571 redef fun accept_typing(v)
1572 do
1573 self.mtype = v.mmodule.model.null_type
1574 end
1575 end
1576
1577 redef class AIsaExpr
1578 # The static type to cast to.
1579 # (different from the static type of the expression that is `Bool`).
1580 var cast_type: nullable MType
1581 redef fun accept_typing(v)
1582 do
1583 v.visit_expr(n_expr)
1584
1585 var mtype = v.resolve_mtype(n_type)
1586
1587 self.cast_type = mtype
1588
1589 var variable = self.n_expr.its_variable
1590 if variable != null then
1591 #var orig = self.n_expr.mtype
1592 #var from = if orig != null then orig.to_s else "invalid"
1593 #var to = if mtype != null then mtype.to_s else "invalid"
1594 #debug("adapt {variable}: {from} -> {to}")
1595 self.after_flow_context.when_true.set_var(v, variable, mtype)
1596 end
1597
1598 self.mtype = v.type_bool(self)
1599 end
1600
1601 redef fun accept_post_typing(v)
1602 do
1603 v.check_expr_cast(self, self.n_expr, self.n_type)
1604 end
1605 end
1606
1607 redef class AAsCastExpr
1608 redef fun accept_typing(v)
1609 do
1610 v.visit_expr(n_expr)
1611
1612 self.mtype = v.resolve_mtype(n_type)
1613 end
1614
1615 redef fun accept_post_typing(v)
1616 do
1617 v.check_expr_cast(self, self.n_expr, self.n_type)
1618 end
1619 end
1620
1621 redef class AAsNotnullExpr
1622 redef fun accept_typing(v)
1623 do
1624 var mtype = v.visit_expr(self.n_expr)
1625 if mtype == null then return # Forward error
1626
1627 if mtype isa MNullType then
1628 v.error(self, "Type Error: `as(not null)` on `null`.")
1629 return
1630 end
1631
1632 if v.can_be_null(mtype) then
1633 mtype = mtype.as_notnull
1634 end
1635
1636 self.mtype = mtype
1637 end
1638
1639 redef fun accept_post_typing(v)
1640 do
1641 var mtype = n_expr.mtype
1642 if mtype == null then return
1643 v.check_can_be_null(n_expr, mtype)
1644 end
1645 end
1646
1647 redef class AParExpr
1648 redef fun accept_typing(v)
1649 do
1650 self.mtype = v.visit_expr(self.n_expr)
1651 end
1652 end
1653
1654 redef class AOnceExpr
1655 redef fun accept_typing(v)
1656 do
1657 self.mtype = v.visit_expr(self.n_expr)
1658 end
1659 end
1660
1661 redef class ASelfExpr
1662 redef var its_variable: nullable Variable
1663 redef fun accept_typing(v)
1664 do
1665 if v.is_toplevel_context and not self isa AImplicitSelfExpr then
1666 v.error(self, "Error: `self` cannot be used in top-level method.")
1667 end
1668 var variable = v.selfvariable
1669 self.its_variable = variable
1670 self.mtype = v.get_variable(self, variable)
1671 end
1672 end
1673
1674 redef class AImplicitSelfExpr
1675 # Is the implicit receiver `sys`?
1676 #
1677 # By default, the implicit receiver is `self`.
1678 # But when there is not method for `self`, `sys` is used as a fall-back.
1679 # Is this case this flag is set to `true`.
1680 var is_sys = false
1681 end
1682
1683 ## MESSAGE SENDING AND PROPERTY
1684
1685 redef class ASendExpr
1686 # The property invoked by the send.
1687 var callsite: nullable CallSite
1688
1689 redef fun accept_typing(v)
1690 do
1691 var nrecv = self.n_expr
1692 var recvtype = v.visit_expr(nrecv)
1693 var name = self.property_name
1694 var node = self.property_node
1695
1696 if recvtype == null then return # Forward error
1697
1698 var callsite = null
1699 var unsafe_type = v.anchor_to(recvtype)
1700 var mproperty = v.try_get_mproperty_by_name2(node, unsafe_type, name)
1701 if mproperty == null and nrecv isa AImplicitSelfExpr then
1702 # Special fall-back search in `sys` when noting found in the implicit receiver.
1703 var sysclass = v.try_get_mclass(node, "Sys")
1704 if sysclass != null then
1705 var systype = sysclass.mclass_type
1706 mproperty = v.try_get_mproperty_by_name2(node, systype, name)
1707 if mproperty != null then
1708 callsite = v.get_method(node, systype, name, false)
1709 if callsite == null then return # Forward error
1710 # Update information, we are looking at `sys` now, not `self`
1711 nrecv.is_sys = true
1712 nrecv.its_variable = null
1713 nrecv.mtype = systype
1714 recvtype = systype
1715 end
1716 end
1717 end
1718 if callsite == null then
1719 # If still nothing, just exit
1720 callsite = v.get_method(node, recvtype, name, nrecv isa ASelfExpr)
1721 if callsite == null then return
1722 end
1723
1724 self.callsite = callsite
1725 var msignature = callsite.msignature
1726
1727 var args = compute_raw_arguments
1728
1729 callsite.check_signature(v, node, args)
1730
1731 if callsite.mproperty.is_init then
1732 var vmpropdef = v.mpropdef
1733 if not (vmpropdef isa MMethodDef and vmpropdef.mproperty.is_init) then
1734 v.error(node, "Error: an `init` can only be called from another `init`.")
1735 end
1736 if vmpropdef isa MMethodDef and vmpropdef.mproperty.is_root_init and not callsite.mproperty.is_root_init then
1737 v.error(node, "Error: `{vmpropdef}` cannot call a factory `{callsite.mproperty}`.")
1738 end
1739 end
1740
1741 var ret = msignature.return_mtype
1742 if ret != null then
1743 self.mtype = ret
1744 else
1745 self.is_typed = true
1746 end
1747 end
1748
1749 # The name of the property
1750 # Each subclass simply provide the correct name.
1751 private fun property_name: String is abstract
1752
1753 # The node identifying the name (id, operator, etc) for messages.
1754 #
1755 # Is `self` by default
1756 private fun property_node: ANode do return self
1757
1758 # An array of all arguments (excluding self)
1759 fun raw_arguments: Array[AExpr] do return compute_raw_arguments
1760
1761 private fun compute_raw_arguments: Array[AExpr] is abstract
1762 end
1763
1764 redef class ABinopExpr
1765 redef fun compute_raw_arguments do return [n_expr2]
1766 redef fun property_name do return operator
1767 redef fun property_node do return n_op
1768 end
1769
1770 redef class AEqFormExpr
1771 redef fun accept_typing(v)
1772 do
1773 super
1774 v.null_test(self)
1775 end
1776
1777 redef fun accept_post_typing(v)
1778 do
1779 var mtype = n_expr.mtype
1780 var mtype2 = n_expr2.mtype
1781
1782 if mtype == null or mtype2 == null then return
1783
1784 if not mtype2 isa MNullType then return
1785
1786 v.check_can_be_null(n_expr, mtype)
1787 end
1788 end
1789
1790 redef class AUnaryopExpr
1791 redef fun property_name do return "unary {operator}"
1792 redef fun compute_raw_arguments do return new Array[AExpr]
1793 end
1794
1795
1796 redef class ACallExpr
1797 redef fun property_name do return n_qid.n_id.text
1798 redef fun property_node do return n_qid
1799 redef fun compute_raw_arguments do return n_args.to_a
1800 end
1801
1802 redef class ACallAssignExpr
1803 redef fun property_name do return n_qid.n_id.text + "="
1804 redef fun property_node do return n_qid
1805 redef fun compute_raw_arguments
1806 do
1807 var res = n_args.to_a
1808 res.add(n_value)
1809 return res
1810 end
1811 end
1812
1813 redef class ABraExpr
1814 redef fun property_name do return "[]"
1815 redef fun compute_raw_arguments do return n_args.to_a
1816 end
1817
1818 redef class ABraAssignExpr
1819 redef fun property_name do return "[]="
1820 redef fun compute_raw_arguments
1821 do
1822 var res = n_args.to_a
1823 res.add(n_value)
1824 return res
1825 end
1826 end
1827
1828 redef class ASendReassignFormExpr
1829 # The property invoked for the writing
1830 var write_callsite: nullable CallSite
1831
1832 redef fun accept_typing(v)
1833 do
1834 var recvtype = v.visit_expr(self.n_expr)
1835 var name = self.property_name
1836 var node = self.property_node
1837
1838 if recvtype == null then return # Forward error
1839
1840 var for_self = self.n_expr isa ASelfExpr
1841 var callsite = v.get_method(node, recvtype, name, for_self)
1842
1843 if callsite == null then return
1844 self.callsite = callsite
1845
1846 var args = compute_raw_arguments
1847
1848 callsite.check_signature(v, node, args)
1849
1850 var readtype = callsite.msignature.return_mtype
1851 if readtype == null then
1852 v.error(node, "Error: `{name}` is not a function.")
1853 return
1854 end
1855
1856 var wcallsite = v.get_method(node, recvtype, name + "=", self.n_expr isa ASelfExpr)
1857 if wcallsite == null then return
1858 self.write_callsite = wcallsite
1859
1860 var wtype = self.resolve_reassignment(v, readtype, wcallsite.msignature.mparameters.last.mtype)
1861 if wtype == null then return
1862
1863 args = args.to_a # duplicate so raw_arguments keeps only the getter args
1864 args.add(self.n_value)
1865 wcallsite.check_signature(v, node, args)
1866
1867 self.is_typed = true
1868 end
1869 end
1870
1871 redef class ACallReassignExpr
1872 redef fun property_name do return n_qid.n_id.text
1873 redef fun property_node do return n_qid.n_id
1874 redef fun compute_raw_arguments do return n_args.to_a
1875 end
1876
1877 redef class ABraReassignExpr
1878 redef fun property_name do return "[]"
1879 redef fun compute_raw_arguments do return n_args.to_a
1880 end
1881
1882 redef class AInitExpr
1883 redef fun property_name do return "init"
1884 redef fun property_node do return n_kwinit
1885 redef fun compute_raw_arguments do return n_args.to_a
1886 end
1887
1888 redef class AExprs
1889 fun to_a: Array[AExpr] do return self.n_exprs.to_a
1890 end
1891
1892 ###
1893
1894 redef class ASuperExpr
1895 # The method to call if the super is in fact a 'super init call'
1896 # Note: if the super is a normal call-next-method, then this attribute is null
1897 var callsite: nullable CallSite
1898
1899 # The method to call is the super is a standard `call-next-method` super-call
1900 # Note: if the super is a special super-init-call, then this attribute is null
1901 var mpropdef: nullable MMethodDef
1902
1903 redef fun accept_typing(v)
1904 do
1905 var anchor = v.anchor
1906 assert anchor != null
1907 var recvtype = v.get_variable(self, v.selfvariable)
1908 assert recvtype != null
1909 var mproperty = v.mpropdef.mproperty
1910 if not mproperty isa MMethod then
1911 v.error(self, "Error: `super` only usable in a `method`.")
1912 return
1913 end
1914 var superprops = mproperty.lookup_super_definitions(v.mmodule, anchor)
1915 if superprops.length == 0 then
1916 if mproperty.is_init and v.mpropdef.is_intro then
1917 process_superinit(v)
1918 return
1919 end
1920 v.error(self, "Error: no super method to call for `{mproperty}`.")
1921 return
1922 end
1923 # FIXME: covariance of return type in linear extension?
1924 var superprop = superprops.first
1925
1926 var msignature = superprop.msignature.as(not null)
1927 msignature = v.resolve_for(msignature, recvtype, true).as(MSignature)
1928 var args = self.n_args.to_a
1929 if args.length > 0 then
1930 signaturemap = v.check_signature(self, args, mproperty, msignature)
1931 end
1932 self.mtype = msignature.return_mtype
1933 self.is_typed = true
1934 v.mpropdef.has_supercall = true
1935 mpropdef = v.mpropdef.as(MMethodDef)
1936 end
1937
1938 # The mapping used on the call to associate arguments to parameters.
1939 # If null then no specific association is required.
1940 var signaturemap: nullable SignatureMap
1941
1942 private fun process_superinit(v: TypeVisitor)
1943 do
1944 var anchor = v.anchor
1945 assert anchor != null
1946 var recvtype = v.get_variable(self, v.selfvariable)
1947 assert recvtype != null
1948 var mpropdef = v.mpropdef
1949 assert mpropdef isa MMethodDef
1950 var mproperty = mpropdef.mproperty
1951 var superprop: nullable MMethodDef = null
1952 for msupertype in mpropdef.mclassdef.supertypes do
1953 msupertype = msupertype.anchor_to(v.mmodule, anchor)
1954 var errcount = v.modelbuilder.toolcontext.error_count
1955 var candidate = v.try_get_mproperty_by_name2(self, msupertype, mproperty.name).as(nullable MMethod)
1956 if candidate == null then
1957 if v.modelbuilder.toolcontext.error_count > errcount then return # Forward error
1958 continue # Try next super-class
1959 end
1960 if superprop != null and candidate.is_root_init then
1961 continue
1962 end
1963 if superprop != null and superprop.mproperty != candidate and not superprop.mproperty.is_root_init then
1964 v.error(self, "Error: conflicting super constructor to call for `{mproperty}`: `{candidate.full_name}`, `{superprop.mproperty.full_name}`")
1965 return
1966 end
1967 var candidatedefs = candidate.lookup_definitions(v.mmodule, anchor)
1968 if superprop != null and superprop.mproperty == candidate then
1969 if superprop == candidatedefs.first then continue
1970 candidatedefs.add(superprop)
1971 end
1972 if candidatedefs.length > 1 then
1973 v.error(self, "Error: conflicting property definitions for property `{mproperty}` in `{recvtype}`: {candidatedefs.join(", ")}")
1974 return
1975 end
1976 superprop = candidatedefs.first
1977 end
1978 if superprop == null then
1979 v.error(self, "Error: no super method to call for `{mproperty}`.")
1980 return
1981 end
1982
1983 var msignature = superprop.new_msignature or else superprop.msignature.as(not null)
1984 msignature = v.resolve_for(msignature, recvtype, true).as(MSignature)
1985
1986 var callsite = new CallSite(hot_location, recvtype, v.mmodule, v.anchor, true, superprop.mproperty, superprop, msignature, false)
1987 self.callsite = callsite
1988
1989 var args = self.n_args.to_a
1990 if args.length > 0 then
1991 callsite.check_signature(v, self, args)
1992 else
1993 # Check there is at least enough parameters
1994 if mpropdef.msignature.arity < msignature.arity then
1995 v.error(self, "Error: not enough implicit arguments to pass. Got `{mpropdef.msignature.arity}`, expected at least `{msignature.arity}`. Signature is `{msignature}`.")
1996 return
1997 end
1998 # Check that each needed parameter is conform
1999 var i = 0
2000 for sp in msignature.mparameters do
2001 var p = mpropdef.msignature.mparameters[i]
2002 if not v.is_subtype(p.mtype, sp.mtype) then
2003 v.error(self, "Type Error: expected argument #{i} of type `{sp.mtype}`, got implicit argument `{p.name}` of type `{p.mtype}`. Signature is {msignature}")
2004 return
2005 end
2006 i += 1
2007 end
2008 end
2009
2010 self.is_typed = true
2011 end
2012 end
2013
2014 ####
2015
2016 redef class ANewExpr
2017 # The constructor invoked by the new.
2018 var callsite: nullable CallSite
2019
2020 # The designated type
2021 var recvtype: nullable MClassType
2022
2023 redef fun accept_typing(v)
2024 do
2025 var recvtype = v.resolve_mtype(self.n_type)
2026 if recvtype == null then return
2027
2028 if not recvtype isa MClassType then
2029 if recvtype isa MNullableType then
2030 v.error(self, "Type Error: cannot instantiate the nullable type `{recvtype}`.")
2031 return
2032 else if recvtype isa MFormalType then
2033 v.error(self, "Type Error: cannot instantiate the formal type `{recvtype}`.")
2034 return
2035 else
2036 v.error(self, "Type Error: cannot instantiate the type `{recvtype}`.")
2037 return
2038 end
2039 end
2040
2041 self.recvtype = recvtype
2042 var kind = recvtype.mclass.kind
2043
2044 var name: String
2045 var nqid = self.n_qid
2046 var node: ANode
2047 if nqid != null then
2048 name = nqid.n_id.text
2049 node = nqid
2050 else
2051 name = "new"
2052 node = self.n_kwnew
2053 end
2054 if name == "intern" then
2055 if kind != concrete_kind then
2056 v.error(self, "Type Error: cannot instantiate {kind} {recvtype}.")
2057 return
2058 end
2059 if n_args.n_exprs.not_empty then
2060 v.error(n_args, "Type Error: the intern constructor expects no arguments.")
2061 return
2062 end
2063 # Our job is done
2064 self.mtype = recvtype
2065 return
2066 end
2067
2068 var callsite = v.get_method(node, recvtype, name, false)
2069 if callsite == null then return
2070
2071 if not callsite.mproperty.is_new then
2072 if kind != concrete_kind then
2073 v.error(self, "Type Error: cannot instantiate {kind} `{recvtype}`.")
2074 return
2075 end
2076 self.mtype = recvtype
2077 else
2078 self.mtype = callsite.msignature.return_mtype
2079 assert self.mtype != null
2080 end
2081
2082 self.callsite = callsite
2083
2084 if not callsite.mproperty.is_init_for(recvtype.mclass) then
2085 v.error(self, "Error: `{name}` is not a constructor.")
2086 return
2087 end
2088
2089 var args = n_args.to_a
2090 callsite.check_signature(v, node, args)
2091 end
2092 end
2093
2094 ####
2095
2096 redef class AAttrFormExpr
2097 # The attribute accessed.
2098 var mproperty: nullable MAttribute
2099
2100 # The static type of the attribute.
2101 var attr_type: nullable MType
2102
2103 # Resolve the attribute accessed.
2104 private fun resolve_property(v: TypeVisitor)
2105 do
2106 var recvtype = v.visit_expr(self.n_expr)
2107 if recvtype == null then return # Skip error
2108 var node = self.n_id
2109 var name = node.text
2110 if recvtype isa MNullType then
2111 v.error(node, "Error: attribute `{name}` access on `null`.")
2112 return
2113 end
2114
2115 var unsafe_type = v.anchor_to(recvtype)
2116 var mproperty = v.try_get_mproperty_by_name2(node, unsafe_type, name)
2117 if mproperty == null then
2118 v.modelbuilder.error(node, "Error: attribute `{name}` does not exist in `{recvtype}`.")
2119 return
2120 end
2121 assert mproperty isa MAttribute
2122 self.mproperty = mproperty
2123
2124 var mpropdefs = mproperty.lookup_definitions(v.mmodule, unsafe_type)
2125 assert mpropdefs.length == 1
2126 var mpropdef = mpropdefs.first
2127 var attr_type = mpropdef.static_mtype
2128 if attr_type == null then return # skip error
2129 attr_type = v.resolve_for(attr_type, recvtype, self.n_expr isa ASelfExpr)
2130 self.attr_type = attr_type
2131 end
2132 end
2133
2134 redef class AAttrExpr
2135 redef fun accept_typing(v)
2136 do
2137 self.resolve_property(v)
2138 self.mtype = self.attr_type
2139 end
2140 end
2141
2142
2143 redef class AAttrAssignExpr
2144 redef fun accept_typing(v)
2145 do
2146 self.resolve_property(v)
2147 var mtype = self.attr_type
2148
2149 v.visit_expr_subtype(self.n_value, mtype)
2150 self.is_typed = mtype != null
2151 end
2152 end
2153
2154 redef class AAttrReassignExpr
2155 redef fun accept_typing(v)
2156 do
2157 self.resolve_property(v)
2158 var mtype = self.attr_type
2159 if mtype == null then return # Skip error
2160
2161 var rettype = self.resolve_reassignment(v, mtype, mtype)
2162
2163 self.is_typed = rettype != null
2164 end
2165 end
2166
2167 redef class AIssetAttrExpr
2168 redef fun accept_typing(v)
2169 do
2170 self.resolve_property(v)
2171 var mtype = self.attr_type
2172 if mtype == null then return # Skip error
2173
2174 var recvtype = self.n_expr.mtype.as(not null)
2175 var bound = v.resolve_for(mtype, recvtype, false)
2176 if bound isa MNullableType then
2177 v.error(n_id, "Type Error: `isset` on a nullable attribute.")
2178 end
2179 self.mtype = v.type_bool(self)
2180 end
2181 end
2182
2183 redef class AVarargExpr
2184 redef fun accept_typing(v)
2185 do
2186 # This kind of pseudo-expression can be only processed trough a signature
2187 # See `check_signature`
2188 # Other cases are a syntax error.
2189 v.error(self, "Syntax Error: unexpected `...`.")
2190 end
2191 end
2192
2193 ###
2194
2195 redef class ADebugTypeExpr
2196 redef fun accept_typing(v)
2197 do
2198 var expr = v.visit_expr(self.n_expr)
2199 if expr == null then return
2200 var unsafe = v.anchor_to(expr)
2201 var ntype = self.n_type
2202 var mtype = v.resolve_mtype(ntype)
2203 if mtype != null and mtype != expr then
2204 var umtype = v.anchor_to(mtype)
2205 v.modelbuilder.warning(self, "debug", "Found type {expr} (-> {unsafe}), expected {mtype} (-> {umtype})")
2206 end
2207 self.is_typed = true
2208 end
2209 end