From: Alexandre Terrasa <alexandre@moz-code.org>
Date: Tue, 9 Dec 2014 21:45:32 +0000 (-0500)
Subject: model_utils: creates a replacement for MModule::in_nesting.
X-Git-Tag: v0.7~61^2~2
X-Git-Url: http://nitlanguage.org

model_utils: creates a replacement for MModule::in_nesting.

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

diff --git a/src/model_utils.nit b/src/model_utils.nit
index 7af444d..1466d52 100644
--- a/src/model_utils.nit
+++ b/src/model_utils.nit
@@ -209,6 +209,23 @@ redef class MModule
 		end
 		return max + 1
 	end
+
+	# Find all mmodules nested in `self` if `self` is the default module of a `MGroup`.
+	fun nested_mmodules: Array[MModule] do
+		var res = new Array[MModule]
+		var mgroup = mgroup
+		if mgroup == null or self != mgroup.default_mmodule then return res
+		for mmodule in mgroup.mmodules do
+			if mmodule == self then continue
+			res.add mmodule
+		end
+		for nested in mgroup.in_nesting.direct_smallers do
+			var default = nested.default_mmodule
+			if default == null then continue
+			res.add default
+		end
+		return res
+	end
 end
 
 redef class MClass