nituml: use model filters
authorAlexandre Terrasa <alexandre@moz-code.org>
Sun, 22 Oct 2017 21:11:32 +0000 (17:11 -0400)
committerAlexandre Terrasa <alexandre@moz-code.org>
Fri, 10 Nov 2017 17:29:12 +0000 (12:29 -0500)
Signed-off-by: Alexandre Terrasa <alexandre@moz-code.org>

src/nituml.nit

index ea838aa..0b9bf72 100644 (file)
@@ -41,11 +41,12 @@ private class UMLPhase
        super Phase
        redef fun process_mainmodule(mainmodule, mmodules)
        do
-               var view = new ModelView(mainmodule.model, mainmodule)
+               var filters = new ModelFilter
                if not toolcontext.opt_privacy.value then
-                       view.min_visibility = protected_visibility
+                       filters.min_visibility = protected_visibility
                end
 
+               var view = new ModelView(mainmodule.model, mainmodule, filters)
                var d = new UMLModel(view, mainmodule)
                if toolcontext.opt_gen.value == 0 then
                        print d.generate_class_uml.write_to_string