| use crate::value::Complex; |
|
|
| #[derive(Debug, PartialEq, Eq)] |
| pub struct Unit { |
| |
| pub length: i32, |
| |
| pub mass: i32, |
| |
| pub time: i32, |
| } |
|
|
| impl Default for Unit { |
| fn default() -> Self { |
| Self::BASE_UNIT |
| } |
| } |
|
|
| impl Unit { |
| pub const BASE_UNIT: Unit = Unit { length: 0, mass: 0, time: 0 }; |
|
|
| pub const LENGTH: Unit = Unit { length: 1, mass: 0, time: 0 }; |
| pub const MASS: Unit = Unit { length: 0, mass: 1, time: 0 }; |
| pub const TIME: Unit = Unit { length: 0, mass: 0, time: 1 }; |
|
|
| pub const VELOCITY: Unit = Unit { length: 1, mass: 0, time: -1 }; |
| pub const ACCELERATION: Unit = Unit { length: 1, mass: 0, time: -2 }; |
|
|
| pub const FORCE: Unit = Unit { length: 1, mass: 1, time: -2 }; |
|
|
| pub fn base_unit() -> Self { |
| Self::BASE_UNIT |
| } |
|
|
| pub fn is_base(&self) -> bool { |
| *self == Self::BASE_UNIT |
| } |
| } |
|
|
| #[derive(Debug, PartialEq)] |
| pub enum Literal { |
| Float(f64), |
| Complex(Complex), |
| } |
|
|
| impl From<f64> for Literal { |
| fn from(value: f64) -> Self { |
| Self::Float(value) |
| } |
| } |
|
|
| #[derive(Debug, Clone, Copy, PartialEq)] |
| pub enum BinaryOp { |
| Add, |
| Sub, |
| Mul, |
| Div, |
| Pow, |
| } |
|
|
| #[derive(Debug, PartialEq, Clone, Copy)] |
| pub enum UnaryOp { |
| Neg, |
| Sqrt, |
| Fac, |
| } |
|
|
| #[derive(Debug, PartialEq)] |
| pub enum Node { |
| Lit(Literal), |
| Var(String), |
| FnCall { name: String, expr: Vec<Node> }, |
| BinOp { lhs: Box<Node>, op: BinaryOp, rhs: Box<Node> }, |
| UnaryOp { expr: Box<Node>, op: UnaryOp }, |
| } |
|
|
