model: add MVirtualType::lookup_single_definition
[nit.git] / src / model / model.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 # Classes, types and properties
18 #
19 # All three concepts are defined in this same module because these are strongly connected:
20 # * types are based on classes
21 # * classes contains properties
22 # * some properties are types (virtual types)
23 #
24 # TODO: liearization, extern stuff
25 # FIXME: better handling of the types
26 module model
27
28 import mmodule
29 import mdoc
30 import ordered_tree
31 private import more_collections
32
33 redef class Model
34 # All known classes
35 var mclasses = new Array[MClass]
36
37 # All known properties
38 var mproperties = new Array[MProperty]
39
40 # Hierarchy of class definition.
41 #
42 # Each classdef is associated with its super-classdefs in regard to
43 # its module of definition.
44 var mclassdef_hierarchy = new POSet[MClassDef]
45
46 # Class-type hierarchy restricted to the introduction.
47 #
48 # The idea is that what is true on introduction is always true whatever
49 # the module considered.
50 # Therefore, this hierarchy is used for a fast positive subtype check.
51 #
52 # This poset will evolve in a monotonous way:
53 # * Two non connected nodes will remain unconnected
54 # * New nodes can appear with new edges
55 private var intro_mtype_specialization_hierarchy = new POSet[MClassType]
56
57 # Global overlapped class-type hierarchy.
58 # The hierarchy when all modules are combined.
59 # Therefore, this hierarchy is used for a fast negative subtype check.
60 #
61 # This poset will evolve in an anarchic way. Loops can even be created.
62 #
63 # FIXME decide what to do on loops
64 private var full_mtype_specialization_hierarchy = new POSet[MClassType]
65
66 # Collections of classes grouped by their short name
67 private var mclasses_by_name = new MultiHashMap[String, MClass]
68
69 # Return all class named `name`.
70 #
71 # If such a class does not exist, null is returned
72 # (instead of an empty array)
73 #
74 # Visibility or modules are not considered
75 fun get_mclasses_by_name(name: String): nullable Array[MClass]
76 do
77 if mclasses_by_name.has_key(name) then
78 return mclasses_by_name[name]
79 else
80 return null
81 end
82 end
83
84 # Collections of properties grouped by their short name
85 private var mproperties_by_name = new MultiHashMap[String, MProperty]
86
87 # Return all properties named `name`.
88 #
89 # If such a property does not exist, null is returned
90 # (instead of an empty array)
91 #
92 # Visibility or modules are not considered
93 fun get_mproperties_by_name(name: String): nullable Array[MProperty]
94 do
95 if not mproperties_by_name.has_key(name) then
96 return null
97 else
98 return mproperties_by_name[name]
99 end
100 end
101
102 # The only null type
103 var null_type = new MNullType(self)
104
105 # Build an ordered tree with from `concerns`
106 fun concerns_tree(mconcerns: Collection[MConcern]): ConcernsTree do
107 var seen = new HashSet[MConcern]
108 var res = new ConcernsTree
109
110 var todo = new Array[MConcern]
111 todo.add_all mconcerns
112
113 while not todo.is_empty do
114 var c = todo.pop
115 if seen.has(c) then continue
116 var pc = c.parent_concern
117 if pc == null then
118 res.add(null, c)
119 else
120 res.add(pc, c)
121 todo.add(pc)
122 end
123 seen.add(c)
124 end
125
126 return res
127 end
128 end
129
130 # An OrderedTree that can be easily refined for display purposes
131 class ConcernsTree
132 super OrderedTree[MConcern]
133 end
134
135 redef class MModule
136 # All the classes introduced in the module
137 var intro_mclasses = new Array[MClass]
138
139 # All the class definitions of the module
140 # (introduction and refinement)
141 var mclassdefs = new Array[MClassDef]
142
143 # Does the current module has a given class `mclass`?
144 # Return true if the mmodule introduces, refines or imports a class.
145 # Visibility is not considered.
146 fun has_mclass(mclass: MClass): Bool
147 do
148 return self.in_importation <= mclass.intro_mmodule
149 end
150
151 # Full hierarchy of introduced ans imported classes.
152 #
153 # Create a new hierarchy got by flattening the classes for the module
154 # and its imported modules.
155 # Visibility is not considered.
156 #
157 # Note: this function is expensive and is usually used for the main
158 # module of a program only. Do not use it to do you own subtype
159 # functions.
160 fun flatten_mclass_hierarchy: POSet[MClass]
161 do
162 var res = self.flatten_mclass_hierarchy_cache
163 if res != null then return res
164 res = new POSet[MClass]
165 for m in self.in_importation.greaters do
166 for cd in m.mclassdefs do
167 var c = cd.mclass
168 res.add_node(c)
169 for s in cd.supertypes do
170 res.add_edge(c, s.mclass)
171 end
172 end
173 end
174 self.flatten_mclass_hierarchy_cache = res
175 return res
176 end
177
178 # Sort a given array of classes using the linearization order of the module
179 # The most general is first, the most specific is last
180 fun linearize_mclasses(mclasses: Array[MClass])
181 do
182 self.flatten_mclass_hierarchy.sort(mclasses)
183 end
184
185 # Sort a given array of class definitions using the linearization order of the module
186 # the refinement link is stronger than the specialisation link
187 # The most general is first, the most specific is last
188 fun linearize_mclassdefs(mclassdefs: Array[MClassDef])
189 do
190 var sorter = new MClassDefSorter(self)
191 sorter.sort(mclassdefs)
192 end
193
194 # Sort a given array of property definitions using the linearization order of the module
195 # the refinement link is stronger than the specialisation link
196 # The most general is first, the most specific is last
197 fun linearize_mpropdefs(mpropdefs: Array[MPropDef])
198 do
199 var sorter = new MPropDefSorter(self)
200 sorter.sort(mpropdefs)
201 end
202
203 private var flatten_mclass_hierarchy_cache: nullable POSet[MClass] = null
204
205 # The primitive type `Object`, the root of the class hierarchy
206 fun object_type: MClassType
207 do
208 var res = self.object_type_cache
209 if res != null then return res
210 res = self.get_primitive_class("Object").mclass_type
211 self.object_type_cache = res
212 return res
213 end
214
215 private var object_type_cache: nullable MClassType
216
217 # The type `Pointer`, super class to all extern classes
218 var pointer_type: MClassType = self.get_primitive_class("Pointer").mclass_type is lazy
219
220 # The primitive type `Bool`
221 fun bool_type: MClassType
222 do
223 var res = self.bool_type_cache
224 if res != null then return res
225 res = self.get_primitive_class("Bool").mclass_type
226 self.bool_type_cache = res
227 return res
228 end
229
230 private var bool_type_cache: nullable MClassType
231
232 # The primitive type `Sys`, the main type of the program, if any
233 fun sys_type: nullable MClassType
234 do
235 var clas = self.model.get_mclasses_by_name("Sys")
236 if clas == null then return null
237 return get_primitive_class("Sys").mclass_type
238 end
239
240 # The primitive type `Finalizable`
241 # Used to tag classes that need to be finalized.
242 fun finalizable_type: nullable MClassType
243 do
244 var clas = self.model.get_mclasses_by_name("Finalizable")
245 if clas == null then return null
246 return get_primitive_class("Finalizable").mclass_type
247 end
248
249 # Force to get the primitive class named `name` or abort
250 fun get_primitive_class(name: String): MClass
251 do
252 var cla = self.model.get_mclasses_by_name(name)
253 if cla == null then
254 if name == "Bool" then
255 var c = new MClass(self, name, null, enum_kind, public_visibility)
256 var cladef = new MClassDef(self, c.mclass_type, new Location(null, 0,0,0,0))
257 return c
258 end
259 print("Fatal Error: no primitive class {name}")
260 exit(1)
261 end
262 if cla.length != 1 then
263 var msg = "Fatal Error: more than one primitive class {name}:"
264 for c in cla do msg += " {c.full_name}"
265 print msg
266 exit(1)
267 end
268 return cla.first
269 end
270
271 # Try to get the primitive method named `name` on the type `recv`
272 fun try_get_primitive_method(name: String, recv: MClass): nullable MMethod
273 do
274 var props = self.model.get_mproperties_by_name(name)
275 if props == null then return null
276 var res: nullable MMethod = null
277 for mprop in props do
278 assert mprop isa MMethod
279 var intro = mprop.intro_mclassdef
280 for mclassdef in recv.mclassdefs do
281 if not self.in_importation.greaters.has(mclassdef.mmodule) then continue
282 if not mclassdef.in_hierarchy.greaters.has(intro) then continue
283 if res == null then
284 res = mprop
285 else if res != mprop then
286 print("Fatal Error: ambigous property name '{name}'; conflict between {mprop.full_name} and {res.full_name}")
287 abort
288 end
289 end
290 end
291 return res
292 end
293 end
294
295 private class MClassDefSorter
296 super Comparator
297 redef type COMPARED: MClassDef
298 var mmodule: MModule
299 redef fun compare(a, b)
300 do
301 var ca = a.mclass
302 var cb = b.mclass
303 if ca != cb then return mmodule.flatten_mclass_hierarchy.compare(ca, cb)
304 return mmodule.model.mclassdef_hierarchy.compare(a, b)
305 end
306 end
307
308 private class MPropDefSorter
309 super Comparator
310 redef type COMPARED: MPropDef
311 var mmodule: MModule
312 redef fun compare(pa, pb)
313 do
314 var a = pa.mclassdef
315 var b = pb.mclassdef
316 var ca = a.mclass
317 var cb = b.mclass
318 if ca != cb then return mmodule.flatten_mclass_hierarchy.compare(ca, cb)
319 return mmodule.model.mclassdef_hierarchy.compare(a, b)
320 end
321 end
322
323 # A named class
324 #
325 # `MClass` are global to the model; it means that a `MClass` is not bound to a
326 # specific `MModule`.
327 #
328 # This characteristic helps the reasoning about classes in a program since a
329 # single `MClass` object always denote the same class.
330 #
331 # The drawback is that classes (`MClass`) contain almost nothing by themselves.
332 # These do not really have properties nor belong to a hierarchy since the property and the
333 # hierarchy of a class depends of the refinement in the modules.
334 #
335 # Most services on classes require the precision of a module, and no one can asks what are
336 # the super-classes of a class nor what are properties of a class without precising what is
337 # the module considered.
338 #
339 # For instance, during the typing of a source-file, the module considered is the module of the file.
340 # eg. the question *is the method `foo` exists in the class `Bar`?* must be reformulated into
341 # *is the method `foo` exists in the class `Bar` in the current module?*
342 #
343 # During some global analysis, the module considered may be the main module of the program.
344 class MClass
345 super MEntity
346
347 # The module that introduce the class
348 # While classes are not bound to a specific module,
349 # the introducing module is used for naming an visibility
350 var intro_mmodule: MModule
351
352 # The short name of the class
353 # In Nit, the name of a class cannot evolve in refinements
354 redef var name: String
355
356 # The canonical name of the class
357 # Example: `"owner::module::MyClass"`
358 fun full_name: String
359 do
360 return "{self.intro_mmodule.full_name}::{name}"
361 end
362
363 # The number of generic formal parameters
364 # 0 if the class is not generic
365 var arity: Int is noinit
366
367 # Each generic formal parameters in order.
368 # is empty if the class is not generic
369 var mparameters = new Array[MParameterType]
370
371 protected fun setup_parameter_names(parameter_names: nullable Array[String]) is
372 autoinit
373 do
374 if parameter_names == null then
375 self.arity = 0
376 else
377 self.arity = parameter_names.length
378 end
379
380 # Create the formal parameter types
381 if arity > 0 then
382 assert parameter_names != null
383 var mparametertypes = new Array[MParameterType]
384 for i in [0..arity[ do
385 var mparametertype = new MParameterType(self, i, parameter_names[i])
386 mparametertypes.add(mparametertype)
387 end
388 self.mparameters = mparametertypes
389 var mclass_type = new MGenericType(self, mparametertypes)
390 self.mclass_type = mclass_type
391 self.get_mtype_cache.add(mclass_type)
392 else
393 self.mclass_type = new MClassType(self)
394 end
395 end
396
397 # The kind of the class (interface, abstract class, etc.)
398 # In Nit, the kind of a class cannot evolve in refinements
399 var kind: MClassKind
400
401 # The visibility of the class
402 # In Nit, the visibility of a class cannot evolve in refinements
403 var visibility: MVisibility
404
405 init
406 do
407 intro_mmodule.intro_mclasses.add(self)
408 var model = intro_mmodule.model
409 model.mclasses_by_name.add_one(name, self)
410 model.mclasses.add(self)
411 end
412
413 redef fun model do return intro_mmodule.model
414
415 # All class definitions (introduction and refinements)
416 var mclassdefs = new Array[MClassDef]
417
418 # Alias for `name`
419 redef fun to_s do return self.name
420
421 # The definition that introduces the class.
422 #
423 # Warning: such a definition may not exist in the early life of the object.
424 # In this case, the method will abort.
425 var intro: MClassDef is noinit
426
427 # Return the class `self` in the class hierarchy of the module `mmodule`.
428 #
429 # SEE: `MModule::flatten_mclass_hierarchy`
430 # REQUIRE: `mmodule.has_mclass(self)`
431 fun in_hierarchy(mmodule: MModule): POSetElement[MClass]
432 do
433 return mmodule.flatten_mclass_hierarchy[self]
434 end
435
436 # The principal static type of the class.
437 #
438 # For non-generic class, mclass_type is the only `MClassType` based
439 # on self.
440 #
441 # For a generic class, the arguments are the formal parameters.
442 # i.e.: for the class Array[E:Object], the `mclass_type` is Array[E].
443 # If you want Array[Object] the see `MClassDef::bound_mtype`
444 #
445 # For generic classes, the mclass_type is also the way to get a formal
446 # generic parameter type.
447 #
448 # To get other types based on a generic class, see `get_mtype`.
449 #
450 # ENSURE: `mclass_type.mclass == self`
451 var mclass_type: MClassType is noinit
452
453 # Return a generic type based on the class
454 # Is the class is not generic, then the result is `mclass_type`
455 #
456 # REQUIRE: `mtype_arguments.length == self.arity`
457 fun get_mtype(mtype_arguments: Array[MType]): MClassType
458 do
459 assert mtype_arguments.length == self.arity
460 if self.arity == 0 then return self.mclass_type
461 for t in self.get_mtype_cache do
462 if t.arguments == mtype_arguments then
463 return t
464 end
465 end
466 var res = new MGenericType(self, mtype_arguments)
467 self.get_mtype_cache.add res
468 return res
469 end
470
471 private var get_mtype_cache = new Array[MGenericType]
472 end
473
474
475 # A definition (an introduction or a refinement) of a class in a module
476 #
477 # A `MClassDef` is associated with an explicit (or almost) definition of a
478 # class. Unlike `MClass`, a `MClassDef` is a local definition that belong to
479 # a specific class and a specific module, and contains declarations like super-classes
480 # or properties.
481 #
482 # It is the class definitions that are the backbone of most things in the model:
483 # ClassDefs are defined with regard with other classdefs.
484 # Refinement and specialization are combined to produce a big poset called the `Model::mclassdef_hierarchy`.
485 #
486 # Moreover, the extension and the intention of types is defined by looking at the MClassDefs.
487 class MClassDef
488 super MEntity
489
490 # The module where the definition is
491 var mmodule: MModule
492
493 # The associated `MClass`
494 var mclass: MClass is noinit
495
496 # The bounded type associated to the mclassdef
497 #
498 # For a non-generic class, `bound_mtype` and `mclass.mclass_type`
499 # are the same type.
500 #
501 # Example:
502 # For the classdef Array[E: Object], the bound_mtype is Array[Object].
503 # If you want Array[E], then see `mclass.mclass_type`
504 #
505 # ENSURE: `bound_mtype.mclass == self.mclass`
506 var bound_mtype: MClassType
507
508 # The origin of the definition
509 var location: Location
510
511 # Internal name combining the module and the class
512 # Example: "mymodule#MyClass"
513 redef var to_s: String is noinit
514
515 init
516 do
517 self.mclass = bound_mtype.mclass
518 mmodule.mclassdefs.add(self)
519 mclass.mclassdefs.add(self)
520 if mclass.intro_mmodule == mmodule then
521 assert not isset mclass._intro
522 mclass.intro = self
523 end
524 self.to_s = "{mmodule}#{mclass}"
525 end
526
527 # Actually the name of the `mclass`
528 redef fun name do return mclass.name
529
530 redef fun model do return mmodule.model
531
532 # All declared super-types
533 # FIXME: quite ugly but not better idea yet
534 var supertypes = new Array[MClassType]
535
536 # Register some super-types for the class (ie "super SomeType")
537 #
538 # The hierarchy must not already be set
539 # REQUIRE: `self.in_hierarchy == null`
540 fun set_supertypes(supertypes: Array[MClassType])
541 do
542 assert unique_invocation: self.in_hierarchy == null
543 var mmodule = self.mmodule
544 var model = mmodule.model
545 var mtype = self.bound_mtype
546
547 for supertype in supertypes do
548 self.supertypes.add(supertype)
549
550 # Register in full_type_specialization_hierarchy
551 model.full_mtype_specialization_hierarchy.add_edge(mtype, supertype)
552 # Register in intro_type_specialization_hierarchy
553 if mclass.intro_mmodule == mmodule and supertype.mclass.intro_mmodule == mmodule then
554 model.intro_mtype_specialization_hierarchy.add_edge(mtype, supertype)
555 end
556 end
557
558 end
559
560 # Collect the super-types (set by set_supertypes) to build the hierarchy
561 #
562 # This function can only invoked once by class
563 # REQUIRE: `self.in_hierarchy == null`
564 # ENSURE: `self.in_hierarchy != null`
565 fun add_in_hierarchy
566 do
567 assert unique_invocation: self.in_hierarchy == null
568 var model = mmodule.model
569 var res = model.mclassdef_hierarchy.add_node(self)
570 self.in_hierarchy = res
571 var mtype = self.bound_mtype
572
573 # Here we need to connect the mclassdef to its pairs in the mclassdef_hierarchy
574 # The simpliest way is to attach it to collect_mclassdefs
575 for mclassdef in mtype.collect_mclassdefs(mmodule) do
576 res.poset.add_edge(self, mclassdef)
577 end
578 end
579
580 # The view of the class definition in `mclassdef_hierarchy`
581 var in_hierarchy: nullable POSetElement[MClassDef] = null
582
583 # Is the definition the one that introduced `mclass`?
584 fun is_intro: Bool do return mclass.intro == self
585
586 # All properties introduced by the classdef
587 var intro_mproperties = new Array[MProperty]
588
589 # All property definitions in the class (introductions and redefinitions)
590 var mpropdefs = new Array[MPropDef]
591 end
592
593 # A global static type
594 #
595 # MType are global to the model; it means that a `MType` is not bound to a
596 # specific `MModule`.
597 # This characteristic helps the reasoning about static types in a program
598 # since a single `MType` object always denote the same type.
599 #
600 # However, because a `MType` is global, it does not really have properties
601 # nor have subtypes to a hierarchy since the property and the class hierarchy
602 # depends of a module.
603 # Moreover, virtual types an formal generic parameter types also depends on
604 # a receiver to have sense.
605 #
606 # Therefore, most method of the types require a module and an anchor.
607 # The module is used to know what are the classes and the specialization
608 # links.
609 # The anchor is used to know what is the bound of the virtual types and formal
610 # generic parameter types.
611 #
612 # MType are not directly usable to get properties. See the `anchor_to` method
613 # and the `MClassType` class.
614 #
615 # FIXME: the order of the parameters is not the best. We mus pick on from:
616 # * foo(mmodule, anchor, othertype)
617 # * foo(othertype, anchor, mmodule)
618 # * foo(anchor, mmodule, othertype)
619 # * foo(othertype, mmodule, anchor)
620 abstract class MType
621 super MEntity
622
623 redef fun name do return to_s
624
625 # Return true if `self` is an subtype of `sup`.
626 # The typing is done using the standard typing policy of Nit.
627 #
628 # REQUIRE: `anchor == null implies not self.need_anchor and not sup.need_anchor`
629 # REQUIRE: `anchor != null implies self.can_resolve_for(anchor, null, mmodule) and sup.can_resolve_for(anchor, null, mmodule)`
630 fun is_subtype(mmodule: MModule, anchor: nullable MClassType, sup: MType): Bool
631 do
632 var sub = self
633 if sub == sup then return true
634 if anchor == null then
635 assert not sub.need_anchor
636 assert not sup.need_anchor
637 else
638 assert sub.can_resolve_for(anchor, null, mmodule)
639 assert sup.can_resolve_for(anchor, null, mmodule)
640 end
641
642 # First, resolve the formal types to a common version in the receiver
643 # The trick here is that fixed formal type will be associated to the bound
644 # And unfixed formal types will be associated to a canonical formal type.
645 if sub isa MParameterType or sub isa MVirtualType then
646 assert anchor != null
647 sub = sub.resolve_for(anchor.mclass.mclass_type, anchor, mmodule, false)
648 end
649 if sup isa MParameterType or sup isa MVirtualType then
650 assert anchor != null
651 sup = sup.resolve_for(anchor.mclass.mclass_type, anchor, mmodule, false)
652 end
653
654 # Does `sup` accept null or not?
655 # Discard the nullable marker if it exists
656 var sup_accept_null = false
657 if sup isa MNullableType then
658 sup_accept_null = true
659 sup = sup.mtype
660 else if sup isa MNullType then
661 sup_accept_null = true
662 end
663
664 # Can `sub` provide null or not?
665 # Thus we can match with `sup_accept_null`
666 # Also discard the nullable marker if it exists
667 if sub isa MNullableType then
668 if not sup_accept_null then return false
669 sub = sub.mtype
670 else if sub isa MNullType then
671 return sup_accept_null
672 end
673 # Now the case of direct null and nullable is over.
674
675 # A unfixed formal type can only accept itself
676 if sup isa MParameterType or sup isa MVirtualType then
677 return sub == sup
678 end
679
680 # If `sub` is a formal type, then it is accepted if its bound is accepted
681 if sub isa MParameterType or sub isa MVirtualType then
682 assert anchor != null
683 sub = sub.anchor_to(mmodule, anchor)
684
685 # Manage the second layer of null/nullable
686 if sub isa MNullableType then
687 if not sup_accept_null then return false
688 sub = sub.mtype
689 else if sub isa MNullType then
690 return sup_accept_null
691 end
692 end
693
694 assert sub isa MClassType # It is the only remaining type
695
696 if sup isa MNullType then
697 # `sup` accepts only null
698 return false
699 end
700
701 assert sup isa MClassType # It is the only remaining type
702
703 # Now both are MClassType, we need to dig
704
705 if sub == sup then return true
706
707 if anchor == null then anchor = sub # UGLY: any anchor will work
708 var resolved_sub = sub.anchor_to(mmodule, anchor)
709 var res = resolved_sub.collect_mclasses(mmodule).has(sup.mclass)
710 if res == false then return false
711 if not sup isa MGenericType then return true
712 var sub2 = sub.supertype_to(mmodule, anchor, sup.mclass)
713 assert sub2.mclass == sup.mclass
714 for i in [0..sup.mclass.arity[ do
715 var sub_arg = sub2.arguments[i]
716 var sup_arg = sup.arguments[i]
717 res = sub_arg.is_subtype(mmodule, anchor, sup_arg)
718 if res == false then return false
719 end
720 return true
721 end
722
723 # The base class type on which self is based
724 #
725 # This base type is used to get property (an internally to perform
726 # unsafe type comparison).
727 #
728 # Beware: some types (like null) are not based on a class thus this
729 # method will crash
730 #
731 # Basically, this function transform the virtual types and parameter
732 # types to their bounds.
733 #
734 # Example
735 # class A end
736 # class B super A end
737 # class X end
738 # class Y super X end
739 # class G[T: A]
740 # type U: X
741 # end
742 # class H
743 # super G[B]
744 # redef type U: Y
745 # end
746 # Map[T,U] anchor_to H #-> Map[B,Y]
747 #
748 # Explanation of the example:
749 # In H, T is set to B, because "H super G[B]", and U is bound to Y,
750 # because "redef type U: Y". Therefore, Map[T, U] is bound to
751 # Map[B, Y]
752 #
753 # ENSURE: `not self.need_anchor implies result == self`
754 # ENSURE: `not result.need_anchor`
755 fun anchor_to(mmodule: MModule, anchor: MClassType): MType
756 do
757 if not need_anchor then return self
758 assert not anchor.need_anchor
759 # Just resolve to the anchor and clear all the virtual types
760 var res = self.resolve_for(anchor, null, mmodule, true)
761 assert not res.need_anchor
762 return res
763 end
764
765 # Does `self` contain a virtual type or a formal generic parameter type?
766 # In order to remove those types, you usually want to use `anchor_to`.
767 fun need_anchor: Bool do return true
768
769 # Return the supertype when adapted to a class.
770 #
771 # In Nit, for each super-class of a type, there is a equivalent super-type.
772 #
773 # Example:
774 # class G[T, U] end
775 # class H[V] super G[V, Bool] end
776 # H[Int] supertype_to G #-> G[Int, Bool]
777 #
778 # REQUIRE: `super_mclass` is a super-class of `self`
779 # REQUIRE: `self.need_anchor implies anchor != null and self.can_resolve_for(anchor, null, mmodule)`
780 # ENSURE: `result.mclass = super_mclass`
781 fun supertype_to(mmodule: MModule, anchor: nullable MClassType, super_mclass: MClass): MClassType
782 do
783 if super_mclass.arity == 0 then return super_mclass.mclass_type
784 if self isa MClassType and self.mclass == super_mclass then return self
785 var resolved_self
786 if self.need_anchor then
787 assert anchor != null
788 resolved_self = self.anchor_to(mmodule, anchor)
789 else
790 resolved_self = self
791 end
792 var supertypes = resolved_self.collect_mtypes(mmodule)
793 for supertype in supertypes do
794 if supertype.mclass == super_mclass then
795 # FIXME: Here, we stop on the first goal. Should we check others and detect inconsistencies?
796 return supertype.resolve_for(self, anchor, mmodule, false)
797 end
798 end
799 abort
800 end
801
802 # Replace formals generic types in self with resolved values in `mtype`
803 # If `cleanup_virtual` is true, then virtual types are also replaced
804 # with their bounds.
805 #
806 # This function returns self if `need_anchor` is false.
807 #
808 # ## Example 1
809 #
810 # class G[E] end
811 # class H[F] super G[F] end
812 # class X[Z] end
813 #
814 # * Array[E].resolve_for(H[Int]) #-> Array[Int]
815 # * Array[E].resolve_for(G[Z], X[Int]) #-> Array[Z]
816 #
817 # Explanation of the example:
818 # * Array[E].need_anchor is true because there is a formal generic parameter type E
819 # * E makes sense for H[Int] because E is a formal parameter of G and H specialize G
820 # * Since "H[F] super G[F]", E is in fact F for H
821 # * More specifically, in H[Int], E is Int
822 # * So, in H[Int], Array[E] is Array[Int]
823 #
824 # This function is mainly used to inherit a signature.
825 # Because, unlike `anchor_to`, we do not want a full resolution of
826 # a type but only an adapted version of it.
827 #
828 # ## Example 2
829 #
830 # class A[E]
831 # fun foo(e:E):E is abstract
832 # end
833 # class B super A[Int] end
834 #
835 # The signature on foo is (e: E): E
836 # If we resolve the signature for B, we get (e:Int):Int
837 #
838 # ## Example 3
839 #
840 # class A[E]
841 # fun foo(e:E) is abstract
842 # end
843 # class B[F]
844 # var a: A[Array[F]]
845 # fun bar do a.foo(x) # <- x is here
846 # end
847 #
848 # The first question is: is foo available on `a`?
849 #
850 # The static type of a is `A[Array[F]]`, that is an open type.
851 # in order to find a method `foo`, whe must look at a resolved type.
852 #
853 # A[Array[F]].anchor_to(B[nullable Object]) #-> A[Array[nullable Object]]
854 #
855 # the method `foo` exists in `A[Array[nullable Object]]`, therefore `foo` exists for `a`.
856 #
857 # The next question is: what is the accepted types for `x`?
858 #
859 # the signature of `foo` is `foo(e:E)`, thus we must resolve the type E
860 #
861 # E.resolve_for(A[Array[F]],B[nullable Object]) #-> Array[F]
862 #
863 # The resolution can be done because `E` make sense for the class A (see `can_resolve_for`)
864 #
865 # FIXME: the parameter `cleanup_virtual` is just a bad idea, but having
866 # two function instead of one seems also to be a bad idea.
867 #
868 # REQUIRE: `can_resolve_for(mtype, anchor, mmodule)`
869 # ENSURE: `not self.need_anchor implies result == self`
870 fun resolve_for(mtype: MType, anchor: nullable MClassType, mmodule: MModule, cleanup_virtual: Bool): MType is abstract
871
872 # Resolve formal type to its verbatim bound.
873 # If the type is not formal, just return self
874 #
875 # The result is returned exactly as declared in the "type" property (verbatim).
876 # So it could be another formal type.
877 #
878 # In case of conflict, the method aborts.
879 fun lookup_bound(mmodule: MModule, resolved_receiver: MType): MType do return self
880 # Can the type be resolved?
881 #
882 # In order to resolve open types, the formal types must make sence.
883 #
884 # ## Example
885 #
886 # class A[E]
887 # end
888 # class B[F]
889 # end
890 #
891 # * E.can_resolve_for(A[Int]) #-> true, E make sense in A
892 # * E.can_resolve_for(B[Int]) #-> false, E does not make sense in B
893 # * B[E].can_resolve_for(A[F], B[Object]) #-> true,
894 # B[E] is a red hearing only the E is important,
895 # E make sense in A
896 #
897 # REQUIRE: `anchor != null implies not anchor.need_anchor`
898 # REQUIRE: `mtype.need_anchor implies anchor != null and mtype.can_resolve_for(anchor, null, mmodule)`
899 # ENSURE: `not self.need_anchor implies result == true`
900 fun can_resolve_for(mtype: MType, anchor: nullable MClassType, mmodule: MModule): Bool is abstract
901
902 # Return the nullable version of the type
903 # If the type is already nullable then self is returned
904 fun as_nullable: MType
905 do
906 var res = self.as_nullable_cache
907 if res != null then return res
908 res = new MNullableType(self)
909 self.as_nullable_cache = res
910 return res
911 end
912
913 # Return the not nullable version of the type
914 # Is the type is already not nullable, then self is returned.
915 #
916 # Note: this just remove the `nullable` notation, but the result can still contains null.
917 # For instance if `self isa MNullType` or self is a formal type bounded by a nullable type.
918 fun as_notnullable: MType
919 do
920 return self
921 end
922
923 private var as_nullable_cache: nullable MType = null
924
925
926 # The depth of the type seen as a tree.
927 #
928 # * A -> 1
929 # * G[A] -> 2
930 # * H[A, B] -> 2
931 # * H[G[A], B] -> 3
932 #
933 # Formal types have a depth of 1.
934 fun depth: Int
935 do
936 return 1
937 end
938
939 # The length of the type seen as a tree.
940 #
941 # * A -> 1
942 # * G[A] -> 2
943 # * H[A, B] -> 3
944 # * H[G[A], B] -> 4
945 #
946 # Formal types have a length of 1.
947 fun length: Int
948 do
949 return 1
950 end
951
952 # Compute all the classdefs inherited/imported.
953 # The returned set contains:
954 # * the class definitions from `mmodule` and its imported modules
955 # * the class definitions of this type and its super-types
956 #
957 # This function is used mainly internally.
958 #
959 # REQUIRE: `not self.need_anchor`
960 fun collect_mclassdefs(mmodule: MModule): Set[MClassDef] is abstract
961
962 # Compute all the super-classes.
963 # This function is used mainly internally.
964 #
965 # REQUIRE: `not self.need_anchor`
966 fun collect_mclasses(mmodule: MModule): Set[MClass] is abstract
967
968 # Compute all the declared super-types.
969 # Super-types are returned as declared in the classdefs (verbatim).
970 # This function is used mainly internally.
971 #
972 # REQUIRE: `not self.need_anchor`
973 fun collect_mtypes(mmodule: MModule): Set[MClassType] is abstract
974
975 # Is the property in self for a given module
976 # This method does not filter visibility or whatever
977 #
978 # REQUIRE: `not self.need_anchor`
979 fun has_mproperty(mmodule: MModule, mproperty: MProperty): Bool
980 do
981 assert not self.need_anchor
982 return self.collect_mclassdefs(mmodule).has(mproperty.intro_mclassdef)
983 end
984 end
985
986 # A type based on a class.
987 #
988 # `MClassType` have properties (see `has_mproperty`).
989 class MClassType
990 super MType
991
992 # The associated class
993 var mclass: MClass
994
995 redef fun model do return self.mclass.intro_mmodule.model
996
997 # TODO: private init because strongly bounded to its mclass. see `mclass.mclass_type`
998
999 # The formal arguments of the type
1000 # ENSURE: `result.length == self.mclass.arity`
1001 var arguments = new Array[MType]
1002
1003 redef fun to_s do return mclass.to_s
1004
1005 redef fun need_anchor do return false
1006
1007 redef fun anchor_to(mmodule: MModule, anchor: MClassType): MClassType
1008 do
1009 return super.as(MClassType)
1010 end
1011
1012 redef fun resolve_for(mtype: MType, anchor: nullable MClassType, mmodule: MModule, cleanup_virtual: Bool): MClassType do return self
1013
1014 redef fun can_resolve_for(mtype, anchor, mmodule) do return true
1015
1016 redef fun collect_mclassdefs(mmodule)
1017 do
1018 assert not self.need_anchor
1019 var cache = self.collect_mclassdefs_cache
1020 if not cache.has_key(mmodule) then
1021 self.collect_things(mmodule)
1022 end
1023 return cache[mmodule]
1024 end
1025
1026 redef fun collect_mclasses(mmodule)
1027 do
1028 assert not self.need_anchor
1029 var cache = self.collect_mclasses_cache
1030 if not cache.has_key(mmodule) then
1031 self.collect_things(mmodule)
1032 end
1033 return cache[mmodule]
1034 end
1035
1036 redef fun collect_mtypes(mmodule)
1037 do
1038 assert not self.need_anchor
1039 var cache = self.collect_mtypes_cache
1040 if not cache.has_key(mmodule) then
1041 self.collect_things(mmodule)
1042 end
1043 return cache[mmodule]
1044 end
1045
1046 # common implementation for `collect_mclassdefs`, `collect_mclasses`, and `collect_mtypes`.
1047 private fun collect_things(mmodule: MModule)
1048 do
1049 var res = new HashSet[MClassDef]
1050 var seen = new HashSet[MClass]
1051 var types = new HashSet[MClassType]
1052 seen.add(self.mclass)
1053 var todo = [self.mclass]
1054 while not todo.is_empty do
1055 var mclass = todo.pop
1056 #print "process {mclass}"
1057 for mclassdef in mclass.mclassdefs do
1058 if not mmodule.in_importation <= mclassdef.mmodule then continue
1059 #print " process {mclassdef}"
1060 res.add(mclassdef)
1061 for supertype in mclassdef.supertypes do
1062 types.add(supertype)
1063 var superclass = supertype.mclass
1064 if seen.has(superclass) then continue
1065 #print " add {superclass}"
1066 seen.add(superclass)
1067 todo.add(superclass)
1068 end
1069 end
1070 end
1071 collect_mclassdefs_cache[mmodule] = res
1072 collect_mclasses_cache[mmodule] = seen
1073 collect_mtypes_cache[mmodule] = types
1074 end
1075
1076 private var collect_mclassdefs_cache = new HashMap[MModule, Set[MClassDef]]
1077 private var collect_mclasses_cache = new HashMap[MModule, Set[MClass]]
1078 private var collect_mtypes_cache = new HashMap[MModule, Set[MClassType]]
1079
1080 end
1081
1082 # A type based on a generic class.
1083 # A generic type a just a class with additional formal generic arguments.
1084 class MGenericType
1085 super MClassType
1086
1087 redef var arguments
1088
1089 # TODO: private init because strongly bounded to its mclass. see `mclass.get_mtype`
1090
1091 init
1092 do
1093 assert self.mclass.arity == arguments.length
1094
1095 self.need_anchor = false
1096 for t in arguments do
1097 if t.need_anchor then
1098 self.need_anchor = true
1099 break
1100 end
1101 end
1102
1103 self.to_s = "{mclass}[{arguments.join(", ")}]"
1104 end
1105
1106 # Recursively print the type of the arguments within brackets.
1107 # Example: `"Map[String, List[Int]]"`
1108 redef var to_s: String is noinit
1109
1110 redef var need_anchor: Bool is noinit
1111
1112 redef fun resolve_for(mtype, anchor, mmodule, cleanup_virtual)
1113 do
1114 if not need_anchor then return self
1115 assert can_resolve_for(mtype, anchor, mmodule)
1116 var types = new Array[MType]
1117 for t in arguments do
1118 types.add(t.resolve_for(mtype, anchor, mmodule, cleanup_virtual))
1119 end
1120 return mclass.get_mtype(types)
1121 end
1122
1123 redef fun can_resolve_for(mtype, anchor, mmodule)
1124 do
1125 if not need_anchor then return true
1126 for t in arguments do
1127 if not t.can_resolve_for(mtype, anchor, mmodule) then return false
1128 end
1129 return true
1130 end
1131
1132
1133 redef fun depth
1134 do
1135 var dmax = 0
1136 for a in self.arguments do
1137 var d = a.depth
1138 if d > dmax then dmax = d
1139 end
1140 return dmax + 1
1141 end
1142
1143 redef fun length
1144 do
1145 var res = 1
1146 for a in self.arguments do
1147 res += a.length
1148 end
1149 return res
1150 end
1151 end
1152
1153 # A virtual formal type.
1154 class MVirtualType
1155 super MType
1156
1157 # The property associated with the type.
1158 # Its the definitions of this property that determine the bound or the virtual type.
1159 var mproperty: MVirtualTypeProp
1160
1161 redef fun model do return self.mproperty.intro_mclassdef.mmodule.model
1162
1163 redef fun lookup_bound(mmodule: MModule, resolved_receiver: MType): MType
1164 do
1165 return lookup_single_definition(mmodule, resolved_receiver).bound.as(not null)
1166 end
1167
1168 private fun lookup_single_definition(mmodule: MModule, resolved_receiver: MType): MVirtualTypeDef
1169 do
1170 assert not resolved_receiver.need_anchor
1171 var props = self.mproperty.lookup_definitions(mmodule, resolved_receiver)
1172 if props.is_empty then
1173 abort
1174 else if props.length == 1 then
1175 return props.first
1176 end
1177 var types = new ArraySet[MType]
1178 var res = props.first
1179 for p in props do
1180 types.add(p.bound.as(not null))
1181 end
1182 if types.length == 1 then
1183 return res
1184 end
1185 abort
1186 end
1187
1188 # Is the virtual type fixed for a given resolved_receiver?
1189 fun is_fixed(mmodule: MModule, resolved_receiver: MType): Bool
1190 do
1191 assert not resolved_receiver.need_anchor
1192 var props = self.mproperty.lookup_definitions(mmodule, resolved_receiver)
1193 if props.is_empty then
1194 abort
1195 end
1196 for p in props do
1197 if p.as(MVirtualTypeDef).is_fixed then return true
1198 end
1199 return false
1200 end
1201
1202 redef fun resolve_for(mtype, anchor, mmodule, cleanup_virtual)
1203 do
1204 assert can_resolve_for(mtype, anchor, mmodule)
1205 # self is a virtual type declared (or inherited) in mtype
1206 # The point of the function it to get the bound of the virtual type that make sense for mtype
1207 # But because mtype is maybe a virtual/formal type, we need to get a real receiver first
1208 #print "{class_name}: {self}/{mtype}/{anchor}?"
1209 var resolved_reciever
1210 if mtype.need_anchor then
1211 assert anchor != null
1212 resolved_reciever = mtype.resolve_for(anchor, null, mmodule, true)
1213 else
1214 resolved_reciever = mtype
1215 end
1216 # Now, we can get the bound
1217 var verbatim_bound = lookup_bound(mmodule, resolved_reciever)
1218 # The bound is exactly as declared in the "type" property, so we must resolve it again
1219 var res = verbatim_bound.resolve_for(mtype, anchor, mmodule, cleanup_virtual)
1220 #print "{class_name}: {self}/{mtype}/{anchor} -> {self}/{resolved_receiver}/{anchor} -> {verbatim_bound}/{mtype}/{anchor} -> {res}"
1221
1222 # What to return here? There is a bunch a special cases:
1223 # If 'cleanup_virtual' we must return the resolved type, since we cannot return self
1224 if cleanup_virtual then return res
1225 # If the receiver is a intern class, then the virtual type cannot be redefined since there is no possible subclass. self is just fixed. so simply return the resolution
1226 if resolved_reciever isa MNullableType then resolved_reciever = resolved_reciever.mtype
1227 if resolved_reciever.as(MClassType).mclass.kind == enum_kind then return res
1228 # If the resolved type isa MVirtualType, it means that self was bound to it, and cannot be unbound. self is just fixed. so return the resolution.
1229 if res isa MVirtualType then return res
1230 # If we are final, just return the resolution
1231 if is_fixed(mmodule, resolved_reciever) then return res
1232 # If the resolved type isa intern class, then there is no possible valid redefinition in any potential subclass. self is just fixed. so simply return the resolution
1233 if res isa MClassType and res.mclass.kind == enum_kind then return res
1234 # TODO: What if bound to a MParameterType?
1235 # Note that Nullable types can always be redefined by the non nullable version, so there is no specific case on it.
1236
1237 # If anything apply, then `self' cannot be resolved, so return self
1238 return self
1239 end
1240
1241 redef fun can_resolve_for(mtype, anchor, mmodule)
1242 do
1243 if mtype.need_anchor then
1244 assert anchor != null
1245 mtype = mtype.anchor_to(mmodule, anchor)
1246 end
1247 return mtype.has_mproperty(mmodule, mproperty)
1248 end
1249
1250 redef fun to_s do return self.mproperty.to_s
1251 end
1252
1253 # The type associated to a formal parameter generic type of a class
1254 #
1255 # Each parameter type is associated to a specific class.
1256 # It means that all refinements of a same class "share" the parameter type,
1257 # but that a generic subclass has its own parameter types.
1258 #
1259 # However, in the sense of the meta-model, a parameter type of a class is
1260 # a valid type in a subclass. The "in the sense of the meta-model" is
1261 # important because, in the Nit language, the programmer cannot refers
1262 # directly to the parameter types of the super-classes.
1263 #
1264 # Example:
1265 # class A[E]
1266 # fun e: E is abstract
1267 # end
1268 # class B[F]
1269 # super A[Array[F]]
1270 # end
1271 # In the class definition B[F], `F` is a valid type but `E` is not.
1272 # However, `self.e` is a valid method call, and the signature of `e` is
1273 # declared `e: E`.
1274 #
1275 # Note that parameter types are shared among class refinements.
1276 # Therefore parameter only have an internal name (see `to_s` for details).
1277 class MParameterType
1278 super MType
1279
1280 # The generic class where the parameter belong
1281 var mclass: MClass
1282
1283 redef fun model do return self.mclass.intro_mmodule.model
1284
1285 # The position of the parameter (0 for the first parameter)
1286 # FIXME: is `position` a better name?
1287 var rank: Int
1288
1289 redef var name
1290
1291 redef fun to_s do return name
1292
1293 redef fun lookup_bound(mmodule: MModule, resolved_receiver: MType): MType
1294 do
1295 assert not resolved_receiver.need_anchor
1296 var goalclass = self.mclass
1297 var supertypes = resolved_receiver.collect_mtypes(mmodule)
1298 for t in supertypes do
1299 if t.mclass == goalclass then
1300 # Yeah! c specialize goalclass with a "super `t'". So the question is what is the argument of f
1301 # FIXME: Here, we stop on the first goal. Should we check others and detect inconsistencies?
1302 var res = t.arguments[self.rank]
1303 return res
1304 end
1305 end
1306 abort
1307 end
1308
1309 redef fun resolve_for(mtype, anchor, mmodule, cleanup_virtual)
1310 do
1311 assert can_resolve_for(mtype, anchor, mmodule)
1312 #print "{class_name}: {self}/{mtype}/{anchor}?"
1313
1314 if mtype isa MGenericType and mtype.mclass == self.mclass then
1315 var res = mtype.arguments[self.rank]
1316 if anchor != null and res.need_anchor then
1317 # Maybe the result can be resolved more if are bound to a final class
1318 var r2 = res.anchor_to(mmodule, anchor)
1319 if r2 isa MClassType and r2.mclass.kind == enum_kind then return r2
1320 end
1321 return res
1322 end
1323
1324 # self is a parameter type of mtype (or of a super-class of mtype)
1325 # The point of the function it to get the bound of the virtual type that make sense for mtype
1326 # But because mtype is maybe a virtual/formal type, we need to get a real receiver first
1327 # FIXME: What happens here is far from clear. Thus this part must be validated and clarified
1328 var resolved_receiver
1329 if mtype.need_anchor then
1330 assert anchor != null
1331 resolved_receiver = mtype.resolve_for(anchor.mclass.mclass_type, anchor, mmodule, true)
1332 else
1333 resolved_receiver = mtype
1334 end
1335 if resolved_receiver isa MNullableType then resolved_receiver = resolved_receiver.mtype
1336 if resolved_receiver isa MParameterType then
1337 assert resolved_receiver.mclass == anchor.mclass
1338 resolved_receiver = anchor.arguments[resolved_receiver.rank]
1339 if resolved_receiver isa MNullableType then resolved_receiver = resolved_receiver.mtype
1340 end
1341 assert resolved_receiver isa MClassType
1342
1343 # Eh! The parameter is in the current class.
1344 # So we return the corresponding argument, no mater what!
1345 if resolved_receiver.mclass == self.mclass then
1346 var res = resolved_receiver.arguments[self.rank]
1347 #print "{class_name}: {self}/{mtype}/{anchor} -> direct {res}"
1348 return res
1349 end
1350
1351 if resolved_receiver.need_anchor then
1352 assert anchor != null
1353 resolved_receiver = resolved_receiver.resolve_for(anchor, null, mmodule, false)
1354 end
1355 # Now, we can get the bound
1356 var verbatim_bound = lookup_bound(mmodule, resolved_receiver)
1357 # The bound is exactly as declared in the "type" property, so we must resolve it again
1358 var res = verbatim_bound.resolve_for(mtype, anchor, mmodule, cleanup_virtual)
1359
1360 #print "{class_name}: {self}/{mtype}/{anchor} -> indirect {res}"
1361
1362 return res
1363 end
1364
1365 redef fun can_resolve_for(mtype, anchor, mmodule)
1366 do
1367 if mtype.need_anchor then
1368 assert anchor != null
1369 mtype = mtype.anchor_to(mmodule, anchor)
1370 end
1371 return mtype.collect_mclassdefs(mmodule).has(mclass.intro)
1372 end
1373 end
1374
1375 # A type prefixed with "nullable"
1376 class MNullableType
1377 super MType
1378
1379 # The base type of the nullable type
1380 var mtype: MType
1381
1382 redef fun model do return self.mtype.model
1383
1384 init
1385 do
1386 self.to_s = "nullable {mtype}"
1387 end
1388
1389 redef var to_s: String is noinit
1390
1391 redef fun need_anchor do return mtype.need_anchor
1392 redef fun as_nullable do return self
1393 redef fun as_notnullable do return mtype
1394 redef fun resolve_for(mtype, anchor, mmodule, cleanup_virtual)
1395 do
1396 var res = self.mtype.resolve_for(mtype, anchor, mmodule, cleanup_virtual)
1397 return res.as_nullable
1398 end
1399
1400 redef fun can_resolve_for(mtype, anchor, mmodule)
1401 do
1402 return self.mtype.can_resolve_for(mtype, anchor, mmodule)
1403 end
1404
1405 redef fun depth do return self.mtype.depth
1406
1407 redef fun length do return self.mtype.length
1408
1409 redef fun collect_mclassdefs(mmodule)
1410 do
1411 assert not self.need_anchor
1412 return self.mtype.collect_mclassdefs(mmodule)
1413 end
1414
1415 redef fun collect_mclasses(mmodule)
1416 do
1417 assert not self.need_anchor
1418 return self.mtype.collect_mclasses(mmodule)
1419 end
1420
1421 redef fun collect_mtypes(mmodule)
1422 do
1423 assert not self.need_anchor
1424 return self.mtype.collect_mtypes(mmodule)
1425 end
1426 end
1427
1428 # The type of the only value null
1429 #
1430 # The is only one null type per model, see `MModel::null_type`.
1431 class MNullType
1432 super MType
1433 redef var model: Model
1434 redef fun to_s do return "null"
1435 redef fun as_nullable do return self
1436 redef fun need_anchor do return false
1437 redef fun resolve_for(mtype, anchor, mmodule, cleanup_virtual) do return self
1438 redef fun can_resolve_for(mtype, anchor, mmodule) do return true
1439
1440 redef fun collect_mclassdefs(mmodule) do return new HashSet[MClassDef]
1441
1442 redef fun collect_mclasses(mmodule) do return new HashSet[MClass]
1443
1444 redef fun collect_mtypes(mmodule) do return new HashSet[MClassType]
1445 end
1446
1447 # A signature of a method
1448 class MSignature
1449 super MType
1450
1451 # The each parameter (in order)
1452 var mparameters: Array[MParameter]
1453
1454 # The return type (null for a procedure)
1455 var return_mtype: nullable MType
1456
1457 redef fun depth
1458 do
1459 var dmax = 0
1460 var t = self.return_mtype
1461 if t != null then dmax = t.depth
1462 for p in mparameters do
1463 var d = p.mtype.depth
1464 if d > dmax then dmax = d
1465 end
1466 return dmax + 1
1467 end
1468
1469 redef fun length
1470 do
1471 var res = 1
1472 var t = self.return_mtype
1473 if t != null then res += t.length
1474 for p in mparameters do
1475 res += p.mtype.length
1476 end
1477 return res
1478 end
1479
1480 # REQUIRE: 1 <= mparameters.count p -> p.is_vararg
1481 init
1482 do
1483 var vararg_rank = -1
1484 for i in [0..mparameters.length[ do
1485 var parameter = mparameters[i]
1486 if parameter.is_vararg then
1487 assert vararg_rank == -1
1488 vararg_rank = i
1489 end
1490 end
1491 self.vararg_rank = vararg_rank
1492 end
1493
1494 # The rank of the ellipsis (`...`) for vararg (starting from 0).
1495 # value is -1 if there is no vararg.
1496 # Example: for "(a: Int, b: Bool..., c: Char)" #-> vararg_rank=1
1497 var vararg_rank: Int is noinit
1498
1499 # The number or parameters
1500 fun arity: Int do return mparameters.length
1501
1502 redef fun to_s
1503 do
1504 var b = new FlatBuffer
1505 if not mparameters.is_empty then
1506 b.append("(")
1507 for i in [0..mparameters.length[ do
1508 var mparameter = mparameters[i]
1509 if i > 0 then b.append(", ")
1510 b.append(mparameter.name)
1511 b.append(": ")
1512 b.append(mparameter.mtype.to_s)
1513 if mparameter.is_vararg then
1514 b.append("...")
1515 end
1516 end
1517 b.append(")")
1518 end
1519 var ret = self.return_mtype
1520 if ret != null then
1521 b.append(": ")
1522 b.append(ret.to_s)
1523 end
1524 return b.to_s
1525 end
1526
1527 redef fun resolve_for(mtype: MType, anchor: nullable MClassType, mmodule: MModule, cleanup_virtual: Bool): MSignature
1528 do
1529 var params = new Array[MParameter]
1530 for p in self.mparameters do
1531 params.add(p.resolve_for(mtype, anchor, mmodule, cleanup_virtual))
1532 end
1533 var ret = self.return_mtype
1534 if ret != null then
1535 ret = ret.resolve_for(mtype, anchor, mmodule, cleanup_virtual)
1536 end
1537 var res = new MSignature(params, ret)
1538 return res
1539 end
1540 end
1541
1542 # A parameter in a signature
1543 class MParameter
1544 super MEntity
1545
1546 # The name of the parameter
1547 redef var name: String
1548
1549 # The static type of the parameter
1550 var mtype: MType
1551
1552 # Is the parameter a vararg?
1553 var is_vararg: Bool
1554
1555 redef fun to_s
1556 do
1557 if is_vararg then
1558 return "{name}: {mtype}..."
1559 else
1560 return "{name}: {mtype}"
1561 end
1562 end
1563
1564 fun resolve_for(mtype: MType, anchor: nullable MClassType, mmodule: MModule, cleanup_virtual: Bool): MParameter
1565 do
1566 if not self.mtype.need_anchor then return self
1567 var newtype = self.mtype.resolve_for(mtype, anchor, mmodule, cleanup_virtual)
1568 var res = new MParameter(self.name, newtype, self.is_vararg)
1569 return res
1570 end
1571
1572 redef fun model do return mtype.model
1573 end
1574
1575 # A service (global property) that generalize method, attribute, etc.
1576 #
1577 # `MProperty` are global to the model; it means that a `MProperty` is not bound
1578 # to a specific `MModule` nor a specific `MClass`.
1579 #
1580 # A MProperty gather definitions (see `mpropdefs`) ; one for the introduction
1581 # and the other in subclasses and in refinements.
1582 #
1583 # A `MProperty` is used to denotes services in polymorphic way (ie. independent
1584 # of any dynamic type).
1585 # For instance, a call site "x.foo" is associated to a `MProperty`.
1586 abstract class MProperty
1587 super MEntity
1588
1589 # The associated MPropDef subclass.
1590 # The two specialization hierarchy are symmetric.
1591 type MPROPDEF: MPropDef
1592
1593 # The classdef that introduce the property
1594 # While a property is not bound to a specific module, or class,
1595 # the introducing mclassdef is used for naming and visibility
1596 var intro_mclassdef: MClassDef
1597
1598 # The (short) name of the property
1599 redef var name: String
1600
1601 # The canonical name of the property
1602 # Example: "owner::my_module::MyClass::my_method"
1603 fun full_name: String
1604 do
1605 return "{self.intro_mclassdef.mmodule.full_name}::{self.intro_mclassdef.mclass.name}::{name}"
1606 end
1607
1608 # The visibility of the property
1609 var visibility: MVisibility
1610
1611 init
1612 do
1613 intro_mclassdef.intro_mproperties.add(self)
1614 var model = intro_mclassdef.mmodule.model
1615 model.mproperties_by_name.add_one(name, self)
1616 model.mproperties.add(self)
1617 end
1618
1619 # All definitions of the property.
1620 # The first is the introduction,
1621 # The other are redefinitions (in refinements and in subclasses)
1622 var mpropdefs = new Array[MPROPDEF]
1623
1624 # The definition that introduces the property.
1625 #
1626 # Warning: such a definition may not exist in the early life of the object.
1627 # In this case, the method will abort.
1628 var intro: MPROPDEF is noinit
1629
1630 redef fun model do return intro.model
1631
1632 # Alias for `name`
1633 redef fun to_s do return name
1634
1635 # Return the most specific property definitions defined or inherited by a type.
1636 # The selection knows that refinement is stronger than specialization;
1637 # however, in case of conflict more than one property are returned.
1638 # If mtype does not know mproperty then an empty array is returned.
1639 #
1640 # If you want the really most specific property, then look at `lookup_first_definition`
1641 fun lookup_definitions(mmodule: MModule, mtype: MType): Array[MPROPDEF]
1642 do
1643 assert not mtype.need_anchor
1644 mtype = mtype.as_notnullable
1645
1646 var cache = self.lookup_definitions_cache[mmodule, mtype]
1647 if cache != null then return cache
1648
1649 #print "select prop {mproperty} for {mtype} in {self}"
1650 # First, select all candidates
1651 var candidates = new Array[MPROPDEF]
1652 for mpropdef in self.mpropdefs do
1653 # If the definition is not imported by the module, then skip
1654 if not mmodule.in_importation <= mpropdef.mclassdef.mmodule then continue
1655 # If the definition is not inherited by the type, then skip
1656 if not mtype.is_subtype(mmodule, null, mpropdef.mclassdef.bound_mtype) then continue
1657 # Else, we keep it
1658 candidates.add(mpropdef)
1659 end
1660 # Fast track for only one candidate
1661 if candidates.length <= 1 then
1662 self.lookup_definitions_cache[mmodule, mtype] = candidates
1663 return candidates
1664 end
1665
1666 # Second, filter the most specific ones
1667 return select_most_specific(mmodule, candidates)
1668 end
1669
1670 private var lookup_definitions_cache = new HashMap2[MModule, MType, Array[MPROPDEF]]
1671
1672 # Return the most specific property definitions inherited by a type.
1673 # The selection knows that refinement is stronger than specialization;
1674 # however, in case of conflict more than one property are returned.
1675 # If mtype does not know mproperty then an empty array is returned.
1676 #
1677 # If you want the really most specific property, then look at `lookup_next_definition`
1678 #
1679 # FIXME: Move to `MPropDef`?
1680 fun lookup_super_definitions(mmodule: MModule, mtype: MType): Array[MPROPDEF]
1681 do
1682 assert not mtype.need_anchor
1683 mtype = mtype.as_notnullable
1684
1685 # First, select all candidates
1686 var candidates = new Array[MPROPDEF]
1687 for mpropdef in self.mpropdefs do
1688 # If the definition is not imported by the module, then skip
1689 if not mmodule.in_importation <= mpropdef.mclassdef.mmodule then continue
1690 # If the definition is not inherited by the type, then skip
1691 if not mtype.is_subtype(mmodule, null, mpropdef.mclassdef.bound_mtype) then continue
1692 # If the definition is defined by the type, then skip (we want the super, so e skip the current)
1693 if mtype == mpropdef.mclassdef.bound_mtype and mmodule == mpropdef.mclassdef.mmodule then continue
1694 # Else, we keep it
1695 candidates.add(mpropdef)
1696 end
1697 # Fast track for only one candidate
1698 if candidates.length <= 1 then return candidates
1699
1700 # Second, filter the most specific ones
1701 return select_most_specific(mmodule, candidates)
1702 end
1703
1704 # Return an array containing olny the most specific property definitions
1705 # This is an helper function for `lookup_definitions` and `lookup_super_definitions`
1706 private fun select_most_specific(mmodule: MModule, candidates: Array[MPROPDEF]): Array[MPROPDEF]
1707 do
1708 var res = new Array[MPROPDEF]
1709 for pd1 in candidates do
1710 var cd1 = pd1.mclassdef
1711 var c1 = cd1.mclass
1712 var keep = true
1713 for pd2 in candidates do
1714 if pd2 == pd1 then continue # do not compare with self!
1715 var cd2 = pd2.mclassdef
1716 var c2 = cd2.mclass
1717 if c2.mclass_type == c1.mclass_type then
1718 if cd2.mmodule.in_importation < cd1.mmodule then
1719 # cd2 refines cd1; therefore we skip pd1
1720 keep = false
1721 break
1722 end
1723 else if cd2.bound_mtype.is_subtype(mmodule, null, cd1.bound_mtype) and cd2.bound_mtype != cd1.bound_mtype then
1724 # cd2 < cd1; therefore we skip pd1
1725 keep = false
1726 break
1727 end
1728 end
1729 if keep then
1730 res.add(pd1)
1731 end
1732 end
1733 if res.is_empty then
1734 print "All lost! {candidates.join(", ")}"
1735 # FIXME: should be abort!
1736 end
1737 return res
1738 end
1739
1740 # Return the most specific definition in the linearization of `mtype`.
1741 #
1742 # If you want to know the next properties in the linearization,
1743 # look at `MPropDef::lookup_next_definition`.
1744 #
1745 # FIXME: the linearization is still unspecified
1746 #
1747 # REQUIRE: `not mtype.need_anchor`
1748 # REQUIRE: `mtype.has_mproperty(mmodule, self)`
1749 fun lookup_first_definition(mmodule: MModule, mtype: MType): MPROPDEF
1750 do
1751 assert mtype.has_mproperty(mmodule, self)
1752 return lookup_all_definitions(mmodule, mtype).first
1753 end
1754
1755 # Return all definitions in a linearization order
1756 # Most specific first, most general last
1757 fun lookup_all_definitions(mmodule: MModule, mtype: MType): Array[MPROPDEF]
1758 do
1759 assert not mtype.need_anchor
1760 mtype = mtype.as_notnullable
1761
1762 var cache = self.lookup_all_definitions_cache[mmodule, mtype]
1763 if cache != null then return cache
1764
1765 #print "select prop {mproperty} for {mtype} in {self}"
1766 # First, select all candidates
1767 var candidates = new Array[MPROPDEF]
1768 for mpropdef in self.mpropdefs do
1769 # If the definition is not imported by the module, then skip
1770 if not mmodule.in_importation <= mpropdef.mclassdef.mmodule then continue
1771 # If the definition is not inherited by the type, then skip
1772 if not mtype.is_subtype(mmodule, null, mpropdef.mclassdef.bound_mtype) then continue
1773 # Else, we keep it
1774 candidates.add(mpropdef)
1775 end
1776 # Fast track for only one candidate
1777 if candidates.length <= 1 then
1778 self.lookup_all_definitions_cache[mmodule, mtype] = candidates
1779 return candidates
1780 end
1781
1782 mmodule.linearize_mpropdefs(candidates)
1783 candidates = candidates.reversed
1784 self.lookup_all_definitions_cache[mmodule, mtype] = candidates
1785 return candidates
1786 end
1787
1788 private var lookup_all_definitions_cache = new HashMap2[MModule, MType, Array[MPROPDEF]]
1789 end
1790
1791 # A global method
1792 class MMethod
1793 super MProperty
1794
1795 redef type MPROPDEF: MMethodDef
1796
1797 # Is the property defined at the top_level of the module?
1798 # Currently such a property are stored in `Object`
1799 var is_toplevel: Bool = false is writable
1800
1801 # Is the property a constructor?
1802 # Warning, this property can be inherited by subclasses with or without being a constructor
1803 # therefore, you should use `is_init_for` the verify if the property is a legal constructor for a given class
1804 var is_init: Bool = false is writable
1805
1806 # The constructor is a (the) root init with empty signature but a set of initializers
1807 var is_root_init: Bool = false is writable
1808
1809 # Is the property a 'new' constructor?
1810 var is_new: Bool = false is writable
1811
1812 # Is the property a legal constructor for a given class?
1813 # As usual, visibility is not considered.
1814 # FIXME not implemented
1815 fun is_init_for(mclass: MClass): Bool
1816 do
1817 return self.is_init
1818 end
1819 end
1820
1821 # A global attribute
1822 class MAttribute
1823 super MProperty
1824
1825 redef type MPROPDEF: MAttributeDef
1826
1827 end
1828
1829 # A global virtual type
1830 class MVirtualTypeProp
1831 super MProperty
1832
1833 redef type MPROPDEF: MVirtualTypeDef
1834
1835 # The formal type associated to the virtual type property
1836 var mvirtualtype = new MVirtualType(self)
1837 end
1838
1839 # A definition of a property (local property)
1840 #
1841 # Unlike `MProperty`, a `MPropDef` is a local definition that belong to a
1842 # specific class definition (which belong to a specific module)
1843 abstract class MPropDef
1844 super MEntity
1845
1846 # The associated `MProperty` subclass.
1847 # the two specialization hierarchy are symmetric
1848 type MPROPERTY: MProperty
1849
1850 # Self class
1851 type MPROPDEF: MPropDef
1852
1853 # The class definition where the property definition is
1854 var mclassdef: MClassDef
1855
1856 # The associated global property
1857 var mproperty: MPROPERTY
1858
1859 # The origin of the definition
1860 var location: Location
1861
1862 init
1863 do
1864 mclassdef.mpropdefs.add(self)
1865 mproperty.mpropdefs.add(self)
1866 if mproperty.intro_mclassdef == mclassdef then
1867 assert not isset mproperty._intro
1868 mproperty.intro = self
1869 end
1870 self.to_s = "{mclassdef}#{mproperty}"
1871 end
1872
1873 # Actually the name of the `mproperty`
1874 redef fun name do return mproperty.name
1875
1876 redef fun model do return mclassdef.model
1877
1878 # Internal name combining the module, the class and the property
1879 # Example: "mymodule#MyClass#mymethod"
1880 redef var to_s: String is noinit
1881
1882 # Is self the definition that introduce the property?
1883 fun is_intro: Bool do return mproperty.intro == self
1884
1885 # Return the next definition in linearization of `mtype`.
1886 #
1887 # This method is used to determine what method is called by a super.
1888 #
1889 # REQUIRE: `not mtype.need_anchor`
1890 fun lookup_next_definition(mmodule: MModule, mtype: MType): MPROPDEF
1891 do
1892 assert not mtype.need_anchor
1893
1894 var mpropdefs = self.mproperty.lookup_all_definitions(mmodule, mtype)
1895 var i = mpropdefs.iterator
1896 while i.is_ok and i.item != self do i.next
1897 assert has_property: i.is_ok
1898 i.next
1899 assert has_next_property: i.is_ok
1900 return i.item
1901 end
1902 end
1903
1904 # A local definition of a method
1905 class MMethodDef
1906 super MPropDef
1907
1908 redef type MPROPERTY: MMethod
1909 redef type MPROPDEF: MMethodDef
1910
1911 # The signature attached to the property definition
1912 var msignature: nullable MSignature = null is writable
1913
1914 # The signature attached to the `new` call on a root-init
1915 # This is a concatenation of the signatures of the initializers
1916 #
1917 # REQUIRE `mproperty.is_root_init == (new_msignature != null)`
1918 var new_msignature: nullable MSignature = null is writable
1919
1920 # List of initialisers to call in root-inits
1921 #
1922 # They could be setters or attributes
1923 #
1924 # REQUIRE `mproperty.is_root_init == (new_msignature != null)`
1925 var initializers = new Array[MProperty]
1926
1927 # Is the method definition abstract?
1928 var is_abstract: Bool = false is writable
1929
1930 # Is the method definition intern?
1931 var is_intern = false is writable
1932
1933 # Is the method definition extern?
1934 var is_extern = false is writable
1935
1936 # An optional constant value returned in functions.
1937 #
1938 # Only some specific primitife value are accepted by engines.
1939 # Is used when there is no better implementation available.
1940 #
1941 # Currently used only for the implementation of the `--define`
1942 # command-line option.
1943 # SEE: module `mixin`.
1944 var constant_value: nullable Object = null is writable
1945 end
1946
1947 # A local definition of an attribute
1948 class MAttributeDef
1949 super MPropDef
1950
1951 redef type MPROPERTY: MAttribute
1952 redef type MPROPDEF: MAttributeDef
1953
1954 # The static type of the attribute
1955 var static_mtype: nullable MType = null is writable
1956 end
1957
1958 # A local definition of a virtual type
1959 class MVirtualTypeDef
1960 super MPropDef
1961
1962 redef type MPROPERTY: MVirtualTypeProp
1963 redef type MPROPDEF: MVirtualTypeDef
1964
1965 # The bound of the virtual type
1966 var bound: nullable MType = null is writable
1967
1968 # Is the bound fixed?
1969 var is_fixed = false is writable
1970 end
1971
1972 # A kind of class.
1973 #
1974 # * `abstract_kind`
1975 # * `concrete_kind`
1976 # * `interface_kind`
1977 # * `enum_kind`
1978 # * `extern_kind`
1979 #
1980 # Note this class is basically an enum.
1981 # FIXME: use a real enum once user-defined enums are available
1982 class MClassKind
1983 redef var to_s: String
1984
1985 # Is a constructor required?
1986 var need_init: Bool
1987
1988 # TODO: private init because enumeration.
1989
1990 # Can a class of kind `self` specializes a class of kine `other`?
1991 fun can_specialize(other: MClassKind): Bool
1992 do
1993 if other == interface_kind then return true # everybody can specialize interfaces
1994 if self == interface_kind or self == enum_kind then
1995 # no other case for interfaces
1996 return false
1997 else if self == extern_kind then
1998 # only compatible with themselves
1999 return self == other
2000 else if other == enum_kind or other == extern_kind then
2001 # abstract_kind and concrete_kind are incompatible
2002 return false
2003 end
2004 # remain only abstract_kind and concrete_kind
2005 return true
2006 end
2007 end
2008
2009 # The class kind `abstract`
2010 fun abstract_kind: MClassKind do return once new MClassKind("abstract class", true)
2011 # The class kind `concrete`
2012 fun concrete_kind: MClassKind do return once new MClassKind("class", true)
2013 # The class kind `interface`
2014 fun interface_kind: MClassKind do return once new MClassKind("interface", false)
2015 # The class kind `enum`
2016 fun enum_kind: MClassKind do return once new MClassKind("enum", false)
2017 # The class kind `extern`
2018 fun extern_kind: MClassKind do return once new MClassKind("extern class", false)