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