| 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 | // 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 "Pass.hpp"
 | 
|---|
| 18 | 
 | 
|---|
| 19 | namespace ast {
 | 
|---|
| 20 | template<typename pass_type>
 | 
|---|
| 21 | class Pass;
 | 
|---|
| 22 | 
 | 
|---|
| 23 | namespace __pass {
 | 
|---|
| 24 |         typedef std::function<void( void * )> cleanup_func_t;
 | 
|---|
| 25 |         typedef std::function<void( cleanup_func_t, void * )> at_cleanup_t;
 | 
|---|
| 26 | 
 | 
|---|
| 27 | 
 | 
|---|
| 28 |         // boolean reference that may be null
 | 
|---|
| 29 |         // either refers to a boolean value or is null and returns true
 | 
|---|
| 30 |         class bool_ref {
 | 
|---|
| 31 |         public:
 | 
|---|
| 32 |                 bool_ref() = default;
 | 
|---|
| 33 |                 ~bool_ref() = default;
 | 
|---|
| 34 | 
 | 
|---|
| 35 |                 operator bool() { return m_ref ? *m_ref : true; }
 | 
|---|
| 36 |                 bool operator=( bool val ) { assert(m_ref); return *m_ref = val; }
 | 
|---|
| 37 | 
 | 
|---|
| 38 |         private:
 | 
|---|
| 39 | 
 | 
|---|
| 40 |                 friend class visit_children_guard;
 | 
|---|
| 41 | 
 | 
|---|
| 42 |                 bool * set( bool * val ) {
 | 
|---|
| 43 |                         bool * prev = m_ref;
 | 
|---|
| 44 |                         m_ref = val;
 | 
|---|
| 45 |                         return prev;
 | 
|---|
| 46 |                 }
 | 
|---|
| 47 | 
 | 
|---|
| 48 |                 bool * m_ref = nullptr;
 | 
|---|
| 49 |         };
 | 
|---|
| 50 | 
 | 
|---|
| 51 |         // Implementation of the guard value
 | 
|---|
| 52 |         // Created inside the visit scope
 | 
|---|
| 53 |         class guard_value {
 | 
|---|
| 54 |         public:
 | 
|---|
| 55 |                 /// Push onto the cleanup
 | 
|---|
| 56 |                 guard_value( at_cleanup_t * at_cleanup ) {
 | 
|---|
| 57 |                         if( at_cleanup ) {
 | 
|---|
| 58 |                                 *at_cleanup = [this]( cleanup_func_t && func, void* val ) {
 | 
|---|
| 59 |                                         push( std::move( func ), val );
 | 
|---|
| 60 |                                 };
 | 
|---|
| 61 |                         }
 | 
|---|
| 62 |                 }
 | 
|---|
| 63 | 
 | 
|---|
| 64 |                 ~guard_value() {
 | 
|---|
| 65 |                         while( !cleanups.empty() ) {
 | 
|---|
| 66 |                                 auto& cleanup = cleanups.top();
 | 
|---|
| 67 |                                 cleanup.func( cleanup.val );
 | 
|---|
| 68 |                                 cleanups.pop();
 | 
|---|
| 69 |                         }
 | 
|---|
| 70 |                 }
 | 
|---|
| 71 | 
 | 
|---|
| 72 |                 void push( cleanup_func_t && func, void* val ) {
 | 
|---|
| 73 |                         cleanups.emplace( std::move(func), val );
 | 
|---|
| 74 |                 }
 | 
|---|
| 75 | 
 | 
|---|
| 76 |         private:
 | 
|---|
| 77 |                 struct cleanup_t {
 | 
|---|
| 78 |                         cleanup_func_t func;
 | 
|---|
| 79 |                         void * val;
 | 
|---|
| 80 | 
 | 
|---|
| 81 |                         cleanup_t( cleanup_func_t&& func, void * val ) : func(func), val(val) {}
 | 
|---|
| 82 |                 };
 | 
|---|
| 83 | 
 | 
|---|
| 84 |                 std::stack< cleanup_t > cleanups;
 | 
|---|
| 85 |         };
 | 
|---|
| 86 | 
 | 
|---|
| 87 |         // Guard structure implementation for whether or not children should be visited
 | 
|---|
| 88 |         class visit_children_guard {
 | 
|---|
| 89 |         public:
 | 
|---|
| 90 | 
 | 
|---|
| 91 |                 visit_children_guard( bool_ref * ref )
 | 
|---|
| 92 |                         : m_val ( true )
 | 
|---|
| 93 |                         , m_prev( ref ? ref->set( &m_val ) : nullptr )
 | 
|---|
| 94 |                         , m_ref ( ref )
 | 
|---|
| 95 |                 {}
 | 
|---|
| 96 | 
 | 
|---|
| 97 |                 ~visit_children_guard() {
 | 
|---|
| 98 |                         if( m_ref ) {
 | 
|---|
| 99 |                                 m_ref->set( m_prev );
 | 
|---|
| 100 |                         }
 | 
|---|
| 101 |                 }
 | 
|---|
| 102 | 
 | 
|---|
| 103 |                 operator bool() { return m_val; }
 | 
|---|
| 104 | 
 | 
|---|
| 105 |         private:
 | 
|---|
| 106 |                 bool       m_val;
 | 
|---|
| 107 |                 bool     * m_prev;
 | 
|---|
| 108 |                 bool_ref * m_ref;
 | 
|---|
| 109 |         };
 | 
|---|
| 110 | 
 | 
|---|
| 111 |         //-------------------------------------------------------------------------------------------------------------------------------------------------------------------------
 | 
|---|
| 112 |         // Deep magic (a.k.a template meta programming) to make the templated visitor work
 | 
|---|
| 113 |         // Basically the goal is to make 2 previsit
 | 
|---|
| 114 |         // 1 - Use when a pass implements a valid previsit. This uses overloading which means the any overload of
 | 
|---|
| 115 |         //     'pass.previsit( node )' that compiles will be used for that node for that type
 | 
|---|
| 116 |         //     This requires that this option only compile for passes that actually define an appropriate visit.
 | 
|---|
| 117 |         //     SFINAE will make sure the compilation errors in this function don't halt the build.
 | 
|---|
| 118 |         //     See http://en.cppreference.com/w/cpp/language/sfinae for details on SFINAE
 | 
|---|
| 119 |         // 2 - Since the first implementation might not be specilizable, the second implementation exists and does nothing.
 | 
|---|
| 120 |         //     This is needed only to eliminate the need for passes to specify any kind of handlers.
 | 
|---|
| 121 |         //     The second implementation only works because it has a lower priority. This is due to the bogus last parameter.
 | 
|---|
| 122 |         //     The second implementation takes a long while the first takes an int. Since the caller always passes an literal 0
 | 
|---|
| 123 |         //     the first implementation takes priority in regards to overloading.
 | 
|---|
| 124 |         //-------------------------------------------------------------------------------------------------------------------------------------------------------------------------
 | 
|---|
| 125 |         // PreVisit : may mutate the pointer passed in if the node is mutated in the previsit call
 | 
|---|
| 126 |         template<typename pass_t, typename node_t>
 | 
|---|
| 127 |         static inline auto previsit( pass_t & pass, const node_t * & node, int ) -> decltype( pass.previsit( node ), void() ) {
 | 
|---|
| 128 |                 node = pass.previsit( node );
 | 
|---|
| 129 |                 assert(node);
 | 
|---|
| 130 |         }
 | 
|---|
| 131 | 
 | 
|---|
| 132 |         template<typename pass_t, typename node_t>
 | 
|---|
| 133 |         static inline auto previsit( pass_t &, const node_t *, long ) {}
 | 
|---|
| 134 | 
 | 
|---|
| 135 |         // PostVisit : never mutates the passed pointer but may return a different node
 | 
|---|
| 136 |         template<typename pass_t, typename node_t>
 | 
|---|
| 137 |         static inline auto postvisit( pass_t & pass, const node_t * node, int ) -> decltype( pass.postvisit( node ), (const node_t *)nullptr ) {
 | 
|---|
| 138 |                 return pass.postvisit( node );
 | 
|---|
| 139 |         }
 | 
|---|
| 140 | 
 | 
|---|
| 141 |         template<typename pass_t, typename node_t>
 | 
|---|
| 142 |         static inline const node_t * postvisit( pass_t &, const node_t * node, long ) { return node; }
 | 
|---|
| 143 | 
 | 
|---|
| 144 |         //-------------------------------------------------------------------------------------------------------------------------------------------------------------------------
 | 
|---|
| 145 |         // Deep magic (a.k.a template meta programming) continued
 | 
|---|
| 146 |         // To make the templated visitor be more expressive, we allow 'accessories' : classes/structs the implementation can inherit
 | 
|---|
| 147 |         // from in order to get extra functionallity for example
 | 
|---|
| 148 |         // class ErrorChecker : WithShortCircuiting { ... };
 | 
|---|
| 149 |         // Pass<ErrorChecker> checker;
 | 
|---|
| 150 |         // this would define a pass that uses the templated visitor with the additionnal feature that it has short circuiting
 | 
|---|
| 151 |         // Note that in all cases the accessories are not required but guarantee the requirements of the feature is matched
 | 
|---|
| 152 |         //-------------------------------------------------------------------------------------------------------------------------------------------------------------------------
 | 
|---|
| 153 |         // For several accessories, the feature is enabled by detecting that a specific field is present
 | 
|---|
| 154 |         // Use a macro the encapsulate the logic of detecting a particular field
 | 
|---|
| 155 |         // The type is not strictly enforced but does match the accessory
 | 
|---|
| 156 |         #define FIELD_PTR( name, default_type ) \
 | 
|---|
| 157 |         template< typename pass_t > \
 | 
|---|
| 158 |         static inline auto name( pass_t & pass, int ) -> decltype( &pass.name ) { return &pass.name; } \
 | 
|---|
| 159 |         \
 | 
|---|
| 160 |         template< typename pass_t > \
 | 
|---|
| 161 |         static inline default_type * name( pass_t &, long ) { return nullptr; }
 | 
|---|
| 162 | 
 | 
|---|
| 163 |         // List of fields and their expected types
 | 
|---|
| 164 |         FIELD_PTR( env, const ast::TypeSubstitution * )
 | 
|---|
| 165 |         FIELD_PTR( stmtsToAddBefore, std::list< ast::ptr< ast::Stmt > > )
 | 
|---|
| 166 |         FIELD_PTR( stmtsToAddAfter , std::list< ast::ptr< ast::Stmt > > )
 | 
|---|
| 167 |         FIELD_PTR( declsToAddBefore, std::list< ast::ptr< ast::Decl > > )
 | 
|---|
| 168 |         FIELD_PTR( declsToAddAfter , std::list< ast::ptr< ast::Decl > > )
 | 
|---|
| 169 |         FIELD_PTR( visit_children, __pass::bool_ref )
 | 
|---|
| 170 |         FIELD_PTR( at_cleanup, __pass::at_cleanup_t )
 | 
|---|
| 171 |         FIELD_PTR( visitor, ast::Pass<pass_t> * const )
 | 
|---|
| 172 | 
 | 
|---|
| 173 |         // Remove the macro to make sure we don't clash
 | 
|---|
| 174 |         #undef FIELD_PTR
 | 
|---|
| 175 | 
 | 
|---|
| 176 |         // Another feature of the templated visitor is that it calls beginScope()/endScope() for compound statement.
 | 
|---|
| 177 |         // All passes which have such functions are assumed desire this behaviour
 | 
|---|
| 178 |         // detect it using the same strategy
 | 
|---|
| 179 |         namespace scope {
 | 
|---|
| 180 |                 template<typename pass_t>
 | 
|---|
| 181 |                 static inline auto enter( pass_t & pass, int ) -> decltype( pass.beginScope(), void() ) {
 | 
|---|
| 182 |                         pass.beginScope();
 | 
|---|
| 183 |                 }
 | 
|---|
| 184 | 
 | 
|---|
| 185 |                 template<typename pass_t>
 | 
|---|
| 186 |                 static inline void enter( pass_t &, long ) {}
 | 
|---|
| 187 | 
 | 
|---|
| 188 |                 template<typename pass_t>
 | 
|---|
| 189 |                 static inline auto leave( pass_t & pass, int ) -> decltype( pass.endScope(), void() ) {
 | 
|---|
| 190 |                         pass.endScope();
 | 
|---|
| 191 |                 }
 | 
|---|
| 192 | 
 | 
|---|
| 193 |                 template<typename pass_t>
 | 
|---|
| 194 |                 static inline void leave( pass_t &, long ) {}
 | 
|---|
| 195 |         };
 | 
|---|
| 196 | 
 | 
|---|
| 197 |         // Finally certain pass desire an up to date indexer automatically
 | 
|---|
| 198 |         // detect the presence of a member name indexer and call all the members appropriately
 | 
|---|
| 199 |         namespace indexer {
 | 
|---|
| 200 |                 // Some simple scoping rules
 | 
|---|
| 201 |                 template<typename pass_t>
 | 
|---|
| 202 |                 static inline auto enter( pass_t & pass, int ) -> decltype( pass.indexer.enterScope(), void() ) {
 | 
|---|
| 203 |                         pass.indexer.enterScope();
 | 
|---|
| 204 |                 }
 | 
|---|
| 205 | 
 | 
|---|
| 206 |                 template<typename pass_t>
 | 
|---|
| 207 |                 static inline auto enter( pass_t &, long ) {}
 | 
|---|
| 208 | 
 | 
|---|
| 209 |                 template<typename pass_t>
 | 
|---|
| 210 |                 static inline auto leave( pass_t & pass, int ) -> decltype( pass.indexer.leaveScope(), void() ) {
 | 
|---|
| 211 |                         pass.indexer.leaveScope();
 | 
|---|
| 212 |                 }
 | 
|---|
| 213 | 
 | 
|---|
| 214 |                 template<typename pass_t>
 | 
|---|
| 215 |                 static inline auto leave( pass_t &, long ) {}
 | 
|---|
| 216 | 
 | 
|---|
| 217 |                 // The indexer has 2 kind of functions mostly, 1 argument and 2 arguments
 | 
|---|
| 218 |                 // Create macro to condense these common patterns
 | 
|---|
| 219 |                 #define INDEXER_FUNC1( func, type ) \
 | 
|---|
| 220 |                 template<typename pass_t> \
 | 
|---|
| 221 |                 static inline auto func( pass_t & pass, int, type arg ) -> decltype( pass.indexer.func( arg ), void() ) {\
 | 
|---|
| 222 |                         pass.indexer.func( arg ); \
 | 
|---|
| 223 |                 } \
 | 
|---|
| 224 |                 \
 | 
|---|
| 225 |                 template<typename pass_t> \
 | 
|---|
| 226 |                 static inline void func( pass_t &, long, type ) {}
 | 
|---|
| 227 | 
 | 
|---|
| 228 |                 #define INDEXER_FUNC2( func, type1, type2 ) \
 | 
|---|
| 229 |                 template<typename pass_t> \
 | 
|---|
| 230 |                 static inline auto func( pass_t & pass, int, type1 arg1, type2 arg2 ) -> decltype( pass.indexer.func( arg1, arg2 ), void () ) {\
 | 
|---|
| 231 |                         pass.indexer.func( arg1, arg2 ); \
 | 
|---|
| 232 |                 } \
 | 
|---|
| 233 |                         \
 | 
|---|
| 234 |                 template<typename pass_t> \
 | 
|---|
| 235 |                 static inline void func( pass_t &, long, type1, type2 ) {}
 | 
|---|
| 236 | 
 | 
|---|
| 237 |                 INDEXER_FUNC1( addId     , const DeclWithType *  );
 | 
|---|
| 238 |                 INDEXER_FUNC1( addType   , const NamedTypeDecl * );
 | 
|---|
| 239 |                 INDEXER_FUNC1( addStruct , const StructDecl *    );
 | 
|---|
| 240 |                 INDEXER_FUNC1( addEnum   , const EnumDecl *      );
 | 
|---|
| 241 |                 INDEXER_FUNC1( addUnion  , const UnionDecl *     );
 | 
|---|
| 242 |                 INDEXER_FUNC1( addTrait  , const TraitDecl *     );
 | 
|---|
| 243 |                 INDEXER_FUNC2( addWith   , const std::list< Expression * > &, const Node * );
 | 
|---|
| 244 | 
 | 
|---|
| 245 |                 // A few extra functions have more complicated behaviour, they are hand written
 | 
|---|
| 246 |                 template<typename pass_t>
 | 
|---|
| 247 |                 static inline auto addStructFwd( pass_t & pass, int, const ast::StructDecl * decl ) -> decltype( pass.indexer.addStruct( decl ), void() ) {
 | 
|---|
| 248 |                         ast::StructDecl * fwd = new ast::StructDecl( decl->location, decl->name );
 | 
|---|
| 249 |                         fwd->parameters = decl->parameters;
 | 
|---|
| 250 |                         pass.indexer.addStruct( fwd );
 | 
|---|
| 251 |                 }
 | 
|---|
| 252 | 
 | 
|---|
| 253 |                 template<typename pass_t>
 | 
|---|
| 254 |                 static inline void addStructFwd( pass_t &, long, const ast::StructDecl * ) {}
 | 
|---|
| 255 | 
 | 
|---|
| 256 |                 template<typename pass_t>
 | 
|---|
| 257 |                 static inline auto addUnionFwd( pass_t & pass, int, const ast::UnionDecl * decl ) -> decltype( pass.indexer.addUnion( decl ), void() ) {
 | 
|---|
| 258 |                         UnionDecl * fwd = new UnionDecl( decl->location, decl->name );
 | 
|---|
| 259 |                         fwd->parameters = decl->parameters;
 | 
|---|
| 260 |                         pass.indexer.addUnion( fwd );
 | 
|---|
| 261 |                 }
 | 
|---|
| 262 | 
 | 
|---|
| 263 |                 template<typename pass_t>
 | 
|---|
| 264 |                 static inline void addUnionFwd( pass_t &, long, const ast::UnionDecl * ) {}
 | 
|---|
| 265 | 
 | 
|---|
| 266 |                 template<typename pass_t>
 | 
|---|
| 267 |                 static inline auto addStruct( pass_t & pass, int, const std::string & str ) -> decltype( pass.indexer.addStruct( str ), void() ) {
 | 
|---|
| 268 |                         if ( ! pass.indexer.lookupStruct( str ) ) {
 | 
|---|
| 269 |                                 pass.indexer.addStruct( str );
 | 
|---|
| 270 |                         }
 | 
|---|
| 271 |                 }
 | 
|---|
| 272 | 
 | 
|---|
| 273 |                 template<typename pass_t>
 | 
|---|
| 274 |                 static inline void addStruct( pass_t &, long, const std::string & ) {}
 | 
|---|
| 275 | 
 | 
|---|
| 276 |                 template<typename pass_t>
 | 
|---|
| 277 |                 static inline auto addUnion( pass_t & pass, int, const std::string & str ) -> decltype( pass.indexer.addUnion( str ), void() ) {
 | 
|---|
| 278 |                         if ( ! pass.indexer.lookupUnion( str ) ) {
 | 
|---|
| 279 |                                 pass.indexer.addUnion( str );
 | 
|---|
| 280 |                         }
 | 
|---|
| 281 |                 }
 | 
|---|
| 282 | 
 | 
|---|
| 283 |                 template<typename pass_t>
 | 
|---|
| 284 |                 static inline void addUnion( pass_t &, long, const std::string & ) {}
 | 
|---|
| 285 | 
 | 
|---|
| 286 |                 #undef INDEXER_FUNC1
 | 
|---|
| 287 |                 #undef INDEXER_FUNC2
 | 
|---|
| 288 |         };
 | 
|---|
| 289 | };
 | 
|---|
| 290 | };
 | 
|---|