From 704d8f070038c53dc2350c886a0e377bc6de7653 Mon Sep 17 00:00:00 2001
From: BlackMinou <romain.chanoir@viacesi.fr>
Date: Wed, 1 Apr 2015 08:57:27 +0200
Subject: [PATCH] Add the support of parametrized functions for the threaded
 annotation

Signed-off-by: BlackMinou <romain.chanoir@viacesi.fr>
---
 src/frontend/parallelization_phase.nit |   39 ++++++++++++++++++++++++++------
 1 file changed, 32 insertions(+), 7 deletions(-)

diff --git a/src/frontend/parallelization_phase.nit b/src/frontend/parallelization_phase.nit
index 7da80ff..8cd7965 100644
--- a/src/frontend/parallelization_phase.nit
+++ b/src/frontend/parallelization_phase.nit
@@ -39,10 +39,9 @@ private class ParallelizationPhase
 			toolcontext.error(nmethdef.location, "Syntax error: only a method can be threaded.")
 			return
 		end
-		if nmethdef.n_signature.n_params.length != 0 then
-			toolcontext.error(nmethdef.location, "Syntax error: parametrized method not supported yet.")
-			return
-		end
+
+		#TODO: check for self calls
+
 		if nmethdef.n_signature.n_type != null then
 			toolcontext.error(nmethdef.location, "Syntax error: method with a return value not supported yet.")
 			return
@@ -66,11 +65,24 @@ private class ParallelizationPhase
 			classname += nmethdef.n_methid.as(AIdMethid).n_id.text
 		end
 
-		# Create a string corresponding to the threaded class
-		var s ="""
+
+		var params = new Array[String]
+		# case if the method has parameters
+		if nmethdef.n_signature.n_params.not_empty then
+			for param in nmethdef.n_signature.n_params do
+				params.add("""
+var {{{param.n_id.text}}} : {{{param.n_type.n_id.text}}}
+
+""")
+			end
+		end
+
+		# String corresponding to the generated class
+		var s="""
 class {{{classname}}}
 	super Thread
 
+	{{{params.join("\n")}}}
 	redef fun main do
 	end
 end
@@ -95,11 +107,24 @@ end
 		assert nullreturn isa AExpr
 		mainfun.n_block.as(ABlockExpr).n_expr.add(nullreturn)
 
+
 		# Create new body for the annotated fun
-		var s_newbody ="""
+		var s_newbody : String
+		if nmethdef.n_signature.n_params.not_empty then
+			var init_params = new Array[String]
+			for param in nmethdef.n_signature.n_params do
+				init_params.add(param.n_id.text)
+			end
+			s_newbody ="""
+var thread = new {{{classname}}}({{{init_params.join(",")}}})
+thread.start
+"""
+		else
+			s_newbody = """
 var thread = new {{{classname}}}
 thread.start
 """
+		end
 
 		var newbody = toolcontext.parse_something(s_newbody)
 		nmethdef.n_block = newbody.as(ABlockExpr)
-- 
1.7.9.5