- var d = new UMLModel(mainmodule.model, mainmodule, toolcontext)
- print d.generate_class_uml.write_to_string
+ 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
+ print d.generate_package_uml.write_to_string
+ end