Changeset d85141f for src/GenPoly/SpecializeNew.cpp

Timestamp:

Oct 16, 2023, 7:01:37 PM (9 months ago)

Author:

Andrew Beach <ajbeach@…>

Branches:

master

Children:

1b41219

Parents:

52a5262e

Message:

Small refactoring of some helper functions to avoid repeating a loop.

File:

• src/GenPoly/SpecializeNew.cpp (modified) (1 diff)

src/GenPoly/SpecializeNew.cpp

@@ -81 +81 @@
 }
 
-// The number of elements in a type if it is a flattened tuple.
-size_t flatTupleSize( const ast::Type * type ) {
-        if ( auto tuple = dynamic_cast<const ast::TupleType *>( type ) ) {
-                size_t sum = 0;
-                for ( auto t : *tuple ) {
-                        sum += flatTupleSize( t );
-                }
-                return sum;
-        } else {
-                return 1;
-        }
+// The number of elements in a list, if all tuples had been flattened.
+size_t flatTypeListSize( const std::vector<ast::ptr<ast::Type>> & types ) {
+        size_t sum = 0;
+        for ( const ast::ptr<ast::Type> & type : types ) {
+                if ( const ast::TupleType * tuple = type.as<ast::TupleType>() ) {
+                        sum += flatTypeListSize( tuple->types );
+                } else {
+                        sum += 1;
+                }
+        }
+        return sum;
 }
 
 // Find the total number of components in a parameter list.
 size_t functionParameterSize( const ast::FunctionType * type ) {
-        size_t sum = 0;
-        for ( auto param : type->params ) {
-                sum += flatTupleSize( param );
-        }
-        return sum;
+        return flatTypeListSize( type->params );
 }