Changeset 1cb934d for src/InitTweak

Timestamp:

Nov 30, 2017, 5:11:59 PM (7 years ago)

Author:

Rob Schluntz <rschlunt@…>

Branches:

ADT, aaron-thesis, arm-eh, ast-experimental, cleanup-dtors, deferred_resn, demangler, enum, forall-pointer-decay, jacob/cs343-translation, jenkins-sandbox, master, new-ast, new-ast-unique-expr, new-env, no_list, persistent-indexer, pthread-emulation, qualifiedEnum, resolv-new, with_gc

Children:

1fbeebd

Parents:

fd236ed

Message:

Convert FixGlobalInit? to PassVisitor?

File:

• src/InitTweak/FixGlobalInit.cc (modified) (5 diffs)

src/InitTweak/FixGlobalInit.cc

@@ -20 +20 @@
 #include <algorithm>               // for replace_if
 
+#include "Common/PassVisitor.h"
 #include "Common/SemanticError.h"  // for SemanticError
 #include "Common/UniqueName.h"     // for UniqueName
@@ -35 +36 @@
 
 namespace InitTweak {
-        class GlobalFixer : public Visitor {
+        class GlobalFixer : public WithShortCircuiting {
           public:
                 GlobalFixer( const std::string & name, bool inLibrary );
 
-                virtual void visit( ObjectDecl *objDecl );
-                virtual void visit( FunctionDecl *functionDecl );
-                virtual void visit( StructDecl *aggregateDecl );
-                virtual void visit( UnionDecl *aggregateDecl );
-                virtual void visit( EnumDecl *aggregateDecl );
-                virtual void visit( TraitDecl *aggregateDecl );
-                virtual void visit( TypeDecl *typeDecl );
+                void previsit( ObjectDecl *objDecl );
+                void previsit( FunctionDecl *functionDecl );
+                void previsit( StructDecl *aggregateDecl );
+                void previsit( UnionDecl *aggregateDecl );
+                void previsit( EnumDecl *aggregateDecl );
+                void previsit( TraitDecl *aggregateDecl );
+                void previsit( TypeDecl *typeDecl );
 
                 UniqueName tempNamer;
@@ -53 +54 @@
 
         void fixGlobalInit( std::list< Declaration * > & translationUnit, const std::string & name, bool inLibrary ) {
-                GlobalFixer fixer( name, inLibrary );
-                acceptAll( translationUnit, fixer );
+                PassVisitor<GlobalFixer> visitor( name, inLibrary );
+                acceptAll( translationUnit, visitor );
+                GlobalFixer & fixer = visitor.pass;
                 // don't need to include function if it's empty
                 if ( fixer.initFunction->get_statements()->get_kids().empty() ) {
@@ -98 +100 @@
         }
 
-        void GlobalFixer::visit( ObjectDecl *objDecl ) {
+        void GlobalFixer::previsit( ObjectDecl *objDecl ) {
                 std::list< Statement * > & initStatements = initFunction->get_statements()->get_kids();
                 std::list< Statement * > & destroyStatements = destroyFunction->get_statements()->get_kids();
@@ -134 +136 @@
 
         // only modify global variables
-        void GlobalFixer::visit( __attribute__((unused)) FunctionDecl *functionDecl ) {}
-        void GlobalFixer::visit( __attribute__((unused)) StructDecl *aggregateDecl ) {}
-        void GlobalFixer::visit( __attribute__((unused)) UnionDecl *aggregateDecl ) {}
-        void GlobalFixer::visit( __attribute__((unused)) EnumDecl *aggregateDecl ) {}
-        void GlobalFixer::visit( __attribute__((unused)) TraitDecl *aggregateDecl ) {}
-        void GlobalFixer::visit( __attribute__((unused)) TypeDecl *typeDecl ) {}
+        void GlobalFixer::previsit( FunctionDecl * ) { visit_children = false; }
+        void GlobalFixer::previsit( StructDecl * ) { visit_children = false; }
+        void GlobalFixer::previsit( UnionDecl * ) { visit_children = false; }
+        void GlobalFixer::previsit( EnumDecl * ) { visit_children = false; }
+        void GlobalFixer::previsit( TraitDecl * ) { visit_children = false; }
+        void GlobalFixer::previsit( TypeDecl * ) { visit_children = false; }
 
 } // namespace InitTweak