Changes in src/CodeGen/FixNames.cc [c50d54d:fa4c094]

File:

• src/CodeGen/FixNames.cc (modified) (5 diffs)

src/CodeGen/FixNames.cc

@@ -19 +19 @@
 #include <string>                  // for string, operator!=, operator==
 
-#include "Common/PassVisitor.h"
 #include "Common/SemanticError.h"  // for SemanticError
 #include "FixMain.h"               // for FixMain
@@ -33 +32 @@
 
 namespace CodeGen {
-        class FixNames : public WithGuards {
+        class FixNames : public Visitor {
           public:
-                void postvisit( ObjectDecl *objectDecl );
-                void postvisit( FunctionDecl *functionDecl );
+                virtual void visit( ObjectDecl *objectDecl );
+                virtual void visit( FunctionDecl *functionDecl );
 
-                void previsit( CompoundStmt *compoundStmt );
+                virtual void visit( CompoundStmt *compoundStmt );
           private:
                 int scopeLevel = 1;
@@ -94 +93 @@
         }
 
-        void fixNames( std::list< Declaration* > & translationUnit ) {
-                PassVisitor<FixNames> fixer;
+        void fixNames( std::list< Declaration* > translationUnit ) {
+                FixNames fixer;
                 acceptAll( translationUnit, fixer );
         }
 
-        void FixNames::fixDWT( DeclarationWithType * dwt ) {
+        void FixNames::fixDWT( DeclarationWithType *dwt ) {
                 if ( dwt->get_name() != "" ) {
                         if ( LinkageSpec::isMangled( dwt->get_linkage() ) ) {
@@ -108 +107 @@
         }
 
-        void FixNames::postvisit( ObjectDecl * objectDecl ) {
+        void FixNames::visit( ObjectDecl *objectDecl ) {
+                Visitor::visit( objectDecl );
                 fixDWT( objectDecl );
         }
 
-        void FixNames::postvisit( FunctionDecl * functionDecl ) {
+        void FixNames::visit( FunctionDecl *functionDecl ) {
+                Visitor::visit( functionDecl );
                 fixDWT( functionDecl );
 
@@ -120 +121 @@
                                 throw SemanticError("Main expected to have 0, 2 or 3 arguments\n", functionDecl);
                         }
-                        functionDecl->get_statements()->get_kids().push_back( new ReturnStmt( new ConstantExpr( Constant::from_int( 0 ) ) ) );
+                        functionDecl->get_statements()->get_kids().push_back( new ReturnStmt( noLabels, new ConstantExpr( Constant::from_int( 0 ) ) ) );
                         CodeGen::FixMain::registerMain( functionDecl );
                 }
         }
 
-        void FixNames::previsit( CompoundStmt * ) {
+        void FixNames::visit( CompoundStmt *compoundStmt ) {
                 scopeLevel++;
-                GuardAction( [this](){ scopeLevel--; } );
+                Visitor::visit( compoundStmt );
+                scopeLevel--;
         }
 } // namespace CodeGen