X-Git-Url: http://nitlanguage.org

diff --git a/src/uml/uml_base.nit b/src/uml/uml_base.nit
index e54ccef..04538f0 100644
--- a/src/uml/uml_base.nit
+++ b/src/uml/uml_base.nit
@@ -16,23 +16,17 @@
 module uml_base
 
 import toolcontext
-import model_utils
-
-redef class ToolContext
-	# -p
-	var opt_privacy = new OptionBool("Generates private API", "-p", "--private")
-
-	# Shortcut for the value of `self.opt_privacy`
-	fun private_gen: Bool do return opt_privacy.value
-
-	redef init do
-		option_context.add_option opt_privacy
-		super
-	end
-end
+import model::model_collect
 
+# UML model builder.
 class UMLModel
+
+	# Model to access mentities
 	var model: Model
+
+	# Main module used for linearization.
 	var mainmodule: MModule
-	var ctx: ToolContext
+
+	# Filters to apply on model if any
+	var filter: nullable ModelFilter
 end