Changes in src/InitTweak/FixInit.cc [52c14b3:092528b]

File:

• src/InitTweak/FixInit.cc (modified) (9 diffs)

src/InitTweak/FixInit.cc

@@ -104 +104 @@
                         virtual void visit( CompoundStmt *compoundStmt ) override;
                         virtual void visit( DeclStmt *stmt ) override;
-
-                        // don't go into other functions
-                        virtual void visit( FunctionDecl *decl ) override {}
-
                   protected:
                         ObjectSet curVars;
@@ -170 +166 @@
                         typedef std::list< OrderedDecls > OrderedDeclsStack;
 
-                        InsertDtors( LabelFinder & finder ) : finder( finder ), labelVars( finder.vars ) {}
+                        InsertDtors( LabelFinder & finder ) : labelVars( finder.vars ) {}
 
                         using Parent::visit;
 
                         virtual void visit( ObjectDecl * objDecl ) override;
-                        virtual void visit( FunctionDecl * funcDecl ) override;
 
                         virtual void visit( CompoundStmt * compoundStmt ) override;
@@ -183 +178 @@
                         void handleGoto( BranchStmt * stmt );
 
-                        LabelFinder & finder;
                         LabelFinder::LabelMap & labelVars;
                         OrderedDeclsStack reverseDeclOrder;
@@ -324 +318 @@
                         LabelFinder finder;
                         InsertDtors inserter( finder );
+                        acceptAll( translationUnit, finder );
                         acceptAll( translationUnit, inserter );
                 }
@@ -783 +778 @@
                 }
 
-                void ObjDeclCollector::visit( CompoundStmt * compoundStmt ) {
+                void ObjDeclCollector::visit( CompoundStmt *compoundStmt ) {
                         std::set< ObjectDecl * > prevVars = curVars;
                         Parent::visit( compoundStmt );
@@ -789 +784 @@
                 }
 
-                void ObjDeclCollector::visit( DeclStmt * stmt ) {
+                void ObjDeclCollector::visit( DeclStmt *stmt ) {
                         // keep track of all variables currently in scope
                         if ( ObjectDecl * objDecl = dynamic_cast< ObjectDecl * > ( stmt->get_decl() ) ) {
@@ -833 +828 @@
                         } // if
                         Parent::visit( objDecl );
-                }
-
-                template< typename Visitor >
-                void handleFuncDecl( FunctionDecl * funcDecl, Visitor & visitor ) {
-                        maybeAccept( funcDecl->get_functionType(), visitor );
-                        acceptAll( funcDecl->get_oldDecls(), visitor );
-                        maybeAccept( funcDecl->get_statements(), visitor );
-                }
-
-                void InsertDtors::visit( FunctionDecl * funcDecl ) {
-                        // each function needs to have its own set of labels
-                        ValueGuard< LabelFinder::LabelMap > oldLabels( labelVars );
-                        labelVars.clear();
-                        handleFuncDecl( funcDecl, finder );
-
-                        // all labels for this function have been collected, insert destructors as appropriate.
-                        // can't be Parent::mutate, because ObjDeclCollector bottoms out on FunctionDecl
-                        handleFuncDecl( funcDecl, *this );
                 }
 
@@ -975 +952 @@
                         std::set_difference( usedUninit.begin(), usedUninit.end(), unhandled.begin(), unhandled.end(), std::inserter( diff, diff.begin() ) );
                         for ( DeclarationWithType * member : diff ) {
-                                emit( "in ", CodeGen::genPrettyType( function->get_functionType(), function->get_name() ), ", field ", member->get_name(), " used before being constructed" );
+                                emit( "in ", CodeGen::genType( function->get_functionType(), function->get_name(), false ), ", field ", member->get_name(), " used before being constructed" );
                         }
 
@@ -1020 +997 @@
                                                         }
                                                 } catch ( SemanticError & error ) {
-                                                        emit( "in ", CodeGen::genPrettyType( function->get_functionType(), function->get_name() ), ", field ", field->get_name(), " not explicitly ", isCtor ? "constructed" : "destructed",  " and no ", isCtor ? "default constructor" : "destructor", " found" );
+                                                        emit( "in ", CodeGen::genType( function->get_functionType(), function->get_name(), false ), ", field ", field->get_name(), " not explicitly ", isCtor ? "constructed" : "destructed",  " and no ", isCtor ? "default constructor" : "destructor", " found" );
                                                 }
                                         }