extern/kalker
1
0
Fork 0
mirror of https://github.com/PaddiM8/kalker.git synced 2024-12-13 10:00:51 +01:00
Code Issues Packages Projects Releases Wiki Activity

Made sum() and ans compatible with complex numbers

Browse Source
This commit is contained in:
bakk 2021-05-22 22:03:35 +02:00
parent 637c3e331b
commit cff40be924
1 changed files with 21 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
kalk/src/interpreter.rs
View File

@ -50,13 +50,29 @@ impl<'a> Context<'a> {
Identifier::from_full_name("ans"), Identifier::from_full_name("ans"),
Box::new(Expr::Unit( Box::new(Expr::Unit(
num.unit.clone(), num.unit.clone(),
Box::new(Expr::Literal(num.to_f64())), Box::new(Expr::Binary(
Box::new(Expr::Literal(num.to_f64())),
TokenKind::Plus,
Box::new(Expr::Binary(
Box::new(Expr::Literal(num.imaginary_to_f64())),
TokenKind::Star,
Box::new(Expr::Var(Identifier::from_full_name("i"))),
)),
)),
)), )),
) )
} else { } else {
Stmt::VarDecl( Stmt::VarDecl(
Identifier::from_full_name("ans"), Identifier::from_full_name("ans"),
Box::new(Expr::Literal(num.to_f64())), Box::new(Expr::Binary(
Box::new(Expr::Literal(num.to_f64())),
TokenKind::Plus,
Box::new(Expr::Binary(
Box::new(Expr::Literal(num.imaginary_to_f64())),
TokenKind::Star,
Box::new(Expr::Var(Identifier::from_full_name("i"))),
)),
)),
) )
}); });
@ -338,7 +354,9 @@ pub(crate) fn eval_fn_call_expr(
for n in start..=end { for n in start..=end {
context.sum_n_value = Some(n); context.sum_n_value = Some(n);
sum.value += eval_expr(context, &expressions[2], "")?.value; let eval = eval_expr(context, &expressions[2], "")?;
sum.value += eval.value;
sum.imaginary_value += eval.imaginary_value;
} }
context.sum_n_value = None; context.sum_n_value = None;