dnlmlr/nek-lang
1
0
Fork 1
Code Issues Pull Requests Projects Releases Wiki Activity

Allow endless loops with no condition

Browse Source
This commit is contained in:
Daniel M
2022-02-10 20:36:26 +01:00
parent 2880ba81ab
commit 3806a61756
4 changed files with 36 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/astoptimizer.rs
View File

@@ -1,4 +1,4 @@
use crate::ast::{Ast, BlockScope, Expression, If, Loop, Statement, BinOpType, UnOpType};
use crate::ast::{Ast, BlockScope, Expression, If, Loop, Statement, BinOpType, UnOpType, VarDecl};
pub trait AstOptimizer {
fn optimize(ast: Ast) -> Ast;
@@ -24,7 +24,9 @@ impl SimpleAstOptimizer {
advancement,
body,
}) => {
Self::optimize_expr(condition);
if let Some(condition) = condition {
Self::optimize_expr(condition);
}
if let Some(advancement) = advancement {
Self::optimize_expr(advancement)
}
@@ -40,7 +42,7 @@ impl SimpleAstOptimizer {
Self::optimize_block(body_false);
}
Statement::Print(expr) => Self::optimize_expr(expr),
Statement::Declaration(_, _, expr) => Self::optimize_expr(expr),
Statement::Declaration(VarDecl { name: _, var_stackpos: _, rhs}) => Self::optimize_expr(rhs),
Statement::FunDeclare(_) => (),
Statement::Return(expr) => Self::optimize_expr(expr),
Statement::Break | Statement::Continue => (),