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

diff --git a/src/metrics/nullables_metrics.nit b/src/metrics/nullables_metrics.nit
index 62c8f71..0cae25d 100644
--- a/src/metrics/nullables_metrics.nit
+++ b/src/metrics/nullables_metrics.nit
@@ -22,6 +22,8 @@ import mclasses_metrics
 import semantize
 
 redef class ToolContext
+
+	# Nullable types related metrics
 	var nullables_metrics_phase: Phase = new NullablesMetricsPhase(self, null)
 end
 
@@ -37,18 +39,19 @@ private class NullablesMetricsPhase
 
 		print toolcontext.format_h1("\n# Nullable metrics")
 
+		var model = toolcontext.modelbuilder.model
+		var filter = new ModelFilter(private_visibility)
+
 		var metrics = new MetricSet
-		var min_vis = private_visibility
-		metrics.register(new CNBA(mainmodule, min_vis))
-		metrics.register(new CNBNA(mainmodule, min_vis))
+		metrics.register(new CNBA(model, mainmodule, filter))
+		metrics.register(new CNBNA(model, mainmodule, filter))
 
-		var model = toolcontext.modelbuilder.model
 		var mclasses = new HashSet[MClass]
-		for mproject in model.mprojects do
+		for mpackage in model.mpackages do
 
-			print toolcontext.format_h2("\n ## project {mproject}")
+			print toolcontext.format_h2("\n ## package {mpackage}")
 
-			for mgroup in mproject.mgroups do
+			for mgroup in mpackage.mgroups do
 				if mgroup.mmodules.is_empty then continue
 				metrics.clear
 
@@ -60,7 +63,7 @@ private class NullablesMetricsPhase
 				mclasses.add_all(mod_mclasses)
 				metrics.collect(new HashSet[MClass].from(mod_mclasses))
 				metrics.to_console(1, not toolcontext.opt_nocolors.value)
-				if csv then metrics.to_csv.save("{out}/{mgroup}.csv")
+				if csv then metrics.to_csv.write_to_file("{out}/{mgroup}.csv")
 			end
 		end
 		if not mclasses.is_empty then
@@ -69,7 +72,7 @@ private class NullablesMetricsPhase
 			print toolcontext.format_h2("\n ## global metrics")
 			metrics.collect(mclasses)
 			metrics.to_console(1, not toolcontext.opt_nocolors.value)
-			if csv then metrics.to_csv.save("{out}/summary.csv")
+			if csv then metrics.to_csv.write_to_file("{out}/summary.csv")
 		end
 
 		compute_nullables_metrics(toolcontext.modelbuilder)
@@ -83,17 +86,9 @@ class CNBNA
 	redef fun name do return "cnbna"
 	redef fun desc do return "number of accessible nullable attributes (inherited + local)"
 
-	var mainmodule: MModule
-	var min_visibility: MVisibility
-
-	init(mainmodule: MModule, min_visibility: MVisibility) do
-		self.mainmodule = mainmodule
-		self.min_visibility = min_visibility
-	end
-
 	redef fun collect(mclasses) do
 		for mclass in mclasses do
-			var all = mclass.collect_accessible_mattributes(min_visibility)
+			var all = mclass.collect_accessible_mattributes(mainmodule, filter)
 			for mattr in all do
 				if mattr.is_nullable then values.inc(mclass)
 			end
@@ -113,7 +108,6 @@ end
 
 private class NullableSends
 	super Visitor
-	var modelbuilder: ModelBuilder
 	var nclassdef: AClassdef
 
 	var total_sends: Int = 0
@@ -121,13 +115,6 @@ private class NullableSends
 	var nullable_eq_sends: Int = 0
 	var buggy_sends: Int = 0
 
-	# Get a new visitor on a classef to add type count in `typecount`.
-	init(modelbuilder: ModelBuilder, nclassdef: AClassdef)
-	do
-		self.modelbuilder = modelbuilder
-		self.nclassdef = nclassdef
-	end
-
 	redef fun visit(n)
 	do
 		n.visit_all(self)
@@ -167,7 +154,7 @@ do
 	# Visit all the source code to collect data
 	for nmodule in modelbuilder.nmodules do
 		for nclassdef in nmodule.n_classdefs do
-			var visitor = new NullableSends(modelbuilder, nclassdef)
+			var visitor = new NullableSends(nclassdef)
 			visitor.enter_visit(nclassdef)
 			total_sends += visitor.total_sends
 			nullable_sends += visitor.nullable_sends