[13932f14] | 1 | #pragma once |
---|
| 2 | |
---|
[08fc48f] | 3 | // IWYU pragma: private, include "Common/PassVisitor.h" |
---|
| 4 | |
---|
[296b2be] | 5 | #include <stack> |
---|
| 6 | |
---|
[3c398b6] | 7 | #include "Common/utility.h" |
---|
| 8 | |
---|
[ab904dc] | 9 | #include "SynTree/Mutator.h" |
---|
[13932f14] | 10 | #include "SynTree/Visitor.h" |
---|
| 11 | |
---|
[e0886db] | 12 | #include "SymTab/Indexer.h" |
---|
| 13 | |
---|
[5ea7a22] | 14 | #include "SynTree/Attribute.h" |
---|
[ab904dc] | 15 | #include "SynTree/Initializer.h" |
---|
| 16 | #include "SynTree/Statement.h" |
---|
| 17 | #include "SynTree/Type.h" |
---|
| 18 | #include "SynTree/Declaration.h" |
---|
| 19 | #include "SynTree/Expression.h" |
---|
| 20 | #include "SynTree/Constant.h" |
---|
[07c178f0] | 21 | |
---|
| 22 | class TypeSubstitution; |
---|
[ab904dc] | 23 | |
---|
[296b2be] | 24 | #include "PassVisitor.proto.h" |
---|
[ab904dc] | 25 | |
---|
[13932f14] | 26 | //------------------------------------------------------------------------------------------------------------------------------------------------------------------------- |
---|
[5d88a0a] | 27 | // Templated visitor type |
---|
| 28 | // To use declare a PassVisitor< YOUR VISITOR TYPE > |
---|
[5e298d7] | 29 | // The visitor type should specify the previsit/postvisit/premutate/postmutate for types that are desired. |
---|
[4551a6e] | 30 | // Note: previsit/postvisit/premutate/postmutate must be **public** members |
---|
[5e298d7] | 31 | // |
---|
| 32 | // Several additional features are available through inheritance |
---|
| 33 | // | WithTypeSubstitution - provides polymorphic TypeSubstitution * env for the current expression |
---|
| 34 | // | WithStmtsToAdd - provides the ability to insert statements before or after the current statement by adding new statements into |
---|
| 35 | // stmtsToAddBefore or stmtsToAddAfter respectively. |
---|
[6ca154b] | 36 | // | WithShortCircuiting - provides the ability to skip visiting child nodes; set visit_children to false in pre{visit,mutate} to skip visiting children |
---|
[d24d4e1] | 37 | // | WithGuards - provides the ability to save/restore data like a LIFO stack; to save, call GuardValue with the variable to save, the variable |
---|
[5e298d7] | 38 | // will automatically be restored to its previous value after the corresponding postvisit/postmutate teminates. |
---|
[13932f14] | 39 | //------------------------------------------------------------------------------------------------------------------------------------------------------------------------- |
---|
| 40 | template< typename pass_type > |
---|
[ab904dc] | 41 | class PassVisitor final : public Visitor, public Mutator { |
---|
[13932f14] | 42 | public: |
---|
| 43 | |
---|
| 44 | template< typename... Args > |
---|
[aff3af4] | 45 | PassVisitor(Args &&... args) |
---|
[13932f14] | 46 | : pass( std::forward<Args>( args )... ) |
---|
[b73bd70] | 47 | { |
---|
| 48 | typedef PassVisitor<pass_type> this_t; |
---|
| 49 | this_t * const * visitor = visitor_impl(pass, 0); |
---|
| 50 | if(visitor) { |
---|
| 51 | *const_cast<this_t **>( visitor ) = this; |
---|
| 52 | } |
---|
| 53 | } |
---|
[13932f14] | 54 | |
---|
| 55 | virtual ~PassVisitor() = default; |
---|
| 56 | |
---|
[aff3af4] | 57 | pass_type pass; |
---|
[13932f14] | 58 | |
---|
[5ea7a22] | 59 | virtual void visit( ObjectDecl * objectDecl ) override final; |
---|
| 60 | virtual void visit( FunctionDecl * functionDecl ) override final; |
---|
| 61 | virtual void visit( StructDecl * aggregateDecl ) override final; |
---|
| 62 | virtual void visit( UnionDecl * aggregateDecl ) override final; |
---|
| 63 | virtual void visit( EnumDecl * aggregateDecl ) override final; |
---|
| 64 | virtual void visit( TraitDecl * aggregateDecl ) override final; |
---|
| 65 | virtual void visit( TypeDecl * typeDecl ) override final; |
---|
| 66 | virtual void visit( TypedefDecl * typeDecl ) override final; |
---|
| 67 | virtual void visit( AsmDecl * asmDecl ) override final; |
---|
[f6e3e34] | 68 | virtual void visit( StaticAssertDecl * assertDecl ) override final; |
---|
[5ea7a22] | 69 | |
---|
| 70 | virtual void visit( CompoundStmt * compoundStmt ) override final; |
---|
| 71 | virtual void visit( ExprStmt * exprStmt ) override final; |
---|
| 72 | virtual void visit( AsmStmt * asmStmt ) override final; |
---|
[cc32d83] | 73 | virtual void visit( DirectiveStmt * dirStmt ) override final; |
---|
[5ea7a22] | 74 | virtual void visit( IfStmt * ifStmt ) override final; |
---|
| 75 | virtual void visit( WhileStmt * whileStmt ) override final; |
---|
| 76 | virtual void visit( ForStmt * forStmt ) override final; |
---|
| 77 | virtual void visit( SwitchStmt * switchStmt ) override final; |
---|
| 78 | virtual void visit( CaseStmt * caseStmt ) override final; |
---|
| 79 | virtual void visit( BranchStmt * branchStmt ) override final; |
---|
| 80 | virtual void visit( ReturnStmt * returnStmt ) override final; |
---|
| 81 | virtual void visit( ThrowStmt * throwStmt ) override final; |
---|
| 82 | virtual void visit( TryStmt * tryStmt ) override final; |
---|
| 83 | virtual void visit( CatchStmt * catchStmt ) override final; |
---|
| 84 | virtual void visit( FinallyStmt * finallyStmt ) override final; |
---|
| 85 | virtual void visit( WaitForStmt * waitforStmt ) override final; |
---|
[61255ad] | 86 | virtual void visit( WithStmt * withStmt ) override final; |
---|
[5ea7a22] | 87 | virtual void visit( NullStmt * nullStmt ) override final; |
---|
| 88 | virtual void visit( DeclStmt * declStmt ) override final; |
---|
| 89 | virtual void visit( ImplicitCtorDtorStmt * impCtorDtorStmt ) override final; |
---|
| 90 | |
---|
| 91 | virtual void visit( ApplicationExpr * applicationExpr ) override final; |
---|
| 92 | virtual void visit( UntypedExpr * untypedExpr ) override final; |
---|
| 93 | virtual void visit( NameExpr * nameExpr ) override final; |
---|
| 94 | virtual void visit( CastExpr * castExpr ) override final; |
---|
[9a705dc8] | 95 | virtual void visit( KeywordCastExpr * castExpr ) override final; |
---|
[5ea7a22] | 96 | virtual void visit( VirtualCastExpr * castExpr ) override final; |
---|
| 97 | virtual void visit( AddressExpr * addressExpr ) override final; |
---|
| 98 | virtual void visit( LabelAddressExpr * labAddressExpr ) override final; |
---|
| 99 | virtual void visit( UntypedMemberExpr * memberExpr ) override final; |
---|
| 100 | virtual void visit( MemberExpr * memberExpr ) override final; |
---|
| 101 | virtual void visit( VariableExpr * variableExpr ) override final; |
---|
| 102 | virtual void visit( ConstantExpr * constantExpr ) override final; |
---|
| 103 | virtual void visit( SizeofExpr * sizeofExpr ) override final; |
---|
| 104 | virtual void visit( AlignofExpr * alignofExpr ) override final; |
---|
| 105 | virtual void visit( UntypedOffsetofExpr * offsetofExpr ) override final; |
---|
| 106 | virtual void visit( OffsetofExpr * offsetofExpr ) override final; |
---|
| 107 | virtual void visit( OffsetPackExpr * offsetPackExpr ) override final; |
---|
| 108 | virtual void visit( AttrExpr * attrExpr ) override final; |
---|
| 109 | virtual void visit( LogicalExpr * logicalExpr ) override final; |
---|
| 110 | virtual void visit( ConditionalExpr * conditionalExpr ) override final; |
---|
| 111 | virtual void visit( CommaExpr * commaExpr ) override final; |
---|
| 112 | virtual void visit( TypeExpr * typeExpr ) override final; |
---|
| 113 | virtual void visit( AsmExpr * asmExpr ) override final; |
---|
| 114 | virtual void visit( ImplicitCopyCtorExpr * impCpCtorExpr ) override final; |
---|
| 115 | virtual void visit( ConstructorExpr * ctorExpr ) override final; |
---|
| 116 | virtual void visit( CompoundLiteralExpr * compLitExpr ) override final; |
---|
| 117 | virtual void visit( RangeExpr * rangeExpr ) override final; |
---|
| 118 | virtual void visit( UntypedTupleExpr * tupleExpr ) override final; |
---|
| 119 | virtual void visit( TupleExpr * tupleExpr ) override final; |
---|
| 120 | virtual void visit( TupleIndexExpr * tupleExpr ) override final; |
---|
| 121 | virtual void visit( TupleAssignExpr * assignExpr ) override final; |
---|
| 122 | virtual void visit( StmtExpr * stmtExpr ) override final; |
---|
| 123 | virtual void visit( UniqueExpr * uniqueExpr ) override final; |
---|
[73367a8] | 124 | virtual void visit( UntypedInitExpr * initExpr ) override final; |
---|
| 125 | virtual void visit( InitExpr * initExpr ) override final; |
---|
[44b4114] | 126 | virtual void visit( DeletedExpr * delExpr ) override final; |
---|
[0f79853] | 127 | virtual void visit( DefaultArgExpr * argExpr ) override final; |
---|
[d807ca28] | 128 | virtual void visit( GenericExpr * genExpr ) override final; |
---|
[5ea7a22] | 129 | |
---|
| 130 | virtual void visit( VoidType * basicType ) override final; |
---|
| 131 | virtual void visit( BasicType * basicType ) override final; |
---|
| 132 | virtual void visit( PointerType * pointerType ) override final; |
---|
| 133 | virtual void visit( ArrayType * arrayType ) override final; |
---|
| 134 | virtual void visit( ReferenceType * referenceType ) override final; |
---|
| 135 | virtual void visit( FunctionType * functionType ) override final; |
---|
| 136 | virtual void visit( StructInstType * aggregateUseType ) override final; |
---|
| 137 | virtual void visit( UnionInstType * aggregateUseType ) override final; |
---|
| 138 | virtual void visit( EnumInstType * aggregateUseType ) override final; |
---|
| 139 | virtual void visit( TraitInstType * aggregateUseType ) override final; |
---|
| 140 | virtual void visit( TypeInstType * aggregateUseType ) override final; |
---|
| 141 | virtual void visit( TupleType * tupleType ) override final; |
---|
| 142 | virtual void visit( TypeofType * typeofType ) override final; |
---|
| 143 | virtual void visit( AttrType * attrType ) override final; |
---|
| 144 | virtual void visit( VarArgsType * varArgsType ) override final; |
---|
| 145 | virtual void visit( ZeroType * zeroType ) override final; |
---|
| 146 | virtual void visit( OneType * oneType ) override final; |
---|
| 147 | |
---|
| 148 | virtual void visit( Designation * designation ) override final; |
---|
| 149 | virtual void visit( SingleInit * singleInit ) override final; |
---|
| 150 | virtual void visit( ListInit * listInit ) override final; |
---|
| 151 | virtual void visit( ConstructorInit * ctorInit ) override final; |
---|
| 152 | |
---|
| 153 | virtual void visit( Subrange * subrange ) override final; |
---|
| 154 | |
---|
| 155 | virtual void visit( Constant * constant ) override final; |
---|
| 156 | |
---|
| 157 | virtual void visit( Attribute * attribute ) override final; |
---|
| 158 | |
---|
| 159 | virtual DeclarationWithType * mutate( ObjectDecl * objectDecl ) override final; |
---|
| 160 | virtual DeclarationWithType * mutate( FunctionDecl * functionDecl ) override final; |
---|
| 161 | virtual Declaration * mutate( StructDecl * aggregateDecl ) override final; |
---|
| 162 | virtual Declaration * mutate( UnionDecl * aggregateDecl ) override final; |
---|
| 163 | virtual Declaration * mutate( EnumDecl * aggregateDecl ) override final; |
---|
| 164 | virtual Declaration * mutate( TraitDecl * aggregateDecl ) override final; |
---|
| 165 | virtual Declaration * mutate( TypeDecl * typeDecl ) override final; |
---|
| 166 | virtual Declaration * mutate( TypedefDecl * typeDecl ) override final; |
---|
| 167 | virtual AsmDecl * mutate( AsmDecl * asmDecl ) override final; |
---|
[f6e3e34] | 168 | virtual StaticAssertDecl * mutate( StaticAssertDecl * assertDecl ) override final; |
---|
[5ea7a22] | 169 | |
---|
| 170 | virtual CompoundStmt * mutate( CompoundStmt * compoundStmt ) override final; |
---|
| 171 | virtual Statement * mutate( ExprStmt * exprStmt ) override final; |
---|
| 172 | virtual Statement * mutate( AsmStmt * asmStmt ) override final; |
---|
[cc32d83] | 173 | virtual Statement * mutate( DirectiveStmt * dirStmt ) override final; |
---|
[5ea7a22] | 174 | virtual Statement * mutate( IfStmt * ifStmt ) override final; |
---|
| 175 | virtual Statement * mutate( WhileStmt * whileStmt ) override final; |
---|
| 176 | virtual Statement * mutate( ForStmt * forStmt ) override final; |
---|
| 177 | virtual Statement * mutate( SwitchStmt * switchStmt ) override final; |
---|
| 178 | virtual Statement * mutate( CaseStmt * caseStmt ) override final; |
---|
| 179 | virtual Statement * mutate( BranchStmt * branchStmt ) override final; |
---|
| 180 | virtual Statement * mutate( ReturnStmt * returnStmt ) override final; |
---|
| 181 | virtual Statement * mutate( ThrowStmt * throwStmt ) override final; |
---|
| 182 | virtual Statement * mutate( TryStmt * tryStmt ) override final; |
---|
| 183 | virtual Statement * mutate( CatchStmt * catchStmt ) override final; |
---|
| 184 | virtual Statement * mutate( FinallyStmt * finallyStmt ) override final; |
---|
| 185 | virtual Statement * mutate( WaitForStmt * waitforStmt ) override final; |
---|
[61255ad] | 186 | virtual Statement * mutate( WithStmt * withStmt ) override final; |
---|
[5ea7a22] | 187 | virtual NullStmt * mutate( NullStmt * nullStmt ) override final; |
---|
| 188 | virtual Statement * mutate( DeclStmt * declStmt ) override final; |
---|
| 189 | virtual Statement * mutate( ImplicitCtorDtorStmt * impCtorDtorStmt ) override final; |
---|
| 190 | |
---|
| 191 | virtual Expression * mutate( ApplicationExpr * applicationExpr ) override final; |
---|
| 192 | virtual Expression * mutate( UntypedExpr * untypedExpr ) override final; |
---|
| 193 | virtual Expression * mutate( NameExpr * nameExpr ) override final; |
---|
[9a705dc8] | 194 | virtual Expression * mutate( AddressExpr * addrExpr ) override final; |
---|
[5ea7a22] | 195 | virtual Expression * mutate( LabelAddressExpr * labAddressExpr ) override final; |
---|
| 196 | virtual Expression * mutate( CastExpr * castExpr ) override final; |
---|
[9a705dc8] | 197 | virtual Expression * mutate( KeywordCastExpr * castExpr ) override final; |
---|
[5ea7a22] | 198 | virtual Expression * mutate( VirtualCastExpr * castExpr ) override final; |
---|
| 199 | virtual Expression * mutate( UntypedMemberExpr * memberExpr ) override final; |
---|
| 200 | virtual Expression * mutate( MemberExpr * memberExpr ) override final; |
---|
| 201 | virtual Expression * mutate( VariableExpr * variableExpr ) override final; |
---|
| 202 | virtual Expression * mutate( ConstantExpr * constantExpr ) override final; |
---|
| 203 | virtual Expression * mutate( SizeofExpr * sizeofExpr ) override final; |
---|
| 204 | virtual Expression * mutate( AlignofExpr * alignofExpr ) override final; |
---|
| 205 | virtual Expression * mutate( UntypedOffsetofExpr * offsetofExpr ) override final; |
---|
| 206 | virtual Expression * mutate( OffsetofExpr * offsetofExpr ) override final; |
---|
| 207 | virtual Expression * mutate( OffsetPackExpr * offsetPackExpr ) override final; |
---|
| 208 | virtual Expression * mutate( AttrExpr * attrExpr ) override final; |
---|
| 209 | virtual Expression * mutate( LogicalExpr * logicalExpr ) override final; |
---|
| 210 | virtual Expression * mutate( ConditionalExpr * conditionalExpr ) override final; |
---|
| 211 | virtual Expression * mutate( CommaExpr * commaExpr ) override final; |
---|
| 212 | virtual Expression * mutate( TypeExpr * typeExpr ) override final; |
---|
| 213 | virtual Expression * mutate( AsmExpr * asmExpr ) override final; |
---|
| 214 | virtual Expression * mutate( ImplicitCopyCtorExpr * impCpCtorExpr ) override final; |
---|
| 215 | virtual Expression * mutate( ConstructorExpr * ctorExpr ) override final; |
---|
| 216 | virtual Expression * mutate( CompoundLiteralExpr * compLitExpr ) override final; |
---|
| 217 | virtual Expression * mutate( RangeExpr * rangeExpr ) override final; |
---|
| 218 | virtual Expression * mutate( UntypedTupleExpr * tupleExpr ) override final; |
---|
| 219 | virtual Expression * mutate( TupleExpr * tupleExpr ) override final; |
---|
| 220 | virtual Expression * mutate( TupleIndexExpr * tupleExpr ) override final; |
---|
| 221 | virtual Expression * mutate( TupleAssignExpr * assignExpr ) override final; |
---|
| 222 | virtual Expression * mutate( StmtExpr * stmtExpr ) override final; |
---|
| 223 | virtual Expression * mutate( UniqueExpr * uniqueExpr ) override final; |
---|
[73367a8] | 224 | virtual Expression * mutate( UntypedInitExpr * initExpr ) override final; |
---|
| 225 | virtual Expression * mutate( InitExpr * initExpr ) override final; |
---|
[44b4114] | 226 | virtual Expression * mutate( DeletedExpr * delExpr ) override final; |
---|
[0f79853] | 227 | virtual Expression * mutate( DefaultArgExpr * argExpr ) override final; |
---|
[d807ca28] | 228 | virtual Expression * mutate( GenericExpr * genExpr ) override final; |
---|
[5ea7a22] | 229 | |
---|
| 230 | virtual Type * mutate( VoidType * basicType ) override final; |
---|
| 231 | virtual Type * mutate( BasicType * basicType ) override final; |
---|
| 232 | virtual Type * mutate( PointerType * pointerType ) override final; |
---|
| 233 | virtual Type * mutate( ArrayType * arrayType ) override final; |
---|
| 234 | virtual Type * mutate( ReferenceType * referenceType ) override final; |
---|
| 235 | virtual Type * mutate( FunctionType * functionType ) override final; |
---|
| 236 | virtual Type * mutate( StructInstType * aggregateUseType ) override final; |
---|
| 237 | virtual Type * mutate( UnionInstType * aggregateUseType ) override final; |
---|
| 238 | virtual Type * mutate( EnumInstType * aggregateUseType ) override final; |
---|
| 239 | virtual Type * mutate( TraitInstType * aggregateUseType ) override final; |
---|
| 240 | virtual Type * mutate( TypeInstType * aggregateUseType ) override final; |
---|
| 241 | virtual Type * mutate( TupleType * tupleType ) override final; |
---|
| 242 | virtual Type * mutate( TypeofType * typeofType ) override final; |
---|
| 243 | virtual Type * mutate( AttrType * attrType ) override final; |
---|
| 244 | virtual Type * mutate( VarArgsType * varArgsType ) override final; |
---|
| 245 | virtual Type * mutate( ZeroType * zeroType ) override final; |
---|
| 246 | virtual Type * mutate( OneType * oneType ) override final; |
---|
| 247 | |
---|
| 248 | virtual Designation * mutate( Designation * designation ) override final; |
---|
| 249 | virtual Initializer * mutate( SingleInit * singleInit ) override final; |
---|
| 250 | virtual Initializer * mutate( ListInit * listInit ) override final; |
---|
| 251 | virtual Initializer * mutate( ConstructorInit * ctorInit ) override final; |
---|
| 252 | |
---|
| 253 | virtual Subrange * mutate( Subrange * subrange ) override final; |
---|
| 254 | |
---|
| 255 | virtual Constant * mutate( Constant * constant ) override final; |
---|
| 256 | |
---|
| 257 | virtual Attribute * mutate( Attribute * attribute ) override final; |
---|
[447c356] | 258 | |
---|
| 259 | virtual TypeSubstitution * mutate( TypeSubstitution * sub ) final; |
---|
[ab904dc] | 260 | |
---|
[13932f14] | 261 | private: |
---|
[61d9b4b] | 262 | bool inFunction = false; |
---|
| 263 | |
---|
[6ca154b] | 264 | template<typename pass_t> friend void acceptAll( std::list< Declaration* > &decls, PassVisitor< pass_t >& visitor ); |
---|
| 265 | template<typename pass_t> friend void mutateAll( std::list< Declaration* > &decls, PassVisitor< pass_t >& visitor ); |
---|
[3c398b6] | 266 | template< typename TreeType, typename pass_t > friend void maybeAccept_impl( TreeType * tree, PassVisitor< pass_t > & visitor ); |
---|
| 267 | template< typename TreeType, typename pass_t > friend void maybeMutate_impl( TreeType *& tree, PassVisitor< pass_t > & mutator ); |
---|
| 268 | template< typename Container, typename pass_t > friend void maybeAccept_impl( Container & container, PassVisitor< pass_t > & visitor ); |
---|
| 269 | template< typename Container, typename pass_t > friend void maybeMutate_impl( Container & container, PassVisitor< pass_t > & mutator ); |
---|
[6ca154b] | 270 | |
---|
[296b2be] | 271 | template<typename node_type> void call_previsit ( node_type * node ) { previsit_impl ( pass, node, 0 ); } |
---|
| 272 | template<typename node_type> void call_postvisit( node_type * node ) { postvisit_impl( pass, node, 0 ); } |
---|
[13932f14] | 273 | |
---|
[296b2be] | 274 | template<typename node_type> void call_premutate ( node_type * node ) { premutate_impl( pass, node, 0 ); } |
---|
| 275 | template<typename return_type, typename node_type> return_type call_postmutate ( node_type * node ) { return postmutate_impl<return_type>( pass, node, 0 ); } |
---|
[ab904dc] | 276 | |
---|
[296b2be] | 277 | void call_beginScope() { begin_scope_impl( pass, 0 ); } |
---|
| 278 | void call_endScope () { end_scope_impl ( pass, 0 ); } |
---|
| 279 | |
---|
[3fb9a83] | 280 | void set_env( TypeSubstitution * env ) { set_env_impl( pass, env, 0); } |
---|
| 281 | |
---|
[6ca154b] | 282 | template< typename func_t > |
---|
| 283 | void handleStatementList( std::list< Statement * > & statements, func_t func ); |
---|
| 284 | void visitStatementList ( std::list< Statement* > &statements ); |
---|
[296b2be] | 285 | void mutateStatementList( std::list< Statement* > &statements ); |
---|
[134322e] | 286 | |
---|
[6ca154b] | 287 | template< typename func_t > |
---|
| 288 | Statement * handleStatement( Statement * stmt, func_t func ); |
---|
| 289 | Statement * visitStatement ( Statement * stmt ); |
---|
[296b2be] | 290 | Statement * mutateStatement( Statement * stmt ); |
---|
[134322e] | 291 | |
---|
[6ca154b] | 292 | template< typename func_t > |
---|
| 293 | Expression * handleExpression( Expression * expr, func_t func ); |
---|
| 294 | Expression * visitExpression ( Expression * expr ); |
---|
[296b2be] | 295 | Expression * mutateExpression( Expression * expr ); |
---|
| 296 | |
---|
[ab904dc] | 297 | |
---|
[134322e] | 298 | TypeSubstitution ** get_env_ptr () { return env_impl ( pass, 0); } |
---|
| 299 | std::list< Statement* > * get_beforeStmts() { return stmtsToAddBefore_impl( pass, 0); } |
---|
| 300 | std::list< Statement* > * get_afterStmts () { return stmtsToAddAfter_impl ( pass, 0); } |
---|
[35df560] | 301 | std::list< Declaration* > * get_beforeDecls() { return declsToAddBefore_impl( pass, 0); } |
---|
| 302 | std::list< Declaration* > * get_afterDecls () { return declsToAddAfter_impl ( pass, 0); } |
---|
[b73bd70] | 303 | |
---|
[3c398b6] | 304 | bool get_visit_children () { bool_ref * ptr = visit_children_impl(pass, 0); return ptr ? *ptr : true; } |
---|
| 305 | bool_ref * get_visit_children_ptr() { return visit_children_impl(pass, 0); } |
---|
[e0886db] | 306 | |
---|
[4670c79] | 307 | void indexerScopeEnter () { indexer_impl_enterScope ( pass, 0 ); } |
---|
| 308 | void indexerScopeLeave () { indexer_impl_leaveScope ( pass, 0 ); } |
---|
| 309 | void indexerAddId ( DeclarationWithType * node ) { indexer_impl_addId ( pass, 0, node ); } |
---|
| 310 | void indexerAddType ( NamedTypeDecl * node ) { indexer_impl_addType ( pass, 0, node ); } |
---|
| 311 | void indexerAddStruct ( const std::string & id ) { indexer_impl_addStruct ( pass, 0, id ); } |
---|
| 312 | void indexerAddStruct ( StructDecl * node ) { indexer_impl_addStruct ( pass, 0, node ); } |
---|
| 313 | void indexerAddStructFwd( StructDecl * node ) { indexer_impl_addStructFwd( pass, 0, node ); } |
---|
| 314 | void indexerAddEnum ( EnumDecl * node ) { indexer_impl_addEnum ( pass, 0, node ); } |
---|
| 315 | void indexerAddUnion ( const std::string & id ) { indexer_impl_addUnion ( pass, 0, id ); } |
---|
| 316 | void indexerAddUnion ( UnionDecl * node ) { indexer_impl_addUnion ( pass, 0, node ); } |
---|
| 317 | void indexerAddUnionFwd ( UnionDecl * node ) { indexer_impl_addUnionFwd ( pass, 0, node ); } |
---|
| 318 | void indexerAddTrait ( TraitDecl * node ) { indexer_impl_addTrait ( pass, 0, node ); } |
---|
[3ed994e] | 319 | void indexerAddWith ( std::list< Expression * > & exprs, BaseSyntaxNode * withStmt ) { indexer_impl_addWith( pass, 0, exprs, withStmt ); } |
---|
[d8893ca] | 320 | |
---|
[e0886db] | 321 | |
---|
| 322 | template< typename TreeType, typename VisitorType > |
---|
| 323 | friend inline void indexerScopedAccept( TreeType * tree, VisitorType &visitor ); |
---|
| 324 | |
---|
| 325 | template< typename TreeType, typename VisitorType > |
---|
| 326 | friend inline void indexerScopedMutate( TreeType *& tree, VisitorType &visitor ); |
---|
[13932f14] | 327 | }; |
---|
| 328 | |
---|
[6e09f211] | 329 | template<typename pass_type, typename T> |
---|
| 330 | void GuardValue( pass_type * pass, T& val ) { |
---|
| 331 | pass->at_cleanup( [ val ]( void * newVal ) { |
---|
| 332 | * static_cast< T * >( newVal ) = val; |
---|
| 333 | }, static_cast< void * >( & val ) ); |
---|
| 334 | } |
---|
| 335 | |
---|
[637568b] | 336 | class WithTypeSubstitution { |
---|
| 337 | protected: |
---|
| 338 | WithTypeSubstitution() = default; |
---|
| 339 | ~WithTypeSubstitution() = default; |
---|
| 340 | |
---|
[4eb31f2b] | 341 | public: |
---|
[d24d4e1] | 342 | TypeSubstitution * env = nullptr; |
---|
[637568b] | 343 | }; |
---|
| 344 | |
---|
| 345 | class WithStmtsToAdd { |
---|
| 346 | protected: |
---|
| 347 | WithStmtsToAdd() = default; |
---|
| 348 | ~WithStmtsToAdd() = default; |
---|
| 349 | |
---|
[4eb31f2b] | 350 | public: |
---|
[637568b] | 351 | std::list< Statement* > stmtsToAddBefore; |
---|
| 352 | std::list< Statement* > stmtsToAddAfter; |
---|
| 353 | }; |
---|
[d24d4e1] | 354 | |
---|
| 355 | class WithDeclsToAdd { |
---|
| 356 | protected: |
---|
| 357 | WithDeclsToAdd() = default; |
---|
[e0886db] | 358 | ~WithDeclsToAdd() { |
---|
| 359 | assert( declsToAddBefore.empty() ); |
---|
| 360 | } |
---|
[d24d4e1] | 361 | |
---|
| 362 | public: |
---|
| 363 | std::list< Declaration* > declsToAddBefore; |
---|
| 364 | std::list< Declaration* > declsToAddAfter; |
---|
| 365 | }; |
---|
| 366 | |
---|
[637568b] | 367 | class WithShortCircuiting { |
---|
| 368 | protected: |
---|
| 369 | WithShortCircuiting() = default; |
---|
| 370 | ~WithShortCircuiting() = default; |
---|
| 371 | |
---|
[4eb31f2b] | 372 | public: |
---|
[b73bd70] | 373 | bool_ref visit_children; |
---|
[637568b] | 374 | }; |
---|
| 375 | |
---|
[d24d4e1] | 376 | class WithGuards { |
---|
[637568b] | 377 | protected: |
---|
[d24d4e1] | 378 | WithGuards() = default; |
---|
| 379 | ~WithGuards() = default; |
---|
[637568b] | 380 | |
---|
[4eb31f2b] | 381 | public: |
---|
[637568b] | 382 | at_cleanup_t at_cleanup; |
---|
| 383 | |
---|
[4eb31f2b] | 384 | template< typename T > |
---|
[637568b] | 385 | void GuardValue( T& val ) { |
---|
| 386 | at_cleanup( [ val ]( void * newVal ) { |
---|
| 387 | * static_cast< T * >( newVal ) = val; |
---|
| 388 | }, static_cast< void * >( & val ) ); |
---|
| 389 | } |
---|
[5dd0704] | 390 | |
---|
[d24d4e1] | 391 | template< typename T > |
---|
| 392 | void GuardScope( T& val ) { |
---|
| 393 | val.beginScope(); |
---|
| 394 | at_cleanup( []( void * val ) { |
---|
| 395 | static_cast< T * >( val )->endScope(); |
---|
| 396 | }, static_cast< void * >( & val ) ); |
---|
| 397 | } |
---|
| 398 | |
---|
[5dd0704] | 399 | template< typename Func > |
---|
[d24d4e1] | 400 | void GuardAction( Func func ) { |
---|
| 401 | at_cleanup( [func](__attribute__((unused)) void *) { func(); }, nullptr ); |
---|
[5dd0704] | 402 | } |
---|
[637568b] | 403 | }; |
---|
| 404 | |
---|
[b73bd70] | 405 | template<typename pass_type> |
---|
| 406 | class WithVisitorRef { |
---|
| 407 | protected: |
---|
[d24d4e1] | 408 | WithVisitorRef() {} |
---|
| 409 | ~WithVisitorRef() {} |
---|
[b73bd70] | 410 | |
---|
| 411 | public: |
---|
[d24d4e1] | 412 | PassVisitor<pass_type> * const visitor = nullptr; |
---|
[b73bd70] | 413 | }; |
---|
[637568b] | 414 | |
---|
[e0886db] | 415 | class WithIndexer { |
---|
| 416 | protected: |
---|
| 417 | WithIndexer() {} |
---|
| 418 | ~WithIndexer() {} |
---|
| 419 | |
---|
| 420 | public: |
---|
| 421 | SymTab::Indexer indexer; |
---|
| 422 | }; |
---|
| 423 | |
---|
[07c178f0] | 424 | #include "SynTree/TypeSubstitution.h" |
---|
[aff3af4] | 425 | #include "PassVisitor.impl.h" |
---|