model_collect: collect all mmodules for a mgroup
authorAlexandre Terrasa <alexandre@moz-code.org>
Mon, 7 May 2018 21:02:21 +0000 (17:02 -0400)
committerAlexandre Terrasa <alexandre@moz-code.org>
Thu, 10 May 2018 14:16:57 +0000 (10:16 -0400)
Signed-off-by: Alexandre Terrasa <alexandre@moz-code.org>

src/model/model_collect.nit

index 6edd044..4df13c6 100644 (file)
@@ -380,6 +380,15 @@ redef class MGroup
        end
 
        # Collect all modules contained in `self`
+       fun collect_all_mmodules(view: ModelView): HashSet[MModule] do
+               var res = new HashSet[MModule]
+               for mgroup in collect_mgroups(view) do
+                       res.add_all mgroup.collect_all_mmodules(view)
+               end
+               return res
+       end
+
+       # Collect all modules contained in `self`
        fun collect_mmodules(view: ModelView): HashSet[MModule] do
                var res = new HashSet[MModule]
                for mmodule in mmodules do