src/model: collect package 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 d4e40f4..bfe5fdf 100644 (file)
--- a/src/model/model_collect.nit
+++ b/src/model/model_collect.nit
@@ -154,6 +154,18 @@ redef class MPackage
                end
                return res
        end
+
+       # `MModules` contained in `self`.
+       fun collect_mmodules(view: ModelView): HashSet[MModule] do
+               var res = new HashSet[MModule]
+               for mgroup in mgroups do
+                       for mmodule in mgroup.mmodules do
+                               if not view.accept_mentity(mmodule) then continue
+                               res.add(mmodule)
+                       end
+               end
+               return res
+       end
 end
 
 redef class MGroup