From: Alexandre Terrasa <alexandre@moz-code.org>
Date: Wed, 25 Jun 2014 15:24:43 +0000 (-0400)
Subject: model_utils: collect mmodules in mproject and mgroups
X-Git-Tag: v0.6.6~16^2~1
X-Git-Url: http://nitlanguage.org

model_utils: collect mmodules in mproject and mgroups

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

diff --git a/src/model_utils.nit b/src/model_utils.nit
index f73ed8b..e1d0f52 100644
--- a/src/model_utils.nit
+++ b/src/model_utils.nit
@@ -51,6 +51,16 @@ redef class MGroup
 		for mgrp in lst do res.add_all mgrp.in_nesting_redef_mclassdefs(min_visibility)
 		return res
 	end
+
+	# Collect nested modules
+	fun collect_mmodules: Set[MModule] do
+		var res = new HashSet[MModule]
+		res.add_all mmodules
+		for mgroup in in_nesting.direct_smallers do
+			res.add_all mgroup.collect_mmodules
+		end
+		return res
+	end
 end
 
 redef class MModule