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