Changes in src/ControlStruct/MultiLevelExit.cpp [400b8be:f75e25b]

File:

• src/ControlStruct/MultiLevelExit.cpp (modified) (1 diff)

src/ControlStruct/MultiLevelExit.cpp

@@ -588 +588 @@
                 }
 
+                ptr<Stmt> else_stmt = nullptr;
+                Stmt * loop_kid = nullptr;
+                // check if loop node and if so add else clause if it exists
+                const WhileDoStmt * whilePtr = dynamic_cast<const WhileDoStmt *>(kid.get());
+                if ( whilePtr && whilePtr->else_) {
+                        else_stmt = whilePtr->else_;
+                        WhileDoStmt * mutate_ptr = mutate(whilePtr);
+                        mutate_ptr->else_ = nullptr;
+                        loop_kid = mutate_ptr;
+                }
+                const ForStmt * forPtr = dynamic_cast<const ForStmt *>(kid.get());
+                if ( forPtr && forPtr->else_) {
+                        else_stmt = forPtr->else_;
+                        ForStmt * mutate_ptr = mutate(forPtr);
+                        mutate_ptr->else_ = nullptr;
+                        loop_kid = mutate_ptr;
+                }
+
                 try {
-                        ret.push_back( kid->accept( *visitor ) );
+                        if (else_stmt) ret.push_back( loop_kid->accept( *visitor ) );
+                        else ret.push_back( kid->accept( *visitor ) );
                 } catch ( SemanticErrorException & e ) {
                         errors.append( e );
                 }
+
+                if (else_stmt) ret.push_back(else_stmt);
 
                 if ( ! break_label.empty() ) {