use flycatcher_lexer::Token;
#[derive(Clone, Copy, Debug, PartialEq)]
pub enum Opcode {
EqualsEquals,
Equals,
Greater,
Less,
GreaterEquals,
LessEquals,
Add,
Subtract,
Multiply,
Divide,
}
impl Opcode {
pub fn precedence(&self) -> usize {
match self {
Opcode::Equals => 0,
Opcode::EqualsEquals => 1,
Opcode::Greater => 1,
Opcode::GreaterEquals => 1,
Opcode::Less => 1,
Opcode::LessEquals => 1,
Opcode::Add => 2,
Opcode::Subtract => 2,
Opcode::Multiply => 3,
Opcode::Divide => 3
}
}
}
pub fn is_operator(tok: Token) -> bool {
match tok {
Token::Equals |
Token::EqualsEquals |
Token::GreaterThan |
Token::GreaterThanOrEqual |
Token::LessThan |
Token::LessThanOrEqual |
Token::Plus |
Token::Dash |
Token::Star |
Token::Slash => true,
_ => false
}
}
pub fn get_operator(tok: Token) -> Option<Opcode> {
Some(match tok {
Token::Equals => Opcode::Equals,
Token::EqualsEquals => Opcode::EqualsEquals,
Token::GreaterThan => Opcode::Greater,
Token::LessThan => Opcode::Less,
Token::GreaterThanOrEqual => Opcode::GreaterEquals,
Token::LessThanOrEqual => Opcode::LessEquals,
Token::Plus => Opcode::Add,
Token::Dash => Opcode::Subtract,
Token::Star => Opcode::Multiply,
Token::Slash => Opcode::Divide,
_ => return None
})
}