From 97ca04f317250f23b61064090f1214e534767b6a Mon Sep 17 00:00:00 2001 From: Alexandre Terrasa Date: Wed, 25 Jun 2014 11:24:43 -0400 Subject: [PATCH] model_utils: collect mmodules in mproject and mgroups Signed-off-by: Alexandre Terrasa --- src/model_utils.nit | 10 ++++++++++ 1 file changed, 10 insertions(+) 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 -- 1.7.9.5