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