| 1 | #pragma once
 | 
|---|
| 2 | 
 | 
|---|
| 3 | // IWYU pragma: private, include "Common/PassVisitor.h"
 | 
|---|
| 4 | 
 | 
|---|
| 5 | #include <stack>
 | 
|---|
| 6 | #include <type_traits>
 | 
|---|
| 7 | 
 | 
|---|
| 8 | #include "Common/Stats.h"
 | 
|---|
| 9 | #include "Common/utility.h"
 | 
|---|
| 10 | 
 | 
|---|
| 11 | #include "SynTree/Mutator.h"
 | 
|---|
| 12 | #include "SynTree/Visitor.h"
 | 
|---|
| 13 | 
 | 
|---|
| 14 | #include "SymTab/Indexer.h"
 | 
|---|
| 15 | 
 | 
|---|
| 16 | #include "SynTree/Attribute.h"
 | 
|---|
| 17 | #include "SynTree/Initializer.h"
 | 
|---|
| 18 | #include "SynTree/Statement.h"
 | 
|---|
| 19 | #include "SynTree/Type.h"
 | 
|---|
| 20 | #include "SynTree/Declaration.h"
 | 
|---|
| 21 | #include "SynTree/Expression.h"
 | 
|---|
| 22 | #include "SynTree/Constant.h"
 | 
|---|
| 23 | 
 | 
|---|
| 24 | class TypeSubstitution;
 | 
|---|
| 25 | 
 | 
|---|
| 26 | #include "PassVisitor.proto.h"
 | 
|---|
| 27 | 
 | 
|---|
| 28 | //-------------------------------------------------------------------------------------------------------------------------------------------------------------------------
 | 
|---|
| 29 | // Templated visitor type
 | 
|---|
| 30 | // To use declare a PassVisitor< YOUR VISITOR TYPE >
 | 
|---|
| 31 | // The visitor type should specify the previsit/postvisit/premutate/postmutate for types that are desired.
 | 
|---|
| 32 | // Note: previsit/postvisit/premutate/postmutate must be **public** members
 | 
|---|
| 33 | //
 | 
|---|
| 34 | // Several additional features are available through inheritance
 | 
|---|
| 35 | // | WithTypeSubstitution - provides polymorphic TypeSubstitution * env for the current expression
 | 
|---|
| 36 | // | WithStmtsToAdd       - provides the ability to insert statements before or after the current statement by adding new statements into
 | 
|---|
| 37 | //                          stmtsToAddBefore or stmtsToAddAfter respectively.
 | 
|---|
| 38 | // | WithShortCircuiting  - provides the ability to skip visiting child nodes; set visit_children to false in pre{visit,mutate} to skip visiting children
 | 
|---|
| 39 | // | WithGuards           - provides the ability to save/restore data like a LIFO stack; to save, call GuardValue with the variable to save, the variable
 | 
|---|
| 40 | //                          will automatically be restored to its previous value after the corresponding postvisit/postmutate teminates.
 | 
|---|
| 41 | //-------------------------------------------------------------------------------------------------------------------------------------------------------------------------
 | 
|---|
| 42 | template< typename pass_type >
 | 
|---|
| 43 | class PassVisitor final : public Visitor, public Mutator {
 | 
|---|
| 44 | public:
 | 
|---|
| 45 | 
 | 
|---|
| 46 |         template< typename... Args >
 | 
|---|
| 47 |         PassVisitor(Args &&... args)
 | 
|---|
| 48 |                 : pass( std::forward<Args>( args )... )
 | 
|---|
| 49 |         {
 | 
|---|
| 50 |                 typedef PassVisitor<pass_type> this_t;
 | 
|---|
| 51 |                 this_t * const * visitor = visitor_impl(pass, 0);
 | 
|---|
| 52 |                 if(visitor) {
 | 
|---|
| 53 |                         *const_cast<this_t **>( visitor ) = this;
 | 
|---|
| 54 |                 }
 | 
|---|
| 55 |         }
 | 
|---|
| 56 | 
 | 
|---|
| 57 |         virtual ~PassVisitor() = default;
 | 
|---|
| 58 | 
 | 
|---|
| 59 |         pass_type pass;
 | 
|---|
| 60 | 
 | 
|---|
| 61 |         virtual void visit( ObjectDecl * objectDecl ) override final;
 | 
|---|
| 62 |         virtual void visit( const ObjectDecl * objectDecl ) override final;
 | 
|---|
| 63 |         virtual void visit( FunctionDecl * functionDecl ) override final;
 | 
|---|
| 64 |         virtual void visit( const FunctionDecl * functionDecl ) override final;
 | 
|---|
| 65 |         virtual void visit( StructDecl * aggregateDecl ) override final;
 | 
|---|
| 66 |         virtual void visit( const StructDecl * aggregateDecl ) override final;
 | 
|---|
| 67 |         virtual void visit( UnionDecl * aggregateDecl ) override final;
 | 
|---|
| 68 |         virtual void visit( const UnionDecl * aggregateDecl ) override final;
 | 
|---|
| 69 |         virtual void visit( EnumDecl * aggregateDecl ) override final;
 | 
|---|
| 70 |         virtual void visit( const EnumDecl * aggregateDecl ) override final;
 | 
|---|
| 71 |         virtual void visit( TraitDecl * aggregateDecl ) override final;
 | 
|---|
| 72 |         virtual void visit( const TraitDecl * aggregateDecl ) override final;
 | 
|---|
| 73 |         virtual void visit( TypeDecl * typeDecl ) override final;
 | 
|---|
| 74 |         virtual void visit( const TypeDecl * typeDecl ) override final;
 | 
|---|
| 75 |         virtual void visit( TypedefDecl * typeDecl ) override final;
 | 
|---|
| 76 |         virtual void visit( const TypedefDecl * typeDecl ) override final;
 | 
|---|
| 77 |         virtual void visit( AsmDecl * asmDecl ) override final;
 | 
|---|
| 78 |         virtual void visit( const AsmDecl * asmDecl ) override final;
 | 
|---|
| 79 |         virtual void visit( DirectiveDecl * directiveDecl ) override final;
 | 
|---|
| 80 |         virtual void visit( const DirectiveDecl * directiveDecl ) override final;
 | 
|---|
| 81 |         virtual void visit( StaticAssertDecl * assertDecl ) override final;
 | 
|---|
| 82 |         virtual void visit( const StaticAssertDecl * assertDecl ) override final;
 | 
|---|
| 83 | 
 | 
|---|
| 84 |         virtual void visit( CompoundStmt * compoundStmt ) override final;
 | 
|---|
| 85 |         virtual void visit( const CompoundStmt * compoundStmt ) override final;
 | 
|---|
| 86 |         virtual void visit( ExprStmt * exprStmt ) override final;
 | 
|---|
| 87 |         virtual void visit( const ExprStmt * exprStmt ) override final;
 | 
|---|
| 88 |         virtual void visit( AsmStmt * asmStmt ) override final;
 | 
|---|
| 89 |         virtual void visit( const AsmStmt * asmStmt ) override final;
 | 
|---|
| 90 |         virtual void visit( DirectiveStmt * dirStmt ) override final;
 | 
|---|
| 91 |         virtual void visit( const DirectiveStmt * dirStmt ) override final;
 | 
|---|
| 92 |         virtual void visit( IfStmt * ifStmt ) override final;
 | 
|---|
| 93 |         virtual void visit( const IfStmt * ifStmt ) override final;
 | 
|---|
| 94 |         virtual void visit( WhileDoStmt * whileDoStmt ) override final;
 | 
|---|
| 95 |         virtual void visit( const WhileDoStmt * whileDoStmt ) override final;
 | 
|---|
| 96 |         virtual void visit( ForStmt * forStmt ) override final;
 | 
|---|
| 97 |         virtual void visit( const ForStmt * forStmt ) override final;
 | 
|---|
| 98 |         virtual void visit( SwitchStmt * switchStmt ) override final;
 | 
|---|
| 99 |         virtual void visit( const SwitchStmt * switchStmt ) override final;
 | 
|---|
| 100 |         virtual void visit( CaseStmt * caseStmt ) override final;
 | 
|---|
| 101 |         virtual void visit( const CaseStmt * caseStmt ) override final;
 | 
|---|
| 102 |         virtual void visit( BranchStmt * branchStmt ) override final;
 | 
|---|
| 103 |         virtual void visit( const BranchStmt * branchStmt ) override final;
 | 
|---|
| 104 |         virtual void visit( ReturnStmt * returnStmt ) override final;
 | 
|---|
| 105 |         virtual void visit( const ReturnStmt * returnStmt ) override final;
 | 
|---|
| 106 |         virtual void visit( ThrowStmt * throwStmt ) override final;
 | 
|---|
| 107 |         virtual void visit( const ThrowStmt * throwStmt ) override final;
 | 
|---|
| 108 |         virtual void visit( TryStmt * tryStmt ) override final;
 | 
|---|
| 109 |         virtual void visit( const TryStmt * tryStmt ) override final;
 | 
|---|
| 110 |         virtual void visit( CatchStmt * catchStmt ) override final;
 | 
|---|
| 111 |         virtual void visit( const CatchStmt * catchStmt ) override final;
 | 
|---|
| 112 |         virtual void visit( FinallyStmt * finallyStmt ) override final;
 | 
|---|
| 113 |         virtual void visit( const FinallyStmt * finallyStmt ) override final;
 | 
|---|
| 114 |         virtual void visit( SuspendStmt * suspendStmt ) override final;
 | 
|---|
| 115 |         virtual void visit( const SuspendStmt * suspendStmt ) override final;
 | 
|---|
| 116 |         virtual void visit( WaitForStmt * waitforStmt ) override final;
 | 
|---|
| 117 |         virtual void visit( const WaitForStmt * waitforStmt ) override final;
 | 
|---|
| 118 |         virtual void visit( WithStmt * withStmt ) override final;
 | 
|---|
| 119 |         virtual void visit( const WithStmt * withStmt ) override final;
 | 
|---|
| 120 |         virtual void visit( NullStmt * nullStmt ) override final;
 | 
|---|
| 121 |         virtual void visit( const NullStmt * nullStmt ) override final;
 | 
|---|
| 122 |         virtual void visit( DeclStmt * declStmt ) override final;
 | 
|---|
| 123 |         virtual void visit( const DeclStmt * declStmt ) override final;
 | 
|---|
| 124 |         virtual void visit( ImplicitCtorDtorStmt * impCtorDtorStmt ) override final;
 | 
|---|
| 125 |         virtual void visit( const ImplicitCtorDtorStmt * impCtorDtorStmt ) override final;
 | 
|---|
| 126 |         virtual void visit( MutexStmt * mutexStmt ) override final;
 | 
|---|
| 127 |         virtual void visit( const MutexStmt * mutexStmt ) override final;
 | 
|---|
| 128 | 
 | 
|---|
| 129 |         virtual void visit( ApplicationExpr * applicationExpr ) override final;
 | 
|---|
| 130 |         virtual void visit( const ApplicationExpr * applicationExpr ) override final;
 | 
|---|
| 131 |         virtual void visit( UntypedExpr * untypedExpr ) override final;
 | 
|---|
| 132 |         virtual void visit( const UntypedExpr * untypedExpr ) override final;
 | 
|---|
| 133 |         virtual void visit( NameExpr * nameExpr ) override final;
 | 
|---|
| 134 |         virtual void visit( const NameExpr * nameExpr ) override final;
 | 
|---|
| 135 |         virtual void visit ( QualifiedNameExpr * qualifiedNameExpr ) override final;
 | 
|---|
| 136 |         virtual void visit ( const QualifiedNameExpr * qualifiedNameExpr ) override final;
 | 
|---|
| 137 |         virtual void visit( CastExpr * castExpr ) override final;
 | 
|---|
| 138 |         virtual void visit( const CastExpr * castExpr ) override final;
 | 
|---|
| 139 |         virtual void visit( KeywordCastExpr * castExpr ) override final;
 | 
|---|
| 140 |         virtual void visit( const KeywordCastExpr * castExpr ) override final;
 | 
|---|
| 141 |         virtual void visit( VirtualCastExpr * castExpr ) override final;
 | 
|---|
| 142 |         virtual void visit( const VirtualCastExpr * castExpr ) override final;
 | 
|---|
| 143 |         virtual void visit( AddressExpr * addressExpr ) override final;
 | 
|---|
| 144 |         virtual void visit( const AddressExpr * addressExpr ) override final;
 | 
|---|
| 145 |         virtual void visit( LabelAddressExpr * labAddressExpr ) override final;
 | 
|---|
| 146 |         virtual void visit( const LabelAddressExpr * labAddressExpr ) override final;
 | 
|---|
| 147 |         virtual void visit( UntypedMemberExpr * memberExpr ) override final;
 | 
|---|
| 148 |         virtual void visit( const UntypedMemberExpr * memberExpr ) override final;
 | 
|---|
| 149 |         virtual void visit( MemberExpr * memberExpr ) override final;
 | 
|---|
| 150 |         virtual void visit( const MemberExpr * memberExpr ) override final;
 | 
|---|
| 151 |         virtual void visit( VariableExpr * variableExpr ) override final;
 | 
|---|
| 152 |         virtual void visit( const VariableExpr * variableExpr ) override final;
 | 
|---|
| 153 |         virtual void visit( ConstantExpr * constantExpr ) override final;
 | 
|---|
| 154 |         virtual void visit( const ConstantExpr * constantExpr ) override final;
 | 
|---|
| 155 |         virtual void visit( SizeofExpr * sizeofExpr ) override final;
 | 
|---|
| 156 |         virtual void visit( const SizeofExpr * sizeofExpr ) override final;
 | 
|---|
| 157 |         virtual void visit( AlignofExpr * alignofExpr ) override final;
 | 
|---|
| 158 |         virtual void visit( const AlignofExpr * alignofExpr ) override final;
 | 
|---|
| 159 |         virtual void visit( UntypedOffsetofExpr * offsetofExpr ) override final;
 | 
|---|
| 160 |         virtual void visit( const UntypedOffsetofExpr * offsetofExpr ) override final;
 | 
|---|
| 161 |         virtual void visit( OffsetofExpr * offsetofExpr ) override final;
 | 
|---|
| 162 |         virtual void visit( const OffsetofExpr * offsetofExpr ) override final;
 | 
|---|
| 163 |         virtual void visit( OffsetPackExpr * offsetPackExpr ) override final;
 | 
|---|
| 164 |         virtual void visit( const OffsetPackExpr * offsetPackExpr ) override final;
 | 
|---|
| 165 |         virtual void visit( LogicalExpr * logicalExpr ) override final;
 | 
|---|
| 166 |         virtual void visit( const LogicalExpr * logicalExpr ) override final;
 | 
|---|
| 167 |         virtual void visit( ConditionalExpr * conditionalExpr ) override final;
 | 
|---|
| 168 |         virtual void visit( const ConditionalExpr * conditionalExpr ) override final;
 | 
|---|
| 169 |         virtual void visit( CommaExpr * commaExpr ) override final;
 | 
|---|
| 170 |         virtual void visit( const CommaExpr * commaExpr ) override final;
 | 
|---|
| 171 |         virtual void visit( TypeExpr * typeExpr ) override final;
 | 
|---|
| 172 |         virtual void visit( const TypeExpr * typeExpr ) override final;
 | 
|---|
| 173 |         virtual void visit( DimensionExpr * dimensionExpr ) override final;
 | 
|---|
| 174 |         virtual void visit( const DimensionExpr * dimensionExpr ) override final;
 | 
|---|
| 175 |         virtual void visit( AsmExpr * asmExpr ) override final;
 | 
|---|
| 176 |         virtual void visit( const AsmExpr * asmExpr ) override final;
 | 
|---|
| 177 |         virtual void visit( ImplicitCopyCtorExpr * impCpCtorExpr ) override final;
 | 
|---|
| 178 |         virtual void visit( const ImplicitCopyCtorExpr * impCpCtorExpr ) override final;
 | 
|---|
| 179 |         virtual void visit( ConstructorExpr *  ctorExpr ) override final;
 | 
|---|
| 180 |         virtual void visit( const ConstructorExpr *  ctorExpr ) override final;
 | 
|---|
| 181 |         virtual void visit( CompoundLiteralExpr * compLitExpr ) override final;
 | 
|---|
| 182 |         virtual void visit( const CompoundLiteralExpr * compLitExpr ) override final;
 | 
|---|
| 183 |         virtual void visit( RangeExpr * rangeExpr ) override final;
 | 
|---|
| 184 |         virtual void visit( const RangeExpr * rangeExpr ) override final;
 | 
|---|
| 185 |         virtual void visit( UntypedTupleExpr * tupleExpr ) override final;
 | 
|---|
| 186 |         virtual void visit( const UntypedTupleExpr * tupleExpr ) override final;
 | 
|---|
| 187 |         virtual void visit( TupleExpr * tupleExpr ) override final;
 | 
|---|
| 188 |         virtual void visit( const TupleExpr * tupleExpr ) override final;
 | 
|---|
| 189 |         virtual void visit( TupleIndexExpr * tupleExpr ) override final;
 | 
|---|
| 190 |         virtual void visit( const TupleIndexExpr * tupleExpr ) override final;
 | 
|---|
| 191 |         virtual void visit( TupleAssignExpr * assignExpr ) override final;
 | 
|---|
| 192 |         virtual void visit( const TupleAssignExpr * assignExpr ) override final;
 | 
|---|
| 193 |         virtual void visit( StmtExpr *  stmtExpr ) override final;
 | 
|---|
| 194 |         virtual void visit( const StmtExpr *  stmtExpr ) override final;
 | 
|---|
| 195 |         virtual void visit( UniqueExpr *  uniqueExpr ) override final;
 | 
|---|
| 196 |         virtual void visit( const UniqueExpr *  uniqueExpr ) override final;
 | 
|---|
| 197 |         virtual void visit( UntypedInitExpr *  initExpr ) override final;
 | 
|---|
| 198 |         virtual void visit( const UntypedInitExpr *  initExpr ) override final;
 | 
|---|
| 199 |         virtual void visit( InitExpr *  initExpr ) override final;
 | 
|---|
| 200 |         virtual void visit( const InitExpr *  initExpr ) override final;
 | 
|---|
| 201 |         virtual void visit( DeletedExpr *  delExpr ) override final;
 | 
|---|
| 202 |         virtual void visit( const DeletedExpr *  delExpr ) override final;
 | 
|---|
| 203 |         virtual void visit( DefaultArgExpr * argExpr ) override final;
 | 
|---|
| 204 |         virtual void visit( const DefaultArgExpr * argExpr ) override final;
 | 
|---|
| 205 |         virtual void visit( GenericExpr * genExpr ) override final;
 | 
|---|
| 206 |         virtual void visit( const GenericExpr * genExpr ) override final;
 | 
|---|
| 207 | 
 | 
|---|
| 208 |         virtual void visit( VoidType * basicType ) override final;
 | 
|---|
| 209 |         virtual void visit( const VoidType * basicType ) override final;
 | 
|---|
| 210 |         virtual void visit( BasicType * basicType ) override final;
 | 
|---|
| 211 |         virtual void visit( const BasicType * basicType ) override final;
 | 
|---|
| 212 |         virtual void visit( PointerType * pointerType ) override final;
 | 
|---|
| 213 |         virtual void visit( const PointerType * pointerType ) override final;
 | 
|---|
| 214 |         virtual void visit( ArrayType * arrayType ) override final;
 | 
|---|
| 215 |         virtual void visit( const ArrayType * arrayType ) override final;
 | 
|---|
| 216 |         virtual void visit( ReferenceType * referenceType ) override final;
 | 
|---|
| 217 |         virtual void visit( const ReferenceType * referenceType ) override final;
 | 
|---|
| 218 |         virtual void visit( QualifiedType * qualType ) override final;
 | 
|---|
| 219 |         virtual void visit( const QualifiedType * qualType ) override final;
 | 
|---|
| 220 |         virtual void visit( FunctionType * functionType ) override final;
 | 
|---|
| 221 |         virtual void visit( const FunctionType * functionType ) override final;
 | 
|---|
| 222 |         virtual void visit( StructInstType * aggregateUseType ) override final;
 | 
|---|
| 223 |         virtual void visit( const StructInstType * aggregateUseType ) override final;
 | 
|---|
| 224 |         virtual void visit( UnionInstType * aggregateUseType ) override final;
 | 
|---|
| 225 |         virtual void visit( const UnionInstType * aggregateUseType ) override final;
 | 
|---|
| 226 |         virtual void visit( EnumInstType * aggregateUseType ) override final;
 | 
|---|
| 227 |         virtual void visit( const EnumInstType * aggregateUseType ) override final;
 | 
|---|
| 228 |         virtual void visit( TraitInstType * aggregateUseType ) override final;
 | 
|---|
| 229 |         virtual void visit( const TraitInstType * aggregateUseType ) override final;
 | 
|---|
| 230 |         virtual void visit( TypeInstType * aggregateUseType ) override final;
 | 
|---|
| 231 |         virtual void visit( const TypeInstType * aggregateUseType ) override final;
 | 
|---|
| 232 |         virtual void visit( TupleType * tupleType ) override final;
 | 
|---|
| 233 |         virtual void visit( const TupleType * tupleType ) override final;
 | 
|---|
| 234 |         virtual void visit( TypeofType * typeofType ) override final;
 | 
|---|
| 235 |         virtual void visit( const TypeofType * typeofType ) override final;
 | 
|---|
| 236 |         virtual void visit( VTableType * vtableType ) override final;
 | 
|---|
| 237 |         virtual void visit( const VTableType * vtableType ) override final;
 | 
|---|
| 238 |         virtual void visit( AttrType * attrType ) override final;
 | 
|---|
| 239 |         virtual void visit( const AttrType * attrType ) override final;
 | 
|---|
| 240 |         virtual void visit( VarArgsType * varArgsType ) override final;
 | 
|---|
| 241 |         virtual void visit( const VarArgsType * varArgsType ) override final;
 | 
|---|
| 242 |         virtual void visit( ZeroType * zeroType ) override final;
 | 
|---|
| 243 |         virtual void visit( const ZeroType * zeroType ) override final;
 | 
|---|
| 244 |         virtual void visit( OneType * oneType ) override final;
 | 
|---|
| 245 |         virtual void visit( const OneType * oneType ) override final;
 | 
|---|
| 246 |         virtual void visit( GlobalScopeType * globalType ) override final;
 | 
|---|
| 247 |         virtual void visit( const GlobalScopeType * globalType ) override final;
 | 
|---|
| 248 | 
 | 
|---|
| 249 |         virtual void visit( Designation * designation ) override final;
 | 
|---|
| 250 |         virtual void visit( const Designation * designation ) override final;
 | 
|---|
| 251 |         virtual void visit( SingleInit * singleInit ) override final;
 | 
|---|
| 252 |         virtual void visit( const SingleInit * singleInit ) override final;
 | 
|---|
| 253 |         virtual void visit( ListInit * listInit ) override final;
 | 
|---|
| 254 |         virtual void visit( const ListInit * listInit ) override final;
 | 
|---|
| 255 |         virtual void visit( ConstructorInit * ctorInit ) override final;
 | 
|---|
| 256 |         virtual void visit( const ConstructorInit * ctorInit ) override final;
 | 
|---|
| 257 | 
 | 
|---|
| 258 |         virtual void visit( Constant * constant ) override final;
 | 
|---|
| 259 |         virtual void visit( const Constant * constant ) override final;
 | 
|---|
| 260 | 
 | 
|---|
| 261 |         virtual void visit( Attribute * attribute ) override final;
 | 
|---|
| 262 |         virtual void visit( const Attribute * attribute ) override final;
 | 
|---|
| 263 | 
 | 
|---|
| 264 |         virtual DeclarationWithType * mutate( ObjectDecl * objectDecl ) override final;
 | 
|---|
| 265 |         virtual DeclarationWithType * mutate( FunctionDecl * functionDecl ) override final;
 | 
|---|
| 266 |         virtual Declaration * mutate( StructDecl * aggregateDecl ) override final;
 | 
|---|
| 267 |         virtual Declaration * mutate( UnionDecl * aggregateDecl ) override final;
 | 
|---|
| 268 |         virtual Declaration * mutate( EnumDecl * aggregateDecl ) override final;
 | 
|---|
| 269 |         virtual Declaration * mutate( TraitDecl * aggregateDecl ) override final;
 | 
|---|
| 270 |         virtual Declaration * mutate( TypeDecl * typeDecl ) override final;
 | 
|---|
| 271 |         virtual Declaration * mutate( TypedefDecl * typeDecl ) override final;
 | 
|---|
| 272 |         virtual AsmDecl * mutate( AsmDecl * asmDecl ) override final;
 | 
|---|
| 273 |         virtual DirectiveDecl * mutate( DirectiveDecl * directiveDecl ) override final;
 | 
|---|
| 274 |         virtual StaticAssertDecl * mutate( StaticAssertDecl * assertDecl ) override final;
 | 
|---|
| 275 | 
 | 
|---|
| 276 |         virtual CompoundStmt * mutate( CompoundStmt * compoundStmt ) override final;
 | 
|---|
| 277 |         virtual Statement * mutate( ExprStmt * exprStmt ) override final;
 | 
|---|
| 278 |         virtual Statement * mutate( AsmStmt * asmStmt ) override final;
 | 
|---|
| 279 |         virtual Statement * mutate( DirectiveStmt * dirStmt ) override final;
 | 
|---|
| 280 |         virtual Statement * mutate( IfStmt * ifStmt ) override final;
 | 
|---|
| 281 |         virtual Statement * mutate( WhileDoStmt * whileDoStmt ) override final;
 | 
|---|
| 282 |         virtual Statement * mutate( ForStmt * forStmt ) override final;
 | 
|---|
| 283 |         virtual Statement * mutate( SwitchStmt * switchStmt ) override final;
 | 
|---|
| 284 |         virtual Statement * mutate( CaseStmt * caseStmt ) override final;
 | 
|---|
| 285 |         virtual Statement * mutate( BranchStmt * branchStmt ) override final;
 | 
|---|
| 286 |         virtual Statement * mutate( ReturnStmt * returnStmt ) override final;
 | 
|---|
| 287 |         virtual Statement * mutate( ThrowStmt * throwStmt ) override final;
 | 
|---|
| 288 |         virtual Statement * mutate( TryStmt * tryStmt ) override final;
 | 
|---|
| 289 |         virtual Statement * mutate( CatchStmt * catchStmt ) override final;
 | 
|---|
| 290 |         virtual Statement * mutate( FinallyStmt * finallyStmt ) override final;
 | 
|---|
| 291 |         virtual Statement * mutate( SuspendStmt * suspendStmt ) override final;
 | 
|---|
| 292 |         virtual Statement * mutate( WaitForStmt * waitforStmt ) override final;
 | 
|---|
| 293 |         virtual Declaration * mutate( WithStmt * withStmt ) override final;
 | 
|---|
| 294 |         virtual NullStmt * mutate( NullStmt * nullStmt ) override final;
 | 
|---|
| 295 |         virtual Statement * mutate( DeclStmt * declStmt ) override final;
 | 
|---|
| 296 |         virtual Statement * mutate( ImplicitCtorDtorStmt * impCtorDtorStmt ) override final;
 | 
|---|
| 297 |         virtual Statement * mutate( MutexStmt * mutexStmt ) override final;
 | 
|---|
| 298 | 
 | 
|---|
| 299 |         virtual Expression * mutate( ApplicationExpr * applicationExpr ) override final;
 | 
|---|
| 300 |         virtual Expression * mutate( UntypedExpr * untypedExpr ) override final;
 | 
|---|
| 301 |         virtual Expression * mutate( NameExpr * nameExpr ) override final;
 | 
|---|
| 302 |         virtual Expression * mutate( AddressExpr * addrExpr ) override final;
 | 
|---|
| 303 |         virtual Expression * mutate( LabelAddressExpr * labAddressExpr ) override final;
 | 
|---|
| 304 |         virtual Expression * mutate( CastExpr * castExpr ) override final;
 | 
|---|
| 305 |         virtual Expression * mutate( KeywordCastExpr * castExpr ) override final;
 | 
|---|
| 306 |         virtual Expression * mutate( VirtualCastExpr * castExpr ) override final;
 | 
|---|
| 307 |         virtual Expression * mutate( UntypedMemberExpr * memberExpr ) override final;
 | 
|---|
| 308 |         virtual Expression * mutate( MemberExpr * memberExpr ) override final;
 | 
|---|
| 309 |         virtual Expression * mutate( VariableExpr * variableExpr ) override final;
 | 
|---|
| 310 |         virtual Expression * mutate( ConstantExpr * constantExpr ) override final;
 | 
|---|
| 311 |         virtual Expression * mutate( SizeofExpr * sizeofExpr ) override final;
 | 
|---|
| 312 |         virtual Expression * mutate( AlignofExpr * alignofExpr ) override final;
 | 
|---|
| 313 |         virtual Expression * mutate( UntypedOffsetofExpr * offsetofExpr ) override final;
 | 
|---|
| 314 |         virtual Expression * mutate( OffsetofExpr * offsetofExpr ) override final;
 | 
|---|
| 315 |         virtual Expression * mutate( OffsetPackExpr * offsetPackExpr ) override final;
 | 
|---|
| 316 |         virtual Expression * mutate( LogicalExpr * logicalExpr ) override final;
 | 
|---|
| 317 |         virtual Expression * mutate( ConditionalExpr * conditionalExpr ) override final;
 | 
|---|
| 318 |         virtual Expression * mutate( CommaExpr * commaExpr ) override final;
 | 
|---|
| 319 |         virtual Expression * mutate( TypeExpr * typeExpr ) override final;
 | 
|---|
| 320 |         virtual Expression * mutate( DimensionExpr * dimensionExpr ) override final;
 | 
|---|
| 321 |         virtual Expression * mutate( AsmExpr * asmExpr ) override final;
 | 
|---|
| 322 |         virtual Expression * mutate( ImplicitCopyCtorExpr * impCpCtorExpr ) override final;
 | 
|---|
| 323 |         virtual Expression * mutate( ConstructorExpr * ctorExpr ) override final;
 | 
|---|
| 324 |         virtual Expression * mutate( CompoundLiteralExpr * compLitExpr ) override final;
 | 
|---|
| 325 |         virtual Expression * mutate( RangeExpr * rangeExpr ) override final;
 | 
|---|
| 326 |         virtual Expression * mutate( UntypedTupleExpr * tupleExpr ) override final;
 | 
|---|
| 327 |         virtual Expression * mutate( TupleExpr * tupleExpr ) override final;
 | 
|---|
| 328 |         virtual Expression * mutate( TupleIndexExpr * tupleExpr ) override final;
 | 
|---|
| 329 |         virtual Expression * mutate( TupleAssignExpr * assignExpr ) override final; 
 | 
|---|
| 330 |         virtual Expression * mutate( StmtExpr *  stmtExpr ) override final;
 | 
|---|
| 331 |         virtual Expression * mutate( UniqueExpr *  uniqueExpr ) override final;
 | 
|---|
| 332 |         virtual Expression * mutate( UntypedInitExpr *  initExpr ) override final;
 | 
|---|
| 333 |         virtual Expression * mutate( InitExpr *  initExpr ) override final;
 | 
|---|
| 334 |         virtual Expression * mutate( DeletedExpr *  delExpr ) override final;
 | 
|---|
| 335 |         virtual Expression * mutate( DefaultArgExpr * argExpr ) override final;
 | 
|---|
| 336 |         virtual Expression * mutate( GenericExpr * genExpr ) override final;
 | 
|---|
| 337 |         virtual Expression * mutate( QualifiedNameExpr * qualifiedNameExpr ) override final;
 | 
|---|
| 338 | 
 | 
|---|
| 339 |         virtual Type * mutate( VoidType * basicType ) override final;
 | 
|---|
| 340 |         virtual Type * mutate( BasicType * basicType ) override final;
 | 
|---|
| 341 |         virtual Type * mutate( PointerType * pointerType ) override final;
 | 
|---|
| 342 |         virtual Type * mutate( ArrayType * arrayType ) override final;
 | 
|---|
| 343 |         virtual Type * mutate( ReferenceType * referenceType ) override final;
 | 
|---|
| 344 |         virtual Type * mutate( QualifiedType * qualType ) override final;
 | 
|---|
| 345 |         virtual Type * mutate( FunctionType * functionType ) override final;
 | 
|---|
| 346 |         virtual Type * mutate( StructInstType * aggregateUseType ) override final;
 | 
|---|
| 347 |         virtual Type * mutate( UnionInstType * aggregateUseType ) override final;
 | 
|---|
| 348 |         virtual Type * mutate( EnumInstType * aggregateUseType ) override final;
 | 
|---|
| 349 |         virtual Type * mutate( TraitInstType * aggregateUseType ) override final;
 | 
|---|
| 350 |         virtual Type * mutate( TypeInstType * aggregateUseType ) override final;
 | 
|---|
| 351 |         virtual Type * mutate( TupleType * tupleType ) override final;
 | 
|---|
| 352 |         virtual Type * mutate( TypeofType * typeofType ) override final;
 | 
|---|
| 353 |         virtual Type * mutate( VTableType * vtableType ) override final;
 | 
|---|
| 354 |         virtual Type * mutate( AttrType * attrType ) override final;
 | 
|---|
| 355 |         virtual Type * mutate( VarArgsType * varArgsType ) override final;
 | 
|---|
| 356 |         virtual Type * mutate( ZeroType * zeroType ) override final;
 | 
|---|
| 357 |         virtual Type * mutate( OneType * oneType ) override final;
 | 
|---|
| 358 |         virtual Type * mutate( GlobalScopeType * globalType ) override final;
 | 
|---|
| 359 | 
 | 
|---|
| 360 |         virtual Designation * mutate( Designation * designation ) override final;
 | 
|---|
| 361 |         virtual Initializer * mutate( SingleInit * singleInit ) override final;
 | 
|---|
| 362 |         virtual Initializer * mutate( ListInit * listInit ) override final;
 | 
|---|
| 363 |         virtual Initializer * mutate( ConstructorInit * ctorInit ) override final;
 | 
|---|
| 364 | 
 | 
|---|
| 365 |         virtual Constant * mutate( Constant * constant ) override final;
 | 
|---|
| 366 | 
 | 
|---|
| 367 |         virtual Attribute * mutate( Attribute * attribute ) override final;
 | 
|---|
| 368 | 
 | 
|---|
| 369 |         virtual TypeSubstitution * mutate( TypeSubstitution * sub ) final;
 | 
|---|
| 370 | 
 | 
|---|
| 371 |         bool isInFunction() const {
 | 
|---|
| 372 |                 return inFunction;
 | 
|---|
| 373 |         }
 | 
|---|
| 374 | 
 | 
|---|
| 375 | private:
 | 
|---|
| 376 |         bool inFunction = false;
 | 
|---|
| 377 |         bool atFunctionTop = false;
 | 
|---|
| 378 | 
 | 
|---|
| 379 |         template<typename pass_t> friend void acceptAll( std::list< Declaration* > &decls, PassVisitor< pass_t >& visitor );
 | 
|---|
| 380 |         template<typename pass_t> friend void acceptAll( const std::list< const Declaration * > &decls, PassVisitor< pass_t >& visitor );
 | 
|---|
| 381 |         template<typename pass_t> friend void mutateAll( std::list< Declaration* > &decls, PassVisitor< pass_t >& visitor );
 | 
|---|
| 382 |         template< typename TreeType, typename pass_t > friend void maybeAccept_impl( TreeType * tree, PassVisitor< pass_t > & visitor );
 | 
|---|
| 383 |         template< typename TreeType, typename pass_t > friend void maybeAccept_impl( const TreeType * tree, PassVisitor< pass_t > & visitor );
 | 
|---|
| 384 |         template< typename TreeType, typename pass_t > friend void maybeMutate_impl( TreeType *& tree, PassVisitor< pass_t > & mutator );
 | 
|---|
| 385 |         template< typename Container, typename pass_t > friend void maybeAccept_impl( Container & container, PassVisitor< pass_t > & visitor );
 | 
|---|
| 386 |         template< typename Container, typename pass_t > friend void maybeAccept_impl( const Container & container, PassVisitor< pass_t > & visitor );
 | 
|---|
| 387 |         template< typename Container, typename pass_t > friend void maybeMutate_impl( Container & container, PassVisitor< pass_t > & mutator );
 | 
|---|
| 388 | 
 | 
|---|
| 389 |         template<typename node_type> void call_previsit ( node_type * node ) { previsit_impl ( pass, node, 0 ); }
 | 
|---|
| 390 |         template<typename node_type> void call_previsit ( const node_type * node ) { previsit_impl ( pass, node, 0 ); }
 | 
|---|
| 391 |         template<typename node_type> void call_postvisit( node_type * node ) { postvisit_impl( pass, node, 0 ); }
 | 
|---|
| 392 |         template<typename node_type> void call_postvisit( const node_type * node ) { postvisit_impl( pass, node, 0 ); }
 | 
|---|
| 393 | 
 | 
|---|
| 394 |         template<typename node_type> void call_premutate ( node_type * node ) { premutate_impl( pass, node, 0 ); }
 | 
|---|
| 395 |         template<typename return_type, typename node_type> return_type call_postmutate ( node_type * node ) { return postmutate_impl<return_type>( pass, node, 0 ); }
 | 
|---|
| 396 | 
 | 
|---|
| 397 |         void call_beginScope() { begin_scope_impl( pass, 0 ); }
 | 
|---|
| 398 |         void call_endScope  () { end_scope_impl  ( pass, 0 ); }
 | 
|---|
| 399 | 
 | 
|---|
| 400 |         void set_env( TypeSubstitution * env ) { set_env_impl( pass, env, 0); }
 | 
|---|
| 401 | 
 | 
|---|
| 402 |         template< typename func_t >
 | 
|---|
| 403 |         void handleStatementList( std::list< Statement * > & statements, func_t func );
 | 
|---|
| 404 |         void visitStatementList ( std::list< Statement* > &statements );
 | 
|---|
| 405 |         void mutateStatementList( std::list< Statement* > &statements );
 | 
|---|
| 406 |         void visitStatementList ( const std::list< Statement * > & statements );
 | 
|---|
| 407 | 
 | 
|---|
| 408 |         template< typename func_t >
 | 
|---|
| 409 |         Statement * handleStatement( Statement * stmt, func_t func );
 | 
|---|
| 410 |         Statement * visitStatement ( Statement * stmt );
 | 
|---|
| 411 |         Statement * mutateStatement( Statement * stmt );
 | 
|---|
| 412 |         void visitStatement ( const Statement * stmt );
 | 
|---|
| 413 | 
 | 
|---|
| 414 |         template< typename func_t >
 | 
|---|
| 415 |         Expression * handleExpression( Expression * expr, func_t func );
 | 
|---|
| 416 |         Expression * visitExpression ( Expression * expr );
 | 
|---|
| 417 |         Expression * mutateExpression( Expression * expr );
 | 
|---|
| 418 |         void visitExpression ( const Expression * expr );
 | 
|---|
| 419 | 
 | 
|---|
| 420 | 
 | 
|---|
| 421 |         auto                                    get_env_ptr    () -> decltype(env_impl( pass, 0)) { return env_impl( pass, 0); }
 | 
|---|
| 422 |         std::list< Statement* > *       get_beforeStmts() { return stmtsToAddBefore_impl( pass, 0); }
 | 
|---|
| 423 |         std::list< Statement* > *       get_afterStmts () { return stmtsToAddAfter_impl ( pass, 0); }
 | 
|---|
| 424 |         std::list< Declaration* > *     get_beforeDecls() { return declsToAddBefore_impl( pass, 0); }
 | 
|---|
| 425 |         std::list< Declaration* > *     get_afterDecls () { return declsToAddAfter_impl ( pass, 0); }
 | 
|---|
| 426 | 
 | 
|---|
| 427 |         bool       get_visit_children    () { bool_ref * ptr = visit_children_impl(pass, 0); return ptr ? *ptr : true; }
 | 
|---|
| 428 |         bool_ref * get_visit_children_ptr() { return visit_children_impl(pass, 0); }
 | 
|---|
| 429 | 
 | 
|---|
| 430 |         void indexerScopeEnter  ()                                    { indexer_impl_enterScope  ( pass, 0       ); }
 | 
|---|
| 431 |         void indexerScopeLeave  ()                                    { indexer_impl_leaveScope  ( pass, 0       ); }
 | 
|---|
| 432 |         void indexerAddId       ( const DeclarationWithType * node  ) { indexer_impl_addId       ( pass, 0, node ); }
 | 
|---|
| 433 |         void indexerAddType     ( const NamedTypeDecl       * node  ) { indexer_impl_addType     ( pass, 0, node ); }
 | 
|---|
| 434 |         void indexerAddStruct   ( const std::string         & id    ) { indexer_impl_addStruct   ( pass, 0, id   ); }
 | 
|---|
| 435 |         void indexerAddStruct   ( const StructDecl          * node  ) { indexer_impl_addStruct   ( pass, 0, node ); }
 | 
|---|
| 436 |         void indexerAddStructFwd( const StructDecl          * node  ) { indexer_impl_addStructFwd( pass, 0, node ); }
 | 
|---|
| 437 |         void indexerAddEnum     ( const EnumDecl            * node  ) { indexer_impl_addEnum     ( pass, 0, node ); }
 | 
|---|
| 438 |         void indexerAddUnion    ( const std::string         & id    ) { indexer_impl_addUnion    ( pass, 0, id   ); }
 | 
|---|
| 439 |         void indexerAddUnion    ( const UnionDecl           * node  ) { indexer_impl_addUnion    ( pass, 0, node ); }
 | 
|---|
| 440 |         void indexerAddUnionFwd ( const UnionDecl           * node  ) { indexer_impl_addUnionFwd ( pass, 0, node ); }
 | 
|---|
| 441 |         void indexerAddTrait    ( const TraitDecl           * node  ) { indexer_impl_addTrait    ( pass, 0, node ); }
 | 
|---|
| 442 |         void indexerAddWith     ( const std::list< Expression * > & exprs, const Declaration * withStmt ) { indexer_impl_addWith( pass, 0, exprs, withStmt ); }
 | 
|---|
| 443 | 
 | 
|---|
| 444 | 
 | 
|---|
| 445 |         template< typename TreeType, typename VisitorType >
 | 
|---|
| 446 |         friend inline void indexerScopedAccept( TreeType * tree, VisitorType & visitor );
 | 
|---|
| 447 | 
 | 
|---|
| 448 |         template< typename TreeType, typename VisitorType >
 | 
|---|
| 449 |         friend inline void indexerScopedAccept( const TreeType * tree, VisitorType & visitor );
 | 
|---|
| 450 | 
 | 
|---|
| 451 |         template< typename TreeType, typename VisitorType >
 | 
|---|
| 452 |         friend inline void indexerScopedMutate( TreeType *& tree, VisitorType & visitor );
 | 
|---|
| 453 | };
 | 
|---|
| 454 | 
 | 
|---|
| 455 | template<typename pass_type, typename T>
 | 
|---|
| 456 | void GuardValue( pass_type * pass, T& val ) {
 | 
|---|
| 457 |         pass->at_cleanup( [ val ]( void * newVal ) {
 | 
|---|
| 458 |                 * static_cast< T * >( newVal ) = val;
 | 
|---|
| 459 |         }, static_cast< void * >( & val ) );
 | 
|---|
| 460 | }
 | 
|---|
| 461 | 
 | 
|---|
| 462 | class WithTypeSubstitution {
 | 
|---|
| 463 | protected:
 | 
|---|
| 464 |         WithTypeSubstitution() = default;
 | 
|---|
| 465 |         ~WithTypeSubstitution() = default;
 | 
|---|
| 466 | 
 | 
|---|
| 467 | public:
 | 
|---|
| 468 |         TypeSubstitution * env = nullptr;
 | 
|---|
| 469 | };
 | 
|---|
| 470 | 
 | 
|---|
| 471 | class WithConstTypeSubstitution {
 | 
|---|
| 472 | protected:
 | 
|---|
| 473 |         WithConstTypeSubstitution() = default;
 | 
|---|
| 474 |         ~WithConstTypeSubstitution() = default;
 | 
|---|
| 475 | 
 | 
|---|
| 476 | public:
 | 
|---|
| 477 |         const TypeSubstitution * env = nullptr;
 | 
|---|
| 478 | };
 | 
|---|
| 479 | 
 | 
|---|
| 480 | class WithStmtsToAdd {
 | 
|---|
| 481 | protected:
 | 
|---|
| 482 |         WithStmtsToAdd() = default;
 | 
|---|
| 483 |         ~WithStmtsToAdd() = default;
 | 
|---|
| 484 | 
 | 
|---|
| 485 | public:
 | 
|---|
| 486 |         std::list< Statement* > stmtsToAddBefore;
 | 
|---|
| 487 |         std::list< Statement* > stmtsToAddAfter;
 | 
|---|
| 488 | };
 | 
|---|
| 489 | 
 | 
|---|
| 490 | class WithDeclsToAdd {
 | 
|---|
| 491 | protected:
 | 
|---|
| 492 |         WithDeclsToAdd() = default;
 | 
|---|
| 493 |         ~WithDeclsToAdd() {
 | 
|---|
| 494 |                 assert( declsToAddBefore.empty() );
 | 
|---|
| 495 |         }
 | 
|---|
| 496 | 
 | 
|---|
| 497 | public:
 | 
|---|
| 498 |         std::list< Declaration* > declsToAddBefore;
 | 
|---|
| 499 |         std::list< Declaration* > declsToAddAfter;
 | 
|---|
| 500 | };
 | 
|---|
| 501 | 
 | 
|---|
| 502 | class WithShortCircuiting {
 | 
|---|
| 503 | protected:
 | 
|---|
| 504 |         WithShortCircuiting() = default;
 | 
|---|
| 505 |         ~WithShortCircuiting() = default;
 | 
|---|
| 506 | 
 | 
|---|
| 507 | public:
 | 
|---|
| 508 |         bool_ref visit_children;
 | 
|---|
| 509 | };
 | 
|---|
| 510 | 
 | 
|---|
| 511 | class WithGuards {
 | 
|---|
| 512 | protected:
 | 
|---|
| 513 |         WithGuards() = default;
 | 
|---|
| 514 |         ~WithGuards() = default;
 | 
|---|
| 515 | 
 | 
|---|
| 516 | public:
 | 
|---|
| 517 |         at_cleanup_t at_cleanup;
 | 
|---|
| 518 | 
 | 
|---|
| 519 |         template< typename T >
 | 
|---|
| 520 |         void GuardValue( T& val ) {
 | 
|---|
| 521 |                 at_cleanup( [ val ]( void * newVal ) {
 | 
|---|
| 522 |                         * static_cast< T * >( newVal ) = val;
 | 
|---|
| 523 |                 }, static_cast< void * >( & val ) );
 | 
|---|
| 524 |         }
 | 
|---|
| 525 | 
 | 
|---|
| 526 |         template< typename T >
 | 
|---|
| 527 |         void GuardScope( T& val ) {
 | 
|---|
| 528 |                 val.beginScope();
 | 
|---|
| 529 |                 at_cleanup( []( void * val ) {
 | 
|---|
| 530 |                         static_cast< T * >( val )->endScope();
 | 
|---|
| 531 |                 }, static_cast< void * >( & val ) );
 | 
|---|
| 532 |         }
 | 
|---|
| 533 | 
 | 
|---|
| 534 |         template< typename Func >
 | 
|---|
| 535 |         void GuardAction( Func func ) {
 | 
|---|
| 536 |                 at_cleanup( [func](__attribute__((unused)) void *) { func(); }, nullptr );
 | 
|---|
| 537 |         }
 | 
|---|
| 538 | };
 | 
|---|
| 539 | 
 | 
|---|
| 540 | template<typename pass_type>
 | 
|---|
| 541 | class WithVisitorRef {
 | 
|---|
| 542 | protected:
 | 
|---|
| 543 |         WithVisitorRef() {}
 | 
|---|
| 544 |         ~WithVisitorRef() {}
 | 
|---|
| 545 | 
 | 
|---|
| 546 | public:
 | 
|---|
| 547 |         PassVisitor<pass_type> * const visitor = nullptr;
 | 
|---|
| 548 | 
 | 
|---|
| 549 |         bool isInFunction() const {
 | 
|---|
| 550 |                 return visitor->isInFunction();
 | 
|---|
| 551 |         }
 | 
|---|
| 552 | };
 | 
|---|
| 553 | 
 | 
|---|
| 554 | class WithIndexer {
 | 
|---|
| 555 | protected:
 | 
|---|
| 556 |         WithIndexer( bool trackIdentifiers = true ) : indexer(trackIdentifiers) {}
 | 
|---|
| 557 |         ~WithIndexer() {}
 | 
|---|
| 558 | 
 | 
|---|
| 559 | public:
 | 
|---|
| 560 |         SymTab::Indexer indexer;
 | 
|---|
| 561 | };
 | 
|---|
| 562 | 
 | 
|---|
| 563 | #include "Common/Stats.h"
 | 
|---|
| 564 | 
 | 
|---|
| 565 | extern struct PassVisitorStats {
 | 
|---|
| 566 |         size_t depth = 0;
 | 
|---|
| 567 |         Stats::Counters::MaxCounter<double> * max = nullptr;
 | 
|---|
| 568 |         Stats::Counters::AverageCounter<double> * avg = nullptr;
 | 
|---|
| 569 | } pass_visitor_stats;
 | 
|---|
| 570 | 
 | 
|---|
| 571 | #include "SynTree/TypeSubstitution.h"
 | 
|---|
| 572 | #include "PassVisitor.impl.h"
 | 
|---|