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