source: src/AST/Pass.impl.hpp @ 0dd9a5e

ADTarm-ehast-experimentalenumforall-pointer-decayjacob/cs343-translationnew-ast-unique-exprpthread-emulationqualifiedEnum
Last change on this file since 0dd9a5e was 0dd9a5e, checked in by Fangren Yu <f37yu@…>, 3 years ago

delay autogen resolve
  • Property mode set to 100644
File size: 58.5 KB
Line 
1//
2// Cforall Version 1.0.0 Copyright (C) 2019 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// ast::Pass.impl.hpp --
8//
9// Author           : Thierry Delisle
10// Created On       : Thu May 09 15::37::05 2019
11// Last Modified By :
12// Last Modified On :
13// Update Count     :
14//
15
16#pragma once
17// IWYU pragma: private, include "AST/Pass.hpp"
18
19#include <type_traits>
20#include <unordered_map>
21
22#include "AST/TranslationUnit.hpp"
23#include "AST/TypeSubstitution.hpp"
24
25#define VISIT_START( node ) \
26        using namespace ast; \
27        /* back-up the visit children */ \
28        __attribute__((unused)) ast::__pass::visit_children_guard guard1( ast::__pass::visit_children(core, 0) ); \
29        /* setup the scope for passes that want to run code at exit */ \
30        __attribute__((unused)) ast::__pass::guard_value          guard2( ast::__pass::at_cleanup    (core, 0) ); \
31        /* begin tracing memory allocation if requested by this pass */ \
32        __pass::beginTrace( core, 0 ); \
33        /* call the implementation of the previsit of this pass */ \
34        __pass::previsit( core, node, 0 );
35
36#define VISIT( code... ) \
37        /* if this node should visit its children */ \
38        if ( __visit_children() ) { \
39                /* visit the children */ \
40                code \
41        }
42
43#define VISIT_END( type, node ) \
44        /* call the implementation of the postvisit of this pass */ \
45        auto __return = __pass::postvisit( core, node, 0 ); \
46        assertf(__return, "post visit should never return null"); \
47        /* end tracing memory allocation if requested by this pass */ \
48        __pass::endTrace( core, 0 ); \
49        return __return;
50
51#ifdef PEDANTIC_PASS_ASSERT
52#define __pedantic_pass_assert(...) assert (__VA_ARGS__)
53#define __pedantic_pass_assertf(...) assertf(__VA_ARGS__)
54#else
55#define __pedantic_pass_assert(...)
56#define __pedantic_pass_assertf(...)
57#endif
58
59namespace ast {
60        template<typename node_t>
61        node_t * shallowCopy( const node_t * node );
62
63        namespace __pass {
64                // Check if this is either a null pointer or a pointer to an empty container
65                template<typename T>
66                static inline bool empty( T * ptr ) {
67                        return !ptr || ptr->empty();
68                }
69
70                template< typename core_t, typename node_t >
71                static inline node_t* mutate(const node_t *node) {
72                        return std::is_base_of<PureVisitor, core_t>::value ? ::ast::shallowCopy(node) : ::ast::mutate(node);
73                }
74
75                //------------------------------
76                template<typename it_t, template <class...> class container_t>
77                static inline void take_all( it_t it, container_t<ast::ptr<ast::Decl>> * decls, bool * mutated = nullptr ) {
78                        if(empty(decls)) return;
79
80                        std::transform(decls->begin(), decls->end(), it, [](const ast::Decl * decl) -> auto {
81                                        return new DeclStmt( decl->location, decl );
82                                });
83                        decls->clear();
84                        if(mutated) *mutated = true;
85                }
86
87                template<typename it_t, template <class...> class container_t>
88                static inline void take_all( it_t it, container_t<ast::ptr<ast::Stmt>> * decls, bool * mutated = nullptr ) {
89                        if(empty(decls)) return;
90
91                        std::move(decls->begin(), decls->end(), it);
92                        decls->clear();
93                        if(mutated) *mutated = true;
94                }
95
96                //------------------------------
97                /// Check if should be skipped, different for pointers and containers
98                template<typename node_t>
99                bool skip( const ast::ptr<node_t> & val) {
100                        return !val;
101                }
102
103                template< template <class...> class container_t, typename node_t >
104                bool skip( const container_t<ast::ptr< node_t >> & val ) {
105                        return val.empty();
106                }
107
108                //------------------------------
109                /// Get the value to visit, different for pointers and containers
110                template<typename node_t>
111                auto get( const ast::ptr<node_t> & val, int ) -> decltype(val.get()) {
112                        return val.get();
113                }
114
115                template<typename node_t>
116                const node_t & get( const node_t & val, long) {
117                        return val;
118                }
119
120
121                //------------------------------
122                /// Check if value was mutated, different for pointers and containers
123                template<typename lhs_t, typename rhs_t>
124                bool differs( const lhs_t * old_val, const rhs_t * new_val ) {
125                        return old_val != new_val;
126                }
127
128                template< template <class...> class container_t, typename node_t >
129                bool differs( const container_t<ast::ptr< node_t >> &, const container_t<ast::ptr< node_t >> & new_val ) {
130                        return !new_val.empty();
131                }
132        }
133
134        template< typename core_t >
135        template< typename node_t >
136        auto ast::Pass< core_t >::call_accept( const node_t * node )
137                -> typename std::enable_if<
138                                !std::is_base_of<ast::Expr, node_t>::value &&
139                                !std::is_base_of<ast::Stmt, node_t>::value
140                        , decltype( node->accept(*this) )
141                >::type
142        {
143                __pedantic_pass_assert( __visit_children() );
144                __pedantic_pass_assert( node );
145
146                static_assert( !std::is_base_of<ast::Expr, node_t>::value, "ERROR");
147                static_assert( !std::is_base_of<ast::Stmt, node_t>::value, "ERROR");
148
149                return node->accept( *this );
150        }
151
152        template< typename core_t >
153        const ast::Expr * ast::Pass< core_t >::call_accept( const ast::Expr * expr ) {
154                __pedantic_pass_assert( __visit_children() );
155                __pedantic_pass_assert( expr );
156
157                const ast::TypeSubstitution ** typeSubs_ptr = __pass::typeSubs( core, 0 );
158                if ( typeSubs_ptr && expr->env ) {
159                        *typeSubs_ptr = expr->env;
160                }
161
162                return expr->accept( *this );
163        }
164
165        template< typename core_t >
166        const ast::Stmt * ast::Pass< core_t >::call_accept( const ast::Stmt * stmt ) {
167                __pedantic_pass_assert( __visit_children() );
168                __pedantic_pass_assert( stmt );
169
170                return stmt->accept( *this );
171        }
172
173        template< typename core_t >
174        const ast::Stmt * ast::Pass< core_t >::call_accept_as_compound( const ast::Stmt * stmt ) {
175                __pedantic_pass_assert( __visit_children() );
176                __pedantic_pass_assert( stmt );
177
178                // add a few useful symbols to the scope
179                using __pass::empty;
180
181                // get the stmts/decls that will need to be spliced in
182                auto stmts_before = __pass::stmtsToAddBefore( core, 0);
183                auto stmts_after  = __pass::stmtsToAddAfter ( core, 0);
184                auto decls_before = __pass::declsToAddBefore( core, 0);
185                auto decls_after  = __pass::declsToAddAfter ( core, 0);
186
187                // These may be modified by subnode but most be restored once we exit this statemnet.
188                ValueGuardPtr< const ast::TypeSubstitution * > __old_env         ( __pass::typeSubs( core, 0 ) );
189                ValueGuardPtr< typename std::remove_pointer< decltype(stmts_before) >::type > __old_decls_before( stmts_before );
190                ValueGuardPtr< typename std::remove_pointer< decltype(stmts_after ) >::type > __old_decls_after ( stmts_after  );
191                ValueGuardPtr< typename std::remove_pointer< decltype(decls_before) >::type > __old_stmts_before( decls_before );
192                ValueGuardPtr< typename std::remove_pointer< decltype(decls_after ) >::type > __old_stmts_after ( decls_after  );
193
194                // Now is the time to actually visit the node
195                const ast::Stmt * nstmt = stmt->accept( *this );
196
197                // If the pass doesn't want to add anything then we are done
198                if( empty(stmts_before) && empty(stmts_after) && empty(decls_before) && empty(decls_after) ) {
199                        return nstmt;
200                }
201
202                // Make sure that it is either adding statements or declartions but not both
203                // this is because otherwise the order would be awkward to predict
204                assert(( empty( stmts_before ) && empty( stmts_after ))
205                    || ( empty( decls_before ) && empty( decls_after )) );
206
207                // Create a new Compound Statement to hold the new decls/stmts
208                ast::CompoundStmt * compound = new ast::CompoundStmt( stmt->location );
209
210                // Take all the declarations that go before
211                __pass::take_all( std::back_inserter( compound->kids ), decls_before );
212                __pass::take_all( std::back_inserter( compound->kids ), stmts_before );
213
214                // Insert the original declaration
215                compound->kids.emplace_back( nstmt );
216
217                // Insert all the declarations that go before
218                __pass::take_all( std::back_inserter( compound->kids ), decls_after );
219                __pass::take_all( std::back_inserter( compound->kids ), stmts_after );
220
221                return compound;
222        }
223
224        template< typename core_t >
225        template< template <class...> class container_t >
226        container_t< ptr<Stmt> > ast::Pass< core_t >::call_accept( const container_t< ptr<Stmt> > & statements ) {
227                __pedantic_pass_assert( __visit_children() );
228                if( statements.empty() ) return {};
229
230                // We are going to aggregate errors for all these statements
231                SemanticErrorException errors;
232
233                // add a few useful symbols to the scope
234                using __pass::empty;
235
236                // get the stmts/decls that will need to be spliced in
237                auto stmts_before = __pass::stmtsToAddBefore( core, 0);
238                auto stmts_after  = __pass::stmtsToAddAfter ( core, 0);
239                auto decls_before = __pass::declsToAddBefore( core, 0);
240                auto decls_after  = __pass::declsToAddAfter ( core, 0);
241
242                // These may be modified by subnode but most be restored once we exit this statemnet.
243                ValueGuardPtr< typename std::remove_pointer< decltype(stmts_before) >::type > __old_decls_before( stmts_before );
244                ValueGuardPtr< typename std::remove_pointer< decltype(stmts_after ) >::type > __old_decls_after ( stmts_after  );
245                ValueGuardPtr< typename std::remove_pointer< decltype(decls_before) >::type > __old_stmts_before( decls_before );
246                ValueGuardPtr< typename std::remove_pointer< decltype(decls_after ) >::type > __old_stmts_after ( decls_after  );
247
248                // update pass statitistics
249                pass_visitor_stats.depth++;
250                pass_visitor_stats.max->push(pass_visitor_stats.depth);
251                pass_visitor_stats.avg->push(pass_visitor_stats.depth);
252
253                bool mutated = false;
254                container_t< ptr<Stmt> > new_kids;
255                for( const Stmt * stmt : statements ) {
256                        try {
257                                __pedantic_pass_assert( stmt );
258                                const ast::Stmt * new_stmt = stmt->accept( *this );
259                                assert( new_stmt );
260                                if(new_stmt != stmt ) mutated = true;
261
262                                // Make sure that it is either adding statements or declartions but not both
263                                // this is because otherwise the order would be awkward to predict
264                                assert(( empty( stmts_before ) && empty( stmts_after ))
265                                    || ( empty( decls_before ) && empty( decls_after )) );
266
267
268
269                                // Take all the statements which should have gone after, N/A for first iteration
270                                __pass::take_all( std::back_inserter( new_kids ), decls_before, &mutated );
271                                __pass::take_all( std::back_inserter( new_kids ), stmts_before, &mutated );
272
273                                // Now add the statement if there is one
274                                new_kids.emplace_back( new_stmt );
275
276                                // Take all the declarations that go before
277                                __pass::take_all( std::back_inserter( new_kids ), decls_after, &mutated );
278                                __pass::take_all( std::back_inserter( new_kids ), stmts_after, &mutated );
279                        }
280                        catch ( SemanticErrorException &e ) {
281                                errors.append( e );
282                        }
283                }
284                pass_visitor_stats.depth--;
285                if ( !errors.isEmpty() ) { throw errors; }
286
287                return mutated ? new_kids : container_t< ptr<Stmt> >();
288        }
289
290        template< typename core_t >
291        template< template <class...> class container_t, typename node_t >
292        container_t< ast::ptr<node_t> > ast::Pass< core_t >::call_accept( const container_t< ast::ptr<node_t> > & container ) {
293                __pedantic_pass_assert( __visit_children() );
294                if( container.empty() ) return {};
295                SemanticErrorException errors;
296
297                pass_visitor_stats.depth++;
298                pass_visitor_stats.max->push(pass_visitor_stats.depth);
299                pass_visitor_stats.avg->push(pass_visitor_stats.depth);
300
301                bool mutated = false;
302                container_t< ast::ptr<node_t> > new_kids;
303                for ( const node_t * node : container ) {
304                        try {
305                                __pedantic_pass_assert( node );
306                                const node_t * new_stmt = strict_dynamic_cast< const node_t * >( node->accept( *this ) );
307                                if(new_stmt != node ) mutated = true;
308
309                                new_kids.emplace_back( new_stmt );
310                        }
311                        catch( SemanticErrorException &e ) {
312                                errors.append( e );
313                        }
314                }
315                pass_visitor_stats.depth--;
316                if ( ! errors.isEmpty() ) { throw errors; }
317
318                return mutated ? new_kids : container_t< ast::ptr<node_t> >();
319        }
320
321        template< typename core_t >
322        template<typename node_t, typename parent_t, typename child_t>
323        void ast::Pass< core_t >::maybe_accept(
324                const node_t * & parent,
325                child_t parent_t::*child
326        ) {
327                static_assert( std::is_base_of<parent_t, node_t>::value, "Error deducing member object" );
328
329                if(__pass::skip(parent->*child)) return;
330                const auto & old_val = __pass::get(parent->*child, 0);
331
332                static_assert( !std::is_same<const ast::Node * &, decltype(old_val)>::value, "ERROR");
333
334                auto new_val = call_accept( old_val );
335
336                static_assert( !std::is_same<const ast::Node *, decltype(new_val)>::value || std::is_same<int, decltype(old_val)>::value, "ERROR");
337
338                if( __pass::differs(old_val, new_val) ) {
339                        auto new_parent = __pass::mutate<core_t>(parent);
340                        new_parent->*child = new_val;
341                        parent = new_parent;
342                }
343        }
344
345        template< typename core_t >
346        template<typename node_t, typename parent_t, typename child_t>
347        void ast::Pass< core_t >::maybe_accept_as_compound(
348                const node_t * & parent,
349                child_t parent_t::*child
350        ) {
351                static_assert( std::is_base_of<parent_t, node_t>::value, "Error deducing member object" );
352
353                if(__pass::skip(parent->*child)) return;
354                const auto & old_val = __pass::get(parent->*child, 0);
355
356                static_assert( !std::is_same<const ast::Node * &, decltype(old_val)>::value, "ERROR");
357
358                auto new_val = call_accept_as_compound( old_val );
359
360                static_assert( !std::is_same<const ast::Node *, decltype(new_val)>::value || std::is_same<int, decltype(old_val)>::value, "ERROR");
361
362                if( __pass::differs(old_val, new_val) ) {
363                        auto new_parent = __pass::mutate<core_t>(parent);
364                        new_parent->*child = new_val;
365                        parent = new_parent;
366                }
367        }
368
369
370        template< typename core_t >
371        template< typename node_t >
372        void ast::Pass< core_t >::mutate_forall( const node_t *& node ) {
373                if ( auto subs = __pass::forall::subs( core, 0 ) ) {
374                        // tracking TypeDecl substitution, full clone
375                        if ( node->forall.empty() ) return;
376
377                        node_t * mut = __pass::mutate<core_t>( node );
378                        mut->forall = subs->clone( node->forall, *this );
379                        node = mut;
380                } else {
381                        // not tracking TypeDecl substitution, just mutate
382                        maybe_accept( node, &node_t::forall );
383                }
384        }
385}
386
387//------------------------------------------------------------------------------------------------------------------------------------------------------------------------
388//========================================================================================================================================================================
389//========================================================================================================================================================================
390//========================================================================================================================================================================
391//========================================================================================================================================================================
392//========================================================================================================================================================================
393//------------------------------------------------------------------------------------------------------------------------------------------------------------------------
394
395template< typename core_t >
396inline void ast::accept_all( std::list< ast::ptr<ast::Decl> > & decls, ast::Pass< core_t > & visitor ) {
397        // We are going to aggregate errors for all these statements
398        SemanticErrorException errors;
399
400        // add a few useful symbols to the scope
401        using __pass::empty;
402
403        // get the stmts/decls that will need to be spliced in
404        auto decls_before = __pass::declsToAddBefore( visitor.core, 0);
405        auto decls_after  = __pass::declsToAddAfter ( visitor.core, 0);
406
407        // update pass statitistics
408        pass_visitor_stats.depth++;
409        pass_visitor_stats.max->push(pass_visitor_stats.depth);
410        pass_visitor_stats.avg->push(pass_visitor_stats.depth);
411
412        for ( std::list< ast::ptr<ast::Decl> >::iterator i = decls.begin(); ; ++i ) {
413                // splice in new declarations after previous decl
414                if ( !empty( decls_after ) ) { decls.splice( i, *decls_after ); }
415
416                if ( i == decls.end() ) break;
417
418                try {
419                        // run visitor on declaration
420                        ast::ptr<ast::Decl> & node = *i;
421                        assert( node );
422                        node = node->accept( visitor );
423                }
424                catch( SemanticErrorException &e ) {
425                        if (__pass::onError (visitor.core, *i, 0))
426                                errors.append( e );
427                }
428
429                // splice in new declarations before current decl
430                if ( !empty( decls_before ) ) { decls.splice( i, *decls_before ); }
431        }
432        pass_visitor_stats.depth--;
433        if ( !errors.isEmpty() ) { throw errors; }
434}
435
436template< typename core_t >
437inline void ast::accept_all( ast::TranslationUnit & unit, ast::Pass< core_t > & visitor ) {
438        return ast::accept_all( unit.decls, visitor );
439}
440
441// A NOTE ON THE ORDER OF TRAVERSAL
442//
443// Types and typedefs have their base types visited before they are added to the type table.  This is ok, since there is
444// no such thing as a recursive type or typedef.
445//
446//             typedef struct { T *x; } T; // never allowed
447//
448// for structs/unions, it is possible to have recursion, so the decl should be added as if it's incomplete to begin, the
449// members are traversed, and then the complete type should be added (assuming the type is completed by this particular
450// declaration).
451//
452//             struct T { struct T *x; }; // allowed
453//
454// It is important to add the complete type to the symbol table *after* the members/base has been traversed, since that
455// traversal may modify the definition of the type and these modifications should be visible when the symbol table is
456// queried later in this pass.
457
458//--------------------------------------------------------------------------
459// ObjectDecl
460template< typename core_t >
461const ast::DeclWithType * ast::Pass< core_t >::visit( const ast::ObjectDecl * node ) {
462        VISIT_START( node );
463
464        VISIT(
465                {
466                        guard_symtab guard { *this };
467                        maybe_accept( node, &ObjectDecl::type );
468                }
469                maybe_accept( node, &ObjectDecl::init          );
470                maybe_accept( node, &ObjectDecl::bitfieldWidth );
471                maybe_accept( node, &ObjectDecl::attributes    );
472        )
473
474        __pass::symtab::addId( core, 0, node );
475
476        VISIT_END( DeclWithType, node );
477}
478
479//--------------------------------------------------------------------------
480// FunctionDecl
481template< typename core_t >
482const ast::DeclWithType * ast::Pass< core_t >::visit( const ast::FunctionDecl * node ) {
483        VISIT_START( node );
484
485        __pass::symtab::addId( core, 0, node );
486
487        VISIT(maybe_accept( node, &FunctionDecl::withExprs );)
488        {
489                // with clause introduces a level of scope (for the with expression members).
490                // with clause exprs are added to the symbol table before parameters so that parameters
491                // shadow with exprs and not the other way around.
492                guard_symtab guard { *this };
493                __pass::symtab::addWith( core, 0, node->withExprs, node );
494                {
495                        guard_symtab guard { *this };
496                        // implicit add __func__ identifier as specified in the C manual 6.4.2.2
497                        static ast::ptr< ast::ObjectDecl > func{ new ast::ObjectDecl{ 
498                                CodeLocation{}, "__func__",
499                                new ast::ArrayType{
500                                        new ast::BasicType{ ast::BasicType::Char, ast::CV::Const },
501                                        nullptr, VariableLen, DynamicDim
502                                }
503                        } };
504                        __pass::symtab::addId( core, 0, func );
505                        VISIT(
506                                // parameter declarations are now directly here
507                                maybe_accept( node, &FunctionDecl::params );
508                                maybe_accept( node, &FunctionDecl::returns );
509                                // foralls are still in function type
510                                maybe_accept( node, &FunctionDecl::type );
511                                // First remember that we are now within a function.
512                                ValueGuard< bool > oldInFunction( inFunction );
513                                inFunction = true;
514                                // The function body needs to have the same scope as parameters.
515                                // A CompoundStmt will not enter a new scope if atFunctionTop is true.
516                                ValueGuard< bool > oldAtFunctionTop( atFunctionTop );
517                                atFunctionTop = true;
518                                maybe_accept( node, &FunctionDecl::stmts );
519                                maybe_accept( node, &FunctionDecl::attributes );
520                        )
521                }
522        }
523
524        VISIT_END( DeclWithType, node );
525}
526
527//--------------------------------------------------------------------------
528// StructDecl
529template< typename core_t >
530const ast::Decl * ast::Pass< core_t >::visit( const ast::StructDecl * node ) {
531        VISIT_START( node );
532
533        // make up a forward declaration and add it before processing the members
534        // needs to be on the heap because addStruct saves the pointer
535        __pass::symtab::addStructFwd( core, 0, node );
536
537        VISIT({
538                guard_symtab guard { * this };
539                maybe_accept( node, &StructDecl::params  );
540                maybe_accept( node, &StructDecl::members );
541        })
542
543        // this addition replaces the forward declaration
544        __pass::symtab::addStruct( core, 0, node );
545
546        VISIT_END( Decl, node );
547}
548
549//--------------------------------------------------------------------------
550// UnionDecl
551template< typename core_t >
552const ast::Decl * ast::Pass< core_t >::visit( const ast::UnionDecl * node ) {
553        VISIT_START( node );
554
555        // make up a forward declaration and add it before processing the members
556        __pass::symtab::addUnionFwd( core, 0, node );
557
558        VISIT({
559                guard_symtab guard { * this };
560                maybe_accept( node, &UnionDecl::params  );
561                maybe_accept( node, &UnionDecl::members );
562        })
563
564        __pass::symtab::addUnion( core, 0, node );
565
566        VISIT_END( Decl, node );
567}
568
569//--------------------------------------------------------------------------
570// EnumDecl
571template< typename core_t >
572const ast::Decl * ast::Pass< core_t >::visit( const ast::EnumDecl * node ) {
573        VISIT_START( node );
574
575        __pass::symtab::addEnum( core, 0, node );
576
577        VISIT(
578                // unlike structs, traits, and unions, enums inject their members into the global scope
579                maybe_accept( node, &EnumDecl::params  );
580                maybe_accept( node, &EnumDecl::members );
581        )
582
583        VISIT_END( Decl, node );
584}
585
586//--------------------------------------------------------------------------
587// TraitDecl
588template< typename core_t >
589const ast::Decl * ast::Pass< core_t >::visit( const ast::TraitDecl * node ) {
590        VISIT_START( node );
591
592        VISIT({
593                guard_symtab guard { *this };
594                maybe_accept( node, &TraitDecl::params  );
595                maybe_accept( node, &TraitDecl::members );
596        })
597
598        __pass::symtab::addTrait( core, 0, node );
599
600        VISIT_END( Decl, node );
601}
602
603//--------------------------------------------------------------------------
604// TypeDecl
605template< typename core_t >
606const ast::Decl * ast::Pass< core_t >::visit( const ast::TypeDecl * node ) {
607        VISIT_START( node );
608
609        VISIT({
610                guard_symtab guard { *this };
611                maybe_accept( node, &TypeDecl::params );
612                maybe_accept( node, &TypeDecl::base   );
613        })
614
615        // see A NOTE ON THE ORDER OF TRAVERSAL, above
616        // note that assertions come after the type is added to the symtab, since they are not part of the type proper
617        // and may depend on the type itself
618        __pass::symtab::addType( core, 0, node );
619
620        VISIT(
621                maybe_accept( node, &TypeDecl::assertions );
622
623                {
624                        guard_symtab guard { *this };
625                        maybe_accept( node, &TypeDecl::init );
626                }
627        )
628
629        VISIT_END( Decl, node );
630}
631
632//--------------------------------------------------------------------------
633// TypedefDecl
634template< typename core_t >
635const ast::Decl * ast::Pass< core_t >::visit( const ast::TypedefDecl * node ) {
636        VISIT_START( node );
637
638        VISIT({
639                guard_symtab guard { *this };
640                maybe_accept( node, &TypedefDecl::params );
641                maybe_accept( node, &TypedefDecl::base   );
642        })
643
644        __pass::symtab::addType( core, 0, node );
645
646        VISIT( maybe_accept( node, &TypedefDecl::assertions ); )
647
648        VISIT_END( Decl, node );
649}
650
651//--------------------------------------------------------------------------
652// AsmDecl
653template< typename core_t >
654const ast::AsmDecl * ast::Pass< core_t >::visit( const ast::AsmDecl * node ) {
655        VISIT_START( node );
656
657        VISIT(
658                maybe_accept( node, &AsmDecl::stmt );
659        )
660
661        VISIT_END( AsmDecl, node );
662}
663
664//--------------------------------------------------------------------------
665// StaticAssertDecl
666template< typename core_t >
667const ast::StaticAssertDecl * ast::Pass< core_t >::visit( const ast::StaticAssertDecl * node ) {
668        VISIT_START( node );
669
670        VISIT(
671                maybe_accept( node, &StaticAssertDecl::cond );
672                maybe_accept( node, &StaticAssertDecl::msg  );
673        )
674
675        VISIT_END( StaticAssertDecl, node );
676}
677
678//--------------------------------------------------------------------------
679// CompoundStmt
680template< typename core_t >
681const ast::CompoundStmt * ast::Pass< core_t >::visit( const ast::CompoundStmt * node ) {
682        VISIT_START( node );
683        VISIT(
684                // Do not enter (or leave) a new scope if atFunctionTop. Remember to save the result.
685                auto guard1 = makeFuncGuard( [this, enterScope = !this->atFunctionTop]() {
686                        if ( enterScope ) {
687                                __pass::symtab::enter(core, 0);
688                                __pass::scope::enter(core, 0);
689                        }
690                }, [this, leaveScope = !this->atFunctionTop]() {
691                        if ( leaveScope ) {
692                                __pass::symtab::leave(core, 0);
693                                __pass::scope::leave(core, 0);
694                        }
695                });
696                ValueGuard< bool > guard2( atFunctionTop );
697                atFunctionTop = false;
698                guard_scope guard3 { *this };
699                maybe_accept( node, &CompoundStmt::kids );
700        )
701        VISIT_END( CompoundStmt, node );
702}
703
704//--------------------------------------------------------------------------
705// ExprStmt
706template< typename core_t >
707const ast::Stmt * ast::Pass< core_t >::visit( const ast::ExprStmt * node ) {
708        VISIT_START( node );
709
710        VISIT(
711                maybe_accept( node, &ExprStmt::expr );
712        )
713
714        VISIT_END( Stmt, node );
715}
716
717//--------------------------------------------------------------------------
718// AsmStmt
719template< typename core_t >
720const ast::Stmt * ast::Pass< core_t >::visit( const ast::AsmStmt * node ) {
721        VISIT_START( node )
722
723        VISIT(
724                maybe_accept( node, &AsmStmt::instruction );
725                maybe_accept( node, &AsmStmt::output      );
726                maybe_accept( node, &AsmStmt::input       );
727                maybe_accept( node, &AsmStmt::clobber     );
728        )
729
730        VISIT_END( Stmt, node );
731}
732
733//--------------------------------------------------------------------------
734// DirectiveStmt
735template< typename core_t >
736const ast::Stmt * ast::Pass< core_t >::visit( const ast::DirectiveStmt * node ) {
737        VISIT_START( node )
738
739        VISIT_END( Stmt, node );
740}
741
742//--------------------------------------------------------------------------
743// IfStmt
744template< typename core_t >
745const ast::Stmt * ast::Pass< core_t >::visit( const ast::IfStmt * node ) {
746        VISIT_START( node );
747
748        VISIT({
749                // if statements introduce a level of scope (for the initialization)
750                guard_symtab guard { *this };
751                maybe_accept( node, &IfStmt::inits    );
752                maybe_accept( node, &IfStmt::cond     );
753                maybe_accept_as_compound( node, &IfStmt::thenPart );
754                maybe_accept_as_compound( node, &IfStmt::elsePart );
755        })
756
757        VISIT_END( Stmt, node );
758}
759
760//--------------------------------------------------------------------------
761// WhileStmt
762template< typename core_t >
763const ast::Stmt * ast::Pass< core_t >::visit( const ast::WhileStmt * node ) {
764        VISIT_START( node );
765
766        VISIT({
767                // while statements introduce a level of scope (for the initialization)
768                guard_symtab guard { *this };
769                maybe_accept( node, &WhileStmt::inits );
770                maybe_accept( node, &WhileStmt::cond  );
771                maybe_accept_as_compound( node, &WhileStmt::body  );
772        })
773
774        VISIT_END( Stmt, node );
775}
776
777//--------------------------------------------------------------------------
778// ForStmt
779template< typename core_t >
780const ast::Stmt * ast::Pass< core_t >::visit( const ast::ForStmt * node ) {
781        VISIT_START( node );
782
783        VISIT({
784                // for statements introduce a level of scope (for the initialization)
785                guard_symtab guard { *this };
786                // xxx - old ast does not create WithStmtsToAdd scope for loop inits. should revisit this later.
787                maybe_accept( node, &ForStmt::inits );
788                maybe_accept( node, &ForStmt::cond  );
789                maybe_accept( node, &ForStmt::inc   );
790                maybe_accept_as_compound( node, &ForStmt::body  );
791        })
792
793        VISIT_END( Stmt, node );
794}
795
796//--------------------------------------------------------------------------
797// SwitchStmt
798template< typename core_t >
799const ast::Stmt * ast::Pass< core_t >::visit( const ast::SwitchStmt * node ) {
800        VISIT_START( node );
801
802        VISIT(
803                maybe_accept( node, &SwitchStmt::cond  );
804                maybe_accept( node, &SwitchStmt::stmts );
805        )
806
807        VISIT_END( Stmt, node );
808}
809
810//--------------------------------------------------------------------------
811// CaseStmt
812template< typename core_t >
813const ast::Stmt * ast::Pass< core_t >::visit( const ast::CaseStmt * node ) {
814        VISIT_START( node );
815
816        VISIT(
817                maybe_accept( node, &CaseStmt::cond  );
818                maybe_accept( node, &CaseStmt::stmts );
819        )
820
821        VISIT_END( Stmt, node );
822}
823
824//--------------------------------------------------------------------------
825// BranchStmt
826template< typename core_t >
827const ast::Stmt * ast::Pass< core_t >::visit( const ast::BranchStmt * node ) {
828        VISIT_START( node );
829        VISIT_END( Stmt, node );
830}
831
832//--------------------------------------------------------------------------
833// ReturnStmt
834template< typename core_t >
835const ast::Stmt * ast::Pass< core_t >::visit( const ast::ReturnStmt * node ) {
836        VISIT_START( node );
837
838        VISIT(
839                maybe_accept( node, &ReturnStmt::expr );
840        )
841
842        VISIT_END( Stmt, node );
843}
844
845//--------------------------------------------------------------------------
846// ThrowStmt
847template< typename core_t >
848const ast::Stmt * ast::Pass< core_t >::visit( const ast::ThrowStmt * node ) {
849        VISIT_START( node );
850
851        VISIT(
852                maybe_accept( node, &ThrowStmt::expr   );
853                maybe_accept( node, &ThrowStmt::target );
854        )
855
856        VISIT_END( Stmt, node );
857}
858
859//--------------------------------------------------------------------------
860// TryStmt
861template< typename core_t >
862const ast::Stmt * ast::Pass< core_t >::visit( const ast::TryStmt * node ) {
863        VISIT_START( node );
864
865        VISIT(
866                maybe_accept( node, &TryStmt::body     );
867                maybe_accept( node, &TryStmt::handlers );
868                maybe_accept( node, &TryStmt::finally  );
869        )
870
871        VISIT_END( Stmt, node );
872}
873
874//--------------------------------------------------------------------------
875// CatchStmt
876template< typename core_t >
877const ast::Stmt * ast::Pass< core_t >::visit( const ast::CatchStmt * node ) {
878        VISIT_START( node );
879
880        VISIT({
881                // catch statements introduce a level of scope (for the caught exception)
882                guard_symtab guard { *this };
883                maybe_accept( node, &CatchStmt::decl );
884                maybe_accept( node, &CatchStmt::cond );
885                maybe_accept_as_compound( node, &CatchStmt::body );
886        })
887
888        VISIT_END( Stmt, node );
889}
890
891//--------------------------------------------------------------------------
892// FinallyStmt
893template< typename core_t >
894const ast::Stmt * ast::Pass< core_t >::visit( const ast::FinallyStmt * node ) {
895        VISIT_START( node );
896
897        VISIT(
898                maybe_accept( node, &FinallyStmt::body );
899        )
900
901        VISIT_END( Stmt, node );
902}
903
904//--------------------------------------------------------------------------
905// FinallyStmt
906template< typename core_t >
907const ast::Stmt * ast::Pass< core_t >::visit( const ast::SuspendStmt * node ) {
908        VISIT_START( node );
909
910        VISIT(
911                maybe_accept( node, &SuspendStmt::then   );
912        )
913
914        VISIT_END( Stmt, node );
915}
916
917//--------------------------------------------------------------------------
918// WaitForStmt
919template< typename core_t >
920const ast::Stmt * ast::Pass< core_t >::visit( const ast::WaitForStmt * node ) {
921        VISIT_START( node );
922                // for( auto & clause : node->clauses ) {
923                //      maybeAccept_impl( clause.target.function, *this );
924                //      maybeAccept_impl( clause.target.arguments, *this );
925
926                //      maybeAccept_impl( clause.statement, *this );
927                //      maybeAccept_impl( clause.condition, *this );
928                // }
929
930        VISIT({
931                std::vector<WaitForStmt::Clause> new_clauses;
932                new_clauses.reserve( node->clauses.size() );
933                bool mutated = false;
934                for( const auto & clause : node->clauses ) {
935
936                        const Expr * func = clause.target.func ? clause.target.func->accept(*this) : nullptr;
937                        if(func != clause.target.func) mutated = true;
938
939                        std::vector<ptr<Expr>> new_args;
940                        new_args.reserve(clause.target.args.size());
941                        for( const auto & arg : clause.target.args ) {
942                                auto a = arg->accept(*this);
943                                new_args.push_back( a );
944                                if( a != arg ) mutated = true;
945                        }
946
947                        const Stmt * stmt = clause.stmt ? clause.stmt->accept(*this) : nullptr;
948                        if(stmt != clause.stmt) mutated = true;
949
950                        const Expr * cond = clause.cond ? clause.cond->accept(*this) : nullptr;
951                        if(cond != clause.cond) mutated = true;
952
953                        new_clauses.push_back( WaitForStmt::Clause{ {func, std::move(new_args) }, stmt, cond } );
954                }
955
956                if(mutated) {
957                        auto n = __pass::mutate<core_t>(node);
958                        n->clauses = std::move( new_clauses );
959                        node = n;
960                }
961        })
962
963        #define maybe_accept(field) \
964                if(node->field) { \
965                        auto nval = call_accept( node->field ); \
966                        if(nval != node->field ) { \
967                                auto nparent = __pass::mutate<core_t>(node); \
968                                nparent->field = nval; \
969                                node = nparent; \
970                        } \
971                }
972
973        VISIT(
974                maybe_accept( timeout.time );
975                maybe_accept( timeout.stmt );
976                maybe_accept( timeout.cond );
977                maybe_accept( orElse.stmt  );
978                maybe_accept( orElse.cond  );
979        )
980
981        #undef maybe_accept
982
983        VISIT_END( Stmt, node );
984}
985
986//--------------------------------------------------------------------------
987// WithStmt
988template< typename core_t >
989const ast::Decl * ast::Pass< core_t >::visit( const ast::WithStmt * node ) {
990        VISIT_START( node );
991
992        VISIT(
993                maybe_accept( node, &WithStmt::exprs );
994                {
995                        // catch statements introduce a level of scope (for the caught exception)
996                        guard_symtab guard { *this };
997                        __pass::symtab::addWith( core, 0, node->exprs, node );
998                        maybe_accept( node, &WithStmt::stmt );
999                }
1000        )
1001        VISIT_END( Stmt, node );
1002}
1003
1004//--------------------------------------------------------------------------
1005// NullStmt
1006template< typename core_t >
1007const ast::NullStmt * ast::Pass< core_t >::visit( const ast::NullStmt * node ) {
1008        VISIT_START( node );
1009        VISIT_END( NullStmt, node );
1010}
1011
1012//--------------------------------------------------------------------------
1013// DeclStmt
1014template< typename core_t >
1015const ast::Stmt * ast::Pass< core_t >::visit( const ast::DeclStmt * node ) {
1016        VISIT_START( node );
1017
1018        VISIT(
1019                maybe_accept( node, &DeclStmt::decl );
1020        )
1021
1022        VISIT_END( Stmt, node );
1023}
1024
1025//--------------------------------------------------------------------------
1026// ImplicitCtorDtorStmt
1027template< typename core_t >
1028const ast::Stmt * ast::Pass< core_t >::visit( const ast::ImplicitCtorDtorStmt * node ) {
1029        VISIT_START( node );
1030
1031        // For now this isn't visited, it is unclear if this causes problem
1032        // if all tests are known to pass, remove this code
1033        VISIT(
1034                maybe_accept( node, &ImplicitCtorDtorStmt::callStmt );
1035        )
1036
1037        VISIT_END( Stmt, node );
1038}
1039
1040//--------------------------------------------------------------------------
1041// ApplicationExpr
1042template< typename core_t >
1043const ast::Expr * ast::Pass< core_t >::visit( const ast::ApplicationExpr * node ) {
1044        VISIT_START( node );
1045
1046        VISIT(
1047                {
1048                        guard_symtab guard { *this };
1049                        maybe_accept( node, &ApplicationExpr::result );
1050                }
1051                maybe_accept( node, &ApplicationExpr::func );
1052                maybe_accept( node, &ApplicationExpr::args );
1053        )
1054
1055        VISIT_END( Expr, node );
1056}
1057
1058//--------------------------------------------------------------------------
1059// UntypedExpr
1060template< typename core_t >
1061const ast::Expr * ast::Pass< core_t >::visit( const ast::UntypedExpr * node ) {
1062        VISIT_START( node );
1063
1064        VISIT(
1065                {
1066                        guard_symtab guard { *this };
1067                        maybe_accept( node, &UntypedExpr::result );
1068                }
1069
1070                maybe_accept( node, &UntypedExpr::args );
1071        )
1072
1073        VISIT_END( Expr, node );
1074}
1075
1076//--------------------------------------------------------------------------
1077// NameExpr
1078template< typename core_t >
1079const ast::Expr * ast::Pass< core_t >::visit( const ast::NameExpr * node ) {
1080        VISIT_START( node );
1081
1082        VISIT({
1083                guard_symtab guard { *this };
1084                maybe_accept( node, &NameExpr::result );
1085        })
1086
1087        VISIT_END( Expr, node );
1088}
1089
1090//--------------------------------------------------------------------------
1091// CastExpr
1092template< typename core_t >
1093const ast::Expr * ast::Pass< core_t >::visit( const ast::CastExpr * node ) {
1094        VISIT_START( node );
1095
1096        VISIT({
1097                        guard_symtab guard { *this };
1098                        maybe_accept( node, &CastExpr::result );
1099                }
1100                maybe_accept( node, &CastExpr::arg );
1101        )
1102
1103        VISIT_END( Expr, node );
1104}
1105
1106//--------------------------------------------------------------------------
1107// KeywordCastExpr
1108template< typename core_t >
1109const ast::Expr * ast::Pass< core_t >::visit( const ast::KeywordCastExpr * node ) {
1110        VISIT_START( node );
1111
1112        VISIT({
1113                        guard_symtab guard { *this };
1114                        maybe_accept( node, &KeywordCastExpr::result );
1115                }
1116                maybe_accept( node, &KeywordCastExpr::arg );
1117        )
1118
1119        VISIT_END( Expr, node );
1120}
1121
1122//--------------------------------------------------------------------------
1123// VirtualCastExpr
1124template< typename core_t >
1125const ast::Expr * ast::Pass< core_t >::visit( const ast::VirtualCastExpr * node ) {
1126        VISIT_START( node );
1127
1128        VISIT({
1129                        guard_symtab guard { *this };
1130                        maybe_accept( node, &VirtualCastExpr::result );
1131                }
1132                maybe_accept( node, &VirtualCastExpr::arg );
1133        )
1134
1135        VISIT_END( Expr, node );
1136}
1137
1138//--------------------------------------------------------------------------
1139// AddressExpr
1140template< typename core_t >
1141const ast::Expr * ast::Pass< core_t >::visit( const ast::AddressExpr * node ) {
1142        VISIT_START( node );
1143
1144        VISIT({
1145                        guard_symtab guard { *this };
1146                        maybe_accept( node, &AddressExpr::result );
1147                }
1148                maybe_accept( node, &AddressExpr::arg );
1149        )
1150
1151        VISIT_END( Expr, node );
1152}
1153
1154//--------------------------------------------------------------------------
1155// LabelAddressExpr
1156template< typename core_t >
1157const ast::Expr * ast::Pass< core_t >::visit( const ast::LabelAddressExpr * node ) {
1158        VISIT_START( node );
1159
1160        VISIT({
1161                guard_symtab guard { *this };
1162                maybe_accept( node, &LabelAddressExpr::result );
1163        })
1164
1165        VISIT_END( Expr, node );
1166}
1167
1168//--------------------------------------------------------------------------
1169// UntypedMemberExpr
1170template< typename core_t >
1171const ast::Expr * ast::Pass< core_t >::visit( const ast::UntypedMemberExpr * node ) {
1172        VISIT_START( node );
1173
1174        VISIT({
1175                        guard_symtab guard { *this };
1176                        maybe_accept( node, &UntypedMemberExpr::result );
1177                }
1178                maybe_accept( node, &UntypedMemberExpr::aggregate );
1179                maybe_accept( node, &UntypedMemberExpr::member    );
1180        )
1181
1182        VISIT_END( Expr, node );
1183}
1184
1185//--------------------------------------------------------------------------
1186// MemberExpr
1187template< typename core_t >
1188const ast::Expr * ast::Pass< core_t >::visit( const ast::MemberExpr * node ) {
1189        VISIT_START( node );
1190
1191        VISIT({
1192                        guard_symtab guard { *this };
1193                        maybe_accept( node, &MemberExpr::result );
1194                }
1195                maybe_accept( node, &MemberExpr::aggregate );
1196        )
1197
1198        VISIT_END( Expr, node );
1199}
1200
1201//--------------------------------------------------------------------------
1202// VariableExpr
1203template< typename core_t >
1204const ast::Expr * ast::Pass< core_t >::visit( const ast::VariableExpr * node ) {
1205        VISIT_START( node );
1206
1207        VISIT({
1208                guard_symtab guard { *this };
1209                maybe_accept( node, &VariableExpr::result );
1210        })
1211
1212        VISIT_END( Expr, node );
1213}
1214
1215//--------------------------------------------------------------------------
1216// ConstantExpr
1217template< typename core_t >
1218const ast::Expr * ast::Pass< core_t >::visit( const ast::ConstantExpr * node ) {
1219        VISIT_START( node );
1220
1221        VISIT({
1222                guard_symtab guard { *this };
1223                maybe_accept( node, &ConstantExpr::result );
1224        })
1225
1226        VISIT_END( Expr, node );
1227}
1228
1229//--------------------------------------------------------------------------
1230// SizeofExpr
1231template< typename core_t >
1232const ast::Expr * ast::Pass< core_t >::visit( const ast::SizeofExpr * node ) {
1233        VISIT_START( node );
1234
1235        VISIT({
1236                        guard_symtab guard { *this };
1237                        maybe_accept( node, &SizeofExpr::result );
1238                }
1239                if ( node->type ) {
1240                        maybe_accept( node, &SizeofExpr::type );
1241                } else {
1242                        maybe_accept( node, &SizeofExpr::expr );
1243                }
1244        )
1245
1246        VISIT_END( Expr, node );
1247}
1248
1249//--------------------------------------------------------------------------
1250// AlignofExpr
1251template< typename core_t >
1252const ast::Expr * ast::Pass< core_t >::visit( const ast::AlignofExpr * node ) {
1253        VISIT_START( node );
1254
1255        VISIT({
1256                        guard_symtab guard { *this };
1257                        maybe_accept( node, &AlignofExpr::result );
1258                }
1259                if ( node->type ) {
1260                        maybe_accept( node, &AlignofExpr::type );
1261                } else {
1262                        maybe_accept( node, &AlignofExpr::expr );
1263                }
1264        )
1265
1266        VISIT_END( Expr, node );
1267}
1268
1269//--------------------------------------------------------------------------
1270// UntypedOffsetofExpr
1271template< typename core_t >
1272const ast::Expr * ast::Pass< core_t >::visit( const ast::UntypedOffsetofExpr * node ) {
1273        VISIT_START( node );
1274
1275        VISIT({
1276                        guard_symtab guard { *this };
1277                        maybe_accept( node, &UntypedOffsetofExpr::result );
1278                }
1279                maybe_accept( node, &UntypedOffsetofExpr::type   );
1280        )
1281
1282        VISIT_END( Expr, node );
1283}
1284
1285//--------------------------------------------------------------------------
1286// OffsetofExpr
1287template< typename core_t >
1288const ast::Expr * ast::Pass< core_t >::visit( const ast::OffsetofExpr * node ) {
1289        VISIT_START( node );
1290
1291        VISIT({
1292                        guard_symtab guard { *this };
1293                        maybe_accept( node, &OffsetofExpr::result );
1294                }
1295                maybe_accept( node, &OffsetofExpr::type   );
1296        )
1297
1298        VISIT_END( Expr, node );
1299}
1300
1301//--------------------------------------------------------------------------
1302// OffsetPackExpr
1303template< typename core_t >
1304const ast::Expr * ast::Pass< core_t >::visit( const ast::OffsetPackExpr * node ) {
1305        VISIT_START( node );
1306
1307        VISIT({
1308                        guard_symtab guard { *this };
1309                        maybe_accept( node, &OffsetPackExpr::result );
1310                }
1311                maybe_accept( node, &OffsetPackExpr::type   );
1312        )
1313
1314        VISIT_END( Expr, node );
1315}
1316
1317//--------------------------------------------------------------------------
1318// LogicalExpr
1319template< typename core_t >
1320const ast::Expr * ast::Pass< core_t >::visit( const ast::LogicalExpr * node ) {
1321        VISIT_START( node );
1322
1323        VISIT({
1324                        guard_symtab guard { *this };
1325                        maybe_accept( node, &LogicalExpr::result );
1326                }
1327                maybe_accept( node, &LogicalExpr::arg1 );
1328                maybe_accept( node, &LogicalExpr::arg2 );
1329        )
1330
1331        VISIT_END( Expr, node );
1332}
1333
1334//--------------------------------------------------------------------------
1335// ConditionalExpr
1336template< typename core_t >
1337const ast::Expr * ast::Pass< core_t >::visit( const ast::ConditionalExpr * node ) {
1338        VISIT_START( node );
1339
1340        VISIT({
1341                        guard_symtab guard { *this };
1342                        maybe_accept( node, &ConditionalExpr::result );
1343                }
1344                maybe_accept( node, &ConditionalExpr::arg1 );
1345                maybe_accept( node, &ConditionalExpr::arg2 );
1346                maybe_accept( node, &ConditionalExpr::arg3 );
1347        )
1348
1349        VISIT_END( Expr, node );
1350}
1351
1352//--------------------------------------------------------------------------
1353// CommaExpr
1354template< typename core_t >
1355const ast::Expr * ast::Pass< core_t >::visit( const ast::CommaExpr * node ) {
1356        VISIT_START( node );
1357
1358        VISIT({
1359                        guard_symtab guard { *this };
1360                        maybe_accept( node, &CommaExpr::result );
1361                }
1362                maybe_accept( node, &CommaExpr::arg1 );
1363                maybe_accept( node, &CommaExpr::arg2 );
1364        )
1365
1366        VISIT_END( Expr, node );
1367}
1368
1369//--------------------------------------------------------------------------
1370// TypeExpr
1371template< typename core_t >
1372const ast::Expr * ast::Pass< core_t >::visit( const ast::TypeExpr * node ) {
1373        VISIT_START( node );
1374
1375        VISIT({
1376                        guard_symtab guard { *this };
1377                        maybe_accept( node, &TypeExpr::result );
1378                }
1379                maybe_accept( node, &TypeExpr::type );
1380        )
1381
1382        VISIT_END( Expr, node );
1383}
1384
1385//--------------------------------------------------------------------------
1386// AsmExpr
1387template< typename core_t >
1388const ast::Expr * ast::Pass< core_t >::visit( const ast::AsmExpr * node ) {
1389        VISIT_START( node );
1390
1391        VISIT({
1392                        guard_symtab guard { *this };
1393                        maybe_accept( node, &AsmExpr::result );
1394                }
1395                maybe_accept( node, &AsmExpr::constraint );
1396                maybe_accept( node, &AsmExpr::operand    );
1397        )
1398
1399        VISIT_END( Expr, node );
1400}
1401
1402//--------------------------------------------------------------------------
1403// ImplicitCopyCtorExpr
1404template< typename core_t >
1405const ast::Expr * ast::Pass< core_t >::visit( const ast::ImplicitCopyCtorExpr * node ) {
1406        VISIT_START( node );
1407
1408        VISIT({
1409                        guard_symtab guard { *this };
1410                        maybe_accept( node, &ImplicitCopyCtorExpr::result );
1411                }
1412                maybe_accept( node, &ImplicitCopyCtorExpr::callExpr    );
1413        )
1414
1415        VISIT_END( Expr, node );
1416}
1417
1418//--------------------------------------------------------------------------
1419// ConstructorExpr
1420template< typename core_t >
1421const ast::Expr * ast::Pass< core_t >::visit( const ast::ConstructorExpr * node ) {
1422        VISIT_START( node );
1423
1424        VISIT({
1425                        guard_symtab guard { *this };
1426                        maybe_accept( node, &ConstructorExpr::result );
1427                }
1428                maybe_accept( node, &ConstructorExpr::callExpr );
1429        )
1430
1431        VISIT_END( Expr, node );
1432}
1433
1434//--------------------------------------------------------------------------
1435// CompoundLiteralExpr
1436template< typename core_t >
1437const ast::Expr * ast::Pass< core_t >::visit( const ast::CompoundLiteralExpr * node ) {
1438        VISIT_START( node );
1439
1440        VISIT({
1441                        guard_symtab guard { *this };
1442                        maybe_accept( node, &CompoundLiteralExpr::result );
1443                }
1444                maybe_accept( node, &CompoundLiteralExpr::init );
1445        )
1446
1447        VISIT_END( Expr, node );
1448}
1449
1450//--------------------------------------------------------------------------
1451// RangeExpr
1452template< typename core_t >
1453const ast::Expr * ast::Pass< core_t >::visit( const ast::RangeExpr * node ) {
1454        VISIT_START( node );
1455
1456        VISIT({
1457                        guard_symtab guard { *this };
1458                        maybe_accept( node, &RangeExpr::result );
1459                }
1460                maybe_accept( node, &RangeExpr::low    );
1461                maybe_accept( node, &RangeExpr::high   );
1462        )
1463
1464        VISIT_END( Expr, node );
1465}
1466
1467//--------------------------------------------------------------------------
1468// UntypedTupleExpr
1469template< typename core_t >
1470const ast::Expr * ast::Pass< core_t >::visit( const ast::UntypedTupleExpr * node ) {
1471        VISIT_START( node );
1472
1473        VISIT({
1474                        guard_symtab guard { *this };
1475                        maybe_accept( node, &UntypedTupleExpr::result );
1476                }
1477                maybe_accept( node, &UntypedTupleExpr::exprs  );
1478        )
1479
1480        VISIT_END( Expr, node );
1481}
1482
1483//--------------------------------------------------------------------------
1484// TupleExpr
1485template< typename core_t >
1486const ast::Expr * ast::Pass< core_t >::visit( const ast::TupleExpr * node ) {
1487        VISIT_START( node );
1488
1489        VISIT({
1490                        guard_symtab guard { *this };
1491                        maybe_accept( node, &TupleExpr::result );
1492                }
1493                maybe_accept( node, &TupleExpr::exprs  );
1494        )
1495
1496        VISIT_END( Expr, node );
1497}
1498
1499//--------------------------------------------------------------------------
1500// TupleIndexExpr
1501template< typename core_t >
1502const ast::Expr * ast::Pass< core_t >::visit( const ast::TupleIndexExpr * node ) {
1503        VISIT_START( node );
1504
1505        VISIT({
1506                        guard_symtab guard { *this };
1507                        maybe_accept( node, &TupleIndexExpr::result );
1508                }
1509                maybe_accept( node, &TupleIndexExpr::tuple  );
1510        )
1511
1512        VISIT_END( Expr, node );
1513}
1514
1515//--------------------------------------------------------------------------
1516// TupleAssignExpr
1517template< typename core_t >
1518const ast::Expr * ast::Pass< core_t >::visit( const ast::TupleAssignExpr * node ) {
1519        VISIT_START( node );
1520
1521        VISIT({
1522                        guard_symtab guard { *this };
1523                        maybe_accept( node, &TupleAssignExpr::result );
1524                }
1525                maybe_accept( node, &TupleAssignExpr::stmtExpr );
1526        )
1527
1528        VISIT_END( Expr, node );
1529}
1530
1531//--------------------------------------------------------------------------
1532// StmtExpr
1533template< typename core_t >
1534const ast::Expr * ast::Pass< core_t >::visit( const ast::StmtExpr * node ) {
1535        VISIT_START( node );
1536
1537        VISIT(// don't want statements from outer CompoundStmts to be added to this StmtExpr
1538                // get the stmts that will need to be spliced in
1539                auto stmts_before = __pass::stmtsToAddBefore( core, 0);
1540                auto stmts_after  = __pass::stmtsToAddAfter ( core, 0);
1541
1542                // These may be modified by subnode but most be restored once we exit this statemnet.
1543                ValueGuardPtr< const ast::TypeSubstitution * > __old_env( __pass::typeSubs( core, 0 ) );
1544                ValueGuardPtr< typename std::remove_pointer< decltype(stmts_before) >::type > __old_decls_before( stmts_before );
1545                ValueGuardPtr< typename std::remove_pointer< decltype(stmts_after ) >::type > __old_decls_after ( stmts_after  );
1546
1547                {
1548                        guard_symtab guard { *this };
1549                        maybe_accept( node, &StmtExpr::result );
1550                }
1551                maybe_accept( node, &StmtExpr::stmts       );
1552                maybe_accept( node, &StmtExpr::returnDecls );
1553                maybe_accept( node, &StmtExpr::dtors       );
1554        )
1555
1556        VISIT_END( Expr, node );
1557}
1558
1559//--------------------------------------------------------------------------
1560// UniqueExpr
1561template< typename core_t >
1562const ast::Expr * ast::Pass< core_t >::visit( const ast::UniqueExpr * node ) {
1563        VISIT_START( node );
1564
1565        VISIT({
1566                        guard_symtab guard { *this };
1567                        maybe_accept( node, &UniqueExpr::result );
1568                }
1569                maybe_accept( node, &UniqueExpr::expr   );
1570        )
1571
1572        VISIT_END( Expr, node );
1573}
1574
1575//--------------------------------------------------------------------------
1576// UntypedInitExpr
1577template< typename core_t >
1578const ast::Expr * ast::Pass< core_t >::visit( const ast::UntypedInitExpr * node ) {
1579        VISIT_START( node );
1580
1581        VISIT({
1582                        guard_symtab guard { *this };
1583                        maybe_accept( node, &UntypedInitExpr::result );
1584                }
1585                maybe_accept( node, &UntypedInitExpr::expr   );
1586                // not currently visiting initAlts, but this doesn't matter since this node is only used in the resolver.
1587        )
1588
1589        VISIT_END( Expr, node );
1590}
1591
1592//--------------------------------------------------------------------------
1593// InitExpr
1594template< typename core_t >
1595const ast::Expr * ast::Pass< core_t >::visit( const ast::InitExpr * node ) {
1596        VISIT_START( node );
1597
1598        VISIT({
1599                        guard_symtab guard { *this };
1600                        maybe_accept( node, &InitExpr::result );
1601                }
1602                maybe_accept( node, &InitExpr::expr   );
1603                maybe_accept( node, &InitExpr::designation );
1604        )
1605
1606        VISIT_END( Expr, node );
1607}
1608
1609//--------------------------------------------------------------------------
1610// DeletedExpr
1611template< typename core_t >
1612const ast::Expr * ast::Pass< core_t >::visit( const ast::DeletedExpr * node ) {
1613        VISIT_START( node );
1614
1615        VISIT({
1616                        guard_symtab guard { *this };
1617                        maybe_accept( node, &DeletedExpr::result );
1618                }
1619                maybe_accept( node, &DeletedExpr::expr );
1620                // don't visit deleteStmt, because it is a pointer to somewhere else in the tree.
1621        )
1622
1623        VISIT_END( Expr, node );
1624}
1625
1626//--------------------------------------------------------------------------
1627// DefaultArgExpr
1628template< typename core_t >
1629const ast::Expr * ast::Pass< core_t >::visit( const ast::DefaultArgExpr * node ) {
1630        VISIT_START( node );
1631
1632        VISIT({
1633                        guard_symtab guard { *this };
1634                        maybe_accept( node, &DefaultArgExpr::result );
1635                }
1636                maybe_accept( node, &DefaultArgExpr::expr );
1637        )
1638
1639        VISIT_END( Expr, node );
1640}
1641
1642//--------------------------------------------------------------------------
1643// GenericExpr
1644template< typename core_t >
1645const ast::Expr * ast::Pass< core_t >::visit( const ast::GenericExpr * node ) {
1646        VISIT_START( node );
1647
1648        VISIT({
1649                        guard_symtab guard { *this };
1650                        maybe_accept( node, &GenericExpr::result );
1651                }
1652                maybe_accept( node, &GenericExpr::control );
1653
1654                std::vector<GenericExpr::Association> new_kids;
1655                new_kids.reserve(node->associations.size());
1656                bool mutated = false;
1657                for( const auto & assoc : node->associations ) {
1658                        const Type * type = nullptr;
1659                        if( assoc.type ) {
1660                                guard_symtab guard { *this };
1661                                type = assoc.type->accept( *this );
1662                                if( type != assoc.type ) mutated = true;
1663                        }
1664                        const Expr * expr = nullptr;
1665                        if( assoc.expr ) {
1666                                expr = assoc.expr->accept( *this );
1667                                if( expr != assoc.expr ) mutated = true;
1668                        }
1669                        new_kids.emplace_back( type, expr );
1670                }
1671
1672                if(mutated) {
1673                        auto n = __pass::mutate<core_t>(node);
1674                        n->associations = std::move( new_kids );
1675                        node = n;
1676                }
1677        )
1678
1679        VISIT_END( Expr, node );
1680}
1681
1682//--------------------------------------------------------------------------
1683// VoidType
1684template< typename core_t >
1685const ast::Type * ast::Pass< core_t >::visit( const ast::VoidType * node ) {
1686        VISIT_START( node );
1687
1688        VISIT_END( Type, node );
1689}
1690
1691//--------------------------------------------------------------------------
1692// BasicType
1693template< typename core_t >
1694const ast::Type * ast::Pass< core_t >::visit( const ast::BasicType * node ) {
1695        VISIT_START( node );
1696
1697        VISIT_END( Type, node );
1698}
1699
1700//--------------------------------------------------------------------------
1701// PointerType
1702template< typename core_t >
1703const ast::Type * ast::Pass< core_t >::visit( const ast::PointerType * node ) {
1704        VISIT_START( node );
1705
1706        VISIT(
1707                // xxx - should PointerType visit/mutate dimension?
1708                maybe_accept( node, &PointerType::base );
1709        )
1710
1711        VISIT_END( Type, node );
1712}
1713
1714//--------------------------------------------------------------------------
1715// ArrayType
1716template< typename core_t >
1717const ast::Type * ast::Pass< core_t >::visit( const ast::ArrayType * node ) {
1718        VISIT_START( node );
1719
1720        VISIT(
1721                maybe_accept( node, &ArrayType::dimension );
1722                maybe_accept( node, &ArrayType::base );
1723        )
1724
1725        VISIT_END( Type, node );
1726}
1727
1728//--------------------------------------------------------------------------
1729// ReferenceType
1730template< typename core_t >
1731const ast::Type * ast::Pass< core_t >::visit( const ast::ReferenceType * node ) {
1732        VISIT_START( node );
1733
1734        VISIT(
1735                maybe_accept( node, &ReferenceType::base );
1736        )
1737
1738        VISIT_END( Type, node );
1739}
1740
1741//--------------------------------------------------------------------------
1742// QualifiedType
1743template< typename core_t >
1744const ast::Type * ast::Pass< core_t >::visit( const ast::QualifiedType * node ) {
1745        VISIT_START( node );
1746
1747        VISIT(
1748                maybe_accept( node, &QualifiedType::parent );
1749                maybe_accept( node, &QualifiedType::child );
1750        )
1751
1752        VISIT_END( Type, node );
1753}
1754
1755//--------------------------------------------------------------------------
1756// FunctionType
1757template< typename core_t >
1758const ast::Type * ast::Pass< core_t >::visit( const ast::FunctionType * node ) {
1759        VISIT_START( node );
1760
1761        VISIT({
1762                guard_forall_subs forall_guard { *this, node };
1763                mutate_forall( node );
1764                maybe_accept( node, &FunctionType::returns );
1765                maybe_accept( node, &FunctionType::params  );
1766        })
1767
1768        VISIT_END( Type, node );
1769}
1770
1771//--------------------------------------------------------------------------
1772// StructInstType
1773template< typename core_t >
1774const ast::Type * ast::Pass< core_t >::visit( const ast::StructInstType * node ) {
1775        VISIT_START( node );
1776
1777        __pass::symtab::addStruct( core, 0, node->name );
1778
1779        VISIT({
1780                guard_symtab guard { *this };
1781                guard_forall_subs forall_guard { *this, node };
1782                mutate_forall( node );
1783                maybe_accept( node, &StructInstType::params );
1784        })
1785
1786        VISIT_END( Type, node );
1787}
1788
1789//--------------------------------------------------------------------------
1790// UnionInstType
1791template< typename core_t >
1792const ast::Type * ast::Pass< core_t >::visit( const ast::UnionInstType * node ) {
1793        VISIT_START( node );
1794
1795        __pass::symtab::addUnion( core, 0, node->name );
1796
1797        VISIT({
1798                guard_symtab guard { *this };
1799                guard_forall_subs forall_guard { *this, node };
1800                mutate_forall( node );
1801                maybe_accept( node, &UnionInstType::params );
1802        })
1803
1804        VISIT_END( Type, node );
1805}
1806
1807//--------------------------------------------------------------------------
1808// EnumInstType
1809template< typename core_t >
1810const ast::Type * ast::Pass< core_t >::visit( const ast::EnumInstType * node ) {
1811        VISIT_START( node );
1812
1813        VISIT({
1814                guard_forall_subs forall_guard { *this, node };
1815                mutate_forall( node );
1816                maybe_accept( node, &EnumInstType::params );
1817        })
1818
1819        VISIT_END( Type, node );
1820}
1821
1822//--------------------------------------------------------------------------
1823// TraitInstType
1824template< typename core_t >
1825const ast::Type * ast::Pass< core_t >::visit( const ast::TraitInstType * node ) {
1826        VISIT_START( node );
1827
1828        VISIT({
1829                guard_forall_subs forall_guard { *this, node };
1830                mutate_forall( node );
1831                maybe_accept( node, &TraitInstType::params );
1832        })
1833
1834        VISIT_END( Type, node );
1835}
1836
1837//--------------------------------------------------------------------------
1838// TypeInstType
1839template< typename core_t >
1840const ast::Type * ast::Pass< core_t >::visit( const ast::TypeInstType * node ) {
1841        VISIT_START( node );
1842
1843        VISIT(
1844                {
1845                        guard_forall_subs forall_guard { *this, node };
1846                        mutate_forall( node );
1847                        maybe_accept( node, &TypeInstType::params );
1848                }
1849                // ensure that base re-bound if doing substitution
1850                __pass::forall::replace( core, 0, node );
1851        )
1852
1853        VISIT_END( Type, node );
1854}
1855
1856//--------------------------------------------------------------------------
1857// TupleType
1858template< typename core_t >
1859const ast::Type * ast::Pass< core_t >::visit( const ast::TupleType * node ) {
1860        VISIT_START( node );
1861
1862        VISIT(
1863                maybe_accept( node, &TupleType::types );
1864                maybe_accept( node, &TupleType::members );
1865        )
1866
1867        VISIT_END( Type, node );
1868}
1869
1870//--------------------------------------------------------------------------
1871// TypeofType
1872template< typename core_t >
1873const ast::Type * ast::Pass< core_t >::visit( const ast::TypeofType * node ) {
1874        VISIT_START( node );
1875
1876        VISIT(
1877                maybe_accept( node, &TypeofType::expr );
1878        )
1879
1880        VISIT_END( Type, node );
1881}
1882
1883//--------------------------------------------------------------------------
1884// VarArgsType
1885template< typename core_t >
1886const ast::Type * ast::Pass< core_t >::visit( const ast::VarArgsType * node ) {
1887        VISIT_START( node );
1888
1889        VISIT_END( Type, node );
1890}
1891
1892//--------------------------------------------------------------------------
1893// ZeroType
1894template< typename core_t >
1895const ast::Type * ast::Pass< core_t >::visit( const ast::ZeroType * node ) {
1896        VISIT_START( node );
1897
1898        VISIT_END( Type, node );
1899}
1900
1901//--------------------------------------------------------------------------
1902// OneType
1903template< typename core_t >
1904const ast::Type * ast::Pass< core_t >::visit( const ast::OneType * node ) {
1905        VISIT_START( node );
1906
1907        VISIT_END( Type, node );
1908}
1909
1910//--------------------------------------------------------------------------
1911// GlobalScopeType
1912template< typename core_t >
1913const ast::Type * ast::Pass< core_t >::visit( const ast::GlobalScopeType * node ) {
1914        VISIT_START( node );
1915
1916        VISIT_END( Type, node );
1917}
1918
1919
1920//--------------------------------------------------------------------------
1921// Designation
1922template< typename core_t >
1923const ast::Designation * ast::Pass< core_t >::visit( const ast::Designation * node ) {
1924        VISIT_START( node );
1925
1926        VISIT( maybe_accept( node, &Designation::designators ); )
1927
1928        VISIT_END( Designation, node );
1929}
1930
1931//--------------------------------------------------------------------------
1932// SingleInit
1933template< typename core_t >
1934const ast::Init * ast::Pass< core_t >::visit( const ast::SingleInit * node ) {
1935        VISIT_START( node );
1936
1937        VISIT(
1938                maybe_accept( node, &SingleInit::value );
1939        )
1940
1941        VISIT_END( Init, node );
1942}
1943
1944//--------------------------------------------------------------------------
1945// ListInit
1946template< typename core_t >
1947const ast::Init * ast::Pass< core_t >::visit( const ast::ListInit * node ) {
1948        VISIT_START( node );
1949
1950        VISIT(
1951                maybe_accept( node, &ListInit::designations );
1952                maybe_accept( node, &ListInit::initializers );
1953        )
1954
1955        VISIT_END( Init, node );
1956}
1957
1958//--------------------------------------------------------------------------
1959// ConstructorInit
1960template< typename core_t >
1961const ast::Init * ast::Pass< core_t >::visit( const ast::ConstructorInit * node ) {
1962        VISIT_START( node );
1963
1964        VISIT(
1965                maybe_accept( node, &ConstructorInit::ctor );
1966                maybe_accept( node, &ConstructorInit::dtor );
1967                maybe_accept( node, &ConstructorInit::init );
1968        )
1969
1970        VISIT_END( Init, node );
1971}
1972
1973//--------------------------------------------------------------------------
1974// Attribute
1975template< typename core_t >
1976const ast::Attribute * ast::Pass< core_t >::visit( const ast::Attribute * node  )  {
1977        VISIT_START( node );
1978
1979        VISIT(
1980                maybe_accept( node, &Attribute::params );
1981        )
1982
1983        VISIT_END( Attribute, node );
1984}
1985
1986//--------------------------------------------------------------------------
1987// TypeSubstitution
1988template< typename core_t >
1989const ast::TypeSubstitution * ast::Pass< core_t >::visit( const ast::TypeSubstitution * node ) {
1990        VISIT_START( node );
1991
1992        VISIT(
1993                {
1994                        bool mutated = false;
1995                        std::unordered_map< std::string, ast::ptr< ast::Type > > new_map;
1996                        for ( const auto & p : node->typeEnv ) {
1997                                guard_symtab guard { *this };
1998                                auto new_node = p.second->accept( *this );
1999                                if (new_node != p.second) mutated = true;
2000                                new_map.insert({ p.first, new_node });
2001                        }
2002                        if (mutated) {
2003                                auto new_node = __pass::mutate<core_t>( node );
2004                                new_node->typeEnv.swap( new_map );
2005                                node = new_node;
2006                        }
2007                }
2008
2009                {
2010                        bool mutated = false;
2011                        std::unordered_map< std::string, ast::ptr< ast::Expr > > new_map;
2012                        for ( const auto & p : node->varEnv ) {
2013                                guard_symtab guard { *this };
2014                                auto new_node = p.second->accept( *this );
2015                                if (new_node != p.second) mutated = true;
2016                                new_map.insert({ p.first, new_node });
2017                        }
2018                        if (mutated) {
2019                                auto new_node = __pass::mutate<core_t>( node );
2020                                new_node->varEnv.swap( new_map );
2021                                node = new_node;
2022                        }
2023                }
2024        )
2025
2026        VISIT_END( TypeSubstitution, node );
2027}
2028
2029#undef VISIT_START
2030#undef VISIT
2031#undef VISIT_END
Note: See TracBrowser for help on using the repository browser.