extern/kalker
1
0
Fork 0
mirror of https://github.com/PaddiM8/kalker.git synced 2024-12-12 17:40:52 +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

20
kalk/src/interpreter.rs
View File

@ -50,13 +50,29 @@ impl<'a> Context<'a> {
Identifier::from_full_name("ans"),
Box::new(Expr::Unit(
num.unit.clone(),
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 {
Stmt::VarDecl(
Identifier::from_full_name("ans"),
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 {
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;