@@ -1424,6 +1424,50 @@ impl AstVisitor for GraphVizVisitor {
14241424
14251425 //* --------------------------------------------------------------------- *//
14261426
1427+ fn visit_loop_stmt_node ( & mut self , loop_stmt_node : & LoopStmtNode ) {
1428+ match & loop_stmt_node. loop_types {
1429+ LoopStmtTypes :: LoopForStmt {
1430+ loop_for_stmt_node : loop_for_expr_node,
1431+ } => {
1432+ loop_for_expr_node. accept ( self ) ;
1433+ }
1434+ LoopStmtTypes :: LoopInStmt { loop_in_stmt_node } => {
1435+ loop_in_stmt_node. accept ( self ) ;
1436+ }
1437+ LoopStmtTypes :: LoopInfiniteStmt {
1438+ loop_infinite_stmt_node,
1439+ } => {
1440+ loop_infinite_stmt_node. accept ( self ) ;
1441+ }
1442+ }
1443+ }
1444+
1445+ //* --------------------------------------------------------------------- *//
1446+
1447+ fn visit_loop_for_stmt_node ( & mut self , loop_for_expr_node : & LoopForStmtNode ) {
1448+ // only call if there are statements
1449+ if loop_for_expr_node. statements . len ( ) != 0 {
1450+ self . visit_decl_stmts ( & loop_for_expr_node. statements ) ;
1451+ }
1452+ }
1453+
1454+ //* --------------------------------------------------------------------- *//
1455+
1456+ fn visit_loop_in_stmt_node ( & mut self , loop_in_stmt_node : & LoopInStmtNode ) {
1457+ // only call if there are statements
1458+ if loop_in_stmt_node. statements . len ( ) != 0 {
1459+ self . visit_decl_stmts ( & loop_in_stmt_node. statements ) ;
1460+ }
1461+ }
1462+
1463+ //* --------------------------------------------------------------------- *//
1464+
1465+ fn visit_loop_infinite_stmt_node ( & mut self , loop_in_expr_node : & LoopInfiniteStmtNode ) {
1466+ self . visit_decl_stmts ( & loop_in_expr_node. statements ) ;
1467+ }
1468+
1469+ //* --------------------------------------------------------------------- *//
1470+
14271471 fn visit_bool_test_conditional_branch_node (
14281472 & mut self ,
14291473 bool_test_true_branch_node : & BoolTestConditionalBranchNode ,
0 commit comments