astutils.h: use pre-sized SmallVector in visitAstNodes()

firewave · firewave · commit fe25c1550fdc · 2022-09-27T20:26:04.000+02:00
diff --git a/lib/astutils.h b/lib/astutils.h
@@ -33,6 +33,7 @@
 #include "config.h"
 #include "errortypes.h"
 #include "library.h"
+#include "smallvector.h"
 #include "symboldatabase.h"
 
 class Settings;
@@ -55,11 +56,9 @@ void visitAstNodes(T *ast, const TFunc &visitor)
     if (!ast)
         return;
 
-    std::vector<T *> tokensContainer;
     // the size of 8 was determined in tests to be sufficient to avoid excess allocations. also add 1 as a buffer.
     // we might need to increase that value in the future.
-    tokensContainer.reserve(8 + 1);
-    std::stack<T *, std::vector<T *>> tokens(std::move(tokensContainer));
+    std::stack<T *, SmallVector<T *, 8 + 1>> tokens;
     T *tok = ast;
     do {
         ChildrenToVisit c = visitor(tok);
