Changeset 525f7ad for src/AST/Pass.impl.hpp

Timestamp:

Jun 19, 2024, 3:20:39 PM (2 months ago)

Author:

JiadaL <j82liang@…>

Branches:

master

Children:

57e43cd

Parents:

1725989

Message:

1. Add count_e( enum_name ), a pseudo function that return the number of element in an enum; 2. Implementation of enum range loop.

File:

• src/AST/Pass.impl.hpp (modified) (2 diffs)

src/AST/Pass.impl.hpp

@@ -802 +802 @@
                 maybe_accept_top( node, &ForStmt::cond  );
                 maybe_accept_top( node, &ForStmt::inc   );
+                maybe_accept_top( node, &ForStmt::range_over );
                 maybe_accept_as_compound( node, &ForStmt::body  );
         }
@@ -1323 +1324 @@
         }
 
+        VISIT_END( Expr, node );
+}
+
+//--------------------------------------------------------------------------
+// CountExpr
+template< typename core_t >
+const ast::Expr * ast::Pass< core_t >::visit( const ast::CountExpr * node ) {
+        VISIT_START( node );
+        if ( __visit_children() ) {
+                {
+                        guard_symtab guard { *this };
+                        maybe_accept( node, &CountExpr::result );
+                }
+                maybe_accept( node, &CountExpr::type );
+        }
         VISIT_END( Expr, node );
 }