1 # This file is part of NIT ( http://www.nitlanguage.org ).
3 # Copyright 2015 Romain Chanoir <romain.chanoir@viacesi.fr>
5 # Licensed under the Apache License, Version 2.0 (the "License");
6 # you may not use this file except in compliance with the License.
7 # You may obtain a copy of the License at
9 # http://www.apache.org/licenses/LICENSE-2.0
11 # Unless required by applicable law or agreed to in writing, software
12 # distributed under the License is distributed on an "AS IS" BASIS,
13 # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14 # See the License for the specific language governing permissions and
15 # limitations under the License.
18 # Phase generating threads for functions annotated with `threaded` annotation
19 module parallelization_phase
21 private import parser_util
24 private import annotation
25 private import astvalidation
27 redef class ToolContext
28 # Transforms a function annotated with "threaded"
29 var parallelization_phase
: Phase = new ParallelizationPhase(self, null)
32 private class ParallelizationPhase
35 redef fun process_annotated_node
(nmethdef
, nat
)
37 if nat
.n_atid
.n_id
.text
!= "threaded" then return
39 if not nmethdef
isa AMethPropdef then
40 toolcontext
.error
(nat
.location
, "Syntax Error: only a method can be threaded.")
44 #TODO: check for self calls
46 # Get the module associated with this method
47 var amod
= nmethdef
.parent
.parent
48 assert amod
isa AModule
50 # Construct the name of the generated class
51 var classname
= "Threaded"
53 # Try to get the name of the class
54 if nmethdef
.parent
isa AStdClassdef then
55 classname
+= nmethdef
.parent
.as(AStdClassdef).n_id
.text
58 # Try to get the name of the method
59 if nmethdef
.n_methid
isa AIdMethid then
60 classname
+= nmethdef
.n_methid
.as(AIdMethid).n_id
.text
63 # Handle methods with a return value
64 var has_rvalue
= nmethdef
.n_signature
.n_type
!= null
67 vtype
= "redef type E: " + nmethdef
.n_signature
.n_type
.n_id
.text
70 # create a return type
71 var n_id
= new TClassid
73 var n_type
= new AType
75 nmethdef
.n_signature
.n_type
= n_type
77 var params
= new Array[String]
78 for param
in nmethdef
.n_signature
.n_params
do
79 var typ
= param
.n_type
.n_id
.text
80 if param
.n_type
.n_kwnullable
!= null then typ
= "nullable {typ}"
82 var {{{param.n_id.text}}}: {{{typ}}}
86 # String corresponding to the generated class
87 var classdef_source
= """
93 {{{params.join("\n")}}}
99 # Parse newly obtained classdef
100 var classdef
= toolcontext
.parse_classdef
(classdef_source
)
101 assert classdef
isa AStdClassdef
103 # Get the `main` fun of the class
104 var mainfun
: nullable AMethPropdef = null
105 for prop
in classdef
.n_propdefs
do
106 if prop
isa AMethPropdef then mainfun
= prop
108 assert mainfun
!= null
110 # Make the statements from `main` fun be the statements from the "threaded" fun
111 mainfun
.n_block
= nmethdef
.n_block
113 # Add "return null" to the end of the `main` function
114 if not has_rvalue
then
115 var s_nullreturn
= "return null"
116 var nullreturn
= toolcontext
.parse_something
(s_nullreturn
)
117 assert nullreturn
isa AExpr
118 mainfun
.n_block
.as(ABlockExpr).n_expr
.add
(nullreturn
)
121 # Create new body for the annotated fun
122 var s_newbody
: String
123 if nmethdef
.n_signature
.n_params
.not_empty
then
124 var init_params
= new Array[String]
125 for param
in nmethdef
.n_signature
.n_params
do
126 init_params
.add
(param
.n_id
.text
)
129 var thread = new {{{classname}}}({{{init_params.join(",")}}})
135 var thread = new {{{classname}}}
141 var newbody
= toolcontext
.parse_something
(s_newbody
)
142 nmethdef
.n_block
= newbody
.as(ABlockExpr)
146 # Add the new class to the module
147 amod
.n_classdefs
.add
(classdef
)