Fixed pre- and post statements of nested ternary operators

ftsrg · Oct 19, 2024 · 8749d32 · 8749d32
1 parent ad26bfb
commit 8749d32
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/.../main/java/hu/bme/mit/theta/frontend/transformation/grammar/function/FunctionVisitor.java b/.../main/java/hu/bme/mit/theta/frontend/transformation/grammar/function/FunctionVisitor.java
@@ -566,7 +566,7 @@ private void resetPostStatements(CStatement statement) {
 
     private List<CStatement> getStatementList(CStatement statement) {
         if (statement instanceof CCompound compound) {
-            return compound.getcStatementList();
+            return compound.getcStatementList().stream().flatMap(i -> getStatementList(i).stream()).toList();
         } else if (statement != null) {
             return List.of(statement);
         } else {
@@ -587,7 +587,7 @@ private List<CStatement> collectPreStatements(CStatement cStatement) {
                             collectPreStatements(cStatement.getPreStatements()).stream(),
                             getStatementList(cStatement.getPreStatements()).stream()),
                     ((CCompound) cStatement).getcStatementList().stream().flatMap(cStatement1 -> collectPreStatements(cStatement1).stream())
-            ).toList();
+            ).filter(i -> !(i instanceof CExpr)).toList();
         } else return List.of();
     }
 
@@ -604,7 +604,7 @@ private List<CStatement> collectPostStatements(CStatement cStatement) {
                     Stream.concat(
                             getStatementList(cStatement.getPostStatements()).stream(),
                             collectPostStatements(cStatement.getPostStatements()).stream())
-            ).toList();
+            ).filter(i -> !(i instanceof CExpr)).toList();
         } else return List.of();
     }