Changes in src/InitTweak/InitTweak.cc [d180746:65dc863]

File:

• src/InitTweak/InitTweak.cc (modified) (2 diffs)

src/InitTweak/InitTweak.cc

@@ -1 @@
-#include <stddef.h>                // for NULL
-#include <algorithm>               // for find, all_of
-#include <cassert>                 // for assertf, assert, safe_dynamic_cast
-#include <iostream>                // for ostream, cerr, endl
-#include <iterator>                // for back_insert_iterator, back_inserter
-#include <memory>                  // for __shared_ptr
-
-#include "Common/SemanticError.h"  // for SemanticError
-#include "Common/UniqueName.h"     // for UniqueName
-#include "Common/utility.h"        // for toString, deleteAll, maybeClone
-#include "GenPoly/GenPoly.h"       // for getFunctionType
+#include <algorithm>
 #include "InitTweak.h"
-#include "Parser/LinkageSpec.h"    // for Spec, isBuiltin, Intrinsic
-#include "ResolvExpr/typeops.h"    // for typesCompatibleIgnoreQualifiers
-#include "SymTab/Indexer.h"        // for Indexer
-#include "SynTree/Attribute.h"     // for Attribute
-#include "SynTree/Constant.h"      // for Constant
-#include "SynTree/Declaration.h"   // for ObjectDecl, DeclarationWithType
-#include "SynTree/Expression.h"    // for Expression, UntypedExpr, Applicati...
-#include "SynTree/Initializer.h"   // for Initializer, ListInit, Designation
-#include "SynTree/Label.h"         // for Label, noLabels
-#include "SynTree/Statement.h"     // for CompoundStmt, ExprStmt, BranchStmt
-#include "SynTree/Type.h"          // for FunctionType, ArrayType, PointerType
-#include "SynTree/Visitor.h"       // for Visitor, maybeAccept
-
-class UntypedValofExpr;
+#include "SynTree/Visitor.h"
+#include "SynTree/Statement.h"
+#include "SynTree/Initializer.h"
+#include "SynTree/Expression.h"
+#include "SynTree/Attribute.h"
+#include "GenPoly/GenPoly.h"
+#include "ResolvExpr/typeops.h"
 
 namespace InitTweak {
@@ -31 +14 @@
                 public:
                         bool hasDesignations = false;
-                        virtual void visit( Designation * des ) {
-                                if ( ! des->get_designators().empty() ) hasDesignations = true;
-                                else Visitor::visit( des );
-                        }
+                        template<typename Init>
+                        void handleInit( Init * init ) {
+                                if ( ! init->get_designators().empty() ) hasDesignations = true;
+                                else Visitor::visit( init );
+                        }
+                        virtual void visit( SingleInit * singleInit ) { handleInit( singleInit); }
+                        virtual void visit( ListInit * listInit ) { handleInit( listInit); }
                 };