Merge: doc: fixed some typos and other misc. corrections

[nit.git] / src / nituml.nit

diff --git a/src/nituml.nit b/src/nituml.nit
index 9ea9048..2723a8b 100644 (file)
--- a/src/nituml.nit
+++ b/src/nituml.nit
@@ -20,12 +20,19 @@ import frontend
 import uml
 
 redef class ToolContext
+
+       # Phase that generates UML diagrams from model entities.
        var umlphase: Phase = new UMLPhase(self, null)
 
+       # What to generate?
        var opt_gen = new OptionEnum(["class", "package"], "Choose which type of uml diagram to generate", 0, "--diagram")
 
+       # Generate private?
+       var opt_privacy = new OptionBool("Generates private API", "-p", "--private")
+
        redef init do
                option_context.add_option opt_gen
+               option_context.add_option opt_privacy
                super
        end
 end
@@ -34,7 +41,12 @@ private class UMLPhase
        super Phase
        redef fun process_mainmodule(mainmodule, mmodules)
        do
-               var d = new UMLModel(mainmodule.model, mainmodule, toolcontext)
+               var filters = new ModelFilter
+               if not toolcontext.opt_privacy.value then
+                       filters.min_visibility = protected_visibility
+               end
+
+               var d = new UMLModel(toolcontext.modelbuilder.model, mainmodule, filters)
                if toolcontext.opt_gen.value == 0 then
                        print d.generate_class_uml.write_to_string
                else if toolcontext.opt_gen.value == 1 then