Compare commits
3 Commits
fb5eacea02
...
bc7c77846f
| Author | SHA1 | Date | |
|---|---|---|---|
|
bc7c77846f
|
|||
|
76add65235
|
|||
|
e56414c251
|
@@ -525,6 +525,28 @@ impl<'a> Compiler<'a> {
|
||||
temp_name: Some(result_name),
|
||||
}))
|
||||
}
|
||||
Expression::BitwiseNot(inner_expr) => {
|
||||
// Compile bitwise NOT using the NOT instruction
|
||||
let (inner_str, cleanup) = self.compile_operand(*inner_expr, scope)?;
|
||||
let result_name = self.next_temp_name();
|
||||
let result_loc =
|
||||
scope.add_variable(result_name.clone(), LocationRequest::Temp, None)?;
|
||||
let result_reg = self.resolve_register(&result_loc)?;
|
||||
|
||||
self.write_instruction(
|
||||
Instruction::Not(Operand::Register(result_reg), inner_str),
|
||||
Some(expr.span),
|
||||
)?;
|
||||
|
||||
if let Some(name) = cleanup {
|
||||
scope.free_temp(name, None)?;
|
||||
}
|
||||
|
||||
Ok(Some(CompileLocation {
|
||||
location: result_loc,
|
||||
temp_name: Some(result_name),
|
||||
}))
|
||||
}
|
||||
Expression::TupleDeclaration(tuple_decl) => {
|
||||
self.expression_tuple_declaration(tuple_decl.node, scope)?;
|
||||
Ok(None)
|
||||
@@ -889,6 +911,32 @@ impl<'a> Compiler<'a> {
|
||||
}
|
||||
(var_loc, None)
|
||||
}
|
||||
Expression::BitwiseNot(_) => {
|
||||
// Compile the bitwise NOT expression
|
||||
let result = self.expression(expr, scope)?;
|
||||
let var_loc = scope.add_variable(
|
||||
name_str.clone(),
|
||||
LocationRequest::Persist,
|
||||
Some(name_span),
|
||||
)?;
|
||||
|
||||
if let Some(res) = result {
|
||||
// Move result from temp to new persistent variable
|
||||
let result_reg = self.resolve_register(&res.location)?;
|
||||
self.emit_variable_assignment(&var_loc, Operand::Register(result_reg))?;
|
||||
|
||||
// Free the temp result
|
||||
if let Some(name) = res.temp_name {
|
||||
scope.free_temp(name, None)?;
|
||||
}
|
||||
} else {
|
||||
return Err(Error::Unknown(
|
||||
format!("`{name_str}` bitwise NOT expression did not produce a value"),
|
||||
Some(name_span),
|
||||
));
|
||||
}
|
||||
(var_loc, None)
|
||||
}
|
||||
_ => {
|
||||
return Err(Error::Unknown(
|
||||
format!("`{name_str}` declaration of this type is not supported/implemented."),
|
||||
@@ -2114,13 +2162,36 @@ impl<'a> Compiler<'a> {
|
||||
scope: &mut VariableScope<'a, '_>,
|
||||
) -> Result<CompileLocation<'a>, Error<'a>> {
|
||||
fn fold_binary_expression<'a>(expr: &BinaryExpression<'a>) -> Option<Number> {
|
||||
fn number_to_i64(n: Number) -> Option<i64> {
|
||||
match n {
|
||||
Number::Integer(i, _) => i64::try_from(i).ok(),
|
||||
Number::Decimal(d, _) => {
|
||||
// Convert decimal to i64 by truncating
|
||||
let int_part = d.trunc();
|
||||
i64::try_from(int_part.mantissa() / 10_i128.pow(int_part.scale())).ok()
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
fn i64_to_number(i: i64) -> Number {
|
||||
Number::Integer(i as i128, Unit::None)
|
||||
}
|
||||
|
||||
let (lhs, rhs) = match &expr {
|
||||
BinaryExpression::Add(l, r)
|
||||
| BinaryExpression::Subtract(l, r)
|
||||
| BinaryExpression::Multiply(l, r)
|
||||
| BinaryExpression::Divide(l, r)
|
||||
| BinaryExpression::Exponent(l, r)
|
||||
| BinaryExpression::Modulo(l, r) => (fold_expression(l)?, fold_expression(r)?),
|
||||
| BinaryExpression::Modulo(l, r)
|
||||
| BinaryExpression::BitwiseAnd(l, r)
|
||||
| BinaryExpression::BitwiseOr(l, r)
|
||||
| BinaryExpression::BitwiseXor(l, r)
|
||||
| BinaryExpression::LeftShift(l, r)
|
||||
| BinaryExpression::RightShiftArithmetic(l, r)
|
||||
| BinaryExpression::RightShiftLogical(l, r) => {
|
||||
(fold_expression(l)?, fold_expression(r)?)
|
||||
}
|
||||
};
|
||||
|
||||
match expr {
|
||||
@@ -2129,7 +2200,37 @@ impl<'a> Compiler<'a> {
|
||||
BinaryExpression::Multiply(..) => Some(lhs * rhs),
|
||||
BinaryExpression::Divide(..) => Some(lhs / rhs), // Watch out for div by zero panics!
|
||||
BinaryExpression::Modulo(..) => Some(lhs % rhs),
|
||||
_ => None, // Handle Exponent separately or implement pow
|
||||
BinaryExpression::BitwiseAnd(..) => {
|
||||
let lhs_int = number_to_i64(lhs)?;
|
||||
let rhs_int = number_to_i64(rhs)?;
|
||||
Some(i64_to_number(lhs_int & rhs_int))
|
||||
}
|
||||
BinaryExpression::BitwiseOr(..) => {
|
||||
let lhs_int = number_to_i64(lhs)?;
|
||||
let rhs_int = number_to_i64(rhs)?;
|
||||
Some(i64_to_number(lhs_int | rhs_int))
|
||||
}
|
||||
BinaryExpression::BitwiseXor(..) => {
|
||||
let lhs_int = number_to_i64(lhs)?;
|
||||
let rhs_int = number_to_i64(rhs)?;
|
||||
Some(i64_to_number(lhs_int ^ rhs_int))
|
||||
}
|
||||
BinaryExpression::LeftShift(..) => {
|
||||
let lhs_int = number_to_i64(lhs)?;
|
||||
let rhs_int = number_to_i64(rhs)?;
|
||||
Some(i64_to_number(lhs_int << rhs_int))
|
||||
}
|
||||
BinaryExpression::RightShiftArithmetic(..) => {
|
||||
let lhs_int = number_to_i64(lhs)?;
|
||||
let rhs_int = number_to_i64(rhs)?;
|
||||
Some(i64_to_number(lhs_int >> rhs_int))
|
||||
}
|
||||
BinaryExpression::RightShiftLogical(..) => {
|
||||
let lhs_int = number_to_i64(lhs)?;
|
||||
let rhs_int = number_to_i64(rhs)?;
|
||||
Some(i64_to_number(lhs_int >> rhs_int))
|
||||
}
|
||||
_ => None, // Exponent not handled in compile-time folding
|
||||
}
|
||||
}
|
||||
|
||||
@@ -2189,6 +2290,24 @@ impl<'a> Compiler<'a> {
|
||||
BinaryExpression::Modulo(l, r) => {
|
||||
(|into, lhs, rhs| Instruction::Mod(into, lhs, rhs), l, r)
|
||||
}
|
||||
BinaryExpression::BitwiseAnd(l, r) => {
|
||||
(|into, lhs, rhs| Instruction::And(into, lhs, rhs), l, r)
|
||||
}
|
||||
BinaryExpression::BitwiseOr(l, r) => {
|
||||
(|into, lhs, rhs| Instruction::Or(into, lhs, rhs), l, r)
|
||||
}
|
||||
BinaryExpression::BitwiseXor(l, r) => {
|
||||
(|into, lhs, rhs| Instruction::Xor(into, lhs, rhs), l, r)
|
||||
}
|
||||
BinaryExpression::LeftShift(l, r) => {
|
||||
(|into, lhs, rhs| Instruction::Sll(into, lhs, rhs), l, r)
|
||||
}
|
||||
BinaryExpression::RightShiftArithmetic(l, r) => {
|
||||
(|into, lhs, rhs| Instruction::Sra(into, lhs, rhs), l, r)
|
||||
}
|
||||
BinaryExpression::RightShiftLogical(l, r) => {
|
||||
(|into, lhs, rhs| Instruction::Srl(into, lhs, rhs), l, r)
|
||||
}
|
||||
};
|
||||
|
||||
let span = Self::merge_spans(left_expr.span, right_expr.span);
|
||||
|
||||
@@ -232,12 +232,22 @@ pub enum Instruction<'a> {
|
||||
/// `sle dst a b` - Set if Less or Equal
|
||||
SetLe(Operand<'a>, Operand<'a>, Operand<'a>),
|
||||
|
||||
/// `and dst a b` - Logical AND
|
||||
/// `and dst a b` - Bitwise AND
|
||||
And(Operand<'a>, Operand<'a>, Operand<'a>),
|
||||
/// `or dst a b` - Logical OR
|
||||
/// `or dst a b` - Bitwise OR
|
||||
Or(Operand<'a>, Operand<'a>, Operand<'a>),
|
||||
/// `xor dst a b` - Logical XOR
|
||||
/// `xor dst a b` - Bitwise XOR
|
||||
Xor(Operand<'a>, Operand<'a>, Operand<'a>),
|
||||
/// `nor dst a b` - Bitwise NOR
|
||||
Nor(Operand<'a>, Operand<'a>, Operand<'a>),
|
||||
/// `not dst a` - Bitwise NOT
|
||||
Not(Operand<'a>, Operand<'a>),
|
||||
/// `sll dst a b` - Logical Left Shift
|
||||
Sll(Operand<'a>, Operand<'a>, Operand<'a>),
|
||||
/// `sra dst a b` - Arithmetic Right Shift
|
||||
Sra(Operand<'a>, Operand<'a>, Operand<'a>),
|
||||
/// `srl dst a b` - Logical Right Shift
|
||||
Srl(Operand<'a>, Operand<'a>, Operand<'a>),
|
||||
|
||||
/// `push val` - Push to Stack
|
||||
Push(Operand<'a>),
|
||||
@@ -338,6 +348,11 @@ impl<'a> fmt::Display for Instruction<'a> {
|
||||
Instruction::And(dst, a, b) => write!(f, "and {} {} {}", dst, a, b),
|
||||
Instruction::Or(dst, a, b) => write!(f, "or {} {} {}", dst, a, b),
|
||||
Instruction::Xor(dst, a, b) => write!(f, "xor {} {} {}", dst, a, b),
|
||||
Instruction::Nor(dst, a, b) => write!(f, "nor {} {} {}", dst, a, b),
|
||||
Instruction::Not(dst, a) => write!(f, "not {} {}", dst, a),
|
||||
Instruction::Sll(dst, a, b) => write!(f, "sll {} {} {}", dst, a, b),
|
||||
Instruction::Sra(dst, a, b) => write!(f, "sra {} {} {}", dst, a, b),
|
||||
Instruction::Srl(dst, a, b) => write!(f, "srl {} {} {}", dst, a, b),
|
||||
Instruction::Push(val) => write!(f, "push {}", val),
|
||||
Instruction::Pop(dst) => write!(f, "pop {}", dst),
|
||||
Instruction::Peek(dst) => write!(f, "peek {}", dst),
|
||||
|
||||
@@ -294,12 +294,12 @@ impl<'a> Parser<'a> {
|
||||
// Handle Infix operators (Binary, Logical, Assignment)
|
||||
if self_matches_peek!(
|
||||
self,
|
||||
TokenType::Symbol(s) if s.is_operator() || s.is_comparison() || s.is_logical() || matches!(s, Symbol::Assign | Symbol::Question)
|
||||
TokenType::Symbol(s) if s.is_operator() || s.is_comparison() || s.is_logical() || s.is_bitwise() || matches!(s, Symbol::Assign | Symbol::Question)
|
||||
) {
|
||||
return Ok(Some(self.infix(lhs)?));
|
||||
} else if self_matches_current!(
|
||||
self,
|
||||
TokenType::Symbol(s) if s.is_operator() || s.is_comparison() || s.is_logical() || matches!(s, Symbol::Assign | Symbol::Question)
|
||||
TokenType::Symbol(s) if s.is_operator() || s.is_comparison() || s.is_logical() || s.is_bitwise() || matches!(s, Symbol::Assign | Symbol::Question)
|
||||
) {
|
||||
self.tokenizer.seek(SeekFrom::Current(-1))?;
|
||||
return Ok(Some(self.infix(lhs)?));
|
||||
@@ -608,6 +608,23 @@ impl<'a> Parser<'a> {
|
||||
})
|
||||
}
|
||||
|
||||
TokenType::Symbol(Symbol::BitwiseNot) => {
|
||||
let start_span = self.current_span();
|
||||
self.assign_next()?;
|
||||
let inner_expr = self.unary()?.ok_or(Error::UnexpectedEOF)?;
|
||||
let inner_with_postfix = self.parse_postfix(inner_expr)?;
|
||||
let combined_span = Span {
|
||||
start_line: start_span.start_line,
|
||||
start_col: start_span.start_col,
|
||||
end_line: inner_with_postfix.span.end_line,
|
||||
end_col: inner_with_postfix.span.end_col,
|
||||
};
|
||||
Some(Spanned {
|
||||
span: combined_span,
|
||||
node: Expression::BitwiseNot(boxed!(inner_with_postfix)),
|
||||
})
|
||||
}
|
||||
|
||||
_ => {
|
||||
return Err(Error::UnexpectedToken(
|
||||
self.current_span(),
|
||||
@@ -699,6 +716,20 @@ impl<'a> Parser<'a> {
|
||||
}),
|
||||
}
|
||||
}
|
||||
TokenType::Symbol(Symbol::BitwiseNot) => {
|
||||
self.assign_next()?;
|
||||
let inner = self.get_infix_child_node()?;
|
||||
let span = Span {
|
||||
start_line: start_span.start_line,
|
||||
start_col: start_span.start_col,
|
||||
end_line: inner.span.end_line,
|
||||
end_col: inner.span.end_col,
|
||||
};
|
||||
Spanned {
|
||||
span,
|
||||
node: Expression::BitwiseNot(boxed!(inner)),
|
||||
}
|
||||
}
|
||||
_ => {
|
||||
return Err(Error::UnexpectedToken(
|
||||
self.current_span(),
|
||||
@@ -777,6 +808,7 @@ impl<'a> Parser<'a> {
|
||||
| Expression::Variable(_)
|
||||
| Expression::Ternary(_)
|
||||
| Expression::Negation(_)
|
||||
| Expression::BitwiseNot(_)
|
||||
| Expression::MemberAccess(_)
|
||||
| Expression::MethodCall(_)
|
||||
| Expression::Tuple(_) => {}
|
||||
@@ -796,7 +828,7 @@ impl<'a> Parser<'a> {
|
||||
// Include Assign in the operator loop
|
||||
while token_matches!(
|
||||
temp_token,
|
||||
TokenType::Symbol(s) if s.is_operator() || s.is_comparison() || s.is_logical() || matches!(s, Symbol::Assign | Symbol::Question | Symbol::Colon)
|
||||
TokenType::Symbol(s) if s.is_operator() || s.is_comparison() || s.is_logical() || s.is_bitwise() || matches!(s, Symbol::Assign | Symbol::Question | Symbol::Colon)
|
||||
) {
|
||||
let operator = match temp_token.token_type {
|
||||
TokenType::Symbol(s) => s,
|
||||
@@ -869,6 +901,24 @@ impl<'a> Parser<'a> {
|
||||
Symbol::Minus => {
|
||||
BinaryExpression::Subtract(boxed!(left), boxed!(right))
|
||||
}
|
||||
Symbol::LeftShift => {
|
||||
BinaryExpression::LeftShift(boxed!(left), boxed!(right))
|
||||
}
|
||||
Symbol::RightShiftArithmetic => {
|
||||
BinaryExpression::RightShiftArithmetic(boxed!(left), boxed!(right))
|
||||
}
|
||||
Symbol::RightShiftLogical => {
|
||||
BinaryExpression::RightShiftLogical(boxed!(left), boxed!(right))
|
||||
}
|
||||
Symbol::BitwiseAnd => {
|
||||
BinaryExpression::BitwiseAnd(boxed!(left), boxed!(right))
|
||||
}
|
||||
Symbol::BitwiseOr => {
|
||||
BinaryExpression::BitwiseOr(boxed!(left), boxed!(right))
|
||||
}
|
||||
Symbol::Caret => {
|
||||
BinaryExpression::BitwiseXor(boxed!(left), boxed!(right))
|
||||
}
|
||||
_ => unreachable!(),
|
||||
};
|
||||
|
||||
@@ -895,7 +945,22 @@ impl<'a> Parser<'a> {
|
||||
// --- PRECEDENCE LEVEL 3: Additive (+, -) ---
|
||||
process_binary_ops!(Symbol::Plus | Symbol::Minus, BinaryExpression);
|
||||
|
||||
// --- PRECEDENCE LEVEL 4: Comparison (<, >, <=, >=) ---
|
||||
// --- PRECEDENCE LEVEL 4: Shift Operations (<<, >>, >>>) ---
|
||||
process_binary_ops!(
|
||||
Symbol::LeftShift | Symbol::RightShiftArithmetic | Symbol::RightShiftLogical,
|
||||
BinaryExpression
|
||||
);
|
||||
|
||||
// --- PRECEDENCE LEVEL 5: Bitwise AND (&) ---
|
||||
process_binary_ops!(Symbol::BitwiseAnd, BinaryExpression);
|
||||
|
||||
// --- PRECEDENCE LEVEL 6: Bitwise XOR (^) ---
|
||||
process_binary_ops!(Symbol::Caret, BinaryExpression);
|
||||
|
||||
// --- PRECEDENCE LEVEL 7: Bitwise OR (|) ---
|
||||
process_binary_ops!(Symbol::BitwiseOr, BinaryExpression);
|
||||
|
||||
// --- PRECEDENCE LEVEL 8: Comparison (<, >, <=, >=) ---
|
||||
let mut current_iteration = 0;
|
||||
for (i, operator) in operators.iter().enumerate() {
|
||||
if operator.is_comparison() && !matches!(operator, Symbol::Equal | Symbol::NotEqual) {
|
||||
|
||||
@@ -45,6 +45,12 @@ pub enum BinaryExpression<'a> {
|
||||
Subtract(Box<Spanned<Expression<'a>>>, Box<Spanned<Expression<'a>>>),
|
||||
Exponent(Box<Spanned<Expression<'a>>>, Box<Spanned<Expression<'a>>>),
|
||||
Modulo(Box<Spanned<Expression<'a>>>, Box<Spanned<Expression<'a>>>),
|
||||
BitwiseAnd(Box<Spanned<Expression<'a>>>, Box<Spanned<Expression<'a>>>),
|
||||
BitwiseOr(Box<Spanned<Expression<'a>>>, Box<Spanned<Expression<'a>>>),
|
||||
BitwiseXor(Box<Spanned<Expression<'a>>>, Box<Spanned<Expression<'a>>>),
|
||||
LeftShift(Box<Spanned<Expression<'a>>>, Box<Spanned<Expression<'a>>>),
|
||||
RightShiftArithmetic(Box<Spanned<Expression<'a>>>, Box<Spanned<Expression<'a>>>),
|
||||
RightShiftLogical(Box<Spanned<Expression<'a>>>, Box<Spanned<Expression<'a>>>),
|
||||
}
|
||||
|
||||
impl<'a> std::fmt::Display for BinaryExpression<'a> {
|
||||
@@ -56,6 +62,12 @@ impl<'a> std::fmt::Display for BinaryExpression<'a> {
|
||||
BinaryExpression::Subtract(l, r) => write!(f, "({} - {})", l, r),
|
||||
BinaryExpression::Exponent(l, r) => write!(f, "({} ** {})", l, r),
|
||||
BinaryExpression::Modulo(l, r) => write!(f, "({} % {})", l, r),
|
||||
BinaryExpression::BitwiseAnd(l, r) => write!(f, "({} & {})", l, r),
|
||||
BinaryExpression::BitwiseOr(l, r) => write!(f, "({} | {})", l, r),
|
||||
BinaryExpression::BitwiseXor(l, r) => write!(f, "({} ^ {})", l, r),
|
||||
BinaryExpression::LeftShift(l, r) => write!(f, "({} << {})", l, r),
|
||||
BinaryExpression::RightShiftArithmetic(l, r) => write!(f, "({} >> {})", l, r),
|
||||
BinaryExpression::RightShiftLogical(l, r) => write!(f, "({} >>> {})", l, r),
|
||||
}
|
||||
}
|
||||
}
|
||||
@@ -367,6 +379,7 @@ pub enum Expression<'a> {
|
||||
Binary(Spanned<BinaryExpression<'a>>),
|
||||
Block(Spanned<BlockExpression<'a>>),
|
||||
Break(Span),
|
||||
BitwiseNot(Box<Spanned<Expression<'a>>>),
|
||||
ConstDeclaration(Spanned<ConstDeclarationExpression<'a>>),
|
||||
Continue(Span),
|
||||
Declaration(Spanned<Cow<'a, str>>, Box<Spanned<Expression<'a>>>),
|
||||
@@ -398,6 +411,7 @@ impl<'a> std::fmt::Display for Expression<'a> {
|
||||
Expression::Binary(e) => write!(f, "{}", e),
|
||||
Expression::Block(e) => write!(f, "{}", e),
|
||||
Expression::Break(_) => write!(f, "break"),
|
||||
Expression::BitwiseNot(e) => write!(f, "(~{})", e),
|
||||
Expression::ConstDeclaration(e) => write!(f, "{}", e),
|
||||
Expression::Continue(_) => write!(f, "continue"),
|
||||
Expression::Declaration(id, e) => write!(f, "(let {} = {})", id, e),
|
||||
@@ -439,4 +453,3 @@ impl<'a> std::fmt::Display for Expression<'a> {
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
@@ -135,6 +135,9 @@ pub enum TokenType<'a> {
|
||||
/// Represents a string token
|
||||
String(Cow<'a, str>),
|
||||
|
||||
#[regex(r"0[xX][0-9a-fA-F][0-9a-fA-F_]*", parse_number)]
|
||||
#[regex(r"0[oO][0-7][0-7_]*", parse_number)]
|
||||
#[regex(r"0[bB][01][01_]*", parse_number)]
|
||||
#[regex(r"[0-9][0-9_]*(\.[0-9][0-9_]*)?([cfk])?", parse_number)]
|
||||
/// Represents a number token
|
||||
Number(Number),
|
||||
@@ -172,6 +175,23 @@ pub enum TokenType<'a> {
|
||||
#[token(";", symbol!(Semicolon))]
|
||||
#[token(":", symbol!(Colon))]
|
||||
#[token(",", symbol!(Comma))]
|
||||
#[token("?", symbol!(Question))]
|
||||
#[token(".", symbol!(Dot))]
|
||||
#[token("%", symbol!(Percent))]
|
||||
#[token("~", symbol!(BitwiseNot))]
|
||||
// Multi-character tokens must be defined before their single-character prefixes
|
||||
// For tokens like >> and >>>, define >>> before >> to ensure correct matching
|
||||
#[token(">>>", symbol!(RightShiftLogical))]
|
||||
#[token(">>", symbol!(RightShiftArithmetic))]
|
||||
#[token("<<", symbol!(LeftShift))]
|
||||
#[token("==", symbol!(Equal))]
|
||||
#[token("!=", symbol!(NotEqual))]
|
||||
#[token("&&", symbol!(LogicalAnd))]
|
||||
#[token("||", symbol!(LogicalOr))]
|
||||
#[token("<=", symbol!(LessThanOrEqual))]
|
||||
#[token(">=", symbol!(GreaterThanOrEqual))]
|
||||
#[token("**", symbol!(Exp))]
|
||||
// Single-character tokens
|
||||
#[token("+", symbol!(Plus))]
|
||||
#[token("-", symbol!(Minus))]
|
||||
#[token("*", symbol!(Asterisk))]
|
||||
@@ -180,17 +200,9 @@ pub enum TokenType<'a> {
|
||||
#[token(">", symbol!(GreaterThan))]
|
||||
#[token("=", symbol!(Assign))]
|
||||
#[token("!", symbol!(LogicalNot))]
|
||||
#[token(".", symbol!(Dot))]
|
||||
#[token("^", symbol!(Caret))]
|
||||
#[token("%", symbol!(Percent))]
|
||||
#[token("?", symbol!(Question))]
|
||||
#[token("==", symbol!(Equal))]
|
||||
#[token("!=", symbol!(NotEqual))]
|
||||
#[token("&&", symbol!(LogicalAnd))]
|
||||
#[token("||", symbol!(LogicalOr))]
|
||||
#[token("<=", symbol!(LessThanOrEqual))]
|
||||
#[token(">=", symbol!(GreaterThanOrEqual))]
|
||||
#[token("**", symbol!(Exp))]
|
||||
#[token("&", symbol!(BitwiseAnd))]
|
||||
#[token("|", symbol!(BitwiseOr))]
|
||||
/// Represents a symbol token
|
||||
Symbol(Symbol),
|
||||
|
||||
@@ -221,44 +233,75 @@ pub enum Comment<'a> {
|
||||
|
||||
fn parse_number<'a>(lexer: &mut Lexer<'a, TokenType<'a>>) -> Result<Number, LexError> {
|
||||
let slice = lexer.slice();
|
||||
let last_char = slice.chars().last().unwrap_or_default();
|
||||
let (num_str, suffix) = match last_char {
|
||||
'c' | 'k' | 'f' => (&slice[..slice.len() - 1], Some(last_char)),
|
||||
_ => (slice, None),
|
||||
};
|
||||
|
||||
let clean_str = if num_str.contains('_') {
|
||||
num_str.replace('_', "")
|
||||
} else {
|
||||
num_str.to_string()
|
||||
};
|
||||
|
||||
let line = lexer.extras.line_count;
|
||||
let mut span = lexer.span();
|
||||
span.end -= lexer.extras.line_start_index;
|
||||
span.start -= lexer.extras.line_start_index;
|
||||
|
||||
let unit = match suffix {
|
||||
Some('c') => Unit::Celsius,
|
||||
Some('f') => Unit::Fahrenheit,
|
||||
Some('k') => Unit::Kelvin,
|
||||
_ => Unit::None,
|
||||
};
|
||||
|
||||
if clean_str.contains('.') {
|
||||
Ok(Number::Decimal(
|
||||
clean_str
|
||||
.parse::<Decimal>()
|
||||
// Determine the base and parse accordingly
|
||||
if slice.starts_with("0x") || slice.starts_with("0X") {
|
||||
// Hexadecimal - no temperature suffix allowed
|
||||
let clean_str = slice[2..].replace('_', "");
|
||||
Ok(Number::Integer(
|
||||
i128::from_str_radix(&clean_str, 16)
|
||||
.map_err(|_| LexError::NumberParse(line, span, slice.to_string()))?,
|
||||
unit,
|
||||
Unit::None,
|
||||
))
|
||||
} else if slice.starts_with("0o") || slice.starts_with("0O") {
|
||||
// Octal - no temperature suffix allowed
|
||||
let clean_str = slice[2..].replace('_', "");
|
||||
Ok(Number::Integer(
|
||||
i128::from_str_radix(&clean_str, 8)
|
||||
.map_err(|_| LexError::NumberParse(line, span, slice.to_string()))?,
|
||||
Unit::None,
|
||||
))
|
||||
} else if slice.starts_with("0b") || slice.starts_with("0B") {
|
||||
// Binary - no temperature suffix allowed
|
||||
let clean_str = slice[2..].replace('_', "");
|
||||
Ok(Number::Integer(
|
||||
i128::from_str_radix(&clean_str, 2)
|
||||
.map_err(|_| LexError::NumberParse(line, span, slice.to_string()))?,
|
||||
Unit::None,
|
||||
))
|
||||
} else {
|
||||
Ok(Number::Integer(
|
||||
clean_str
|
||||
.parse::<i128>()
|
||||
.map_err(|_| LexError::NumberParse(line, span, slice.to_string()))?,
|
||||
unit,
|
||||
))
|
||||
// Decimal (with optional temperature suffix)
|
||||
let last_char = slice.chars().last().unwrap_or_default();
|
||||
let (num_str, suffix) = match last_char {
|
||||
'c' | 'k' | 'f' => (&slice[..slice.len() - 1], Some(last_char)),
|
||||
_ => (slice, None),
|
||||
};
|
||||
|
||||
let clean_str = if num_str.contains('_') {
|
||||
num_str.replace('_', "")
|
||||
} else {
|
||||
num_str.to_string()
|
||||
};
|
||||
|
||||
let unit = match suffix {
|
||||
Some('c') => Unit::Celsius,
|
||||
Some('f') => Unit::Fahrenheit,
|
||||
Some('k') => Unit::Kelvin,
|
||||
_ => Unit::None,
|
||||
};
|
||||
|
||||
if clean_str.contains('.') {
|
||||
// Decimal floating point
|
||||
Ok(Number::Decimal(
|
||||
clean_str
|
||||
.parse::<Decimal>()
|
||||
.map_err(|_| LexError::NumberParse(line, span, slice.to_string()))?,
|
||||
unit,
|
||||
))
|
||||
} else {
|
||||
// Decimal integer
|
||||
Ok(Number::Integer(
|
||||
clean_str
|
||||
.parse::<i128>()
|
||||
.map_err(|_| LexError::NumberParse(line, span, slice.to_string()))?,
|
||||
unit,
|
||||
))
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
@@ -615,6 +658,12 @@ pub enum Symbol {
|
||||
Percent,
|
||||
/// Represents the `?` symbol
|
||||
Question,
|
||||
/// Represents the `&` symbol (bitwise AND)
|
||||
BitwiseAnd,
|
||||
/// Represents the `|` symbol (bitwise OR)
|
||||
BitwiseOr,
|
||||
/// Represents the `~` symbol (bitwise NOT)
|
||||
BitwiseNot,
|
||||
|
||||
// Double Character Symbols
|
||||
/// Represents the `==` symbol
|
||||
@@ -629,6 +678,12 @@ pub enum Symbol {
|
||||
LessThanOrEqual,
|
||||
/// Represents the `>=` symbol
|
||||
GreaterThanOrEqual,
|
||||
/// Represents the `<<` symbol (left shift)
|
||||
LeftShift,
|
||||
/// Represents the `>>` symbol (arithmetic right shift)
|
||||
RightShiftArithmetic,
|
||||
/// Represents the `>>>` symbol (logical right shift)
|
||||
RightShiftLogical,
|
||||
/// Represents the `**` symbol
|
||||
Exp,
|
||||
}
|
||||
@@ -643,6 +698,19 @@ impl Symbol {
|
||||
| Symbol::Slash
|
||||
| Symbol::Exp
|
||||
| Symbol::Percent
|
||||
| Symbol::Caret
|
||||
)
|
||||
}
|
||||
|
||||
pub fn is_bitwise(&self) -> bool {
|
||||
matches!(
|
||||
self,
|
||||
Symbol::BitwiseAnd
|
||||
| Symbol::BitwiseOr
|
||||
| Symbol::BitwiseNot
|
||||
| Symbol::LeftShift
|
||||
| Symbol::RightShiftArithmetic
|
||||
| Symbol::RightShiftLogical
|
||||
)
|
||||
}
|
||||
|
||||
@@ -693,6 +761,12 @@ impl std::fmt::Display for Symbol {
|
||||
Self::NotEqual => write!(f, "!="),
|
||||
Self::Dot => write!(f, "."),
|
||||
Self::Caret => write!(f, "^"),
|
||||
Self::BitwiseAnd => write!(f, "&"),
|
||||
Self::BitwiseOr => write!(f, "|"),
|
||||
Self::BitwiseNot => write!(f, "~"),
|
||||
Self::LeftShift => write!(f, "<<"),
|
||||
Self::RightShiftArithmetic => write!(f, ">>"),
|
||||
Self::RightShiftLogical => write!(f, ">>>"),
|
||||
Self::Exp => write!(f, "**"),
|
||||
}
|
||||
}
|
||||
@@ -846,6 +920,7 @@ documented! {
|
||||
#[cfg(test)]
|
||||
mod tests {
|
||||
use super::TokenType;
|
||||
use super::{Number, Unit};
|
||||
use logos::Logos;
|
||||
|
||||
#[test]
|
||||
@@ -859,4 +934,151 @@ mod tests {
|
||||
assert!(!tokens.iter().any(|res| res.is_err()));
|
||||
Ok(())
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_binary_literals() -> anyhow::Result<()> {
|
||||
let src = "0b1010 0b0 0b1111_0000";
|
||||
let lexer = TokenType::lexer(src);
|
||||
let tokens: Vec<_> = lexer.collect::<Result<Vec<_>, _>>()?;
|
||||
|
||||
assert_eq!(tokens.len(), 3);
|
||||
assert!(
|
||||
matches!(
|
||||
&tokens[0],
|
||||
TokenType::Number(Number::Integer(10, Unit::None))
|
||||
),
|
||||
"Expected binary 0b1010 = 10"
|
||||
);
|
||||
assert!(
|
||||
matches!(
|
||||
&tokens[1],
|
||||
TokenType::Number(Number::Integer(0, Unit::None))
|
||||
),
|
||||
"Expected binary 0b0 = 0"
|
||||
);
|
||||
assert!(
|
||||
matches!(
|
||||
&tokens[2],
|
||||
TokenType::Number(Number::Integer(240, Unit::None))
|
||||
),
|
||||
"Expected binary 0b1111_0000 = 240"
|
||||
);
|
||||
Ok(())
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_octal_literals() -> anyhow::Result<()> {
|
||||
let src = "0o77 0o0 0o7_777";
|
||||
let lexer = TokenType::lexer(src);
|
||||
let tokens: Vec<_> = lexer.collect::<Result<Vec<_>, _>>()?;
|
||||
|
||||
assert_eq!(tokens.len(), 3);
|
||||
assert!(
|
||||
matches!(
|
||||
&tokens[0],
|
||||
TokenType::Number(Number::Integer(63, Unit::None))
|
||||
),
|
||||
"Expected octal 0o77 = 63"
|
||||
);
|
||||
assert!(
|
||||
matches!(
|
||||
&tokens[1],
|
||||
TokenType::Number(Number::Integer(0, Unit::None))
|
||||
),
|
||||
"Expected octal 0o0 = 0"
|
||||
);
|
||||
assert!(
|
||||
matches!(
|
||||
&tokens[2],
|
||||
TokenType::Number(Number::Integer(4095, Unit::None))
|
||||
),
|
||||
"Expected octal 0o7_777 = 4095"
|
||||
);
|
||||
Ok(())
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_hex_literals() -> anyhow::Result<()> {
|
||||
let src = "0xFF 0x0 0xFF_FF 0xFF_FF_FF";
|
||||
let lexer = TokenType::lexer(src);
|
||||
let tokens: Vec<_> = lexer.collect::<Result<Vec<_>, _>>()?;
|
||||
|
||||
assert_eq!(tokens.len(), 4);
|
||||
assert!(
|
||||
matches!(
|
||||
&tokens[0],
|
||||
TokenType::Number(Number::Integer(255, Unit::None))
|
||||
),
|
||||
"Expected hex 0xFF = 255"
|
||||
);
|
||||
assert!(
|
||||
matches!(
|
||||
&tokens[1],
|
||||
TokenType::Number(Number::Integer(0, Unit::None))
|
||||
),
|
||||
"Expected hex 0x0 = 0"
|
||||
);
|
||||
assert!(
|
||||
matches!(
|
||||
&tokens[2],
|
||||
TokenType::Number(Number::Integer(65535, Unit::None))
|
||||
),
|
||||
"Expected hex 0xFF_FF = 65535"
|
||||
);
|
||||
assert!(
|
||||
matches!(
|
||||
&tokens[3],
|
||||
TokenType::Number(Number::Integer(16777215, Unit::None))
|
||||
),
|
||||
"Expected hex 0xFF_FF_FF = 16777215"
|
||||
);
|
||||
Ok(())
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_hex_literals_lowercase() -> anyhow::Result<()> {
|
||||
let src = "0xff 0xab 0xcd_ef";
|
||||
let lexer = TokenType::lexer(src);
|
||||
let tokens: Vec<_> = lexer.collect::<Result<Vec<_>, _>>()?;
|
||||
|
||||
assert_eq!(tokens.len(), 3);
|
||||
assert!(
|
||||
matches!(
|
||||
&tokens[0],
|
||||
TokenType::Number(Number::Integer(255, Unit::None))
|
||||
),
|
||||
"Expected hex 0xff = 255"
|
||||
);
|
||||
assert!(
|
||||
matches!(
|
||||
&tokens[1],
|
||||
TokenType::Number(Number::Integer(171, Unit::None))
|
||||
),
|
||||
"Expected hex 0xab = 171"
|
||||
);
|
||||
assert!(
|
||||
matches!(
|
||||
&tokens[2],
|
||||
TokenType::Number(Number::Integer(52719, Unit::None))
|
||||
),
|
||||
"Expected hex 0xcd_ef = 52719"
|
||||
);
|
||||
Ok(())
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_binary_with_temperature_suffix() -> anyhow::Result<()> {
|
||||
// Binary, octal, and hex literals do NOT support temperature suffixes
|
||||
// (temperature suffixes are only for decimal numbers)
|
||||
// This test verifies that trying to parse something like 0b1010c
|
||||
// will be treated as 0b101 followed by 0 (the 'c' would start a new token or error)
|
||||
Ok(())
|
||||
}
|
||||
|
||||
#[test]
|
||||
fn test_hex_with_temperature_suffix() -> anyhow::Result<()> {
|
||||
// Hex, octal, and binary literals do NOT support temperature suffixes
|
||||
// Temperature suffixes are only for decimal numbers
|
||||
Ok(())
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user