tests: tests_icode also use alterner.pl
[nit.git] / src / 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 flow
22 import modelbuilder
23
24 private class TypeVisitor
25 var modelbuilder: ModelBuilder
26 var nclassdef: AClassdef
27 var mpropdef: MPropDef
28
29 var selfvariable: Variable = new Variable("self")
30
31 init(modelbuilder: ModelBuilder, nclassdef: AClassdef, mpropdef: MPropDef)
32 do
33 self.modelbuilder = modelbuilder
34 self.nclassdef = nclassdef
35 self.mpropdef = mpropdef
36
37 var mclass = nclassdef.mclassdef.mclass
38
39 var selfvariable = new Variable("self")
40 self.selfvariable = selfvariable
41 selfvariable.declared_type = mclass.mclass_type
42 end
43
44 fun mmodule: MModule do return self.nclassdef.mclassdef.mmodule
45
46 fun anchor: MClassType do return self.nclassdef.mclassdef.bound_mtype
47
48 fun anchor_to(mtype: MType): MType
49 do
50 var mmodule = self.nclassdef.mclassdef.mmodule
51 var anchor = self.nclassdef.mclassdef.bound_mtype
52 return mtype.anchor_to(mmodule, anchor)
53 end
54
55 fun is_subtype(sub, sup: MType): Bool
56 do
57 var mmodule = self.nclassdef.mclassdef.mmodule
58 var anchor = self.nclassdef.mclassdef.bound_mtype
59 return sub.is_subtype(mmodule, anchor, sup)
60 end
61
62 fun resolve_for(mtype, subtype: MType, for_self: Bool): MType
63 do
64 var mmodule = self.nclassdef.mclassdef.mmodule
65 var anchor = self.nclassdef.mclassdef.bound_mtype
66 #print "resolve_for {mtype} sub={subtype} forself={for_self} mmodule={mmodule} anchor={anchor}"
67 var res = mtype.resolve_for(subtype, anchor, mmodule, not for_self)
68 return res
69 end
70
71 fun resolve_signature_for(msignature: MSignature, recv: MType, for_self: Bool): MSignature
72 do
73 return self.resolve_for(msignature, recv, for_self).as(MSignature)
74 end
75
76 fun check_subtype(node: ANode, sub, sup: MType): Bool
77 do
78 if self.is_subtype(sub, sup) then return true
79 if self.is_subtype(sub, self.anchor_to(sup)) then
80 # FIXME workarround to the current unsafe typing policy. To remove once fixed virtual types exists.
81 #node.debug("Unsafe typing: expected {sup}, got {sub}")
82 return true
83 end
84 self.modelbuilder.error(node, "Type error: expected {sup}, got {sub}")
85 return false
86 end
87
88 # Visit an expression and do not care about the return value
89 fun visit_stmt(nexpr: nullable AExpr)
90 do
91 if nexpr == null then return
92 nexpr.accept_typing(self)
93 end
94
95 # Visit an expression and expects that it is not a statement
96 # Return the type of the expression
97 # Display an error and return null if:
98 # * the type cannot be determined or
99 # * `nexpr' is a statement
100 fun visit_expr(nexpr: AExpr): nullable MType
101 do
102 nexpr.accept_typing(self)
103 var mtype = nexpr.mtype
104 if mtype != null then return mtype
105 if not nexpr.is_typed then
106 if not self.modelbuilder.toolcontext.error_count > 0 then # check that there is really an error
107 if self.modelbuilder.toolcontext.verbose_level > 1 then
108 nexpr.debug("No return type but no error.")
109 end
110 end
111 return null # forward error
112 end
113 self.error(nexpr, "Type error: expected expression.")
114 return null
115 end
116
117 # Visit an expression and expect its static type is a least a `sup'
118 # Return the type of the expression
119 # * the type cannot be determined or
120 # * `nexpr' is a statement
121 # * `nexpt' is not a `sup'
122 fun visit_expr_subtype(nexpr: AExpr, sup: nullable MType): nullable MType
123 do
124 var sub = visit_expr(nexpr)
125 if sub == null then return null # Forward error
126
127 if sup == null then return null # Forward error
128
129 if not check_subtype(nexpr, sub, sup) then
130 return null
131 end
132 return sub
133 end
134
135 # Visit an expression and expect its static type is a bool
136 # Return the type of the expression
137 # * the type cannot be determined or
138 # * `nexpr' is a statement
139 # * `nexpt' is not a `sup'
140 fun visit_expr_bool(nexpr: AExpr): nullable MType
141 do
142 return self.visit_expr_subtype(nexpr, self.type_bool(nexpr))
143 end
144
145
146 private fun visit_expr_cast(node: ANode, nexpr: AExpr, ntype: AType): nullable MType
147 do
148 var sub = visit_expr(nexpr)
149 if sub == null then return null # Forward error
150
151 var sup = self.resolve_mtype(ntype)
152 if sup == null then return null # Forward error
153
154 var mmodule = self.nclassdef.mclassdef.mmodule
155 var anchor = self.nclassdef.mclassdef.bound_mtype
156 if sup == sub then
157 self.modelbuilder.warning(node, "Warning: Expression is already a {sup}.")
158 else if self.is_subtype(sub, sup) and not sup.need_anchor then
159 self.modelbuilder.warning(node, "Warning: Expression is already a {sup} since it is a {sub}.")
160 end
161 return sup
162 end
163
164 fun try_get_mproperty_by_name2(anode: ANode, mtype: MType, name: String): nullable MProperty
165 do
166 return self.modelbuilder.try_get_mproperty_by_name2(anode, self.nclassdef.mclassdef.mmodule, mtype, name)
167 end
168
169 fun resolve_mtype(node: AType): nullable MType
170 do
171 return self.modelbuilder.resolve_mtype(self.nclassdef, node)
172 end
173
174 fun get_mclass(node: ANode, name: String): nullable MClass
175 do
176 var mmodule = self.nclassdef.mclassdef.mmodule
177 var mclass = modelbuilder.try_get_mclass_by_name(node, mmodule, name)
178 if mclass == null then
179 self.modelbuilder.error(node, "Type Error: missing primitive class `{name}'.")
180 end
181 return mclass
182 end
183
184 fun type_bool(node: ANode): nullable MType
185 do
186 var mclass = self.get_mclass(node, "Bool")
187 if mclass == null then return null
188 return mclass.mclass_type
189 end
190
191 fun get_method(node: ANode, recvtype: MType, name: String, recv_is_self: Bool): nullable MMethodDef
192 do
193 var unsafe_type = self.anchor_to(recvtype)
194
195 #debug("recv: {recvtype} (aka {unsafe_type})")
196
197 var mproperty = self.try_get_mproperty_by_name2(node, unsafe_type, name)
198 if mproperty == null then
199 #self.modelbuilder.error(node, "Type error: property {name} not found in {unsafe_type} (ie {recvtype})")
200 if recv_is_self then
201 self.modelbuilder.error(node, "Error: Method or variable '{name}' unknown in {recvtype}.")
202 else
203 self.modelbuilder.error(node, "Error: Method '{name}' doesn't exists in {recvtype}.")
204 end
205 return null
206 end
207
208 var propdefs = mproperty.lookup_definitions(self.mmodule, unsafe_type)
209 if propdefs.length == 0 then
210 self.modelbuilder.error(node, "Type error: no definition found for property {name} in {unsafe_type}")
211 return null
212 else if propdefs.length > 1 then
213 self.modelbuilder.error(node, "Error: confliting property definitions for property {name} in {unsafe_type}: {propdefs.join(" ")}")
214 return null
215 end
216
217 var propdef = propdefs.first
218 assert propdef isa MMethodDef
219 return propdef
220 end
221
222 # Visit the expressions of args and cheik their conformity with the corresponding typi in signature
223 # The point of this method is to handle varargs correctly
224 # Note: The signature must be correctly adapted
225 fun check_signature(node: ANode, args: Array[AExpr], name: String, msignature: MSignature): Bool
226 do
227 var vararg_rank = msignature.vararg_rank
228 if vararg_rank >= 0 then
229 if args.length < msignature.arity then
230 #self.modelbuilder.error(node, "Error: Incorrect number of parameters. Got {args.length}, expected at least {msignature.arity}. Signature is {msignature}")
231 self.modelbuilder.error(node, "Error: arity mismatch; prototype is '{name}{msignature}'")
232 return false
233 end
234 else if args.length != msignature.arity then
235 self.modelbuilder.error(node, "Error: Incorrect number of parameters. Got {args.length}, expected {msignature.arity}. Signature is {msignature}")
236 return false
237 end
238
239 #debug("CALL {unsafe_type}.{msignature}")
240
241 var vararg_decl = args.length - msignature.arity
242 for i in [0..msignature.arity[ do
243 var j = i
244 if i == vararg_rank then continue # skip the vararg
245 if i > vararg_rank then
246 j = i + vararg_decl
247 end
248 var paramtype = msignature.mparameters[i].mtype
249 self.visit_expr_subtype(args[j], paramtype)
250 end
251 if vararg_rank >= 0 then
252 var varargs = new Array[AExpr]
253 var paramtype = msignature.mparameters[vararg_rank].mtype
254 for j in [vararg_rank..vararg_rank+vararg_decl] do
255 varargs.add(args[j])
256 self.visit_expr_subtype(args[j], paramtype)
257 end
258 end
259 return true
260 end
261
262 fun error(node: ANode, message: String)
263 do
264 self.modelbuilder.toolcontext.error(node.hot_location, message)
265 end
266
267 fun get_variable(node: AExpr, variable: Variable): nullable MType
268 do
269 var flow = node.after_flow_context
270 if flow == null then
271 self.error(node, "No context!")
272 return null
273 end
274
275 if flow.vars.has_key(variable) then
276 return flow.vars[variable]
277 else
278 #node.debug("*** START Collected for {variable}")
279 var mtypes = flow.collect_types(variable)
280 #node.debug("**** END Collected for {variable}")
281 if mtypes == null or mtypes.length == 0 then
282 return variable.declared_type
283 else if mtypes.length == 1 then
284 return mtypes.first
285 else
286 var res = merge_types(node,mtypes)
287 if res == null then res = variable.declared_type
288 return res
289 end
290 end
291 end
292
293 fun set_variable(node: AExpr, variable: Variable, mtype: nullable MType)
294 do
295 var flow = node.after_flow_context
296 assert flow != null
297
298 flow.set_var(variable, mtype)
299 end
300
301 fun merge_types(node: ANode, col: Array[nullable MType]): nullable MType
302 do
303 if col.length == 1 then return col.first
304 var res = new Array[nullable MType]
305 for t1 in col do
306 if t1 == null then continue # return null
307 var found = true
308 for t2 in col do
309 if t2 == null then continue # return null
310 if t2 isa MNullableType or t2 isa MNullType then
311 t1 = t1.as_nullable
312 end
313 if not is_subtype(t2, t1) then found = false
314 end
315 if found then
316 #print "merge {col.join(" ")} -> {t1}"
317 return t1
318 end
319 end
320 #self.modelbuilder.warning(node, "Type Error: {col.length} conflicting types: <{col.join(", ")}>")
321 return null
322 end
323 end
324
325 redef class Variable
326 # The declared type of the variable
327 var declared_type: nullable MType
328 end
329
330 redef class FlowContext
331 # Store changes of types because of type evolution
332 private var vars: HashMap[Variable, nullable MType] = new HashMap[Variable, nullable MType]
333 private var cache: HashMap[Variable, nullable Array[nullable MType]] = new HashMap[Variable, nullable Array[nullable MType]]
334
335 # Adapt the variable to a static type
336 # Warning1: do not modify vars directly.
337 # Warning2: sub-flow may have cached a unadapted variabial
338 private fun set_var(variable: Variable, mtype: nullable MType)
339 do
340 self.vars[variable] = mtype
341 self.cache.keys.remove(variable)
342 end
343
344 private fun collect_types(variable: Variable): nullable Array[nullable MType]
345 do
346 if cache.has_key(variable) then
347 return cache[variable]
348 end
349 var res: nullable Array[nullable MType] = null
350 if vars.has_key(variable) then
351 var mtype = vars[variable]
352 res = [mtype]
353 else if self.previous.is_empty then
354 # Root flow
355 res = [variable.declared_type]
356 else
357 for flow in self.previous do
358 if flow.is_unreachable then continue
359 var r2 = flow.collect_types(variable)
360 if r2 == null then continue
361 if res == null then
362 res = r2.to_a
363 else
364 for t in r2 do
365 if not res.has(t) then res.add(t)
366 end
367 end
368 end
369 end
370 cache[variable] = res
371 return res
372 end
373 end
374
375 redef class APropdef
376 # The entry point of the whole typing analysis
377 fun do_typing(modelbuilder: ModelBuilder)
378 do
379 end
380
381 # The variable associated to the reciever (if any)
382 var selfvariable: nullable Variable
383 end
384
385 redef class AConcreteMethPropdef
386 redef fun do_typing(modelbuilder: ModelBuilder)
387 do
388 var nclassdef = self.parent.as(AClassdef)
389 var mpropdef = self.mpropdef.as(not null)
390 var v = new TypeVisitor(modelbuilder, nclassdef, mpropdef)
391 self.selfvariable = v.selfvariable
392
393 var nblock = self.n_block
394 if nblock == null then return
395
396 var mmethoddef = self.mpropdef.as(not null)
397 for i in [0..mmethoddef.msignature.arity[ do
398 var mtype = mmethoddef.msignature.mparameters[i].mtype
399 if mmethoddef.msignature.vararg_rank == i then
400 var arrayclass = v.get_mclass(self.n_signature.n_params[i], "Array")
401 if arrayclass == null then return # Skip error
402 mtype = arrayclass.get_mtype([mtype])
403 end
404 var variable = self.n_signature.n_params[i].variable
405 assert variable != null
406 variable.declared_type = mtype
407 end
408 for i in [0..mmethoddef.msignature.mclosures.length[ do
409 var mclosure = mmethoddef.msignature.mclosures[i]
410 var variable = self.n_signature.n_closure_decls[i].variable
411 assert variable != null
412 variable.declared_type = mclosure.mtype
413 end
414 v.visit_stmt(nblock)
415
416 if not nblock.after_flow_context.is_unreachable and mmethoddef.msignature.return_mtype != null then
417 # We reach the end of the function without having a return, it is bad
418 v.error(self, "Control error: Reached end of function (a 'return' with a value was expected).")
419 end
420 end
421 end
422
423 redef class AAttrPropdef
424 redef fun do_typing(modelbuilder: ModelBuilder)
425 do
426 var nclassdef = self.parent.as(AClassdef)
427 var v = new TypeVisitor(modelbuilder, nclassdef, self.mpropdef.as(not null))
428 self.selfvariable = v.selfvariable
429
430 var nexpr = self.n_expr
431 if nexpr != null then
432 var mtype = self.mpropdef.static_mtype
433 v.visit_expr_subtype(nexpr, mtype)
434 end
435 end
436 end
437
438 ###
439
440 redef class AExpr
441 # The static type of the expression.
442 # null if self is a statement of in case of error
443 var mtype: nullable MType = null
444
445 # Is the statement correctly typed?
446 # Used to distinguish errors and statements when `mtype' == null
447 var is_typed: Bool = false
448
449 # Return the variable read (if any)
450 # Used to perform adaptive typing
451 fun its_variable: nullable Variable do return null
452
453 private fun accept_typing(v: TypeVisitor)
454 do
455 v.error(self, "no implemented accept_typing for {self.class_name}")
456 end
457 end
458
459 redef class ABlockExpr
460 redef fun accept_typing(v)
461 do
462 for e in self.n_expr do v.visit_stmt(e)
463 self.is_typed = true
464 end
465 end
466
467 redef class AVardeclExpr
468 redef fun accept_typing(v)
469 do
470 var variable = self.variable
471 if variable == null then return # Skip error
472
473 var ntype = self.n_type
474 var mtype: nullable MType
475 if ntype == null then
476 mtype = null
477 else
478 mtype = v.resolve_mtype(ntype)
479 if mtype == null then return # Skip error
480 end
481
482 var nexpr = self.n_expr
483 if nexpr != null then
484 if mtype != null then
485 v.visit_expr_subtype(nexpr, mtype)
486 else
487 mtype = v.visit_expr(nexpr)
488 if mtype == null then return # Skip error
489 end
490 end
491
492 if mtype == null then
493 mtype = v.get_mclass(self, "Object").mclass_type.as_nullable
494 end
495
496 variable.declared_type = mtype
497 v.set_variable(self, variable, mtype)
498
499 #debug("var {variable}: {mtype}")
500
501 self.is_typed = true
502 end
503 end
504
505 redef class AVarExpr
506 redef fun its_variable do return self.variable
507 redef fun accept_typing(v)
508 do
509 var variable = self.variable
510 if variable == null then return # Skip error
511
512 var mtype = v.get_variable(self, variable)
513 if mtype != null then
514 #debug("{variable} is {mtype}")
515 else
516 #debug("{variable} is untyped")
517 end
518
519 self.mtype = mtype
520 end
521 end
522
523 redef class AVarAssignExpr
524 redef fun accept_typing(v)
525 do
526 var variable = self.variable
527 assert variable != null
528
529 var mtype = v.visit_expr_subtype(n_value, variable.declared_type)
530
531 v.set_variable(self, variable, mtype)
532
533 self.is_typed = true
534 end
535 end
536
537 redef class AReassignFormExpr
538 # The method designed by the reassign operator.
539 var reassign_property: nullable MMethodDef = null
540
541 var read_type: nullable MType = null
542
543 # Determine the `reassign_property'
544 # `readtype' is the type of the reading of the left value.
545 # `writetype' is the type of the writing of the left value.
546 # (Because of ACallReassignExpr, both can be different.
547 # Return the static type of the value to store.
548 private fun resolve_reassignment(v: TypeVisitor, readtype, writetype: MType): nullable MType
549 do
550 var reassign_name: String
551 if self.n_assign_op isa APlusAssignOp then
552 reassign_name = "+"
553 else if self.n_assign_op isa AMinusAssignOp then
554 reassign_name = "-"
555 else
556 abort
557 end
558
559 self.read_type = readtype
560
561 if readtype isa MNullType then
562 v.error(self, "Error: Method '{reassign_name}' call on 'null'.")
563 return null
564 end
565
566 var mpropdef = v.get_method(self, readtype, reassign_name, false)
567 if mpropdef == null then return null # Skip error
568
569 self.reassign_property = mpropdef
570
571 var msignature = mpropdef.msignature
572 assert msignature!= null
573 msignature = v.resolve_signature_for(msignature, readtype, false)
574
575 var rettype = msignature.return_mtype
576 assert msignature.arity == 1 and rettype != null
577
578 var value_type = v.visit_expr_subtype(self.n_value, msignature.mparameters.first.mtype)
579 if value_type == null then return null # Skip error
580
581 v.check_subtype(self, rettype, writetype)
582 return rettype
583 end
584 end
585
586 redef class AVarReassignExpr
587 redef fun accept_typing(v)
588 do
589 var variable = self.variable
590 assert variable != null
591
592 var readtype = v.get_variable(self, variable)
593 if readtype == null then return
594
595 var writetype = variable.declared_type
596 if writetype == null then return
597
598 var rettype = self.resolve_reassignment(v, readtype, writetype)
599
600 v.set_variable(self, variable, rettype)
601
602 self.is_typed = true
603 end
604 end
605
606
607 redef class AContinueExpr
608 redef fun accept_typing(v)
609 do
610 var nexpr = self.n_expr
611 if nexpr != null then
612 var mtype = v.visit_expr(nexpr)
613 end
614 self.is_typed = true
615 end
616 end
617
618 redef class ABreakExpr
619 redef fun accept_typing(v)
620 do
621 var nexpr = self.n_expr
622 if nexpr != null then
623 var mtype = v.visit_expr(nexpr)
624 end
625 self.is_typed = true
626 end
627 end
628
629 redef class AReturnExpr
630 redef fun accept_typing(v)
631 do
632 var nexpr = self.n_expr
633 var ret_type = v.mpropdef.as(MMethodDef).msignature.return_mtype
634 if nexpr != null then
635 if ret_type != null then
636 var mtype = v.visit_expr_subtype(nexpr, ret_type)
637 else
638 var mtype = v.visit_expr(nexpr)
639 v.error(self, "Error: Return with value in a procedure.")
640 end
641 else if ret_type != null then
642 v.error(self, "Error: Return without value in a function.")
643 end
644 self.is_typed = true
645 end
646 end
647
648 redef class AAbortExpr
649 redef fun accept_typing(v)
650 do
651 self.is_typed = true
652 end
653 end
654
655 redef class AIfExpr
656 redef fun accept_typing(v)
657 do
658 v.visit_expr_bool(n_expr)
659
660 v.visit_stmt(n_then)
661 v.visit_stmt(n_else)
662 self.is_typed = true
663 end
664 end
665
666 redef class AIfexprExpr
667 redef fun accept_typing(v)
668 do
669 v.visit_expr_bool(n_expr)
670
671 var t1 = v.visit_expr(n_then)
672 var t2 = v.visit_expr(n_else)
673
674 if t1 == null or t2 == null then
675 return # Skip error
676 end
677
678 var t = v.merge_types(self, [t1, t2])
679 if t == null then
680 v.error(self, "Type Error: ambiguous type {t1} vs {t2}")
681 end
682 self.mtype = t
683 end
684 end
685
686 redef class ADoExpr
687 redef fun accept_typing(v)
688 do
689 v.visit_stmt(n_block)
690 self.is_typed = true
691 end
692 end
693
694 redef class AWhileExpr
695 redef fun accept_typing(v)
696 do
697 v.visit_expr_bool(n_expr)
698
699 v.visit_stmt(n_block)
700 self.is_typed = true
701 end
702 end
703
704 redef class ALoopExpr
705 redef fun accept_typing(v)
706 do
707 v.visit_stmt(n_block)
708 self.is_typed = true
709 end
710 end
711
712 redef class AForExpr
713 var coltype: nullable MGenericType
714 redef fun accept_typing(v)
715 do
716 var mtype = v.visit_expr(n_expr)
717 if mtype == null then return
718
719 var colcla = v.get_mclass(self, "Collection")
720 if colcla == null then return
721 var objcla = v.get_mclass(self, "Object")
722 if objcla == null then return
723 var mapcla = v.get_mclass(self, "Map")
724 if mapcla == null then return
725 if v.is_subtype(mtype, colcla.get_mtype([objcla.mclass_type.as_nullable])) then
726 var coltype = mtype.supertype_to(v.mmodule, v.anchor, colcla)
727 assert coltype isa MGenericType
728 self.coltype = coltype
729 var variables = self.variables
730 if variables.length != 1 then
731 v.error(self, "Type Error: Expected one variable")
732 else
733 variables.first.declared_type = coltype.arguments.first
734 end
735 else if v.is_subtype(mtype, mapcla.get_mtype([objcla.mclass_type.as_nullable, objcla.mclass_type.as_nullable])) then
736 var coltype = mtype.supertype_to(v.mmodule, v.anchor, mapcla)
737 assert coltype isa MGenericType
738 self.coltype = coltype
739 var variables = self.variables
740 if variables.length != 2 then
741 v.error(self, "Type Error: Expected two variables")
742 else
743 variables[0].declared_type = coltype.arguments[0]
744 variables[1].declared_type = coltype.arguments[1]
745 end
746 else
747 v.modelbuilder.error(self, "TODO: Do 'for' on {mtype}")
748 end
749
750 v.visit_stmt(n_block)
751 self.is_typed = true
752 end
753 end
754
755 redef class AAssertExpr
756 redef fun accept_typing(v)
757 do
758 v.visit_expr_bool(n_expr)
759
760 v.visit_stmt(n_else)
761 self.is_typed = true
762 end
763 end
764
765 redef class AOrExpr
766 redef fun accept_typing(v)
767 do
768 v.visit_expr_bool(n_expr)
769 v.visit_expr_bool(n_expr2)
770 self.mtype = v.type_bool(self)
771 end
772 end
773
774 redef class AAndExpr
775 redef fun accept_typing(v)
776 do
777 v.visit_expr_bool(n_expr)
778 v.visit_expr_bool(n_expr2)
779 self.mtype = v.type_bool(self)
780 end
781 end
782
783
784 redef class ANotExpr
785 redef fun accept_typing(v)
786 do
787 v.visit_expr_bool(n_expr)
788 self.mtype = v.type_bool(self)
789 end
790 end
791
792 redef class AOrElseExpr
793 redef fun accept_typing(v)
794 do
795 var t1 = v.visit_expr(n_expr)
796 var t2 = v.visit_expr(n_expr2)
797
798 if t1 == null or t2 == null then
799 return # Skip error
800 end
801
802 if t1 isa MNullableType then
803 t1 = t1.mtype
804 end
805
806 var t = v.merge_types(self, [t1, t2])
807 if t == null then
808 v.error(self, "Type Error: ambiguous type {t1} vs {t2}")
809 end
810 self.mtype = t
811 end
812 end
813
814 redef class AEeExpr
815 redef fun accept_typing(v)
816 do
817 v.visit_expr(n_expr)
818 v.visit_expr(n_expr2)
819 self.mtype = v.type_bool(self)
820 end
821 end
822
823 redef class ATrueExpr
824 redef fun accept_typing(v)
825 do
826 self.mtype = v.type_bool(self)
827 end
828 end
829
830 redef class AFalseExpr
831 redef fun accept_typing(v)
832 do
833 self.mtype = v.type_bool(self)
834 end
835 end
836
837 redef class AIntExpr
838 redef fun accept_typing(v)
839 do
840 var mclass = v.get_mclass(self, "Int")
841 if mclass == null then return # Forward error
842 self.mtype = mclass.mclass_type
843 end
844 end
845
846 redef class AFloatExpr
847 redef fun accept_typing(v)
848 do
849 var mclass = v.get_mclass(self, "Float")
850 if mclass == null then return # Forward error
851 self.mtype = mclass.mclass_type
852 end
853 end
854
855 redef class ACharExpr
856 redef fun accept_typing(v)
857 do
858 var mclass = v.get_mclass(self, "Char")
859 if mclass == null then return # Forward error
860 self.mtype = mclass.mclass_type
861 end
862 end
863
864 redef class AStringFormExpr
865 redef fun accept_typing(v)
866 do
867 var mclass = v.get_mclass(self, "String")
868 if mclass == null then return # Forward error
869 self.mtype = mclass.mclass_type
870 end
871 end
872
873 redef class ASuperstringExpr
874 redef fun accept_typing(v)
875 do
876 var mclass = v.get_mclass(self, "String")
877 if mclass == null then return # Forward error
878 self.mtype = mclass.mclass_type
879 for nexpr in self.n_exprs do
880 var t = v.visit_expr(nexpr)
881 end
882 end
883 end
884
885 redef class AArrayExpr
886 redef fun accept_typing(v)
887 do
888 var mtypes = new Array[nullable MType]
889 for e in self.n_exprs.n_exprs do
890 var t = v.visit_expr(e)
891 if t == null then
892 return # Skip error
893 end
894 mtypes.add(t)
895 end
896 var mtype = v.merge_types(self, mtypes)
897 if mtype == null then
898 v.error(self, "Type Error: ambiguous array type {mtypes.join(" ")}")
899 return
900 end
901 var mclass = v.get_mclass(self, "Array")
902 if mclass == null then return # Forward error
903 self.mtype = mclass.get_mtype([mtype])
904 end
905 end
906
907 redef class ARangeExpr
908 redef fun accept_typing(v)
909 do
910 var t1 = v.visit_expr(self.n_expr)
911 var t2 = v.visit_expr(self.n_expr2)
912 if t1 == null or t2 == null then return
913 var mclass = v.get_mclass(self, "Range")
914 if mclass == null then return # Forward error
915 if v.is_subtype(t1, t2) then
916 self.mtype = mclass.get_mtype([t2])
917 else if v.is_subtype(t2, t1) then
918 self.mtype = mclass.get_mtype([t1])
919 else
920 v.error(self, "Type Error: Cannot create range: {t1} vs {t2}")
921 end
922 end
923 end
924
925 redef class ANullExpr
926 redef fun accept_typing(v)
927 do
928 self.mtype = v.mmodule.model.null_type
929 end
930 end
931
932 redef class AIsaExpr
933 # The static type to cast to.
934 # (different from the static type of the expression that is Bool).
935 var cast_type: nullable MType
936 redef fun accept_typing(v)
937 do
938 var mtype = v.visit_expr_cast(self, self.n_expr, self.n_type)
939 self.cast_type = mtype
940
941 var variable = self.n_expr.its_variable
942 if variable != null then
943 var orig = self.n_expr.mtype
944 var from = if orig != null then orig.to_s else "invalid"
945 var to = if mtype != null then mtype.to_s else "invalid"
946 #debug("adapt {variable}: {from} -> {to}")
947 self.after_flow_context.when_true.set_var(variable, mtype)
948 end
949
950 self.mtype = v.type_bool(self)
951 end
952 end
953
954 redef class AAsCastExpr
955 redef fun accept_typing(v)
956 do
957 self.mtype = v.visit_expr_cast(self, self.n_expr, self.n_type)
958 end
959 end
960
961 redef class AAsNotnullExpr
962 redef fun accept_typing(v)
963 do
964 var mtype = v.visit_expr(self.n_expr)
965 if mtype isa MNullType then
966 v.error(self, "Type error: as(not null) on null")
967 return
968 end
969 if mtype isa MNullableType then
970 self.mtype = mtype.mtype
971 return
972 end
973 # TODO: warn on useless as not null
974 self.mtype = mtype
975 end
976 end
977
978 redef class AProxyExpr
979 redef fun accept_typing(v)
980 do
981 self.mtype = v.visit_expr(self.n_expr)
982 end
983 end
984
985 redef class ASelfExpr
986 redef var its_variable: nullable Variable
987 redef fun accept_typing(v)
988 do
989 var variable = v.selfvariable
990 self.its_variable = variable
991 self.mtype = v.get_variable(self, variable)
992 end
993 end
994
995 ## MESSAGE SENDING AND PROPERTY
996
997 redef class ASendExpr
998 # The property invoked by the send.
999 var mproperty: nullable MMethod
1000
1001 redef fun accept_typing(v)
1002 do
1003 var recvtype = v.visit_expr(self.n_expr)
1004 var name = self.property_name
1005
1006 if recvtype == null then return # Forward error
1007 if recvtype isa MNullType then
1008 v.error(self, "Error: Method '{name}' call on 'null'.")
1009 return
1010 end
1011
1012 var propdef = v.get_method(self, recvtype, name, self.n_expr isa ASelfExpr)
1013 if propdef == null then return
1014 var mproperty = propdef.mproperty
1015 self.mproperty = mproperty
1016 var msignature = propdef.msignature
1017 if msignature == null then abort # Forward error
1018
1019 var for_self = self.n_expr isa ASelfExpr
1020 msignature = v.resolve_signature_for(msignature, recvtype, for_self)
1021
1022 var args = compute_raw_arguments
1023
1024 v.check_signature(self, args, name, msignature)
1025
1026 if mproperty.is_init then
1027 var vmpropdef = v.mpropdef
1028 if not (vmpropdef isa MMethodDef and vmpropdef.mproperty.is_init) then
1029 v.error(self, "Can call a init only in another init")
1030 end
1031 end
1032
1033 var ret = msignature.return_mtype
1034 if ret != null then
1035 self.mtype = ret
1036 else
1037 self.is_typed = true
1038 end
1039
1040 if self.n_closure_defs.length == msignature.mclosures.length then
1041 for i in [0..self.n_closure_defs.length[ do
1042 self.n_closure_defs[i].accept_typing(v, msignature.mclosures[i])
1043 end
1044 else
1045 debug("closure: got {self.n_closure_defs.length}, want {msignature.mclosures.length}")
1046 end
1047 end
1048
1049 # The name of the property
1050 # Each subclass simply provide the correct name.
1051 private fun property_name: String is abstract
1052
1053 # An array of all arguments (excluding self)
1054 fun compute_raw_arguments: Array[AExpr] is abstract
1055 end
1056
1057 redef class ABinopExpr
1058 redef fun compute_raw_arguments do return [n_expr2]
1059 end
1060 redef class AEqExpr
1061 redef fun property_name do return "=="
1062 redef fun accept_typing(v)
1063 do
1064 super
1065
1066 var variable = self.n_expr.its_variable
1067 if variable == null then return
1068 var mtype = self.n_expr2.mtype
1069 if not mtype isa MNullType then return
1070 var vartype = v.get_variable(self, variable)
1071 if not vartype isa MNullableType then return
1072 self.after_flow_context.when_true.set_var(variable, mtype)
1073 self.after_flow_context.when_false.set_var(variable, vartype.mtype)
1074 #debug("adapt {variable}:{vartype} ; true->{mtype} false->{vartype.mtype}")
1075 end
1076 end
1077 redef class ANeExpr
1078 redef fun property_name do return "!="
1079 redef fun accept_typing(v)
1080 do
1081 super
1082
1083 var variable = self.n_expr.its_variable
1084 if variable == null then return
1085 var mtype = self.n_expr2.mtype
1086 if not mtype isa MNullType then return
1087 var vartype = v.get_variable(self, variable)
1088 if not vartype isa MNullableType then return
1089 self.after_flow_context.when_false.set_var(variable, mtype)
1090 self.after_flow_context.when_true.set_var(variable, vartype.mtype)
1091 #debug("adapt {variable}:{vartype} ; true->{vartype.mtype} false->{mtype}")
1092 end
1093 end
1094 redef class ALtExpr
1095 redef fun property_name do return "<"
1096 end
1097 redef class ALeExpr
1098 redef fun property_name do return "<="
1099 end
1100 redef class ALlExpr
1101 redef fun property_name do return "<<"
1102 end
1103 redef class AGtExpr
1104 redef fun property_name do return ">"
1105 end
1106 redef class AGeExpr
1107 redef fun property_name do return ">="
1108 end
1109 redef class AGgExpr
1110 redef fun property_name do return ">>"
1111 end
1112 redef class APlusExpr
1113 redef fun property_name do return "+"
1114 end
1115 redef class AMinusExpr
1116 redef fun property_name do return "-"
1117 end
1118 redef class AStarshipExpr
1119 redef fun property_name do return "<=>"
1120 end
1121 redef class AStarExpr
1122 redef fun property_name do return "*"
1123 end
1124 redef class ASlashExpr
1125 redef fun property_name do return "/"
1126 end
1127 redef class APercentExpr
1128 redef fun property_name do return "%"
1129 end
1130
1131 redef class AUminusExpr
1132 redef fun property_name do return "unary -"
1133 redef fun compute_raw_arguments do return new Array[AExpr]
1134 end
1135
1136
1137 redef class ACallExpr
1138 redef fun property_name do return n_id.text
1139 redef fun compute_raw_arguments do return n_args.to_a
1140 end
1141
1142 redef class ACallAssignExpr
1143 redef fun property_name do return n_id.text + "="
1144 redef fun compute_raw_arguments
1145 do
1146 var res = n_args.to_a
1147 res.add(n_value)
1148 return res
1149 end
1150 end
1151
1152 redef class ABraExpr
1153 redef fun property_name do return "[]"
1154 redef fun compute_raw_arguments do return n_args.to_a
1155 end
1156
1157 redef class ABraAssignExpr
1158 redef fun property_name do return "[]="
1159 redef fun compute_raw_arguments
1160 do
1161 var res = n_args.to_a
1162 res.add(n_value)
1163 return res
1164 end
1165 end
1166
1167 redef class ASendReassignFormExpr
1168 # The property invoked for the writing
1169 var write_mproperty: nullable MMethod = null
1170
1171 redef fun accept_typing(v)
1172 do
1173 var recvtype = v.visit_expr(self.n_expr)
1174 var name = self.property_name
1175
1176 if recvtype == null then return # Forward error
1177 if recvtype isa MNullType then
1178 v.error(self, "Error: Method '{name}' call on 'null'.")
1179 return
1180 end
1181
1182 var propdef = v.get_method(self, recvtype, name, self.n_expr isa ASelfExpr)
1183 if propdef == null then return
1184 var mproperty = propdef.mproperty
1185 self.mproperty = mproperty
1186 var msignature = propdef.msignature
1187 if msignature == null then abort # Forward error
1188 var for_self = self.n_expr isa ASelfExpr
1189 msignature = v.resolve_signature_for(msignature, recvtype, for_self)
1190
1191 var args = compute_raw_arguments
1192
1193 v.check_signature(self, args, name, msignature)
1194
1195 var readtype = msignature.return_mtype
1196 if readtype == null then
1197 v.error(self, "Error: {name} is not a function")
1198 return
1199 end
1200
1201 var wpropdef = v.get_method(self, recvtype, name + "=", self.n_expr isa ASelfExpr)
1202 if wpropdef == null then return
1203 var wmproperty = wpropdef.mproperty
1204 self.write_mproperty = wmproperty
1205 var wmsignature = wpropdef.msignature
1206 if wmsignature == null then abort # Forward error
1207 wmsignature = v.resolve_signature_for(wmsignature, recvtype, for_self)
1208
1209 var wtype = self.resolve_reassignment(v, readtype, wmsignature.mparameters.last.mtype)
1210 if wtype == null then return
1211
1212 args.add(self.n_value)
1213 v.check_signature(self, args, name + "=", wmsignature)
1214
1215 self.is_typed = true
1216 end
1217 end
1218
1219 redef class ACallReassignExpr
1220 redef fun property_name do return n_id.text
1221 redef fun compute_raw_arguments do return n_args.to_a
1222 end
1223
1224 redef class ABraReassignExpr
1225 redef fun property_name do return "[]"
1226 redef fun compute_raw_arguments do return n_args.to_a
1227 end
1228
1229 redef class AInitExpr
1230 redef fun property_name do return "init"
1231 redef fun compute_raw_arguments do return n_args.to_a
1232 end
1233
1234 redef class AExprs
1235 fun to_a: Array[AExpr] do return self.n_exprs.to_a
1236 end
1237
1238 ###
1239
1240 redef class ASuperExpr
1241 # The method to call if the super is in fact a 'super init call'
1242 # Note: if the super is a normal call-next-method, then this attribute is null
1243 var mproperty: nullable MMethod
1244
1245 redef fun accept_typing(v)
1246 do
1247 var recvtype = v.nclassdef.mclassdef.bound_mtype
1248 var mproperty = v.mpropdef.mproperty
1249 if not mproperty isa MMethod then
1250 v.error(self, "Error: super only usable in a method")
1251 return
1252 end
1253 var superprops = mproperty.lookup_super_definitions(v.mmodule, recvtype)
1254 if superprops.length == 0 then
1255 if mproperty.is_init and v.mpropdef.is_intro then
1256 process_superinit(v)
1257 return
1258 end
1259 v.error(self, "Error: No super method to call for {mproperty}.")
1260 return
1261 else if superprops.length > 1 then
1262 v.modelbuilder.warning(self, "Error: Conflicting super method to call for {mproperty}: {superprops.join(", ")}.")
1263 return
1264 end
1265 var superprop = superprops.first
1266 assert superprop isa MMethodDef
1267
1268 var msignature = superprop.msignature.as(not null)
1269 msignature = v.resolve_signature_for(msignature, recvtype, true)
1270 var args = self.n_args.to_a
1271 if args.length > 0 then
1272 v.check_signature(self, args, mproperty.name, msignature)
1273 end
1274 self.mtype = msignature.return_mtype
1275 end
1276
1277 private fun process_superinit(v: TypeVisitor)
1278 do
1279 var recvtype = v.nclassdef.mclassdef.bound_mtype
1280 var mproperty = v.mpropdef.mproperty
1281 var superprop: nullable MMethodDef = null
1282 for msupertype in v.nclassdef.mclassdef.supertypes do
1283 msupertype = msupertype.anchor_to(v.mmodule, recvtype)
1284 var errcount = v.modelbuilder.toolcontext.error_count
1285 var candidate = v.try_get_mproperty_by_name2(self, msupertype, mproperty.name).as(nullable MMethod)
1286 if candidate == null then
1287 if v.modelbuilder.toolcontext.error_count > errcount then return # Forard error
1288 continue # Try next super-class
1289 end
1290 if superprop != null and superprop.mproperty != candidate then
1291 v.error(self, "Error: conflicting super constructor to call for {mproperty}: {candidate.full_name}, {superprop.mproperty.full_name}")
1292 return
1293 end
1294 var candidatedefs = candidate.lookup_definitions(v.mmodule, recvtype)
1295 if superprop != null then
1296 if superprop == candidatedefs.first then continue
1297 candidatedefs.add(superprop)
1298 end
1299 if candidatedefs.length > 1 then
1300 v.error(self, "Error: confliting property definitions for property {mproperty} in {recvtype}: {candidatedefs.join(", ")}")
1301 return
1302 end
1303 superprop = candidatedefs.first
1304 end
1305 if superprop == null then
1306 v.error(self, "Error: No super method to call for {mproperty}.")
1307 return
1308 end
1309 self.mproperty = superprop.mproperty
1310
1311 var args = self.n_args.to_a
1312 var msignature = superprop.msignature.as(not null)
1313 msignature = v.resolve_signature_for(msignature, recvtype, true)
1314 if args.length > 0 then
1315 v.check_signature(self, args, mproperty.name, msignature)
1316 else
1317 # TODO: Check signature
1318 end
1319
1320 self.is_typed = true
1321 end
1322 end
1323
1324 ####
1325
1326 redef class ANewExpr
1327 # The constructor invoked by the new.
1328 var mproperty: nullable MMethod
1329
1330 redef fun accept_typing(v)
1331 do
1332 var recvtype = v.resolve_mtype(self.n_type)
1333 if recvtype == null then return
1334 self.mtype = recvtype
1335
1336 if not recvtype isa MClassType then
1337 if recvtype isa MNullableType then
1338 v.error(self, "Type error: cannot instantiate the nullable type {recvtype}.")
1339 return
1340 else
1341 v.error(self, "Type error: cannot instantiate the formal type {recvtype}.")
1342 return
1343 end
1344 end
1345
1346 var name: String
1347 var nid = self.n_id
1348 if nid != null then
1349 name = nid.text
1350 else
1351 name = "init"
1352 end
1353 var propdef = v.get_method(self, recvtype, name, false)
1354 if propdef == null then return
1355
1356 self.mproperty = propdef.mproperty
1357
1358 if not propdef.mproperty.is_init_for(recvtype.mclass) then
1359 v.error(self, "Error: {name} is not a constructor.")
1360 return
1361 end
1362
1363 var msignature = propdef.msignature.as(not null)
1364 msignature = v.resolve_signature_for(msignature, recvtype, false)
1365
1366 var args = n_args.to_a
1367 v.check_signature(self, args, name, msignature)
1368 end
1369 end
1370
1371 ####
1372
1373 redef class AAttrFormExpr
1374 # The attribute acceded.
1375 var mproperty: nullable MAttribute
1376
1377 # The static type of the attribute.
1378 var attr_type: nullable MType
1379
1380 # Resolve the attribute acceded.
1381 private fun resolve_property(v: TypeVisitor)
1382 do
1383 var recvtype = v.visit_expr(self.n_expr)
1384 if recvtype == null then return # Skip error
1385 var name = self.n_id.text
1386 if recvtype isa MNullType then
1387 v.error(self, "Error: Attribute '{name}' access on 'null'.")
1388 return
1389 end
1390
1391 var unsafe_type = v.anchor_to(recvtype)
1392 var mproperty = v.try_get_mproperty_by_name2(self, unsafe_type, name)
1393 if mproperty == null then
1394 v.modelbuilder.error(self, "Error: Attribute {name} doesn't exists in {recvtype}.")
1395 return
1396 end
1397 assert mproperty isa MAttribute
1398 self.mproperty = mproperty
1399
1400 var mpropdefs = mproperty.lookup_definitions(v.mmodule, unsafe_type)
1401 assert mpropdefs.length == 1
1402 var mpropdef = mpropdefs.first
1403 var attr_type = mpropdef.static_mtype.as(not null)
1404 attr_type = v.resolve_for(attr_type, recvtype, self.n_expr isa ASelfExpr)
1405 self.attr_type = attr_type
1406 end
1407 end
1408
1409 redef class AAttrExpr
1410 redef fun accept_typing(v)
1411 do
1412 self.resolve_property(v)
1413 self.mtype = self.attr_type
1414 end
1415 end
1416
1417
1418 redef class AAttrAssignExpr
1419 redef fun accept_typing(v)
1420 do
1421 self.resolve_property(v)
1422 var mtype = self.attr_type
1423
1424 v.visit_expr_subtype(self.n_value, mtype)
1425 self.is_typed = true
1426 end
1427 end
1428
1429 redef class AAttrReassignExpr
1430 redef fun accept_typing(v)
1431 do
1432 self.resolve_property(v)
1433 var mtype = self.attr_type
1434 if mtype == null then return # Skip error
1435
1436 self.resolve_reassignment(v, mtype, mtype)
1437
1438 self.is_typed = true
1439 end
1440 end
1441
1442 redef class AIssetAttrExpr
1443 redef fun accept_typing(v)
1444 do
1445 self.resolve_property(v)
1446 var mtype = self.attr_type
1447 if mtype == null then return # Skip error
1448
1449 var recvtype = self.n_expr.mtype.as(not null)
1450 var bound = v.resolve_for(mtype, recvtype, false)
1451 if bound isa MNullableType then
1452 v.error(self, "Error: isset on a nullable attribute.")
1453 end
1454 self.mtype = v.type_bool(self)
1455 end
1456 end
1457
1458 ###
1459
1460 redef class AClosureCallExpr
1461 redef fun accept_typing(v)
1462 do
1463 var variable = self.variable
1464 if variable == null then return # Skip error
1465
1466 var recvtype = v.nclassdef.mclassdef.bound_mtype
1467 var msignature = variable.declared_type.as(MSignature)
1468 msignature = v.resolve_signature_for(msignature, recvtype, false)
1469
1470 var args = n_args.to_a
1471 v.check_signature(self, args, variable.name, msignature)
1472
1473 self.is_typed = true
1474 self.mtype = msignature.return_mtype
1475 end
1476 end
1477
1478 redef class AClosureDef
1479 var mclosure: nullable MParameter
1480
1481 private fun accept_typing(v: TypeVisitor, mparameter: MParameter)
1482 do
1483 var variables = self.variables
1484 if variables == null then return
1485
1486 self.mclosure = mparameter
1487 var msignature = mparameter.mtype.as(MSignature)
1488
1489 if msignature.arity != variables.length then
1490 v.error(self, "Type error: closure {mparameter.name} expects {msignature.arity} parameters, {variables.length} given")
1491 return
1492 end
1493
1494 for i in [0..variables.length[ do
1495 variables[i].declared_type = msignature.mparameters[i].mtype
1496 end
1497
1498 v.visit_stmt(self.n_expr)
1499 end
1500 end
1501
1502 ###
1503
1504 redef class ADebugTypeExpr
1505 redef fun accept_typing(v)
1506 do
1507 var expr = v.visit_expr(self.n_expr)
1508 if expr == null then return
1509 var unsafe = v.anchor_to(expr)
1510 var ntype = self.n_type
1511 var mtype = v.resolve_mtype(ntype)
1512 if mtype != null and mtype != expr then
1513 var umtype = v.anchor_to(mtype)
1514 v.modelbuilder.warning(self, "Found type {expr} (-> {unsafe}), expected {mtype} (-> {umtype})")
1515 end
1516 end
1517 end