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