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

diff --git a/src/rapid_type_analysis.nit b/src/rapid_type_analysis.nit
index 897d7b7..7fcb4bc 100644
--- a/src/rapid_type_analysis.nit
+++ b/src/rapid_type_analysis.nit
@@ -215,7 +215,7 @@ class RapidTypeAnalysis
 
 			var vararg_rank = mmethoddef.msignature.vararg_rank
 			if vararg_rank > -1 then
-				var node = self.modelbuilder.mpropdef2npropdef[mmethoddef]
+				var node = self.modelbuilder.mpropdef2node(mmethoddef)
 				var elttype = mmethoddef.msignature.mparameters[vararg_rank].mtype
 				#elttype = elttype.anchor_to(self.mainmodule, v.receiver)
 				var vararg = self.mainmodule.get_primitive_class("Array").get_mtype([elttype])
@@ -235,27 +235,25 @@ class RapidTypeAnalysis
 				add_cast(paramtype)
 			end
 
-			if not modelbuilder.mpropdef2npropdef.has_key(mmethoddef) then
-				# It is an init for a class?
-				if mmeth.name == "init" then
-					var nclassdef = self.modelbuilder.mclassdef2nclassdef[mmethoddef.mclassdef]
-					assert mmethoddef == nclassdef.mfree_init
+			var npropdef = modelbuilder.mpropdef2node(mmethoddef)
 
-					if mmethoddef.mproperty.is_root_init and not mmethoddef.is_intro then
-						self.add_super_send(v.receiver, mmethoddef)
-					end
-				else if mmethoddef.constant_value != null then
-					# Make the return type live
-					v.add_type(mmethoddef.msignature.return_mtype.as(MClassType))
-				else
-					abort
+			if npropdef isa AClassdef then
+				# It is an init for a class
+				assert mmethoddef == npropdef.mfree_init
+
+				if mmethoddef.mproperty.is_root_init and not mmethoddef.is_intro then
+					self.add_super_send(v.receiver, mmethoddef)
 				end
 				continue
+			else if mmethoddef.constant_value != null then
+				# Make the return type live
+				v.add_type(mmethoddef.msignature.return_mtype.as(MClassType))
+				continue
+			else if npropdef == null then
+				abort
 			end
 
-			var npropdef = modelbuilder.mpropdef2npropdef[mmethoddef]
-
-			if npropdef isa AMethPropdef  then
+			if npropdef isa AMethPropdef then
 				var auto_super_inits = npropdef.auto_super_inits
 				if auto_super_inits != null then
 					for auto_super_init in auto_super_inits do
@@ -267,9 +265,7 @@ class RapidTypeAnalysis
 				end
 			end
 
-			if mmeth.is_new then
-				v.add_type(v.receiver)
-			else if mmethoddef.is_intern or mmethoddef.is_extern then
+			if mmethoddef.is_intern or mmethoddef.is_extern then
 				# UGLY: We force the "instantation" of the concrete return type if any
 				var ret = mmethoddef.msignature.return_mtype
 				if ret != null and ret isa MClassType and ret.mclass.kind != abstract_kind and ret.mclass.kind != interface_kind then
@@ -351,11 +347,12 @@ class RapidTypeAnalysis
 			var nclassdef = self.modelbuilder.mclassdef2nclassdef[cd]
 			for npropdef in nclassdef.n_propdefs do
 				if not npropdef isa AAttrPropdef then continue
-				var nexpr = npropdef.n_expr
-				if nexpr == null then continue
+				if not npropdef.has_value then continue
+
 				var mpropdef = npropdef.mpropdef.as(not null)
 				var v = new RapidTypeVisitor(self, bound_mtype, mpropdef)
-				v.enter_visit(nexpr)
+				v.enter_visit(npropdef.n_expr)
+				v.enter_visit(npropdef.n_block)
 			end
 		end
 
@@ -442,11 +439,8 @@ class RapidTypeVisitor
 	var receiver: MClassType
 	var mpropdef: MPropDef
 
-	init(analysis: RapidTypeAnalysis, receiver: MClassType, mpropdef: MPropDef)
+	init
 	do
-		self.analysis = analysis
-		self.receiver = receiver
-		self.mpropdef = mpropdef
 		assert not receiver.need_anchor
 	end
 
@@ -686,7 +680,7 @@ end
 redef class ANewExpr
 	redef fun accept_rapid_type_visitor(v)
 	do
-		var mtype = self.mtype.as(MClassType)
+		var mtype = self.recvtype.as(not null)
 		v.add_type(mtype)
 		v.add_callsite(callsite)
 	end