bitwise operators, hex/bin/seximal literals, changed infix boxing notation, fixed semicolons in repl

This commit is contained in:
TriMill 2022-10-18 08:35:44 -04:00
parent 443262f711
commit 7be878b97a
10 changed files with 180 additions and 69 deletions

View file

@ -229,6 +229,10 @@ let add7 = add(7);
println(add7(3)); # prints 10 println(add7(3)); # prints 10
``` ```
### Boxed infix operators
Some infix operators can be converted to functions by prefixing them with a backslash `\`. For example, `\+` is a function that computes the sum of its two arguments (the same as `fn(x, y) (x + y)`). This can be done for arithmetic, comparison, and bitwise operators only.
## Iterators ## Iterators
An iterator is simply a function with the following properties: An iterator is simply a function with the following properties:

View file

@ -144,7 +144,9 @@ pub fn eval_expr(expr: &Expr, env: EnvRef) -> Result<Value, RuntimeError> {
Some(OpType::Additive) Some(OpType::Additive)
| Some(OpType::Multiplicative) | Some(OpType::Multiplicative)
| Some(OpType::Exponential) | Some(OpType::Exponential)
| Some(OpType::Comparison) => { | Some(OpType::Comparison)
| Some(OpType::BitwiseAnd)
| Some(OpType::BitwiseOr) => {
let l = eval_expr(lhs, env.clone())?; let l = eval_expr(lhs, env.clone())?;
let r = eval_expr(rhs, env)?; let r = eval_expr(rhs, env)?;
eval_standard_binary(l, r, &op.ty, &op.pos) eval_standard_binary(l, r, &op.ty, &op.pos)
@ -361,6 +363,8 @@ pub fn eval_standard_binary(l: Value, r: Value, opty: &TokenType, pos: &Position
TokenType::Star => &l * &r, TokenType::Star => &l * &r,
TokenType::Slash => &l / &r, TokenType::Slash => &l / &r,
TokenType::Percent => &l % &r, TokenType::Percent => &l % &r,
TokenType::Amper => &l & &r,
TokenType::Pipe => &l | &r,
TokenType::Caret => l.pow(&r), TokenType::Caret => l.pow(&r),
TokenType::DoubleSlash => l.fracdiv(&r), TokenType::DoubleSlash => l.fracdiv(&r),
TokenType::DoubleEqual => Ok(Value::Bool(l == r)), TokenType::DoubleEqual => Ok(Value::Bool(l == r)),
@ -511,8 +515,9 @@ pub fn eval_ternary(arg1: &Expr, arg2: &Expr, arg3: &Expr, op: &Token, env: EnvR
pub fn eval_unary(arg: &Expr, op: &Token, env: EnvRef) -> Result<Value, RuntimeError> { pub fn eval_unary(arg: &Expr, op: &Token, env: EnvRef) -> Result<Value, RuntimeError> {
let a = eval_expr(arg, env)?; let a = eval_expr(arg, env)?;
match op.ty { match op.ty {
TokenType::Minus => -a, TokenType::Minus => -&a,
TokenType::Bang => Ok(Value::Bool(!a.truthy())), TokenType::Bang => Ok(Value::Bool(!a.truthy())),
TokenType::Tilde => !&a,
_ => todo!(), _ => todo!(),
}.map_err(|e| RuntimeError::new(e, op.pos.clone())) }.map_err(|e| RuntimeError::new(e, op.pos.clone()))
} }

View file

@ -6,7 +6,8 @@ pub fn interpret(src: &str, fname: Option<String>, env: Option<EnvRef>, repl: bo
let ctx_name = if repl { "<interactive input>" } else { fname.as_ref().map(|s| s.as_ref()).unwrap_or("<unknown>") }; let ctx_name = if repl { "<interactive input>" } else { fname.as_ref().map(|s| s.as_ref()).unwrap_or("<unknown>") };
let mut lexer = Lexer::new(src, fname.clone()); let mut lexer = Lexer::new(src, fname.clone());
lexer.lex()?; lexer.lex()?;
let mut parser = Parser::new(lexer.into_tokens(), repl); let tokens = lexer.into_tokens();
let mut parser = Parser::new(tokens);
let ast = parser.parse()?; let ast = parser.parse()?;
let environ; let environ;
if let Some(env) = env { if let Some(env) = env {

View file

@ -2,6 +2,19 @@ use std::rc::Rc;
use crate::{ParserError, Position, token::{Token, TokenType}}; use crate::{ParserError, Position, token::{Token, TokenType}};
fn is_in_base(c: char, base: u32) -> bool {
match (c, base) {
('0' | '1', 2) => true,
('0'..='5', 6) => true,
('0'..='7', 8) => true,
('0'..='9', 10) => true,
('0'..='9' | 'a' | 'b' | 'A' | 'B', 12) => true,
('0'..='9' | 'a'..='f' | 'A'..='F', 16) => true,
_ => false
}
}
pub struct Lexer { pub struct Lexer {
// name of file being lexed // name of file being lexed
filename: Option<Rc<str>>, filename: Option<Rc<str>>,
@ -220,6 +233,7 @@ impl Lexer {
_ => self.add_token(TokenType::Pipe, "|"), _ => self.add_token(TokenType::Pipe, "|"),
}, },
'~' => self.add_token(TokenType::Tilde, "~"), '~' => self.add_token(TokenType::Tilde, "~"),
'\\' => self.add_token(TokenType::Backslash, "\\"),
',' => self.add_token(TokenType::Comma, ","), ',' => self.add_token(TokenType::Comma, ","),
';' => self.add_token(TokenType::Semicolon, ";"), ';' => self.add_token(TokenType::Semicolon, ";"),
':' => self.add_token(TokenType::Colon, ":"), ':' => self.add_token(TokenType::Colon, ":"),
@ -256,7 +270,15 @@ impl Lexer {
'"' => self.string()?, '"' => self.string()?,
'\'' => self.char()?, '\'' => self.char()?,
' ' | '\t' | '\r' | '\n' => (), ' ' | '\t' | '\r' | '\n' => (),
'0'..='9' => self.number()?, '0' => match self.expect(&['b', 's', 'o', 'd', 'x']) {
Some('b') => self.number(2)?,
Some('s') => self.number(6)?,
Some('o') => self.number(8)?,
Some('d') => self.number(12)?,
Some('x') => self.number(16)?,
_ => self.number(10)?,
},
'0'..='9' => self.number(10)?,
'a'..='z' | 'A'..='Z' | '_' => self.ident()?, 'a'..='z' | 'A'..='Z' | '_' => self.ident()?,
c => return Err(self.mk_error(format!("Unexpected character: {}", c))) c => return Err(self.mk_error(format!("Unexpected character: {}", c)))
} }
@ -306,9 +328,9 @@ impl Lexer {
Ok(()) Ok(())
} }
fn number(&mut self) -> Result<(), ParserError> { fn number(&mut self, base: u32) -> Result<(), ParserError> {
let mut has_dot = false; let mut has_dot = false;
while !self.at_end() && (self.peek().is_numeric() || self.peek() == '.') { while !self.at_end() && (is_in_base(self.peek(), base) || self.peek() == '.') {
if self.peek() == '.' { if self.peek() == '.' {
if has_dot { if has_dot {
break; break;
@ -316,6 +338,9 @@ impl Lexer {
if self.peek_ahead(1) == Some('.') { if self.peek_ahead(1) == Some('.') {
break; break;
} }
if base != 10 {
return Err(self.mk_error("Numeric literals using bases other than 10 must be integers."))
}
has_dot = true; has_dot = true;
} }
} }
@ -334,10 +359,16 @@ impl Lexer {
Ok(num) => self.add_token(TokenType::Float(num), literal), Ok(num) => self.add_token(TokenType::Float(num), literal),
Err(e) => return Err(self.mk_error(format!("Error parsing float: {}", e))) Err(e) => return Err(self.mk_error(format!("Error parsing float: {}", e)))
} }
} else if base != 10 {
match i64::from_str_radix(&literal[2..literal.len()], base) {
Ok(num) => self.add_token(TokenType::Int(num), literal),
Err(e) => return Err(self.mk_error(format!("Error parsing integer: {}", e)))
}
} else { } else {
match literal.parse::<i64>() { match literal.parse::<i64>() {
Ok(num) => self.add_token(TokenType::Int(num), literal), Ok(num) => self.add_token(TokenType::Int(num), literal),
Err(e) => return Err(self.mk_error(format!("Error parsing float: {}", e))) Err(e) => return Err(self.mk_error(format!("Error parsing integer: {}", e)))
} }
} }
Ok(()) Ok(())

View file

@ -4,19 +4,18 @@ use crate::{token::{Token, TokenType, OpType}, ParserError, expr::{Stmt, Expr},
pub struct Parser { pub struct Parser {
tokens: Vec<Token>, tokens: Vec<Token>,
repl: bool,
idx: usize idx: usize
} }
impl Parser { impl Parser {
pub fn new(tokens: Vec<Token>, repl: bool) -> Self { pub fn new(tokens: Vec<Token>) -> Self {
Self { tokens, repl, idx: 0 } Self { tokens, idx: 0 }
} }
pub fn parse(&mut self) -> Result<Vec<Stmt>, ParserError> { pub fn parse(&mut self) -> Result<Vec<Stmt>, ParserError> {
let mut stmts = vec![]; let mut stmts = vec![];
while !self.at_end() { while !self.at_end() {
stmts.push(self.statement(!self.repl)?); stmts.push(self.statement()?);
} }
Ok(stmts) Ok(stmts)
} }
@ -96,14 +95,19 @@ impl Parser {
// // // //
////////////////// //////////////////
fn statement(&mut self, req_semicolon: bool) -> Result<Stmt, ParserError> { fn statement(&mut self) -> Result<Stmt, ParserError> {
let next_ty = &self.peek().ty; let next_ty = &self.peek().ty;
match next_ty { match next_ty {
TokenType::Semicolon => {
// skip lonely semicolon
self.next();
self.statement()
}
TokenType::Let => { TokenType::Let => {
// let statement // let statement
self.next(); self.next();
self.letstmt(req_semicolon) self.letstmt()
}, },
TokenType::LBrace => { TokenType::LBrace => {
// block // block
@ -127,16 +131,16 @@ impl Parser {
}, },
TokenType::Break => { TokenType::Break => {
let tok = self.next(); let tok = self.next();
self.terminate_stmt(Stmt::Break{ pos: tok.pos }, req_semicolon) self.terminate_stmt(Stmt::Break{ pos: tok.pos })
}, },
TokenType::Continue => { TokenType::Continue => {
let tok = self.next(); let tok = self.next();
self.terminate_stmt(Stmt::Continue{ pos: tok.pos }, req_semicolon) self.terminate_stmt(Stmt::Continue{ pos: tok.pos })
}, },
TokenType::Return => { TokenType::Return => {
let tok = self.next(); let tok = self.next();
let expr = self.assignment()?; let expr = self.assignment()?;
self.terminate_stmt(Stmt::Return{ pos: tok.pos, expr }, req_semicolon) self.terminate_stmt(Stmt::Return{ pos: tok.pos, expr })
}, },
TokenType::Fn => { TokenType::Fn => {
self.next(); self.next();
@ -149,39 +153,34 @@ impl Parser {
_ => { _ => {
// fallback to an expression terminated with a semicolon // fallback to an expression terminated with a semicolon
let expr = self.assignment()?; let expr = self.assignment()?;
self.terminate_stmt(Stmt::Expr{ expr }, req_semicolon) self.terminate_stmt(Stmt::Expr{ expr })
} }
} }
} }
fn terminate_stmt(&mut self, stmt: Stmt, req_semicolon: bool) -> Result<Stmt, ParserError> { fn terminate_stmt(&mut self, stmt: Stmt) -> Result<Stmt, ParserError> {
if self.at_end() { if self.at_end() {
if req_semicolon {
self.err_on_eof()?; self.err_on_eof()?;
} else{
return Ok(stmt)
}
} }
match self.expect(TokenType::Semicolon).0 { match self.expect(TokenType::Semicolon).0 {
true => Ok(stmt), true => Ok(stmt),
false if !req_semicolon => Ok(stmt),
false => Err(self.mk_error("Missing semicolon after statement")) false => Err(self.mk_error("Missing semicolon after statement"))
} }
} }
fn letstmt(&mut self, req_semicolon: bool) -> Result<Stmt, ParserError> { fn letstmt(&mut self) -> Result<Stmt, ParserError> {
let expr = self.assignment()?; let expr = self.assignment()?;
// must be followed by an assignment expression // must be followed by an assignment expression
if let Expr::Binary{lhs, rhs, op: Token{ty: TokenType::Equal,..}} = expr { if let Expr::Binary{lhs, rhs, op: Token{ty: TokenType::Equal,..}} = expr {
if let Expr::Ident{value: tok} = *lhs { if let Expr::Ident{value: tok} = *lhs {
self.terminate_stmt(Stmt::Let{lhs: tok, rhs: Some(*rhs)}, req_semicolon) self.terminate_stmt(Stmt::Let{lhs: tok, rhs: Some(*rhs)})
} else { } else {
Err(self.mk_error("Invalid expression after 'let'".to_owned())) Err(self.mk_error("Invalid expression after 'let'".to_owned()))
} }
} else if let Expr::Ident{value: tok} = expr { } else if let Expr::Ident{value: tok} = expr {
self.terminate_stmt(Stmt::Let{lhs: tok, rhs: None}, req_semicolon) self.terminate_stmt(Stmt::Let{lhs: tok, rhs: None})
} else { } else {
Err(self.mk_error("Invalid expression after 'let'".to_owned())) Err(self.mk_error("Invalid expression after 'let'".to_owned()))
} }
@ -192,7 +191,7 @@ impl Parser {
let mut ec = false; let mut ec = false;
loop { loop {
let condition = self.assignment()?; let condition = self.assignment()?;
let body = self.statement(true)?; let body = self.statement()?;
if_clauses.push((condition, body)); if_clauses.push((condition, body));
match self.peek().ty { match self.peek().ty {
TokenType::Elif => { self.next(); continue }, TokenType::Elif => { self.next(); continue },
@ -201,7 +200,7 @@ impl Parser {
} }
} }
let else_clause = if ec { let else_clause = if ec {
Some(Box::new(self.statement(true)?)) Some(Box::new(self.statement()?))
} else { } else {
None None
}; };
@ -222,7 +221,7 @@ impl Parser {
self.err_on_eof()?; self.err_on_eof()?;
let expr = self.assignment()?; let expr = self.assignment()?;
self.err_on_eof()?; self.err_on_eof()?;
let stmt = self.statement(true)?; let stmt = self.statement()?;
Ok(Stmt::For{ var, expr, stmt: Box::new(stmt), iter_pos: colon.pos }) Ok(Stmt::For{ var, expr, stmt: Box::new(stmt), iter_pos: colon.pos })
} else { } else {
Err(self.mk_error("Expected identifier after for")) Err(self.mk_error("Expected identifier after for"))
@ -233,7 +232,7 @@ impl Parser {
self.err_on_eof()?; self.err_on_eof()?;
let expr = self.assignment()?; let expr = self.assignment()?;
self.err_on_eof()?; self.err_on_eof()?;
let stmt = self.statement(true)?; let stmt = self.statement()?;
Ok(Stmt::While{ expr, stmt: Box::new(stmt) }) Ok(Stmt::While{ expr, stmt: Box::new(stmt) })
} }
@ -273,7 +272,7 @@ impl Parser {
fn block(&mut self) -> Result<Stmt, ParserError> { fn block(&mut self) -> Result<Stmt, ParserError> {
let mut stmts = vec![]; let mut stmts = vec![];
while !self.at_end() && self.peek().ty != TokenType::RBrace { while !self.at_end() && self.peek().ty != TokenType::RBrace {
stmts.push(self.statement(true)?) stmts.push(self.statement()?)
} }
self.err_on_eof()?; self.err_on_eof()?;
self.next(); self.next();
@ -357,7 +356,15 @@ impl Parser {
} }
fn comparison(&mut self) -> Result<Expr, ParserError> { fn comparison(&mut self) -> Result<Expr, ParserError> {
self.expr(OpType::Comparison, Self::additive) self.expr(OpType::Comparison, Self::bitwise_or)
}
fn bitwise_or(&mut self) -> Result<Expr, ParserError> {
self.expr(OpType::BitwiseOr, Self::bitwise_and)
}
fn bitwise_and(&mut self) -> Result<Expr, ParserError> {
self.expr(OpType::BitwiseAnd, Self::additive)
} }
fn additive(&mut self) -> Result<Expr, ParserError> { fn additive(&mut self) -> Result<Expr, ParserError> {
@ -422,9 +429,9 @@ impl Parser {
// unary: !x, -x // unary: !x, -x
fn unary(&mut self) -> Result<Expr, ParserError> { fn unary(&mut self) -> Result<Expr, ParserError> {
self.err_on_eof()?; self.err_on_eof()?;
if matches!(self.peek().ty, TokenType::Bang | TokenType::Minus) { if matches!(self.peek().ty, TokenType::Minus | TokenType::Bang | TokenType::Tilde) {
let op = self.next(); let op = self.next();
Ok(Expr::Unary { arg: Box::new(self.fieldaccess()?), op }) Ok(Expr::Unary { arg: Box::new(self.unary()?), op })
} else { } else {
self.fieldaccess() self.fieldaccess()
} }
@ -511,29 +518,26 @@ impl Parser {
// An identifier // An identifier
Ok(Expr::Ident { value: next }) Ok(Expr::Ident { value: next })
} else if next.ty == TokenType::LParen { } else if next.ty == TokenType::LParen {
// Special case for "boxed infix operators" // Grouping with parentheses
if !self.at_end() && self.peek().ty.is_infix_op() {
let op = self.next();
self.err_on_eof()?;
if self.peek().ty != TokenType::RParen {
return Err(self.mk_error("Expected right parenthesis after enclosed operator"))
}
self.next();
let func = Func::BuiltinClosure {
arg_count: 2,
func: Rc::new(move |args| {
eval_standard_binary(args[0].clone(), args[1].clone(), &op.ty, &op.pos)
})
};
return Ok(Expr::BoxedInfix { func })
}
// general case: parentheses as grouping symbols
let expr = self.assignment()?; let expr = self.assignment()?;
if self.at_end() || TokenType::RParen != self.next().ty { if self.at_end() || TokenType::RParen != self.next().ty {
Err(self.mk_error("Left parenthesis never closed")) Err(self.mk_error("Left parenthesis never closed"))
} else { } else {
Ok(expr) Ok(expr)
} }
} else if next.ty == TokenType::Backslash {
self.err_on_eof()?;
let op = self.next();
if !op.ty.is_infix_op() {
return Err(self.mk_error("Expected infix operator after backslash"))
}
let func = Func::BuiltinClosure {
arg_count: 2,
func: Rc::new(move |args| {
eval_standard_binary(args[0].clone(), args[1].clone(), &op.ty, &op.pos)
})
};
Ok(Expr::BoxedInfix { func })
} else if next.ty == TokenType::LBrack { } else if next.ty == TokenType::LBrack {
// list literal // list literal
let items = self.commalist(TokenType::RBrack, Self::assignment)?; let items = self.commalist(TokenType::RBrack, Self::assignment)?;

View file

@ -32,6 +32,7 @@ pub enum TokenType {
PipeColon, PipePoint, PipeQuestion, PipeAmper, PipeColon, PipePoint, PipeQuestion, PipeAmper,
PipeSlash, PipeBackslash, PipeDoubleSlash, PipeDoubleBackslash, PipeSlash, PipeBackslash, PipeDoubleSlash, PipeDoubleBackslash,
Backslash,
Comma, Semicolon, Colon, Comma, Semicolon, Colon,
@ -63,6 +64,9 @@ impl TokenType {
| Self::StarEqual | Self::SlashEqual | Self::DoubleSlashEqual | Self::StarEqual | Self::SlashEqual | Self::DoubleSlashEqual
| Self::CaretEqual | Self::PercentEqual => Some(OpType::Assignment), | Self::CaretEqual | Self::PercentEqual => Some(OpType::Assignment),
Self::Amper => Some(OpType::BitwiseAnd),
Self::Pipe => Some(OpType::BitwiseOr),
Self::DoubleAmper => Some(OpType::LogicalAnd), Self::DoubleAmper => Some(OpType::LogicalAnd),
Self::DoublePipe => Some(OpType::LogicalOr), Self::DoublePipe => Some(OpType::LogicalOr),
@ -83,6 +87,8 @@ impl TokenType {
| OpType::Multiplicative | OpType::Multiplicative
| OpType::Exponential | OpType::Exponential
| OpType::Comparison | OpType::Comparison
| OpType::BitwiseAnd
| OpType::BitwiseOr
// TODO | OpType::Pipeline // TODO | OpType::Pipeline
)) ))
} }
@ -90,7 +96,10 @@ impl TokenType {
#[derive(Clone,Copy,Debug,PartialEq,Eq)] #[derive(Clone,Copy,Debug,PartialEq,Eq)]
pub enum OpType { pub enum OpType {
Assignment, Comparison, Pipeline, Additive, Multiplicative, Exponential, LogicalAnd, LogicalOr Assignment, Comparison, Pipeline,
Additive, Multiplicative, Exponential,
LogicalAnd, LogicalOr,
BitwiseAnd, BitwiseOr,
} }
impl OpType { impl OpType {

View file

@ -480,7 +480,7 @@ value_from!(Map, RefCell<HashMap<Value,Value>>);
macro_rules! impl_numeric_op { macro_rules! impl_numeric_op {
($optrait:ty, $fnname:ident, { $($bonus:tt)* }) => { ($optrait:ty, $fnname:ident, $op:literal, { $($bonus:tt)* }) => {
impl $optrait for &Value { impl $optrait for &Value {
type Output = Result<Value, String>; type Output = Result<Value, String>;
fn $fnname(self, other: Self) -> Self::Output { fn $fnname(self, other: Self) -> Self::Output {
@ -504,14 +504,14 @@ macro_rules! impl_numeric_op {
(Rational(a), Complex(b)) => Ok(self::Complex::from(a.to_f64().ok_or(RATIO_CAST_FAIL)?).$fnname(b).into()), (Rational(a), Complex(b)) => Ok(self::Complex::from(a.to_f64().ok_or(RATIO_CAST_FAIL)?).$fnname(b).into()),
(Complex(a), Rational(b)) => Ok(a.$fnname(self::Complex::from(b.to_f64().ok_or(RATIO_CAST_FAIL)?)).into()), (Complex(a), Rational(b)) => Ok(a.$fnname(self::Complex::from(b.to_f64().ok_or(RATIO_CAST_FAIL)?)).into()),
(Complex(a), Complex(b)) => Ok(a.$fnname(b).into()), (Complex(a), Complex(b)) => Ok(a.$fnname(b).into()),
(lhs, rhs) => Err(format!("Unsupported operation '{}' between {} and {}", stringify!($fnname), lhs.repr(), rhs.repr())) (lhs, rhs) => Err(format!("Unsupported operation '{}' between {} and {}", $op, lhs.repr(), rhs.repr()))
} }
} }
} }
} }
} }
impl Neg for Value { impl Neg for &Value {
type Output = Result<Value, String>; type Output = Result<Value, String>;
fn neg(self) -> Self::Output { fn neg(self) -> Self::Output {
match self { match self {
@ -524,7 +524,7 @@ impl Neg for Value {
} }
} }
impl_numeric_op!(Add, add, { impl_numeric_op!(Add, add, "+", {
(String(a), String(b)) => Ok(((**a).to_owned() + b).into()), (String(a), String(b)) => Ok(((**a).to_owned() + b).into()),
(String(a), Char(c)) => { (String(a), Char(c)) => {
let mut s = (**a).to_owned(); let mut s = (**a).to_owned();
@ -543,20 +543,20 @@ impl_numeric_op!(Add, add, {
Ok(a.into()) Ok(a.into())
}, },
}); });
impl_numeric_op!(Sub, sub, {}); impl_numeric_op!(Sub, sub, "-", {});
impl_numeric_op!(Mul, mul, { impl_numeric_op!(Mul, mul, "*", {
(String(a), Int(b)) | (Int(b), String(a)) (String(a), Int(b)) | (Int(b), String(a))
=> Ok(Value::from(a.chars().cycle().take(a.chars().count()*(*b as usize)).collect::<std::string::String>())), => Ok(Value::from(a.chars().cycle().take(a.chars().count()*(*b as usize)).collect::<std::string::String>())),
(List(a), Int(b)) | (Int(b), List(a)) (List(a), Int(b)) | (Int(b), List(a))
=> Ok(Value::from(a.borrow().iter().cycle().take(a.borrow().len()*(*b as usize)).cloned().collect::<Vec<Value>>())), => Ok(Value::from(a.borrow().iter().cycle().take(a.borrow().len()*(*b as usize)).cloned().collect::<Vec<Value>>())),
}); });
impl_numeric_op!(Div, div, { impl_numeric_op!(Div, div, "/", {
(Int(_), Int(b)) if *b == 0 => Err("Integer division by zero".into()), (Int(_), Int(b)) if *b == 0 => Err("Integer division by zero".into()),
(Rational(_), Int(b)) if *b == 0 => Err("Rational division by zero".into()), (Rational(_), Int(b)) if *b == 0 => Err("Rational division by zero".into()),
(Int(_), Rational(b)) if b.is_zero() => Err("Rational division by zero".into()), (Int(_), Rational(b)) if b.is_zero() => Err("Rational division by zero".into()),
(Rational(_), Rational(b)) if b.is_zero() => Err("Rational division by zero".into()), (Rational(_), Rational(b)) if b.is_zero() => Err("Rational division by zero".into()),
}); });
impl_numeric_op!(Rem, rem, { impl_numeric_op!(Rem, rem, "%", {
(Int(_), Int(b)) if *b == 0 => Err("Integer modulo by zero".into()), (Int(_), Int(b)) if *b == 0 => Err("Integer modulo by zero".into()),
(Rational(_), Int(b)) if *b == 0 => Err("Rational modulo by zero".into()), (Rational(_), Int(b)) if *b == 0 => Err("Rational modulo by zero".into()),
(Int(_), Rational(b)) if b.is_zero() => Err("Rational modulo by zero".into()), (Int(_), Rational(b)) if b.is_zero() => Err("Rational modulo by zero".into()),
@ -594,7 +594,46 @@ impl Pow<&Value> for &Value {
(Rational(a), Complex(b)) => Ok(self::Complex::from(a.to_f64().ok_or(RATIO_CAST_FAIL)?).pow(b).into()), (Rational(a), Complex(b)) => Ok(self::Complex::from(a.to_f64().ok_or(RATIO_CAST_FAIL)?).pow(b).into()),
(Complex(a), Rational(b)) => Ok(a.pow(self::Complex::from(b.to_f64().ok_or(RATIO_CAST_FAIL)?)).into()), (Complex(a), Rational(b)) => Ok(a.pow(self::Complex::from(b.to_f64().ok_or(RATIO_CAST_FAIL)?)).into()),
(Complex(a), Complex(b)) => Ok(a.pow(b).into()), (Complex(a), Complex(b)) => Ok(a.pow(b).into()),
(lhs, rhs) => Err(format!("Unsupported operation 'pow' between {} and {}", lhs.repr(), rhs.repr())) (lhs, rhs) => Err(format!("Unsupported operation '^' between {} and {}", lhs.repr(), rhs.repr()))
}
}
}
impl BitAnd for &Value {
type Output = Result<Value, String>;
fn bitand(self, other: Self) -> Self::Output {
use Value::*;
match (self, other) {
(Int(a), Int(b)) => Ok(Int(a & b)),
(Bool(a), Bool(b)) => Ok(Bool(a & b)),
_ => Err(format!("Unsupported operation '&' between {} and {}", self.repr(), other.repr()))
}
}
}
impl BitOr for &Value {
type Output = Result<Value, String>;
fn bitor(self, other: Self) -> Self::Output {
use Value::*;
match (self, other) {
(Int(a), Int(b)) => Ok(Int(a | b)),
(Bool(a), Bool(b)) => Ok(Bool(a | b)),
_ => Err(format!("Unsupported operation '|' between {} and {}", self.repr(), other.repr()))
}
}
}
impl Not for &Value {
type Output = Result<Value, String>;
fn not(self) -> Self::Output {
use Value::*;
match self {
Int(a) => Ok(Int(!a)),
Bool(a) => Ok(Bool(!a)),
_ => Err(format!("Unsupported operation '~' for {}", self.repr())),
} }
} }
} }

View file

@ -1,3 +1,3 @@
while true { while true {
print(input()); println(input());
} }

View file

@ -1,11 +1,29 @@
fn factorial(n) { fn fact_recursive(n) {
if n <= 1 { if n <= 1 {
return 1; return 1;
} else {
return n * factorial(n-1);
} }
return n * fact_recursive(n-1);
} }
for n: 0..10 { fn fact_imperative(n) {
println(factorial(n)); if n <= 1 {
return 1;
}
let res = 1;
for i: 1..=n {
res *= i;
}
return res;
} }
fn fact_pipeline(n) {
if n <= 1 {
return 1;
}
return 1..=n |// \*;
}
# 10*9*8*7*6*5*4*3*2*1 = 10
println(fact_recursive(10));
println(fact_imperative(10));
println(fact_pipeline(10));

View file

@ -1,6 +1,6 @@
# compute the primorial of n - the product of all primes <= n # compute the primorial of n - the product of all primes <= n
fn primorial(n) { fn primorial(n) {
return 2..(n+1) |? is_prime |// fn(x,y) { return x*y; }; return 2..(n+1) |? is_prime |// \*;
} }
println(primorial(10)); # 2*3*5*7 = 210 println(primorial(10)); # 2*3*5*7 = 210