model_collect: collect importation poset for modules

Signed-off-by: Alexandre Terrasa <alexandre@moz-code.org>

diff --git a/src/model/model_collect.nit b/src/model/model_collect.nit
index 65107d9..c8d3766 100644 (file)
--- a/src/model/model_collect.nit
+++ b/src/model/model_collect.nit
@@ -77,6 +77,17 @@ redef class MModule
                return res
        end
 
+       # Build the importation poset for `self`
+       fun importation_poset(view: ModelView): POSet[MModule] do
+               var mmodules = new HashSet[MModule]
+               mmodules.add self
+               mmodules.add_all collect_ancestors(view)
+               mmodules.add_all collect_parents(view)
+               mmodules.add_all collect_children(view)
+               mmodules.add_all collect_descendants(view)
+               return view.mmodules_poset(mmodules)
+       end
+
        # Collect mclassdefs introduced in `self` with `visibility >= to min_visibility`.
        fun collect_intro_mclassdefs(view: ModelView): Set[MClassDef] do
                var res = new HashSet[MClassDef]