1 | // |
---|
2 | // Cforall Version 1.0.0 Copyright (C) 2015 University of Waterloo |
---|
3 | // |
---|
4 | // The contents of this file are covered under the licence agreement in the |
---|
5 | // file "LICENCE" distributed with Cforall. |
---|
6 | // |
---|
7 | // GenType.cpp -- |
---|
8 | // |
---|
9 | // Author : Richard C. Bilson |
---|
10 | // Created On : Mon May 18 07:44:20 2015 |
---|
11 | // Last Modified By : Andrew Beach |
---|
12 | // Last Modified On : Fri May 20 11:18:00 2022 |
---|
13 | // Update Count : 24 |
---|
14 | // |
---|
15 | |
---|
16 | #include "GenType.hpp" |
---|
17 | |
---|
18 | #include <cassert> // for assert, assertf |
---|
19 | #include <list> // for _List_iterator, _List_const_iterator |
---|
20 | #include <sstream> // for operator<<, ostringstream, basic_os... |
---|
21 | |
---|
22 | #include "AST/Print.hpp" // for print |
---|
23 | #include "AST/Vector.hpp" // for vector |
---|
24 | #include "CodeGenerator.hpp" // for CodeGenerator |
---|
25 | #include "Common/UniqueName.hpp" // for UniqueName |
---|
26 | |
---|
27 | namespace CodeGen { |
---|
28 | |
---|
29 | namespace { |
---|
30 | |
---|
31 | struct GenType final : |
---|
32 | public ast::WithShortCircuiting, |
---|
33 | public ast::WithVisitorRef<GenType> { |
---|
34 | std::string result; |
---|
35 | GenType( const std::string &typeString, const Options &options ); |
---|
36 | |
---|
37 | void previsit( ast::Node const * ); |
---|
38 | void postvisit( ast::Node const * ); |
---|
39 | |
---|
40 | void postvisit( ast::FunctionType const * type ); |
---|
41 | void postvisit( ast::VoidType const * type ); |
---|
42 | void postvisit( ast::BasicType const * type ); |
---|
43 | void postvisit( ast::PointerType const * type ); |
---|
44 | void postvisit( ast::ArrayType const * type ); |
---|
45 | void postvisit( ast::ReferenceType const * type ); |
---|
46 | void postvisit( ast::StructInstType const * type ); |
---|
47 | void postvisit( ast::UnionInstType const * type ); |
---|
48 | void postvisit( ast::EnumInstType const * type ); |
---|
49 | void postvisit( ast::EnumAttrType const * type ); |
---|
50 | void postvisit( ast::TypeInstType const * type ); |
---|
51 | void postvisit( ast::TupleType const * type ); |
---|
52 | void postvisit( ast::VarArgsType const * type ); |
---|
53 | void postvisit( ast::ZeroType const * type ); |
---|
54 | void postvisit( ast::OneType const * type ); |
---|
55 | void postvisit( ast::GlobalScopeType const * type ); |
---|
56 | void postvisit( ast::TraitInstType const * type ); |
---|
57 | void postvisit( ast::TypeofType const * type ); |
---|
58 | void postvisit( ast::VTableType const * type ); |
---|
59 | void postvisit( ast::QualifiedType const * type ); |
---|
60 | |
---|
61 | private: |
---|
62 | void handleQualifiers( ast::Type const *type ); |
---|
63 | std::string handleGeneric( ast::BaseInstType const * type ); |
---|
64 | void genArray( const ast::CV::Qualifiers &qualifiers, ast::Type const *base, ast::Expr const *dimension, bool isVarLen, bool isStatic ); |
---|
65 | std::string genParamList( const ast::vector<ast::Type> & ); |
---|
66 | |
---|
67 | Options options; |
---|
68 | }; |
---|
69 | |
---|
70 | GenType::GenType( const std::string &typeString, const Options &options ) : result( typeString ), options( options ) {} |
---|
71 | |
---|
72 | void GenType::previsit( ast::Node const * ) { |
---|
73 | // Turn off automatic recursion for all nodes, to allow each visitor to |
---|
74 | // precisely control the order in which its children are visited. |
---|
75 | visit_children = false; |
---|
76 | } |
---|
77 | |
---|
78 | void GenType::postvisit( ast::Node const * node ) { |
---|
79 | std::stringstream ss; |
---|
80 | ast::print( ss, node ); |
---|
81 | assertf( false, "Unhandled node reached in GenType: %s", ss.str().c_str() ); |
---|
82 | } |
---|
83 | |
---|
84 | void GenType::postvisit( ast::VoidType const * type ) { |
---|
85 | result = "void " + result; |
---|
86 | handleQualifiers( type ); |
---|
87 | } |
---|
88 | |
---|
89 | void GenType::postvisit( ast::BasicType const * type ) { |
---|
90 | ast::BasicKind kind = type->kind; |
---|
91 | assert( 0 <= kind && kind < ast::BasicKind::NUMBER_OF_BASIC_TYPES ); |
---|
92 | result = std::string( ast::BasicType::typeNames[kind] ) + " " + result; |
---|
93 | handleQualifiers( type ); |
---|
94 | } |
---|
95 | |
---|
96 | void GenType::genArray( const ast::CV::Qualifiers & qualifiers, ast::Type const * base, ast::Expr const *dimension, bool isVarLen, bool isStatic ) { |
---|
97 | std::ostringstream os; |
---|
98 | if ( result != "" ) { |
---|
99 | if ( result[ 0 ] == '*' ) { |
---|
100 | os << "(" << result << ")"; |
---|
101 | } else { |
---|
102 | os << result; |
---|
103 | } |
---|
104 | } |
---|
105 | os << "["; |
---|
106 | if ( isStatic ) { |
---|
107 | os << "static "; |
---|
108 | } |
---|
109 | if ( qualifiers.is_const ) { |
---|
110 | os << "const "; |
---|
111 | } |
---|
112 | if ( qualifiers.is_volatile ) { |
---|
113 | os << "volatile "; |
---|
114 | } |
---|
115 | if ( qualifiers.is_restrict ) { |
---|
116 | os << "__restrict "; |
---|
117 | } |
---|
118 | if ( qualifiers.is_atomic ) { |
---|
119 | os << "_Atomic "; |
---|
120 | } |
---|
121 | if ( dimension != 0 ) { |
---|
122 | ast::Pass<CodeGenerator>::read( dimension, os, options ); |
---|
123 | } else if ( isVarLen ) { |
---|
124 | // no dimension expression on a VLA means it came in with the * token |
---|
125 | os << "*"; |
---|
126 | } |
---|
127 | os << "]"; |
---|
128 | |
---|
129 | result = os.str(); |
---|
130 | |
---|
131 | base->accept( *visitor ); |
---|
132 | } |
---|
133 | |
---|
134 | void GenType::postvisit( ast::PointerType const * type ) { |
---|
135 | if ( type->isStatic || type->isVarLen || type->dimension ) { |
---|
136 | genArray( type->qualifiers, type->base, type->dimension, type->isVarLen, type->isStatic ); |
---|
137 | } else { |
---|
138 | handleQualifiers( type ); |
---|
139 | if ( result[ 0 ] == '?' ) { |
---|
140 | result = "* " + result; |
---|
141 | } else { |
---|
142 | result = "*" + result; |
---|
143 | } |
---|
144 | type->base->accept( *visitor ); |
---|
145 | } |
---|
146 | } |
---|
147 | |
---|
148 | void GenType::postvisit( ast::ArrayType const * type ) { |
---|
149 | genArray( type->qualifiers, type->base, type->dimension, type->isVarLen, type->isStatic ); |
---|
150 | } |
---|
151 | |
---|
152 | void GenType::postvisit( ast::ReferenceType const * type ) { |
---|
153 | assertf( !options.genC, "Reference types should not reach code generation." ); |
---|
154 | handleQualifiers( type ); |
---|
155 | result = "&" + result; |
---|
156 | type->base->accept( *visitor ); |
---|
157 | } |
---|
158 | |
---|
159 | void GenType::postvisit( ast::FunctionType const * type ) { |
---|
160 | std::ostringstream os; |
---|
161 | |
---|
162 | if ( result != "" ) { |
---|
163 | if ( result[ 0 ] == '*' ) { |
---|
164 | os << "(" << result << ")"; |
---|
165 | } else { |
---|
166 | os << result; |
---|
167 | } |
---|
168 | } |
---|
169 | |
---|
170 | if ( type->params.empty() ) { |
---|
171 | if ( type->isVarArgs ) { |
---|
172 | os << "()"; |
---|
173 | } else { |
---|
174 | os << "(void)"; |
---|
175 | } |
---|
176 | } else { |
---|
177 | os << "(" ; |
---|
178 | |
---|
179 | os << genParamList( type->params ); |
---|
180 | |
---|
181 | if ( type->isVarArgs ) { |
---|
182 | os << ", ..."; |
---|
183 | } |
---|
184 | os << ")"; |
---|
185 | } |
---|
186 | |
---|
187 | result = os.str(); |
---|
188 | |
---|
189 | if ( type->returns.size() == 0 ) { |
---|
190 | result = "void " + result; |
---|
191 | } else { |
---|
192 | type->returns.front()->accept( *visitor ); |
---|
193 | } |
---|
194 | |
---|
195 | // Add forall clause. |
---|
196 | if( !type->forall.empty() && !options.genC ) { |
---|
197 | //assertf( !options.genC, "FunctionDecl type parameters should not reach code generation." ); |
---|
198 | std::ostringstream os; |
---|
199 | ast::Pass<CodeGenerator> cg( os, options ); |
---|
200 | os << "forall("; |
---|
201 | cg.core.genCommaList( type->forall ); |
---|
202 | os << ")" << std::endl; |
---|
203 | result = os.str() + result; |
---|
204 | } |
---|
205 | } |
---|
206 | |
---|
207 | std::string GenType::handleGeneric( ast::BaseInstType const * type ) { |
---|
208 | if ( !type->params.empty() ) { |
---|
209 | std::ostringstream os; |
---|
210 | ast::Pass<CodeGenerator> cg( os, options ); |
---|
211 | os << "("; |
---|
212 | cg.core.genCommaList( type->params ); |
---|
213 | os << ") "; |
---|
214 | return os.str(); |
---|
215 | } |
---|
216 | return ""; |
---|
217 | } |
---|
218 | |
---|
219 | void GenType::postvisit( ast::StructInstType const * type ) { |
---|
220 | result = type->name + handleGeneric( type ) + " " + result; |
---|
221 | if ( options.genC ) result = "struct " + result; |
---|
222 | handleQualifiers( type ); |
---|
223 | } |
---|
224 | |
---|
225 | void GenType::postvisit( ast::UnionInstType const * type ) { |
---|
226 | result = type->name + handleGeneric( type ) + " " + result; |
---|
227 | if ( options.genC ) result = "union " + result; |
---|
228 | handleQualifiers( type ); |
---|
229 | } |
---|
230 | |
---|
231 | void GenType::postvisit( ast::EnumInstType const * type ) { |
---|
232 | // if ( type->base && type->base->base ) { |
---|
233 | // result = genType( type->base->base, result, options ); |
---|
234 | // } else { |
---|
235 | result = type->name + " " + result; |
---|
236 | if ( options.genC ) { |
---|
237 | result = "enum " + result; |
---|
238 | } |
---|
239 | // } |
---|
240 | handleQualifiers( type ); |
---|
241 | } |
---|
242 | |
---|
243 | void GenType::postvisit( ast::EnumAttrType const * type ) { |
---|
244 | postvisit( type->instance ); |
---|
245 | } |
---|
246 | |
---|
247 | void GenType::postvisit( ast::TypeInstType const * type ) { |
---|
248 | assertf( !options.genC, "TypeInstType should not reach code generation." ); |
---|
249 | result = type->name + " " + result; |
---|
250 | handleQualifiers( type ); |
---|
251 | } |
---|
252 | |
---|
253 | void GenType::postvisit( ast::TupleType const * type ) { |
---|
254 | assertf( !options.genC, "TupleType should not reach code generation." ); |
---|
255 | unsigned int i = 0; |
---|
256 | std::ostringstream os; |
---|
257 | os << "["; |
---|
258 | for ( ast::ptr<ast::Type> const & t : type->types ) { |
---|
259 | i++; |
---|
260 | os << genType( t, "", options ) << (i == type->size() ? "" : ", "); |
---|
261 | } |
---|
262 | os << "] "; |
---|
263 | result = os.str() + result; |
---|
264 | } |
---|
265 | |
---|
266 | void GenType::postvisit( ast::VarArgsType const * type ) { |
---|
267 | result = "__builtin_va_list " + result; |
---|
268 | handleQualifiers( type ); |
---|
269 | } |
---|
270 | |
---|
271 | void GenType::postvisit( ast::ZeroType const * type ) { |
---|
272 | // Ideally these wouldn't hit codegen at all, but should be safe to make them ints. |
---|
273 | result = (options.pretty ? "zero_t " : "long int ") + result; |
---|
274 | handleQualifiers( type ); |
---|
275 | } |
---|
276 | |
---|
277 | void GenType::postvisit( ast::OneType const * type ) { |
---|
278 | // Ideally these wouldn't hit codegen at all, but should be safe to make them ints. |
---|
279 | result = (options.pretty ? "one_t " : "long int ") + result; |
---|
280 | handleQualifiers( type ); |
---|
281 | } |
---|
282 | |
---|
283 | void GenType::postvisit( ast::GlobalScopeType const * type ) { |
---|
284 | assertf( !options.genC, "GlobalScopeType should not reach code generation." ); |
---|
285 | handleQualifiers( type ); |
---|
286 | } |
---|
287 | |
---|
288 | void GenType::postvisit( ast::TraitInstType const * type ) { |
---|
289 | assertf( !options.genC, "TraitInstType should not reach code generation." ); |
---|
290 | result = type->name + " " + result; |
---|
291 | handleQualifiers( type ); |
---|
292 | } |
---|
293 | |
---|
294 | void GenType::postvisit( ast::TypeofType const * type ) { |
---|
295 | std::ostringstream os; |
---|
296 | os << "typeof("; |
---|
297 | ast::Pass<CodeGenerator>::read( type->expr.get(), os, options ); |
---|
298 | os << ") " << result; |
---|
299 | result = os.str(); |
---|
300 | handleQualifiers( type ); |
---|
301 | } |
---|
302 | |
---|
303 | void GenType::postvisit( ast::VTableType const * type ) { |
---|
304 | assertf( !options.genC, "Virtual table types should not reach code generation." ); |
---|
305 | std::ostringstream os; |
---|
306 | os << "vtable(" << genType( type->base, "", options ) << ") " << result; |
---|
307 | result = os.str(); |
---|
308 | handleQualifiers( type ); |
---|
309 | } |
---|
310 | |
---|
311 | void GenType::postvisit( ast::QualifiedType const * type ) { |
---|
312 | assertf( !options.genC, "QualifiedType should not reach code generation." ); |
---|
313 | std::ostringstream os; |
---|
314 | os << genType( type->parent, "", options ) << "." << genType( type->child, "", options ) << result; |
---|
315 | result = os.str(); |
---|
316 | handleQualifiers( type ); |
---|
317 | } |
---|
318 | |
---|
319 | void GenType::handleQualifiers( ast::Type const * type ) { |
---|
320 | if ( type->is_const() ) { |
---|
321 | result = "const " + result; |
---|
322 | } |
---|
323 | if ( type->is_volatile() ) { |
---|
324 | result = "volatile " + result; |
---|
325 | } |
---|
326 | if ( type->is_restrict() ) { |
---|
327 | result = "__restrict " + result; |
---|
328 | } |
---|
329 | if ( type->is_atomic() ) { |
---|
330 | result = "_Atomic " + result; |
---|
331 | } |
---|
332 | } |
---|
333 | |
---|
334 | std::string GenType::genParamList( const ast::vector<ast::Type> & range ) { |
---|
335 | auto cur = range.begin(); |
---|
336 | auto end = range.end(); |
---|
337 | if ( cur == end ) return ""; |
---|
338 | std::ostringstream oss; |
---|
339 | UniqueName param( "__param_" ); |
---|
340 | while ( true ) { |
---|
341 | oss << genType( *cur++, options.genC ? param.newName() : "", options ); |
---|
342 | if ( cur == end ) break; |
---|
343 | oss << ", "; |
---|
344 | } |
---|
345 | return oss.str(); |
---|
346 | } |
---|
347 | |
---|
348 | } // namespace |
---|
349 | |
---|
350 | std::string genType( ast::Type const * type, const std::string & base, const Options & options ) { |
---|
351 | std::ostringstream os; |
---|
352 | if ( !type->attributes.empty() ) { |
---|
353 | ast::Pass<CodeGenerator> cg( os, options ); |
---|
354 | cg.core.genAttributes( type->attributes ); |
---|
355 | } |
---|
356 | |
---|
357 | return os.str() + ast::Pass<GenType>::read( type, base, options ); |
---|
358 | } |
---|
359 | |
---|
360 | std::string genTypeNoAttr( ast::Type const * type, const std::string & base, const Options & options ) { |
---|
361 | return ast::Pass<GenType>::read( type, base, options ); |
---|
362 | } |
---|
363 | |
---|
364 | } // namespace CodeGen |
---|
365 | |
---|
366 | // Local Variables: // |
---|
367 | // tab-width: 4 // |
---|
368 | // mode: c++ // |
---|
369 | // compile-command: "make install" // |
---|
370 | // End: // |
---|